setproctitle.c revision 98937
1/* 2 * Modified for OpenSSH by Kevin Steves 3 * October 2000 4 */ 5 6/* 7 * Copyright (c) 1994, 1995 Christopher G. Demetriou 8 * All rights reserved. 9 * 10 * Redistribution and use in source and binary forms, with or without 11 * modification, are permitted provided that the following conditions 12 * are met: 13 * 1. Redistributions of source code must retain the above copyright 14 * notice, this list of conditions and the following disclaimer. 15 * 2. Redistributions in binary form must reproduce the above copyright 16 * notice, this list of conditions and the following disclaimer in the 17 * documentation and/or other materials provided with the distribution. 18 * 3. All advertising materials mentioning features or use of this software 19 * must display the following acknowledgement: 20 * This product includes software developed by Christopher G. Demetriou 21 * for the NetBSD Project. 22 * 4. The name of the author may not be used to endorse or promote products 23 * derived from this software without specific prior written permission 24 * 25 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR 26 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES 27 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. 28 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, 29 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT 30 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 31 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 32 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 33 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF 34 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 35 */ 36 37#if defined(LIBC_SCCS) && !defined(lint) 38static char rcsid[] = "$OpenBSD: setproctitle.c,v 1.8 2001/11/06 19:21:40 art Exp $"; 39#endif /* LIBC_SCCS and not lint */ 40 41#include "includes.h" 42 43#ifndef HAVE_SETPROCTITLE 44 45#define SPT_NONE 0 46#define SPT_PSTAT 1 47 48#ifndef SPT_TYPE 49#define SPT_TYPE SPT_NONE 50#endif 51 52#if SPT_TYPE == SPT_PSTAT 53#include <sys/param.h> 54#include <sys/pstat.h> 55#endif /* SPT_TYPE == SPT_PSTAT */ 56 57#define MAX_PROCTITLE 2048 58 59extern char *__progname; 60 61/* 62 * Set Process Title (SPT) defines. Modeled after sendmail's 63 * SPT type definition strategy. 64 * 65 * SPT_TYPE: 66 * 67 * SPT_NONE: Don't set the process title. Default. 68 * SPT_PSTAT: Use pstat(PSTAT_SETCMD). HP-UX specific. 69 */ 70 71void 72setproctitle(const char *fmt, ...) 73{ 74#if SPT_TYPE != SPT_NONE 75 va_list ap; 76 77 char buf[MAX_PROCTITLE]; 78 size_t used; 79 80#if SPT_TYPE == SPT_PSTAT 81 union pstun pst; 82#endif /* SPT_TYPE == SPT_PSTAT */ 83 84 va_start(ap, fmt); 85 if (fmt != NULL) { 86 used = snprintf(buf, MAX_PROCTITLE, "%s: ", __progname); 87 if (used >= MAX_PROCTITLE) 88 used = MAX_PROCTITLE - 1; 89 (void)vsnprintf(buf + used, MAX_PROCTITLE - used, fmt, ap); 90 } else 91 (void)snprintf(buf, MAX_PROCTITLE, "%s", __progname); 92 va_end(ap); 93 used = strlen(buf); 94 95#if SPT_TYPE == SPT_PSTAT 96 pst.pst_command = buf; 97 pstat(PSTAT_SETCMD, pst, used, 0, 0); 98#endif /* SPT_TYPE == SPT_PSTAT */ 99 100#endif /* SPT_TYPE != SPT_NONE */ 101} 102#endif /* HAVE_SETPROCTITLE */ 103