Bool PrsAsmImm(CCmpCtrl *cc,CAsmArg *arg) { if (arg->imm_or_off_present) LexExcept(cc,"Already one immediate at "); arg->imm_or_off_present=TRUE; arg->num.local_asm_undef_hash=NULL; arg->num.glbl_asm_undef_hash=NULL; cc->asm_undef_hash=NULL; cc->abs_cnts=0; cc->flags&=~(CCF_UNRESOLVED+CCF_LOCAL); if (!IsLexExpression2Bin(cc,&arg->num.machine_code)) LexSkipEol(cc); else { if (cc->abs_cnts.externs) LexExcept(cc,"Extern Not Allowed at "); if (cc->flags & CCF_UNRESOLVED) { if (cc->flags & CCF_LOCAL) { arg->num.local_asm_undef_hash=cc->asm_undef_hash; cc->asm_undef_hash=NULL; } else { arg->num.glbl_asm_undef_hash=cc->asm_undef_hash; cc->asm_undef_hash=NULL; } } else { arg->num.i=Call(arg->num.machine_code); arg->num.glbl_asm_undef_hash=cc->asm_undef_hash; cc->asm_undef_hash=NULL; Free(arg->num.machine_code); arg->num.machine_code=NULL; } } return TRUE; } U0 PrsAsmArg(CCmpCtrl *cc,CAsmArg *arg,Bool rel) { CHashGeneric *tmph,*tmph1; CHashReg *tmpr; MemSet(arg,0,sizeof(CAsmArg)); arg->seg =REG_NONE; arg->reg1=REG_NONE; arg->reg2=REG_NONE; arg->scale=1; while (TRUE) { if (cc->token==TK_IDENT) { if (tmph=cc->hash_entry) { if (tmph->type&HTG_TYPE_MASK==HTT_REG) { tmpr=tmph; arg->reg1_type=tmpr->reg_type; switch (tmpr->reg_type) { start: case REGT_R8: arg->size=1; break; case REGT_R16: arg->size=2; break; case REGT_R32: arg->size=4; break; case REGT_R64: arg->size=8; break; end: arg->reg1=tmpr->reg_num; Lex(cc); return; case REGT_SEG: arg->seg=tmpr->reg_num; if (Lex(cc)!=':') { arg->just_seg=TRUE; return; } else Lex(cc); //skip ":" break; case REGT_FSTK: case REGT_MM: case REGT_XMM: arg->size=8; arg->reg1=tmpr->reg_num; Lex(cc); return; } } else { if ((tmph->type&HTG_TYPE_MASK==HTT_CLASS|| tmph->type&HTG_TYPE_MASK==HTT_INTERNAL_TYPE) && (tmph1=HashFind(cc->cur_str,cmp.asm_hash,HTT_ASM_KEYWORD))) tmph=tmph1; if (tmph->type&HTG_TYPE_MASK==HTT_ASM_KEYWORD) { switch (tmph->user_data0) { case AKW_I8: case AKW_U8: arg->size=1; break; case AKW_I16: case AKW_U16: arg->size=2; break; case AKW_I32: case AKW_U32: arg->size=4; break; case AKW_I64: case AKW_U64: arg->size=8; break; default: LexExcept(cc,"syntax error at "); } Lex(cc); //skip keyword } else goto pa_asm_direct_imm; } } else { pa_asm_direct_imm: PrsAsmImm(cc,arg); arg->num.abs_cnts=cc->abs_cnts; if (arg->size<=1 && !rel && arg->num.abs_cnts&1) { if (cc->aotc->seg_size==16) arg->size=2; else arg->size=4; } if (cc->token!='[') return; } } else if (cc->token=='[') { arg->indirect=TRUE; Lex(cc); // skip [ while (cc->token && cc->token!=']') { if (cc->token==TK_IDENT) { if (tmph=cc->hash_entry) { if (tmph->type&HTG_TYPE_MASK==HTT_REG && REGT_R16<=tmph(CHashReg *)->reg_type<=REGT_R64) { tmpr=tmph; arg->reg2_type=tmpr->reg_type; if (arg->reg1==REG_NONE) { if (tmpr->reg_num&7==REG_RSP) { arg->reg1=4; arg->reg2=tmpr->reg_num; } else arg->reg1=tmpr->reg_num; } else arg->reg2=tmpr->reg_num; Lex(cc); } else goto pa_asm_indirect_imm; } else goto pa_asm_indirect_imm; } else if (cc->token=='*') { Lex(cc); if (cc->token!=TK_I64) LexExcept(cc,"Expecting scale factor at "); arg->scale=cc->cur_i64; Lex(cc); //skip scale if (arg->reg2!=REG_NONE) { SwapI64(&arg->reg1,&arg->reg2); SwapI64(&arg->reg1_type,&arg->reg2_type); } } else if (cc->token=='+') { Lex(cc); //skip '+' } else { pa_asm_indirect_imm: PrsAsmImm(cc,arg); arg->num.abs_cnts=cc->abs_cnts; } } if (cc->token!=']') LexExcept(cc,"Missing ']' at "); Lex(cc); //skip ] return; } else goto pa_asm_direct_imm; } } I64 AsmMakeArgMask(CCmpCtrl *cc,CAsmArg *arg) { CAOTCtrl *aotc=cc->aotc; I64 res; if (arg->just_seg) { switch (arg->seg) { case 0: res=1<<ARGT_ES|1<<ARGT_SREG; break; case 1: res=1<<ARGT_CS|1<<ARGT_SREG; break; case 2: res=1<<ARGT_SS|1<<ARGT_SREG; break; case 3: res=1<<ARGT_DS|1<<ARGT_SREG; break; case 4: res=1<<ARGT_FS|1<<ARGT_SREG; break; case 5: res=1<<ARGT_GS|1<<ARGT_SREG; break; } goto mm_done; } if (arg->reg1_type==REGT_FSTK) { if (arg->reg1) res=1<<ARGT_STI; else res=1<<ARGT_ST0|1<<ARGT_STI; goto mm_done; } res=cmp.size_arg_mask[arg->size]; if (aotc->seg_size==64) res&=0xFF0FFFFFFF; if (arg->reg1!=REG_NONE && arg->imm_or_off_present && !arg->num.i && !arg->num.glbl_asm_undef_hash && !arg->num.local_asm_undef_hash) arg->imm_or_off_present=FALSE; //Zero displacement if (arg->reg2!=REG_NONE || arg->scale!=1) { res&=0x0000FF0000; goto mm_done; } if (arg->indirect) { if (arg->imm_or_off_present) res&=0x00FFFF0000; else res&=0x000FFF0000; } else { if (arg->imm_or_off_present) res&=0x000F000FFE; else res&=0x3F0FFFF000; } if (arg->seg!=REG_NONE) res&=0x00FFFF0000; if (arg->reg1==REG_NONE) { if (arg->indirect) res&=0x00FFFF0000; else if (arg->num.i<0) { if (arg->num.i>=I8_MIN) res&=0x8FE; else if (arg->num.i>=I16_MIN) res&=0x8EE; else if (arg->num.i>=I32_MIN) res&=0x8CE; else res&=0x88E; } else { if (arg->num.i<=I8_MAX) res&=0xFFE; else if (arg->num.i<=U8_MAX) res&=0xFEE; else if (arg->num.i<=I16_MAX) res&=0xEEE; else if (arg->num.i<=U16_MAX) res&=0xECE; else if (arg->num.i<=I32_MAX) res&=0xCCE; else if (arg->num.i<=U32_MAX) res&=0xC8E; else res&=0x88E; } } else { res&= 0x3F00FFF000; if (!arg->indirect) //M8-M64 res&=0xFFFF0FFFFF; } switch (arg->reg1) { case REG_RAX: res&=~0x3000000000; break; case REG_RCX: res&=~0x2F00000000; break; case REG_RDX: res&=~0x1F00000000; break; default: res&=~0x3F00000000; } mm_done: return res; } Bool AsmStoreNum(CCmpCtrl *cc,CAsmNum2 *num2,I64 cnt,Bool U8_avail) { CAOTCtrl *aotc=cc->aotc; I64 i; CAOTAbsAddr *tmpa; if (!num2->imm_flag) num2->num.i-=num2->rel; for (i=0;i<cnt;i++) { if (num2->U8_cnt==1) { if (num2->num.local_asm_undef_hash||num2->num.glbl_asm_undef_hash) AsmUnresolvedAdd(cc,num2->num.machine_code,IET_REL_I8+num2->imm_flag, aotc->rip,num2->rel,num2->num.local_asm_undef_hash, num2->num.glbl_asm_undef_hash,cc->lex_include_stk->line_num, U8_avail); else if (!num2->imm_flag && !(I8_MIN<=num2->num.i<=I8_MAX)) LexExcept(cc,"Branch out of range at "); if (num2->imm_flag) { if (num2->num.abs_cnts.abs_addres&1) { tmpa=CAlloc(sizeof(CAOTAbsAddr)); tmpa->next=aotc->abss; aotc->abss=tmpa; tmpa->rip=aotc->rip; tmpa->type=AAT_ADD_U8; } } else { if (num2->num.abs_cnts.c_addres&1) { tmpa=CAlloc(sizeof(CAOTAbsAddr)); tmpa->next=aotc->abss; aotc->abss=tmpa; tmpa->rip=aotc->rip; tmpa->type=AAT_SUB_U8; } } AOTStoreCodeU8(cc,num2->num.i); } else { if (num2->U8_cnt==2) { if (num2->num.local_asm_undef_hash||num2->num.glbl_asm_undef_hash) AsmUnresolvedAdd(cc,num2->num.machine_code,IET_REL_I16+num2->imm_flag, aotc->rip,num2->rel,num2->num.local_asm_undef_hash, num2->num.glbl_asm_undef_hash,cc->lex_include_stk->line_num, U8_avail); else if (!num2->imm_flag && !(I16_MIN<=num2->num.i<=I16_MAX)) LexExcept(cc,"Branch out of range at "); if (num2->imm_flag) { if (num2->num.abs_cnts.abs_addres&1) { tmpa=CAlloc(sizeof(CAOTAbsAddr)); tmpa->next=aotc->abss; aotc->abss=tmpa; tmpa->rip=aotc->rip; tmpa->type=AAT_ADD_U16; } } else { if (num2->num.abs_cnts.c_addres&1) { tmpa=CAlloc(sizeof(CAOTAbsAddr)); tmpa->next=aotc->abss; aotc->abss=tmpa; tmpa->rip=aotc->rip; tmpa->type=AAT_SUB_U16; } } AOTStoreCodeU8(cc,num2->num.i.u8[0]); AOTStoreCodeU8(cc,num2->num.i.u8[1]); } else if (num2->U8_cnt==4) { if (num2->num.local_asm_undef_hash||num2->num.glbl_asm_undef_hash) AsmUnresolvedAdd(cc,num2->num.machine_code,IET_REL_I32+num2->imm_flag, aotc->rip,num2->rel,num2->num.local_asm_undef_hash, num2->num.glbl_asm_undef_hash,cc->lex_include_stk->line_num, U8_avail); else if (!num2->imm_flag && !(I32_MIN<=num2->num.i<=I32_MAX)) LexExcept(cc,"Branch out of range at "); if (num2->imm_flag) { if (num2->num.abs_cnts.abs_addres&1) { tmpa=CAlloc(sizeof(CAOTAbsAddr)); tmpa->next=aotc->abss; aotc->abss=tmpa; tmpa->rip=aotc->rip; tmpa->type=AAT_ADD_U32; } } else { if (num2->num.abs_cnts.c_addres&1) { tmpa=CAlloc(sizeof(CAOTAbsAddr)); tmpa->next=aotc->abss; aotc->abss=tmpa; tmpa->rip=aotc->rip; tmpa->type=AAT_SUB_U32; } } AOTStoreCodeU32(cc,num2->num.i); } else if (num2->U8_cnt==8) { if (num2->num.local_asm_undef_hash||num2->num.glbl_asm_undef_hash) AsmUnresolvedAdd(cc,num2->num.machine_code,IET_REL_I64+num2->imm_flag, aotc->rip,num2->rel,num2->num.local_asm_undef_hash, num2->num.glbl_asm_undef_hash,cc->lex_include_stk->line_num, U8_avail); if (num2->imm_flag) { if (num2->num.abs_cnts.abs_addres&1) { tmpa=CAlloc(sizeof(CAOTAbsAddr)); tmpa->next=aotc->abss; aotc->abss=tmpa; tmpa->rip=aotc->rip; tmpa->type=AAT_ADD_U64; } } else { if (num2->num.abs_cnts.c_addres&1) { tmpa=CAlloc(sizeof(CAOTAbsAddr)); tmpa->next=aotc->abss; aotc->abss=tmpa; tmpa->rip=aotc->rip; tmpa->type=AAT_SUB_U64; } } AOTStoreCodeU64(cc,num2->num.i); } if (U8_avail && !num2->num.local_asm_undef_hash && !num2->num.glbl_asm_undef_hash && !num2->imm_flag && -124<=num2->num.i<=123) { LexWarn(cc,"could use I8 displacement at "); return FALSE; } } } return TRUE; } U8 asm_seg_prefixes[6]={0x26,0x2E,0x36,0x3E,0x64,0x65}; Bool PrsAsmInst(CCmpCtrl *cc,CHashOpcode *tmpo,I64 argcnt) { CAOTCtrl *aotc=cc->aotc; I64 i,j,arg1,arg2,om,seg,arg1mask,arg2mask; CAsmArg *tmpa1,*tmpa2; Bool ModrM_complete,U8_avail=FALSE,found_second_possible=FALSE; CInst *tmpins; CAsmIns cur,best; best.U8_cnt=255; if (argcnt>0) arg1mask=AsmMakeArgMask(cc,&aotc->arg1); else arg1mask=1; if (argcnt>1) arg2mask=AsmMakeArgMask(cc,&aotc->arg2); else arg2mask=1; for (i=0;i<tmpo->inst_entry_cnt;i++) { tmpins=&tmpo->ins[i]; if (tmpins->arg1==ARGT_REL8 || tmpins->arg2==ARGT_REL8) U8_avail=TRUE; if (Bt(&arg1mask,tmpins->arg1) && Bt(&arg2mask,tmpins->arg2) && (!(tmpins->flags&IEF_NOT_IN_64_BIT) || aotc->seg_size!=64)) { MemSet(&cur,0,sizeof(CAsmIns)); cur.tmpins=tmpins; ModrM_complete=FALSE; cur.is_dft=ToBool(tmpins->flags & IEF_DFT); if (aotc->seg_size==64) { if (tmpins->flags & IEF_48_REX) cur.REX=0x48; else cur.REX=0x40; } cur.disp.imm_flag=TRUE; cur.imm.imm_flag=TRUE; om=tmpins->opcode_modifier; arg1=tmpins->arg1; arg2=tmpins->arg2; tmpa1=&aotc->arg1; tmpa2=&aotc->arg2; cur.last_opcode_U8=tmpins->opcode[tmpins->opcode_cnt-1]; if (tmpins->slash_val<8) { cur.ModrM|=tmpins->slash_val<<3; cur.has_ModrM=TRUE; } if (aotc->seg_size==16 && tmpins->flags & IEF_OP_SIZE32 || aotc->seg_size!=16 && tmpins->flags & IEF_OP_SIZE16) cur.has_operand_prefix=TRUE; if (om==OM_IB) cur.imm.U8_cnt=1; else if (om==OM_IW) cur.imm.U8_cnt=2; else if (om==OM_ID) cur.imm.U8_cnt=4; if (om==OM_CB) { cur.imm.U8_cnt=1; cur.imm.imm_flag=FALSE; } else if (om==OM_CW) { cur.imm.U8_cnt=2; cur.imm.imm_flag=FALSE; } else if (om==OM_CD) { cur.imm.U8_cnt=4; cur.imm.imm_flag=FALSE; } if (argcnt==1) { if (best.U8_cnt!=255 && !found_second_possible && !best.is_dft) { found_second_possible=TRUE; if (!aotc->arg1.size) PrintWarn("no size specified at %s,%04d\n", cc->lex_include_stk->full_name, cc->lex_include_stk->line_num-1); } if (tmpins->flags & IEF_PLUS_OPCODE) { if (tmpins->slash_val==SV_R_REG) { cur.last_opcode_U8|=tmpa1->reg1&7; if (tmpa1->reg1&15>7) cur.REX|=1; if (tmpa1->reg1>=20) //RBPu8,RSPu8,RSIu8,RDIu8? cur.has_REX=TRUE; } else {//SV_I_REG if (tmpa1->reg1_type==REGT_FSTK) cur.last_opcode_U8+=tmpa1->reg1; } } if (arg1==ARGT_R64 || arg1==ARGT_RM64 || arg1==ARGT_M64) cur.REX|=8; if (ARGT_RM8<=arg1<=ARGT_RM64 || ARGT_M8<=arg1<=ARGT_M64) { if (aotc->seg_size==16) cur.has_addr_prefix=TRUE; cur.has_ModrM=TRUE; if (tmpa1->imm_or_off_present && tmpa1->indirect && tmpa1->reg1==REG_NONE) { cur.ModrM=cur.ModrM+5; MemCpy(&cur.disp.num,&tmpa1->num,sizeof(CAsmNum)); cur.disp.U8_cnt=4; if (aotc->seg_size==64) cur.disp.imm_flag=FALSE; } else { if (tmpa1->reg2==REG_NONE && tmpa1->scale==1) { cur.ModrM|=tmpa1->reg1&7; if (tmpa1->reg1&15>7) cur.REX|=1; if (tmpa1->reg1>=20) //RBPu8,RSPu8,RSIu8,RDIu8? cur.has_REX=TRUE; } else { cur.ModrM|=4; cur.has_SIB=TRUE; if (tmpa1->scale==1) cur.SIB=0; else if (tmpa1->scale==2) cur.SIB=0x40; else if (tmpa1->scale==4) cur.SIB=0x80; else if (tmpa1->scale==8) cur.SIB=0xC0; if (tmpa1->reg2==REG_NONE) { ModrM_complete=TRUE; cur.SIB|=(tmpa1->reg1&7)<<3+REG_RBP; if (tmpa1->reg1&15>7) cur.REX|=2; if (tmpa1->reg1>=20) //RBPu8,RSPu8,RSIu8,RDIu8? cur.has_REX=TRUE; MemCpy(&cur.disp.num,&tmpa1->num,sizeof(CAsmNum)); cur.disp.U8_cnt=4; } else { cur.SIB|=(tmpa1->reg1&7)<<3+tmpa1->reg2&7; if (tmpa1->reg1&15>7) cur.REX|=2; if (tmpa1->reg1>=20) //RBPu8,RSPu8,RSIu8,RDIu8? cur.has_REX=TRUE; if (tmpa1->reg2&15>7) cur.REX|=1; if (tmpa1->reg2>=20) //RBPu8,RSPu8,RSIu8,RDIu8? cur.has_REX=TRUE; if (tmpa1->reg2&7==REG_RBP && !tmpa1->imm_or_off_present && tmpa1->indirect) { cur.ModrM|=0x40; cur.disp.U8_cnt=1; ModrM_complete=TRUE; } } } if (!ModrM_complete) { if (tmpa1->imm_or_off_present) { MemCpy(&cur.disp.num,&tmpa1->num,sizeof(CAsmNum)); if (!cur.disp.num.machine_code && I8_MIN<=cur.disp.num.i<=I8_MAX) { cur.ModrM|=0x40; cur.disp.U8_cnt=1; } else if (aotc->seg_size==16) { cur.ModrM|=0x80; cur.disp.U8_cnt=2; } else { cur.ModrM|=0x80; cur.disp.U8_cnt=4; } } else if (!tmpa1->indirect) { cur.has_addr_prefix=FALSE; cur.ModrM|=0xC0; } else { if (tmpa1->reg1&7==REG_RBP) { cur.ModrM|=0x40; cur.disp.U8_cnt=1; } } } } } else if (ARGT_REL8<=arg1<=ARGT_REL32 || ARGT_IMM8<=arg1<=ARGT_IMM64 || ARGT_UIMM8<=arg1<=ARGT_UIMM64) { if (arg1==ARGT_IMM64 || arg2==ARGT_UIMM64) cur.REX|=8; MemCpy(&cur.imm.num,&tmpa1->num,sizeof(CAsmNum)); } } else if (argcnt==2) { if (best.U8_cnt!=255 && !found_second_possible && !best.is_dft) { found_second_possible=TRUE; if (!aotc->arg1.size && !aotc->arg2.size) PrintWarn("no size specified at %s,%04d\n", cc->lex_include_stk->full_name, cc->lex_include_stk->line_num-1); } if (tmpins->flags & IEF_PLUS_OPCODE) { if (tmpins->slash_val==SV_R_REG) { if (ARGT_AL<=arg1<=ARGT_RAX) { cur.last_opcode_U8|=tmpa2->reg1&7; if (tmpa2->reg1&15>7) cur.REX|=1; if (tmpa2->reg1>=20) //RBPu8,RSPu8,RSIu8,RDIu8? cur.has_REX=TRUE; } else { cur.last_opcode_U8|=tmpa1->reg1&7; if (tmpa1->reg1&15>7) cur.REX|=1; if (tmpa1->reg1>=20) //RBPu8,RSPu8,RSIu8,RDIu8? cur.has_REX=TRUE; } } else {//SV_I_REG if (tmpa1->reg1_type==REGT_FSTK) cur.last_opcode_U8|=tmpa1->reg1; if (tmpa2->reg1_type==REGT_FSTK) cur.last_opcode_U8|=tmpa2->reg1; } } if (arg1==ARGT_RM64 || arg2==ARGT_RM64 || arg1==ARGT_M64 || arg2==ARGT_M64 || arg1==ARGT_R64 || arg2==ARGT_R64) cur.REX|=8; if (ARGT_RM8<=arg1<=ARGT_RM64 || ARGT_RM8<=arg2<=ARGT_RM64 || ARGT_M8<=arg1<=ARGT_M64 || ARGT_M8<=arg2<=ARGT_M64) { if (aotc->seg_size==16) cur.has_addr_prefix=TRUE; cur.has_ModrM=TRUE; if (ARGT_RM8<=arg2<=ARGT_RM64 || ARGT_M8<=arg2<=ARGT_M64) { tmpa1=&aotc->arg2; tmpa2=&aotc->arg1; } if (tmpins->slash_val==SV_R_REG) { if (tmpa2->just_seg) cur.ModrM|=tmpa2->seg<<3; else { if (tmpa2->reg1==REG_NONE) { cur.ModrM|=(tmpa1->reg1&7)<<3; if (tmpa1->reg1&15>7) cur.REX|=4; if (tmpa1->reg1>=20) //RBPu8,RSPu8,RSIu8,RDIu8? cur.has_REX=TRUE; } else { cur.ModrM|=(tmpa2->reg1&7)<<3; if (tmpa2->reg1&15>7) cur.REX|=4; if (tmpa2->reg1>=20) //RBPu8,RSPu8,RSIu8,RDIu8? cur.has_REX=TRUE; } } } if (tmpa1->reg2==REG_NONE && tmpa1->scale==1) { if (tmpa1->reg1!=REG_NONE) { cur.ModrM|=tmpa1->reg1&7; if (tmpa1->reg1&15>7) cur.REX|=1; if (tmpa1->reg1>=20) //RBPu8,RSPu8,RSIu8,RDIu8? cur.has_REX=TRUE; } } else { cur.ModrM|=4; cur.has_SIB=TRUE; if (tmpa1->scale==1) cur.SIB=0; else if (tmpa1->scale==2) cur.SIB=0x40; else if (tmpa1->scale==4) cur.SIB=0x80; else if (tmpa1->scale==8) cur.SIB=0xC0; if (tmpa1->reg2==REG_NONE) { ModrM_complete=TRUE; cur.SIB|=(tmpa1->reg1&7)<<3+5; if (tmpa1->reg1&15>7) cur.REX|=2; if (tmpa1->reg1>=20) //RBPu8,RSPu8,RSIu8,RDIu8? cur.has_REX=TRUE; MemCpy(&cur.disp.num,&tmpa1->num,sizeof(CAsmNum)); cur.disp.U8_cnt=4; } else { cur.SIB|=(tmpa1->reg1&7)<<3+tmpa1->reg2&7; if (tmpa1->reg1&15>7) cur.REX|=2; if (tmpa1->reg1>=20) //RBPu8,RSPu8,RSIu8,RDIu8? cur.has_REX=TRUE; if (tmpa1->reg2&15>7) cur.REX|=1; if (tmpa1->reg2>=20) //RBPu8,RSPu8,RSIu8,RDIu8? cur.has_REX=TRUE; if (tmpa1->reg2&7==REG_RBP && !tmpa1->imm_or_off_present && tmpa1->indirect) { cur.ModrM|=0x40; cur.disp.U8_cnt=1; ModrM_complete=TRUE; } } } if (!ModrM_complete) { if (tmpa1->imm_or_off_present && tmpa1->indirect && tmpa1->reg1==REG_NONE) { cur.ModrM=cur.ModrM&0xF8+5; MemCpy(&cur.disp.num,&tmpa1->num,sizeof(CAsmNum)); cur.disp.U8_cnt=4; if (aotc->seg_size==64) cur.disp.imm_flag=FALSE; } else { if (tmpa1->imm_or_off_present) { MemCpy(&cur.disp.num,&tmpa1->num,sizeof(CAsmNum)); if (!cur.disp.num.machine_code && I8_MIN<=cur.disp.num.i<=I8_MAX) { cur.ModrM|=0x40; cur.disp.U8_cnt=1; } else if (aotc->seg_size==16) { cur.ModrM|=0x80; cur.disp.U8_cnt=2; } else { cur.ModrM|=0x80; cur.disp.U8_cnt=4; } } else if (!tmpa1->indirect) { cur.has_addr_prefix=FALSE; cur.ModrM|=0xC0; } else { if (tmpa1->reg1&7==REG_RBP) { cur.ModrM|=0x40; cur.disp.U8_cnt=1; } } } } } else if (ARGT_MOFFS8<=arg1<=ARGT_MOFFS64) { MemCpy(&cur.disp.num,&tmpa1->num,sizeof(CAsmNum)); if (aotc->seg_size==16) cur.disp.U8_cnt=2; else cur.disp.U8_cnt=4; cur.has_addr_prefix=FALSE; } else if (ARGT_MOFFS8<=arg2<=ARGT_MOFFS64) { MemCpy(&cur.disp.num,&tmpa2->num,sizeof(CAsmNum)); if (aotc->seg_size==16) cur.disp.U8_cnt=2; else cur.disp.U8_cnt=4; cur.has_addr_prefix=FALSE; } else if (ARGT_IMM8<=arg1<=ARGT_IMM64 || ARGT_UIMM8<=arg1<=ARGT_UIMM64) { MemCpy(&cur.imm.num,&tmpa1->num,sizeof(CAsmNum)); if (arg1==ARGT_IMM8 || arg1==ARGT_UIMM8) cur.imm.U8_cnt=1; else if (arg1==ARGT_IMM16 || arg1==ARGT_UIMM16) cur.imm.U8_cnt=2; else if (arg1==ARGT_IMM32 || arg1==ARGT_UIMM32) cur.imm.U8_cnt=4; else { cur.imm.U8_cnt=8; cur.REX|=8; } } if (ARGT_IMM8<=arg2<=ARGT_IMM64 || ARGT_UIMM8<=arg2<=ARGT_UIMM64) { MemCpy(&cur.imm.num,&tmpa2->num,sizeof(CAsmNum)); if (arg2==ARGT_IMM8 || arg2==ARGT_UIMM8) cur.imm.U8_cnt=1; else if (arg2==ARGT_IMM16 || arg2==ARGT_UIMM16) cur.imm.U8_cnt=2; else if (arg2==ARGT_IMM32 || arg2==ARGT_UIMM32) { cur.imm.U8_cnt=4; if (tmpins->flags&IEF_REX_ONLY_R8_R15 && arg2==ARGT_UIMM32) cur.REX&=~8; } else { cur.imm.U8_cnt=8; cur.REX|=8; } } } cur.U8_cnt=tmpins->opcode_cnt+cur.disp.U8_cnt+cur.imm.U8_cnt; if (cur.has_ModrM) cur.U8_cnt++; if (cur.has_SIB) cur.U8_cnt++; if (aotc->seg_size==64 && cur.REX&0x40==0x40 && (cur.REX!=0x40 || cur.has_REX) && (cur.REX&7 || !(tmpins->flags&IEF_REX_ONLY_R8_R15 || tmpins->flags&IEF_REX_XOR_LIKE && tmpa1->reg1==tmpa2->reg1 && cur.ModrM&0xC0==0xC0))) cur.U8_cnt++; if (cur.U8_cnt<best.U8_cnt && !(tmpins->flags & IEF_DONT_SWITCH_MODES && (cur.has_addr_prefix || cur.has_operand_prefix))) MemCpy(&best,&cur,sizeof(CAsmIns)); } } if (best.U8_cnt<255) { tmpins=best.tmpins; seg=REG_NONE; if (argcnt>1 && aotc->arg2.seg!=REG_NONE && !aotc->arg2.just_seg) seg=aotc->arg2.seg; else if (argcnt>0 && aotc->arg1.seg!=REG_NONE && !aotc->arg1.just_seg) seg=aotc->arg1.seg; if (seg!=REG_NONE) AOTStoreCodeU8(cc,asm_seg_prefixes[seg]); if (best.has_operand_prefix) AOTStoreCodeU8(cc,OC_OP_SIZE_PREFIX); //Operand size override if (best.has_addr_prefix || aotc->seg_size==16 && cur.has_SIB) AOTStoreCodeU8(cc,OC_ADDR_SIZE_PREFIX); //Operand size override if (aotc->seg_size==64 && best.REX&0x40==0x40 && (best.REX!=0x40 || best.has_REX) && (best.REX&7||!(tmpins->flags&IEF_REX_ONLY_R8_R15 || tmpins->flags&IEF_REX_XOR_LIKE && tmpa1->reg1==tmpa2->reg1 && best.ModrM&0xC0==0xC0))) AOTStoreCodeU8(cc,best.REX); for (j=0;j<tmpins->opcode_cnt-1;j++) AOTStoreCodeU8(cc,tmpins->opcode[j]); AOTStoreCodeU8(cc,best.last_opcode_U8); if (best.has_ModrM) AOTStoreCodeU8(cc,best.ModrM); if (best.has_SIB) AOTStoreCodeU8(cc,best.SIB); if (best.disp.U8_cnt) { best.disp.rel=aotc->rip+best.disp.U8_cnt+best.imm.U8_cnt; if (!AsmStoreNum(cc,&best.disp,1,U8_avail)) return FALSE; } if (best.imm.U8_cnt) { best.imm.rel=aotc->rip+best.imm.U8_cnt; if (!AsmStoreNum(cc,&best.imm,1,U8_avail)) return FALSE; } if (tmpins->flags&IEF_ENDING_ZERO) //ENTER inst AOTStoreCodeU8(cc,0); return TRUE; } LexExcept(cc,"Invalid inst at "); } U0 PrsAsmDefine(CCmpCtrl *cc,I64 U8_cnt) { Bool is_dup; I64 i,dup_val; U8 *ptr; CAsmNum2 num2; num2.U8_cnt=U8_cnt; while (cc->token && cc->token!=';') { num2.num.local_asm_undef_hash=NULL; num2.num.glbl_asm_undef_hash=NULL; if (cc->token==TK_STR) { ptr=cc->cur_str; i=cc->cur_str_len-1; while (i--) AOTStoreCodeU8(cc,*ptr++); Lex(cc); //Skip Str } else { is_dup=FALSE; cc->abs_cnts=0; cc->asm_undef_hash=NULL; cc->flags&=~(CCF_UNRESOLVED+CCF_LOCAL); if (!IsLexExpression2Bin(cc,&num2.num.machine_code)) LexSkipEol(cc); else { if (cc->abs_cnts.externs) LexExcept(cc,"Extern Not Allowed at "); if (cc->flags & CCF_UNRESOLVED) { if (cc->flags & CCF_LOCAL) { num2.num.local_asm_undef_hash=cc->asm_undef_hash; cc->asm_undef_hash=NULL; } else { num2.num.glbl_asm_undef_hash=cc->asm_undef_hash; cc->asm_undef_hash=NULL; } } else { i=Call(num2.num.machine_code); Free(num2.num.machine_code); } } if (cc->token==TK_IDENT && cc->hash_entry) { if (cc->hash_entry->type & HTT_ASM_KEYWORD && cc->hash_entry->user_data0==AKW_DUP) { is_dup=TRUE; if (Lex(cc)!='(') LexExcept(cc,"Expecting '(' at "); Lex(cc); //skip ( dup_val=AsmLexExpression(cc); if (cc->token!=')') LexExcept(cc,"Expecting ')' at "); Lex(cc); //SKIP ) } } num2.rel=0; num2.imm_flag=TRUE; num2.num.abs_cnts=cc->abs_cnts; if (is_dup) { if (num2.num.local_asm_undef_hash || num2.num.glbl_asm_undef_hash) LexExcept(cc,"Undefined DUP cnt at "); num2.num.i=dup_val; AsmStoreNum(cc,&num2,i,FALSE); } else { num2.num.i=i; AsmStoreNum(cc,&num2,1,FALSE); } } if (cc->token==',') Lex(cc); } if (cc->token!=';') LexExcept(cc,"Missing ';' at"); Lex(cc); } U0 PrsBinFile(CCmpCtrl *cc) { I64 i,size; U8 *buf,*st; if (cc->token!=TK_STR) LexExcept(cc,"Expecting string at "); st=ExtDft(cc->cur_str,"BIN"); buf=FileRead(st,&size); Free(st); for (i=0;i<size;i++) AOTStoreCodeU8(cc,buf[i]); if (Lex(cc)!=';') LexExcept(cc,"Missing ';' at"); Lex(cc); } U0 PrsAsmBlk(CCmpCtrl *cc,I64 cmp_flags) { CAOTCtrl *aotc=cc->aotc; I64 i,j,k,argcnt, old_flags=cc->flags & CCF_ASM_EXPRESSIONS; CHashOpcode *tmpo; CHashExport *tmpex; U8 *next_last_label; CCodeMisc *g_lb; aotc->seg_size=64; cc->flags|=CCF_ASM_EXPRESSIONS; if (!(cmp_flags&CMPF_ONE_ASM_INS)) { if (cc->token!='{') LexExcept(cc,"Expecting '{' at "); Lex(cc); } while (cc->token && cc->token!='}') { AsmLineLst(cc); if (cc->token==TK_IDENT && cc->hash_entry) { if (cc->hash_entry->type&HTT_ASM_KEYWORD) { i=cc->hash_entry->user_data0; Lex(cc); //skip keyword switch (i) { case AKW_IMPORT: while (cc->token && cc->token!=';') { if (cc->token!=TK_IDENT) LexExcept(cc,"Expecting identifier at "); else { tmpex=NULL; tmpex=CAlloc(sizeof(CHashExport)); tmpex->str=cc->cur_str; cc->cur_str=0; tmpex->type=HTT_EXPORT_SYS_SYM|HTF_UNRESOLVED; HashAdd(tmpex,cc->htc.glbl_hash_table); tmpex->type|=HTF_IMPORT; if (Lex(cc)==',') Lex(cc); //skip ',' } } if (cc->token!=';') LexExcept(cc,"Missing ';' at"); Lex(cc); //skip ';'; break; case AKW_MODULE_ORG: if (cc->htc.local_var_lst) LexExcept(cc,"ORG not allowed in fun asm blk "); if (aotc->module_org!=INVALID_PTR) LexExcept(cc,"Just one org allowed "); if (aotc->rip) LexExcept(cc,"ORG must be at beginning "); aotc->module_org=AsmLexExpression(cc); break; case AKW_ALIGN: if (cc->htc.local_var_lst) LexExcept(cc,"ALIGN not allowed in fun asm blk "); i=AsmLexExpression(cc); j=Bsf(i); if (!i || j!=Bsr(i)) LexExcept(cc,"ALIGN must be power of two at "); if (!(cc->flags&CCF_AOT_COMPILE) && i>8) LexExcept(cc,"In JIT mode, max ALIGN is 8 "); if (j>aotc->max_align_bits) aotc->max_align_bits=j; i=CeilU64(aotc->rip,i); if (cc->token!=',') LexExcept(cc,"Expecting ',' at "); Lex(cc); k=AsmLexExpression(cc); for (j=aotc->rip;j<i;j++) AOTStoreCodeU8(cc,k); break; case AKW_DU8: PrsAsmDefine(cc,1); break; case AKW_DU16: PrsAsmDefine(cc,2); break; case AKW_DU32: PrsAsmDefine(cc,4); break; case AKW_DU64: PrsAsmDefine(cc,8); break; case AKW_BINFILE: PrsBinFile(cc); break; case AKW_LIST: aotc->lst=TRUE; break; case AKW_NOLIST: aotc->lst=FALSE; break; case AKW_USE16: aotc->seg_size=16; break; case AKW_USE32: aotc->seg_size=32; break; case AKW_USE64: aotc->seg_size=64; break; default: LexExcept(cc,"Syntax error at "); } } else if (cc->hash_entry->type & HTT_OPCODE) { tmpo=cc->hash_entry; Lex(cc); //skip opcode argcnt=0; if (tmpo->ins[0].arg1) { argcnt++; if (ARGT_REL8<=tmpo->ins[0].arg1<=ARGT_REL32) PrsAsmArg(cc,&aotc->arg1,TRUE); else PrsAsmArg(cc,&aotc->arg1,FALSE); if (tmpo->ins[0].arg2) { argcnt++; if (cc->token!=',') LexExcept(cc,"Expecting ',' at "); else { Lex(cc); //skip ',' if (ARGT_REL8<=tmpo->ins[0].arg2<=ARGT_REL32) PrsAsmArg(cc,&aotc->arg2,TRUE); else PrsAsmArg(cc,&aotc->arg2,FALSE); } } } PrsAsmInst(cc,tmpo,argcnt); } else if (cc->hash_entry->type & HTT_EXPORT_SYS_SYM) { if (Btr(&cc->hash_entry->type,HTf_UNRESOLVED)) { if (cc->hash_entry->type & HTF_LOCAL) { cc->hash_entry(CHashExport *)->val=aotc->rip; if (Lex(cc)!=':') LexExcept(cc,"Expecting ':' at "); Lex(cc); } else { if (cc->hash_entry->type & HTF_IMPORT) LexExcept(cc,"attempt to define import at "); cc->hash_entry(CHashExport *)->val=aotc->rip; next_last_label=cc->hash_entry->str; Lex(cc); //Skip cur_str if (cc->token!=':' && cc->token!=TK_DBL_COLON) LexExcept(cc,"Expecting ':' at "); if (cc->token==TK_DBL_COLON) { cc->hash_entry->type|=HTF_EXPORT; HashSrcFileSet(cc,cc->hash_entry); AOTLocalsResolve(cc); aotc->last_label=next_last_label; } Lex(cc); } } else if (cc->hash_entry(CHashExport *)->val==aotc->rip) { Lex(cc); //Skip cur_str if (cc->token!=':' && cc->token!=TK_DBL_COLON) LexExcept(cc,"Expecting ':' at "); Lex(cc); } else LexExcept(cc,"Redefinition at "); } else LexExcept(cc,"Syntax error at "); } else if (cc->token==TK_IDENT) { tmpex=CAlloc(sizeof(CHashExport)); tmpex->str=cc->cur_str; cc->cur_str=0; tmpex->type=HTT_EXPORT_SYS_SYM; tmpex->val=aotc->rip; Lex(cc); //Skip cur_str if (cc->token!=':' && cc->token!=TK_DBL_COLON) LexExcept(cc,"Expecting ':' at "); else { if (*tmpex->str=='@' && tmpex->str[1]=='@') { if (cc->token==TK_DBL_COLON) LexExcept(cc,"No local glbl exports at "); HashAdd(tmpex,cc->htc.local_hash_table); } else HashAdd(tmpex,cc->htc.glbl_hash_table); if (cc->htc.local_var_lst) {//AsmBlk in fun? Also add goto-like label. if (!(g_lb=COCGoToLabelFind(cc,tmpex->str))) { g_lb=COCMiscNew(cc,CMT_ASM_LABEL); g_lb->str=StrNew(tmpex->str); } else if (g_lb->flags&CMF_DEFINED) LexExcept(cc,"Duplicate goto label at "); g_lb->type=CMT_ASM_LABEL; g_lb->flags|=CMF_DEFINED; g_lb->rip=aotc->rip; g_lb->use_cnt++; //Disable warning on unused labels. ICAdd(cc,IC_LABEL,g_lb,0); } if (cc->token==TK_DBL_COLON) { tmpex->type|=HTF_EXPORT; HashSrcFileSet(cc,tmpex); AOTLocalsResolve(cc); aotc->last_label=tmpex->str; } Lex(cc); } } else if (cc->token==';') Lex(cc); else LexExcept(cc,"Syntax error at "); if (cmp_flags&CMPF_ONE_ASM_INS && (cc->token!=TK_IDENT || !(tmpo=cc->hash_entry) || !(tmpo->type&(HTT_OPCODE|HTT_ASM_KEYWORD)))) break; } AOTLocalsResolve(cc); aotc->lst=FALSE; cc->flags&=cc->flags&~CCF_ASM_EXPRESSIONS|old_flags; }