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