1 /***
2 This file is part of avahi.
3
4 avahi is free software; you can redistribute it and/or modify it
5 under the terms of the GNU Lesser General Public License as
6 published by the Free Software Foundation; either version 2.1 of the
7 License, or (at your option) any later version.
8
9 avahi is distributed in the hope that it will be useful, but WITHOUT
10 ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
11 or FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser General
12 Public License for more details.
13
14 You should have received a copy of the GNU Lesser General Public
15 License along with avahi; if not, write to the Free Software
16 Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
17 USA.
18 ***/
19
20 #ifdef HAVE_CONFIG_H
21 #include <config.h>
22 #endif
23
24 #include <stdio.h>
25 #include <stdlib.h>
26 #include <string.h>
27 #include <stdarg.h>
28 #include <unistd.h>
29
30 #ifdef HAVE_SYS_PRCTL_H
31 #include <sys/prctl.h>
32 #endif
33
34 #include "avahi-common/avahi-malloc.h"
35
36 #include "setproctitle.h"
37
38 #if !defined(HAVE_SETPROCTITLE) && defined(__linux__)
39 static char** argv_buffer = NULL;
40 static size_t argv_size = 0;
41
42 #if !HAVE_DECL_ENVIRON
43 extern char **environ;
44 #endif
45
46 #endif
47
avahi_init_proc_title(int argc,char ** argv)48 void avahi_init_proc_title(int argc, char **argv) {
49
50 #if !defined(HAVE_SETPROCTITLE) && defined(__linux__)
51
52 unsigned i;
53 char **new_environ, *endptr;
54
55 /* This code is really really ugly. We make some memory layout
56 * assumptions and reuse the environment array as memory to store
57 * our process title in */
58
59 for (i = 0; environ[i]; i++);
60
61 endptr = i ? environ[i-1] + strlen(environ[i-1]) : argv[argc-1] + strlen(argv[argc-1]);
62
63 argv_buffer = argv;
64 argv_size = endptr - argv_buffer[0];
65
66 /* Make a copy of environ */
67
68 new_environ = avahi_malloc(sizeof(char*) * (i + 1));
69 for (i = 0; environ[i]; i++)
70 new_environ[i] = avahi_strdup(environ[i]);
71 new_environ[i] = NULL;
72
73 environ = new_environ;
74
75 #endif
76 }
77
avahi_set_proc_title(const char * name,const char * fmt,...)78 void avahi_set_proc_title(const char *name, const char *fmt,...) {
79 #ifdef HAVE_SETPROCTITLE
80 char t[256];
81
82 va_list ap;
83 va_start(ap, fmt);
84 vsnprintf(t, sizeof(t), fmt, ap);
85 va_end(ap);
86
87 setproctitle("-%s", t);
88 #elif defined(__linux__)
89 size_t l;
90 va_list ap;
91
92 if (!argv_buffer)
93 return;
94
95 va_start(ap, fmt);
96 vsnprintf(argv_buffer[0], argv_size, fmt, ap);
97 va_end(ap);
98
99 l = strlen(argv_buffer[0]);
100
101 memset(argv_buffer[0] + l, 0, argv_size - l);
102 argv_buffer[1] = NULL;
103 #endif
104
105 #if defined(HAVE_SYS_PRCTL_H) && defined(PR_SET_NAME)
106
107 if (name)
108 prctl(PR_SET_NAME, (unsigned long) name, 0, 0, 0);
109
110 #endif
111 }
112