[openssh.git] / misc.c

/*	$OpenBSD: misc.c,v 1.11 2001/06/16 08:58:34 markus Exp $	*/

/*
 * 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.11 2001/06/16 08:58:34 markus Exp $");

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

char *
chop(char *s)
{
	char *t = s;
	while (*t) {
		if(*t == '\n' || *t == '\r') {
			*t = '\0';
			return s;
		}
		t++;
	}
	return s;

}

void
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;
	}
	if (val & O_NONBLOCK) {
		debug2("fd %d is O_NONBLOCK", fd);
		return;
	}
	debug("fd %d setting O_NONBLOCK", fd);
	val |= O_NONBLOCK;
	if (fcntl(fd, F_SETFL, val) == -1)
		if (errno != ENODEV)
			error("fcntl(%d, F_SETFL, O_NONBLOCK): %s",
			    fd, strerror(errno));
}

void
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;
	}
	if (!(val & O_NONBLOCK)) {
		debug2("fd %d is not O_NONBLOCK", fd);
		return;
	}
	debug("fd %d clearing O_NONBLOCK", fd);
	val &= ~O_NONBLOCK;
	if (fcntl(fd, F_SETFL, val) == -1)
		if (errno != ENODEV)
			error("fcntl(%d, F_SETFL, O_NONBLOCK): %s",
			    fd, strerror(errno));
}

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

char *
strdelim(char **s)
{
	char *old;
	int wspace = 0;

	if (*s == NULL)
		return NULL;

	old = *s;

	*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;
	copy->pw_expire = pw->pw_expire;
	copy->pw_change = pw->pw_change;
#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;
}

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)

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);
}

void
addargs(arglist *args, char *fmt, ...)
{
	va_list ap;
	char buf[1024];

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

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

	args->list = xrealloc(args->list, args->nalloc * sizeof(char *));
	args->list[args->num++] = xstrdup(buf);
	args->list[args->num] = NULL;
}

mysig_t
mysignal(int sig, mysig_t act)
{
#ifdef HAVE_SIGACTION
	struct sigaction sa, osa;

	if (sigaction(sig, NULL, &osa) == -1)
		return (mysig_t) -1;
	if (osa.sa_handler != act) {
		memset(&sa, 0, sizeof(sa));
		sigemptyset(&sa.sa_mask);
		sa.sa_flags = 0;
#if defined(SA_INTERRUPT)
		if (sig == SIGALRM)
			sa.sa_flags |= SA_INTERRUPT;
#endif
		sa.sa_handler = act;
		if (sigaction(sig, &sa, NULL) == -1)
			return (mysig_t) -1;
	}
	return (osa.sa_handler);
#else
	return (signal(sig, act));
#endif
}
Commit	Line	Data
c4d49b85	1	/* $OpenBSD: misc.c,v 1.11 2001/06/16 08:58:34 markus Exp $ */
bcbf86ec	2
	3	/*
	4	* Copyright (c) 2000 Markus Friedl. 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	*
	15	* THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
	16	* IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
	17	* OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
	18	* IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
	19	* INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
	20	* NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
	21	* DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
	22	* THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
	23	* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
	24	* THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
	25	*/
	26
