[gssapi-openssh.git] / openssh / misc.c

/*
 * Copyright (c) 2000 Markus Friedl.  All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 *
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

#include "includes.h"
RCSID("$OpenBSD: misc.c,v 1.25 2004/08/11 21:43:05 avsm Exp $");

#include "misc.h"
#include "log.h"
#include "xmalloc.h"

/* remove newline at end of string */
char *
chop(char *s)
{
	char *t = s;
	while (*t) {
		if (*t == '\n' || *t == '\r') {
			*t = '\0';
			return s;
		}
		t++;
	}
	return s;

}

/* set/unset filedescriptor to non-blocking */
int
set_nonblock(int fd)
{
	int val;

	val = fcntl(fd, F_GETFL, 0);
	if (val < 0) {
		error("fcntl(%d, F_GETFL, 0): %s", fd, strerror(errno));
		return (-1);
	}
	if (val & O_NONBLOCK) {
		debug3("fd %d is O_NONBLOCK", fd);
		return (0);
	}
	debug2("fd %d setting O_NONBLOCK", fd);
	val |= O_NONBLOCK;
	if (fcntl(fd, F_SETFL, val) == -1) {
		debug("fcntl(%d, F_SETFL, O_NONBLOCK): %s", fd,
		    strerror(errno));
		return (-1);
	}
	return (0);
}

int
unset_nonblock(int fd)
{
	int val;

	val = fcntl(fd, F_GETFL, 0);
	if (val < 0) {
		error("fcntl(%d, F_GETFL, 0): %s", fd, strerror(errno));
		return (-1);
	}
	if (!(val & O_NONBLOCK)) {
		debug3("fd %d is not O_NONBLOCK", fd);
		return (0);
	}
	debug("fd %d clearing O_NONBLOCK", fd);
	val &= ~O_NONBLOCK;
	if (fcntl(fd, F_SETFL, val) == -1) {
		debug("fcntl(%d, F_SETFL, ~O_NONBLOCK): %s",
		    fd, strerror(errno));
		return (-1);
	}
	return (0);
}

/* disable nagle on socket */
void
set_nodelay(int fd)
{
	int opt;
	socklen_t optlen;

	optlen = sizeof opt;
	if (getsockopt(fd, IPPROTO_TCP, TCP_NODELAY, &opt, &optlen) == -1) {
		debug("getsockopt TCP_NODELAY: %.100s", strerror(errno));
		return;
	}
	if (opt == 1) {
		debug2("fd %d is TCP_NODELAY", fd);
		return;
	}
	opt = 1;
	debug2("fd %d setting TCP_NODELAY", fd);
	if (setsockopt(fd, IPPROTO_TCP, TCP_NODELAY, &opt, sizeof opt) == -1)
		error("setsockopt TCP_NODELAY: %.100s", strerror(errno));
}

/* Characters considered whitespace in strsep calls. */
#define WHITESPACE " \t\r\n"

/* Characters considered as quotations. */
#define QUOTES "'\""

/* return next token in configuration line */
char *
strdelim(char **s)
{
	char *old, *p, *q;
	int wspace = 0;

	if (*s == NULL)
		return NULL;

	old = *s;

        if ((q=strchr(QUOTES, (int) *old)) && *q)
        {
            /* find next quote character, point old to start of quoted
             * string */
            for (p = ++old;*p && *p!=*q; p++)
                 ;
            
            /* find start of next token */
            *s = (*p) ? p + strspn(p + 1, WHITESPACE) + 1 : NULL;
            
            /* terminate 'old' token */
            *p = '\0';
            return (old);
        }

        *s = strpbrk(*s, WHITESPACE "=");
	if (*s == NULL)
		return (old);

	/* Allow only one '=' to be skipped */
	if (*s[0] == '=')
		wspace = 1;
	*s[0] = '\0';

	*s += strspn(*s + 1, WHITESPACE) + 1;
	if (*s[0] == '=' && !wspace)
		*s += strspn(*s + 1, WHITESPACE) + 1;

	return (old);
}

