1 /*
2  * Copyright 2020 The Android Open Source Project
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  *      http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16 
17 #include <gtest/gtest.h>
18 
19 #include <binder/Binder.h>
20 #include <binder/Parcel.h>
21 
22 #include <gui/LayerState.h>
23 
24 namespace android {
25 using gui::ScreenCaptureResults;
26 
27 namespace test {
28 
TEST(LayerStateTest,ParcellingDisplayCaptureArgs)29 TEST(LayerStateTest, ParcellingDisplayCaptureArgs) {
30     DisplayCaptureArgs args;
31     args.pixelFormat = ui::PixelFormat::RGB_565;
32     args.sourceCrop = Rect(0, 0, 500, 200);
33     args.frameScaleX = 2;
34     args.frameScaleY = 4;
35     args.captureSecureLayers = true;
36     args.displayToken = new BBinder();
37     args.width = 10;
38     args.height = 20;
39     args.useIdentityTransform = true;
40     args.grayscale = true;
41 
42     Parcel p;
43     args.write(p);
44     p.setDataPosition(0);
45 
46     DisplayCaptureArgs args2;
47     args2.read(p);
48 
49     ASSERT_EQ(args.pixelFormat, args2.pixelFormat);
50     ASSERT_EQ(args.sourceCrop, args2.sourceCrop);
51     ASSERT_EQ(args.frameScaleX, args2.frameScaleX);
52     ASSERT_EQ(args.frameScaleY, args2.frameScaleY);
53     ASSERT_EQ(args.captureSecureLayers, args2.captureSecureLayers);
54     ASSERT_EQ(args.displayToken, args2.displayToken);
55     ASSERT_EQ(args.width, args2.width);
56     ASSERT_EQ(args.height, args2.height);
57     ASSERT_EQ(args.useIdentityTransform, args2.useIdentityTransform);
58     ASSERT_EQ(args.grayscale, args2.grayscale);
59 }
60 
TEST(LayerStateTest,ParcellingLayerCaptureArgs)61 TEST(LayerStateTest, ParcellingLayerCaptureArgs) {
62     LayerCaptureArgs args;
63     args.pixelFormat = ui::PixelFormat::RGB_565;
64     args.sourceCrop = Rect(0, 0, 500, 200);
65     args.frameScaleX = 2;
66     args.frameScaleY = 4;
67     args.captureSecureLayers = true;
68     args.layerHandle = new BBinder();
69     args.excludeHandles = {new BBinder(), new BBinder()};
70     args.childrenOnly = false;
71     args.grayscale = true;
72 
73     Parcel p;
74     args.write(p);
75     p.setDataPosition(0);
76 
77     LayerCaptureArgs args2;
78     args2.read(p);
79 
80     ASSERT_EQ(args.pixelFormat, args2.pixelFormat);
81     ASSERT_EQ(args.sourceCrop, args2.sourceCrop);
82     ASSERT_EQ(args.frameScaleX, args2.frameScaleX);
83     ASSERT_EQ(args.frameScaleY, args2.frameScaleY);
84     ASSERT_EQ(args.captureSecureLayers, args2.captureSecureLayers);
85     ASSERT_EQ(args.layerHandle, args2.layerHandle);
86     ASSERT_EQ(args.excludeHandles, args2.excludeHandles);
87     ASSERT_EQ(args.childrenOnly, args2.childrenOnly);
88     ASSERT_EQ(args.grayscale, args2.grayscale);
89 }
90 
TEST(LayerStateTest,ParcellingScreenCaptureResults)91 TEST(LayerStateTest, ParcellingScreenCaptureResults) {
92     ScreenCaptureResults results;
93     results.buffer = new GraphicBuffer(100, 200, PIXEL_FORMAT_RGBA_8888, 1, 0);
94     results.fence = new Fence(dup(fileno(tmpfile())));
95     results.capturedSecureLayers = true;
96     results.capturedDataspace = ui::Dataspace::DISPLAY_P3;
97     results.result = BAD_VALUE;
98 
99     Parcel p;
100     results.writeToParcel(&p);
101     p.setDataPosition(0);
102 
103     ScreenCaptureResults results2;
104     results2.readFromParcel(&p);
105 
106     // GraphicBuffer object is reallocated so compare the data in the graphic buffer
107     // rather than the object itself
108     ASSERT_EQ(results.buffer->getWidth(), results2.buffer->getWidth());
109     ASSERT_EQ(results.buffer->getHeight(), results2.buffer->getHeight());
110     ASSERT_EQ(results.buffer->getPixelFormat(), results2.buffer->getPixelFormat());
111     ASSERT_EQ(results.fence->isValid(), results2.fence->isValid());
112     ASSERT_EQ(results.capturedSecureLayers, results2.capturedSecureLayers);
113     ASSERT_EQ(results.capturedDataspace, results2.capturedDataspace);
114     ASSERT_EQ(results.result, results2.result);
115 }
116 
117 } // namespace test
118 } // namespace android
119