| // Copyright 2014 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 main |
| |
| import ( |
| "bufio" |
| "flag" |
| "fmt" |
| "log" |
| "os" |
| |
| "cmd/asm/internal/arch" |
| "cmd/asm/internal/asm" |
| "cmd/asm/internal/flags" |
| "cmd/asm/internal/lex" |
| |
| "cmd/internal/bio" |
| "cmd/internal/obj" |
| "cmd/internal/objabi" |
| ) |
| |
| func main() { |
| log.SetFlags(0) |
| log.SetPrefix("asm: ") |
| |
| GOARCH := objabi.GOARCH |
| |
| 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 = true |
| } |
| ctxt.Flag_dynlink = *flags.Dynlink |
| ctxt.Flag_shared = *flags.Shared || *flags.Dynlink |
| ctxt.Bso = bufio.NewWriter(os.Stdout) |
| defer ctxt.Bso.Flush() |
| |
| architecture.Init(ctxt) |
| |
| // 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", objabi.GOOS, objabi.GOARCH, objabi.Version) |
| fmt.Fprintf(buf, "!\n") |
| |
| var ok, diag bool |
| var failedFile string |
| for _, f := range flag.Args() { |
| lexer := lex.NewLexer(f) |
| parser := asm.NewParser(ctxt, architecture, lexer) |
| ctxt.DiagFunc = func(format string, args ...interface{}) { |
| diag = true |
| log.Printf(format, args...) |
| } |
| pList := new(obj.Plist) |
| pList.Firstpc, ok = parser.Parse() |
| if !ok { |
| failedFile = f |
| break |
| } |
| // reports errors to parser.Errorf |
| obj.Flushplist(ctxt, pList, nil) |
| } |
| if ok { |
| obj.WriteObjFile(ctxt, buf) |
| } |
| if !ok || diag { |
| if failedFile != "" { |
| log.Printf("assembly of %s failed", failedFile) |
| } else { |
| log.Print("assembly failed") |
| } |
| out.Close() |
| os.Remove(*flags.OutputFile) |
| os.Exit(1) |
| } |
| buf.Flush() |
| } |