[openssh.git] / misc.c

/*	$OpenBSD: misc.c,v 1.3 2001/02/28 05:34:28 deraadt 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.3 2001/02/28 05:34:28 deraadt 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) {
		debug("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));
}

/* 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_uid = pw->pw_uid;
	copy->pw_gid = pw->pw_gid;
	copy->pw_class = xstrdup(pw->pw_class);
	copy->pw_dir = xstrdup(pw->pw_dir);
	copy->pw_shell = xstrdup(pw->pw_shell);
	return copy;
}

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_RESTART)
		if (sig == SIGCHLD)
			sa.sa_flags |= SA_RESTART;
#endif
#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
4ab21f86	1	/* $OpenBSD: misc.c,v 1.3 2001/02/28 05:34:28 deraadt 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"
4ab21f86	28	RCSID("$OpenBSD: misc.c,v 1.3 2001/02/28 05:34:28 deraadt 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;
6d24277f	53	val = fcntl(fd, F_GETFL, 0);
	54	if (val < 0) {
	55	error("fcntl(%d, F_GETFL, 0): %s", fd, strerror(errno));
	56	return;
	57	}
a22aff1f	58	if (val & O_NONBLOCK) {
a22aff1f	59	debug("fd %d IS O_NONBLOCK", fd);
6d24277f	60	return;
a22aff1f	61	}
6d24277f	62	debug("fd %d setting O_NONBLOCK", fd);
	63	val \|= O_NONBLOCK;
	64	if (fcntl(fd, F_SETFL, val) == -1)
14a9a859	65	if (errno != ENODEV)
	66	error("fcntl(%d, F_SETFL, O_NONBLOCK): %s",
	67	fd, strerror(errno));
6d24277f	68	}
	69
	70	/* Characters considered whitespace in strsep calls. */
	71	#define WHITESPACE " \t\r\n"
	72
	73	char *
	74	strdelim(char **s)
	75	{
	76	char *old;
	77	int wspace = 0;
	78
	79	if (*s == NULL)
	80	return NULL;
	81
	82	old = *s;
	83
	84	s = strpbrk(s, WHITESPACE "=");
	85	if (*s == NULL)
	86	return (old);
	87
	88	/* Allow only one '=' to be skipped */
	89	if (*s[0] == '=')
	90	wspace = 1;
	91	*s[0] = '\0';
	92
	93	s += strspn(s + 1, WHITESPACE) + 1;
	94	if (*s[0] == '=' && !wspace)
	95	s += strspn(s + 1, WHITESPACE) + 1;
	96
	97	return (old);
	98	}
1aa00dcb	99
3b1a83df	100	struct passwd *
	101	pwcopy(struct passwd *pw)
	102	{
	103	struct passwd copy = xmalloc(sizeof(copy));
	104	memset(copy, 0, sizeof(*copy));
	105	copy->pw_name = xstrdup(pw->pw_name);
	106	copy->pw_passwd = xstrdup(pw->pw_passwd);
	107	copy->pw_uid = pw->pw_uid;
	108	copy->pw_gid = pw->pw_gid;
	109	copy->pw_class = xstrdup(pw->pw_class);
	110	copy->pw_dir = xstrdup(pw->pw_dir);
	111	copy->pw_shell = xstrdup(pw->pw_shell);
	112	return copy;
	113	}
	114
1aa00dcb	115	mysig_t
	116	mysignal(int sig, mysig_t act)
	117	{
	118	#ifdef HAVE_SIGACTION
	119	struct sigaction sa, osa;
	120
120bf679	121	if (sigaction(sig, NULL, &osa) == -1)
1aa00dcb	122	return (mysig_t) -1;
1aa00dcb	123	if (osa.sa_handler != act) {
120bf679	124	memset(&sa, 0, sizeof(sa));
1aa00dcb	125	sigemptyset(&sa.sa_mask);
1aa00dcb	126	sa.sa_flags = 0;
dfef7e7e	127	#if defined(SA_RESTART)
d54d99a3	128	if (sig == SIGCHLD)
9535dddf	129	sa.sa_flags \|= SA_RESTART;
d54d99a3	130	#endif
	131	#if defined(SA_INTERRUPT)
	132	if (sig == SIGALRM)
df538d55	133	sa.sa_flags \|= SA_INTERRUPT;
25cd3375	134	#endif
1aa00dcb	135	sa.sa_handler = act;
120bf679	136	if (sigaction(sig, &sa, NULL) == -1)
1aa00dcb	137	return (mysig_t) -1;
	138	}
	139	return (osa.sa_handler);
	140	#else
	141	return (signal(sig, act));
	142	#endif
	143	}