1 /*
2  * Copyright 2011 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 "SkBitmap.h"
9 #include "SkCanvas.h"
10 #include "SkData.h"
11 #include "SkDiscardableMemoryPool.h"
12 #include "SkImageGenerator.h"
13 #include "SkMatrixUtils.h"
14 #include "SkPaint.h"
15 #include "SkPath.h"
16 #include "SkPixelRef.h"
17 #include "SkRandom.h"
18 #include "SkShader.h"
19 #include "SkSurface.h"
20 #include "Test.h"
21 
22 class FailurePixelRef : public SkPixelRef {
23 public:
FailurePixelRef(const SkImageInfo & info)24     FailurePixelRef(const SkImageInfo& info) : SkPixelRef(info) {}
25 protected:
onNewLockPixels(LockRec *)26     bool onNewLockPixels(LockRec*) override { return false; }
onUnlockPixels()27     void onUnlockPixels() override {}
28 };
29 
30 // crbug.com/295895
31 // Crashing in skia when a pixelref fails in lockPixels
32 //
test_faulty_pixelref(skiatest::Reporter * reporter)33 static void test_faulty_pixelref(skiatest::Reporter* reporter) {
34     // need a cache, but don't expect to use it, so the budget is not critical
35     sk_sp<SkDiscardableMemoryPool> pool(
36         SkDiscardableMemoryPool::Create(10 * 1000, nullptr));
37 
38     SkBitmap bm;
39     const SkImageInfo info = SkImageInfo::MakeN32Premul(100, 100);
40     bm.setInfo(info);
41     bm.setPixelRef(sk_make_sp<FailurePixelRef>(info), 0, 0);
42     // now our bitmap has a pixelref, but we know it will fail to lock
43 
44     auto surface(SkSurface::MakeRasterN32Premul(200, 200));
45     SkCanvas* canvas = surface->getCanvas();
46 
47     const SkFilterQuality levels[] = {
48         kNone_SkFilterQuality,
49         kLow_SkFilterQuality,
50         kMedium_SkFilterQuality,
51         kHigh_SkFilterQuality,
52     };
53 
54     SkPaint paint;
55     canvas->scale(2, 2);    // need a scale, otherwise we may ignore filtering
56     for (size_t i = 0; i < SK_ARRAY_COUNT(levels); ++i) {
57         paint.setFilterQuality(levels[i]);
58         canvas->drawBitmap(bm, 0, 0, &paint);
59     }
60 }
61 
62 ///////////////////////////////////////////////////////////////////////////////
63 
rand_matrix(SkMatrix * mat,SkRandom & rand,unsigned mask)64 static void rand_matrix(SkMatrix* mat, SkRandom& rand, unsigned mask) {
65     mat->setIdentity();
66     if (mask & SkMatrix::kTranslate_Mask) {
67         mat->postTranslate(rand.nextSScalar1(), rand.nextSScalar1());
68     }
69     if (mask & SkMatrix::kScale_Mask) {
70         mat->postScale(rand.nextSScalar1(), rand.nextSScalar1());
71     }
72     if (mask & SkMatrix::kAffine_Mask) {
73         mat->postRotate(rand.nextSScalar1() * 360);
74     }
75     if (mask & SkMatrix::kPerspective_Mask) {
76         mat->setPerspX(rand.nextSScalar1());
77         mat->setPerspY(rand.nextSScalar1());
78     }
79 }
80 
rand_size(SkISize * size,SkRandom & rand)81 static void rand_size(SkISize* size, SkRandom& rand) {
82     size->set(rand.nextU() & 0xFFFF, rand.nextU() & 0xFFFF);
83 }
84 
test_treatAsSprite(skiatest::Reporter * reporter)85 static void test_treatAsSprite(skiatest::Reporter* reporter) {
86 
87     SkMatrix mat;
88     SkISize  size;
89     SkRandom rand;
90 
91     SkPaint noaaPaint;
92     SkPaint aaPaint;
93     aaPaint.setAntiAlias(true);
94 
95     // assert: translate-only no-aa can always be treated as sprite
96     for (int i = 0; i < 1000; ++i) {
97         rand_matrix(&mat, rand, SkMatrix::kTranslate_Mask);
98         for (int j = 0; j < 1000; ++j) {
99             rand_size(&size, rand);
100             REPORTER_ASSERT(reporter, SkTreatAsSprite(mat, size, noaaPaint));
101         }
102     }
103 
104     // assert: rotate/perspect is never treated as sprite
105     for (int i = 0; i < 1000; ++i) {
106         rand_matrix(&mat, rand, SkMatrix::kAffine_Mask | SkMatrix::kPerspective_Mask);
107         for (int j = 0; j < 1000; ++j) {
108             rand_size(&size, rand);
109             REPORTER_ASSERT(reporter, !SkTreatAsSprite(mat, size, noaaPaint));
110             REPORTER_ASSERT(reporter, !SkTreatAsSprite(mat, size, aaPaint));
111         }
112     }
113 
114     size.set(500, 600);
115 
116     const SkScalar tooMuchSubpixel = 100.1f;
117     mat.setTranslate(tooMuchSubpixel, 0);
118     REPORTER_ASSERT(reporter, !SkTreatAsSprite(mat, size, aaPaint));
119     mat.setTranslate(0, tooMuchSubpixel);
120     REPORTER_ASSERT(reporter, !SkTreatAsSprite(mat, size, aaPaint));
121 
122     const SkScalar tinySubPixel = 100.02f;
123     mat.setTranslate(tinySubPixel, 0);
124     REPORTER_ASSERT(reporter, SkTreatAsSprite(mat, size, aaPaint));
125     mat.setTranslate(0, tinySubPixel);
126     REPORTER_ASSERT(reporter, SkTreatAsSprite(mat, size, aaPaint));
127 
128     const SkScalar twoThirds = SK_Scalar1 * 2 / 3;
129     const SkScalar bigScale = (size.width() + twoThirds) / size.width();
130     mat.setScale(bigScale, bigScale);
131     REPORTER_ASSERT(reporter, !SkTreatAsSprite(mat, size, noaaPaint));
132     REPORTER_ASSERT(reporter, !SkTreatAsSprite(mat, size, aaPaint));
133 
134     const SkScalar oneThird = SK_Scalar1 / 3;
135     const SkScalar smallScale = (size.width() + oneThird) / size.width();
136     mat.setScale(smallScale, smallScale);
137     REPORTER_ASSERT(reporter, SkTreatAsSprite(mat, size, noaaPaint));
138     REPORTER_ASSERT(reporter, !SkTreatAsSprite(mat, size, aaPaint));
139 
140     const SkScalar oneFortyth = SK_Scalar1 / 40;
141     const SkScalar tinyScale = (size.width() + oneFortyth) / size.width();
142     mat.setScale(tinyScale, tinyScale);
143     REPORTER_ASSERT(reporter, SkTreatAsSprite(mat, size, noaaPaint));
144     REPORTER_ASSERT(reporter, SkTreatAsSprite(mat, size, aaPaint));
145 }
146 
assert_ifDrawnTo(skiatest::Reporter * reporter,const SkBitmap & bm,bool shouldBeDrawn)147 static void assert_ifDrawnTo(skiatest::Reporter* reporter,
148                              const SkBitmap& bm, bool shouldBeDrawn) {
149     for (int y = 0; y < bm.height(); ++y) {
150         for (int x = 0; x < bm.width(); ++x) {
151             if (shouldBeDrawn) {
152                 if (SK_ColorTRANSPARENT == *bm.getAddr32(x, y)) {
153                     REPORTER_ASSERT(reporter, false);
154                     return;
155                 }
156             } else {
157                 // should not be drawn
158                 if (SK_ColorTRANSPARENT != *bm.getAddr32(x, y)) {
159                     REPORTER_ASSERT(reporter, false);
160                     return;
161                 }
162             }
163         }
164     }
165 }
166 
test_wacky_bitmapshader(skiatest::Reporter * reporter,int width,int height,bool shouldBeDrawn)167 static void test_wacky_bitmapshader(skiatest::Reporter* reporter,
168                                     int width, int height, bool shouldBeDrawn) {
169     SkBitmap dev;
170     dev.allocN32Pixels(0x56F, 0x4f6);
171     dev.eraseColor(SK_ColorTRANSPARENT);  // necessary, so we know if we draw to it
172 
173     SkMatrix matrix;
174 
175     SkCanvas c(dev);
176     matrix.setAll(-119.34097f,
177                   -43.436558f,
178                   93489.945f,
179                   43.436558f,
180                   -119.34097f,
181                   123.98426f,
182                   0, 0, SK_Scalar1);
183     c.concat(matrix);
184 
185     SkBitmap bm;
186     if (bm.tryAllocN32Pixels(width, height)) {
187         // allow this to fail silently, to test the code downstream
188     }
189     bm.eraseColor(SK_ColorRED);
190 
191     matrix.setAll(0.0078740157f,
192                   0,
193                   SkIntToScalar(249),
194                   0,
195                   0.0078740157f,
196                   SkIntToScalar(239),
197                   0, 0, SK_Scalar1);
198     SkPaint paint;
199     paint.setShader(SkShader::MakeBitmapShader(bm, SkShader::kRepeat_TileMode,
200                                                SkShader::kRepeat_TileMode, &matrix));
201 
202     SkRect r = SkRect::MakeXYWH(681, 239, 695, 253);
203     c.drawRect(r, paint);
204 
205     assert_ifDrawnTo(reporter, dev, shouldBeDrawn);
206 }
207 
208 /*
209  *  Original bug was asserting that the matrix-proc had generated a (Y) value
210  *  that was out of range. This led (in the release build) to the sampler-proc
211  *  reading memory out-of-bounds of the original bitmap.
212  *
213  *  We were numerically overflowing our 16bit coordinates that we communicate
214  *  between these two procs. The fixes was in two parts:
215  *
216  *  1. Just don't draw bitmaps larger than 64K-1 in width or height, since we
217  *     can't represent those coordinates in our transport format (yet).
218  *  2. Perform an unsigned shift during the calculation, so we don't get
219  *     sign-extension bleed when packing the two values (X,Y) into our 32bit
220  *     slot.
221  *
222  *  This tests exercises the original setup, plus 3 more to ensure that we can,
223  *  in fact, handle bitmaps at 64K-1 (assuming we don't exceed the total
224  *  memory allocation limit).
225  */
test_giantrepeat_crbug118018(skiatest::Reporter * reporter)226 static void test_giantrepeat_crbug118018(skiatest::Reporter* reporter) {
227     static const struct {
228         int fWidth;
229         int fHeight;
230         bool fExpectedToDraw;
231     } gTests[] = {
232         { 0x1b294, 0x7f,  false },   // crbug 118018 (width exceeds 64K)
233         { 0xFFFF, 0x7f,    true },   // should draw, test max width
234         { 0x7f, 0xFFFF,    true },   // should draw, test max height
235         { 0xFFFF, 0xFFFF, false },   // allocation fails (too much RAM)
236     };
237 
238     for (size_t i = 0; i < SK_ARRAY_COUNT(gTests); ++i) {
239         test_wacky_bitmapshader(reporter,
240                                 gTests[i].fWidth, gTests[i].fHeight,
241                                 gTests[i].fExpectedToDraw);
242     }
243 }
244 
245 ///////////////////////////////////////////////////////////////////////////////
246 
test_nan_antihair()247 static void test_nan_antihair() {
248     SkBitmap bm;
249     bm.allocN32Pixels(20, 20);
250 
251     SkCanvas canvas(bm);
252 
253     SkPath path;
254     path.moveTo(0, 0);
255     path.lineTo(10, SK_ScalarNaN);
256 
257     SkPaint paint;
258     paint.setAntiAlias(true);
259     paint.setStyle(SkPaint::kStroke_Style);
260 
261     // before our fix to SkScan_Antihair.cpp to check for integral NaN (0x800...)
262     // this would trigger an assert/crash.
263     //
264     // see rev. 3558
265     canvas.drawPath(path, paint);
266 }
267 
check_for_all_zeros(const SkBitmap & bm)268 static bool check_for_all_zeros(const SkBitmap& bm) {
269     SkAutoLockPixels alp(bm);
270 
271     size_t count = bm.width() * bm.bytesPerPixel();
272     for (int y = 0; y < bm.height(); y++) {
273         const uint8_t* ptr = reinterpret_cast<const uint8_t*>(bm.getAddr(0, y));
274         for (size_t i = 0; i < count; i++) {
275             if (ptr[i]) {
276                 return false;
277             }
278         }
279     }
280     return true;
281 }
282 
283 static const int gWidth = 256;
284 static const int gHeight = 256;
285 
create(SkBitmap * bm,SkColor color)286 static void create(SkBitmap* bm, SkColor color) {
287     bm->allocN32Pixels(gWidth, gHeight);
288     bm->eraseColor(color);
289 }
290 
DEF_TEST(DrawBitmapRect,reporter)291 DEF_TEST(DrawBitmapRect, reporter) {
292     SkBitmap src, dst;
293 
294     create(&src, 0xFFFFFFFF);
295     create(&dst, 0);
296 
297     SkCanvas canvas(dst);
298 
299     SkIRect srcR = { gWidth, 0, gWidth + 16, 16 };
300     SkRect  dstR = { 0, 0, SkIntToScalar(16), SkIntToScalar(16) };
301 
302     canvas.drawBitmapRect(src, srcR, dstR, nullptr);
303 
304     // ensure that we draw nothing if srcR does not intersect the bitmap
305     REPORTER_ASSERT(reporter, check_for_all_zeros(dst));
306 
307     test_nan_antihair();
308     test_giantrepeat_crbug118018(reporter);
309 
310     test_treatAsSprite(reporter);
311     test_faulty_pixelref(reporter);
312 }
313