hohn/codeql
1
0
Fork 0
mirror of https://github.com/github/codeql.git synced 2026-04-26 09:15:12 +02:00
Code Issues Packages Projects Releases Wiki Activity

Test if Origin() works (for Var and Func)

Browse Source
This commit is contained in:
Owen Mansel-Chan
2023-07-14 11:11:42 +01:00
parent 1767ca2682
commit cff09d28b8
1 changed files with 23 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
go/extractor/extractor.go
View File

@@ -1946,6 +1946,18 @@ func populateTypeParamParents(tw *trap.Writer, typeparams *types.TypeParamList,
// some changes to the object to avoid returning objects relating to instantiated
// types.
func getObjectBeingUsed(tw *trap.Writer, ident *ast.Ident) types.Object {
obj1 := getObjectBeingUsed1(tw, ident)
obj2 := getObjectBeingUsed2(tw, ident)
if obj1 != obj2 {
log.Fatalf("different results!\nobj1 = %s\nobj2=%s\n", obj1.String(), obj2.String())
}
return obj2
}
// getobjectBeingUsed looks up `ident` in `tw.Package.TypesInfo.Uses` and makes
// some changes to the object to avoid returning objects relating to instantiated
// types.
func getObjectBeingUsed1(tw *trap.Writer, ident *ast.Ident) types.Object {
obj := tw.Package.TypesInfo.Uses[ident]
if obj == nil {
return nil
@@ -1990,6 +2002,17 @@ func getObjectBeingUsed(tw *trap.Writer, ident *ast.Ident) types.Object {
return obj
}
func getObjectBeingUsed2(tw *trap.Writer, ident *ast.Ident) types.Object {
switch obj := tw.Package.TypesInfo.Uses[ident].(type) {
case *types.Var:
return obj.Origin()
case *types.Func:
return obj.Origin()
default:
return obj
}
}
// tryGetGenericType returns the generic type of `tp`, and a boolean indicating
// whether it is the same as `tp`.
func tryGetGenericType(tp types.Type) (*types.Named, bool) {