1 /* 2 * Copyright (c) 1994, 1995 Christopher G. Demetriou 3 * All rights reserved. 4 * 5 * Redistribution and use in source and binary forms, with or without 6 * modification, are permitted provided that the following conditions 7 * are met: 8 * 1. Redistributions of source code must retain the above copyright 9 * notice, this list of conditions and the following disclaimer. 10 * 2. Redistributions in binary form must reproduce the above copyright 11 * notice, this list of conditions and the following disclaimer in the 12 * documentation and/or other materials provided with the distribution. 13 * 3. All advertising materials mentioning features or use of this software 14 * must display the following acknowledgement: 15 * This product includes software developed by Christopher G. Demetriou 16 * for the NetBSD Project. 17 * 4. The name of the author may not be used to endorse or promote products 18 * derived from this software without specific prior written permission 19 * 20 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR 21 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES 22 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. 23 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, 24 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT 25 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 26 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 27 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 28 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF 29 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 30 */ 31 32 #if defined(LIBC_SCCS) && !defined(lint) 33 static char rcsid[] = "$OpenBSD: setproctitle.c,v 1.7 1999/02/25 22:10:12 art Exp $"; 34 #endif /* LIBC_SCCS and not lint */ 35 36 #include <sys/param.h> 37 #include <sys/types.h> 38 #include <sys/exec.h> 39 #include <sys/sysctl.h> 40 #include <vm/vm.h> 41 42 #include <stdio.h> 43 #include <stdlib.h> 44 #include <string.h> 45 46 #ifdef __STDC__ 47 #include <stdarg.h> 48 #else 49 #include <varargs.h> 50 #endif 51 52 #define MAX_PROCTITLE 2048 53 54 extern char *__progname; /* Program name, from crt0. */ 55 56 void 57 #ifdef __STDC__ 58 setproctitle(const char *fmt, ...) 59 #else 60 setproctitle(fmt, va_alist) 61 const char *fmt; 62 va_dcl 63 #endif 64 { 65 static struct ps_strings *ps; 66 va_list ap; 67 68 static char buf[MAX_PROCTITLE], *bufp = buf; 69 size_t used; 70 71 #ifdef __STDC__ 72 va_start(ap, fmt); 73 #else 74 va_start(ap); 75 #endif 76 if (fmt != NULL) { 77 used = snprintf(buf, MAX_PROCTITLE, "%s: ", __progname); 78 if (used >= MAX_PROCTITLE) 79 used = MAX_PROCTITLE - 1; 80 (void)vsnprintf(buf + used, MAX_PROCTITLE - used, fmt, ap); 81 } else 82 (void)snprintf(buf, MAX_PROCTITLE, "%s", __progname); 83 va_end(ap); 84 85 if (ps == NULL) { 86 struct _ps_strings _ps; 87 int mib[2]; 88 size_t len; 89 90 mib[0] = CTL_VM; 91 mib[1] = VM_PSSTRINGS; 92 len = sizeof(_ps); 93 if (sysctl(mib, 2, &_ps, &len, NULL, 0) != 0) 94 return; 95 ps = (struct ps_strings *)_ps.val; 96 } 97 ps->ps_nargvstr = 1; 98 ps->ps_argvstr = &bufp; 99 } 100