1 // SPDX-License-Identifier: GPL-2.0 2 /* 3 * This test covers the PR_SET_NAME functionality of prctl calls 4 */ 5 6 #include <errno.h> 7 #include <sys/prctl.h> 8 #include <string.h> 9 10 #include "../kselftest_harness.h" 11 12 #define CHANGE_NAME "changename" 13 #define EMPTY_NAME "" 14 #define TASK_COMM_LEN 16 15 #define MAX_PATH_LEN 50 16 17 int set_name(char *name) 18 { 19 int res; 20 21 res = prctl(PR_SET_NAME, name, NULL, NULL, NULL); 22 23 if (res < 0) 24 return -errno; 25 return res; 26 } 27 28 int check_is_name_correct(char *check_name) 29 { 30 char name[TASK_COMM_LEN]; 31 int res; 32 33 res = prctl(PR_GET_NAME, name, NULL, NULL, NULL); 34 35 if (res < 0) 36 return -errno; 37 38 return !strcmp(name, check_name); 39 } 40 41 int check_null_pointer(char *check_name) 42 { 43 char *name = NULL; 44 int res; 45 46 res = prctl(PR_GET_NAME, name, NULL, NULL, NULL); 47 48 return res; 49 } 50 51 int check_name(void) 52 { 53 54 int pid; 55 56 pid = getpid(); 57 FILE *fptr = NULL; 58 char path[MAX_PATH_LEN] = {}; 59 char name[TASK_COMM_LEN] = {}; 60 char output[TASK_COMM_LEN] = {}; 61 int j; 62 63 j = snprintf(path, MAX_PATH_LEN, "/proc/self/task/%d/comm", pid); 64 fptr = fopen(path, "r"); 65 if (!fptr) 66 return -EIO; 67 68 fscanf(fptr, "%s", output); 69 if (ferror(fptr)) 70 return -EIO; 71 72 int res = prctl(PR_GET_NAME, name, NULL, NULL, NULL); 73 74 if (res < 0) 75 return -errno; 76 77 return !strcmp(output, name); 78 } 79 80 TEST(rename_process) { 81 82 EXPECT_GE(set_name(CHANGE_NAME), 0); 83 EXPECT_TRUE(check_is_name_correct(CHANGE_NAME)); 84 85 EXPECT_GE(set_name(EMPTY_NAME), 0); 86 EXPECT_TRUE(check_is_name_correct(EMPTY_NAME)); 87 88 EXPECT_GE(set_name(CHANGE_NAME), 0); 89 EXPECT_LT(check_null_pointer(CHANGE_NAME), 0); 90 91 EXPECT_TRUE(check_name()); 92 } 93 94 TEST_HARNESS_MAIN 95