src/mojo/public/cpp/bindings/lib/fixed_buffer.cc

   1 // Copyright 2014 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "mojo/public/cpp/bindings/lib/fixed_buffer.h"
   6
   7 #include <assert.h>
   8 #include <stdlib.h>
   9 #include <string.h>
  10
  11 #include <algorithm>
  12
  13 #include "mojo/public/cpp/bindings/lib/bindings_serialization.h"
  14
  15 namespace mojo {
  16 namespace internal {
  17
  18 FixedBuffer::FixedBuffer(size_t size)
  19     : ptr_(NULL),
  20       cursor_(0),
  21       size_(internal::Align(size)) {
  22   // calloc() required to zero memory and thus avoid info leaks.
  23   ptr_ = static_cast<char*>(calloc(size_, 1));
  24 }
  25
  26 FixedBuffer::~FixedBuffer() {
  27   free(ptr_);
  28 }
  29
  30 void* FixedBuffer::Allocate(size_t delta, Destructor dtor) {
  31   assert(!dtor);
  32
  33   delta = internal::Align(delta);
  34
  35   if (delta == 0 || delta > size_ - cursor_) {
  36     assert(false);
  37     return NULL;
  38   }
  39
  40   char* result = ptr_ + cursor_;
  41   cursor_ += delta;
  42
  43   return result;
  44 }
  45
  46 void* FixedBuffer::Leak() {
  47   char* ptr = ptr_;
  48   ptr_ = NULL;
  49   cursor_ = 0;
  50   size_ = 0;
  51   return ptr;
  52 }
  53
  54 }  // namespace internal
  55 }  // namespace mojo