Fix #1355 Make extension nodes check for their extensions

src/Microsoft.AspNetCore.Mvc.Razor.Extensions/InjectDirectiveIRNode.cs

@@ -38,7 +38,18 @@ namespace Microsoft.AspNetCore.Mvc.Razor.Extensions
                 throw new ArgumentNullException(nameof(target));
             }
 
+            if (context == null)
+            {
+                throw new ArgumentNullException(nameof(context));
+            }
+
             var extension = target.GetExtension<IInjectDirectiveTargetExtension>();
+            if (extension == null)
+            {
+                context.ReportMissingExtension<IInjectDirectiveTargetExtension>();
+                return;
+            }
+
             extension.WriteInjectProperty(context, this);
         }
     }

src/Microsoft.AspNetCore.Razor.Language/Extensions/TemplateIRNode.cs

@@ -28,6 +28,16 @@ namespace Microsoft.AspNetCore.Razor.Language.Extensions
 
         public override void WriteNode(CodeTarget target, CSharpRenderingContext context)
         {
+            if (target == null)
+            {
+                throw new ArgumentNullException(nameof(target));
+            }
+
+            if (context == null)
+            {
+                throw new ArgumentNullException(nameof(context));
+            }
+
             var extension = target.GetExtension<ITemplateTargetExtension>();
             if (extension == null)
             {

src/Microsoft.AspNetCore.Razor.Language/Intermediate/AddPreallocatedTagHelperHtmlAttributeIRNode.cs

@@ -29,7 +29,23 @@ namespace Microsoft.AspNetCore.Razor.Language.Intermediate
 
         public override void WriteNode(CodeTarget target, CSharpRenderingContext context)
         {
+            if (target == null)
+            {
+                throw new ArgumentNullException(nameof(target));
+            }
+
+            if (context == null)
+            {
+                throw new ArgumentNullException(nameof(context));
+            }
+
             var extension = target.GetExtension<IPreallocatedAttributeTargetExtension>();
+            if (extension == null)
+            {
+                context.ReportMissingExtension<IPreallocatedAttributeTargetExtension>();
+                return;
+            }
+
             extension.WriteAddPreallocatedTagHelperHtmlAttribute(context, this);
         }
     }

src/Microsoft.AspNetCore.Razor.Language/Intermediate/DeclarePreallocatedTagHelperAttributeIRNode.cs

@@ -36,7 +36,23 @@ namespace Microsoft.AspNetCore.Razor.Language.Intermediate
 
         public override void WriteNode(CodeTarget target, CSharpRenderingContext context)
         {
+            if (target == null)
+            {
+                throw new ArgumentNullException(nameof(target));
+            }
+
+            if (context == null)
+            {
+                throw new ArgumentNullException(nameof(context));
+            }
+
             var extension = target.GetExtension<IPreallocatedAttributeTargetExtension>();
+            if (extension == null)
+            {
+                context.ReportMissingExtension<IPreallocatedAttributeTargetExtension>();
+                return;
+            }
+
             extension.WriteDeclarePreallocatedTagHelperAttribute(context, this);
         }
     }

src/Microsoft.AspNetCore.Razor.Language/Intermediate/DeclarePreallocatedTagHelperHtmlAttributeIRNode.cs

@@ -36,7 +36,23 @@ namespace Microsoft.AspNetCore.Razor.Language.Intermediate
 
         public override void WriteNode(CodeTarget target, CSharpRenderingContext context)
         {
+            if (target == null)
+            {
+                throw new ArgumentNullException(nameof(target));
+            }
+
+            if (context == null)
+            {
+                throw new ArgumentNullException(nameof(context));
+            }
+
             var extension = target.GetExtension<IPreallocatedAttributeTargetExtension>();
+            if (extension == null)
+            {
+                context.ReportMissingExtension<IPreallocatedAttributeTargetExtension>();
+                return;
+            }
+
             extension.WriteDeclarePreallocatedTagHelperHtmlAttribute(context, this);
         }
     }

src/Microsoft.AspNetCore.Razor.Language/Intermediate/DesignTimeDirectiveIRNode.cs

@@ -27,7 +27,23 @@ namespace Microsoft.AspNetCore.Razor.Language.Intermediate
 
         public override void WriteNode(CodeTarget target, CSharpRenderingContext context)
         {
+            if (target == null)
+            {
+                throw new ArgumentNullException(nameof(target));
+            }
+
+            if (context == null)
+            {
+                throw new ArgumentNullException(nameof(context));
+            }
+
             var extension = target.GetExtension<IDesignTimeDirectiveTargetExtension>();
+            if (extension == null)
+            {
+                context.ReportMissingExtension<IDesignTimeDirectiveTargetExtension>();
+                return;
+            }
+
             extension.WriteDesignTimeDirective(context, this);
         }
     }

src/Microsoft.AspNetCore.Razor.Language/Intermediate/SetPreallocatedTagHelperPropertyIRNode.cs

@@ -42,7 +42,23 @@ namespace Microsoft.AspNetCore.Razor.Language.Intermediate
 
         public override void WriteNode(CodeTarget target, CSharpRenderingContext context)
         {
+            if (target == null)
+            {
+                throw new ArgumentNullException(nameof(target));
+            }
+
+            if (context == null)
+            {
+                throw new ArgumentNullException(nameof(context));
+            }
+
             var extension = target.GetExtension<IPreallocatedAttributeTargetExtension>();
+            if (extension == null)
+            {
+                context.ReportMissingExtension<IPreallocatedAttributeTargetExtension>();
+                return;
+            }
+
             extension.WriteSetPreallocatedTagHelperProperty(context, this);
         }
     }