1 c423c56e 2022-09-25 op /*
 2 c423c56e 2022-09-25 op  * Copyright (c) 2016 Nicholas Marriott <nicholas.marriott@gmail.com>
 3 c423c56e 2022-09-25 op  *
 4 c423c56e 2022-09-25 op  * Permission to use, copy, modify, and distribute this software for any
 5 c423c56e 2022-09-25 op  * purpose with or without fee is hereby granted, provided that the above
 6 c423c56e 2022-09-25 op  * copyright notice and this permission notice appear in all copies.
 7 c423c56e 2022-09-25 op  *
 8 c423c56e 2022-09-25 op  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
 9 c423c56e 2022-09-25 op  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
10 c423c56e 2022-09-25 op  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
11 c423c56e 2022-09-25 op  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
12 c423c56e 2022-09-25 op  * WHATSOEVER RESULTING FROM LOSS OF MIND, USE, DATA OR PROFITS, WHETHER
13 c423c56e 2022-09-25 op  * IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING
14 c423c56e 2022-09-25 op  * OUT OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
15 c423c56e 2022-09-25 op  */
16 c423c56e 2022-09-25 op 
17 c423c56e 2022-09-25 op #include <sys/types.h>
18 c423c56e 2022-09-25 op 
19 c423c56e 2022-09-25 op #include <stdarg.h>
20 c423c56e 2022-09-25 op #include <stdio.h>
21 c423c56e 2022-09-25 op #include <string.h>
22 c423c56e 2022-09-25 op 
23 c423c56e 2022-09-25 op #include "../config.h"
24 c423c56e 2022-09-25 op 
25 c423c56e 2022-09-25 op void	setproctitle(const char *, ...);
26 c423c56e 2022-09-25 op 
27 c423c56e 2022-09-25 op #if HAVE_PR_SET_NAME
28 c423c56e 2022-09-25 op 
29 c423c56e 2022-09-25 op #include <sys/prctl.h>
30 c423c56e 2022-09-25 op 
31 c423c56e 2022-09-25 op void
32 c423c56e 2022-09-25 op setproctitle(const char *fmt, ...)
33 c423c56e 2022-09-25 op {
34 c423c56e 2022-09-25 op 	char	title[16], name[16], *cp;
35 c423c56e 2022-09-25 op 	va_list	ap;
36 c423c56e 2022-09-25 op 	int	used;
37 c423c56e 2022-09-25 op 
38 c423c56e 2022-09-25 op 	va_start(ap, fmt);
39 c423c56e 2022-09-25 op 	vsnprintf(title, sizeof title, fmt, ap);
40 c423c56e 2022-09-25 op 	va_end(ap);
41 c423c56e 2022-09-25 op 
42 c423c56e 2022-09-25 op 	used = snprintf(name, sizeof name, "%s: %s", getprogname(), title);
43 c423c56e 2022-09-25 op 	if (used >= (int)sizeof name) {
44 c423c56e 2022-09-25 op 		cp = strrchr(name, ' ');
45 c423c56e 2022-09-25 op 		if (cp != NULL)
46 c423c56e 2022-09-25 op 			*cp = '\0';
47 c423c56e 2022-09-25 op 	}
48 c423c56e 2022-09-25 op 	prctl(PR_SET_NAME, name);
49 c423c56e 2022-09-25 op }
50 c423c56e 2022-09-25 op #else
51 c423c56e 2022-09-25 op void
52 c423c56e 2022-09-25 op setproctitle(const char *fmt, ...)
53 c423c56e 2022-09-25 op {
54 c423c56e 2022-09-25 op 	(void)fmt;
55 c423c56e 2022-09-25 op }
56 c423c56e 2022-09-25 op #endif