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