func setArch(goarch string) (*arch.Arch, *obj.Link) { os.Setenv("GOOS", "linux") // obj can handle this OS for all architectures. architecture := arch.Set(goarch) if architecture == nil { panic("asm: unrecognized architecture " + goarch) } return architecture, obj.Linknew(architecture.LinkArch) }
func main() { log.SetFlags(0) log.SetPrefix("asm: ") GOARCH := obj.Getgoarch() architecture := arch.Set(GOARCH) if architecture == nil { log.Fatalf("unrecognized architecture %s", GOARCH) } flags.Parse() ctxt := obj.Linknew(architecture.LinkArch) if *flags.PrintOut { ctxt.Debugasm = 1 } ctxt.LineHist.TrimPathPrefix = *flags.TrimPath ctxt.Flag_dynlink = *flags.Dynlink ctxt.Flag_shared = *flags.Shared || *flags.Dynlink ctxt.Bso = bufio.NewWriter(os.Stdout) defer ctxt.Bso.Flush() // Create object file, write header. out, err := os.Create(*flags.OutputFile) if err != nil { log.Fatal(err) } defer bio.MustClose(out) buf := bufio.NewWriter(bio.MustWriter(out)) fmt.Fprintf(buf, "go object %s %s %s\n", obj.Getgoos(), obj.Getgoarch(), obj.Getgoversion()) fmt.Fprintf(buf, "!\n") lexer := lex.NewLexer(flag.Arg(0), ctxt) parser := asm.NewParser(ctxt, architecture, lexer) diag := false ctxt.DiagFunc = func(format string, args ...interface{}) { diag = true log.Printf(format, args...) } pList := obj.Linknewplist(ctxt) var ok bool pList.Firstpc, ok = parser.Parse() if ok { // reports errors to parser.Errorf obj.Writeobjdirect(ctxt, buf) } if !ok || diag { log.Printf("assembly of %s failed", flag.Arg(0)) os.Remove(*flags.OutputFile) os.Exit(1) } buf.Flush() }
func main() { log.SetFlags(0) log.SetPrefix("asm: ") GOARCH := obj.Getgoarch() architecture := arch.Set(GOARCH) if architecture == nil { log.Fatalf("asm: unrecognized architecture %s", GOARCH) } flags.Parse(architecture.Thechar) // Create object file, write header. fd, err := os.Create(*flags.OutputFile) if err != nil { log.Fatal(err) } ctxt := obj.Linknew(architecture.LinkArch) if *flags.PrintOut { ctxt.Debugasm = 1 } ctxt.LineHist.TrimPathPrefix = *flags.TrimPath ctxt.Flag_dynlink = *flags.Dynlink if *flags.Shared || *flags.Dynlink { ctxt.Flag_shared = 1 } ctxt.Bso = obj.Binitw(os.Stdout) defer ctxt.Bso.Flush() ctxt.Diag = log.Fatalf output := obj.Binitw(fd) fmt.Fprintf(output, "go object %s %s %s\n", obj.Getgoos(), obj.Getgoarch(), obj.Getgoversion()) fmt.Fprintf(output, "!\n") lexer := lex.NewLexer(flag.Arg(0), ctxt) parser := asm.NewParser(ctxt, architecture, lexer) pList := obj.Linknewplist(ctxt) var ok bool pList.Firstpc, ok = parser.Parse() if !ok { log.Printf("asm: assembly of %s failed", flag.Arg(0)) os.Remove(*flags.OutputFile) os.Exit(1) } obj.Writeobjdirect(ctxt, output) output.Flush() }
func Fuzz(data []byte) int { f, err := ioutil.TempFile("", "fuzz.asm") if err != nil { return 0 } defer os.Remove(f.Name()) defer f.Close() _, err = f.Write(data) if err != nil { return 0 } f.Close() defer func() { if x := recover(); x != nil { if str, ok := x.(string); ok && str == "os.Exit" { return } panic(x) } }() const GOARCH = "amd64" architecture := arch.Set(GOARCH) fd := new(bytes.Buffer) ctxt := obj.Linknew(architecture.LinkArch) // Try to vary these and other flags: // ctxt.Flag_dynlink // ctxt.Flag_shared ctxt.Bso = obj.Binitw(new(bytes.Buffer)) defer ctxt.Bso.Flush() ctxt.Diag = func(format string, v ...interface{}) { panic("os.Exit") } output := obj.Binitw(fd) lexer := lex.NewLexer(f.Name(), ctxt) parser := asm.NewParser(ctxt, architecture, lexer) pList := obj.Linknewplist(ctxt) var ok bool pList.Firstpc, ok = parser.Parse() if !ok { return 0 } obj.Writeobjdirect(ctxt, output) output.Flush() return 1 }