Extensions: misc checks on receiver parameter and extension members (#77937)

Author: Julien Couvreur

docs/compilers/CSharp/Compiler Breaking Changes - DotNet 10.md

@@ -334,26 +334,33 @@ class partial { }
 
 ## `extension` treated as a contextual keyword
 
-PROTOTYPE record which version this break is introduced in
-
+***Introduced in Visual Studio 2022 version 17.14.***
 Starting with C# 14, the `extension` keyword serves a special purpose in denoting extension containers. 
 This changes how the compiler interprets certain code constructs.
 
 If you need to use "extension" as an identifier rather than a keyword, escape it with the `@` prefix: `@extension`. This tells the compiler to treat it as a regular identifier instead of a keyword.
 
 The compiler will parse this as an extension container rather than a constructor.
 ```csharp
-class extension
+class @extension
 {
     extension(object o) { } // parsed as an extension container
 }
 ```
 
 The compiler will fail to parse this as a method with return type `extension`.
 ```csharp
-class extension
+class @extension
 {
     extension M() { } // will not compile
 }
 ```
 
+***Introduced in Visual Studio 2022 version 17.15.***
+The "extension" identifier may not be used as a type name, so the following will not compile:
+```csharp
+using extension = ...; // alias may not be named "extension"
+class extension { } // type may not be named "extension"
+class C<extension> { } // type parameter may not be named "extension"
+```
+

docs/contributing/Compiler Test Plan.md

@@ -80,6 +80,7 @@ This document provides guidance for thinking about language interactions and tes
 - Readonly members on structs (methods, property/indexer accessors, custom event accessors)
 - SkipLocalsInit
 - Method override or explicit implementation with `where T : { class, struct, default }`
+- `extension` blocks
 
 # Code
 - Operators (see Eric's list below)

eng/todo-check.ps1

@@ -7,7 +7,7 @@ Set-StrictMode -version 2.0
 $ErrorActionPreference="Stop"
 if ($env:SYSTEM_PULLREQUEST_TARGETBRANCH -eq "main") {
   Write-Host "Checking no PROTOTYPE markers in source"
-  $prototypes = Get-ChildItem -Path src, eng, scripts -Exclude *.dll,*.exe,*.pdb,*.xlf,todo-check.ps1 -Recurse | Select-String -Pattern 'PROTOTYPE' -CaseSensitive -SimpleMatch
+  $prototypes = Get-ChildItem -Path src, eng, scripts, docs\compilers -Exclude *.dll,*.exe,*.pdb,*.xlf,todo-check.ps1 -Recurse | Select-String -Pattern 'PROTOTYPE' -CaseSensitive -SimpleMatch
   if ($prototypes) {
     Write-Host "Found PROTOTYPE markers in source:"
     Write-Host $prototypes
@@ -16,7 +16,7 @@ if ($env:SYSTEM_PULLREQUEST_TARGETBRANCH -eq "main") {
 }
 
 # Verify no TODO2 marker left
-$prototypes = Get-ChildItem -Path src, eng, scripts -Exclude *.dll,*.exe,*.pdb,*.xlf,todo-check.ps1 -Recurse | Select-String -Pattern 'TODO2' -CaseSensitive -SimpleMatch
+$prototypes = Get-ChildItem -Path src, eng, scripts, docs\compilers -Exclude *.dll,*.exe,*.pdb,*.xlf,todo-check.ps1 -Recurse | Select-String -Pattern 'TODO2' -CaseSensitive -SimpleMatch
 if ($prototypes) {
   Write-Host "Found TODO2 markers in source:"
   Write-Host $prototypes

src/Compilers/CSharp/Portable/Binder/Binder_Expressions.cs

@@ -2121,7 +2121,7 @@ private BoundExpression BindNonMethod(SimpleNameSyntax node, Symbol symbol, Bind
                         {
                             Error(diagnostics, ErrorCode.ERR_InvalidPrimaryConstructorParameterReference, node, parameter);
                         }
-                        else if (parameter.ContainingSymbol is NamedTypeSymbol { IsExtension: true } &&
+                        else if (parameter.IsExtensionParameter() &&
                                 (InParameterDefaultValue || InAttributeArgument ||
                                  this.ContainingMember() is not { Kind: not SymbolKind.NamedType, IsStatic: false } || // We are not in an instance member
                                  (object)this.ContainingMember().ContainingSymbol != parameter.ContainingSymbol) &&

src/Compilers/CSharp/Portable/Binder/Binder_NameConflicts.cs

@@ -74,7 +74,7 @@ internal void ValidateParameterNameConflicts(
                                 diagnostics.Add(ErrorCode.ERR_LocalSameNameAsExtensionTypeParameter, GetLocation(p), name);
                             }
                         }
-                        else if (p.ContainingSymbol is NamedTypeSymbol { IsExtension: true })
+                        else if (p.IsExtensionParameter())
                         {
                             diagnostics.Add(ErrorCode.ERR_TypeParameterSameNameAsExtensionParameter, tp.GetFirstLocationOrNone(), name);
                         }

src/Compilers/CSharp/Portable/Binder/ExecutableCodeBinder.cs

@@ -107,14 +107,21 @@ public static void ValidateIteratorMethod(CSharpCompilation compilation, MethodS
                 return;
             }
 
-            foreach (var parameter in iterator.Parameters)
+            var parameters = !iterator.IsStatic
+                ? iterator.GetParametersIncludingExtensionParameter()
+                : iterator.Parameters;
+
+            foreach (var parameter in parameters)
             {
+                bool isReceiverParameter = parameter.IsExtensionParameter();
                 if (parameter.RefKind != RefKind.None)
                 {
-                    diagnostics.Add(ErrorCode.ERR_BadIteratorArgType, parameter.GetFirstLocation());
+                    var location = isReceiverParameter ? iterator.GetFirstLocation() : parameter.GetFirstLocation();
+                    diagnostics.Add(ErrorCode.ERR_BadIteratorArgType, location);
                 }
-                else if (parameter.Type.IsPointerOrFunctionPointer())
+                else if (parameter.Type.IsPointerOrFunctionPointer() && !isReceiverParameter)
                 {
+                    // We already reported an error elsewhere if the receiver parameter of an extension is a pointer type.
                     diagnostics.Add(ErrorCode.ERR_UnsafeIteratorArgType, parameter.GetFirstLocation());
                 }
             }

src/Compilers/CSharp/Portable/Binder/Semantics/AccessCheck.cs

@@ -696,7 +696,9 @@ internal static bool HasInternalAccessTo(this AssemblySymbol fromAssembly, Assem
 
         internal static ErrorCode GetProtectedMemberInSealedTypeError(NamedTypeSymbol containingType)
         {
-            return containingType.TypeKind == TypeKind.Struct ? ErrorCode.ERR_ProtectedInStruct : ErrorCode.WRN_ProtectedInSealed;
+            return containingType.IsExtension ? ErrorCode.ERR_ProtectedInExtension
+                : containingType.TypeKind == TypeKind.Struct ? ErrorCode.ERR_ProtectedInStruct
+                : ErrorCode.WRN_ProtectedInSealed;
         }
     }
 }

src/Compilers/CSharp/Portable/Binder/Semantics/OverloadResolution/OverloadResolution.cs

@@ -3878,16 +3878,16 @@ private static EffectiveParameters GetEffectiveParametersInNormalForm<TMember>(
 
             hasAnyRefOmittedArgument = false;
 
-            bool isNewExtensionMember = member.GetIsNewExtensionMember();
-            ImmutableArray<ParameterSymbol> parameters = isNewExtensionMember ? GetParametersIncludingReceiver(member) : member.GetParameters();
+            ImmutableArray<ParameterSymbol> parameters = member.GetParametersIncludingExtensionParameter();
 
             // We simulate an extra parameter for vararg methods 
             int parameterCount = parameters.Length + (member.GetIsVararg() ? 1 : 0);
 
             if (argumentCount == parameterCount && argToParamMap.IsDefaultOrEmpty)
             {
                 bool hasSomeRefKinds = !member.GetParameterRefKinds().IsDefaultOrEmpty;
-                if (member.GetIsNewExtensionMember())
+                bool isNewExtensionMember = member.GetIsNewExtensionMember();
+                if (isNewExtensionMember)
                 {
                     Debug.Assert(member.ContainingType.ExtensionParameter is not null);
                     hasSomeRefKinds |= member.ContainingType.ExtensionParameter.RefKind != RefKind.None;
@@ -4040,7 +4040,7 @@ private static EffectiveParameters GetEffectiveParametersInExpandedForm<TMember>
             var types = ArrayBuilder<TypeWithAnnotations>.GetInstance();
             var refs = ArrayBuilder<RefKind>.GetInstance();
             bool anyRef = false;
-            var parameters = member.GetIsNewExtensionMember() ? GetParametersIncludingReceiver(member) : member.GetParameters();
+            var parameters = member.GetParametersIncludingExtensionParameter();
             bool hasAnyRefArg = argumentRefKinds.Any();
             hasAnyRefOmittedArgument = false;
             TypeWithAnnotations paramsIterationType = default;

src/Compilers/CSharp/Portable/Binder/Semantics/OverloadResolution/OverloadResolutionResult.cs

@@ -895,7 +895,7 @@ private static void ReportMissingRequiredParameter(
             // to required formal parameter 'y'.
 
             TMember badMember = bad.Member;
-            ImmutableArray<ParameterSymbol> parameters = badMember.GetIsNewExtensionMember() ? OverloadResolution.GetParametersIncludingReceiver(badMember) : badMember.GetParameters();
+            ImmutableArray<ParameterSymbol> parameters = badMember.GetParametersIncludingExtensionParameter();
             int badParamIndex = bad.Result.BadParameter;
             string badParamName;
             if (badParamIndex == parameters.Length)
@@ -1115,7 +1115,7 @@ private bool HadBadArguments(
                 // as there is no explicit call to Add method.
 
                 int argumentOffset = arguments.IncludesReceiverAsArgument ? 1 : 0;
-                var parameters = method.GetIsNewExtensionMember() ? OverloadResolution.GetParametersIncludingReceiver(method) : method.GetParameters();
+                var parameters = method.GetParametersIncludingExtensionParameter();
 
                 for (int i = argumentOffset; i < parameters.Length; i++)
                 {
@@ -1170,7 +1170,7 @@ private static void ReportBadArgumentError(
 
             // Early out: if the bad argument is an __arglist parameter then simply report that:
 
-            var parameters = method.GetIsNewExtensionMember() ? OverloadResolution.GetParametersIncludingReceiver(method) : method.GetParameters();
+            var parameters = method.GetParametersIncludingExtensionParameter();
             if (method.GetIsVararg() && parm == parameters.Length)
             {
                 // NOTE: No SymbolDistinguisher required, since one of the arguments is "__arglist".

src/Compilers/CSharp/Portable/Binder/Semantics/OverloadResolution/OverloadResolution_ArgsToParameters.cs

@@ -55,13 +55,6 @@ public ImmutableArray<int> ToImmutableArray()
             }
         }
 
-        internal static ImmutableArray<ParameterSymbol> GetParametersIncludingReceiver(Symbol symbol)
-        {
-            Debug.Assert(symbol.GetIsNewExtensionMember());
-            // Tracked by https://github.com/dotnet/roslyn/issues/76130 : consider optimizing
-            return [symbol.ContainingType.ExtensionParameter, .. symbol.GetParameters()];
-        }
-
         private static ImmutableArray<TypeWithAnnotations> GetParameterTypesIncludingReceiver(Symbol symbol)
         {
             Debug.Assert(symbol.GetIsNewExtensionMember());
@@ -78,8 +71,7 @@ private static ArgumentAnalysisResult AnalyzeArguments(
             Debug.Assert((object)symbol != null);
             Debug.Assert(arguments != null);
 
-            bool isNewExtensionMember = symbol.GetIsNewExtensionMember();
-            ImmutableArray<ParameterSymbol> parameters = isNewExtensionMember ? GetParametersIncludingReceiver(symbol) : symbol.GetParameters();
+            ImmutableArray<ParameterSymbol> parameters = symbol.GetParametersIncludingExtensionParameter();
             bool isVararg = symbol.GetIsVararg();
 
             // The easy out is that we have no named arguments and are in normal form.