signature.go

Documentation: cmd/compile/internal/types2

     1  // Copyright 2021 The Go Authors. All rights reserved.
     2  // Use of this source code is governed by a BSD-style
     3  // license that can be found in the LICENSE file.
     4  
     5  package types2
     6  
     7  import (
     8  	"cmd/compile/internal/syntax"
     9  	"fmt"
    10  	. "internal/types/errors"
    11  )
    12  
    13  // ----------------------------------------------------------------------------
    14  // API
    15  
    16  // A Signature represents a (non-builtin) function or method type.
    17  // The receiver is ignored when comparing signatures for identity.
    18  type Signature struct {
    19  	// We need to keep the scope in Signature (rather than passing it around
    20  	// and store it in the Func Object) because when type-checking a function
    21  	// literal we call the general type checker which returns a general Type.
    22  	// We then unpack the *Signature and use the scope for the literal body.
    23  	rparams  *TypeParamList // receiver type parameters from left to right, or nil
    24  	tparams  *TypeParamList // type parameters from left to right, or nil
    25  	scope    *Scope         // function scope for package-local and non-instantiated signatures; nil otherwise
    26  	recv     *Var           // nil if not a method
    27  	params   *Tuple         // (incoming) parameters from left to right; or nil
    28  	results  *Tuple         // (outgoing) results from left to right; or nil
    29  	variadic bool           // true if the last parameter's type is of the form ...T (or string, for append built-in only)
    30  }
    31  
    32  // NewSignatureType creates a new function type for the given receiver,
    33  // receiver type parameters, type parameters, parameters, and results. If
    34  // variadic is set, params must hold at least one parameter and the last
    35  // parameter's core type must be of unnamed slice or bytestring type.
    36  // If recv is non-nil, typeParams must be empty. If recvTypeParams is
    37  // non-empty, recv must be non-nil.
    38  func NewSignatureType(recv *Var, recvTypeParams, typeParams []*TypeParam, params, results *Tuple, variadic bool) *Signature {
    39  	if variadic {
    40  		n := params.Len()
    41  		if n == 0 {
    42  			panic("variadic function must have at least one parameter")
    43  		}
    44  		core := coreString(params.At(n - 1).typ)
    45  		if _, ok := core.(*Slice); !ok && !isString(core) {
    46  			panic(fmt.Sprintf("got %s, want variadic parameter with unnamed slice type or string as core type", core.String()))
    47  		}
    48  	}
    49  	sig := &Signature{recv: recv, params: params, results: results, variadic: variadic}
    50  	if len(recvTypeParams) != 0 {
    51  		if recv == nil {
    52  			panic("function with receiver type parameters must have a receiver")
    53  		}
    54  		sig.rparams = bindTParams(recvTypeParams)
    55  	}
    56  	if len(typeParams) != 0 {
    57  		if recv != nil {
    58  			panic("function with type parameters cannot have a receiver")
    59  		}
    60  		sig.tparams = bindTParams(typeParams)
    61  	}
    62  	return sig
    63  }
    64  
    65  // Recv returns the receiver of signature s (if a method), or nil if a
    66  // function. It is ignored when comparing signatures for identity.
    67  //
    68  // For an abstract method, Recv returns the enclosing interface either
    69  // as a *Named or an *Interface. Due to embedding, an interface may
    70  // contain methods whose receiver type is a different interface.
    71  func (s *Signature) Recv() *Var { return s.recv }
    72  
    73  // TypeParams returns the type parameters of signature s, or nil.
    74  func (s *Signature) TypeParams() *TypeParamList { return s.tparams }
    75  
    76  // RecvTypeParams returns the receiver type parameters of signature s, or nil.
    77  func (s *Signature) RecvTypeParams() *TypeParamList { return s.rparams }
    78  
    79  // Params returns the parameters of signature s, or nil.
    80  func (s *Signature) Params() *Tuple { return s.params }
    81  
    82  // Results returns the results of signature s, or nil.
    83  func (s *Signature) Results() *Tuple { return s.results }
    84  
    85  // Variadic reports whether the signature s is variadic.
    86  func (s *Signature) Variadic() bool { return s.variadic }
    87  
    88  func (s *Signature) Underlying() Type { return s }
    89  func (s *Signature) String() string   { return TypeString(s, nil) }
    90  
    91  // ----------------------------------------------------------------------------
    92  // Implementation
    93  
    94  // funcType type-checks a function or method type.
    95  func (check *Checker) funcType(sig *Signature, recvPar *syntax.Field, tparams []*syntax.Field, ftyp *syntax.FuncType) {
    96  	check.openScope(ftyp, "function")
    97  	check.scope.isFunc = true
    98  	check.recordScope(ftyp, check.scope)
    99  	sig.scope = check.scope
   100  	defer check.closeScope()
   101  
   102  	if recvPar != nil {
   103  		// collect generic receiver type parameters, if any
   104  		// - a receiver type parameter is like any other type parameter, except that it is declared implicitly
   105  		// - the receiver specification acts as local declaration for its type parameters, which may be blank
   106  		_, rname, rparams := check.unpackRecv(recvPar.Type, true)
   107  		if len(rparams) > 0 {
   108  			// The scope of the type parameter T in "func (r T[T]) f()"
   109  			// starts after f, not at "r"; see #52038.
   110  			scopePos := ftyp.Pos()
   111  			tparams := make([]*TypeParam, len(rparams))
   112  			for i, rparam := range rparams {
   113  				tparams[i] = check.declareTypeParam(rparam, scopePos)
   114  			}
   115  			sig.rparams = bindTParams(tparams)
   116  			// Blank identifiers don't get declared, so naive type-checking of the
   117  			// receiver type expression would fail in Checker.collectParams below,
   118  			// when Checker.ident cannot resolve the _ to a type.
   119  			//
   120  			// Checker.recvTParamMap maps these blank identifiers to their type parameter
   121  			// types, so that they may be resolved in Checker.ident when they fail
   122  			// lookup in the scope.
   123  			for i, p := range rparams {
   124  				if p.Value == "_" {
   125  					if check.recvTParamMap == nil {
   126  						check.recvTParamMap = make(map[*syntax.Name]*TypeParam)
   127  					}
   128  					check.recvTParamMap[p] = tparams[i]
   129  				}
   130  			}
   131  			// determine receiver type to get its type parameters
   132  			// and the respective type parameter bounds
   133  			var recvTParams []*TypeParam
   134  			if rname != nil {
   135  				// recv should be a Named type (otherwise an error is reported elsewhere)
   136  				// Also: Don't report an error via genericType since it will be reported
   137  				//       again when we type-check the signature.
   138  				// TODO(gri) maybe the receiver should be marked as invalid instead?
   139  				if recv := asNamed(check.genericType(rname, nil)); recv != nil {
   140  					recvTParams = recv.TypeParams().list()
   141  				}
   142  			}
   143  			// provide type parameter bounds
   144  			if len(tparams) == len(recvTParams) {
   145  				smap := makeRenameMap(recvTParams, tparams)
   146  				for i, tpar := range tparams {
   147  					recvTPar := recvTParams[i]
   148  					check.mono.recordCanon(tpar, recvTPar)
   149  					// recvTPar.bound is (possibly) parameterized in the context of the
   150  					// receiver type declaration. Substitute parameters for the current
   151  					// context.
   152  					tpar.bound = check.subst(tpar.obj.pos, recvTPar.bound, smap, nil, check.context())
   153  				}
   154  			} else if len(tparams) < len(recvTParams) {
   155  				// Reporting an error here is a stop-gap measure to avoid crashes in the
   156  				// compiler when a type parameter/argument cannot be inferred later. It
   157  				// may lead to follow-on errors (see issues go.dev/issue/51339, go.dev/issue/51343).
   158  				// TODO(gri) find a better solution
   159  				got := measure(len(tparams), "type parameter")
   160  				check.errorf(recvPar, BadRecv, "got %s, but receiver base type declares %d", got, len(recvTParams))
   161  			}
   162  		}
   163  	}
   164  
   165  	if tparams != nil {
   166  		// The parser will complain about invalid type parameters for methods.
   167  		check.collectTypeParams(&sig.tparams, tparams)
   168  	}
   169  
   170  	// Use a temporary scope for all parameter declarations and then
   171  	// squash that scope into the parent scope (and report any
   172  	// redeclarations at that time).
   173  	//
   174  	// TODO(adonovan): now that each declaration has the correct
   175  	// scopePos, there should be no need for scope squashing.
   176  	// Audit to ensure all lookups honor scopePos and simplify.
   177  	scope := NewScope(check.scope, nopos, nopos, "function body (temp. scope)")
   178  	scopePos := syntax.EndPos(ftyp) // all parameters' scopes start after the signature
   179  	var recvList []*Var             // TODO(gri) remove the need for making a list here
   180  	if recvPar != nil {
   181  		recvList, _ = check.collectParams(scope, []*syntax.Field{recvPar}, false, scopePos) // use rewritten receiver type, if any
   182  	}
   183  	params, variadic := check.collectParams(scope, ftyp.ParamList, true, scopePos)
   184  	results, _ := check.collectParams(scope, ftyp.ResultList, false, scopePos)
   185  	scope.Squash(func(obj, alt Object) {
   186  		err := check.newError(DuplicateDecl)
   187  		err.addf(obj, "%s redeclared in this block", obj.Name())
   188  		err.addAltDecl(alt)
   189  		err.report()
   190  	})
   191  
   192  	if recvPar != nil {
   193  		// recv parameter list present (may be empty)
   194  		// spec: "The receiver is specified via an extra parameter section preceding the
   195  		// method name. That parameter section must declare a single parameter, the receiver."
   196  		var recv *Var
   197  		switch len(recvList) {
   198  		case 0:
   199  			// error reported by resolver
   200  			recv = NewParam(nopos, nil, "", Typ[Invalid]) // ignore recv below
   201  		default:
   202  			// more than one receiver
   203  			check.error(recvList[len(recvList)-1].Pos(), InvalidRecv, "method must have exactly one receiver")
   204  			fallthrough // continue with first receiver
   205  		case 1:
   206  			recv = recvList[0]
   207  		}
   208  		sig.recv = recv
   209  
   210  		// Delay validation of receiver type as it may cause premature expansion
   211  		// of types the receiver type is dependent on (see issues go.dev/issue/51232, go.dev/issue/51233).
   212  		check.later(func() {
   213  			// spec: "The receiver type must be of the form T or *T where T is a type name."
   214  			rtyp, _ := deref(recv.typ)
   215  			atyp := Unalias(rtyp)
   216  			if !isValid(atyp) {
   217  				return // error was reported before
   218  			}
   219  			// spec: "The type denoted by T is called the receiver base type; it must not
   220  			// be a pointer or interface type and it must be declared in the same package
   221  			// as the method."
   222  			switch T := atyp.(type) {
   223  			case *Named:
   224  				// The receiver type may be an instantiated type referred to
   225  				// by an alias (which cannot have receiver parameters for now).
   226  				if T.TypeArgs() != nil && sig.RecvTypeParams() == nil {
   227  					check.errorf(recv, InvalidRecv, "cannot define new methods on instantiated type %s", rtyp)
   228  					break
   229  				}
   230  				if T.obj.pkg != check.pkg {
   231  					check.errorf(recv, InvalidRecv, "cannot define new methods on non-local type %s", rtyp)
   232  					break
   233  				}
   234  				var cause string
   235  				switch u := T.under().(type) {
   236  				case *Basic:
   237  					// unsafe.Pointer is treated like a regular pointer
   238  					if u.kind == UnsafePointer {
   239  						cause = "unsafe.Pointer"
   240  					}
   241  				case *Pointer, *Interface:
   242  					cause = "pointer or interface type"
   243  				case *TypeParam:
   244  					// The underlying type of a receiver base type cannot be a
   245  					// type parameter: "type T[P any] P" is not a valid declaration.
   246  					panic("unreachable")
   247  				}
   248  				if cause != "" {
   249  					check.errorf(recv, InvalidRecv, "invalid receiver type %s (%s)", rtyp, cause)
   250  				}
   251  			case *Basic:
   252  				check.errorf(recv, InvalidRecv, "cannot define new methods on non-local type %s", rtyp)
   253  			default:
   254  				check.errorf(recv, InvalidRecv, "invalid receiver type %s", recv.typ)
   255  			}
   256  		}).describef(recv, "validate receiver %s", recv)
   257  	}
   258  
   259  	sig.params = NewTuple(params...)
   260  	sig.results = NewTuple(results...)
   261  	sig.variadic = variadic
   262  }
   263  
   264  // collectParams declares the parameters of list in scope and returns the corresponding
   265  // variable list.
   266  func (check *Checker) collectParams(scope *Scope, list []*syntax.Field, variadicOk bool, scopePos syntax.Pos) (params []*Var, variadic bool) {
   267  	if list == nil {
   268  		return
   269  	}
   270  
   271  	var named, anonymous bool
   272  
   273  	var typ Type
   274  	var prev syntax.Expr
   275  	for i, field := range list {
   276  		ftype := field.Type
   277  		// type-check type of grouped fields only once
   278  		if ftype != prev {
   279  			prev = ftype
   280  			if t, _ := ftype.(*syntax.DotsType); t != nil {
   281  				ftype = t.Elem
   282  				if variadicOk && i == len(list)-1 {
   283  					variadic = true
   284  				} else {
   285  					check.softErrorf(t, MisplacedDotDotDot, "can only use ... with final parameter in list")
   286  					// ignore ... and continue
   287  				}
   288  			}
   289  			typ = check.varType(ftype)
   290  		}
   291  		// The parser ensures that f.Tag is nil and we don't
   292  		// care if a constructed AST contains a non-nil tag.
   293  		if field.Name != nil {
   294  			// named parameter
   295  			name := field.Name.Value
   296  			if name == "" {
   297  				check.error(field.Name, InvalidSyntaxTree, "anonymous parameter")
   298  				// ok to continue
   299  			}
   300  			par := NewParam(field.Name.Pos(), check.pkg, name, typ)
   301  			check.declare(scope, field.Name, par, scopePos)
   302  			params = append(params, par)
   303  			named = true
   304  		} else {
   305  			// anonymous parameter
   306  			par := NewParam(field.Pos(), check.pkg, "", typ)
   307  			check.recordImplicit(field, par)
   308  			params = append(params, par)
   309  			anonymous = true
   310  		}
   311  	}
   312  
   313  	if named && anonymous {
   314  		check.error(list[0], InvalidSyntaxTree, "list contains both named and anonymous parameters")
   315  		// ok to continue
   316  	}
   317  
   318  	// For a variadic function, change the last parameter's type from T to []T.
   319  	// Since we type-checked T rather than ...T, we also need to retro-actively
   320  	// record the type for ...T.
   321  	if variadic {
   322  		last := params[len(params)-1]
   323  		last.typ = &Slice{elem: last.typ}
   324  		check.recordTypeAndValue(list[len(list)-1].Type, typexpr, last.typ, nil)
   325  	}
   326  
   327  	return
   328  }
   329
View as plain text