Better nice name

This commit is contained in:
topjohnwu
2019-02-15 04:31:39 -05:00
parent 8b84039f1f
commit a19c7215d2
5 changed files with 18 additions and 7 deletions

View File

@ -85,6 +85,8 @@ int fork_dont_care();
int fork_no_zombie();
void gen_rand_str(char *buf, int len);
int strend(const char *s1, const char *s2);
void init_argv0(int argc, char **argv);
void set_nice_name(const char *name);
#define getline __getline
#define getdelim __getdelim

View File

@ -224,3 +224,15 @@ int new_daemon_thread(void *(*start_routine) (void *), void *arg, const pthread_
return ret;
}
static char *argv0;
static size_t name_len;
void init_argv0(int argc, char **argv) {
argv0 = argv[0];
name_len = (argv[argc - 1] - argv[0]) + strlen(argv[argc - 1]) + 1;
}
void set_nice_name(const char *name) {
memset(argv0, 0, name_len);
strlcpy(argv0, name, name_len);
prctl(PR_SET_NAME, name);
}