Chromium Code Reviews| Index: src/core/SkOrderedReadBuffer.cpp |
| diff --git a/src/core/SkOrderedReadBuffer.cpp b/src/core/SkOrderedReadBuffer.cpp |
| index d9aa8bd1d751d35c722bb05d79ddd291d0225461..31841182b9c47edbce0e0f85b1b9bef466620fe8 100644 |
| --- a/src/core/SkOrderedReadBuffer.cpp |
| +++ b/src/core/SkOrderedReadBuffer.cpp |
| @@ -137,38 +137,37 @@ void SkOrderedReadBuffer::readPath(SkPath* path) { |
| fReader.readPath(path); |
| } |
| -uint32_t SkOrderedReadBuffer::readByteArray(void* value) { |
| - const uint32_t length = fReader.readU32(); |
| - memcpy(value, fReader.skip(SkAlign4(length)), length); |
| - return length; |
| +bool SkOrderedReadBuffer::readArray(void* value, size_t size, size_t elementSize) { |
| + const size_t count = this->getArrayCount(); |
| + if (count == size) { |
|
Stephen White
2013/10/31 15:19:28
Nit: prefer the early-out to be the error case rat
|
| + (void)fReader.skip(sizeof(uint32_t)); // Skip array count |
| + const size_t byteLength = count * elementSize; |
|
Stephen White
2013/10/31 15:19:28
Should we be checking for overflow here? At worst,
|
| + memcpy(value, fReader.skip(SkAlign4(byteLength)), byteLength); |
| + return true; |
| + } |
| + SkASSERT(false); |
| + fReader.skip(fReader.available()); |
| + return false; |
| +} |
| + |
| +bool SkOrderedReadBuffer::readByteArray(void* value, size_t size) { |
| + return readArray(static_cast<unsigned char*>(value), size, sizeof(unsigned char)); |
| } |
| -uint32_t SkOrderedReadBuffer::readColorArray(SkColor* colors) { |
| - const uint32_t count = fReader.readU32(); |
| - const uint32_t byteLength = count * sizeof(SkColor); |
| - memcpy(colors, fReader.skip(SkAlign4(byteLength)), byteLength); |
| - return count; |
| +bool SkOrderedReadBuffer::readColorArray(SkColor* colors, size_t size) { |
| + return readArray(colors, size, sizeof(SkColor)); |
| } |
| -uint32_t SkOrderedReadBuffer::readIntArray(int32_t* values) { |
| - const uint32_t count = fReader.readU32(); |
| - const uint32_t byteLength = count * sizeof(int32_t); |
| - memcpy(values, fReader.skip(SkAlign4(byteLength)), byteLength); |
| - return count; |
| +bool SkOrderedReadBuffer::readIntArray(int32_t* values, size_t size) { |
| + return readArray(values, size, sizeof(int32_t)); |
| } |
| -uint32_t SkOrderedReadBuffer::readPointArray(SkPoint* points) { |
| - const uint32_t count = fReader.readU32(); |
| - const uint32_t byteLength = count * sizeof(SkPoint); |
| - memcpy(points, fReader.skip(SkAlign4(byteLength)), byteLength); |
| - return count; |
| +bool SkOrderedReadBuffer::readPointArray(SkPoint* points, size_t size) { |
| + return readArray(points, size, sizeof(SkPoint)); |
| } |
| -uint32_t SkOrderedReadBuffer::readScalarArray(SkScalar* values) { |
| - const uint32_t count = fReader.readU32(); |
| - const uint32_t byteLength = count * sizeof(SkScalar); |
| - memcpy(values, fReader.skip(SkAlign4(byteLength)), byteLength); |
| - return count; |
| +bool SkOrderedReadBuffer::readScalarArray(SkScalar* values, size_t size) { |
| + return readArray(values, size, sizeof(SkScalar)); |
| } |
| uint32_t SkOrderedReadBuffer::getArrayCount() { |