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