74 lines
2.1 KiB
C++
74 lines
2.1 KiB
C++
/* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
|
|
/* This Source Code Form is subject to the terms of the Mozilla Public
|
|
* License, v. 2.0. If a copy of the MPL was not distributed with this file,
|
|
* You can obtain one at http://mozilla.org/MPL/2.0/. */
|
|
|
|
// Original author: ekr@rtfm.com
|
|
|
|
#ifndef databuffer_h__
|
|
#define databuffer_h__
|
|
#include <algorithm>
|
|
#include <mozilla/UniquePtr.h>
|
|
#include <m_cpp_utils.h>
|
|
#include <nsISupportsImpl.h>
|
|
|
|
namespace mozilla {
|
|
|
|
class DataBuffer {
|
|
public:
|
|
DataBuffer() : data_(nullptr), len_(0), capacity_(0) {}
|
|
DataBuffer(const uint8_t *data, size_t len) {
|
|
Assign(data, len, len);
|
|
}
|
|
DataBuffer(const uint8_t *data, size_t len, size_t capacity) {
|
|
Assign(data, len, capacity);
|
|
}
|
|
|
|
// to ensure extra space for expansion
|
|
void Assign(const uint8_t *data, size_t len, size_t capacity) {
|
|
MOZ_RELEASE_ASSERT(len <= capacity);
|
|
Allocate(capacity); // sets len_ = capacity
|
|
memcpy(static_cast<void *>(data_.get()),
|
|
static_cast<const void *>(data), len);
|
|
len_ = len;
|
|
}
|
|
|
|
void Allocate(size_t capacity) {
|
|
data_.reset(new uint8_t[capacity ? capacity : 1]); // Don't depend on new [0].
|
|
len_ = capacity_ = capacity;
|
|
}
|
|
|
|
void EnsureCapacity(size_t capacity) {
|
|
if (capacity_ < capacity) {
|
|
uint8_t *new_data = new uint8_t[ capacity ? capacity : 1];
|
|
memcpy(static_cast<void *>(new_data),
|
|
static_cast<const void *>(data_.get()), len_);
|
|
data_.reset(new_data); // after copying! Deletes old data
|
|
capacity_ = capacity;
|
|
}
|
|
}
|
|
|
|
// used when something writes to the buffer (having checked
|
|
// capacity() or used EnsureCapacity()) and increased the length.
|
|
void SetLength(size_t len) {
|
|
MOZ_RELEASE_ASSERT(len <= capacity_);
|
|
len_ = len;
|
|
}
|
|
|
|
const uint8_t *data() const { return data_.get(); }
|
|
uint8_t *data() { return data_.get(); }
|
|
size_t len() const { return len_; }
|
|
size_t capacity() const { return capacity_; }
|
|
|
|
private:
|
|
UniquePtr<uint8_t[]> data_;
|
|
size_t len_;
|
|
size_t capacity_;
|
|
|
|
DISALLOW_COPY_ASSIGN(DataBuffer);
|
|
};
|
|
|
|
}
|
|
|
|
#endif
|