Home | History | Annotate | Download | only in tests 
      1 /*
      2  * Copyright 2017 Google Inc.
      3  *
      4  * Use of this source code is governed by a BSD-style license that can be
      5  * found in the LICENSE file.
      6  */
      7 
      8 #include "Resources.h"
      9 #include "Test.h"
     10 
     11 #include "SkBitmap.h"
     12 #include "SkCodec.h"
     13 #include "SkData.h"
     14 #include "SkMakeUnique.h"
     15 #include "SkStream.h"
     16 
     17 namespace {
     18 // This class wraps another SkStream. It does not own the underlying stream, so
     19 // that the underlying stream can be reused starting from where the first
     20 // client left off. This mimics Android's JavaInputStreamAdaptor.
     21 class UnowningStream : public SkStream {
     22 public:
     23     explicit UnowningStream(SkStream* stream)
     24         : fStream(stream)
     25     {}
     26 
     27     size_t read(void* buf, size_t bytes) override {
     28         return fStream->read(buf, bytes);
     29     }
     30 
     31     bool rewind() override {
     32         return fStream->rewind();
     33     }
     34 
     35     bool isAtEnd() const override {
     36         return fStream->isAtEnd();
     37     }
     38 private:
     39     SkStream* fStream; // Unowned.
     40 };
     41 } // namespace
     42 
     43 // Test that some SkCodecs do not attempt to read input beyond the logical
     44 // end of the data. Some other SkCodecs do, but some Android apps rely on not
     45 // doing so for PNGs. Test on other formats that work.
     46 DEF_TEST(Codec_end, r) {
     47     for (const char* path : { "images/plane.png",
     48                               "images/yellow_rose.png",
     49                               "images/plane_interlaced.png",
     50                               "images/google_chrome.ico",
     51                               "images/color_wheel.ico",
     52                               "images/mandrill.wbmp",
     53                               "images/randPixels.bmp",
     54                               }) {
     55         sk_sp<SkData> data = GetResourceAsData(path);
     56         if (!data) {
     57             continue;
     58         }
     59 
     60         const int kNumImages = 2;
     61         const size_t size = data->size();
     62         sk_sp<SkData> multiData = SkData::MakeUninitialized(size * kNumImages);
     63         void* dst = multiData->writable_data();
     64         for (int i = 0; i < kNumImages; i++) {
     65             memcpy(SkTAddOffset<void>(dst, size * i), data->data(), size);
     66         }
     67         data.reset();
     68 
     69         SkMemoryStream stream(std::move(multiData));
     70         for (int i = 0; i < kNumImages; ++i) {
     71             std::unique_ptr<SkCodec> codec(SkCodec::MakeFromStream(
     72                                                    skstd::make_unique<UnowningStream>(&stream)));
     73             if (!codec) {
     74                 ERRORF(r, "Failed to create a codec from %s, iteration %i", path, i);
     75                 continue;
     76             }
     77 
     78             auto info = codec->getInfo().makeColorType(kN32_SkColorType);
     79             SkBitmap bm;
     80             bm.allocPixels(info);
     81 
     82             auto result = codec->getPixels(bm.info(), bm.getPixels(), bm.rowBytes());
     83             if (result != SkCodec::kSuccess) {
     84                 ERRORF(r, "Failed to getPixels from %s, iteration %i error %i", path, i, result);
     85                 continue;
     86             }
     87         }
     88     }
     89 }
     90