blob: 3bb75be1d449a2f3de242e91e2e9675dceaeec40 [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 main
import (
"archive/tar"
"bufio"
"bytes"
"compress/gzip"
"context"
"crypto/sha1"
"errors"
"flag"
"fmt"
"io"
"log"
"os"
"os/exec"
"path/filepath"
"sort"
"strings"
"golang.org/x/build/buildlet"
)
func push(args []string) error {
fs := flag.NewFlagSet("push", flag.ContinueOnError)
var dryRun bool
fs.BoolVar(&dryRun, "dry-run", false, "print what would be done only")
fs.Usage = func() {
fmt.Fprintln(os.Stderr, "push usage: gomote push <instance>")
fs.PrintDefaults()
os.Exit(1)
}
fs.Parse(args)
goroot := os.Getenv("GOROOT")
if goroot == "" {
slurp, err := exec.Command("go", "env", "GOROOT").Output()
if err != nil {
return fmt.Errorf("failed to get GOROOT from go env: %v", err)
}
goroot = strings.TrimSpace(string(slurp))
if goroot == "" {
return errors.New("Failed to get $GOROOT from environment or go env")
}
}
if fs.NArg() != 1 {
fs.Usage()
}
name := fs.Arg(0)
bc, conf, err := clientAndConf(name)
if err != nil {
return err
}
haveGo14 := false
remote := map[string]buildlet.DirEntry{} // keys like "src/make.bash"
lsOpts := buildlet.ListDirOpts{
Recursive: true,
Digest: true,
Skip: []string{
// Ignore binary output directories:
"go/pkg", "go/bin",
// We don't care about the digest of
// particular source files for Go 1.4. And
// exclude /pkg. This leaves go1.4/bin, which
// is enough to know whether we have Go 1.4 or
// not.
"go1.4/src", "go1.4/pkg",
},
}
ctx := context.Background()
if err := bc.ListDir(ctx, ".", lsOpts, func(ent buildlet.DirEntry) {
name := ent.Name()
if strings.HasPrefix(name, "go1.4/") {
haveGo14 = true
return
}
if strings.HasPrefix(name, "go/") {
remote[name[len("go/"):]] = ent
}
}); err != nil {
return fmt.Errorf("error listing buildlet's existing files: %v", err)
}
if !haveGo14 {
if u := conf.GoBootstrapURL(buildEnv); u != "" {
log.Printf("installing go1.4")
if dryRun {
log.Printf("(Dry-run) Would have pushed go1.4")
} else {
if err := bc.PutTarFromURL(ctx, u, "go1.4"); err != nil {
return err
}
}
}
}
// Invoke 'git check-ignore' and use it to query whether paths have been gitignored.
// If anything goes wrong at any point, fall back to assuming that nothing is gitignored.
var isGitIgnored func(string) bool
gci := exec.Command("git", "check-ignore", "--stdin", "-n", "-v", "-z")
gci.Env = append(os.Environ(), "GIT_FLUSH=1")
gciIn, errIn := gci.StdinPipe()
defer gciIn.Close() // allow git process to exit
gciOut, errOut := gci.StdoutPipe()
errStart := gci.Start()
if errIn != nil || errOut != nil || errStart != nil {
isGitIgnored = func(string) bool { return false }
} else {
var failed bool
br := bufio.NewReader(gciOut)
isGitIgnored = func(path string) bool {
if failed {
return false
}
fmt.Fprintf(gciIn, "%s\x00", path)
// Response is of form "<source> <NULL> <linenum> <NULL> <pattern> <NULL> <pathname> <NULL>"
// Read all four and check that the path is correct.
// If so, the path is ignored iff the source (reason why ignored) is non-empty.
var resp [4][]byte
for i := range resp {
b, err := br.ReadBytes(0)
if err != nil {
failed = true
return false
}
resp[i] = b[:len(b)-1] // drop trailing NULL
}
// Sanity check
if string(resp[3]) != path {
panic("git check-ignore path did not roundtrip, got " + string(resp[3]) + " sent " + path)
}
return len(resp[0]) > 0
}
}
type fileInfo struct {
fi os.FileInfo
sha1 string // if regular file
}
local := map[string]fileInfo{} // keys like "src/make.bash"
if err := filepath.Walk(goroot, func(path string, fi os.FileInfo, err error) error {
if isEditorBackup(path) {
return nil
}
if err != nil {
return err
}
rel := strings.TrimPrefix(filepath.ToSlash(strings.TrimPrefix(path, goroot)), "/")
if rel == "" {
return nil
}
if fi.IsDir() {
switch rel {
case ".git", "pkg", "bin":
return filepath.SkipDir
}
}
inf := fileInfo{fi: fi}
if isGitIgnored(path) {
if fi.Mode().IsDir() {
return filepath.SkipDir
}
return nil
}
if fi.Mode().IsRegular() {
inf.sha1, err = fileSHA1(path)
if err != nil {
return err
}
}
local[rel] = inf
return nil
}); err != nil {
return fmt.Errorf("error enumerating local GOROOT files: %v", err)
}
var toDel []string
for rel := range remote {
if rel == "VERSION" {
// Don't delete this. It's harmless, and
// necessary. Clients can overwrite it if they
// want. But if there's no VERSION file there,
// make.bash/bat assumes there's a git repo in
// place, but there's not only not a git repo
// there with gomote, but there's no git tool
// available either.
continue
}
// Also don't delete the auto-generated files from cmd/dist.
// Otherwise gomote users can't gomote push + gomote run make.bash
// and then iteratively:
// -- hack locally
// -- gomote push
// -- gomote run go test -v ...
// Because the go test would fail remotely without
// these files if they were deleted by gomote push.
switch rel {
case "src/cmd/cgo/zdefaultcc.go",
"src/cmd/go/internal/cfg/zdefaultcc.go",
"src/cmd/go/internal/cfg/zosarch.go",
"src/cmd/internal/objabi/zbootstrap.go",
"src/go/build/zcgo.go",
"src/runtime/internal/sys/zversion.go":
continue
}
if isGitIgnored(rel) {
// Don't delete remote gitignored files; this breaks built toolchains.
continue
}
rel = strings.TrimRight(rel, "/")
if rel == "" {
continue
}
if _, ok := local[rel]; !ok {
toDel = append(toDel, rel)
}
}
if len(toDel) > 0 {
withGo := make([]string, len(toDel)) // with the "go/" prefix
for i, v := range toDel {
withGo[i] = "go/" + v
}
sort.Strings(withGo)
if dryRun {
log.Printf("(Dry-run) Would have deleted remote files: %q", withGo)
} else {
log.Printf("Deleting remote files: %q", withGo)
if err := bc.RemoveAll(ctx, withGo...); err != nil {
return fmt.Errorf("Deleting remote unwanted files: %v", err)
}
}
}
var toSend []string
notHave := 0
const maxNotHavePrint = 5
for rel, inf := range local {
switch rel {
case "VERSION.cache", "src/runtime/internal/sys/zversion.go", "src/cmd/internal/objabi/zbootstrap.go",
"src/go/build/zcgo.go":
continue
}
if !inf.fi.Mode().IsRegular() {
// TODO(bradfitz): this is only doing regular files
// for now, so empty directories, symlinks, etc aren't
// supported. revisit if that's a problem.
if !inf.fi.IsDir() {
log.Printf("Ignoring local non-regular, non-directory file %s: %v", rel, inf.fi.Mode())
}
continue
}
rem, ok := remote[rel]
if !ok {
if notHave++; notHave <= maxNotHavePrint {
log.Printf("Remote doesn't have %q", rel)
}
toSend = append(toSend, rel)
continue
}
if rem.Digest() != inf.sha1 {
log.Printf("Remote's %s digest is %q; want %q", rel, rem.Digest(), inf.sha1)
toSend = append(toSend, rel)
}
}
if notHave > maxNotHavePrint {
log.Printf("Remote doesn't have %d files (only showed %d).", notHave, maxNotHavePrint)
}
if _, hasVersion := remote["VERSION"]; !hasVersion {
log.Printf("Remote lacks a VERSION file; sending a fake one")
toSend = append(toSend, "VERSION")
}
if len(toSend) > 0 {
sort.Strings(toSend)
tgz, err := generateDeltaTgz(goroot, toSend)
if err != nil {
return err
}
log.Printf("Uploading %d new/changed files; %d byte .tar.gz", len(toSend), tgz.Len())
if dryRun {
log.Printf("(Dry-run mode; not doing anything.")
return nil
}
if err := bc.PutTar(ctx, tgz, "go"); err != nil {
return fmt.Errorf("writing tarball to buildlet: %v", err)
}
}
return nil
}
func isEditorBackup(path string) bool {
base := filepath.Base(path)
if strings.HasPrefix(base, ".") && strings.HasSuffix(base, ".swp") {
// vi
return true
}
if strings.HasSuffix(path, "~") || strings.HasSuffix(path, "#") ||
strings.HasPrefix(base, "#") || strings.HasPrefix(base, ".#") {
// emacs
return true
}
return false
}
// file is forward-slash separated
func generateDeltaTgz(goroot string, files []string) (*bytes.Buffer, error) {
var buf bytes.Buffer
zw := gzip.NewWriter(&buf)
tw := tar.NewWriter(zw)
for _, file := range files {
// Special.
if file == "VERSION" {
// TODO(bradfitz): a dummy VERSION file's contents to make things
// happy. Notably it starts with "devel ". Do we care about it
// being accurate beyond that?
version := "devel gomote.XXXXX"
if err := tw.WriteHeader(&tar.Header{
Name: "VERSION",
Mode: 0644,
Size: int64(len(version)),
}); err != nil {
return nil, err
}
if _, err := io.WriteString(tw, version); err != nil {
return nil, err
}
continue
}
f, err := os.Open(filepath.Join(goroot, file))
if err != nil {
return nil, err
}
fi, err := f.Stat()
if err != nil {
f.Close()
return nil, err
}
header, err := tar.FileInfoHeader(fi, "")
if err != nil {
f.Close()
return nil, err
}
header.Name = file // forward slash
if err := tw.WriteHeader(header); err != nil {
f.Close()
return nil, err
}
if _, err := io.CopyN(tw, f, header.Size); err != nil {
f.Close()
return nil, fmt.Errorf("error copying contents of %s: %v", file, err)
}
f.Close()
}
if err := tw.Close(); err != nil {
return nil, err
}
if err := zw.Close(); err != nil {
return nil, err
}
return &buf, nil
}
func fileSHA1(path string) (string, error) {
f, err := os.Open(path)
if err != nil {
return "", err
}
defer f.Close()
s1 := sha1.New()
if _, err := io.Copy(s1, f); err != nil {
return "", err
}
return fmt.Sprintf("%x", s1.Sum(nil)), nil
}