6d24277f	27	#include "includes.h"
c4d49b85	28	RCSID("$OpenBSD: misc.c,v 1.11 2001/06/16 08:58:34 markus Exp $");
6d24277f	29
1aa00dcb	30	#include "misc.h"
42f11eb2	31	#include "log.h"
4ab21f86	32	#include "xmalloc.h"
6d24277f	33
	34	char *
	35	chop(char *s)
	36	{
	37	char *t = s;
	38	while (*t) {
	39	if(t == '\n' \|\| t == '\r') {
	40	*t = '\0';
	41	return s;
	42	}
	43	t++;
	44	}
	45	return s;
	46
	47	}
	48
	49	void
	50	set_nonblock(int fd)
	51	{
	52	int val;
89aa792b	53
6d24277f	54	val = fcntl(fd, F_GETFL, 0);
	55	if (val < 0) {
	56	error("fcntl(%d, F_GETFL, 0): %s", fd, strerror(errno));
	57	return;
	58	}
a22aff1f	59	if (val & O_NONBLOCK) {
89aa792b	60	debug2("fd %d is O_NONBLOCK", fd);
6d24277f	61	return;
a22aff1f	62	}
6d24277f	63	debug("fd %d setting O_NONBLOCK", fd);
	64	val \|= O_NONBLOCK;
	65	if (fcntl(fd, F_SETFL, val) == -1)
14a9a859	66	if (errno != ENODEV)
	67	error("fcntl(%d, F_SETFL, O_NONBLOCK): %s",
	68	fd, strerror(errno));
6d24277f	69	}
6d24277f	70
89aa792b	71	void
	72	unset_nonblock(int fd)
	73	{
	74	int val;
	75
	76	val = fcntl(fd, F_GETFL, 0);
	77	if (val < 0) {
	78	error("fcntl(%d, F_GETFL, 0): %s", fd, strerror(errno));
	79	return;
	80	}
	81	if (!(val & O_NONBLOCK)) {
	82	debug2("fd %d is not O_NONBLOCK", fd);
	83	return;
	84	}
e04e7a19	85	debug("fd %d clearing O_NONBLOCK", fd);
89aa792b	86	val &= ~O_NONBLOCK;
	87	if (fcntl(fd, F_SETFL, val) == -1)
	88	if (errno != ENODEV)
	89	error("fcntl(%d, F_SETFL, O_NONBLOCK): %s",
	90	fd, strerror(errno));
	91	}
	92
6d24277f	93	/* Characters considered whitespace in strsep calls. */
	94	#define WHITESPACE " \t\r\n"
	95
	96	char *
	97	strdelim(char **s)
	98	{
	99	char *old;
	100	int wspace = 0;
	101
	102	if (*s == NULL)
	103	return NULL;
	104
	105	old = *s;
	106
	107	s = strpbrk(s, WHITESPACE "=");
	108	if (*s == NULL)
	109	return (old);
	110
	111	/* Allow only one '=' to be skipped */
	112	if (*s[0] == '=')
	113	wspace = 1;
	114	*s[0] = '\0';
	115
	116	s += strspn(s + 1, WHITESPACE) + 1;
	117	if (*s[0] == '=' && !wspace)
	118	s += strspn(s + 1, WHITESPACE) + 1;
	119
	120	return (old);
	121	}
1aa00dcb	122
3b1a83df	123	struct passwd *
	124	pwcopy(struct passwd *pw)
	125	{
	126	struct passwd copy = xmalloc(sizeof(copy));
5649fbbe	127
3b1a83df	128	memset(copy, 0, sizeof(*copy));
	129	copy->pw_name = xstrdup(pw->pw_name);
	130	copy->pw_passwd = xstrdup(pw->pw_passwd);
5649fbbe	131	copy->pw_gecos = xstrdup(pw->pw_gecos);
3b1a83df	132	copy->pw_uid = pw->pw_uid;
3b1a83df	133	copy->pw_gid = pw->pw_gid;
c4d49b85	134	copy->pw_expire = pw->pw_expire;
c4d49b85	135	copy->pw_change = pw->pw_change;
2605addd	136	#ifdef HAVE_PW_CLASS_IN_PASSWD
3b1a83df	137	copy->pw_class = xstrdup(pw->pw_class);
2605addd	138	#endif
3b1a83df	139	copy->pw_dir = xstrdup(pw->pw_dir);
	140	copy->pw_shell = xstrdup(pw->pw_shell);
	141	return copy;
	142	}
	143
2d2a2c65	144	int a2port(const char *s)
	145	{
	146	long port;
	147	char *endp;
	148
	149	errno = 0;
	150	port = strtol(s, &endp, 0);
	151	if (s == endp \|\| *endp != '\0' \|\|
	152	(errno == ERANGE && (port == LONG_MIN \|\| port == LONG_MAX)) \|\|
	153	port <= 0 \|\| port > 65535)
	154	return 0;
	155
	156	return port;
	157	}
	158
