1 /* $OpenBSD: setproctitle.c,v 1.11 2005/08/08 08:05:34 espie 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/param.h>
34 #include <sys/types.h>
35 #include <sys/exec.h>
36 #include <sys/sysctl.h>
37
38 #include <stdio.h>
39 #include <stdlib.h>
40 #include <string.h>
41 #include <stdarg.h>
42
43 #define MAX_PROCTITLE 2048
44
45 extern char *__progname; /* Program name, from crt0. */
46
47 void
setproctitle(const char * fmt,...)48 setproctitle(const char *fmt, ...)
49 {
50 static struct ps_strings *ps;
51 va_list ap;
52
53 static char buf[MAX_PROCTITLE], *bufp = buf;
54 int used;
55
56 va_start(ap, fmt);
57 if (fmt != NULL) {
58 used = snprintf(buf, MAX_PROCTITLE, "%s: ", __progname);
59 if (used >= MAX_PROCTITLE)
60 used = MAX_PROCTITLE - 1;
61 else if (used < 0)
62 used = 0;
63 (void)vsnprintf(buf + used, MAX_PROCTITLE - used, fmt, ap);
64 } else
65 (void)snprintf(buf, MAX_PROCTITLE, "%s", __progname);
66 va_end(ap);
67
68 if (ps == NULL) {
69 struct _ps_strings _ps;
70 int mib[2];
71 size_t len;
72
73 mib[0] = CTL_VM;
74 mib[1] = VM_PSSTRINGS;
75 len = sizeof(_ps);
76 if (sysctl(mib, 2, &_ps, &len, NULL, 0) != 0)
77 return;
78 ps = (struct ps_strings *)_ps.val;
79 }
80 ps->ps_nargvstr = 1;
81 ps->ps_argvstr = &bufp;
82 }
83