[iOS] Add safe guard when receive network response data too long

zhongwuzw · Dec 4, 2023 · f219904 · f219904
1 parent 3854735
commit f219904
Showing 1 changed file with 14 additions and 1 deletion.
diff --git a/packages/react-native/Libraries/Network/RCTNetworkTask.mm b/packages/react-native/Libraries/Network/RCTNetworkTask.mm
@@ -164,7 +164,20 @@ - (void)URLRequest:(id)requestToken didReceiveData:(NSData *)data
     if (!_data) {
       _data = [NSMutableData new];
     }
-    [_data appendData:data];
+    @try {
+      [_data appendData:data];
+    } @catch (NSException *exception) {
+      _status = RCTNetworkTaskFinished;
+      if (_completionBlock) {
+        RCTURLRequestCompletionBlock completionBlock = _completionBlock;
+        [self dispatchCallback:^{
+          completionBlock(self->_response, nil, RCTErrorWithMessage(@"Request's received data too long."));
+        }];
+      }
+      [self invalidate];
+      return;
+    }
+
     length = _data.length;
   }