struct passwd *
pwcopy(struct passwd *pw)
{
	struct passwd *copy = xmalloc(sizeof(*copy));

	memset(copy, 0, sizeof(*copy));
	copy->pw_name = xstrdup(pw->pw_name);
	copy->pw_passwd = xstrdup(pw->pw_passwd);
	copy->pw_gecos = xstrdup(pw->pw_gecos);
	copy->pw_uid = pw->pw_uid;
	copy->pw_gid = pw->pw_gid;
#ifdef HAVE_PW_EXPIRE_IN_PASSWD
	copy->pw_expire = pw->pw_expire;
#endif
#ifdef HAVE_PW_CHANGE_IN_PASSWD
	copy->pw_change = pw->pw_change;
#endif
#ifdef HAVE_PW_CLASS_IN_PASSWD
	copy->pw_class = xstrdup(pw->pw_class);
#endif
	copy->pw_dir = xstrdup(pw->pw_dir);
	copy->pw_shell = xstrdup(pw->pw_shell);
	return copy;
}

/*
 * Convert ASCII string to TCP/IP port number.
 * Port must be >0 and <=65535.
 * Return 0 if invalid.
 */
int
a2port(const char *s)
{
	long port;
	char *endp;

	errno = 0;
	port = strtol(s, &endp, 0);
	if (s == endp || *endp != '\0' ||
	    (errno == ERANGE && (port == LONG_MIN || port == LONG_MAX)) ||
	    port <= 0 || port > 65535)
		return 0;

	return port;
}

#define SECONDS		1
#define MINUTES		(SECONDS * 60)
#define HOURS		(MINUTES * 60)
#define DAYS		(HOURS * 24)
#define WEEKS		(DAYS * 7)

/*
 * Convert a time string into seconds; format is
 * a sequence of:
 *      time[qualifier]
 *
 * Valid time qualifiers are:
 *      <none>  seconds
 *      s|S     seconds
 *      m|M     minutes
 *      h|H     hours
 *      d|D     days
 *      w|W     weeks
 *
 * Examples:
 *      90m     90 minutes
 *      1h30m   90 minutes
 *      2d      2 days
 *      1w      1 week
 *
 * Return -1 if time string is invalid.
 */
long
convtime(const char *s)
{
	long total, secs;
	const char *p;
	char *endp;

	errno = 0;
	total = 0;
	p = s;

	if (p == NULL || *p == '\0')
		return -1;

	while (*p) {
		secs = strtol(p, &endp, 10);
		if (p == endp ||
		    (errno == ERANGE && (secs == LONG_MIN || secs == LONG_MAX)) ||
		    secs < 0)
			return -1;

		switch (*endp++) {
		case '\0':
			endp--;
		case 's':
		case 'S':
			break;
		case 'm':
		case 'M':
			secs *= MINUTES;
			break;
		case 'h':
		case 'H':
			secs *= HOURS;
			break;
		case 'd':
		case 'D':
			secs *= DAYS;
			break;
		case 'w':
		case 'W':
			secs *= WEEKS;
			break;
		default:
			return -1;
		}
		total += secs;
		if (total < 0)
			return -1;
		p = endp;
	}

	return total;
}

char *
cleanhostname(char *host)
{
	if (*host == '[' && host[strlen(host) - 1] == ']') {
		host[strlen(host) - 1] = '\0';
		return (host + 1);
	} else
		return host;
}

char *
colon(char *cp)
{
	int flag = 0;

	if (*cp == ':')		/* Leading colon is part of file name. */
		return (0);
	if (*cp == '[')
		flag = 1;

	for (; *cp; ++cp) {
		if (*cp == '@' && *(cp+1) == '[')
			flag = 1;
		if (*cp == ']' && *(cp+1) == ':' && flag)
			return (cp+1);
		if (*cp == ':' && !flag)
			return (cp);
		if (*cp == '/')
			return (0);
	}
	return (0);
}

