95 lines
1.6 KiB
C
95 lines
1.6 KiB
C
// SPDX-License-Identifier: GPL-2.0
|
|
/*
|
|
* This test covers the PR_SET_NAME functionality of prctl calls
|
|
*/
|
|
|
|
#include <errno.h>
|
|
#include <sys/prctl.h>
|
|
#include <string.h>
|
|
|
|
#include "../kselftest_harness.h"
|
|
|
|
#define CHANGE_NAME "changename"
|
|
#define EMPTY_NAME ""
|
|
#define TASK_COMM_LEN 16
|
|
#define MAX_PATH_LEN 50
|
|
|
|
int set_name(char *name)
|
|
{
|
|
int res;
|
|
|
|
res = prctl(PR_SET_NAME, name, NULL, NULL, NULL);
|
|
|
|
if (res < 0)
|
|
return -errno;
|
|
return res;
|
|
}
|
|
|
|
int check_is_name_correct(char *check_name)
|
|
{
|
|
char name[TASK_COMM_LEN];
|
|
int res;
|
|
|
|
res = prctl(PR_GET_NAME, name, NULL, NULL, NULL);
|
|
|
|
if (res < 0)
|
|
return -errno;
|
|
|
|
return !strcmp(name, check_name);
|
|
}
|
|
|
|
int check_null_pointer(char *check_name)
|
|
{
|
|
char *name = NULL;
|
|
int res;
|
|
|
|
res = prctl(PR_GET_NAME, name, NULL, NULL, NULL);
|
|
|
|
return res;
|
|
}
|
|
|
|
int check_name(void)
|
|
{
|
|
|
|
int pid;
|
|
|
|
pid = getpid();
|
|
FILE *fptr = NULL;
|
|
char path[MAX_PATH_LEN] = {};
|
|
char name[TASK_COMM_LEN] = {};
|
|
char output[TASK_COMM_LEN] = {};
|
|
int j;
|
|
|
|
j = snprintf(path, MAX_PATH_LEN, "/proc/self/task/%d/comm", pid);
|
|
fptr = fopen(path, "r");
|
|
if (!fptr)
|
|
return -EIO;
|
|
|
|
fscanf(fptr, "%s", output);
|
|
if (ferror(fptr))
|
|
return -EIO;
|
|
|
|
int res = prctl(PR_GET_NAME, name, NULL, NULL, NULL);
|
|
|
|
if (res < 0)
|
|
return -errno;
|
|
|
|
return !strcmp(output, name);
|
|
}
|
|
|
|
TEST(rename_process) {
|
|
|
|
EXPECT_GE(set_name(CHANGE_NAME), 0);
|
|
EXPECT_TRUE(check_is_name_correct(CHANGE_NAME));
|
|
|
|
EXPECT_GE(set_name(EMPTY_NAME), 0);
|
|
EXPECT_TRUE(check_is_name_correct(EMPTY_NAME));
|
|
|
|
EXPECT_GE(set_name(CHANGE_NAME), 0);
|
|
EXPECT_LT(check_null_pointer(CHANGE_NAME), 0);
|
|
|
|
EXPECT_TRUE(check_name());
|
|
}
|
|
|
|
TEST_HARNESS_MAIN
|