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