1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
|
/*
* Unix SMB/CIFS implementation.
*
* Process utils.
*
* Copyright (c) 2013 Andreas Schneider <asn@samba.org>
*
* This program is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation; either version 3 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program; if not, see <http://www.gnu.org/licenses/>.
*/
#include "util_process.h"
#include "replace.h"
#ifdef HAVE_SYS_PRCTL_H
#include <sys/prctl.h>
#endif
/*
* These variables are static so that we can access them
* with process_get_short_title() and process_get_long_title(). The
* purpose of this is to allow smb_panic_log() to print them.
*/
static char short_comment[16] = {0,};
static char long_comment[256] = {0,};
static char binary_name[256];
void process_set_title(const char *short_format, const char *long_format, ...)
{
#if defined(HAVE_PRCTL) && defined(PR_SET_NAME)
if (short_format != NULL) {
va_list ap;
va_start(ap, long_format);
vsnprintf(short_comment, sizeof(short_comment), short_format, ap);
va_end(ap);
prctl(PR_SET_NAME, (unsigned long) short_comment, 0, 0, 0);
}
#endif
if (long_format != NULL) {
va_list ap;
va_start(ap, long_format);
vsnprintf(long_comment, sizeof(long_comment), long_format, ap);
va_end(ap);
setproctitle("%s", long_comment);
}
}
const char *process_get_short_title(void)
{
return short_comment;
}
const char *process_get_long_title(void)
{
return long_comment;
}
/*
* This is just for debugging in a panic, so we don't want to do
* anything more than return a fixed pointer, so we save a copy to a
* static variable.
*/
void process_save_binary_name(const char *progname)
{
strlcpy(binary_name, progname, sizeof(binary_name));
}
/* Samba binaries will set this during popt handling */
const char *process_get_saved_binary_name(void)
{
return binary_name;
}
int prctl_set_comment(const char *comment_format, ...)
{
char comment[16];
va_list ap;
va_start(ap, comment_format);
vsnprintf(comment, sizeof(comment), comment_format, ap);
va_end(ap);
process_set_title("%s", "%s", comment);
return 0;
}
|
