Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(271)

Unified Diff: src/codec/SkCodec.cpp

Issue 1472123002: Make SkCodec support peek() and read() (Closed) Base URL: https://skia.googlesource.com/skia.git@master
Patch Set: Make IsWbmp call read_header Created 5 years, 1 month ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
Index: src/codec/SkCodec.cpp
diff --git a/src/codec/SkCodec.cpp b/src/codec/SkCodec.cpp
index 11eb1f98479461b7307695ac49d44cde6f85cecb..27fe88befb07e2ed377a115001b7edb93f897349 100644
--- a/src/codec/SkCodec.cpp
+++ b/src/codec/SkCodec.cpp
@@ -20,7 +20,7 @@
#include "SkWebpCodec.h"
struct DecoderProc {
- bool (*IsFormat)(SkStream*);
+ bool (*IsFormat)(const char*, size_t);
SkCodec* (*NewFromStream)(SkStream*);
};
@@ -42,23 +42,29 @@ SkCodec* SkCodec::NewFromStream(SkStream* stream,
}
SkAutoTDelete<SkStream> streamDeleter(stream);
-
+
+ // 14 is enough to read all of the supported types.
scroggo 2015/11/25 22:08:35 I've convinced myself that 14 bytes is enough to s
msarett 2015/11/30 14:08:11 This matches my understanding as well. Looks like
scroggo 2015/11/30 20:02:30 Added a test.
+ size_t bytesRead = 14;
+ char buffer[bytesRead];
+ if (!stream->peek(buffer, bytesRead)) {
+ SkCodecPrintf("Could not peek!\n");
+ // It is possible the stream does not support peeking, but does support
+ // rewinding, so try reading.
+ // (Is there a valid image that is less than 14 bytes, causing peek to
+ // fail, even though we read enough?)
msarett 2015/11/30 14:08:11 Good question. I don't know, but I would guess th
scroggo 2015/11/30 20:02:30 Yeah, that should do it: 1 + 1 (headers) + 1 (wid
+ bytesRead = stream->read(buffer, bytesRead);
+ if (!stream->rewind()) {
+ return nullptr;
+ }
+ }
SkAutoTDelete<SkCodec> codec(nullptr);
// PNG is special, since we want to be able to supply an SkPngChunkReader.
// But this code follows the same pattern as the loop.
- const bool isPng = SkPngCodec::IsPng(stream);
- if (!stream->rewind()) {
- return NULL;
- }
- if (isPng) {
+ if (SkPngCodec::IsPng((const char*)buffer, bytesRead)) {
codec.reset(SkPngCodec::NewFromStream(streamDeleter.detach(), chunkReader));
} else {
for (DecoderProc proc : gDecoderProcs) {
- const bool correctFormat = proc.IsFormat(stream);
- if (!stream->rewind()) {
- return nullptr;
- }
- if (correctFormat) {
+ if (proc.IsFormat(buffer, bytesRead)) {
codec.reset(proc.NewFromStream(streamDeleter.detach()));
break;
}
« no previous file with comments | « src/codec/SkBmpCodec.cpp ('k') | src/codec/SkCodec_libgif.h » ('j') | src/codec/SkCodec_wbmp.cpp » ('J')

Powered by Google App Engine
This is Rietveld 408576698