Switch out use of std::string and std::vector for large allocations for a buffer class that doesn...

courgette/win32_x86_generator.h

 // Copyright (c) 2011 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 // This is the transformation and adjustment for Windows X86 executables.
 
 #ifndef COURGETTE_WIN32_X86_GENERATOR_H_
 #define COURGETTE_WIN32_X86_GENERATOR_H_
 
 #include "base/logging.h"
 #include "base/memory/scoped_ptr.h"
 
 #include "courgette/ensemble.h"
 
 namespace courgette {
 
 class CourgetteWin32X86PatchGenerator : public TransformationPatchGenerator {
  public:
   CourgetteWin32X86PatchGenerator(Element* old_element,
                                   Element* new_element,
                                   CourgetteWin32X86Patcher* patcher)
       : TransformationPatchGenerator(old_element, new_element, patcher) {
   }
 
   CourgettePatchFile::TransformationMethodId Kind() {
     return CourgettePatchFile::T_COURGETTE_WIN32_X86;
   }
 
   Status WriteInitialParameters(SinkStream* parameter_stream) {
-    parameter_stream->WriteSizeVarint32(old_element_->offset_in_ensemble());
-    parameter_stream->WriteSizeVarint32(old_element_->region().length());
+    if (!parameter_stream->WriteSizeVarint32(
+            old_element_->offset_in_ensemble()) ||
+        !parameter_stream->WriteSizeVarint32(old_element_->region().length())) {
+      return C_STREAM_ERROR;
+    }
     return C_OK;
     // TODO(sra): Initialize |patcher_| with these parameters.
   }
 
   Status PredictTransformParameters(SinkStreamSet* prediction) {
     return TransformationPatchGenerator::PredictTransformParameters(prediction);
   }
 
   Status CorrectedTransformParameters(SinkStreamSet* parameters) {
     // No code needed to write an 'empty' parameter set.
@@ skipping 79 matching lines @@
   }
 
  private:
   ~CourgetteWin32X86PatchGenerator() { }
 
   DISALLOW_COPY_AND_ASSIGN(CourgetteWin32X86PatchGenerator);
 };
 
 }  // namespace courgette
 #endif  // COURGETTE_WIN32_X86_GENERATOR_H_