bb

README

Go Busybox

bb.go in this package implements a Go source-to-source transformation on pure Go code (no cgo).

This AST transformation does the following:

• Takes a Go command's source files and rewrites them into Go package files without global side effects.
• Writes a main.go file with a main() that calls into the appropriate Go command package based on argv[0].

This allows you to take two Go commands, such as Go implementations of sl and cowsay and compile them into one binary.

Which command is invoked is determined by argv[0] or argv[1] if argv[0] is not recognized. Let's say bb is the compiled binary; the following are equivalent invocations of sl and cowsay:

# Make a symlink sl -> bb
ln -s bb sl
./sl -l

# Make a symlink cowsay -> bb
ln -s bb cowsay
./cowsay Haha

./bb sl -l
./bb cowsay Haha

AST Transformation

Principally, the AST transformation moves all global side-effects into callable package functions. E.g. main becomes Main, each init becomes InitN, and global variable assignments are moved into their own InitN.

Then, these Main and Init functions can be registered with a global map of commands by name and used when called upon.

Let's say a command github.com/org/repo/cmds/sl contains the following main.go:

package main

import (
  "flag"
  "log"
)

var name = flag.String("name", "", "Gimme name")

func init() {
  log.Printf("init %s", *name)
}

func main() {
  log.Printf("train")
}

This would be rewritten to be:

package sl // based on the directory name or bazel-rule go_binary name

import (
  "flag"
  "log"

  // This package holds the global map of commands.
  "github.com/u-root/u-root/pkg/bb"
)

// Type has to be inferred through type checking.
var name *string

func Init0() {
  log.Printf("init %s", *name)
}

func Init1() {
  name = flag.String("name", "", "Gimme name")
}

func Init() {
  // Order is determined by go/types.Info.InitOrder.
  Init1()
  Init0()
}

func Main() {
  log.Printf("train")
}

func init() {
  // Register `sl` as a command.
  bb.Register("sl", Init, Main)
}

Shortcomings

• If there is already a function Main or InitN for some N, there may be a compilation error.
• Any packages imported by commands may still have global side-effects affecting other commands. Done properly, we would have to rewrite all non-standard-library packages as well as commands. This has not been necessary to implement so far. It would likely be necessary if two different imported packages register the same flag unconditionally globally.

Generated main

The main file can be generated based on any template Go files, but the default looks something like the following:

import (
  "os"

  "github.com/u-root/u-root/pkg/bb"

  // Side-effect import registers command with bb.
  _ "github.com/org/repo/cmds/generated/sl"
)

func main() {
  bb.Run(os.Argv[0])
}

The default template will use argv[1] if argv[0] is not in the map.

Documentation

Overview

Package bb implements a Go source-to-source transformation on pure Go code.

This AST transformation does the following:

• Takes a Go command's source files and rewrites them into Go package files without global side effects.
• Writes a `main.go` file with a `main()` that calls into the appropriate Go command package based on `argv[0]`.

This allows you to take two Go commands, such as Go implementations of `sl` and `cowsay` and compile them into one binary.

Which command is invoked is determined by `argv[0]` or `argv[1]` if `argv[0]` is not recognized.

Principally, the AST transformation moves all global side-effects into callable package functions. E.g. `main` becomes `Main`, each `init` becomes `InitN`, and global variable assignments are moved into their own `InitN`.

Index

• func BuildBusybox(env golang.Environ, pkgs []string, binaryPath string) error
• func CreateBBMainSource(fset *token.FileSet, astp *ast.Package, pkgs []string, destDir string) error
• func ParseAST(files []string) (*token.FileSet, *ast.Package, error)
• func RewritePackage(env golang.Environ, pkgPath, bbImportPath string, importer types.Importer) error
• func SrcFiles(p *build.Package) []string
• type Package
  • func NewPackage(name string, pkgPath string, srcFiles []string, importer types.Importer) (*Package, error)
  • func NewPackageFromEnv(env golang.Environ, importPath string, importer types.Importer) (*Package, error)
  • func (p *Package) Rewrite(destDir, bbImportPath string) error

Functions

func BuildBusybox

func BuildBusybox(env golang.Environ, pkgs []string, binaryPath string) error

BuildBusybox builds a busybox of the given Go packages.

pkgs is a list of Go import paths. If nil is returned, binaryPath will hold the busybox-style binary.

func CreateBBMainSource

func CreateBBMainSource(fset *token.FileSet, astp *ast.Package, pkgs []string, destDir string) error

CreateBBMainSource creates a bb Go command that imports all given pkgs.

p must be the bb template.

• For each pkg in pkgs, add import _ "pkg" to astp's first file.
• Write source file out to destDir.

func ParseAST

func ParseAST(files []string) (*token.FileSet, *ast.Package, error)

ParseAST parses the given files for a package named main.

Only files with a matching package statement will be part of the AST returned.

func RewritePackage

func RewritePackage(env golang.Environ, pkgPath, bbImportPath string, importer types.Importer) error

RewritePackage rewrites pkgPath to be bb-mode compatible, where destDir is the file system destination of the written files and bbImportPath is the Go import path of the bb package to register with.

func SrcFiles

func SrcFiles(p *build.Package) []string

Types

type Package

type Package struct {
	// Name is the command name.
	//
	// In the standard Go tool chain, this is usually the base name of the
	// directory containing its source files.
	Name string
	// contains filtered or unexported fields
}

Package is a Go package.

It holds AST, type, file, and Go package information about a Go package.

func NewPackage

func NewPackage(name string, pkgPath string, srcFiles []string, importer types.Importer) (*Package, error)

NewPackage gathers AST, type, and token information about a command.

The given importer is used to resolve dependencies.

func NewPackageFromEnv

func NewPackageFromEnv(env golang.Environ, importPath string, importer types.Importer) (*Package, error)

NewPackageFromEnv finds the package identified by importPath, and gathers AST, type, and token information.

func (*Package) Rewrite

func (p *Package) Rewrite(destDir, bbImportPath string) error

Rewrite rewrites p into destDir as a bb package using bbImportPath for the bb implementation.