diff --git a/test/fuzztest/datatransmitmgr/datatransmitmgr_fuzzer/BUILD.gn b/test/fuzztest/datatransmitmgr/datatransmitmgr_fuzzer/BUILD.gn index e41fa01a81325d3ec33746a267f0c8cb7e8ccd4e..870929359e52442307c94d6a6b22b6b212cf3fd9 100644 --- a/test/fuzztest/datatransmitmgr/datatransmitmgr_fuzzer/BUILD.gn +++ b/test/fuzztest/datatransmitmgr/datatransmitmgr_fuzzer/BUILD.gn @@ -32,18 +32,10 @@ ohos_fuzztest("DataTransmitMgrFuzzTest") { sources = [ "datatransmitmgr_fuzzer.cpp" ] - deps = [ - "//base/security/access_token/interfaces/innerkits/accesstoken:libaccesstoken_sdk", - "//base/security/access_token/interfaces/innerkits/nativetoken:libnativetoken", - "//base/security/access_token/interfaces/innerkits/token_setproc:libtoken_setproc", - "//base/security/dataclassification/interfaces/innerkits/datatransmitmgr:data_transit_mgr", - ] + deps = [ "../../../../interfaces/innerkits/datatransmitmgr:data_transit_mgr" ] external_deps = [ "c_utils:utils", - "dsoftbus:softbus_client", "hilog:libhilog", ] - cflags_cc = [ "--coverage" ] - ldflags = [ "--coverage" ] } diff --git a/test/fuzztest/datatransmitmgr/datatransmitmgr_fuzzer/datatransmitmgr_fuzzer.cpp b/test/fuzztest/datatransmitmgr/datatransmitmgr_fuzzer/datatransmitmgr_fuzzer.cpp index 465ee9ec98e256b0e8d942d24dad855affe128e2..f32377969b62ba5c3b624981ea9dc15308b6fc60 100644 --- a/test/fuzztest/datatransmitmgr/datatransmitmgr_fuzzer/datatransmitmgr_fuzzer.cpp +++ b/test/fuzztest/datatransmitmgr/datatransmitmgr_fuzzer/datatransmitmgr_fuzzer.cpp @@ -15,15 +15,11 @@ #include "datatransmitmgr_fuzzer.h" -#include -#include +#include +#include #include "securec.h" #include "dev_slinfo_mgr.h" -#include "softbus_bus_center.h" -#include "nativetoken_kit.h" -#include "token_setproc.h" -#include "accesstoken_kit.h" namespace OHOS { static void tmpCallback(DEVSLQueryParams *queryParams, int32_t result, uint32_t levelInfo) @@ -31,75 +27,29 @@ namespace OHOS { return; } - static void GetLocalUdid(DEVSLQueryParams *queryParams) - { - const char *pkgName = "ohos.dslm"; - NodeBasicInfo info; - (void)GetLocalNodeDeviceInfo(pkgName, &info); - if (GetNodeKeyInfo(pkgName, info.networkId, NODE_KEY_UDID, (uint8_t *)(queryParams->udid), UDID_BUF_LEN) != 0) { - return; - } - queryParams->udidLen = MAX_UDID_LENGTH; - } - void FuzzDoDataTransmitMgr(const uint8_t* data, size_t size) { - if (data == nullptr) { - return; - } - if (size < MAX_UDID_LENGTH) { + if (data == nullptr || size > MAX_UDID_LENGTH) { return; } + uint32_t levelInfo = 0; DEVSLQueryParams queryParams; (void)memset_s(&queryParams, sizeof(DEVSLQueryParams), 0, sizeof(DEVSLQueryParams)); queryParams.udidLen = MAX_UDID_LENGTH; (void)memcpy_s(queryParams.udid, queryParams.udidLen, data, MAX_UDID_LENGTH); - int32_t ret = DATASL_OnStart(); - if (ret != DEVSL_SUCCESS) { - return; - } - (void)DATASL_GetHighestSecLevelAsync(&queryParams, tmpCallback); - (void)DATASL_GetHighestSecLevelAsync(nullptr, tmpCallback); - (void)DATASL_GetHighestSecLevelAsync(&queryParams, nullptr); - (void)DATASL_GetHighestSecLevel(&queryParams, &levelInfo); - (void)DATASL_GetHighestSecLevel(nullptr, &levelInfo); - (void)DATASL_GetHighestSecLevel(&queryParams, nullptr); - GetLocalUdid(&queryParams); + (void)DATASL_OnStart(); (void)DATASL_GetHighestSecLevelAsync(&queryParams, tmpCallback); (void)DATASL_GetHighestSecLevel(&queryParams, &levelInfo); DATASL_OnStop(); } } -static void NativeTokenGetFUZZ() -{ - uint64_t tokenId; - const char **perms = new const char *[1]; - perms[0] = "ohos.permission.DISTRIBUTED_DATASYNC"; - NativeTokenInfoParams infoInstance = { - .acls = nullptr, - .dcaps = nullptr, - .aclsNum = 0, - .dcapsNum = 0, - .permsNum = 1, - .perms = perms, - .aplStr = "system_basic", - }; - - infoInstance.processName = "DataTransmitMgrFuzzTest"; - tokenId = GetAccessTokenId(&infoInstance); - SetSelfTokenID(tokenId); - OHOS::Security::AccessToken::AccessTokenKit::ReloadNativeTokenInfo(); - delete[] perms; -} - /* Fuzzer entry point */ extern "C" int LLVMFuzzerTestOneInput(const uint8_t* data, size_t size) { /* Run your code on data */ - NativeTokenGetFUZZ(); OHOS::FuzzDoDataTransmitMgr(data, size); return 0; } \ No newline at end of file