123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803 |
- 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
- }
|