1 /*
2  * Copyright (c) 2023 Huawei Device Co., Ltd.
3  * Licensed under the Apache License, Version 2.0 (the "License");
4  * you may not use this file except in compliance with the License.
5  * You may obtain a copy of the License at
6  *
7  *     http://www.apache.org/licenses/LICENSE-2.0
8  *
9  * Unless required by applicable law or agreed to in writing, software
10  * distributed under the License is distributed on an "AS IS" BASIS,
11  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12  * See the License for the specific language governing permissions and
13  * limitations under the License.
14  */
15 
16 #include "smooth_zoom_fuzzer.h"
17 #include "camera_timer.h"
18 #include "hstream_capture.h"
19 #include "message_parcel.h"
20 #include "nativetoken_kit.h"
21 #include "token_setproc.h"
22 #include "accesstoken_kit.h"
23 #include "camera_metadata_info.h"
24 #include "metadata_utils.h"
25 
26 namespace {
27 
28 const int32_t LIMITSIZE = 12;
29 
30 }
31 
32 namespace OHOS {
33 namespace CameraStandard {
34 
35 bool SmoothZoomFuzzer::hasPermission = false;
36 SmoothZoom *SmoothZoomFuzzer::fuzz = nullptr;
37 
CheckPermission()38 void SmoothZoomFuzzer::CheckPermission()
39 {
40     if (!hasPermission) {
41         uint64_t tokenId;
42         const char *perms[0];
43         perms[0] = "ohos.permission.CAMERA";
44         NativeTokenInfoParams infoInstance = { .dcapsNum = 0, .permsNum = 1, .aclsNum = 0, .dcaps = NULL,
45             .perms = perms, .acls = NULL, .processName = "camera_capture", .aplStr = "system_basic",
46         };
47         tokenId = GetAccessTokenId(&infoInstance);
48         SetSelfTokenID(tokenId);
49         OHOS::Security::AccessToken::AccessTokenKit::ReloadNativeTokenInfo();
50         hasPermission = true;
51     }
52 }
53 
Test(uint8_t * rawData,size_t size)54 void SmoothZoomFuzzer::Test(uint8_t *rawData, size_t size)
55 {
56     if (rawData == nullptr || size < LIMITSIZE) {
57         return;
58     }
59     CheckPermission();
60 
61     if (fuzz == nullptr) {
62         fuzz = new SmoothZoom();
63     }
64     MessageParcel data;
65     data.WriteRawData(rawData, size);
66 
67     SmoothZoomType mode = SmoothZoomType::NORMAL;
68     auto alg = fuzz->GetZoomAlgorithm(mode);
69 
70     data.RewindRead(0);
71     float currentZoom = data.ReadFloat();
72     float targetZoom = data.ReadFloat();
73     float frameInterval = data.ReadFloat();
74     alg->GetZoomArray(currentZoom, targetZoom, frameInterval);
75 }
76 
77 } // namespace CameraStandard
78 } // namespace OHOS
79 
80 /* Fuzzer entry point */
LLVMFuzzerTestOneInput(uint8_t * data,size_t size)81 extern "C" int LLVMFuzzerTestOneInput(uint8_t *data, size_t size)
82 {
83     /* Run your code on data */
84     OHOS::CameraStandard::SmoothZoomFuzzer::Test(data, size);
85     return 0;
86 }