| // Copyright 2015 The Go Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style |
| // license that can be found in the LICENSE file. |
| |
| package obj |
| |
| import ( |
| "bytes" |
| "fmt" |
| "log" |
| "os" |
| "strings" |
| "time" |
| ) |
| |
| const REG_NONE = 0 |
| |
| var start time.Time |
| |
| func Cputime() float64 { |
| if start.IsZero() { |
| start = time.Now() |
| } |
| return time.Since(start).Seconds() |
| } |
| |
| func envOr(key, value string) string { |
| if x := os.Getenv(key); x != "" { |
| return x |
| } |
| return value |
| } |
| |
| var ( |
| GOROOT = envOr("GOROOT", defaultGOROOT) |
| GOARCH = envOr("GOARCH", defaultGOARCH) |
| GOOS = envOr("GOOS", defaultGOOS) |
| GO386 = envOr("GO386", defaultGO386) |
| GOARM = goarm() |
| Version = version |
| ) |
| |
| func goarm() int { |
| switch v := envOr("GOARM", defaultGOARM); v { |
| case "5": |
| return 5 |
| case "6": |
| return 6 |
| case "7": |
| return 7 |
| } |
| // Fail here, rather than validate at multiple call sites. |
| log.Fatalf("Invalid GOARM value. Must be 5, 6, or 7.") |
| panic("unreachable") |
| } |
| |
| func Getgoextlinkenabled() string { |
| return envOr("GO_EXTLINK_ENABLED", defaultGO_EXTLINK_ENABLED) |
| } |
| |
| func (p *Prog) Line() string { |
| return p.Ctxt.LineHist.LineString(int(p.Lineno)) |
| } |
| |
| var armCondCode = []string{ |
| ".EQ", |
| ".NE", |
| ".CS", |
| ".CC", |
| ".MI", |
| ".PL", |
| ".VS", |
| ".VC", |
| ".HI", |
| ".LS", |
| ".GE", |
| ".LT", |
| ".GT", |
| ".LE", |
| "", |
| ".NV", |
| } |
| |
| /* ARM scond byte */ |
| const ( |
| C_SCOND = (1 << 4) - 1 |
| C_SBIT = 1 << 4 |
| C_PBIT = 1 << 5 |
| C_WBIT = 1 << 6 |
| C_FBIT = 1 << 7 |
| C_UBIT = 1 << 7 |
| C_SCOND_XOR = 14 |
| ) |
| |
| // CConv formats ARM condition codes. |
| func CConv(s uint8) string { |
| if s == 0 { |
| return "" |
| } |
| sc := armCondCode[(s&C_SCOND)^C_SCOND_XOR] |
| if s&C_SBIT != 0 { |
| sc += ".S" |
| } |
| if s&C_PBIT != 0 { |
| sc += ".P" |
| } |
| if s&C_WBIT != 0 { |
| sc += ".W" |
| } |
| if s&C_UBIT != 0 { /* ambiguous with FBIT */ |
| sc += ".U" |
| } |
| return sc |
| } |
| |
| func (p *Prog) String() string { |
| if p == nil { |
| return "<nil Prog>" |
| } |
| |
| if p.Ctxt == nil { |
| return "<Prog without ctxt>" |
| } |
| |
| sc := CConv(p.Scond) |
| |
| var buf bytes.Buffer |
| |
| fmt.Fprintf(&buf, "%.5d (%v)\t%v%s", p.Pc, p.Line(), p.As, sc) |
| sep := "\t" |
| quadOpAmd64 := p.RegTo2 == -1 |
| if quadOpAmd64 { |
| fmt.Fprintf(&buf, "%s$%d", sep, p.From3.Offset) |
| sep = ", " |
| } |
| if p.From.Type != TYPE_NONE { |
| fmt.Fprintf(&buf, "%s%v", sep, Dconv(p, &p.From)) |
| sep = ", " |
| } |
| if p.Reg != REG_NONE { |
| // Should not happen but might as well show it if it does. |
| fmt.Fprintf(&buf, "%s%v", sep, Rconv(int(p.Reg))) |
| sep = ", " |
| } |
| if p.From3Type() != TYPE_NONE { |
| if p.From3.Type == TYPE_CONST && p.As == ATEXT { |
| // Special case - omit $. |
| fmt.Fprintf(&buf, "%s%d", sep, p.From3.Offset) |
| } else if quadOpAmd64 { |
| fmt.Fprintf(&buf, "%s%v", sep, Rconv(int(p.From3.Reg))) |
| } else { |
| fmt.Fprintf(&buf, "%s%v", sep, Dconv(p, p.From3)) |
| } |
| sep = ", " |
| } |
| if p.To.Type != TYPE_NONE { |
| fmt.Fprintf(&buf, "%s%v", sep, Dconv(p, &p.To)) |
| } |
| if p.RegTo2 != REG_NONE && !quadOpAmd64 { |
| fmt.Fprintf(&buf, "%s%v", sep, Rconv(int(p.RegTo2))) |
| } |
| return buf.String() |
| } |
| |
| func (ctxt *Link) NewProg() *Prog { |
| var p *Prog |
| if i := ctxt.allocIdx; i < len(ctxt.progs) { |
| p = &ctxt.progs[i] |
| ctxt.allocIdx = i + 1 |
| } else { |
| p = new(Prog) // should be the only call to this; all others should use ctxt.NewProg |
| } |
| p.Ctxt = ctxt |
| return p |
| } |
| func (ctxt *Link) freeProgs() { |
| s := ctxt.progs[:ctxt.allocIdx] |
| for i := range s { |
| s[i] = Prog{} |
| } |
| ctxt.allocIdx = 0 |
| } |
| |
| func (ctxt *Link) Line(n int) string { |
| return ctxt.LineHist.LineString(n) |
| } |
| |
| func Getcallerpc(interface{}) uintptr { |
| return 1 |
| } |
| |
| func (ctxt *Link) Dconv(a *Addr) string { |
| return Dconv(nil, a) |
| } |
| |
| func Dconv(p *Prog, a *Addr) string { |
| var str string |
| |
| switch a.Type { |
| default: |
| str = fmt.Sprintf("type=%d", a.Type) |
| |
| case TYPE_NONE: |
| str = "" |
| if a.Name != NAME_NONE || a.Reg != 0 || a.Sym != nil { |
| str = fmt.Sprintf("%v(%v)(NONE)", Mconv(a), Rconv(int(a.Reg))) |
| } |
| |
| case TYPE_REG: |
| // TODO(rsc): This special case is for x86 instructions like |
| // PINSRQ CX,$1,X6 |
| // where the $1 is included in the p->to Addr. |
| // Move into a new field. |
| if a.Offset != 0 { |
| str = fmt.Sprintf("$%d,%v", a.Offset, Rconv(int(a.Reg))) |
| break |
| } |
| |
| str = Rconv(int(a.Reg)) |
| if a.Name != NAME_NONE || a.Sym != nil { |
| str = fmt.Sprintf("%v(%v)(REG)", Mconv(a), Rconv(int(a.Reg))) |
| } |
| |
| case TYPE_BRANCH: |
| if a.Sym != nil { |
| str = fmt.Sprintf("%s(SB)", a.Sym.Name) |
| } else if p != nil && p.Pcond != nil { |
| str = fmt.Sprint(p.Pcond.Pc) |
| } else if a.Val != nil { |
| str = fmt.Sprint(a.Val.(*Prog).Pc) |
| } else { |
| str = fmt.Sprintf("%d(PC)", a.Offset) |
| } |
| |
| case TYPE_INDIR: |
| str = fmt.Sprintf("*%s", Mconv(a)) |
| |
| case TYPE_MEM: |
| str = Mconv(a) |
| if a.Index != REG_NONE { |
| str += fmt.Sprintf("(%v*%d)", Rconv(int(a.Index)), int(a.Scale)) |
| } |
| |
| case TYPE_CONST: |
| if a.Reg != 0 { |
| str = fmt.Sprintf("$%v(%v)", Mconv(a), Rconv(int(a.Reg))) |
| } else { |
| str = fmt.Sprintf("$%v", Mconv(a)) |
| } |
| |
| case TYPE_TEXTSIZE: |
| if a.Val.(int32) == ArgsSizeUnknown { |
| str = fmt.Sprintf("$%d", a.Offset) |
| } else { |
| str = fmt.Sprintf("$%d-%d", a.Offset, a.Val.(int32)) |
| } |
| |
| case TYPE_FCONST: |
| str = fmt.Sprintf("%.17g", a.Val.(float64)) |
| // Make sure 1 prints as 1.0 |
| if !strings.ContainsAny(str, ".e") { |
| str += ".0" |
| } |
| str = fmt.Sprintf("$(%s)", str) |
| |
| case TYPE_SCONST: |
| str = fmt.Sprintf("$%q", a.Val.(string)) |
| |
| case TYPE_ADDR: |
| str = fmt.Sprintf("$%s", Mconv(a)) |
| |
| case TYPE_SHIFT: |
| v := int(a.Offset) |
| ops := "<<>>->@>" |
| switch GOARCH { |
| case "arm": |
| op := ops[((v>>5)&3)<<1:] |
| if v&(1<<4) != 0 { |
| str = fmt.Sprintf("R%d%c%cR%d", v&15, op[0], op[1], (v>>8)&15) |
| } else { |
| str = fmt.Sprintf("R%d%c%c%d", v&15, op[0], op[1], (v>>7)&31) |
| } |
| if a.Reg != 0 { |
| str += fmt.Sprintf("(%v)", Rconv(int(a.Reg))) |
| } |
| case "arm64": |
| op := ops[((v>>22)&3)<<1:] |
| str = fmt.Sprintf("R%d%c%c%d", (v>>16)&31, op[0], op[1], (v>>10)&63) |
| default: |
| panic("TYPE_SHIFT is not supported on " + GOARCH) |
| } |
| |
| case TYPE_REGREG: |
| str = fmt.Sprintf("(%v, %v)", Rconv(int(a.Reg)), Rconv(int(a.Offset))) |
| |
| case TYPE_REGREG2: |
| str = fmt.Sprintf("%v, %v", Rconv(int(a.Reg)), Rconv(int(a.Offset))) |
| |
| case TYPE_REGLIST: |
| str = regListConv(int(a.Offset)) |
| } |
| |
| return str |
| } |
| |
| func Mconv(a *Addr) string { |
| var str string |
| |
| switch a.Name { |
| default: |
| str = fmt.Sprintf("name=%d", a.Name) |
| |
| case NAME_NONE: |
| switch { |
| case a.Reg == REG_NONE: |
| str = fmt.Sprint(a.Offset) |
| case a.Offset == 0: |
| str = fmt.Sprintf("(%v)", Rconv(int(a.Reg))) |
| case a.Offset != 0: |
| str = fmt.Sprintf("%d(%v)", a.Offset, Rconv(int(a.Reg))) |
| } |
| |
| case NAME_EXTERN: |
| if a.Sym != nil { |
| str = fmt.Sprintf("%s%s(SB)", a.Sym.Name, offConv(a.Offset)) |
| } else { |
| str = fmt.Sprintf("%s(SB)", offConv(a.Offset)) |
| } |
| |
| case NAME_GOTREF: |
| if a.Sym != nil { |
| str = fmt.Sprintf("%s%s@GOT(SB)", a.Sym.Name, offConv(a.Offset)) |
| } else { |
| str = fmt.Sprintf("%s@GOT(SB)", offConv(a.Offset)) |
| } |
| |
| case NAME_STATIC: |
| if a.Sym != nil { |
| str = fmt.Sprintf("%s<>%s(SB)", a.Sym.Name, offConv(a.Offset)) |
| } else { |
| str = fmt.Sprintf("<>%s(SB)", offConv(a.Offset)) |
| } |
| |
| case NAME_AUTO: |
| if a.Sym != nil { |
| str = fmt.Sprintf("%s%s(SP)", a.Sym.Name, offConv(a.Offset)) |
| } else { |
| str = fmt.Sprintf("%s(SP)", offConv(a.Offset)) |
| } |
| |
| case NAME_PARAM: |
| if a.Sym != nil { |
| str = fmt.Sprintf("%s%s(FP)", a.Sym.Name, offConv(a.Offset)) |
| } else { |
| str = fmt.Sprintf("%s(FP)", offConv(a.Offset)) |
| } |
| } |
| return str |
| } |
| |
| func offConv(off int64) string { |
| if off == 0 { |
| return "" |
| } |
| return fmt.Sprintf("%+d", off) |
| } |
| |
| type regSet struct { |
| lo int |
| hi int |
| Rconv func(int) string |
| } |
| |
| // Few enough architectures that a linear scan is fastest. |
| // Not even worth sorting. |
| var regSpace []regSet |
| |
| /* |
| Each architecture defines a register space as a unique |
| integer range. |
| Here is the list of architectures and the base of their register spaces. |
| */ |
| |
| const ( |
| // Because of masking operations in the encodings, each register |
| // space should start at 0 modulo some power of 2. |
| RBase386 = 1 * 1024 |
| RBaseAMD64 = 2 * 1024 |
| RBaseARM = 3 * 1024 |
| RBasePPC64 = 4 * 1024 // range [4k, 8k) |
| RBaseARM64 = 8 * 1024 // range [8k, 13k) |
| RBaseMIPS = 13 * 1024 // range [13k, 14k) |
| RBaseS390X = 14 * 1024 // range [14k, 15k) |
| ) |
| |
| // RegisterRegister binds a pretty-printer (Rconv) for register |
| // numbers to a given register number range. Lo is inclusive, |
| // hi exclusive (valid registers are lo through hi-1). |
| func RegisterRegister(lo, hi int, Rconv func(int) string) { |
| regSpace = append(regSpace, regSet{lo, hi, Rconv}) |
| } |
| |
| func Rconv(reg int) string { |
| if reg == REG_NONE { |
| return "NONE" |
| } |
| for i := range regSpace { |
| rs := ®Space[i] |
| if rs.lo <= reg && reg < rs.hi { |
| return rs.Rconv(reg) |
| } |
| } |
| return fmt.Sprintf("R???%d", reg) |
| } |
| |
| func regListConv(list int) string { |
| str := "" |
| |
| for i := 0; i < 16; i++ { // TODO: 16 is ARM-specific. |
| if list&(1<<uint(i)) != 0 { |
| if str == "" { |
| str += "[" |
| } else { |
| str += "," |
| } |
| // This is ARM-specific; R10 is g. |
| if i == 10 { |
| str += "g" |
| } else { |
| str += fmt.Sprintf("R%d", i) |
| } |
| } |
| } |
| |
| str += "]" |
| return str |
| } |
| |
| type opSet struct { |
| lo As |
| names []string |
| } |
| |
| // Not even worth sorting |
| var aSpace []opSet |
| |
| // RegisterOpcode binds a list of instruction names |
| // to a given instruction number range. |
| func RegisterOpcode(lo As, Anames []string) { |
| if len(Anames) > AllowedOpCodes { |
| panic(fmt.Sprintf("too many instructions, have %d max %d", len(Anames), AllowedOpCodes)) |
| } |
| aSpace = append(aSpace, opSet{lo, Anames}) |
| } |
| |
| func (a As) String() string { |
| if 0 <= a && int(a) < len(Anames) { |
| return Anames[a] |
| } |
| for i := range aSpace { |
| as := &aSpace[i] |
| if as.lo <= a && int(a-as.lo) < len(as.names) { |
| return as.names[a-as.lo] |
| } |
| } |
| return fmt.Sprintf("A???%d", a) |
| } |
| |
| var Anames = []string{ |
| "XXX", |
| "CALL", |
| "DUFFCOPY", |
| "DUFFZERO", |
| "END", |
| "FUNCDATA", |
| "JMP", |
| "NOP", |
| "PCDATA", |
| "RET", |
| "TEXT", |
| "TYPE", |
| "UNDEF", |
| "USEFIELD", |
| "VARDEF", |
| "VARKILL", |
| "VARLIVE", |
| } |
| |
| func Bool2int(b bool) int { |
| // The compiler currently only optimizes this form. |
| // See issue 6011. |
| var i int |
| if b { |
| i = 1 |
| } else { |
| i = 0 |
| } |
| return i |
| } |