1
0
mirror of https://github.com/golang/go synced 2024-11-05 15:06:09 -07:00

go.tools/cmd/godex: replicate path logic for gccgo-new importer

LGTM=adonovan
R=adonovan
CC=golang-codereviews
https://golang.org/cl/81990043
This commit is contained in:
Robert Griesemer 2014-03-28 14:50:14 -07:00
parent 250eb11b9b
commit 1faba6e3b8

View File

@ -10,7 +10,10 @@ import (
"debug/elf"
"fmt"
"io"
"io/ioutil"
"os"
"path/filepath"
"strings"
"code.google.com/p/go.tools/go/gccgoimporter"
"code.google.com/p/go.tools/go/importer"
@ -24,31 +27,63 @@ func init() {
register("gccgo", inst.GetImporter(nil))
// importer for gccgo using condensed export format (experimental)
register("gccgo-new", gccgoNewImporter)
register("gccgo-new", getNewImporter(append(inst.SearchPaths(), ".")))
}
func gccgoNewImporter(packages map[string]*types.Package, path string) (*types.Package, error) {
reader, closer, err := openGccgoExportFile(path)
// This function is an adjusted variant of gccgoimporter.GccgoInstallation.GetImporter.
func getNewImporter(searchpaths []string) types.Importer {
return func(imports map[string]*types.Package, pkgpath string) (pkg *types.Package, err error) {
if pkgpath == "unsafe" {
return types.Unsafe, nil
}
fpath, err := findExportFile(searchpaths, pkgpath)
if err != nil {
return
}
reader, closer, err := openExportFile(fpath)
if err != nil {
return nil, err
}
defer closer.Close()
// TODO(gri) importer.ImportData takes a []byte instead of an io.Reader;
// hence the need to read some amount of data. At the same time we don't
// want to read the entire, potentially very large object file. For now,
// read 10K. Fix this!
var data = make([]byte, 10<<10)
n, err := reader.Read(data)
// TODO(gri) At the moment we just read the entire file.
// We should change importer.ImportData to take an io.Reader instead.
data, err := ioutil.ReadAll(reader)
if err != nil && err != io.EOF {
return nil, err
}
return importer.ImportData(packages, data[:n])
return importer.ImportData(packages, data)
}
}
// openGccgoExportFile was copied from gccgoimporter.
func openGccgoExportFile(fpath string) (reader io.ReadSeeker, closer io.Closer, err error) {
// This function is an exact copy of gccgoimporter.findExportFile.
func findExportFile(searchpaths []string, pkgpath string) (string, error) {
for _, spath := range searchpaths {
pkgfullpath := filepath.Join(spath, pkgpath)
pkgdir, name := filepath.Split(pkgfullpath)
for _, filepath := range [...]string{
pkgfullpath,
pkgfullpath + ".gox",
pkgdir + "lib" + name + ".so",
pkgdir + "lib" + name + ".a",
pkgfullpath + ".o",
} {
fi, err := os.Stat(filepath)
if err == nil && !fi.IsDir() {
return filepath, nil
}
}
}
return "", fmt.Errorf("%s: could not find export data (tried %s)", pkgpath, strings.Join(searchpaths, ":"))
}
// This function is an exact copy of gccgoimporter.openExportFile.
func openExportFile(fpath string) (reader io.ReadSeeker, closer io.Closer, err error) {
f, err := os.Open(fpath)
if err != nil {
return