/* function to assist building execv() arguments */
void
addargs(arglist *args, char *fmt, ...)
{
	va_list ap;
	char buf[1024];
	u_int nalloc;

	va_start(ap, fmt);
	vsnprintf(buf, sizeof(buf), fmt, ap);
	va_end(ap);

	nalloc = args->nalloc;
	if (args->list == NULL) {
		nalloc = 32;
		args->num = 0;
	} else if (args->num+2 >= nalloc)
		nalloc *= 2;

	args->list = xrealloc(args->list, nalloc * sizeof(char *));
	args->nalloc = nalloc;
	args->list[args->num++] = xstrdup(buf);
	args->list[args->num] = NULL;
}
Commit	Line	Data
3c0ef626	1	/*
	2	* Copyright (c) 2000 Markus Friedl. All rights reserved.
	3	*
	4	* Redistribution and use in source and binary forms, with or without
	5	* modification, are permitted provided that the following conditions
	6	* are met:
	7	* 1. Redistributions of source code must retain the above copyright
	8	* notice, this list of conditions and the following disclaimer.
	9	* 2. Redistributions in binary form must reproduce the above copyright
	10	* notice, this list of conditions and the following disclaimer in the
	11	* documentation and/or other materials provided with the distribution.
	12	*
	13	* THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
	14	* IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
	15	* OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
	16	* IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
	17	* INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
	18	* NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
	19	* DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
	20	* THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
	21	* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
	22	* THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
	23	*/
	24
	25	#include "includes.h"
1b56ff3d	26	RCSID("$OpenBSD: misc.c,v 1.25 2004/08/11 21:43:05 avsm Exp $");
3c0ef626	27
	28	#include "misc.h"
	29	#include "log.h"
	30	#include "xmalloc.h"
	31
	32	/* remove newline at end of string */
	33	char *
	34	chop(char *s)
	35	{
	36	char *t = s;
	37	while (*t) {
e9a17296	38	if (t == '\n' \|\| t == '\r') {
3c0ef626	39	*t = '\0';
	40	return s;
	41	}
	42	t++;
	43	}
	44	return s;
	45
	46	}
	47
	48	/* set/unset filedescriptor to non-blocking */
1b56ff3d	49	int
3c0ef626	50	set_nonblock(int fd)
	51	{
	52	int val;
	53
	54	val = fcntl(fd, F_GETFL, 0);
	55	if (val < 0) {
	56	error("fcntl(%d, F_GETFL, 0): %s", fd, strerror(errno));
1b56ff3d	57	return (-1);
3c0ef626	58	}
3c0ef626	59	if (val & O_NONBLOCK) {
1b56ff3d	60	debug3("fd %d is O_NONBLOCK", fd);
1b56ff3d	61	return (0);
3c0ef626	62	}
70791e56	63	debug2("fd %d setting O_NONBLOCK", fd);
3c0ef626	64	val \|= O_NONBLOCK;
1b56ff3d	65	if (fcntl(fd, F_SETFL, val) == -1) {
	66	debug("fcntl(%d, F_SETFL, O_NONBLOCK): %s", fd,
	67	strerror(errno));
	68	return (-1);
	69	}
	70	return (0);
3c0ef626	71	}
3c0ef626	72
1b56ff3d	73	int
3c0ef626	74	unset_nonblock(int fd)
	75	{
	76	int val;
	77
	78	val = fcntl(fd, F_GETFL, 0);
	79	if (val < 0) {
	80	error("fcntl(%d, F_GETFL, 0): %s", fd, strerror(errno));
1b56ff3d	81	return (-1);
3c0ef626	82	}
3c0ef626	83	if (!(val & O_NONBLOCK)) {
1b56ff3d	84	debug3("fd %d is not O_NONBLOCK", fd);
1b56ff3d	85	return (0);
3c0ef626	86	}
	87	debug("fd %d clearing O_NONBLOCK", fd);
	88	val &= ~O_NONBLOCK;
1b56ff3d	89	if (fcntl(fd, F_SETFL, val) == -1) {
1b56ff3d	90	debug("fcntl(%d, F_SETFL, ~O_NONBLOCK): %s",
e9a17296	91	fd, strerror(errno));
1b56ff3d	92	return (-1);
	93	}
	94	return (0);
e9a17296	95	}
	96
	97	/* disable nagle on socket */
	98	void
	99	set_nodelay(int fd)
	100	{
	101	int opt;
	102	socklen_t optlen;
	103
	104	optlen = sizeof opt;
	105	if (getsockopt(fd, IPPROTO_TCP, TCP_NODELAY, &opt, &optlen) == -1) {
416fd2a8	106	debug("getsockopt TCP_NODELAY: %.100s", strerror(errno));
e9a17296	107	return;
	108	}
	109	if (opt == 1) {
	110	debug2("fd %d is TCP_NODELAY", fd);
	111	return;
	112	}
	113	opt = 1;
1c14df9e	114	debug2("fd %d setting TCP_NODELAY", fd);
e9a17296	115	if (setsockopt(fd, IPPROTO_TCP, TCP_NODELAY, &opt, sizeof opt) == -1)
e9a17296	116	error("setsockopt TCP_NODELAY: %.100s", strerror(errno));
3c0ef626	117	}
	118
	119	/* Characters considered whitespace in strsep calls. */
	120	#define WHITESPACE " \t\r\n"
	121
