mirror of
https://github.com/golang/go
synced 2024-10-05 01:31:22 -06:00
ed238ca4e5
Fixes #3154. R=golang-dev, bradfitz, rsc CC=golang-dev https://golang.org/cl/5707052
118 lines
2.3 KiB
Go
118 lines
2.3 KiB
Go
// Copyright 2009 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 (
|
|
"fmt"
|
|
"go/token"
|
|
"io/ioutil"
|
|
"os"
|
|
"os/exec"
|
|
)
|
|
|
|
// run runs the command argv, feeding in stdin on standard input.
|
|
// It returns the output to standard output and standard error.
|
|
// ok indicates whether the command exited successfully.
|
|
func run(stdin []byte, argv []string) (stdout, stderr []byte, ok bool) {
|
|
cmd, err := exec.LookPath(argv[0])
|
|
if err != nil {
|
|
fatalf("exec %s: %s", argv[0], err)
|
|
}
|
|
r0, w0, err := os.Pipe()
|
|
if err != nil {
|
|
fatalf("%s", err)
|
|
}
|
|
r1, w1, err := os.Pipe()
|
|
if err != nil {
|
|
fatalf("%s", err)
|
|
}
|
|
r2, w2, err := os.Pipe()
|
|
if err != nil {
|
|
fatalf("%s", err)
|
|
}
|
|
p, err := os.StartProcess(cmd, argv, &os.ProcAttr{Files: []*os.File{r0, w1, w2}})
|
|
if err != nil {
|
|
fatalf("%s", err)
|
|
}
|
|
r0.Close()
|
|
w1.Close()
|
|
w2.Close()
|
|
c := make(chan bool)
|
|
go func() {
|
|
w0.Write(stdin)
|
|
w0.Close()
|
|
c <- true
|
|
}()
|
|
go func() {
|
|
stdout, _ = ioutil.ReadAll(r1)
|
|
r1.Close()
|
|
c <- true
|
|
}()
|
|
stderr, _ = ioutil.ReadAll(r2)
|
|
r2.Close()
|
|
<-c
|
|
<-c
|
|
|
|
state, err := p.Wait()
|
|
if err != nil {
|
|
fatalf("%s", err)
|
|
}
|
|
ok = state.Success()
|
|
return
|
|
}
|
|
|
|
func lineno(pos token.Pos) string {
|
|
return fset.Position(pos).String()
|
|
}
|
|
|
|
// Die with an error message.
|
|
func fatalf(msg string, args ...interface{}) {
|
|
// If we've already printed other errors, they might have
|
|
// caused the fatal condition. Assume they're enough.
|
|
if nerrors == 0 {
|
|
fmt.Fprintf(os.Stderr, msg+"\n", args...)
|
|
}
|
|
os.Exit(2)
|
|
}
|
|
|
|
var nerrors int
|
|
|
|
func error_(pos token.Pos, msg string, args ...interface{}) {
|
|
nerrors++
|
|
if pos.IsValid() {
|
|
fmt.Fprintf(os.Stderr, "%s: ", fset.Position(pos).String())
|
|
}
|
|
fmt.Fprintf(os.Stderr, msg, args...)
|
|
fmt.Fprintf(os.Stderr, "\n")
|
|
}
|
|
|
|
// isName returns true if s is a valid C identifier
|
|
func isName(s string) bool {
|
|
for i, v := range s {
|
|
if v != '_' && (v < 'A' || v > 'Z') && (v < 'a' || v > 'z') && (v < '0' || v > '9') {
|
|
return false
|
|
}
|
|
if i == 0 && '0' <= v && v <= '9' {
|
|
return false
|
|
}
|
|
}
|
|
return s != ""
|
|
}
|
|
|
|
func creat(name string) *os.File {
|
|
f, err := os.Create(name)
|
|
if err != nil {
|
|
fatalf("%s", err)
|
|
}
|
|
return f
|
|
}
|
|
|
|
func slashToUnderscore(c rune) rune {
|
|
if c == '/' || c == '\\' || c == ':' {
|
|
c = '_'
|
|
}
|
|
return c
|
|
}
|