From 73c328542ba6b6ca57cf293edfc57aae5ed66ae9 Mon Sep 17 00:00:00 2001 From: dengliang <15934868816@139.com> Date: Fri, 12 Sep 2025 12:00:57 +0800 Subject: [PATCH 1/3] binder freeze process Signed-off-by: dengliang <15934868816@139.com> --- bundle.json | 3 +- config.gni | 3 +- .../innerkits/ipc_core/include/ipc_skeleton.h | 26 ++++++++ .../innerkits/ipc_core/include/ipc_types.h | 1 + .../ipc_core/include/message_option.h | 1 + interfaces/innerkits/ipc_single/BUILD.gn | 6 ++ .../include/dbinder_base_invoker_define.h | 4 ++ .../include/dbinder_base_invoker_interface.h | 14 +++++ .../src/core/framework/include/sys_binder.h | 24 +++++++- .../core/framework/source/ipc_skeleton.cpp | 20 +++++++ .../src/core/invoker/include/binder_invoker.h | 10 ++++ .../core/invoker/include/iremote_invoker.h | 6 ++ .../core/invoker/source/binder_invoker.cpp | 57 ++++++++++++++++++ ipc/native/test/unittest/common/BUILD.gn | 4 ++ .../common/binder_invoker_unittest.cpp | 54 +++++++++++++++++ .../unittest/common/ipc_skeleton_unittest.cpp | 48 +++++++++++++++ .../unittest/common/mock_iremote_invoker.h | 4 ++ ipc/test/auxiliary/native/BUILD.gn | 13 ++++ .../auxiliary/native/include/test_service.h | 3 + .../native/include/test_service_base.h | 4 ++ .../native/include/test_service_client.h | 3 + .../native/include/test_service_proxy.h | 3 + .../native/include/test_service_stub.h | 3 + ipc/test/auxiliary/native/src/main_client.cpp | 12 ++++ .../auxiliary/native/src/test_service.cpp | 7 +++ .../native/src/test_service_client.cpp | 17 ++++++ .../native/src/test_service_proxy.cpp | 60 +++++++++++++++++++ .../native/src/test_service_stub.cpp | 11 ++++ .../binderinvokernew004_fuzzer.cpp | 1 - .../ipc/native/src/core/source/BUILD.gn | 4 ++ test/unittest/mock/mock_iremote_invoker.h | 4 ++ 31 files changed, 426 insertions(+), 4 deletions(-) diff --git a/bundle.json b/bundle.json index 9c9b4e07..33457e07 100644 --- a/bundle.json +++ b/bundle.json @@ -23,7 +23,8 @@ "features": [ "ipc_feature_rpc_enabled", "ipc_feature_test_enabled", - "ipc_feature_trace_enabled" + "ipc_feature_trace_enabled", + "ipc_feature_freeze_enabled" ], "syscap":[ "SystemCapability.Communication.IPC.Core" diff --git a/config.gni b/config.gni index bf6ea89d..d154289a 100644 --- a/config.gni +++ b/config.gni @@ -23,6 +23,7 @@ declare_args() { hiviewdfx_backtrace_support = false hiviewdfx_hisysevent_support = false ipc_proxy_dfx_backtrace_enabled = false + ipc_feature_freeze_enabled = false } if (defined(global_parts_info) && @@ -43,4 +44,4 @@ if (defined(global_parts_info) && if (defined(global_parts_info) && defined(global_parts_info.hiviewdfx_hisysevent)) { hiviewdfx_hisysevent_support = true -} +} \ No newline at end of file diff --git a/interfaces/innerkits/ipc_core/include/ipc_skeleton.h b/interfaces/innerkits/ipc_core/include/ipc_skeleton.h index a19a0e79..3acdc754 100644 --- a/interfaces/innerkits/ipc_core/include/ipc_skeleton.h +++ b/interfaces/innerkits/ipc_core/include/ipc_skeleton.h @@ -207,6 +207,32 @@ public: * @since 20 */ static uint32_t GetDCallingTokenID(); + +#ifdef FREEZE_PROCESS_ENABLED + /** + * @brief Freeze binder transaction of the specific process. + * @note This function can only be used for local device processes. + * @param pid Process ID to freeze or unfreeze. + * @param freeze True to freeze the process, false to unfreeze it. + * @param timeout Timeout value in milliseconds to wait for the stub to finish processing requests before freezing. + * If the timeout is reached and processing is not completed, the freeze operation will be abandoned. + * @return Returns {@link ERR_NONE} if the operation is successful; returns an error code + * defined in {@link ipc_types.h} otherwise. + * @since 20 + */ + static int32_t Freeze(uint32_t pid, bool freeze, uint32_t timeout); + + /** + * @brief Get binder transaction freeze status of the specific process. + * @param pid Process ID to check. + * @param isFrozen Reference to a boolean variable that will store the freeze status of the process. + * True if the process is frozen, false otherwise. + * @return Returns {@link ERR_NONE} if the operation is successful; returns an error code + * defined in {@link ipc_types.h} otherwise. + * @since 20 + */ + static int32_t GetProcessFreezeInfo(uint32_t pid, bool &isFrozen); +#endif // FREEZE_PROCESS_ENABLED }; class IPCDfx { diff --git a/interfaces/innerkits/ipc_core/include/ipc_types.h b/interfaces/innerkits/ipc_core/include/ipc_types.h index 540aac3b..facfceba 100644 --- a/interfaces/innerkits/ipc_core/include/ipc_types.h +++ b/interfaces/innerkits/ipc_core/include/ipc_types.h @@ -138,6 +138,7 @@ enum { RPC_BASE_INVOKER_TRANSLATE_ERR, RPC_BASE_INVOKER_STUBINDEX_ERR, RPC_BASE_INVOKER_MALLOC_ERR, + RPC_BASE_INVOKER_NOT_SUPPORT_ERR, RPC_DATABUS_INVOKER_ERR = 600, RPC_DATABUS_INVOKER_INVALID_DATA_ERR, RPC_DATABUS_INVOKER_CLOSED_PEER_ERR, diff --git a/interfaces/innerkits/ipc_core/include/message_option.h b/interfaces/innerkits/ipc_core/include/message_option.h index bd9f222f..0fe51ff6 100644 --- a/interfaces/innerkits/ipc_core/include/message_option.h +++ b/interfaces/innerkits/ipc_core/include/message_option.h @@ -26,6 +26,7 @@ public: TF_STATUS_CODE = 0x08, TF_ACCEPT_FDS = 0x10, TF_WAIT_TIME = 0x8, + TF_UPDATE_TXN = 0x40, TF_ASYNC_WAKEUP_LATER = 0x100000, }; MessageOption(int flags = TF_SYNC, int waitTime = TF_WAIT_TIME); diff --git a/interfaces/innerkits/ipc_single/BUILD.gn b/interfaces/innerkits/ipc_single/BUILD.gn index 497d8b40..b81c1be8 100644 --- a/interfaces/innerkits/ipc_single/BUILD.gn +++ b/interfaces/innerkits/ipc_single/BUILD.gn @@ -122,6 +122,9 @@ ohos_shared_library("ipc_single") { if (hiviewdfx_hisysevent_support) { defines += [ "HIVIEWDFX_HISYSEVENT_SUPPORT" ] } + if (ipc_feature_freeze_enabled) { + defines += [ "FREEZE_PROCESS_ENABLED" ] + } innerapi_tags = [ "chipsetsdk", "platformsdk", @@ -164,6 +167,9 @@ ohos_shared_library("ipc_single_test") { if (hiviewdfx_hisysevent_support) { defines += [ "HIVIEWDFX_HISYSEVENT_SUPPORT" ] } + if (ipc_feature_freeze_enabled) { + defines += [ "FREEZE_PROCESS_ENABLED" ] + } innerapi_tags = [ "chipsetsdk_sp", "platformsdk", diff --git a/ipc/native/src/core/dbinder/include/dbinder_base_invoker_define.h b/ipc/native/src/core/dbinder/include/dbinder_base_invoker_define.h index 157db0e7..609f69cf 100644 --- a/ipc/native/src/core/dbinder/include/dbinder_base_invoker_define.h +++ b/ipc/native/src/core/dbinder/include/dbinder_base_invoker_define.h @@ -77,6 +77,10 @@ public: virtual sptr GetSAMgrObject() override; virtual bool SetRegistryObject(sptr &object) override; virtual void FreeBuffer(void *data) override; +#ifdef FREEZE_PROCESS_ENABLED + virtual int32_t Freeze(uint32_t pid, bool freeze, uint32_t timeout) override; + virtual int32_t GetProcessFreezeInfo(uint32_t pid, bool &isFrozen) override; +#endif // FREEZE_PROCESS_ENABLED virtual std::shared_ptr WriteTransaction(int cmd, uint32_t flags, int32_t handle, int32_t socketId, uint32_t code, MessageParcel &data, uint64_t &seqNumber, int status); virtual int SendOrWaitForCompletion(int userWaitTime, uint64_t seqNumber, std::shared_ptr sessionOfPeer, diff --git a/ipc/native/src/core/dbinder/include/dbinder_base_invoker_interface.h b/ipc/native/src/core/dbinder/include/dbinder_base_invoker_interface.h index 7561c06a..59e9aecc 100644 --- a/ipc/native/src/core/dbinder/include/dbinder_base_invoker_interface.h +++ b/ipc/native/src/core/dbinder/include/dbinder_base_invoker_interface.h @@ -188,6 +188,20 @@ template void DBinderBaseInvoker::FreeBuffer(void *data) return; } +#ifdef FREEZE_PROCESS_ENABLED +template +int32_t DBinderBaseInvoker::Freeze(uint32_t pid, bool freeze, uint32_t timeout) +{ + return RPC_BASE_INVOKER_NOT_SUPPORT_ERR; +} + +template +int32_t DBinderBaseInvoker::GetProcessFreezeInfo(uint32_t pid, bool &isFreeze) +{ + return RPC_BASE_INVOKER_NOT_SUPPORT_ERR; +} +#endif // FREEZE_PROCESS_ENABLED + template bool DBinderBaseInvoker::CheckTransactionData(const dbinder_transaction_data *tr) const { if (tr->sizeOfSelf == 0 || tr->sizeOfSelf > SOCKET_MAX_BUFF_SIZE || tr->buffer_size > SOCKET_MAX_BUFF_SIZE || diff --git a/ipc/native/src/core/framework/include/sys_binder.h b/ipc/native/src/core/framework/include/sys_binder.h index 206e09e6..de18e1b7 100644 --- a/ipc/native/src/core/framework/include/sys_binder.h +++ b/ipc/native/src/core/framework/include/sys_binder.h @@ -62,6 +62,21 @@ struct flat_binder_object { }; binder_uintptr_t cookie; }; + +#ifdef FREEZE_PROCESS_ENABLED +struct binder_freeze_info { + __u32 pid; + __u32 enable; + __u32 timeout_ms; +}; + +struct binder_proc_frozen_state { + __u32 pid; + // false: process is fully frozen, true: process is not fully frozen + bool has_not_frozen; +}; +#endif // FREEZE_PROCESS_ENABLED + #ifndef __linux__ #define HMB_ERROR_INFO_LEN 64 struct hmb_detailed_err { @@ -160,6 +175,10 @@ struct binder_sender_info { #define BINDER_GET_NODE_DEBUG_INFO _IOWR('b', 11, struct binder_node_debug_info) #define BINDER_GET_NODE_INFO_FOR_REF _IOWR('b', 12, struct binder_node_info_for_ref) #define BINDER_SET_CONTEXT_MGR_EXT _IOW('b', 13, struct flat_binder_object) +#ifdef FREEZE_PROCESS_ENABLED +#define BINDER_FREEZE _IOW('b', 14, struct binder_freeze_info) +#define BINDER_GET_FROZEN_STATE _IOW('b', 15, struct binder_proc_frozen_state) +#endif // FREEZE_PROCESS_ENABLED #define BINDER_FEATURE_SET _IOWR('b', 30, struct binder_feature_set) #define BINDER_GET_ACCESS_TOKEN _IOWR('b', 31, struct access_token) #define BINDER_GET_SENDER_INFO _IOWR('b', 32, struct binder_sender_info) @@ -241,7 +260,10 @@ enum binder_driver_return_protocol { BR_DEAD_BINDER = _IOR('r', 15, binder_uintptr_t), BR_CLEAR_DEATH_NOTIFICATION_DONE = _IOR('r', 16, binder_uintptr_t), BR_FAILED_REPLY = _IO('r', 17), - BR_RELEASE_NODE = _IO('r', 18), +#ifdef FREEZE_PROCESS_ENABLED + BR_FROZEN_REPLY = _IO('r', 18), + BR_TRANSACTION_PENDING_FROZEN = _IO('r', 20), +#endif // FREEZE_PROCESS_ENABLED }; enum binder_driver_command_protocol { BC_TRANSACTION = _IOW('c', 0, struct binder_transaction_data), diff --git a/ipc/native/src/core/framework/source/ipc_skeleton.cpp b/ipc/native/src/core/framework/source/ipc_skeleton.cpp index 7948b69c..f2ccbed2 100644 --- a/ipc/native/src/core/framework/source/ipc_skeleton.cpp +++ b/ipc/native/src/core/framework/source/ipc_skeleton.cpp @@ -330,6 +330,26 @@ uint32_t IPCSkeleton::GetDCallingTokenID() return static_cast(GetSelfTokenID()); } +#ifdef FREEZE_PROCESS_ENABLED +int32_t IPCSkeleton::Freeze(uint32_t pid, bool freeze, uint32_t timeout) +{ + IRemoteInvoker *invoker = IPCThreadSkeleton::GetDefaultInvoker(); + if (invoker == nullptr) { + return IPC_SKELETON_NULL_OBJECT_ERR; + } + return invoker->Freeze(pid, freeze, timeout); +} + +int32_t IPCSkeleton::GetProcessFreezeInfo(uint32_t pid, bool &isFrozen) +{ + IRemoteInvoker *invoker = IPCThreadSkeleton::GetDefaultInvoker(); + if (invoker == nullptr) { + return IPC_SKELETON_NULL_OBJECT_ERR; + } + return invoker->GetProcessFreezeInfo(pid, isFrozen); +} +#endif // FREEZE_PROCESS_ENABLED + void IPCDfx::BlockUntilThreadAvailable() { IPCProcessSkeleton *current = IPCProcessSkeleton::GetCurrent(); diff --git a/ipc/native/src/core/invoker/include/binder_invoker.h b/ipc/native/src/core/invoker/include/binder_invoker.h index d8743e2b..c4ad9cfb 100644 --- a/ipc/native/src/core/invoker/include/binder_invoker.h +++ b/ipc/native/src/core/invoker/include/binder_invoker.h @@ -90,6 +90,12 @@ public: int SendRequest(int handle, uint32_t code, MessageParcel &data, MessageParcel &reply, MessageOption &option) override; +#ifdef FREEZE_PROCESS_ENABLED + int32_t Freeze(uint32_t pid, bool freeze, uint32_t timeout) override; + + int32_t GetProcessFreezeInfo(uint32_t pid, bool &isFrozen) override; +#endif // FREEZE_PROCESS_ENABLED + int SendReply(MessageParcel &reply, uint32_t flags, int32_t result) override; bool FlattenObject(Parcel &parcel, const IRemoteObject *object) const override; @@ -200,6 +206,10 @@ private: void OnTransactionComplete(MessageParcel *reply, bool &continueLoop, int32_t &error, uint32_t cmd); +#ifdef FREEZE_PROCESS_ENABLED + void OnTransactionPendingFrozen(MessageParcel *reply, bool &continueLoop, int32_t &error, uint32_t cmd); +#endif // FREEZE_PROCESS_ENABLED + void OnDeadOrFailedReply(MessageParcel *reply, bool &continueLoop, int32_t &error, uint32_t cmd); void OnReply(MessageParcel *reply, bool &continueLoop, int32_t &error, uint32_t cmd); diff --git a/ipc/native/src/core/invoker/include/iremote_invoker.h b/ipc/native/src/core/invoker/include/iremote_invoker.h index 9feb8517..3ceae6a8 100644 --- a/ipc/native/src/core/invoker/include/iremote_invoker.h +++ b/ipc/native/src/core/invoker/include/iremote_invoker.h @@ -51,6 +51,12 @@ public: virtual int SendRequest(int handle, uint32_t code, MessageParcel &data, MessageParcel &reply, MessageOption &option) = 0; +#ifdef FREEZE_PROCESS_ENABLED + virtual int32_t Freeze(uint32_t pid, bool freeze, uint32_t timeout) = 0; + + virtual int32_t GetProcessFreezeInfo(uint32_t pid, bool &isFrozen) = 0; +#endif // FREEZE_PROCESS_ENABLED + virtual bool AddDeathRecipient(int32_t handle, void *cookie) = 0; virtual bool RemoveDeathRecipient(int32_t handle, void *cookie) = 0; diff --git a/ipc/native/src/core/invoker/source/binder_invoker.cpp b/ipc/native/src/core/invoker/source/binder_invoker.cpp index c4a0a214..6a5a818c 100644 --- a/ipc/native/src/core/invoker/source/binder_invoker.cpp +++ b/ipc/native/src/core/invoker/source/binder_invoker.cpp @@ -743,6 +743,42 @@ int32_t BinderInvoker::SamgrServiceSendRequest( return error; } +#ifdef FREEZE_PROCESS_ENABLED +int32_t BinderInvoker::Freeze(uint32_t pid, bool freeze, uint32_t timeout) +{ + if ((binderConnector_ == nullptr) || (!binderConnector_->IsDriverAlive())) { + return IPC_INVOKER_CONNECT_ERR; + } + binder_freeze_info info; + info.pid = pid; + info.enable = freeze; + info.timeout_ms = timeout; + int error = binderConnector_->WriteBinder(BINDER_FREEZE, &info); + if (error != ERR_NONE) { + ZLOGE(LABEL, "failed, error:%{public}d", error); + return error; + } + return ERR_NONE; +} + +int32_t BinderInvoker::GetProcessFreezeInfo(uint32_t pid, bool &isFrozen) +{ + if ((binderConnector_ == nullptr) || (!binderConnector_->IsDriverAlive())) { + return IPC_INVOKER_CONNECT_ERR; + } + binder_proc_frozen_state info; + info.pid = pid; + info.has_not_frozen = false; + int error = binderConnector_->WriteBinder(BINDER_GET_FROZEN_STATE, &info); + if (error != ERR_NONE) { + ZLOGE(LABEL, "error:%{public}d", error); + return error; + } + isFrozen = !info.has_not_frozen; + return ERR_NONE; +} +#endif // FREEZE_PROCESS_ENABLED + int32_t BinderInvoker::GeneralServiceSendRequest( const binder_transaction_data &tr, MessageParcel &data, MessageParcel &reply, MessageOption &option) { @@ -1190,6 +1226,19 @@ void BinderInvoker::OnTransactionComplete(MessageParcel *reply, bool &continueLo continueLoop = false; } } + +#ifdef FREEZE_PROCESS_ENABLED +void BinderInvoker::OnTransactionPendingFrozen(MessageParcel *reply, bool &continueLoop, int32_t &error, uint32_t cmd) +{ + (void)error; + (void)cmd; + + if (reply == nullptr) { + continueLoop = false; + } +} +#endif // FREEZE_PROCESS_ENABLED + #ifndef __linux__ bool BinderInvoker::GetDetailedErrorInfo(uint32_t &errorCode, std::string &errDesc) { @@ -1238,8 +1287,16 @@ void BinderInvoker::DealWithCmd(MessageParcel *reply, bool &continueLoop, int32_ case BR_TRANSACTION_COMPLETE: OnTransactionComplete(reply, continueLoop, error, cmd); break; +#ifdef FREEZE_PROCESS_ENABLED + case BR_TRANSACTION_PENDING_FROZEN: + OnTransactionPendingFrozen(reply, continueLoop, error, cmd); + break; +#endif // FREEZE_PROCESS_ENABLED case BR_DEAD_REPLY: case BR_FAILED_REPLY: +#ifdef FREEZE_PROCESS_ENABLED + case BR_FROZEN_REPLY: +#endif // FREEZE_PROCESS_ENABLED OnDeadOrFailedReply(reply, continueLoop, error, cmd); break; case BR_REPLY: diff --git a/ipc/native/test/unittest/common/BUILD.gn b/ipc/native/test/unittest/common/BUILD.gn index f894a221..97ca7ff1 100644 --- a/ipc/native/test/unittest/common/BUILD.gn +++ b/ipc/native/test/unittest/common/BUILD.gn @@ -62,6 +62,10 @@ template("ipc_unittest") { "samgr:samgr_proxy", ] + if (ipc_feature_freeze_enabled) { + defines = ["FREEZE_PROCESS_ENABLED"] + } + if (hiviewdfx_hisysevent_support) { external_deps += [ "hisysevent:libhisysevent" ] } diff --git a/ipc/native/test/unittest/common/binder_invoker_unittest.cpp b/ipc/native/test/unittest/common/binder_invoker_unittest.cpp index f24bce52..f5782f3a 100644 --- a/ipc/native/test/unittest/common/binder_invoker_unittest.cpp +++ b/ipc/native/test/unittest/common/binder_invoker_unittest.cpp @@ -25,6 +25,7 @@ #include "ipc_process_skeleton.h" #include "sys_binder.h" +using namespace testing; using namespace testing::ext; using namespace OHOS; @@ -830,4 +831,57 @@ HWTEST_F(BinderInvokerUnitTest, GetInvocationStateTest001, TestSize.Level1) binderInvoker.isFirstInvoke_ = STATUS_FIRST_INVOKE; EXPECT_EQ(binderInvoker.GetInvocationState(), STATUS_FIRST_INVOKE); } + +#ifdef FREEZE_PROCESS_ENABLED +/** +* @tc.name: FreezeProcessTest +* @tc.desc: cover Freeze branch +* @tc.type: FUNC +*/ +HWTEST_F(BinderInvokerUnitTest, FreezeProcessTest, TestSize.Level1) +{ + BinderInvoker binderInvoker; + BinderConnector::instance_ = nullptr; + binderInvoker.binderConnector_ = BinderConnector::GetInstance(); + EXPECT_EQ(binderInvoker.Freeze(getpid(), false, 1), ERR_NONE); + + bool isFrozen = false; + EXPECT_EQ(binderInvoker.GetProcessFreezeInfo(1, isFrozen), ERR_NONE); +} + +/** +* @tc.name: FreezeProcessTest001 +* @tc.desc: cover Freeze branch +* @tc.type: FUNC +*/ +HWTEST_F(BinderInvokerUnitTest, FreezeProcessTest001, TestSize.Level1) +{ + BinderInvoker binderInvoker; + binderInvoker.binderConnector_ = nullptr; + EXPECT_EQ(binderInvoker.Freeze(1, true, 1), IPC_INVOKER_CONNECT_ERR); + + BinderConnector *binderConnector = BinderConnector::GetInstance(); + binderConnector->driverFD_ = -1; + binderInvoker.binderConnector_ = binderConnector; + EXPECT_EQ(binderInvoker.Freeze(1, true, 1), IPC_INVOKER_CONNECT_ERR); +} + +/** +* @tc.name: GetProcessFreezeInfoTest +* @tc.desc: cover GetProcessFreezeInfo branch +* @tc.type: FUNC +*/ +HWTEST_F(BinderInvokerUnitTest, GetProcessFreezeInfoTest, TestSize.Level1) +{ + bool isFrozen = false; + BinderInvoker binderInvoker; + binderInvoker.binderConnector_ = nullptr; + EXPECT_EQ(binderInvoker.GetProcessFreezeInfo(1, isFrozen), IPC_INVOKER_CONNECT_ERR); + + BinderConnector *binderConnector = BinderConnector::GetInstance(); + binderConnector->driverFD_ = -1; + binderInvoker.binderConnector_ = binderConnector; + EXPECT_EQ(binderInvoker.GetProcessFreezeInfo(1, isFrozen), IPC_INVOKER_CONNECT_ERR); +} +#endif // FREEZE_PROCESS_ENABLED } // namespace OHOS \ No newline at end of file diff --git a/ipc/native/test/unittest/common/ipc_skeleton_unittest.cpp b/ipc/native/test/unittest/common/ipc_skeleton_unittest.cpp index 71223a8f..d7c8a62f 100644 --- a/ipc/native/test/unittest/common/ipc_skeleton_unittest.cpp +++ b/ipc/native/test/unittest/common/ipc_skeleton_unittest.cpp @@ -35,6 +35,7 @@ #undef protected #undef private +using namespace testing; using namespace testing::ext; using namespace OHOS; @@ -906,4 +907,51 @@ HWTEST_F(IPCSkeletonTest, GetDCallingTokenIDInRPC, TestSize.Level1) ASSERT_EQ(tokenId, 0); } + +#ifdef FREEZE_PROCESS_ENABLED +/** + * @tc.name: FreezeProcessTest + * @tc.desc: Verify the Freeze function. + * @tc.type: FUNC + */ +HWTEST_F(IPCSkeletonTest, FreezeProcessTest, TestSize.Level1) +{ + IPCThreadSkeleton *current = IPCThreadSkeleton::GetCurrent(); + ASSERT_NE(current, nullptr); + MockIRemoteInvoker *binderInvoker = new (std::nothrow) MockIRemoteInvoker(); + ASSERT_NE(binderInvoker, nullptr); + + current->invokers_[IRemoteObject::IF_PROT_DEFAULT] = binderInvoker; + + EXPECT_CALL(*binderInvoker, Freeze(_, _, _)) + .WillRepeatedly(testing::Return(ERR_NONE)); + + ASSERT_EQ(IPCSkeleton::Freeze(0, true, 0), ERR_NONE); + current->invokers_[IRemoteObject::IF_PROT_DEFAULT] = nullptr; + delete binderInvoker; +} + +/** + * @tc.name: CheckFreezeTest + * @tc.desc: Verify the GetProcessFreezeInfo function. + * @tc.type: FUNC + */ +HWTEST_F(IPCSkeletonTest, CheckFreezeTest, TestSize.Level1) +{ + IPCThreadSkeleton *current = IPCThreadSkeleton::GetCurrent(); + ASSERT_NE(current, nullptr); + MockIRemoteInvoker *binderInvoker = new (std::nothrow) MockIRemoteInvoker(); + ASSERT_NE(binderInvoker, nullptr); + + current->invokers_[IRemoteObject::IF_PROT_DEFAULT] = binderInvoker; + + EXPECT_CALL(*binderInvoker, GetProcessFreezeInfo(_, _)) + .WillRepeatedly(testing::Return(ERR_NONE)); + + bool isFrozen = false; + ASSERT_EQ(IPCSkeleton::GetProcessFreezeInfo(0, isFrozen), ERR_NONE); + current->invokers_[IRemoteObject::IF_PROT_DEFAULT] = nullptr; + delete binderInvoker; +} +#endif // FREEZE_PROCESS_ENABLED } // namespace OHOS \ No newline at end of file diff --git a/ipc/native/test/unittest/common/mock_iremote_invoker.h b/ipc/native/test/unittest/common/mock_iremote_invoker.h index b53f3bb8..fd4007df 100644 --- a/ipc/native/test/unittest/common/mock_iremote_invoker.h +++ b/ipc/native/test/unittest/common/mock_iremote_invoker.h @@ -41,6 +41,10 @@ public: MOCK_METHOD3(SendReply, int(MessageParcel &reply, uint32_t flags, int32_t result)); MOCK_METHOD5(SendRequest, int(int handle, uint32_t code, MessageParcel &data, MessageParcel &reply, MessageOption &option)); +#ifdef FREEZE_PROCESS_ENABLED + MOCK_METHOD3(Freeze, int32_t(uint32_t pid, bool freeze, uint32_t timeout)); + MOCK_METHOD2(GetProcessFreezeInfo, int32_t(uint32_t pid, bool &isFrozen)); +#endif // FREEZE_PROCESS_ENABLED MOCK_METHOD2(AddDeathRecipient, bool(int32_t handle, void *cookie)); MOCK_METHOD2(RemoveDeathRecipient, bool(int32_t handle, void *cookie)); MOCK_METHOD1(SetMaxWorkThread, bool(int maxThreadNum)); diff --git a/ipc/test/auxiliary/native/BUILD.gn b/ipc/test/auxiliary/native/BUILD.gn index 8832a75c..356d8f40 100644 --- a/ipc/test/auxiliary/native/BUILD.gn +++ b/ipc/test/auxiliary/native/BUILD.gn @@ -11,6 +11,7 @@ # See the License for the specific language governing permissions and # limitations under the License. import("//build/ohos.gni") +import("//foundation/communication/ipc/config.gni") SUBSYSTEM_DIR = "//foundation/communication/ipc" IPC_CORE_ROOT = "$SUBSYSTEM_DIR/ipc/native" @@ -31,6 +32,10 @@ ohos_shared_library("ipc_test_helper") { "./src/test_service_stub.cpp", ] + if (ipc_feature_freeze_enabled) { + defines = [ "FREEZE_PROCESS_ENABLED" ] + } + configs = [ "$IPC_TEST_ROOT:ipc_test_config", "$SUBSYSTEM_DIR/config:ipc_util_config", @@ -90,6 +95,10 @@ ohos_executable("ipc_client_test") { "$SUBSYSTEM_DIR/interfaces/innerkits/c_api:ipc_capi", ] + if (ipc_feature_freeze_enabled) { + defines = [ "FREEZE_PROCESS_ENABLED" ] + } + external_deps = [ "access_token:libnativetoken_shared", "access_token:libtokensetproc_shared", @@ -125,6 +134,10 @@ ohos_shared_library("ipc_test_helper_extra") { "$SUBSYSTEM_DIR/config:ipc_util_config", ] + if (ipc_feature_freeze_enabled) { + defines = [ "FREEZE_PROCESS_ENABLED" ] + } + deps = [ "$SUBSYSTEM_DIR/interfaces/innerkits/c_api:ipc_capi", "$SUBSYSTEM_DIR/interfaces/innerkits/ipc_single:ipc_single_test", diff --git a/ipc/test/auxiliary/native/include/test_service.h b/ipc/test/auxiliary/native/include/test_service.h index c7ece976..2bf519d6 100644 --- a/ipc/test/auxiliary/native/include/test_service.h +++ b/ipc/test/auxiliary/native/include/test_service.h @@ -56,6 +56,9 @@ public: int TestSendTooManyRequest(int data, int &reply) override; int TestMultiThreadSendRequest(int data, int &reply) override; int TestQueryThreadInvocationState() override; +#ifdef FREEZE_PROCESS_ENABLED + int TestFreezeProcess() override; +#endif // FREEZE_PROCESS_ENABLED private: int testFd_; std::mutex remoteObjectsMutex_; diff --git a/ipc/test/auxiliary/native/include/test_service_base.h b/ipc/test/auxiliary/native/include/test_service_base.h index fd969b7c..591712f8 100644 --- a/ipc/test/auxiliary/native/include/test_service_base.h +++ b/ipc/test/auxiliary/native/include/test_service_base.h @@ -56,6 +56,7 @@ public: TRANS_ID_UNREGISTER_REMOTE_STUB_OBJECT = 24, TRANS_ID_QUERY_REMOTE_PROXY_OBJECT = 25, TRANS_ID_QUERY_THREAD_INVOCATION_STATE = 26, + TRANS_ID_TEST_FREEZE_PROCESS = 27, }; public: virtual int TestSyncTransaction(int value, int &reply, int delayTime = 0) = 0; @@ -87,6 +88,9 @@ public: virtual int TestSendTooManyRequest(int data, int &reply) = 0; virtual int TestMultiThreadSendRequest(int data, int &reply) = 0; virtual int TestQueryThreadInvocationState() = 0; +#ifdef FREEZE_PROCESS_ENABLED + virtual int TestFreezeProcess() = 0; +#endif // FREEZE_PROCESS_ENABLED public: DECLARE_INTERFACE_DESCRIPTOR(u"test.ipc.ITestService"); diff --git a/ipc/test/auxiliary/native/include/test_service_client.h b/ipc/test/auxiliary/native/include/test_service_client.h index 605e805d..be2625d3 100644 --- a/ipc/test/auxiliary/native/include/test_service_client.h +++ b/ipc/test/auxiliary/native/include/test_service_client.h @@ -45,6 +45,9 @@ public: bool TestUnRegisterRemoteStub(); bool TestSendTooManyRequest(); bool TestMultiThreadSendRequest(); +#ifdef FREEZE_PROCESS_ENABLED + bool TestFreezeProcess(); +#endif // FREEZE_PROCESS_ENABLED bool TestQueryThreadInvocationState(); private: diff --git a/ipc/test/auxiliary/native/include/test_service_proxy.h b/ipc/test/auxiliary/native/include/test_service_proxy.h index c6334114..f3975684 100644 --- a/ipc/test/auxiliary/native/include/test_service_proxy.h +++ b/ipc/test/auxiliary/native/include/test_service_proxy.h @@ -51,6 +51,9 @@ public: sptr TestQueryRemoteProxy(const char *descriptor) override; int TestSendTooManyRequest(int data, int &reply) override; int TestMultiThreadSendRequest(int data, int &reply) override; +#ifdef FREEZE_PROCESS_ENABLED + int TestFreezeProcess() override; +#endif // FREEZE_PROCESS_ENABLED int TestQueryThreadInvocationState() override; private: diff --git a/ipc/test/auxiliary/native/include/test_service_stub.h b/ipc/test/auxiliary/native/include/test_service_stub.h index b5190036..e926dcb5 100644 --- a/ipc/test/auxiliary/native/include/test_service_stub.h +++ b/ipc/test/auxiliary/native/include/test_service_stub.h @@ -50,6 +50,9 @@ private: int32_t QueryRemoteProxy(MessageParcel &data, MessageParcel &reply); int32_t ServerFlushAsyncCalls(MessageParcel &data, MessageParcel &reply); int32_t ServerThreadInvocationState(MessageParcel &data, MessageParcel &reply); +#ifdef FREEZE_PROCESS_ENABLED + int32_t ServerTestFreezeProcess(MessageParcel &data, MessageParcel &reply); +#endif // FREEZE_PROCESS_ENABLED void InitMessageProcessMap(); using TestServiceStubFunc = int32_t(TestServiceStub::*)(MessageParcel &data, MessageParcel &reply); diff --git a/ipc/test/auxiliary/native/src/main_client.cpp b/ipc/test/auxiliary/native/src/main_client.cpp index 6a486a57..17e0818f 100644 --- a/ipc/test/auxiliary/native/src/main_client.cpp +++ b/ipc/test/auxiliary/native/src/main_client.cpp @@ -82,6 +82,15 @@ void TestCaseSyncTrans(std::shared_ptr &testClient) } } +#ifdef FREEZE_PROCESS_ENABLED +void TestCaseFreezeProcess(std::shared_ptr &testClient) +{ + bool ret = testClient->TestFreezeProcess(); + std::cout << (ret ? "[PASS] Execution of TestFreezeProcess case Successful" : + "[FAILED] Execution of TestFreezeProcess case failed") << std::endl; +} +#endif // FREEZE_PROCESS_ENABLED + void TestCasePingService(std::shared_ptr &testClient) { bool ret = testClient->StartPingService(); @@ -229,6 +238,9 @@ void ExecuteAllTestCase() TestCaseRegisterRemoteStub(testClient); TestCaseTooManyRequests(testClient); TestCaseMultiThreadSendRequest(testClient); +#ifdef FREEZE_PROCESS_ENABLED + TestCaseFreezeProcess(testClient); +#endif // FREEZE_PROCESS_ENABLED ZLOGI(LABEL, "All test cases have been executed"); } diff --git a/ipc/test/auxiliary/native/src/test_service.cpp b/ipc/test/auxiliary/native/src/test_service.cpp index c8b6b110..c20fef86 100644 --- a/ipc/test/auxiliary/native/src/test_service.cpp +++ b/ipc/test/auxiliary/native/src/test_service.cpp @@ -345,4 +345,11 @@ int TestService::TestQueryThreadInvocationState() return state; } + +#ifdef FREEZE_PROCESS_ENABLED +int TestService::TestFreezeProcess() +{ + return 0; +} +#endif // FREEZE_PROCESS_ENABLED } // namespace OHOS diff --git a/ipc/test/auxiliary/native/src/test_service_client.cpp b/ipc/test/auxiliary/native/src/test_service_client.cpp index 154a7b0e..3b3481e6 100644 --- a/ipc/test/auxiliary/native/src/test_service_client.cpp +++ b/ipc/test/auxiliary/native/src/test_service_client.cpp @@ -74,6 +74,23 @@ bool TestServiceClient::StartSyncTransaction() return true; } +#ifdef FREEZE_PROCESS_ENABLED +bool TestServiceClient::TestFreezeProcess() +{ + if (testService_ == nullptr) { + ZLOGE(LABEL, "The testService_ object is an empty object"); + return false; + } + + int ret = testService_->TestFreezeProcess(); + if (ret != 0) { + ZLOGE(LABEL, "TestFreezeProcess function call failed"); + return false; + } + return true; +} +#endif // FREEZE_PROCESS_ENABLED + bool TestServiceClient::StartPingService() { if (testService_ == nullptr) { diff --git a/ipc/test/auxiliary/native/src/test_service_proxy.cpp b/ipc/test/auxiliary/native/src/test_service_proxy.cpp index 7a043511..ce0036b3 100644 --- a/ipc/test/auxiliary/native/src/test_service_proxy.cpp +++ b/ipc/test/auxiliary/native/src/test_service_proxy.cpp @@ -19,6 +19,8 @@ #include #include #include +#include +#include #include #include #include @@ -95,6 +97,64 @@ int TestServiceProxy::TestSyncTransaction(int value, int &reply, int delayTime) return ret; } +#ifdef FREEZE_PROCESS_ENABLED +int TestServiceProxy::TestFreezeProcess() +{ + ZLOGI(LABEL, "TestFreezeProcess"); + + auto remote = Remote(); + if (remote == nullptr) { + ZLOGI(LABEL, "The obtained proxy is a null pointer"); + return -1; + } + + MessageOption option; + MessageParcel dataParcel; + MessageParcel replyParcel; + int ret = remote->SendRequest(TRANS_ID_TEST_FREEZE_PROCESS, dataParcel, replyParcel, option); + if (ret != ERR_NONE) { + return ret; + } + int pid = replyParcel.ReadInt32(); + ZLOGI(LABEL, "Get pid from server, server pid = %{public}d", pid); + + int timeout = 1000; + auto res = IPCSkeleton::Freeze(pid, true, timeout); + if (res != ERR_NONE) { + ZLOGI(LABEL, "Freeze failed, error code = %{public}d", res); + return res; + } + + bool isFrozen = false; + res = IPCSkeleton::GetProcessFreezeInfo(pid, isFrozen); + if (res != ERR_NONE || !isFrozen) { + ZLOGI(LABEL, "GetProcessFreezeInfo failed, error code = %{public}d isFrozen = %{public}d ", res, isFrozen); + return -1; + } + + ret = remote->SendRequest(TRANS_ID_TEST_FREEZE_PROCESS, dataParcel, replyParcel, option); + if (ret == ERR_NONE) { + ZLOGI(LABEL, "SendRequest success, so freeze process fail"); + return -1; + } + + res = IPCSkeleton::Freeze(pid, false, timeout); + if (res != ERR_NONE) { + ZLOGI(LABEL, "Unfreeze failed, error code = %{public}d", res); + return -1; + } + + ret = remote->SendRequest(TRANS_ID_TEST_FREEZE_PROCESS, dataParcel, replyParcel, option); + if (ret != ERR_NONE) { + ZLOGI(LABEL, "SendRequest failed, error code = %{public}d", ret); + return -1; + } + + ZLOGI(LABEL, "TestFreezeProcess success"); + return 0; +} +#endif // FREEZE_PROCESS_ENABLED + int TestServiceProxy::TestSendTooManyRequest(int data, int &reply) { int ret = 0; diff --git a/ipc/test/auxiliary/native/src/test_service_stub.cpp b/ipc/test/auxiliary/native/src/test_service_stub.cpp index d9a082d6..d5575315 100644 --- a/ipc/test/auxiliary/native/src/test_service_stub.cpp +++ b/ipc/test/auxiliary/native/src/test_service_stub.cpp @@ -72,6 +72,9 @@ void TestServiceStub::InitMessageProcessMap() funcMap_[static_cast(TRANS_ID_UNREGISTER_REMOTE_STUB_OBJECT)] = &TestServiceStub::UnRegisterRemoteStub; funcMap_[static_cast(TRANS_ID_QUERY_REMOTE_PROXY_OBJECT)] = &TestServiceStub::QueryRemoteProxy; funcMap_[static_cast(TRANS_ID_QUERY_THREAD_INVOCATION_STATE)] = &TestServiceStub::ServerThreadInvocationState; +#ifdef FREEZE_PROCESS_ENABLED + funcMap_[static_cast(TRANS_ID_TEST_FREEZE_PROCESS)] = &TestServiceStub::ServerTestFreezeProcess; +#endif // FREEZE_PROCESS_ENABLED } int32_t TestServiceStub::ServerSyncTransaction(MessageParcel &data, MessageParcel &reply) @@ -483,4 +486,12 @@ int32_t TestServiceStub::ServerThreadInvocationState(MessageParcel &data, Messag reply.WriteInt32(ret); return ret; } + +#ifdef FREEZE_PROCESS_ENABLED +int32_t TestServiceStub::ServerTestFreezeProcess(MessageParcel &data, MessageParcel &reply) +{ + reply.WriteInt32(getpid()); + return 0; +} +#endif // FREEZE_PROCESS_ENABLED } // namespace OHOS diff --git a/test/fuzztest/ipc/native/src/core/binderinvokernew/binderinvokernew004_fuzzer/binderinvokernew004_fuzzer.cpp b/test/fuzztest/ipc/native/src/core/binderinvokernew/binderinvokernew004_fuzzer/binderinvokernew004_fuzzer.cpp index 3c886ed5..c3b8c88f 100644 --- a/test/fuzztest/ipc/native/src/core/binderinvokernew/binderinvokernew004_fuzzer/binderinvokernew004_fuzzer.cpp +++ b/test/fuzztest/ipc/native/src/core/binderinvokernew/binderinvokernew004_fuzzer/binderinvokernew004_fuzzer.cpp @@ -42,7 +42,6 @@ static const std::vector cmdList = { binder_driver_return_protocol::BR_DEAD_BINDER, binder_driver_return_protocol::BR_CLEAR_DEATH_NOTIFICATION_DONE, binder_driver_return_protocol::BR_FAILED_REPLY, - binder_driver_return_protocol::BR_RELEASE_NODE, }; void HandleReplyFuzzTest(FuzzedDataProvider &provider) diff --git a/test/unittest/ipc/native/src/core/source/BUILD.gn b/test/unittest/ipc/native/src/core/source/BUILD.gn index 8236f82c..b0a93d82 100644 --- a/test/unittest/ipc/native/src/core/source/BUILD.gn +++ b/test/unittest/ipc/native/src/core/source/BUILD.gn @@ -64,6 +64,10 @@ template("ipc_unittest") { "samgr:samgr_proxy", ] + if (ipc_feature_freeze_enabled) { + defines += [ "FREEZE_PROCESS_ENABLED" ] + } + if (hiviewdfx_hisysevent_support) { external_deps += [ "hisysevent:libhisysevent" ] } diff --git a/test/unittest/mock/mock_iremote_invoker.h b/test/unittest/mock/mock_iremote_invoker.h index b524b532..49ae4603 100644 --- a/test/unittest/mock/mock_iremote_invoker.h +++ b/test/unittest/mock/mock_iremote_invoker.h @@ -41,6 +41,10 @@ public: MOCK_METHOD3(SendReply, int(MessageParcel &reply, uint32_t flags, int32_t result)); MOCK_METHOD5(SendRequest, int(int handle, uint32_t code, MessageParcel &data, MessageParcel &reply, MessageOption &option)); +#ifdef FREEZE_PROCESS_ENABLED + MOCK_METHOD3(Freeze, int32_t(uint32_t pid, bool freeze, uint32_t timeout)); + MOCK_METHOD2(GetProcessFreezeInfo, int32_t(uint32_t pid, bool &isFrozen)); +#endif // FREEZE_PROCESS_ENABLED MOCK_METHOD2(AddDeathRecipient, bool(int32_t handle, void *cookie)); MOCK_METHOD2(RemoveDeathRecipient, bool(int32_t handle, void *cookie)); MOCK_METHOD1(SetMaxWorkThread, bool(int maxThreadNum)); -- Gitee From 59ace355df8d9886f21e38699ef45936fff1b8bf Mon Sep 17 00:00:00 2001 From: dengliang <15934868816@139.com> Date: Fri, 12 Sep 2025 14:15:22 +0800 Subject: [PATCH 2/3] binder freeze process Signed-off-by: dengliang <15934868816@139.com> --- interfaces/innerkits/ipc_core/include/ipc_types.h | 2 ++ interfaces/innerkits/ipc_core/include/message_option.h | 2 ++ 2 files changed, 4 insertions(+) diff --git a/interfaces/innerkits/ipc_core/include/ipc_types.h b/interfaces/innerkits/ipc_core/include/ipc_types.h index facfceba..2d91ae8a 100644 --- a/interfaces/innerkits/ipc_core/include/ipc_types.h +++ b/interfaces/innerkits/ipc_core/include/ipc_types.h @@ -136,7 +136,9 @@ enum { RPC_BASE_INVOKER_WRITE_TRANS_ERR, RPC_BASE_INVOKER_SEND_REPLY_ERR, RPC_BASE_INVOKER_TRANSLATE_ERR, +#ifdef FREEZE_PROCESS_ENABLED RPC_BASE_INVOKER_STUBINDEX_ERR, +#endif // FREEZE_PROCESS_ENABLED RPC_BASE_INVOKER_MALLOC_ERR, RPC_BASE_INVOKER_NOT_SUPPORT_ERR, RPC_DATABUS_INVOKER_ERR = 600, diff --git a/interfaces/innerkits/ipc_core/include/message_option.h b/interfaces/innerkits/ipc_core/include/message_option.h index 0fe51ff6..b595e917 100644 --- a/interfaces/innerkits/ipc_core/include/message_option.h +++ b/interfaces/innerkits/ipc_core/include/message_option.h @@ -26,7 +26,9 @@ public: TF_STATUS_CODE = 0x08, TF_ACCEPT_FDS = 0x10, TF_WAIT_TIME = 0x8, +#ifdef FREEZE_PROCESS_ENABLED TF_UPDATE_TXN = 0x40, +#endif // FREEZE_PROCESS_ENABLED TF_ASYNC_WAKEUP_LATER = 0x100000, }; MessageOption(int flags = TF_SYNC, int waitTime = TF_WAIT_TIME); -- Gitee From 5885a7d2ca449379b1fff2e41744f878f1124578 Mon Sep 17 00:00:00 2001 From: dengliang <15934868816@139.com> Date: Fri, 12 Sep 2025 14:56:56 +0800 Subject: [PATCH 3/3] binder freeze process Signed-off-by: dengliang <15934868816@139.com> --- interfaces/innerkits/ipc_core/include/ipc_types.h | 4 ++-- ipc/test/auxiliary/native/src/test_service_proxy.cpp | 2 -- 2 files changed, 2 insertions(+), 4 deletions(-) diff --git a/interfaces/innerkits/ipc_core/include/ipc_types.h b/interfaces/innerkits/ipc_core/include/ipc_types.h index 2d91ae8a..4bd6dcfe 100644 --- a/interfaces/innerkits/ipc_core/include/ipc_types.h +++ b/interfaces/innerkits/ipc_core/include/ipc_types.h @@ -136,11 +136,11 @@ enum { RPC_BASE_INVOKER_WRITE_TRANS_ERR, RPC_BASE_INVOKER_SEND_REPLY_ERR, RPC_BASE_INVOKER_TRANSLATE_ERR, -#ifdef FREEZE_PROCESS_ENABLED RPC_BASE_INVOKER_STUBINDEX_ERR, -#endif // FREEZE_PROCESS_ENABLED RPC_BASE_INVOKER_MALLOC_ERR, +#ifdef FREEZE_PROCESS_ENABLED RPC_BASE_INVOKER_NOT_SUPPORT_ERR, +#endif // FREEZE_PROCESS_ENABLED RPC_DATABUS_INVOKER_ERR = 600, RPC_DATABUS_INVOKER_INVALID_DATA_ERR, RPC_DATABUS_INVOKER_CLOSED_PEER_ERR, diff --git a/ipc/test/auxiliary/native/src/test_service_proxy.cpp b/ipc/test/auxiliary/native/src/test_service_proxy.cpp index ce0036b3..dedcec9a 100644 --- a/ipc/test/auxiliary/native/src/test_service_proxy.cpp +++ b/ipc/test/auxiliary/native/src/test_service_proxy.cpp @@ -19,8 +19,6 @@ #include #include #include -#include -#include #include #include #include -- Gitee