Add some missing null-checks

Remove IsInitialized checks accidentally left in MessageParser Simplify ExtensionCollection.CrossLink
protocolbuffers · Nov 19, 2019 · 71c492d · 71c492d
1 parent 3cae867
commit 71c492d
Show file tree

Hide file tree

Showing 5 changed files with 7 additions and 21 deletions.
diff --git a/csharp/src/Google.Protobuf/CodedOutputStream.cs b/csharp/src/Google.Protobuf/CodedOutputStream.cs
@@ -89,7 +89,7 @@ public CodedOutputStream(byte[] flatArray) : this(flatArray, 0, flatArray.Length
         private CodedOutputStream(byte[] buffer, int offset, int length)
         {
             this.output = null;
-            this.buffer = buffer;
+            this.buffer = ProtoPreconditions.CheckNotNull(buffer, nameof(buffer));
             this.position = offset;
             this.limit = offset + length;
             leaveOpen = true; // Simple way of avoiding trying to dispose of a null reference

diff --git a/csharp/src/Google.Protobuf/JsonFormatter.cs b/csharp/src/Google.Protobuf/JsonFormatter.cs
@@ -133,7 +133,7 @@ static JsonFormatter()
         /// <param name="settings">The settings.</param>
         public JsonFormatter(Settings settings)
         {
-            this.settings = settings;
+            this.settings = ProtoPreconditions.CheckNotNull(settings, nameof(settings));
         }
 
         /// <summary>

diff --git a/csharp/src/Google.Protobuf/JsonParser.cs b/csharp/src/Google.Protobuf/JsonParser.cs
@@ -110,7 +110,7 @@ private static void MergeWrapperField(JsonParser parser, IMessage message, JsonT
         /// <param name="settings">The settings.</param>
         public JsonParser(Settings settings)
         {
-            this.settings = settings;
+            this.settings = ProtoPreconditions.CheckNotNull(settings, nameof(settings));
         }
 
         /// <summary>

diff --git a/csharp/src/Google.Protobuf/MessageParser.cs b/csharp/src/Google.Protobuf/MessageParser.cs
@@ -72,7 +72,6 @@ public IMessage ParseFrom(byte[] data)
         {
             IMessage message = factory();
             message.MergeFrom(data, DiscardUnknownFields, Extensions);
-            CheckMergedRequiredFields(message);
             return message;
         }
 
@@ -87,7 +86,6 @@ public IMessage ParseFrom(byte[] data, int offset, int length)
         {
             IMessage message = factory();
             message.MergeFrom(data, offset, length, DiscardUnknownFields, Extensions);
-            CheckMergedRequiredFields(message);
             return message;
         }
 
@@ -100,7 +98,6 @@ public IMessage ParseFrom(ByteString data)
         {
             IMessage message = factory();
             message.MergeFrom(data, DiscardUnknownFields, Extensions);
-            CheckMergedRequiredFields(message);
             return message;
         }
 
@@ -113,7 +110,6 @@ public IMessage ParseFrom(Stream input)
         {
             IMessage message = factory();
             message.MergeFrom(input, DiscardUnknownFields, Extensions);
-            CheckMergedRequiredFields(message);
             return message;
         }
 
@@ -130,7 +126,6 @@ public IMessage ParseDelimitedFrom(Stream input)
         {
             IMessage message = factory();
             message.MergeDelimitedFrom(input, DiscardUnknownFields, Extensions);
-            CheckMergedRequiredFields(message);
             return message;
         }
 
@@ -143,7 +138,6 @@ public IMessage ParseFrom(CodedInputStream input)
         {
             IMessage message = factory();
             MergeFrom(message, input);
-            CheckMergedRequiredFields(message);
             return message;
         }
 
@@ -176,12 +170,6 @@ internal void MergeFrom(IMessage message, CodedInputStream codedInput)
             }
         }
 
-        internal static void CheckMergedRequiredFields(IMessage message)
-        {
-            if (!message.IsInitialized())
-                throw new InvalidOperationException("Parsed message does not contain all required fields");
-        }
-
         /// <summary>
         /// Creates a new message parser which optionally discards unknown fields when parsing.
         /// </summary>

diff --git a/csharp/src/Google.Protobuf/Reflection/ExtensionCollection.cs b/csharp/src/Google.Protobuf/Reflection/ExtensionCollection.cs
@@ -107,13 +107,11 @@ internal void CrossLink()
             {
                 descriptor.CrossLink();
 
-                IList<FieldDescriptor> _;
-                if (!declarationOrder.TryGetValue(descriptor.ExtendeeType, out _))
-                {
-                    declarationOrder.Add(descriptor.ExtendeeType, new List<FieldDescriptor>());
-                }
+                IList<FieldDescriptor> list;
+                if (!declarationOrder.TryGetValue(descriptor.ExtendeeType, out list))
+                    declarationOrder.Add(descriptor.ExtendeeType, list = new List<FieldDescriptor>());
 
-                declarationOrder[descriptor.ExtendeeType].Add(descriptor);
+                list.Add(descriptor);
             }
 
             extensionsByTypeInDeclarationOrder = declarationOrder