mirror of
https://github.com/autc04/Retro68.git
synced 2024-09-08 20:55:01 +00:00
74 lines
1.3 KiB
Go
74 lines
1.3 KiB
Go
// Copyright 2017 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 (
|
|
"flag"
|
|
"fmt"
|
|
"log"
|
|
"os"
|
|
"strings"
|
|
|
|
"cmd/internal/buildid"
|
|
)
|
|
|
|
func usage() {
|
|
fmt.Fprintf(os.Stderr, "usage: go tool buildid [-w] file\n")
|
|
flag.PrintDefaults()
|
|
os.Exit(2)
|
|
}
|
|
|
|
var wflag = flag.Bool("w", false, "write build ID")
|
|
|
|
func main() {
|
|
log.SetPrefix("buildid: ")
|
|
log.SetFlags(0)
|
|
flag.Usage = usage
|
|
flag.Parse()
|
|
if flag.NArg() != 1 {
|
|
usage()
|
|
}
|
|
|
|
file := flag.Arg(0)
|
|
id, err := buildid.ReadFile(file)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
if !*wflag {
|
|
fmt.Printf("%s\n", id)
|
|
return
|
|
}
|
|
|
|
f, err := os.Open(file)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
matches, hash, err := buildid.FindAndHash(f, id, 0)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
f.Close()
|
|
|
|
tail := id
|
|
if i := strings.LastIndex(id, "."); i >= 0 {
|
|
tail = tail[i+1:]
|
|
}
|
|
if len(tail) != len(hash)*2 {
|
|
log.Fatalf("%s: cannot find %d-byte hash in id %s", file, len(hash), id)
|
|
}
|
|
newID := id[:len(id)-len(tail)] + fmt.Sprintf("%x", hash)
|
|
|
|
f, err = os.OpenFile(file, os.O_WRONLY, 0)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
if err := buildid.Rewrite(f, matches, newID); err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
if err := f.Close(); err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
}
|