native/x64: 完善汇编解析和指令编码

Author: chai2010

internal/native/parser/parser_file.go

@@ -16,6 +16,35 @@ func (p *parser) parseFile() {
 	// 解析开头的关联文档
 	p.prog.Doc = p.parseCommentGroup(true)
 
+	// X64 强制采用 intel 语法
+	if p.cpu == abi.X64Unix || p.cpu == abi.X64Windows {
+		for {
+			if p.err != nil {
+				return
+			}
+			if p.tok == token.EOF {
+				break
+			}
+
+			switch p.tok {
+			case token.COMMENT:
+				commentObj := p.parseCommentGroup(true)
+				p.prog.Comments = append(p.prog.Comments, commentObj)
+				p.prog.Objects = append(p.prog.Objects, commentObj)
+
+			case token.GAS_X64_INTEL_SYNTAX:
+				p.prog.IntelSyntax = &ast.GasIntelSyntaxNoprefix{
+					Pos: p.pos,
+				}
+				p.acceptToken(token.GAS_X64_INTEL_SYNTAX)
+				p.acceptToken(token.GAS_X64_NOPREFIX)
+			}
+		}
+		if p.prog.IntelSyntax == nil {
+			p.errorf(p.pos, "%s missing", token.GAS_X64_INTEL_SYNTAX)
+		}
+	}
+
 	// 解析代码主体
 	for {
 		if p.err != nil {
@@ -31,17 +60,6 @@ func (p *parser) parseFile() {
 			p.prog.Comments = append(p.prog.Comments, commentObj)
 			p.prog.Objects = append(p.prog.Objects, commentObj)
 
-		case token.GAS_X64_INTEL_SYNTAX:
-			if p.cpu == abi.X64Unix || p.cpu == abi.X64Windows {
-				p.prog.IntelSyntax = &ast.GasIntelSyntaxNoprefix{
-					Pos: p.pos,
-				}
-				p.acceptToken(token.GAS_X64_INTEL_SYNTAX)
-				p.acceptToken(token.GAS_X64_NOPREFIX)
-			} else {
-				p.errorf(p.pos, "%v only enabled on X64 CPU", p.tok)
-			}
-
 		case token.GAS_EXTERN:
 			ext := &ast.Extern{Pos: p.pos, Tok: token.GAS_EXTERN}
 

internal/native/x64/lookup.go

@@ -14,6 +14,26 @@ func RegValid(reg abi.RegType) bool {
 	return reg > 0 && reg < REG_END
 }
 
+func RegValid_Int(reg abi.RegType) bool {
+	return RegValid_I8(reg) || RegValid_I16(reg) || RegValid_I32(reg) || RegValid_I64(64)
+}
+func RegValid_I8(reg abi.RegType) bool {
+	return (reg >= REG_AL && reg <= REG_R15B) || (reg >= REG_AH && reg <= REG_BH)
+}
+func RegValid_I16(reg abi.RegType) bool {
+	return reg >= REG_AX && reg <= REG_R15W
+}
+func RegValid_I32(reg abi.RegType) bool {
+	return reg >= REG_EAX && reg <= REG_R15D
+}
+func RegValid_I64(reg abi.RegType) bool {
+	return reg >= REG_RAX && reg <= REG_R15
+}
+
+func RegValid_Float(reg abi.RegType) bool {
+	return reg >= REG_XMM0 && reg <= REG_XMM7
+}
+
 // 寄存器宽度
 func RegXLen(r abi.RegType) int {
 	switch {

internal/native/x64/prog.go

@@ -51,10 +51,18 @@ func (p *Prog) xLen(arg *abi.X64Argument) int {
 
 func (p *Prog) reg2p9Reg(r abi.RegType) int16 {
 	switch {
+	case r == REG_RIP:
+		return 0 // plan9 需要独处理
+	case r >= REG_EAX && r <= REG_R15B:
+		return p9x86.REG_AL + int16(r-REG_AL)
 	case r >= REG_EAX && r <= REG_R15D:
 		return p9x86.REG_AX + int16(r-REG_EAX)
 	case r >= REG_RAX && r <= REG_R15:
 		return p9x86.REG_AX + int16(r-REG_RAX)
+	case r >= REG_AH && r <= REG_BH:
+		return p9x86.REG_AH + int16(r-REG_AH)
+	case r >= REG_XMM0 && r <= REG_XMM7:
+		return p9x86.REG_F0 + int16(r-REG_XMM0)
 	default:
 		panic("unreachable")
 	}