OLD | NEW |
(Empty) | |
| 1 // Copyright (c) 2009 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 #ifndef OTS_MEMORY_STREAM_H_ |
| 6 #define OTS_MEMORY_STREAM_H_ |
| 7 |
| 8 #include <cstring> |
| 9 #include <limits> |
| 10 |
| 11 #include "opentype-sanitiser.h" |
| 12 |
| 13 namespace ots { |
| 14 |
| 15 class MemoryStream : public OTSStream { |
| 16 public: |
| 17 MemoryStream(void *ptr, size_t length) |
| 18 : ptr_(ptr), length_(length), off_(0) { |
| 19 } |
| 20 |
| 21 virtual bool WriteRaw(const void *data, size_t length) { |
| 22 if ((off_ + length > length_) || |
| 23 (length > std::numeric_limits<size_t>::max() - off_)) { |
| 24 return false; |
| 25 } |
| 26 std::memcpy(static_cast<char*>(ptr_) + off_, data, length); |
| 27 off_ += length; |
| 28 return true; |
| 29 } |
| 30 |
| 31 virtual bool Seek(off_t position) { |
| 32 if (position < 0) return false; |
| 33 if (static_cast<size_t>(position) > length_) return false; |
| 34 off_ = position; |
| 35 return true; |
| 36 } |
| 37 |
| 38 virtual off_t Tell() const { |
| 39 return off_; |
| 40 } |
| 41 |
| 42 private: |
| 43 void* const ptr_; |
| 44 size_t length_; |
| 45 off_t off_; |
| 46 }; |
| 47 |
| 48 class ExpandingMemoryStream : public OTSStream { |
| 49 public: |
| 50 ExpandingMemoryStream(size_t initial, size_t limit) |
| 51 : length_(initial), limit_(limit), off_(0) { |
| 52 ptr_ = new uint8_t[length_]; |
| 53 } |
| 54 |
| 55 ~ExpandingMemoryStream() { |
| 56 delete[] static_cast<uint8_t*>(ptr_); |
| 57 } |
| 58 |
| 59 void* get() const { |
| 60 return ptr_; |
| 61 } |
| 62 |
| 63 bool WriteRaw(const void *data, size_t length) { |
| 64 if ((off_ + length > length_) || |
| 65 (length > std::numeric_limits<size_t>::max() - off_)) { |
| 66 if (length_ == limit_) |
| 67 return false; |
| 68 size_t new_length = (length_ + 1) * 2; |
| 69 if (new_length < length_) |
| 70 return false; |
| 71 if (new_length > limit_) |
| 72 new_length = limit_; |
| 73 uint8_t* new_buf = new uint8_t[new_length]; |
| 74 std::memcpy(new_buf, ptr_, length_); |
| 75 length_ = new_length; |
| 76 delete[] static_cast<uint8_t*>(ptr_); |
| 77 ptr_ = new_buf; |
| 78 return WriteRaw(data, length); |
| 79 } |
| 80 std::memcpy(static_cast<char*>(ptr_) + off_, data, length); |
| 81 off_ += length; |
| 82 return true; |
| 83 } |
| 84 |
| 85 bool Seek(off_t position) { |
| 86 if (position < 0) return false; |
| 87 if (static_cast<size_t>(position) > length_) return false; |
| 88 off_ = position; |
| 89 return true; |
| 90 } |
| 91 |
| 92 off_t Tell() const { |
| 93 return off_; |
| 94 } |
| 95 |
| 96 private: |
| 97 void* ptr_; |
| 98 size_t length_; |
| 99 const size_t limit_; |
| 100 off_t off_; |
| 101 }; |
| 102 |
| 103 } // namespace ots |
| 104 |
| 105 #endif // OTS_MEMORY_STREAM_H_ |
OLD | NEW |