diff --git a/av_transport/av_trans_control_center/test/fuzztest/avtransonsessionclosed_fuzzer/avtransonsessionclosed_fuzzer.cpp b/av_transport/av_trans_control_center/test/fuzztest/avtransonsessionclosed_fuzzer/avtransonsessionclosed_fuzzer.cpp index dc775aaf807f5952550fed70544339e3763998e9..92ac049afc010c92b056767d6c8daa6916d29a10 100644 --- a/av_transport/av_trans_control_center/test/fuzztest/avtransonsessionclosed_fuzzer/avtransonsessionclosed_fuzzer.cpp +++ b/av_transport/av_trans_control_center/test/fuzztest/avtransonsessionclosed_fuzzer/avtransonsessionclosed_fuzzer.cpp @@ -1,5 +1,5 @@ /* - * Copyright (c) 2024 Huawei Device Co., Ltd. + * Copyright (c) 2024-2025 Huawei Device Co., Ltd. * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. * You may obtain a copy of the License at @@ -16,6 +16,7 @@ #include "avtransonsessionclosed_fuzzer.h" #include +#include #include #include #include @@ -29,8 +30,10 @@ void AVTransOnSessionClosedFuzzTest(const uint8_t *data, size_t size) if ((data == nullptr) || (size < sizeof(int32_t))) { return; } - int32_t sessionId = *(reinterpret_cast(data)); - ShutdownReason reason = *(reinterpret_cast(data)); + FuzzedDataProvider fdp(data, size); + int32_t sessionId = fdp.ConsumeIntegral(); + int32_t reasonValue = fdp.ConsumeIntegral(); + ShutdownReason reason = static_cast(reasonValue); SoftbusChannelAdapter::GetInstance().OnSoftbusChannelClosed(sessionId, reason); } diff --git a/services/distributedhardwarefwkservice/test/fuzztest/componentmanager_fuzzer/componentmanager_fuzzer.cpp b/services/distributedhardwarefwkservice/test/fuzztest/componentmanager_fuzzer/componentmanager_fuzzer.cpp index 99bcb6a8630a98ddcf318c5863c36065ece22f29..515a8a464a9c0160003a671ff5ab0380c8518b09 100644 --- a/services/distributedhardwarefwkservice/test/fuzztest/componentmanager_fuzzer/componentmanager_fuzzer.cpp +++ b/services/distributedhardwarefwkservice/test/fuzztest/componentmanager_fuzzer/componentmanager_fuzzer.cpp @@ -17,6 +17,7 @@ #include #include +#include #include "component_disable.h" #include "component_enable.h" @@ -51,8 +52,9 @@ void ComponentManagerFuzzTest(const uint8_t* data, size_t size) bool enableSource = false; sptr sinkListener = nullptr; sptr sourceListener = nullptr; - int32_t callingUid = *(reinterpret_cast(data)); - int32_t callingPid = *(reinterpret_cast(data)); + FuzzedDataProvider fdp(data, size); + int32_t callingUid = fdp.ConsumeIntegral(); + int32_t callingPid = fdp.ConsumeIntegral(); DHDescriptor dhDescriptor { .id = std::string(reinterpret_cast(data), size), .dhType = dhType