88928908	122	/* Characters considered as quotations. */
	123	#define QUOTES "'\""
	124
3c0ef626	125	/* return next token in configuration line */
	126	char *
	127	strdelim(char **s)
	128	{
88928908	129	char old, p, *q;
3c0ef626	130	int wspace = 0;
	131
	132	if (*s == NULL)
	133	return NULL;
	134
	135	old = *s;
	136
88928908	137	if ((q=strchr(QUOTES, (int) old)) && q)
	138	{
	139	/* find next quote character, point old to start of quoted
	140	* string */
	141	for (p = ++old;p && p!=*q; p++)
	142	;
	143
	144	/* find start of next token */
	145	s = (p) ? p + strspn(p + 1, WHITESPACE) + 1 : NULL;
	146
	147	/* terminate 'old' token */
	148	*p = '\0';
	149	return (old);
	150	}
	151
	152	s = strpbrk(s, WHITESPACE "=");
3c0ef626	153	if (*s == NULL)
	154	return (old);
	155
	156	/* Allow only one '=' to be skipped */
	157	if (*s[0] == '=')
	158	wspace = 1;
	159	*s[0] = '\0';
	160
	161	s += strspn(s + 1, WHITESPACE) + 1;
	162	if (*s[0] == '=' && !wspace)
	163	s += strspn(s + 1, WHITESPACE) + 1;
	164
	165	return (old);
	166	}
	167
	168	struct passwd *
	169	pwcopy(struct passwd *pw)
	170	{
	171	struct passwd copy = xmalloc(sizeof(copy));
	172
	173	memset(copy, 0, sizeof(*copy));
	174	copy->pw_name = xstrdup(pw->pw_name);
	175	copy->pw_passwd = xstrdup(pw->pw_passwd);
	176	copy->pw_gecos = xstrdup(pw->pw_gecos);
	177	copy->pw_uid = pw->pw_uid;
	178	copy->pw_gid = pw->pw_gid;
	179	#ifdef HAVE_PW_EXPIRE_IN_PASSWD
	180	copy->pw_expire = pw->pw_expire;
	181	#endif
	182	#ifdef HAVE_PW_CHANGE_IN_PASSWD
	183	copy->pw_change = pw->pw_change;
	184	#endif
	185	#ifdef HAVE_PW_CLASS_IN_PASSWD
	186	copy->pw_class = xstrdup(pw->pw_class);
	187	#endif
	188	copy->pw_dir = xstrdup(pw->pw_dir);
	189	copy->pw_shell = xstrdup(pw->pw_shell);
	190	return copy;
	191	}
	192
	193	/*
	194	* Convert ASCII string to TCP/IP port number.
	195	* Port must be >0 and <=65535.
	196	* Return 0 if invalid.
	197	*/
	198	int
	199	a2port(const char *s)
	200	{
	201	long port;
	202	char *endp;
	203
	204	errno = 0;
	205	port = strtol(s, &endp, 0);
	206	if (s == endp \|\| *endp != '\0' \|\|
	207	(errno == ERANGE && (port == LONG_MIN \|\| port == LONG_MAX)) \|\|
	208	port <= 0 \|\| port > 65535)
	209	return 0;
	210
	211	return port;
	212	}
	213
	214	#define SECONDS 1
	215	#define MINUTES (SECONDS * 60)
	216	#define HOURS (MINUTES * 60)
