1 /* 2 * Copyright (c) 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 CODE_SIGN_ACCESS_TOKEN_SETTER_H 17 #define CODE_SIGN_ACCESS_TOKEN_SETTER_H 18 19 #include <cstdint> 20 21 #include "accesstoken_kit.h" 22 #include "access_token.h" 23 #include "nativetoken_kit.h" 24 #include "token_setproc.h" 25 26 namespace OHOS { 27 namespace Security { 28 namespace CodeSign { 29 using namespace OHOS::Security::AccessToken; 30 NativeTokenSet(const char * caller)31inline uint64_t NativeTokenSet(const char *caller) 32 { 33 uint64_t tokenId = GetSelfTokenID(); 34 uint64_t mockTokenId = AccessTokenKit::GetNativeTokenId(caller); 35 SetSelfTokenID(mockTokenId); 36 return tokenId; 37 } 38 NativeTokenReset(uint64_t tokenId)39inline void NativeTokenReset(uint64_t tokenId) 40 { 41 SetSelfTokenID(tokenId); 42 } 43 } 44 } 45 } 46 #endif 47