1570af302Sopenharmony_ci/** 2570af302Sopenharmony_ci * Copyright (c) 2022 Huawei Device Co., Ltd. 3570af302Sopenharmony_ci * Licensed under the Apache License, Version 2.0 (the "License"); 4570af302Sopenharmony_ci * you may not use this file except in compliance with the License. 5570af302Sopenharmony_ci * You may obtain a copy of the License at 6570af302Sopenharmony_ci * 7570af302Sopenharmony_ci * http://www.apache.org/licenses/LICENSE-2.0 8570af302Sopenharmony_ci * 9570af302Sopenharmony_ci * Unless required by applicable law or agreed to in writing, software 10570af302Sopenharmony_ci * distributed under the License is distributed on an "AS IS" BASIS, 11570af302Sopenharmony_ci * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 12570af302Sopenharmony_ci * See the License for the specific language governing permissions and 13570af302Sopenharmony_ci * limitations under the License. 14570af302Sopenharmony_ci */ 15570af302Sopenharmony_ci 16570af302Sopenharmony_ci#include <stdlib.h> 17570af302Sopenharmony_ci#include <sys/mman.h> 18570af302Sopenharmony_ci#include "functionalext.h" 19570af302Sopenharmony_ci 20570af302Sopenharmony_ciconst int SUCCESS = 0; 21570af302Sopenharmony_ciconst int FAILED = -1; 22570af302Sopenharmony_ci 23570af302Sopenharmony_ci/** 24570af302Sopenharmony_ci * @tc.name : mprotect_0100 25570af302Sopenharmony_ci * @tc.desc : The port parameter is PROT_READ,which can set the protection attribute of memory area. 26570af302Sopenharmony_ci * @tc.level : Level 0 27570af302Sopenharmony_ci */ 28570af302Sopenharmony_civoid mprotect_0100(void) 29570af302Sopenharmony_ci{ 30570af302Sopenharmony_ci size_t align = getpagesize(); 31570af302Sopenharmony_ci void *buffer = memalign(align, 8 * align); 32570af302Sopenharmony_ci int ret = mprotect(buffer, getpagesize(), PROT_READ); 33570af302Sopenharmony_ci EXPECT_EQ("mprotect_0100", ret, SUCCESS); 34570af302Sopenharmony_ci} 35570af302Sopenharmony_ci 36570af302Sopenharmony_ci/** 37570af302Sopenharmony_ci * @tc.name : mprotect_0200 38570af302Sopenharmony_ci * @tc.desc : The port parameter is PROT_WRITE,which can set the protection attribute of memory area. 39570af302Sopenharmony_ci * @tc.level : Level 0 40570af302Sopenharmony_ci */ 41570af302Sopenharmony_civoid mprotect_0200(void) 42570af302Sopenharmony_ci{ 43570af302Sopenharmony_ci size_t align = getpagesize(); 44570af302Sopenharmony_ci void *buffer = memalign(align, 8 * align); 45570af302Sopenharmony_ci int ret = mprotect(buffer, getpagesize(), PROT_WRITE); 46570af302Sopenharmony_ci EXPECT_EQ("mprotect_0200", ret, SUCCESS); 47570af302Sopenharmony_ci} 48570af302Sopenharmony_ci 49570af302Sopenharmony_ci/** 50570af302Sopenharmony_ci * @tc.name : mprotect_0300 51570af302Sopenharmony_ci * @tc.desc : The port parameter is PROT_EXEC,which can set the protection attribute of memory area. 52570af302Sopenharmony_ci * @tc.level : Level 0 53570af302Sopenharmony_ci */ 54570af302Sopenharmony_civoid mprotect_0300(void) 55570af302Sopenharmony_ci{ 56570af302Sopenharmony_ci size_t align = getpagesize(); 57570af302Sopenharmony_ci void *buffer = memalign(align, 8 * align); 58570af302Sopenharmony_ci int ret = mprotect(buffer, getpagesize(), PROT_EXEC); 59570af302Sopenharmony_ci EXPECT_EQ("mprotect_0300", ret, SUCCESS); 60570af302Sopenharmony_ci} 61570af302Sopenharmony_ci 62570af302Sopenharmony_ci/** 63570af302Sopenharmony_ci * @tc.name : mprotect_0400 64570af302Sopenharmony_ci * @tc.desc : The port parameter is PROT_NONE,which can set the protection attribute of memory area. 65570af302Sopenharmony_ci * @tc.level : Level 0 66570af302Sopenharmony_ci */ 67570af302Sopenharmony_civoid mprotect_0400(void) 68570af302Sopenharmony_ci{ 69570af302Sopenharmony_ci size_t align = getpagesize(); 70570af302Sopenharmony_ci void *buffer = memalign(align, 8 * align); 71570af302Sopenharmony_ci int ret = mprotect(buffer, getpagesize(), PROT_NONE); 72570af302Sopenharmony_ci EXPECT_EQ("mprotect_0400", ret, SUCCESS); 73570af302Sopenharmony_ci} 74570af302Sopenharmony_ci 75570af302Sopenharmony_ci/** 76570af302Sopenharmony_ci * @tc.name : mprotect_0500 77570af302Sopenharmony_ci * @tc.desc : The port parameter is PROT_READ|PROT_WRITE,which can set the protection attribute of memory area. 78570af302Sopenharmony_ci * @tc.level : Level 1 79570af302Sopenharmony_ci */ 80570af302Sopenharmony_civoid mprotect_0500(void) 81570af302Sopenharmony_ci{ 82570af302Sopenharmony_ci size_t align = getpagesize(); 83570af302Sopenharmony_ci void *buffer = memalign(align, 8 * align); 84570af302Sopenharmony_ci int ret = mprotect(buffer, getpagesize(), PROT_READ | PROT_WRITE); 85570af302Sopenharmony_ci EXPECT_EQ("mprotect_0500", ret, SUCCESS); 86570af302Sopenharmony_ci} 87570af302Sopenharmony_ci 88570af302Sopenharmony_ci/** 89570af302Sopenharmony_ci * @tc.name : mprotect_0600 90570af302Sopenharmony_ci * @tc.desc : The start parameter is not an integer multiple of the page, 91570af302Sopenharmony_ci * and the protection attribute of the memory area cannot be set. 92570af302Sopenharmony_ci * @tc.level : Level 2 93570af302Sopenharmony_ci */ 94570af302Sopenharmony_civoid mprotect_0600(void) 95570af302Sopenharmony_ci{ 96570af302Sopenharmony_ci int ret = mprotect((void *)0x0001, getpagesize(), PROT_READ); 97570af302Sopenharmony_ci EXPECT_EQ("mprotect_0600", ret, FAILED); 98570af302Sopenharmony_ci} 99570af302Sopenharmony_ci 100570af302Sopenharmony_ciint main(int argc, char *argv[]) 101570af302Sopenharmony_ci{ 102570af302Sopenharmony_ci mprotect_0100(); 103570af302Sopenharmony_ci mprotect_0200(); 104570af302Sopenharmony_ci mprotect_0300(); 105570af302Sopenharmony_ci mprotect_0400(); 106570af302Sopenharmony_ci mprotect_0500(); 107570af302Sopenharmony_ci mprotect_0600(); 108570af302Sopenharmony_ci 109570af302Sopenharmony_ci return t_status; 110570af302Sopenharmony_ci}