瀏覽代碼

Check return value on write of raw pointer

Felix Jendrusch 7 年之前
父節點
當前提交
1da9ffe394
共有 1 個文件被更改,包括 4 次插入1 次删除
  1. 4 1
      objectivec/GPBCodedOutputStream.m

+ 4 - 1
objectivec/GPBCodedOutputStream.m

@@ -942,7 +942,10 @@ static void GPBWriteRawLittleEndian64(GPBOutputBufferState *state,
       state_.position = length;
     } else {
       // Write is very big.  Let's do it all at once.
-      [state_.output write:((uint8_t *)value) + offset maxLength:length];
+      NSInteger written = [state_.output write:((uint8_t *)value) + offset maxLength:length];
+      if (written != (NSInteger)length) {
+        [NSException raise:GPBCodedOutputStreamException_WriteFailed format:@""];
+      }
     }
   }
 }