From 815ca30eff0fff632f9329cefa963788933a97ee Mon Sep 17 00:00:00 2001
From: chen0088 <chenmenghuan6@huawei.com>
Date: Wed, 18 Dec 2024 14:24:54 +0800
Subject: [PATCH] modify fuzz

Signed-off-by: chen0088 <chenmenghuan6@huawei.com>
---
 .../encodeonoutputbufferavailable_fuzzer.cpp       | 14 ++++++++------
 .../decodeonoutputbufferavailable_fuzzer.cpp       | 12 +++++++-----
 2 files changed, 15 insertions(+), 11 deletions(-)

diff --git a/interfaces/inner_kits/native_cpp/test/sinkfuzztest/encodeonoutputbufferavailable_fuzzer/encodeonoutputbufferavailable_fuzzer.cpp b/interfaces/inner_kits/native_cpp/test/sinkfuzztest/encodeonoutputbufferavailable_fuzzer/encodeonoutputbufferavailable_fuzzer.cpp
index 0471170c..b8bd518a 100644
--- a/interfaces/inner_kits/native_cpp/test/sinkfuzztest/encodeonoutputbufferavailable_fuzzer/encodeonoutputbufferavailable_fuzzer.cpp
+++ b/interfaces/inner_kits/native_cpp/test/sinkfuzztest/encodeonoutputbufferavailable_fuzzer/encodeonoutputbufferavailable_fuzzer.cpp
@@ -1,5 +1,5 @@
 /*
- * Copyright (c) 2023 Huawei Device Co., Ltd.
+ * Copyright (c) 2023-2024 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
@@ -15,6 +15,7 @@
 
 #include "encodeonoutputbufferavailable_fuzzer.h"
 
+#include <fuzzer/FuzzedDataProvider.h>
 #include "encode_video_callback.h"
 
 namespace OHOS {
@@ -25,11 +26,12 @@ void EncodeOnOutputBufferAvailableFuzzTest(const uint8_t* data, size_t size)
         return;
     }
 
-    uint32_t index = *(reinterpret_cast<const uint32_t*>(data));
-    int64_t presentTimeUs = *(reinterpret_cast<const int64_t*>(data));
-    int32_t infoSize = *(reinterpret_cast<const int32_t*>(data));
-    int32_t offset = *(reinterpret_cast<const int32_t*>(data));
-    int32_t bufferFlag = *(reinterpret_cast<const int32_t*>(data));
+    FuzzedDataProvider fdp(data, size);
+    uint32_t index = fdp.ConsumeIntegral<uint32_t>();
+    int64_t presentTimeUs = fdp.ConsumeIntegral<int64_t>();
+    int32_t infoSize = fdp.ConsumeIntegral<int32_t>();
+    int32_t offset = fdp.ConsumeIntegral<int32_t>();
+    int32_t bufferFlag = fdp.ConsumeIntegral<int32_t>();
     MediaAVCodec::AVCodecBufferInfo info = { presentTimeUs, infoSize, offset};
     MediaAVCodec::AVCodecBufferFlag flag = static_cast<MediaAVCodec::AVCodecBufferFlag>(bufferFlag);
     std::shared_ptr<DCameraPipelineSink> sinkPipeline = std::make_shared<DCameraPipelineSink>();
diff --git a/interfaces/inner_kits/native_cpp/test/sourcefuzztest/decodeonoutputbufferavailable_fuzzer/decodeonoutputbufferavailable_fuzzer.cpp b/interfaces/inner_kits/native_cpp/test/sourcefuzztest/decodeonoutputbufferavailable_fuzzer/decodeonoutputbufferavailable_fuzzer.cpp
index aed0eff5..676b71cd 100644
--- a/interfaces/inner_kits/native_cpp/test/sourcefuzztest/decodeonoutputbufferavailable_fuzzer/decodeonoutputbufferavailable_fuzzer.cpp
+++ b/interfaces/inner_kits/native_cpp/test/sourcefuzztest/decodeonoutputbufferavailable_fuzzer/decodeonoutputbufferavailable_fuzzer.cpp
@@ -15,6 +15,7 @@
 
 #include "decodeonoutputbufferavailable_fuzzer.h"
 
+#include <fuzzer/FuzzedDataProvider.h>
 #include "decode_video_callback.h"
 
 namespace OHOS {
@@ -25,11 +26,12 @@ void DecodeOnOutputBufferAvailableFuzzTest(const uint8_t* data, size_t size)
         return;
     }
 
-    uint32_t index = *(reinterpret_cast<const uint32_t*>(data));
-    int64_t presentTimeUs = *(reinterpret_cast<const int64_t*>(data));
-    int32_t infoSize = *(reinterpret_cast<const int32_t*>(data));
-    int32_t offset = *(reinterpret_cast<const int32_t*>(data));
-    int32_t bufferFlag = *(reinterpret_cast<const int32_t*>(data));
+    FuzzedDataProvider fdp(data, size);
+    uint32_t index = fdp.ConsumeIntegral<uint32_t>();
+    int64_t presentTimeUs = fdp.ConsumeIntegral<int64_t>();
+    int32_t infoSize = fdp.ConsumeIntegral<int32_t>();
+    int32_t offset = fdp.ConsumeIntegral<int32_t>();
+    int32_t bufferFlag = fdp.ConsumeIntegral<int32_t>();
     MediaAVCodec::AVCodecBufferInfo info = { presentTimeUs, infoSize, offset};
     MediaAVCodec::AVCodecBufferFlag flag = static_cast<MediaAVCodec::AVCodecBufferFlag>(bufferFlag);
     std::shared_ptr<DCameraPipelineSource> sourcePipeline = std::make_shared<DCameraPipelineSource>();
-- 
Gitee