blob: 2b15c6e00f8b9d36a4e0596f2a18099bb8bb23b0 [file] [log] [blame]
Damien Miller35276252003-06-03 10:14:28 +10001/* Based on conf.c from UCB sendmail 8.8.8 */
Kevin Steves8848b242000-10-18 13:11:44 +00002
Damien Miller35276252003-06-03 10:14:28 +10003/*
4 * Copyright 2003 Damien Miller
5 * Copyright (c) 1983, 1995-1997 Eric P. Allman
6 * Copyright (c) 1988, 1993
7 * The Regents of the University of California. All rights reserved.
Kevin Steves8848b242000-10-18 13:11:44 +00008 *
Damien Miller35276252003-06-03 10:14:28 +10009 * Redistribution and use in source and binary forms, with or without
10 * modification, are permitted provided that the following conditions
11 * are met:
12 * 1. Redistributions of source code must retain the above copyright
13 * notice, this list of conditions and the following disclaimer.
14 * 2. Redistributions in binary form must reproduce the above copyright
15 * notice, this list of conditions and the following disclaimer in the
16 * documentation and/or other materials provided with the distribution.
Damien Miller4636d932003-06-03 11:23:32 +100017 * 3. Neither the name of the University nor the names of its contributors
Damien Miller35276252003-06-03 10:14:28 +100018 * may be used to endorse or promote products derived from this software
19 * without specific prior written permission.
Kevin Steves8848b242000-10-18 13:11:44 +000020 *
Damien Miller35276252003-06-03 10:14:28 +100021 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
22 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
23 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
24 * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
25 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
26 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
27 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
28 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
29 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
30 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
31 * SUCH DAMAGE.
Kevin Steves8848b242000-10-18 13:11:44 +000032 */
33
Kevin Steves8848b242000-10-18 13:11:44 +000034#include "includes.h"
35
Ben Lindstrom49a79c02000-11-17 03:47:20 +000036#ifndef HAVE_SETPROCTITLE
37
Darren Tucker767e4132006-07-12 22:43:28 +100038#include <stdarg.h>
Darren Tuckere6b641a2006-08-17 18:55:27 +100039#include <stdlib.h>
Damien Millera8ed44b2003-01-10 09:53:12 +110040#include <unistd.h>
41#ifdef HAVE_SYS_PSTAT_H
Damien Miller35276252003-06-03 10:14:28 +100042#include <sys/pstat.h>
Kevin Steves8848b242000-10-18 13:11:44 +000043#endif
Damien Millerb8fe89c2006-07-24 14:51:00 +100044#include <string.h>
Kevin Steves8848b242000-10-18 13:11:44 +000045
Damien Millerc5750222008-05-16 10:01:54 +100046#include <vis.h>
47
Damien Miller35276252003-06-03 10:14:28 +100048#define SPT_NONE 0 /* don't use it at all */
Damien Miller2e45cb02004-02-20 20:37:44 +110049#define SPT_PSTAT 1 /* use pstat(PSTAT_SETCMD, ...) */
50#define SPT_REUSEARGV 2 /* cover argv with title information */
Damien Millerec201962003-01-13 10:04:58 +110051
Damien Miller35276252003-06-03 10:14:28 +100052#ifndef SPT_TYPE
53# define SPT_TYPE SPT_NONE
Damien Millera8ed44b2003-01-10 09:53:12 +110054#endif
55
Damien Miller35276252003-06-03 10:14:28 +100056#ifndef SPT_PADCHAR
57# define SPT_PADCHAR '\0'
Damien Millerec201962003-01-13 10:04:58 +110058#endif
Damien Millera8ed44b2003-01-10 09:53:12 +110059
Damien Miller35276252003-06-03 10:14:28 +100060#if SPT_TYPE == SPT_REUSEARGV
61static char *argv_start = NULL;
62static size_t argv_env_len = 0;
Damien Millera8ed44b2003-01-10 09:53:12 +110063#endif
Damien Millera8ed44b2003-01-10 09:53:12 +110064
Kevin Steves8848b242000-10-18 13:11:44 +000065#endif /* HAVE_SETPROCTITLE */
Damien Millera8ed44b2003-01-10 09:53:12 +110066
Damien Millera8ed44b2003-01-10 09:53:12 +110067void
68compat_init_setproctitle(int argc, char *argv[])
69{
Damien Milleree878382014-01-22 16:30:15 +110070#if !defined(HAVE_SETPROCTITLE) && \
71 defined(SPT_TYPE) && SPT_TYPE == SPT_REUSEARGV
Damien Miller35276252003-06-03 10:14:28 +100072 extern char **environ;
73 char *lastargv = NULL;
74 char **envp = environ;
Damien Millera8ed44b2003-01-10 09:53:12 +110075 int i;
Damien Millera8ed44b2003-01-10 09:53:12 +110076
77 /*
Damien Miller35276252003-06-03 10:14:28 +100078 * NB: This assumes that argv has already been copied out of the
Darren Tuckerc20dccb2016-08-02 09:44:25 +100079 * way. This is true for sshd, but may not be true for other
Damien Miller35276252003-06-03 10:14:28 +100080 * programs. Beware.
Damien Millera8ed44b2003-01-10 09:53:12 +110081 */
Damien Millera8ed44b2003-01-10 09:53:12 +110082
Damien Miller35276252003-06-03 10:14:28 +100083 if (argc == 0 || argv[0] == NULL)
84 return;
85
86 /* Fail if we can't allocate room for the new environment */
87 for (i = 0; envp[i] != NULL; i++)
88 ;
Darren Tuckerd8093e42006-05-04 16:24:34 +100089 if ((environ = calloc(i + 1, sizeof(*environ))) == NULL) {
Damien Miller35276252003-06-03 10:14:28 +100090 environ = envp; /* put it back */
Damien Millera8ed44b2003-01-10 09:53:12 +110091 return;
92 }
93
94 /*
Darren Tuckerc20dccb2016-08-02 09:44:25 +100095 * Find the last argv string or environment variable within
Damien Miller35276252003-06-03 10:14:28 +100096 * our process memory area.
Damien Millera8ed44b2003-01-10 09:53:12 +110097 */
Damien Miller35276252003-06-03 10:14:28 +100098 for (i = 0; i < argc; i++) {
99 if (lastargv == NULL || lastargv + 1 == argv[i])
100 lastargv = argv[i] + strlen(argv[i]);
101 }
102 for (i = 0; envp[i] != NULL; i++) {
103 if (lastargv + 1 == envp[i])
104 lastargv = envp[i] + strlen(envp[i]);
Damien Millera8ed44b2003-01-10 09:53:12 +1100105 }
106
Damien Miller35276252003-06-03 10:14:28 +1000107 argv[1] = NULL;
108 argv_start = argv[0];
109 argv_env_len = lastargv - argv[0] - 1;
Damien Millera8ed44b2003-01-10 09:53:12 +1100110
Darren Tuckerc20dccb2016-08-02 09:44:25 +1000111 /*
112 * Copy environment
Damien Miller35276252003-06-03 10:14:28 +1000113 * XXX - will truncate env on strdup fail
Damien Millera8ed44b2003-01-10 09:53:12 +1100114 */
Damien Miller35276252003-06-03 10:14:28 +1000115 for (i = 0; envp[i] != NULL; i++)
116 environ[i] = strdup(envp[i]);
117 environ[i] = NULL;
118#endif /* SPT_REUSEARGV */
Damien Millera8ed44b2003-01-10 09:53:12 +1100119}
120
Damien Miller35276252003-06-03 10:14:28 +1000121#ifndef HAVE_SETPROCTITLE
122void
123setproctitle(const char *fmt, ...)
124{
125#if SPT_TYPE != SPT_NONE
126 va_list ap;
Damien Millerc5750222008-05-16 10:01:54 +1000127 char buf[1024], ptitle[1024];
Damien Miller35276252003-06-03 10:14:28 +1000128 size_t len;
Darren Tucker710f3742013-11-03 17:20:34 +1100129 int r;
Damien Miller35276252003-06-03 10:14:28 +1000130 extern char *__progname;
131#if SPT_TYPE == SPT_PSTAT
132 union pstun pst;
133#endif
134
135#if SPT_TYPE == SPT_REUSEARGV
136 if (argv_env_len <= 0)
137 return;
138#endif
139
140 strlcpy(buf, __progname, sizeof(buf));
141
Darren Tucker710f3742013-11-03 17:20:34 +1100142 r = -1;
Damien Miller35276252003-06-03 10:14:28 +1000143 va_start(ap, fmt);
144 if (fmt != NULL) {
145 len = strlcat(buf, ": ", sizeof(buf));
146 if (len < sizeof(buf))
Darren Tucker710f3742013-11-03 17:20:34 +1100147 r = vsnprintf(buf + len, sizeof(buf) - len , fmt, ap);
Damien Miller35276252003-06-03 10:14:28 +1000148 }
149 va_end(ap);
Darren Tucker710f3742013-11-03 17:20:34 +1100150 if (r == -1 || (size_t)r >= sizeof(buf) - len)
151 return;
Damien Millerc5750222008-05-16 10:01:54 +1000152 strnvis(ptitle, buf, sizeof(ptitle),
153 VIS_CSTYLE|VIS_NL|VIS_TAB|VIS_OCTAL);
Damien Miller35276252003-06-03 10:14:28 +1000154
155#if SPT_TYPE == SPT_PSTAT
Damien Millerc5750222008-05-16 10:01:54 +1000156 pst.pst_command = ptitle;
157 pstat(PSTAT_SETCMD, pst, strlen(ptitle), 0, 0);
Damien Miller35276252003-06-03 10:14:28 +1000158#elif SPT_TYPE == SPT_REUSEARGV
Darren Tuckerc20dccb2016-08-02 09:44:25 +1000159/* debug("setproctitle: copy \"%s\" into len %d",
Damien Miller35276252003-06-03 10:14:28 +1000160 buf, argv_env_len); */
Damien Millerc5750222008-05-16 10:01:54 +1000161 len = strlcpy(argv_start, ptitle, argv_env_len);
Damien Miller35276252003-06-03 10:14:28 +1000162 for(; len < argv_env_len; len++)
163 argv_start[len] = SPT_PADCHAR;
164#endif
165
166#endif /* SPT_NONE */
167}
168
169#endif /* HAVE_SETPROCTITLE */