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 "createsmsmessage_fuzzer.h"
17
18 #define private public
19
20 #include "addsmstoken_fuzzer.h"
21 #include "sms_service.h"
22
23 using namespace OHOS::Telephony;
24 namespace OHOS {
25 static bool g_isInited = false;
26
IsServiceInited()27 bool IsServiceInited()
28 {
29 if (!g_isInited) {
30 DelayedSingleton<SmsService>::GetInstance()->OnStart();
31 if (DelayedSingleton<SmsService>::GetInstance()->GetServiceRunningState() ==
32 static_cast<int32_t>(Telephony::ServiceRunningState::STATE_RUNNING)) {
33 g_isInited = true;
34 }
35 }
36 return g_isInited;
37 }
38
CreateSmsMessage(const uint8_t * data,size_t size)39 void CreateSmsMessage(const uint8_t *data, size_t size)
40 {
41 if (!IsServiceInited()) {
42 return;
43 }
44 MessageParcel dataParcel;
45 MessageParcel replyParcel;
46 MessageOption option(MessageOption::TF_SYNC);
47 std::string pdu(reinterpret_cast<const char *>(data), size);
48 dataParcel.WriteString(pdu);
49 dataParcel.WriteString("3gpp");
50 DelayedSingleton<SmsService>::GetInstance()->OnCreateMessage(dataParcel, replyParcel, option);
51 }
52
CreateCbMessage(const uint8_t * data,size_t size)53 void CreateCbMessage(const uint8_t *data, size_t size)
54 {
55 if (!IsServiceInited()) {
56 return;
57 }
58 std::string strPdu(reinterpret_cast<const char *>(data), size);
59 GsmCbCodec::CreateCbMessage(strPdu);
60 }
61
DoCreateMessageMyAPI(const uint8_t * data,size_t size)62 void DoCreateMessageMyAPI(const uint8_t *data, size_t size)
63 {
64 if (data == nullptr || size == 0) {
65 return;
66 }
67 CreateSmsMessage(data, size);
68 CreateCbMessage(data, size);
69 }
70 } // namespace OHOS
71
72 /* Fuzzer entry point */
LLVMFuzzerTestOneInput(const uint8_t * data,size_t size)73 extern "C" int LLVMFuzzerTestOneInput(const uint8_t *data, size_t size)
74 {
75 /* Run your code on data */
76 OHOS::AddSmsTokenFuzzer token;
77 OHOS::DoCreateMessageMyAPI(data, size);
78 return 0;
79 }
80