avoid panic when embedding a builtin alias

TypeName.Pkg is documented as:

    Pkg returns the package to which the object belongs.
    The result is nil for labels and objects in the Universe scope.

When a struct type embeds a builtin alias type, such as byte,
this would lead to a panic since we assumed we could use the Pkg method.

Fixes #798.
pull/808/head
Daniel Martí 1 year ago committed by Paul Scheduikat
parent 6f0e46f80b
commit 4271bc45ae

@ -1357,7 +1357,8 @@ type (
objectString = string // as per recordedObjectString objectString = string // as per recordedObjectString
typeName struct { typeName struct {
PkgPath, Name string PkgPath string // empty if builtin
Name string
} }
) )
@ -1532,8 +1533,10 @@ func computePkgCache(fsCache *cache.Cache, lpkg *listedPackage, pkg *types.Packa
continue continue
} }
aliasTypeName := typeName{ aliasTypeName := typeName{
PkgPath: obj.Pkg().Path(), Name: obj.Name(),
Name: obj.Name(), }
if pkg := obj.Pkg(); pkg != nil {
aliasTypeName.PkgPath = pkg.Path()
} }
computed.EmbeddedAliasFields[vrStr] = aliasTypeName computed.EmbeddedAliasFields[vrStr] = aliasTypeName
} }
@ -1894,19 +1897,21 @@ func (tf *transformer) transformGoFile(file *ast.File) *ast.File {
vrStr := recordedObjectString(vr) vrStr := recordedObjectString(vr)
aliasTypeName, ok := tf.curPkgCache.EmbeddedAliasFields[vrStr] aliasTypeName, ok := tf.curPkgCache.EmbeddedAliasFields[vrStr]
if ok { if ok {
pkg2 := tf.pkg aliasScope := tf.pkg.Scope()
if path := aliasTypeName.PkgPath; pkg2.Path() != path { if path := aliasTypeName.PkgPath; path == "" {
aliasScope = types.Universe
} else if path != tf.pkg.Path() {
// If the package is a dependency, import it. // If the package is a dependency, import it.
// We can't grab the package via tf.pkg.Imports, // We can't grab the package via tf.pkg.Imports,
// because some of the packages under there are incomplete. // because some of the packages under there are incomplete.
// ImportFrom will cache complete imports, anyway. // ImportFrom will cache complete imports, anyway.
var err error pkg2, err := tf.origImporter.ImportFrom(path, parentWorkDir, 0)
pkg2, err = tf.origImporter.ImportFrom(path, parentWorkDir, 0)
if err != nil { if err != nil {
panic(err) panic(err)
} }
aliasScope = pkg2.Scope()
} }
tname, ok := pkg2.Scope().Lookup(aliasTypeName.Name).(*types.TypeName) tname, ok := aliasScope.Lookup(aliasTypeName.Name).(*types.TypeName)
if !ok { if !ok {
panic(fmt.Sprintf("EmbeddedAliasFields pointed %q to a missing type %q", vrStr, aliasTypeName)) panic(fmt.Sprintf("EmbeddedAliasFields pointed %q to a missing type %q", vrStr, aliasTypeName))
} }

@ -244,6 +244,13 @@ var _ = embeddingAliasSameName{
Reader: nil, Reader: nil,
} }
type embeddingBuiltinAlias struct {
byte
}
var _ = embeddingBuiltinAlias{3}
var _ = embeddingBuiltinAlias{byte: 3}
-- external/external.go -- -- external/external.go --
package external package external

Loading…
Cancel
Save