From c2e3a5b9e86c7be8a507149c63349dc44df3618d Mon Sep 17 00:00:00 2001
From: hwzhangchuang <zhangchuang.zhang@huawei.com>
Date: Fri, 22 Dec 2023 18:27:45 +0800
Subject: [PATCH] add

Signed-off-by: hwzhangchuang <zhangchuang.zhang@huawei.com>
---
 .../dinputonsessionclosed_fuzzer.cpp                         | 5 ++++-
 .../dinputonsessionopend_fuzzer.cpp                          | 2 +-
 2 files changed, 5 insertions(+), 2 deletions(-)

diff --git a/test/fuzztest/dinputonsessionclosed_fuzzer/dinputonsessionclosed_fuzzer.cpp b/test/fuzztest/dinputonsessionclosed_fuzzer/dinputonsessionclosed_fuzzer.cpp
index 3ddaec7..9470ea4 100644
--- a/test/fuzztest/dinputonsessionclosed_fuzzer/dinputonsessionclosed_fuzzer.cpp
+++ b/test/fuzztest/dinputonsessionclosed_fuzzer/dinputonsessionclosed_fuzzer.cpp
@@ -23,6 +23,8 @@
 #include <unistd.h>
 #include <refbase.h>
 
+#include "socket.h"
+
 #include "distributed_input_transport_base.h"
 
 namespace OHOS {
@@ -33,7 +35,8 @@ void OnSessionClosedFuzzTest(const uint8_t *data, size_t size)
         return;
     }
     int32_t sessionId = *(reinterpret_cast<const int32_t*>(data));
-    DistributedInput::DistributedInputTransportBase::GetInstance().OnSessionClosed(sessionId);
+    ShutdownReason reason = SHUTDOWN_REASON_UNKNOWN;
+    DistributedInput::DistributedInputTransportBase::GetInstance().OnSessionClosed(sessionId, reason);
 }
 } // namespace DistributedHardware
 } // namespace OHOS
diff --git a/test/fuzztest/dinputonsessionopend_fuzzer/dinputonsessionopend_fuzzer.cpp b/test/fuzztest/dinputonsessionopend_fuzzer/dinputonsessionopend_fuzzer.cpp
index c5afee1..ff33f9e 100644
--- a/test/fuzztest/dinputonsessionopend_fuzzer/dinputonsessionopend_fuzzer.cpp
+++ b/test/fuzztest/dinputonsessionopend_fuzzer/dinputonsessionopend_fuzzer.cpp
@@ -23,7 +23,7 @@
 #include <unistd.h>
 #include <refbase.h>
 
-#include "dinput_softbus_define.h"
+#include "socket.h"
 
 #include "distributed_input_transport_base.h"