GoToSocial/vendor/github.com/chenzhuoyu/iasm/x86_64/assembler.go

1804 lines
49 KiB
Go
Raw Normal View History

package x86_64
import (
`bytes`
`errors`
`fmt`
`math`
`strconv`
`strings`
`unicode`
`github.com/chenzhuoyu/iasm/expr`
)
type (
_TokenKind int
_Punctuation int
)
const (
_T_end _TokenKind = iota + 1
_T_int
_T_name
_T_punc
_T_space
)
const (
_P_plus _Punctuation = iota + 1
_P_minus
_P_star
_P_slash
_P_percent
_P_amp
_P_bar
_P_caret
_P_shl
_P_shr
_P_tilde
_P_lbrk
_P_rbrk
_P_dot
_P_comma
_P_colon
_P_dollar
_P_hash
)
var _PUNC_NAME = map[_Punctuation]string {
_P_plus : "+",
_P_minus : "-",
_P_star : "*",
_P_slash : "/",
_P_percent : "%",
_P_amp : "&",
_P_bar : "|",
_P_caret : "^",
_P_shl : "<<",
_P_shr : ">>",
_P_tilde : "~",
_P_lbrk : "(",
_P_rbrk : ")",
_P_dot : ".",
_P_comma : ",",
_P_colon : ":",
_P_dollar : "$",
_P_hash : "#",
}
func (self _Punctuation) String() string {
if v, ok := _PUNC_NAME[self]; ok {
return v
} else {
return fmt.Sprintf("_Punctuation(%d)", self)
}
}
type _Token struct {
pos int
end int
u64 uint64
str string
tag _TokenKind
}
func (self *_Token) punc() _Punctuation {
return _Punctuation(self.u64)
}
func (self *_Token) String() string {
switch self.tag {
case _T_end : return "<END>"
case _T_int : return fmt.Sprintf("<INT %d>", self.u64)
case _T_punc : return fmt.Sprintf("<PUNC %s>", _Punctuation(self.u64))
case _T_name : return fmt.Sprintf("<NAME %s>", strconv.QuoteToASCII(self.str))
case _T_space : return "<SPACE>"
default : return fmt.Sprintf("<UNK:%d %d %s>", self.tag, self.u64, strconv.QuoteToASCII(self.str))
}
}
func tokenEnd(p int, end int) _Token {
return _Token {
pos: p,
end: end,
tag: _T_end,
}
}
func tokenInt(p int, val uint64) _Token {
return _Token {
pos: p,
u64: val,
tag: _T_int,
}
}
func tokenName(p int, name string) _Token {
return _Token {
pos: p,
str: name,
tag: _T_name,
}
}
func tokenPunc(p int, punc _Punctuation) _Token {
return _Token {
pos: p,
tag: _T_punc,
u64: uint64(punc),
}
}
func tokenSpace(p int, end int) _Token {
return _Token {
pos: p,
end: end,
tag: _T_space,
}
}
// SyntaxError represents an error in the assembly syntax.
type SyntaxError struct {
Pos int
Row int
Src []rune
Reason string
}
// Error implements the error interface.
func (self *SyntaxError) Error() string {
if self.Pos < 0 {
return fmt.Sprintf("%s at line %d", self.Reason, self.Row)
} else {
return fmt.Sprintf("%s at %d:%d", self.Reason, self.Row, self.Pos + 1)
}
}
type _Tokenizer struct {
pos int
row int
src []rune
}
func (self *_Tokenizer) ch() rune {
return self.src[self.pos]
}
func (self *_Tokenizer) eof() bool {
return self.pos >= len(self.src)
}
func (self *_Tokenizer) rch() (ret rune) {
ret, self.pos = self.src[self.pos], self.pos + 1
return
}
func (self *_Tokenizer) err(pos int, msg string) *SyntaxError {
return &SyntaxError {
Pos : pos,
Row : self.row,
Src : self.src,
Reason : msg,
}
}
type _TrimState int
const (
_TS_normal _TrimState = iota
_TS_slcomm
_TS_hscomm
_TS_string
_TS_escape
_TS_accept
_TS_nolast
)
func (self *_Tokenizer) init(src string) {
var i int
var ch rune
var st _TrimState
/* set the source */
self.pos = 0
self.src = []rune(src)
/* remove commends, including "//" and "##" */
loop: for i, ch = range self.src {
switch {
case st == _TS_normal && ch == '/' : st = _TS_slcomm
case st == _TS_normal && ch == '"' : st = _TS_string
case st == _TS_normal && ch == ';' : st = _TS_accept; break loop
case st == _TS_normal && ch == '#' : st = _TS_hscomm
case st == _TS_slcomm && ch == '/' : st = _TS_nolast; break loop
case st == _TS_slcomm : st = _TS_normal
case st == _TS_hscomm && ch == '#' : st = _TS_nolast; break loop
case st == _TS_hscomm : st = _TS_normal
case st == _TS_string && ch == '"' : st = _TS_normal
case st == _TS_string && ch == '\\' : st = _TS_escape
case st == _TS_escape : st = _TS_string
}
}
/* check for errors */
switch st {
case _TS_accept: self.src = self.src[:i]
case _TS_nolast: self.src = self.src[:i - 1]
case _TS_string: panic(self.err(i, "string is not terminated"))
case _TS_escape: panic(self.err(i, "escape sequence is not terminated"))
}
}
func (self *_Tokenizer) skip(check func(v rune) bool) {
for !self.eof() && check(self.ch()) {
self.pos++
}
}
func (self *_Tokenizer) find(pos int, check func(v rune) bool) string {
self.skip(check)
return string(self.src[pos:self.pos])
}
func (self *_Tokenizer) chrv(p int) _Token {
var err error
var val uint64
/* starting and ending position */
p0 := p + 1
p1 := p0 + 1
/* find the end of the literal */
for p1 < len(self.src) && self.src[p1] != '\'' {
if p1++; self.src[p1 - 1] == '\\' {
p1++
}
}
/* empty literal */
if p1 == p0 {
panic(self.err(p1, "empty character constant"))
}
/* check for EOF */
if p1 == len(self.src) {
panic(self.err(p1, "unexpected EOF when scanning literals"))
}
/* parse the literal */
if val, err = literal64(string(self.src[p0:p1])); err != nil {
panic(self.err(p0, "cannot parse literal: " + err.Error()))
}
/* skip the closing '\'' */
self.pos = p1 + 1
return tokenInt(p, val)
}
func (self *_Tokenizer) numv(p int) _Token {
if val, err := strconv.ParseUint(self.find(p, isnumber), 0, 64); err != nil {
panic(self.err(p, "invalid immediate value: " + err.Error()))
} else {
return tokenInt(p, val)
}
}
func (self *_Tokenizer) defv(p int, cc rune) _Token {
if isdigit(cc) {
return self.numv(p)
} else if isident0(cc) {
return tokenName(p, self.find(p, isident))
} else {
panic(self.err(p, "invalid char: " + strconv.QuoteRune(cc)))
}
}
func (self *_Tokenizer) rep2(p int, pp _Punctuation, cc rune) _Token {
if self.eof() {
panic(self.err(self.pos, "unexpected EOF when scanning operators"))
} else if c := self.rch(); c != cc {
panic(self.err(p + 1, strconv.QuoteRune(cc) + " expected, got " + strconv.QuoteRune(c)))
} else {
return tokenPunc(p, pp)
}
}
func (self *_Tokenizer) read() _Token {
var p int
var c rune
var t _Token
/* check for EOF */
if self.eof() {
return tokenEnd(self.pos, self.pos)
}
/* skip spaces as needed */
if p = self.pos; unicode.IsSpace(self.src[p]) {
self.skip(unicode.IsSpace)
return tokenSpace(p, self.pos)
}
/* check for line comments */
if p = self.pos; p < len(self.src) - 1 && self.src[p] == '/' && self.src[p + 1] == '/' {
self.pos = len(self.src)
return tokenEnd(p, self.pos)
}
/* read the next character */
p = self.pos
c = self.rch()
/* parse the next character */
switch c {
case '+' : t = tokenPunc(p, _P_plus)
case '-' : t = tokenPunc(p, _P_minus)
case '*' : t = tokenPunc(p, _P_star)
case '/' : t = tokenPunc(p, _P_slash)
case '%' : t = tokenPunc(p, _P_percent)
case '&' : t = tokenPunc(p, _P_amp)
case '|' : t = tokenPunc(p, _P_bar)
case '^' : t = tokenPunc(p, _P_caret)
case '<' : t = self.rep2(p, _P_shl, '<')
case '>' : t = self.rep2(p, _P_shr, '>')
case '~' : t = tokenPunc(p, _P_tilde)
case '(' : t = tokenPunc(p, _P_lbrk)
case ')' : t = tokenPunc(p, _P_rbrk)
case '.' : t = tokenPunc(p, _P_dot)
case ',' : t = tokenPunc(p, _P_comma)
case ':' : t = tokenPunc(p, _P_colon)
case '$' : t = tokenPunc(p, _P_dollar)
case '#' : t = tokenPunc(p, _P_hash)
case '\'' : t = self.chrv(p)
default : t = self.defv(p, c)
}
/* mark the end of token */
t.end = self.pos
return t
}
func (self *_Tokenizer) next() (tk _Token) {
for {
if tk = self.read(); tk.tag != _T_space {
return
}
}
}
// LabelKind indicates the type of label reference.
type LabelKind int
// OperandKind indicates the type of the operand.
type OperandKind int
// InstructionPrefix indicates the prefix bytes prepended to the instruction.
type InstructionPrefix byte
const (
// OpImm means the operand is an immediate value.
OpImm OperandKind = 1 << iota
// OpReg means the operand is a register.
OpReg
// OpMem means the operand is a memory address.
OpMem
// OpLabel means the operand is a label, specifically for
// branch instructions.
OpLabel
)
const (
// Declaration means the label is a declaration.
Declaration LabelKind = iota + 1
// BranchTarget means the label should be treated as a branch target.
BranchTarget
// RelativeAddress means the label should be treated as a reference to
// the code section (e.g. RIP-relative addressing).
RelativeAddress
)
const (
// PrefixLock causes the processor's LOCK# signal to be asserted during execution of
// the accompanying instruction (turns the instruction into an atomic instruction).
// In a multiprocessor environment, the LOCK# signal insures that the processor
// has exclusive use of any shared memory while the signal is asserted.
PrefixLock InstructionPrefix = iota
// PrefixSegmentCS overrides the memory operation of this instruction to CS (Code Segment).
PrefixSegmentCS
// PrefixSegmentDS overrides the memory operation of this instruction to DS (Data Segment),
// this is the default section for most instructions if not specified.
PrefixSegmentDS
// PrefixSegmentES overrides the memory operation of this instruction to ES (Extra Segment).
PrefixSegmentES
// PrefixSegmentFS overrides the memory operation of this instruction to FS.
PrefixSegmentFS
// PrefixSegmentGS overrides the memory operation of this instruction to GS.
PrefixSegmentGS
// PrefixSegmentSS overrides the memory operation of this instruction to SS (Stack Segment).
PrefixSegmentSS
)
// ParsedLabel represents a label in the source, either a jump target or
// an RIP-relative addressing.
type ParsedLabel struct {
Name string
Kind LabelKind
}
// ParsedOperand represents an operand of an instruction in the source.
type ParsedOperand struct {
Op OperandKind
Imm int64
Reg Register
Label ParsedLabel
Memory MemoryAddress
}
// ParsedInstruction represents an instruction in the source.
type ParsedInstruction struct {
Mnemonic string
Operands []ParsedOperand
Prefixes []InstructionPrefix
}
func (self *ParsedInstruction) imm(v int64) {
self.Operands = append(self.Operands, ParsedOperand {
Op : OpImm,
Imm : v,
})
}
func (self *ParsedInstruction) reg(v Register) {
self.Operands = append(self.Operands, ParsedOperand {
Op : OpReg,
Reg : v,
})
}
func (self *ParsedInstruction) mem(v MemoryAddress) {
self.Operands = append(self.Operands, ParsedOperand {
Op : OpMem,
Memory : v,
})
}
func (self *ParsedInstruction) target(v string) {
self.Operands = append(self.Operands, ParsedOperand {
Op : OpLabel,
Label : ParsedLabel {
Name: v,
Kind: BranchTarget,
},
})
}
func (self *ParsedInstruction) reference(v string) {
self.Operands = append(self.Operands, ParsedOperand {
Op : OpLabel,
Label : ParsedLabel {
Name: v,
Kind: RelativeAddress,
},
})
}
// LineKind indicates the type of ParsedLine.
type LineKind int
const (
// LineLabel means the ParsedLine is a label.
LineLabel LineKind = iota + 1
// LineInstr means the ParsedLine is an instruction.
LineInstr
// LineCommand means the ParsedLine is a ParsedCommand.
LineCommand
)
// ParsedLine represents a parsed source line.
type ParsedLine struct {
Row int
Src []rune
Kind LineKind
Label ParsedLabel
Command ParsedCommand
Instruction ParsedInstruction
}
// ParsedCommand represents a parsed assembly directive command.
type ParsedCommand struct {
Cmd string
Args []ParsedCommandArg
}
// ParsedCommandArg represents an argument of a ParsedCommand.
type ParsedCommandArg struct {
Value string
IsString bool
}
// Parser parses the source, and generates a sequence of ParsedInstruction's.
type Parser struct {
lex _Tokenizer
exp expr.Parser
}
const (
rip Register64 = 0xff
)
var _RegBranch = map[string]bool {
"jmp" : true,
"jmpq" : true,
"call" : true,
"callq" : true,
}
var _SegPrefix = map[string]InstructionPrefix {
"cs": PrefixSegmentCS,
"ds": PrefixSegmentDS,
"es": PrefixSegmentES,
"fs": PrefixSegmentFS,
"gs": PrefixSegmentGS,
"ss": PrefixSegmentSS,
}
func (self *Parser) i32(tk _Token, v int64) int32 {
if v >= math.MinInt32 && v <= math.MaxUint32 {
return int32(v)
} else {
panic(self.err(tk.pos, fmt.Sprintf("32-bit integer out ouf range: %d", v)))
}
}
func (self *Parser) err(pos int, msg string) *SyntaxError {
return &SyntaxError {
Pos : pos,
Row : self.lex.row,
Src : self.lex.src,
Reason : msg,
}
}
func (self *Parser) negv() int64 {
tk := self.lex.read()
tt := tk.tag
/* must be an integer */
if tt != _T_int {
panic(self.err(tk.pos, "integer expected after '-'"))
} else {
return -int64(tk.u64)
}
}
func (self *Parser) eval(p int) (r int64) {
var e error
var v *expr.Expr
/* searching start */
n := 1
q := p + 1
/* find the end of expression */
for n > 0 && q < len(self.lex.src) {
switch self.lex.src[q] {
case '(' : q++; n++
case ')' : q++; n--
default : q++
}
}
/* check for EOF */
if n != 0 {
panic(self.err(q, "unexpected EOF when parsing expressions"))
}
/* evaluate the expression */
if v, e = self.exp.SetSource(string(self.lex.src[p:q - 1])).Parse(nil); e != nil {
panic(self.err(p, "cannot evaluate expression: " + e.Error()))
}
/* evaluate the expression */
if r, e = v.Evaluate(); e != nil {
panic(self.err(p, "cannot evaluate expression: " + e.Error()))
}
/* skip the last ')' */
v.Free()
self.lex.pos = q
return
}
func (self *Parser) relx(tk _Token) {
if tk.tag != _T_punc || tk.punc() != _P_lbrk {
panic(self.err(tk.pos, "'(' expected for RIP-relative addressing"))
} else if tk = self.lex.next(); self.regx(tk) != rip {
panic(self.err(tk.pos, "RIP-relative addressing expects %rip as the base register"))
} else if tk = self.lex.next(); tk.tag != _T_punc || tk.punc() != _P_rbrk {
panic(self.err(tk.pos, "RIP-relative addressing does not support indexing or scaling"))
}
}
func (self *Parser) immx(tk _Token) int64 {
if tk.tag != _T_punc || tk.punc() != _P_dollar {
panic(self.err(tk.pos, "'$' expected for registers"))
} else if tk = self.lex.read(); tk.tag == _T_int {
return int64(tk.u64)
} else if tk.tag == _T_punc && tk.punc() == _P_lbrk {
return self.eval(self.lex.pos)
} else if tk.tag == _T_punc && tk.punc() == _P_minus {
return self.negv()
} else {
panic(self.err(tk.pos, "immediate value expected"))
}
}
func (self *Parser) regx(tk _Token) Register {
if tk.tag != _T_punc || tk.punc() != _P_percent {
panic(self.err(tk.pos, "'%' expected for registers"))
} else if tk = self.lex.read(); tk.tag != _T_name {
panic(self.err(tk.pos, "register name expected"))
} else if tk.str == "rip" {
return rip
} else if reg, ok := Registers[tk.str]; ok {
return reg
} else {
panic(self.err(tk.pos, "invalid register name: " + strconv.Quote(tk.str)))
}
}
func (self *Parser) regv(tk _Token) Register {
if reg := self.regx(tk); reg == rip {
panic(self.err(tk.pos, "%rip is not accessable as a dedicated register"))
} else {
return reg
}
}
func (self *Parser) disp(vv int32) MemoryAddress {
switch tk := self.lex.next(); tk.tag {
case _T_end : return MemoryAddress { Displacement: vv }
case _T_punc : return self.relm(tk, vv)
default : panic(self.err(tk.pos, "',' or '(' expected"))
}
}
func (self *Parser) relm(tv _Token, disp int32) MemoryAddress {
var tk _Token
var tt _TokenKind
/* check for absolute addressing */
if tv.punc() == _P_comma {
self.lex.pos--
return MemoryAddress { Displacement: disp }
}
/* must be '(' now */
if tv.punc() != _P_lbrk {
panic(self.err(tv.pos, "',' or '(' expected"))
}
/* read the next token */
tk = self.lex.next()
tt = tk.tag
/* must be a punctuation */
if tt != _T_punc {
panic(self.err(tk.pos, "'%' or ',' expected"))
}
/* check for base */
switch tk.punc() {
case _P_percent : return self.base(tk, disp)
case _P_comma : return self.index(nil, disp)
default : panic(self.err(tk.pos, "'%' or ',' expected"))
}
}
func (self *Parser) base(tk _Token, disp int32) MemoryAddress {
rr := self.regx(tk)
nk := self.lex.next()
/* check for register indirection or base-index addressing */
if !isReg64(rr) {
panic(self.err(tk.pos, "not a valid base register"))
} else if nk.tag != _T_punc {
panic(self.err(nk.pos, "',' or ')' expected"))
} else if nk.punc() == _P_comma {
return self.index(rr, disp)
} else if nk.punc() == _P_rbrk {
return MemoryAddress { Base: rr, Displacement: disp }
} else {
panic(self.err(nk.pos, "',' or ')' expected"))
}
}
func (self *Parser) index(base Register, disp int32) MemoryAddress {
tk := self.lex.next()
rr := self.regx(tk)
nk := self.lex.next()
/* check for scaled indexing */
if base == rip {
panic(self.err(tk.pos, "RIP-relative addressing does not support indexing or scaling"))
} else if !isIndexable(rr) {
panic(self.err(tk.pos, "not a valid index register"))
} else if nk.tag != _T_punc {
panic(self.err(nk.pos, "',' or ')' expected"))
} else if nk.punc() == _P_comma {
return self.scale(base, rr, disp)
} else if nk.punc() == _P_rbrk {
return MemoryAddress { Base: base, Index: rr, Scale: 1, Displacement: disp }
} else {
panic(self.err(nk.pos, "',' or ')' expected"))
}
}
func (self *Parser) scale(base Register, index Register, disp int32) MemoryAddress {
tk := self.lex.next()
tt := tk.tag
tv := tk.u64
/* must be an integer */
if tt != _T_int {
panic(self.err(tk.pos, "integer expected"))
}
/* scale can only be 1, 2, 4 or 8 */
if tv == 0 || (_Scales & (1 << tv)) == 0 {
panic(self.err(tk.pos, "scale can only be 1, 2, 4 or 8"))
}
/* read next token */
tk = self.lex.next()
tt = tk.tag
/* check for the closing ')' */
if tt != _T_punc || tk.punc() != _P_rbrk {
panic(self.err(tk.pos, "')' expected"))
}
/* construct the memory address */
return MemoryAddress {
Base : base,
Index : index,
Scale : uint8(tv),
Displacement : disp,
}
}
func (self *Parser) cmds() *ParsedLine {
cmd := ""
pos := self.lex.pos
buf := []ParsedCommandArg(nil)
/* find the end of command */
for p := pos; pos < len(self.lex.src); pos++ {
if unicode.IsSpace(self.lex.src[pos]) {
cmd = string(self.lex.src[p:pos])
break
}
}
/* parse the arguments */
loop: for {
switch self.next(&pos) {
case 0 : break loop
case '#' : break loop
case '"' : pos = self.strings(&buf, pos)
default : pos = self.expressions(&buf, pos)
}
}
/* construct the line */
return &ParsedLine {
Row : self.lex.row,
Src : self.lex.src,
Kind : LineCommand,
Command : ParsedCommand {
Cmd : cmd,
Args : buf,
},
}
}
func (self *Parser) feed(line string) *ParsedLine {
ff := true
rr := false
lk := false
/* reset the lexer */
self.lex.row++
self.lex.init(line)
/* parse the first token */
tk := self.lex.next()
tt := tk.tag
/* it is a directive if it starts with a dot */
if tk.tag == _T_punc && tk.punc() == _P_dot {
return self.cmds()
}
/* otherwise it could be labels or instructions */
if tt != _T_name {
panic(self.err(tk.pos, "identifier expected"))
}
/* peek the next token */
lex := self.lex
tkx := lex.next()
/* check for labels */
if tkx.tag == _T_punc && tkx.punc() == _P_colon {
tkx = lex.next()
ttx := tkx.tag
/* the line must end here */
if ttx != _T_end {
panic(self.err(tkx.pos, "garbage after label definition"))
}
/* construct the label */
return &ParsedLine {
Row : self.lex.row,
Src : self.lex.src,
Kind : LineLabel,
Label : ParsedLabel {
Kind: Declaration,
Name: tk.str,
},
}
}
/* special case for the "lock" prefix */
if tk.tag == _T_name && strings.ToLower(tk.str) == "lock" {
lk = true
tk = self.lex.next()
/* must be an instruction */
if tk.tag != _T_name {
panic(self.err(tk.pos, "identifier expected"))
}
}
/* set the line kind and mnemonic */
ret := &ParsedLine {
Row : self.lex.row,
Src : self.lex.src,
Kind : LineInstr,
Instruction : ParsedInstruction { Mnemonic: strings.ToLower(tk.str) },
}
/* check for LOCK prefix */
if lk {
ret.Instruction.Prefixes = append(ret.Instruction.Prefixes, PrefixLock)
}
/* parse all the operands */
for {
tk = self.lex.next()
tt = tk.tag
/* check for end of line */
if tt == _T_end {
break
}
/* expect a comma if not the first operand */
if !ff {
if tt == _T_punc && tk.punc() == _P_comma {
tk = self.lex.next()
} else {
panic(self.err(tk.pos, "',' expected"))
}
}
/* not the first operand anymore */
ff = false
tt = tk.tag
/* encountered an integer, must be a SIB memory address */
if tt == _T_int {
ret.Instruction.mem(self.disp(self.i32(tk, int64(tk.u64))))
continue
}
/* encountered an identifier, maybe an expression or a jump target, or a segment override prefix */
if tt == _T_name {
ts := tk.str
tp := self.lex.pos
/* if the next token is EOF or a comma, it's a jumpt target */
if tk = self.lex.next(); tk.tag == _T_end || (tk.tag == _T_punc && tk.punc() == _P_comma) {
self.lex.pos = tp
ret.Instruction.target(ts)
continue
}
/* if it is a colon, it's a segment override prefix, otherwise it must be an RIP-relative addressing operand */
if tk.tag != _T_punc || tk.punc() != _P_colon {
self.relx(tk)
ret.Instruction.reference(ts)
continue
}
/* lookup segment prefixes */
if p, ok := _SegPrefix[strings.ToLower(ts)]; !ok {
panic(self.err(tk.pos, "invalid segment name"))
} else {
ret.Instruction.Prefixes = append(ret.Instruction.Prefixes, p)
}
/* read the next token */
tk = self.lex.next()
tt = tk.tag
/* encountered an integer, must be a SIB memory address */
if tt == _T_int {
ret.Instruction.mem(self.disp(self.i32(tk, int64(tk.u64))))
continue
}
}
/* certain instructions may have a "*" before operands */
if tt == _T_punc && tk.punc() == _P_star {
tk = self.lex.next()
tt = tk.tag
rr = true
}
/* ... otherwise it must be a punctuation */
if tt != _T_punc {
panic(self.err(tk.pos, "'$', '%', '-' or '(' expected"))
}
/* check the operator */
switch tk.punc() {
case _P_lbrk : break
case _P_minus : ret.Instruction.mem(self.disp(self.i32(tk, self.negv()))) ; continue
case _P_dollar : ret.Instruction.imm(self.immx(tk)) ; continue
case _P_percent : ret.Instruction.reg(self.regv(tk)) ; continue
default : panic(self.err(tk.pos, "'$', '%', '-' or '(' expected"))
}
/* special case of '(', might be either `(expr)(SIB)` or just `(SIB)`
* read one more token to confirm */
tk = self.lex.next()
tt = tk.tag
/* the next token is '%', it's a memory address,
* or ',' if it's a memory address without base,
* otherwise it must be in `(expr)(SIB)` form */
if tk.tag == _T_punc && tk.punc() == _P_percent {
ret.Instruction.mem(self.base(tk, 0))
} else if tk.tag == _T_punc && tk.punc() == _P_comma {
ret.Instruction.mem(self.index(nil, 0))
} else {
ret.Instruction.mem(self.disp(self.i32(tk, self.eval(tk.pos))))
}
}
/* check "jmp" and "call" instructions */
if !_RegBranch[ret.Instruction.Mnemonic] {
return ret
} else if len(ret.Instruction.Operands) != 1 {
panic(self.err(tk.pos, fmt.Sprintf(`"%s" requires exact 1 argument`, ret.Instruction.Mnemonic)))
} else if !rr && ret.Instruction.Operands[0].Op != OpReg && ret.Instruction.Operands[0].Op != OpLabel {
panic(self.err(tk.pos, fmt.Sprintf(`invalid operand for "%s" instruction`, ret.Instruction.Mnemonic)))
} else {
return ret
}
}
func (self *Parser) next(p *int) rune {
for {
if *p >= len(self.lex.src) {
return 0
} else if cc := self.lex.src[*p]; !unicode.IsSpace(cc) {
return cc
} else {
*p++
}
}
}
func (self *Parser) delim(p int) int {
if cc := self.next(&p); cc == 0 {
return p
} else if cc == ',' {
return p + 1
} else {
panic(self.err(p, "',' expected"))
}
}
func (self *Parser) strings(argv *[]ParsedCommandArg, p int) int {
var i int
var e error
var v string
/* find the end of string */
for i = p + 1; i < len(self.lex.src) && self.lex.src[i] != '"'; i++ {
if self.lex.src[i] == '\\' {
i++
}
}
/* check for EOF */
if i == len(self.lex.src) {
panic(self.err(i, "unexpected EOF when scanning strings"))
}
/* unquote the string */
if v, e = strconv.Unquote(string(self.lex.src[p:i + 1])); e != nil {
panic(self.err(p, "invalid string: " + e.Error()))
}
/* add the argument to buffer */
*argv = append(*argv, ParsedCommandArg { Value: v, IsString: true })
return self.delim(i + 1)
}
func (self *Parser) directives(line string) {
self.lex.row++
self.lex.init(line)
/* parse the first token */
tk := self.lex.next()
tt := tk.tag
/* check for EOF */
if tt == _T_end {
return
}
/* must be a directive */
if tt != _T_punc || tk.punc() != _P_hash {
panic(self.err(tk.pos, "'#' expected"))
}
/* parse the line number */
tk = self.lex.next()
tt = tk.tag
/* must be a line number, if it is, set the row number, and ignore the rest of the line */
if tt != _T_int {
panic(self.err(tk.pos, "line number expected"))
} else {
self.lex.row = int(tk.u64) - 1
}
}
func (self *Parser) expressions(argv *[]ParsedCommandArg, p int) int {
var i int
var n int
var s int
/* scan until the first standalone ',' or EOF */
loop: for i = p; i < len(self.lex.src); i++ {
switch self.lex.src[i] {
case ',' : if s == 0 { if n == 0 { break loop } }
case ']', '}', '>' : if s == 0 { if n == 0 { break loop } else { n-- } }
case '[', '{', '<' : if s == 0 { n++ }
case '\\' : if s != 0 { i++ }
case '\'' : if s != 2 { s ^= 1 }
case '"' : if s != 1 { s ^= 2 }
}
}
/* check for EOF in strings */
if s != 0 {
panic(self.err(i, "unexpected EOF when scanning strings"))
}
/* check for bracket matching */
if n != 0 {
panic(self.err(i, "unbalanced '{' or '[' or '<'"))
}
/* add the argument to buffer */
*argv = append(*argv, ParsedCommandArg { Value: string(self.lex.src[p:i]) })
return self.delim(i)
}
// Feed feeds the parser with one more line, and the parser
// parses it into a ParsedLine.
//
// NOTE: Feed does not handle empty lines or multiple lines,
// it panics when this happens. Use Parse to parse multiple
// lines of assembly source.
//
func (self *Parser) Feed(src string) (ret *ParsedLine, err error) {
var ok bool
var ss string
var vv interface{}
/* check for multiple lines */
if strings.ContainsRune(src, '\n') {
return nil, errors.New("passing multiple lines to Feed()")
}
/* check for blank lines */
if ss = strings.TrimSpace(src); ss == "" || ss[0] == '#' || strings.HasPrefix(ss, "//") {
return nil, errors.New("blank line or line with only comments or line-marks")
}
/* setup error handler */
defer func() {
if vv = recover(); vv != nil {
if err, ok = vv.(*SyntaxError); !ok {
panic(vv)
}
}
}()
/* call the actual parser */
ret = self.feed(src)
return
}
// Parse parses the entire assembly source (possibly multiple lines) into
// a sequence of *ParsedLine.
func (self *Parser) Parse(src string) (ret []*ParsedLine, err error) {
var ok bool
var ss string
var vv interface{}
/* setup error handler */
defer func() {
if vv = recover(); vv != nil {
if err, ok = vv.(*SyntaxError); !ok {
panic(vv)
}
}
}()
/* feed every line */
for _, line := range strings.Split(src, "\n") {
if ss = strings.TrimSpace(line); ss == "" || strings.HasPrefix(ss, "//") {
self.lex.row++
} else if ss[0] == '#' {
self.directives(line)
} else {
ret = append(ret, self.feed(line))
}
}
/* all done */
err = nil
return
}
// Directive handles the directive.
func (self *Parser) Directive(line string) (err error) {
var ok bool
var ss string
var vv interface{}
/* check for directives */
if ss = strings.TrimSpace(line); ss == "" || ss[0] != '#' {
return errors.New("not a directive")
}
/* setup error handler */
defer func() {
if vv = recover(); vv != nil {
if err, ok = vv.(*SyntaxError); !ok {
panic(vv)
}
}
}()
/* call the directive parser */
self.directives(line)
return
}
type _TermRepo struct {
terms map[string]expr.Term
}
func (self *_TermRepo) Get(name string) (expr.Term, error) {
if ret, ok := self.terms[name]; ok {
return ret, nil
} else {
return nil, errors.New("undefined name: " + name)
}
}
func (self *_TermRepo) label(name string) (*Label, error) {
var ok bool
var lb *Label
var tr expr.Term
/* check for existing terms */
if tr, ok = self.terms[name]; ok {
if lb, ok = tr.(*Label); ok {
return lb, nil
} else {
return nil, errors.New("name is not a label: " + name)
}
}
/* create a new one as needed */
lb = new(Label)
lb.Name = name
/* create the map if needed */
if self.terms == nil {
self.terms = make(map[string]expr.Term, 1)
}
/* register the label */
self.terms[name] = lb
return lb, nil
}
func (self *_TermRepo) define(name string, term expr.Term) {
var ok bool
var tr expr.Term
/* create the map if needed */
if self.terms == nil {
self.terms = make(map[string]expr.Term, 1)
}
/* check for existing terms */
if tr, ok = self.terms[name]; !ok {
self.terms[name] = term
} else if _, ok = tr.(*Label); !ok {
self.terms[name] = term
} else {
panic("conflicting term types: " + name)
}
}
// _Command describes an assembler command.
//
// The _Command.args describes both the arity and argument type with characters,
// the length is the number of arguments, the character itself represents the
// argument type.
//
// Possible values are:
//
// s This argument should be a string
// e This argument should be an expression
// ? The next argument is optional, and must be the last argument.
//
type _Command struct {
args string
handler func(*Assembler, *Program, []ParsedCommandArg) error
}
// Options controls the behavior of Assembler.
type Options struct {
// InstructionAliasing specifies whether to enable instruction aliasing.
// Set to true enables instruction aliasing, and the Assembler will try harder to find instructions.
InstructionAliasing bool
// IgnoreUnknownDirectives specifies whether to report errors when encountered unknown directives.
// Set to true ignores all unknwon directives silently, useful for parsing generated assembly.
IgnoreUnknownDirectives bool
}
// Assembler assembles the entire assembly program and generates the corresponding
// machine code representations.
type Assembler struct {
cc int
ps Parser
pc uintptr
buf []byte
main string
opts Options
repo _TermRepo
expr expr.Parser
line *ParsedLine
}
var asmCommands = map[string]_Command {
"org" : { "e" , (*Assembler).assembleCommandOrg },
"set" : { "ee" , (*Assembler).assembleCommandSet },
"byte" : { "e" , (*Assembler).assembleCommandByte },
"word" : { "e" , (*Assembler).assembleCommandWord },
"long" : { "e" , (*Assembler).assembleCommandLong },
"quad" : { "e" , (*Assembler).assembleCommandQuad },
"fill" : { "e?e" , (*Assembler).assembleCommandFill },
"space" : { "e?e" , (*Assembler).assembleCommandFill },
"align" : { "e?e" , (*Assembler).assembleCommandAlign },
"entry" : { "e" , (*Assembler).assembleCommandEntry },
"ascii" : { "s" , (*Assembler).assembleCommandAscii },
"asciz" : { "s" , (*Assembler).assembleCommandAsciz },
"p2align" : { "e?e" , (*Assembler).assembleCommandP2Align },
}
func (self *Assembler) err(msg string) *SyntaxError {
return &SyntaxError {
Pos : -1,
Row : self.line.Row,
Src : self.line.Src,
Reason : msg,
}
}
func (self *Assembler) eval(expr string) (int64, error) {
if exp, err := self.expr.SetSource(expr).Parse(nil); err != nil {
return 0, err
} else {
return exp.Evaluate()
}
}
func (self *Assembler) checkArgs(i int, n int, v *ParsedCommand, isString bool) error {
if i >= len(v.Args) {
return self.err(fmt.Sprintf("command %s takes exact %d arguments", strconv.Quote(v.Cmd), n))
} else if isString && !v.Args[i].IsString {
return self.err(fmt.Sprintf("argument %d of command %s must be a string", i + 1, strconv.Quote(v.Cmd)))
} else if !isString && v.Args[i].IsString {
return self.err(fmt.Sprintf("argument %d of command %s must be an expression", i + 1, strconv.Quote(v.Cmd)))
} else {
return nil
}
}
func (self *Assembler) assembleLabel(p *Program, lb *ParsedLabel) error {
if v, err := self.repo.label(lb.Name); err != nil {
return err
} else {
p.Link(v)
return nil
}
}
func (self *Assembler) assembleInstr(p *Program, line *ParsedInstruction) (err error) {
var ok bool
var pfx []byte
var ops []interface{}
var enc _InstructionEncoder
/* convert to lower-case */
opts := self.opts
name := strings.ToLower(line.Mnemonic)
/* fix register-addressing branches if needed */
if opts.InstructionAliasing && len(line.Operands) == 1 {
switch {
case name == "retq" : name = "ret"
case name == "movabsq" : name = "movq"
case name == "jmp" && line.Operands[0].Op != OpLabel : name = "jmpq"
case name == "jmpq" && line.Operands[0].Op == OpLabel : name = "jmp"
case name == "call" && line.Operands[0].Op != OpLabel : name = "callq"
case name == "callq" && line.Operands[0].Op == OpLabel : name = "call"
}
}
/* lookup from the alias table if needed */
if opts.InstructionAliasing {
enc, ok = _InstructionAliases[name]
}
/* lookup from the instruction table */
if !ok {
enc, ok = Instructions[name]
}
/* remove size suffix if possible */
if !ok && opts.InstructionAliasing {
switch i := len(name) - 1; name[i] {
case 'b', 'w', 'l', 'q': {
enc, ok = Instructions[name[:i]]
}
}
}
/* check for instruction name */
if !ok {
return self.err("no such instruction: " + strconv.Quote(name))
}
/* allocate memory for prefix if any */
if len(line.Prefixes) != 0 {
pfx = make([]byte, len(line.Prefixes))
}
/* convert the prefixes */
for i, v := range line.Prefixes {
switch v {
case PrefixLock : pfx[i] = _P_lock
case PrefixSegmentCS : pfx[i] = _P_cs
case PrefixSegmentDS : pfx[i] = _P_ds
case PrefixSegmentES : pfx[i] = _P_es
case PrefixSegmentFS : pfx[i] = _P_fs
case PrefixSegmentGS : pfx[i] = _P_gs
case PrefixSegmentSS : pfx[i] = _P_ss
default : panic("unreachable: invalid segment prefix")
}
}
/* convert the operands */
for _, op := range line.Operands {
switch op.Op {
case OpImm : ops = append(ops, op.Imm)
case OpReg : ops = append(ops, op.Reg)
case OpMem : self.assembleInstrMem(&ops, op.Memory)
case OpLabel : self.assembleInstrLabel(&ops, op.Label)
default : panic("parser yields an invalid operand kind")
}
}
/* catch any exceptions in the encoder */
defer func() {
if v := recover(); v != nil {
err = self.err(fmt.Sprint(v))
}
}()
/* encode the instruction */
enc(p, ops...).prefix = pfx
return nil
}
func (self *Assembler) assembleInstrMem(ops *[]interface{}, addr MemoryAddress) {
mem := new(MemoryOperand)
*ops = append(*ops, mem)
/* check for RIP-relative addressing */
if addr.Base != rip {
mem.Addr.Type = Memory
mem.Addr.Memory = addr
} else {
mem.Addr.Type = Offset
mem.Addr.Offset = RelativeOffset(addr.Displacement)
}
}
func (self *Assembler) assembleInstrLabel(ops *[]interface{}, label ParsedLabel) {
vk := label.Kind
tr, err := self.repo.label(label.Name)
/* check for errors */
if err != nil {
panic(err)
}
/* check for branch target */
if vk == BranchTarget {
*ops = append(*ops, tr)
return
}
/* add to ops */
*ops = append(*ops, &MemoryOperand {
Addr: Addressable {
Type : Reference,
Reference : tr,
},
})
}
func (self *Assembler) assembleCommand(p *Program, line *ParsedCommand) error {
var iv int
var cc rune
var ok bool
var va bool
var fn _Command
/* find the command */
if fn, ok = asmCommands[line.Cmd]; !ok {
if self.opts.IgnoreUnknownDirectives {
return nil
} else {
return self.err("no such command: " + strconv.Quote(line.Cmd))
}
}
/* expected & real argument count */
argx := len(fn.args)
argc := len(line.Args)
/* check the arguments */
loop: for iv, cc = range fn.args {
switch cc {
case '?' : va = true; break loop
case 's' : if err := self.checkArgs(iv, argx, line, true) ; err != nil { return err }
case 'e' : if err := self.checkArgs(iv, argx, line, false) ; err != nil { return err }
default : panic("invalid argument descriptor: " + strconv.Quote(fn.args))
}
}
/* simple case: non-variadic command */
if !va {
if argc == argx {
return fn.handler(self, p, line.Args)
} else {
return self.err(fmt.Sprintf("command %s takes exact %d arguments", strconv.Quote(line.Cmd), argx))
}
}
/* check for the descriptor */
if iv != argx - 2 {
panic("invalid argument descriptor: " + strconv.Quote(fn.args))
}
/* variadic command and the final optional argument is set */
if argc == argx - 1 {
switch fn.args[argx - 1] {
case 's' : if err := self.checkArgs(iv, -1, line, true) ; err != nil { return err }
case 'e' : if err := self.checkArgs(iv, -1, line, false) ; err != nil { return err }
default : panic("invalid argument descriptor: " + strconv.Quote(fn.args))
}
}
/* check argument count */
if argc == argx - 1 || argc == argx - 2 {
return fn.handler(self, p, line.Args)
} else {
return self.err(fmt.Sprintf("command %s takes %d or %d arguments", strconv.Quote(line.Cmd), argx - 2, argx - 1))
}
}
func (self *Assembler) assembleCommandInt(p *Program, argv []ParsedCommandArg, addfn func(*Program, *expr.Expr) *Instruction) error {
var err error
var val *expr.Expr
/* parse the expression */
if val, err = self.expr.SetSource(argv[0].Value).Parse(&self.repo); err != nil {
return err
}
/* add to the program */
addfn(p, val)
return nil
}
func (self *Assembler) assembleCommandOrg(_ *Program, argv []ParsedCommandArg) error {
var err error
var val int64
/* evaluate the expression */
if val, err = self.eval(argv[0].Value); err != nil {
return err
}
/* check for origin */
if val < 0 {
return self.err(fmt.Sprintf("negative origin: %d", val))
}
/* ".org" must be the first command if any */
if self.cc != 1 {
return self.err(".org must be the first command if present")
}
/* set the initial program counter */
self.pc = uintptr(val)
return nil
}
func (self *Assembler) assembleCommandSet(_ *Program, argv []ParsedCommandArg) error {
var err error
var val *expr.Expr
/* parse the expression */
if val, err = self.expr.SetSource(argv[1].Value).Parse(&self.repo); err != nil {
return err
}
/* define the new identifier */
self.repo.define(argv[0].Value, val)
return nil
}
func (self *Assembler) assembleCommandByte(p *Program, argv []ParsedCommandArg) error {
return self.assembleCommandInt(p, argv, (*Program).Byte)
}
func (self *Assembler) assembleCommandWord(p *Program, argv []ParsedCommandArg) error {
return self.assembleCommandInt(p, argv, (*Program).Word)
}
func (self *Assembler) assembleCommandLong(p *Program, argv []ParsedCommandArg) error {
return self.assembleCommandInt(p, argv, (*Program).Long)
}
func (self *Assembler) assembleCommandQuad(p *Program, argv []ParsedCommandArg) error {
return self.assembleCommandInt(p, argv, (*Program).Quad)
}
func (self *Assembler) assembleCommandFill(p *Program, argv []ParsedCommandArg) error {
var fv byte
var nb int64
var ex error
/* evaluate the size */
if nb, ex = self.eval(argv[0].Value); ex != nil {
return ex
}
/* check for filling size */
if nb < 0 {
return self.err(fmt.Sprintf("negative filling size: %d", nb))
}
/* check for optional filling value */
if len(argv) == 2 {
if val, err := self.eval(argv[1].Value); err != nil {
return err
} else if val < math.MinInt8 || val > math.MaxUint8 {
return self.err(fmt.Sprintf("value %d cannot be represented with a byte", val))
} else {
fv = byte(val)
}
}
/* fill with specified byte */
p.Data(bytes.Repeat([]byte { fv }, int(nb)))
return nil
}
func (self *Assembler) assembleCommandAlign(p *Program, argv []ParsedCommandArg) error {
var nb int64
var ex error
var fv *expr.Expr
/* evaluate the size */
if nb, ex = self.eval(argv[0].Value); ex != nil {
return ex
}
/* check for alignment value */
if nb <= 0 {
return self.err(fmt.Sprintf("zero or negative alignment: %d", nb))
}
/* alignment must be a power of 2 */
if (nb & (nb - 1)) != 0 {
return self.err(fmt.Sprintf("alignment must be a power of 2: %d", nb))
}
/* check for optional filling value */
if len(argv) == 2 {
if v, err := self.expr.SetSource(argv[1].Value).Parse(&self.repo); err == nil {
fv = v
} else {
return err
}
}
/* fill with specified byte, default to 0 if not specified */
p.Align(uint64(nb), fv)
return nil
}
func (self *Assembler) assembleCommandEntry(_ *Program, argv []ParsedCommandArg) error {
name := argv[0].Value
rbuf := []rune(name)
/* check all the characters */
for i, cc := range rbuf {
if !isident0(cc) && (i == 0 || !isident(cc)) {
return self.err("entry point must be a label name")
}
}
/* set the main entry point */
self.main = name
return nil
}
func (self *Assembler) assembleCommandAscii(p *Program, argv []ParsedCommandArg) error {
p.Data([]byte(argv[0].Value))
return nil
}
func (self *Assembler) assembleCommandAsciz(p *Program, argv []ParsedCommandArg) error {
p.Data(append([]byte(argv[0].Value), 0))
return nil
}
func (self *Assembler) assembleCommandP2Align(p *Program, argv []ParsedCommandArg) error {
var nb int64
var ex error
var fv *expr.Expr
/* evaluate the size */
if nb, ex = self.eval(argv[0].Value); ex != nil {
return ex
}
/* check for alignment value */
if nb <= 0 {
return self.err(fmt.Sprintf("zero or negative alignment: %d", nb))
}
/* check for optional filling value */
if len(argv) == 2 {
if v, err := self.expr.SetSource(argv[1].Value).Parse(&self.repo); err == nil {
fv = v
} else {
return err
}
}
/* fill with specified byte, default to 0 if not specified */
p.Align(1 << nb, fv)
return nil
}
// Base returns the origin.
func (self *Assembler) Base() uintptr {
return self.pc
}
// Code returns the assembled machine code.
func (self *Assembler) Code() []byte {
return self.buf
}
// Entry returns the address of the specified entry point, or the origin if not specified.
func (self *Assembler) Entry() uintptr {
if self.main == "" {
return self.pc
} else if tr, err := self.repo.Get(self.main); err != nil {
panic(err)
} else if val, err := tr.Evaluate(); err != nil {
panic(err)
} else {
return uintptr(val)
}
}
// Options returns the internal options reference, changing it WILL affect this Assembler instance.
func (self *Assembler) Options() *Options {
return &self.opts
}
// WithBase resets the origin to pc.
func (self *Assembler) WithBase(pc uintptr) *Assembler {
self.pc = pc
return self
}
// Assemble assembles the assembly source and save the machine code to internal buffer.
func (self *Assembler) Assemble(src string) error {
var err error
var buf []*ParsedLine
/* parse the source */
if buf, err = self.ps.Parse(src); err != nil {
return err
}
/* create a new program */
p := DefaultArch.CreateProgram()
defer p.Free()
/* process every line */
for _, self.line = range buf {
switch self.cc++; self.line.Kind {
case LineLabel : if err = self.assembleLabel (p, &self.line.Label) ; err != nil { return err }
case LineInstr : if err = self.assembleInstr (p, &self.line.Instruction) ; err != nil { return err }
case LineCommand : if err = self.assembleCommand (p, &self.line.Command) ; err != nil { return err }
default : panic("parser yields an invalid line kind")
}
}
/* assemble the program */
self.buf = p.Assemble(self.pc)
return nil
}