diff --git a/test/fuzztest/device_auth_service/devauthservprocessauthdata_fuzzer/devauthservprocessauthdata_fuzzer.cpp b/test/fuzztest/device_auth_service/devauthservprocessauthdata_fuzzer/devauthservprocessauthdata_fuzzer.cpp index 338e0f03..d2a09b2c 100644 --- a/test/fuzztest/device_auth_service/devauthservprocessauthdata_fuzzer/devauthservprocessauthdata_fuzzer.cpp +++ b/test/fuzztest/device_auth_service/devauthservprocessauthdata_fuzzer/devauthservprocessauthdata_fuzzer.cpp @@ -18,6 +18,7 @@ #include "device_auth_defines.h" #include "hc_dev_info.h" #include "hc_log.h" +#include "hc_types.h" #include "json_utils.h" #include "securec.h" #include @@ -61,7 +62,18 @@ namespace OHOS { InitDeviceAuthService(); const GroupAuthManager *gaInstance = GetGaInstance(); int64_t reqId = 123; - gaInstance->processData(reqId, data, size, &g_gaCallback); + uint8_t *authData = reinterpret_cast(HcMalloc(size + 1, 0)); + if (authData == nullptr) { + DestroyDeviceAuthService(); + return false; + } + if (memcpy_s(authData, size, data, size) != EOK) { + HcFree(authData); + DestroyDeviceAuthService(); + return false; + } + gaInstance->processData(reqId, authData, size, &g_gaCallback); + HcFree(authData); sleep(1); DestroyDeviceAuthService(); return true; diff --git a/test/fuzztest/device_auth_service/devauthservprocessbinddata_fuzzer/devauthservprocessbinddata_fuzzer.cpp b/test/fuzztest/device_auth_service/devauthservprocessbinddata_fuzzer/devauthservprocessbinddata_fuzzer.cpp index 5832b5ce..0e18e4c2 100644 --- a/test/fuzztest/device_auth_service/devauthservprocessbinddata_fuzzer/devauthservprocessbinddata_fuzzer.cpp +++ b/test/fuzztest/device_auth_service/devauthservprocessbinddata_fuzzer/devauthservprocessbinddata_fuzzer.cpp @@ -18,6 +18,7 @@ #include "device_auth_defines.h" #include "hc_dev_info.h" #include "hc_log.h" +#include "hc_types.h" #include "json_utils.h" #include "securec.h" #include @@ -63,7 +64,18 @@ namespace OHOS { const DeviceGroupManager *gmInstance = GetGmInstance(); gmInstance->regCallback(appId.c_str(), &g_gmCallback); int64_t reqId = 123; - gmInstance->processData(reqId, data, size); + uint8_t *bindData = reinterpret_cast(HcMalloc(size + 1, 0)); + if (bindData == nullptr) { + DestroyDeviceAuthService(); + return false; + } + if (memcpy_s(bindData, size, data, size) != EOK) { + HcFree(bindData); + DestroyDeviceAuthService(); + return false; + } + gmInstance->processData(reqId, bindData, size); + HcFree(bindData); sleep(1); DestroyDeviceAuthService(); return true;