217	#define DAYS (HOURS * 24)
218	#define WEEKS (DAYS * 7)
219
220	/*
221	* Convert a time string into seconds; format is
222	* a sequence of:
223	* time[qualifier]
224	*
225	* Valid time qualifiers are:
226	* <none> seconds
227	* s\|S seconds
228	* m\|M minutes
229	* h\|H hours
230	* d\|D days
231	* w\|W weeks
232	*
233	* Examples:
234	* 90m 90 minutes
235	* 1h30m 90 minutes
236	* 2d 2 days
237	* 1w 1 week
238	*
239	* Return -1 if time string is invalid.
240	*/
241	long
242	convtime(const char *s)
243	{
244	long total, secs;
245	const char *p;
246	char *endp;
247
248	errno = 0;
249	total = 0;
250	p = s;
251
252	if (p == NULL \|\| *p == '\0')
253	return -1;
254
255	while (*p) {
256	secs = strtol(p, &endp, 10);
257	if (p == endp \|\|
258	(errno == ERANGE && (secs == LONG_MIN \|\| secs == LONG_MAX)) \|\|
259	secs < 0)
260	return -1;
261
262	switch (*endp++) {
263	case '\0':
264	endp--;
265	case 's':
266	case 'S':
267	break;
268	case 'm':
269	case 'M':
270	secs *= MINUTES;
271	break;
272	case 'h':
273	case 'H':
274	secs *= HOURS;
275	break;
276	case 'd':
277	case 'D':
278	secs *= DAYS;
279	break;
280	case 'w':
281	case 'W':
282	secs *= WEEKS;
283	break;
284	default:
285	return -1;
286	}
287	total += secs;
288	if (total < 0)
289	return -1;
290	p = endp;
291	}
292
293	return total;
294	}
295
296	char *
297	cleanhostname(char *host)
298	{
299	if (*host == '[' && host[strlen(host) - 1] == ']') {
300	host[strlen(host) - 1] = '\0';
301	return (host + 1);
302	} else
303	return host;
304	}
305
306	char *
307	colon(char *cp)
308	{
309	int flag = 0;
310
311	if (cp == ':') / Leading colon is part of file name. */
312	return (0);
313	if (*cp == '[')
314	flag = 1;
315
316	for (; *cp; ++cp) {
317	if (cp == '@' && (cp+1) == '[')
318	flag = 1;
319	if (cp == ']' && (cp+1) == ':' && flag)
320	return (cp+1);
321	if (*cp == ':' && !flag)
322	return (cp);
323	if (*cp == '/')
324	return (0);
325	}
326	return (0);
327	}
328
329	/* function to assist building execv() arguments */
330	void
331	addargs(arglist args, char fmt, ...)
332	{
333	va_list ap;
334	char buf[1024];
1b56ff3d	335	u_int nalloc;
3c0ef626	336
	337	va_start(ap, fmt);
	338	vsnprintf(buf, sizeof(buf), fmt, ap);
	339	va_end(ap);
	340
70791e56	341	nalloc = args->nalloc;
3c0ef626	342	if (args->list == NULL) {
70791e56	343	nalloc = 32;
3c0ef626	344	args->num = 0;
70791e56	345	} else if (args->num+2 >= nalloc)
70791e56	346	nalloc *= 2;
3c0ef626	347
70791e56	348	args->list = xrealloc(args->list, nalloc * sizeof(char *));
70791e56	349	args->nalloc = nalloc;
3c0ef626	350	args->list[args->num++] = xstrdup(buf);
	351	args->list[args->num] = NULL;
	352	}