e2b1fb42	159	#define SECONDS 1
	160	#define MINUTES (SECONDS * 60)
	161	#define HOURS (MINUTES * 60)
	162	#define DAYS (HOURS * 24)
	163	#define WEEKS (DAYS * 7)
	164
	165	long convtime(const char *s)
	166	{
	167	long total, secs;
	168	const char *p;
	169	char *endp;
	170
	171	errno = 0;
	172	total = 0;
	173	p = s;
	174
	175	if (p == NULL \|\| *p == '\0')
	176	return -1;
	177
	178	while (*p) {
	179	secs = strtol(p, &endp, 10);
	180	if (p == endp \|\|
	181	(errno == ERANGE && (secs == LONG_MIN \|\| secs == LONG_MAX)) \|\|
	182	secs < 0)
	183	return -1;
	184
	185	switch (*endp++) {
	186	case '\0':
	187	endp--;
	188	case 's':
	189	case 'S':
	190	break;
	191	case 'm':
	192	case 'M':
	193	secs *= MINUTES;
	194	break;
	195	case 'h':
	196	case 'H':
	197	secs *= HOURS;
	198	break;
	199	case 'd':
	200	case 'D':
	201	secs *= DAYS;
	202	break;
	203	case 'w':
	204	case 'W':
	205	secs *= WEEKS;
	206	break;
	207	default:
	208	return -1;
	209	}
	210	total += secs;
	211	if (total < 0)
	212	return -1;
	213	p = endp;
	214	}
	215
	216	return total;
	217	}
	218
1fcde3fe	219	char *
	220	cleanhostname(char *host)
	221	{
	222	if (*host == '[' && host[strlen(host) - 1] == ']') {
	223	host[strlen(host) - 1] = '\0';
	224	return (host + 1);
	225	} else
	226	return host;
	227	}
	228
	229	char *
	230	colon(char *cp)
	231	{
	232	int flag = 0;
	233
	234	if (cp == ':') / Leading colon is part of file name. */
	235	return (0);
	236	if (*cp == '[')
	237	flag = 1;
	238
	239	for (; *cp; ++cp) {
	240	if (cp == '@' && (cp+1) == '[')
	241	flag = 1;
	242	if (cp == ']' && (cp+1) == ':' && flag)
	243	return (cp+1);
	244	if (*cp == ':' && !flag)
	245	return (cp);
	246	if (*cp == '/')
	247	return (0);
	248	}
	249	return (0);
	250	}
	251
8a624ebf	252	void
	253	addargs(arglist args, char fmt, ...)
	254	{
	255	va_list ap;
	256	char buf[1024];
	257
	258	va_start(ap, fmt);
	259	vsnprintf(buf, sizeof(buf), fmt, ap);
	260	va_end(ap);
	261
	262	if (args->list == NULL) {
	263	args->nalloc = 32;
	264	args->num = 0;
	265	} else if (args->num+2 >= args->nalloc)
	266	args->nalloc *= 2;
	267
	268	args->list = xrealloc(args->list, args->nalloc * sizeof(char *));
	269	args->list[args->num++] = xstrdup(buf);
	270	args->list[args->num] = NULL;
	271	}
	272
1aa00dcb	273	mysig_t
	274	mysignal(int sig, mysig_t act)
	275	{
	276	#ifdef HAVE_SIGACTION
	277	struct sigaction sa, osa;
	278
120bf679	279	if (sigaction(sig, NULL, &osa) == -1)
1aa00dcb	280	return (mysig_t) -1;
1aa00dcb	281	if (osa.sa_handler != act) {
120bf679	282	memset(&sa, 0, sizeof(sa));
1aa00dcb	283	sigemptyset(&sa.sa_mask);
1aa00dcb	284	sa.sa_flags = 0;
d54d99a3	285	#if defined(SA_INTERRUPT)
d54d99a3	286	if (sig == SIGALRM)
df538d55	287	sa.sa_flags \|= SA_INTERRUPT;
25cd3375	288	#endif
1aa00dcb	289	sa.sa_handler = act;
120bf679	290	if (sigaction(sig, &sa, NULL) == -1)
1aa00dcb	291	return (mysig_t) -1;
	292	}
	293	return (osa.sa_handler);
	294	#else
	295	return (signal(sig, act));
	296	#endif
	297	}