py: Allocate memory for assembled code at start of PASS_EMIT.

Previously was allocating at end of PASS_COMPUTE, and this pass was
being run twice, so memory was being allocated twice.
This commit is contained in:
Damien George 2015-01-14 00:38:33 +00:00
parent bc47c287df
commit d9dc6fff21
4 changed files with 32 additions and 55 deletions

View File

@ -70,20 +70,20 @@ void asm_arm_free(asm_arm_t *as, bool free_code) {
} }
void asm_arm_start_pass(asm_arm_t *as, uint pass) { void asm_arm_start_pass(asm_arm_t *as, uint pass) {
as->pass = pass;
as->code_offset = 0;
if (pass == ASM_ARM_PASS_COMPUTE) { if (pass == ASM_ARM_PASS_COMPUTE) {
memset(as->label_offsets, -1, as->max_num_labels * sizeof(mp_uint_t)); memset(as->label_offsets, -1, as->max_num_labels * sizeof(mp_uint_t));
} else if (pass == ASM_ARM_PASS_EMIT) {
MP_PLAT_ALLOC_EXEC(as->code_offset, (void**)&as->code_base, &as->code_size);
if (as->code_base == NULL) {
assert(0);
}
} }
as->pass = pass;
as->code_offset = 0;
} }
void asm_arm_end_pass(asm_arm_t *as) { void asm_arm_end_pass(asm_arm_t *as) {
if (as->pass == ASM_ARM_PASS_COMPUTE) { if (as->pass == ASM_ARM_PASS_EMIT) {
MP_PLAT_ALLOC_EXEC(as->code_offset, (void**) &as->code_base, &as->code_size);
if(as->code_base == NULL) {
assert(0);
}
} else if(as->pass == ASM_ARM_PASS_EMIT) {
#ifdef __arm__ #ifdef __arm__
// flush I- and D-cache // flush I- and D-cache
asm volatile( asm volatile(

View File

@ -73,32 +73,21 @@ void asm_thumb_free(asm_thumb_t *as, bool free_code) {
} }
void asm_thumb_start_pass(asm_thumb_t *as, uint pass) { void asm_thumb_start_pass(asm_thumb_t *as, uint pass) {
as->pass = pass;
as->code_offset = 0;
if (pass == ASM_THUMB_PASS_COMPUTE) { if (pass == ASM_THUMB_PASS_COMPUTE) {
memset(as->label_offsets, -1, as->max_num_labels * sizeof(mp_uint_t)); memset(as->label_offsets, -1, as->max_num_labels * sizeof(mp_uint_t));
} } else if (pass == ASM_THUMB_PASS_EMIT) {
} MP_PLAT_ALLOC_EXEC(as->code_offset, (void**)&as->code_base, &as->code_size);
if (as->code_base == NULL) {
void asm_thumb_end_pass(asm_thumb_t *as) {
if (as->pass == ASM_THUMB_PASS_COMPUTE) {
MP_PLAT_ALLOC_EXEC(as->code_offset, (void**) &as->code_base, &as->code_size);
if(as->code_base == NULL) {
assert(0); assert(0);
} }
//printf("code_size: %u\n", as->code_size); //printf("code_size: %u\n", as->code_size);
} }
as->pass = pass;
as->code_offset = 0;
}
/* void asm_thumb_end_pass(asm_thumb_t *as) {
// check labels are resolved // could check labels are resolved...
if (as->label != NULL)
{
int i;
for (i = 0; i < as->label->len; ++i)
if (g_array_index(as->label, Label, i).unresolved != NULL)
return false;
}
*/
} }
// all functions must go through this one to emit bytes // all functions must go through this one to emit bytes

View File

@ -143,33 +143,22 @@ void asm_x64_free(asm_x64_t *as, bool free_code) {
} }
void asm_x64_start_pass(asm_x64_t *as, uint pass) { void asm_x64_start_pass(asm_x64_t *as, uint pass) {
as->pass = pass;
as->code_offset = 0;
if (pass == ASM_X64_PASS_COMPUTE) { if (pass == ASM_X64_PASS_COMPUTE) {
// reset all labels // reset all labels
memset(as->label_offsets, -1, as->max_num_labels * sizeof(mp_uint_t)); memset(as->label_offsets, -1, as->max_num_labels * sizeof(mp_uint_t));
} } if (pass == ASM_X64_PASS_EMIT) {
} MP_PLAT_ALLOC_EXEC(as->code_offset, (void**)&as->code_base, &as->code_size);
if (as->code_base == NULL) {
void asm_x64_end_pass(asm_x64_t *as) {
if (as->pass == ASM_X64_PASS_COMPUTE) {
MP_PLAT_ALLOC_EXEC(as->code_offset, (void**) &as->code_base, &as->code_size);
if(as->code_base == NULL) {
assert(0); assert(0);
} }
//printf("code_size: %u\n", as->code_size); //printf("code_size: %u\n", as->code_size);
} }
as->pass = pass;
as->code_offset = 0;
}
/* void asm_x64_end_pass(asm_x64_t *as) {
// check labels are resolved // could check labels are resolved...
if (as->label != NULL)
{
int i;
for (i = 0; i < as->label->len; ++i)
if (g_array_index(as->label, Label, i).unresolved != NULL)
return false;
}
*/
} }
// all functions must go through this one to emit bytes // all functions must go through this one to emit bytes

View File

@ -131,21 +131,20 @@ void asm_x86_free(asm_x86_t *as, bool free_code) {
} }
void asm_x86_start_pass(asm_x86_t *as, mp_uint_t pass) { void asm_x86_start_pass(asm_x86_t *as, mp_uint_t pass) {
as->pass = pass;
as->code_offset = 0;
if (pass == ASM_X86_PASS_COMPUTE) { if (pass == ASM_X86_PASS_COMPUTE) {
// reset all labels // reset all labels
memset(as->label_offsets, -1, as->max_num_labels * sizeof(mp_uint_t)); memset(as->label_offsets, -1, as->max_num_labels * sizeof(mp_uint_t));
} } else if (pass == ASM_X86_PASS_EMIT) {
} MP_PLAT_ALLOC_EXEC(as->code_offset, (void**)&as->code_base, &as->code_size);
if (as->code_base == NULL) {
void asm_x86_end_pass(asm_x86_t *as) {
if (as->pass == ASM_X86_PASS_COMPUTE) {
MP_PLAT_ALLOC_EXEC(as->code_offset, (void**) &as->code_base, &as->code_size);
if(as->code_base == NULL) {
assert(0); assert(0);
} }
} }
as->pass = pass;
as->code_offset = 0;
}
void asm_x86_end_pass(asm_x86_t *as) {
} }
// all functions must go through this one to emit bytes // all functions must go through this one to emit bytes