blob: 0fa997f06ece667bff053224b414cd1513874178 [file] [log] [blame]
// 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 flags implements top-level flags and the usage message for the assembler.
package flags
import (
"flag"
"fmt"
"os"
"path/filepath"
"strings"
)
var (
Debug = flag.Bool("debug", false, "dump instructions as they are parsed")
OutputFile = flag.String("o", "", "output file; default foo.6 for /a/b/c/foo.s on amd64")
PrintOut = flag.Bool("S", false, "print assembly and machine code")
TrimPath = flag.String("trimpath", "", "remove prefix from recorded source file paths")
Shared = flag.Bool("shared", false, "generate code that can be linked into a shared library")
)
var (
D MultiFlag
I MultiFlag
)
func init() {
flag.Var(&D, "D", "predefined symbol with optional simple value -D=identifer=value; can be set multiple times")
flag.Var(&I, "I", "include directory; can be set multiple times")
}
// MultiFlag allows setting a value multiple times to collect a list, as in -I=dir1 -I=dir2.
type MultiFlag []string
func (m *MultiFlag) String() string {
return fmt.Sprint(*m)
}
func (m *MultiFlag) Set(val string) error {
(*m) = append(*m, val)
return nil
}
func Usage() {
fmt.Fprintf(os.Stderr, "usage: asm [options] file.s\n")
fmt.Fprintf(os.Stderr, "Flags:\n")
flag.PrintDefaults()
os.Exit(2)
}
func Parse(theChar int) {
flag.Usage = Usage
flag.Parse()
if flag.NArg() != 1 {
flag.Usage()
}
// Flag refinement.
if *OutputFile == "" {
input := filepath.Base(flag.Arg(0))
if strings.HasSuffix(input, ".s") {
input = input[:len(input)-2]
}
*OutputFile = fmt.Sprintf("%s.%c", input, theChar)
}
}