diff --git a/config.gni b/config.gni index bf6ea89d5e4db387c541611b8956b9a55862c6f6..5d3c8bbdfa4976a149e6cd6c356a10490e86a230 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_freeze_process_support = false } if (defined(global_parts_info) && @@ -44,3 +45,8 @@ if (defined(global_parts_info) && defined(global_parts_info.hiviewdfx_hisysevent)) { hiviewdfx_hisysevent_support = true } + +if (defined(global_parts_info) && + defined(global_parts_info.officeservice_space_mgr_service)) { + ipc_freeze_process_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 a19a0e7996b64a0a2e0274bec0a0192064a5f1bf..3cf4e1e7ec7ea7e09e3c3ebe5cd1168818bf8de0 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_SUPPORT + /** + * @brief Freeze or unfreeze a process. + * @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 FreezeProcess(uint32_t pid, bool freeze, uint32_t timeout); + + /** + * @brief Check if a process is frozen. + * @param pid Process ID to check. + * @param isFreeze 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 CheckFreeze(uint32_t pid, bool &isFreeze); +#endif // FREEZE_PROCESS_SUPPORT }; class IPCDfx { diff --git a/interfaces/innerkits/ipc_core/include/message_option.h b/interfaces/innerkits/ipc_core/include/message_option.h index bd9f222fe7c2bb19d9681e26212281b8676309c4..0fe51ff64d759ced062ecc0e19da0221b8a0434e 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 497d8b40c6f8e9cbe452d90d0bdca6f9317fe4bd..23d8913c9d7a84e4545af7066593f66e7efa957f 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_freeze_process_support) { + defines += [ "FREEZE_PROCESS_SUPPORT" ] + } innerapi_tags = [ "chipsetsdk", "platformsdk", @@ -164,6 +167,9 @@ ohos_shared_library("ipc_single_test") { if (hiviewdfx_hisysevent_support) { defines += [ "HIVIEWDFX_HISYSEVENT_SUPPORT" ] } + if (ipc_freeze_process_support) { + defines += [ "FREEZE_PROCESS_SUPPORT" ] + } 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 157db0e7728d928026372b0bfefa8af4368a7bc0..fe6add346e2a565b7f01e966777588ab0600a750 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_SUPPORT + virtual int32_t FreezeProcess(uint32_t pid, bool freeze, uint32_t timeout) override; + virtual int32_t CheckFreeze(uint32_t pid, bool &isFreeze) override; +#endif // FREEZE_PROCESS_SUPPORT 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 7561c06a6bfc6634e6c0ad54f704dc5290b40b59..e179c40223cb118a97e75f9c211aa9c551791861 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_SUPPORT +template +int32_t bool DBinderBaseInvoker::FreezeProcess(uint32_t pid, bool freeze, uint32_t timeout) +{ + return ERR_NONE; +} + +template +int32_t bool DBinderBaseInvoker::CheckFreeze(uint32_t pid, bool &isFreeze) +{ + return ERR_NONE; +} +#endif // FREEZE_PROCESS_SUPPORT + 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 206e09e698eabe0d64d4df1f0b7e0a7113148bbc..54dd232d8c34feed744b73e1af15d4f1d3bbedfb 100644 --- a/ipc/native/src/core/framework/include/sys_binder.h +++ b/ipc/native/src/core/framework/include/sys_binder.h @@ -62,6 +62,18 @@ struct flat_binder_object { }; binder_uintptr_t cookie; }; + +struct binder_freeze_info { + __u32 pid; + __u32 enable; + __u32 timeout_ms; +}; + +struct binder_frozen_state_info { + __u32 pid; + bool has_unfrozen; +}; + #ifndef __linux__ #define HMB_ERROR_INFO_LEN 64 struct hmb_detailed_err { @@ -160,6 +172,8 @@ 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) +#define BINDER_FREEZE _IOW('b', 14, struct binder_freeze_info) +#define BINDER_GET_FROZEN_STATE _IOW('b', 15, struct binder_frozen_state_info) #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 +255,8 @@ 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), + BR_FROZEN_REPLY = _IO('r', 18), + BR_TRANSACTION_PENDING_FROZEN = _IO('r', 20), }; 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 7948b69c5910baa26112e0ffd1941fd3db49bb38..94c93f851ef318812cd83656e342731f9d1f7302 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_SUPPORT +int32_t IPCSkeleton::FreezeProcess(uint32_t pid, bool freeze, uint32_t timeout) +{ + IRemoteInvoker *invoker = IPCThreadSkeleton::GetDefaultInvoker(); + if (invoker == nullptr) { + return IPC_SKELETON_NULL_OBJECT_ERR; + } + return invoker->FreezeProcess(pid, freeze, timeout); +} + +int32_t IPCSkeleton::CheckFreeze(uint32_t pid, bool &isFreeze) +{ + IRemoteInvoker *invoker = IPCThreadSkeleton::GetDefaultInvoker(); + if (invoker == nullptr) { + return IPC_SKELETON_NULL_OBJECT_ERR; + } + return invoker->CheckFreeze(pid, isFreeze); +} +#endif // FREEZE_PROCESS_SUPPORT + 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 d8743e2bb4b6c8550c036523d51b8f88cf361ebc..d2362351f5fa26aba17785cdf6bc6f67c3b67819 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_SUPPORT + int32_t FreezeProcess(uint32_t pid, bool freeze, uint32_t timeout) override; + + int32_t CheckFreeze(uint32_t pid, bool &isFreeze) override; +#endif // FREEZE_PROCESS_SUPPORT + int SendReply(MessageParcel &reply, uint32_t flags, int32_t result) override; bool FlattenObject(Parcel &parcel, const IRemoteObject *object) const override; diff --git a/ipc/native/src/core/invoker/include/iremote_invoker.h b/ipc/native/src/core/invoker/include/iremote_invoker.h index 9feb8517ab665335ed2be1e252b5c9dcbe1f5c24..3d5753f900c22cc8df39f59050499520bc65647f 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_SUPPORT + virtual int32_t FreezeProcess(uint32_t pid, bool freeze, uint32_t timeout) = 0; + + virtual int32_t CheckFreeze(uint32_t pid, bool &isFreeze) = 0; +#endif // FREEZE_PROCESS_SUPPORT + 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 df701131b3fa4e1632eeb1cf3be604f227f5c53e..94ab240c7ea47872963cbdbfb1bbf50a9de7298b 100644 --- a/ipc/native/src/core/invoker/source/binder_invoker.cpp +++ b/ipc/native/src/core/invoker/source/binder_invoker.cpp @@ -737,6 +737,42 @@ int32_t BinderInvoker::SamgrServiceSendRequest( return error; } +#ifdef FREEZE_PROCESS_SUPPORT +int32_t BinderInvoker::FreezeProcess(uint32_t pid, bool freeze, uint32_t timeout) +{ + if ((binderConnector_ == nullptr) || (!binderConnector_->IsDriverAlive())) { + return IPC_INVOKER_CONNECT_ERR; + } + binder_freeze_info info; + info.enable = pid; + info.enable = freeze; + info.timeout_ms = timeout; + int error = binderConnector_->WriteBinder(BINDER_FREEZE, &info); + if (error != ERR_NONE) { + ZLOGE(LABEL, "Freeze process failed, error:%{public}d", error); + return error; + } + return ERR_NONE; +} + +int32_t BinderInvoker::CheckFreeze(uint32_t pid, bool &isFreeze) +{ + if ((binderConnector_ == nullptr) || (!binderConnector_->IsDriverAlive())) { + return IPC_INVOKER_CONNECT_ERR; + } + binder_frozen_state_info info; + info.pid = pid; + info.has_unfrozen = false; + int error = binderConnector_->WriteBinder(BINDER_GET_FROZEN_STATE, &info); + if (error != ERR_NONE) { + ZLOGE(LABEL, "Get freeze state failed, error:%{public}d", error); + return error; + } + isFreeze = !info.has_unfrozen; + return ERR_NONE; +} +#endif // FREEZE_PROCESS_SUPPORT + int32_t BinderInvoker::GeneralServiceSendRequest( const binder_transaction_data &tr, MessageParcel &data, MessageParcel &reply, MessageOption &option) { @@ -1226,10 +1262,16 @@ void BinderInvoker::DealWithCmd(MessageParcel *reply, bool &continueLoop, int32_ { switch (cmd) { case BR_TRANSACTION_COMPLETE: +#ifdef FREEZE_PROCESS_SUPPORT + case BR_TRANSACTION_PENDING_FROZEN: +#endif // FREEZE_PROCESS_SUPPORT OnTransactionComplete(reply, continueLoop, error, cmd); break; case BR_DEAD_REPLY: case BR_FAILED_REPLY: +#ifdef FREEZE_PROCESS_SUPPORT + case BR_FROZEN_REPLY: +#endif // FREEZE_PROCESS_SUPPORT 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 f894a22145de6361a498749cd2c4ef60d184e470..699de142d20a6b600a8d124ad9fa2b31eb51436b 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_freeze_process_support) { + defines = ["FREEZE_PROCESS_SUPPORT"] + } + 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 f24bce52332dac687b4fbccd95b60a7d1f04ab9e..48113a26e7c6564443c3b237397dba92b478b8fd 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,41 @@ HWTEST_F(BinderInvokerUnitTest, GetInvocationStateTest001, TestSize.Level1) binderInvoker.isFirstInvoke_ = STATUS_FIRST_INVOKE; EXPECT_EQ(binderInvoker.GetInvocationState(), STATUS_FIRST_INVOKE); } + +#ifdef FREEZE_PROCESS_SUPPORT +/** +* @tc.name: FreezeProcessTest001 +* @tc.desc: cover FreezeProcess branch +* @tc.type: FUNC +*/ +HWTEST_F(BinderInvokerUnitTest, FreezeProcessTest, TestSize.Level1) +{ + BinderInvoker binderInvoker; + binderInvoker.binderConnector_ = nullptr; + EXPECT_EQ(binderInvoker.FreezeProcess(1, true, 1), IPC_INVOKER_CONNECT_ERR); + + BinderConnector *binderConnector = BinderConnector::GetInstance(); + binderConnector->driverFD_ = -1; + binderInvoker.binderConnector_ = binderConnector; + EXPECT_EQ(binderInvoker.FreezeProcess(1, true, 1), IPC_INVOKER_CONNECT_ERR); +} + +/** +* @tc.name: CheckFreezeTest +* @tc.desc: cover CheckFreeze branch +* @tc.type: FUNC +*/ +HWTEST_F(BinderInvokerUnitTest, CheckFreezeTest, TestSize.Level1) +{ + bool isFreeze = false; + BinderInvoker binderInvoker; + binderInvoker.binderConnector_ = nullptr; + EXPECT_EQ(binderInvoker.CheckFreeze(1, isFreeze), IPC_INVOKER_CONNECT_ERR); + + BinderConnector *binderConnector = BinderConnector::GetInstance(); + binderConnector->driverFD_ = -1; + binderInvoker.binderConnector_ = binderConnector; + EXPECT_EQ(binderInvoker.CheckFreeze(1, isFreeze), IPC_INVOKER_CONNECT_ERR); +} +#endif // FREEZE_PROCESS_SUPPORT } // 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 71223a8fb0ba3f168528ce4384477b45784a6a28..8e35ca666c4d54148c600368c3b17a06e0a90388 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,59 @@ HWTEST_F(IPCSkeletonTest, GetDCallingTokenIDInRPC, TestSize.Level1) ASSERT_EQ(tokenId, 0); } + +#ifdef FREEZE_PROCESS_SUPPORT +/** + * @tc.name: FreezeProcessTest + * @tc.desc: Verify the FreezeProcess 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); + MockIRemoteInvoker *dbinderInvoker = new (std::nothrow) MockIRemoteInvoker(); + ASSERT_NE(dbinderInvoker, nullptr); + + current->invokers_[IRemoteObject::IF_PROT_DEFAULT] = binderInvoker; + current->invokers_[IRemoteObject::IF_PROT_DATABUS] = dbinderInvoker; + + EXPECT_CALL(*binderInvoker, FreezeProcess(testing::_, testing::_, testing::_)()) + .WillRepeatedly(testing::Return(ERR_NONE)); + + ASSERT_EQ(IPCSkeleton::FreezeProcess(0, true, 0), ERR_NONE); + std::fill(current->invokers_, current->invokers_ + IPCThreadSkeleton::INVOKER_MAX_COUNT, nullptr); + delete binderInvoker; + delete dbinderInvoker; +} + +/** + * @tc.name: CheckFreezeTest + * @tc.desc: Verify the CheckFreeze 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); + MockIRemoteInvoker *dbinderInvoker = new (std::nothrow) MockIRemoteInvoker(); + ASSERT_NE(dbinderInvoker, nullptr); + + current->invokers_[IRemoteObject::IF_PROT_DEFAULT] = binderInvoker; + current->invokers_[IRemoteObject::IF_PROT_DATABUS] = dbinderInvoker; + + EXPECT_CALL(*binderInvoker, CheckFreeze(testing::_, testing::_)()) + .WillRepeatedly(testing::Return(ERR_NONE)); + + bool isFreeze = false; + ASSERT_EQ(IPCSkeleton::CheckFreeze(0, isFreeze), ERR_NONE); + std::fill(current->invokers_, current->invokers_ + IPCThreadSkeleton::INVOKER_MAX_COUNT, nullptr); + delete binderInvoker; + delete dbinderInvoker; +} +#endif // FREEZE_PROCESS_SUPPORT } // 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 b53f3bb8425b0ef8bb3aae5942c8e0bd8cb99b86..e0402d68453809a22ee752f749b50af03ea22791 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_SUPPORT + MOCK_METHOD3(FreezeProcess, int32_t(uint32_t pid, bool freeze, uint32_t timeout)); + MOCK_METHOD2(CheckFreeze, int32_t(uint32_t pid, bool &isFreeze)); +#endif // FREEZE_PROCESS_SUPPORT 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 8832a75c09feeb9be939a7abb491be26a66415bb..e5d4b12a784f9d6727076405ab11d90b361cfe40 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_freeze_process_support) { + defines = [ "FREEZE_PROCESS_SUPPORT" ] + } + 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_freeze_process_support) { + defines = [ "FREEZE_PROCESS_SUPPORT" ] + } + 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_freeze_process_support) { + defines = [ "FREEZE_PROCESS_SUPPORT" ] + } + 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 c7ece9764da7b2efd88555dc33264539019fc89b..a522e2aff01ed5032f4de11d9d609a0aaa81fc38 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_SUPPORT + int TestFreezeProcess() override; +#endif // FREEZE_PROCESS_SUPPORT 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 fd969b7c23903d417e2ad1857176f33526ef66f4..e0670ca34cce40c3e88427575d1c8086e8b1d38e 100644 --- a/ipc/test/auxiliary/native/include/test_service_base.h +++ b/ipc/test/auxiliary/native/include/test_service_base.h @@ -56,6 +56,8 @@ 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, + TRANS_ID_TEST_TXN_ASYNC_TRANS = 28, }; public: virtual int TestSyncTransaction(int value, int &reply, int delayTime = 0) = 0; @@ -87,6 +89,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_SUPPORT + virtual int TestFreezeProcess() = 0; +#endif // FREEZE_PROCESS_SUPPORT 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 605e805dd5e7a3f266bf43a4ede9c863e85c4bde..bc22e773a692449662756a1701e9b3d63fac46f5 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_SUPPORT + int TestFreezeProcess(); +#endif // FREEZE_PROCESS_SUPPORT 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 c633411411e7f8a7cbb781b9f59d9b550b5b407d..15d0cd679111abae5160cb937d881f1410f40631 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_SUPPORT + int TestFreezeProcess() override; +#endif // FREEZE_PROCESS_SUPPORT 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 b51900369a308a7b5b4068839295d89fdd537cf5..32bdfaa88e29c3d3868dc9e074341142953a0543 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_SUPPORT + int32_t ServerTestFreezeProcess(MessageParcel &data, MessageParcel &reply); +#endif // FREEZE_PROCESS_SUPPORT 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 6a486a5798d85dce2925b16f567b6282c5c63a30..e5d66bc5acec115f9788125dba6618857247a781 100644 --- a/ipc/test/auxiliary/native/src/main_client.cpp +++ b/ipc/test/auxiliary/native/src/main_client.cpp @@ -82,6 +82,18 @@ void TestCaseSyncTrans(std::shared_ptr &testClient) } } +#ifdef FREEZE_PROCESS_SUPPORT +void TestCaseFreezeProcess(std::shared_ptr &testClient) +{ + bool ret = testClient->TestFreezeProcess(); + if (!ret) { + std::cout << "[FAILED] Execution of TestFreezeProcess case failed" < &testClient) { bool ret = testClient->StartPingService(); @@ -229,6 +241,9 @@ void ExecuteAllTestCase() TestCaseRegisterRemoteStub(testClient); TestCaseTooManyRequests(testClient); TestCaseMultiThreadSendRequest(testClient); +#ifdef FREEZE_PROCESS_SUPPORT + TestCaseFreezeProcess(testClient); +#endif // FREEZE_PROCESS_SUPPORT 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 c8b6b1103f71c27c0ac9fc6393934c6e4a92d863..379b26954d38cd6e7adb2c862e5549b2451a7906 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_SUPPORT +int TestService::TestFreezeProcess() +{ + return 0; +} +#endif // FREEZE_PROCESS_SUPPORT } // 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 154a7b0e2a787d87d2e611cd306dec9d28f9fc9c..479c965f0a0eb84f39541ab17850a48ad078ff7f 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_SUPPORT +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_SUPPORT + 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 7a04351199fde2d3997853831b4a3ae3034cb81b..10ed413a9f73d1769ec906b3df56472f6a3234a6 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,52 @@ int TestServiceProxy::TestSyncTransaction(int value, int &reply, int delayTime) return ret; } +#ifdef FREEZE_PROCESS_SUPPORT +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::FreezeProcess(pid, true, timeout); + if (res != ERR_NONE) { + ZLOGI(LABEL, "FreezeProcess failed, error code = %{public}d", res); + return res; + } + + bool isFreeze = false; + res = IPCSkeleton::CheckFreeze(pid, isFreeze); + if (res != ERR_NONE || !isFreeze) { + ZLOGI(LABEL, "CheckFreeze failed, error code = %{public}d isFreeze = %{public}d ",res, isFreeze); + 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_SUPPORT + 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 d9a082d6abb6bc7e515ad989377163a9fbf8a42f..549216b60634368f931cde4e16af5f559408a6ab 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_SUPPORT + funcMap_[static_cast(TRANS_ID_TEST_FREEZE_PROCESS)] = &TestServiceStub::ServerTestFreezeProcess; +#endif // FREEZE_PROCESS_SUPPORT } 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_SUPPORT +int32_t TestServiceStub::ServerTestFreezeProcess(MessageParcel &data, MessageParcel &reply) +{ + reply.WriteInt32(getpid()); + return 0; +} +#endif // FREEZE_PROCESS_SUPPORT } // namespace OHOS diff --git a/test/fuzztest/ipc/native/src/core/binderinvokernew_fuzzer/binderinvokernew_fuzzer.cpp b/test/fuzztest/ipc/native/src/core/binderinvokernew_fuzzer/binderinvokernew_fuzzer.cpp index 89196cb964122cf320c58b8c052f090d0e2e8268..57467c555bc6ddd8baf84e07f28a0fdde50c9a69 100644 --- a/test/fuzztest/ipc/native/src/core/binderinvokernew_fuzzer/binderinvokernew_fuzzer.cpp +++ b/test/fuzztest/ipc/native/src/core/binderinvokernew_fuzzer/binderinvokernew_fuzzer.cpp @@ -44,7 +44,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, }; static const std::vector binderTypeList = { BINDER_TYPE_BINDER, diff --git a/test/unittest/ipc/native/src/core/source/BUILD.gn b/test/unittest/ipc/native/src/core/source/BUILD.gn index 8236f82ce7cba97d0638439c14e5226c331851a3..c843dc2dc361ab167ae966935fed77acbbddf1ae 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_freeze_process_support) { + defines += [ "FREEZE_PROCESS_SUPPORT" ] + } + 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 b524b532688a84ccdf06e26c8ce2a86296de6ffa..bb2d41a626a8119a8400253817472e8a14570099 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_SUPPORT + MOCK_METHOD3(FreezeProcess, int32_t(uint32_t pid, bool freeze, uint32_t timeout)); + MOCK_METHOD2(CheckFreeze, int32_t(uint32_t pid, bool &isFreeze)); +#endif // FREEZE_PROCESS_SUPPORT MOCK_METHOD2(AddDeathRecipient, bool(int32_t handle, void *cookie)); MOCK_METHOD2(RemoveDeathRecipient, bool(int32_t handle, void *cookie)); MOCK_METHOD1(SetMaxWorkThread, bool(int maxThreadNum));