1 /* 2 * Copyright (C) 2021-2023 Huawei Device Co., Ltd. 3 * Licensed under the Apache License, Version 2.0 (the "License"); 4 * you may not use this file except in compliance with the License. 5 * You may obtain a copy of the License at 6 * 7 * http://www.apache.org/licenses/LICENSE-2.0 8 * 9 * Unless required by applicable law or agreed to in writing, software 10 * distributed under the License is distributed on an "AS IS" BASIS, 11 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 12 * See the License for the specific language governing permissions and 13 * limitations under the License. 14 */ 15 16 #ifndef OHOS_IPC_IPC_SKELETON_H 17 #define OHOS_IPC_IPC_SKELETON_H 18 19 #include <cstdint> 20 #include "gmock/gmock.h" 21 22 namespace OHOS { 23 #define getpid() 1 24 #define getuid() 1 25 class MockMidIPCSkeleton { 26 public: 27 virtual ~MockMidIPCSkeleton() = default; 28 virtual uint32_t GetSelfTokenID() = 0; 29 virtual uint32_t GetCallingUid() = 0; 30 virtual uint32_t GetCallingTokenID() = 0; 31 virtual uint32_t GetCallingPid() = 0; 32 }; 33 34 class MockIPCSkeleton : public MockMidIPCSkeleton { 35 public: 36 MockIPCSkeleton() = default; 37 ~MockIPCSkeleton() override = default; 38 MOCK_METHOD(uint32_t, GetSelfTokenID, ()); 39 MOCK_METHOD(uint32_t, GetCallingUid, ()); 40 MOCK_METHOD(uint32_t, GetCallingTokenID, ()); 41 MOCK_METHOD(uint32_t, GetCallingPid, ()); 42 }; 43 44 class IPCSkeleton { 45 public: 46 static MockIPCSkeleton* obj; 47 ~IPCSkeleton() = default; GetSelfTokenID()48 static uint32_t GetSelfTokenID() 49 { 50 return obj->GetSelfTokenID(); 51 } GetCallingUid()52 static uint32_t GetCallingUid() 53 { 54 return obj->GetCallingUid(); 55 } GetCallingPid()56 static uint32_t GetCallingPid() 57 { 58 return obj->GetCallingPid(); 59 } GetCallingTokenID()60 static uint32_t GetCallingTokenID() 61 { 62 return obj->GetCallingTokenID(); 63 } 64 }; 65 66 } 67 #endif 68