libutempter - zygaena/hammerhead

root / base / usr / src / lib / libutempter
libutempter Plain Text 324 lines 9.2 KB
#
# This file and its contents are supplied under the terms of the
# Common Development and Distribution License ("CDDL"), version 1.0.
# You may only use this file in accordance with the terms of version
# 1.0 of the CDDL.
#
# A full copy of the text of the CDDL should have accompanied this
# source.  A copy of the CDDL is also available via the Internet at
# http://www.illumos.org/license/CDDL.
#

include		$(SRC)/lib/Makefile.lib

SUBDIRS = $(MACH64)

all:		TARGET= all
clean:		TARGET= clean
clobber:	TARGET= clobber
install:	TARGET= install

HDRDIR=		common
HDRS=		utempter.h

.KEEP_STATE:

all clean clobber install: $(SUBDIRS)

check:		$(CHECKHDRS)

install_h:	$(ROOTHDRS)

$(SUBDIRS):	FRC
		@cd $@; pwd; $(MAKE) $(TARGET)

FRC:

include		$(SRC)/lib/Makefile.targ
#
# This file and its contents are supplied under the terms of the
# Common Development and Distribution License ("CDDL"), version 1.0.
# You may only use this file in accordance with the terms of version
# 1.0 of the CDDL.
#
# A full copy of the text of the CDDL should have accompanied this
# source.  A copy of the CDDL is also available via the Internet at
# http://www.illumos.org/license/CDDL.
#

LIBRARY=	libutempter.a
VERS=		.1

OBJECTS=	utempter.o

include		$(SRC)/lib/Makefile.lib

SRCDIR=		../common

LDLIBS +=	-lc
LIBS=		$(DYNLIB)

CSTD=		$(CSTD_GNU99)

.KEEP_STATE:

all:		$(LIBS)

include		$(SRC)/lib/Makefile.targ
Copyright (c) 2009 Ed Schouten <ed@FreeBSD.org>
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 AND CONTRIBUTORS ``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 OR CONTRIBUTORS 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.
UTEMPTER LIBRARY
#
# This file and its contents are supplied under the terms of the
# Common Development and Distribution License ("CDDL"), version 1.0.
# You may only use this file in accordance with the terms of version
# 1.0 of the CDDL.
#
# A full copy of the text of the CDDL should have accompanied this
# source.  A copy of the CDDL is also available via the Internet at
# http://www.illumos.org/license/CDDL.
#

include		../Makefile.com
include		../../Makefile.lib.64

install:	all $(ROOTLIBS64) $(ROOTLINKS64)
#
# This file and its contents are supplied under the terms of the
# Common Development and Distribution License ("CDDL"), version 1.0.
# You may only use this file in accordance with the terms of version
# 1.0 of the CDDL.
#
# A full copy of the text of the CDDL should have accompanied this
# source.  A copy of the CDDL is also available via the Internet at
# http://www.illumos.org/license/CDDL.
#

#
# MAPFILE HEADER START
#
# WARNING:  STOP NOW.  DO NOT MODIFY THIS FILE.
# Object versioning must comply with the rules detailed in
#
#	usr/src/lib/README.mapfiles
#
# You should not be making modifications here until you've read the most current
# copy of that file. If you need help, contact a gatekeeper for guidance.
#
# MAPFILE HEADER END
#

$mapfile_version 2

SYMBOL_VERSION ILLUMOS_0.1 {
    global:
	addToUtmp;
	removeFromUtmp;
	removeLineFromUtmp;
	utempter_add_record;
	utempter_remove_added_record;
	utempter_remove_record;
    local:
	*;
};
/*
 * SPDX-License-Identifier: BSD-2-Clause-FreeBSD
 *
 * Copyright (c) 2009 Ed Schouten <ed@FreeBSD.org>
 * 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 AND CONTRIBUTORS ``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 OR CONTRIBUTORS 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 <sys/types.h>

#include <pwd.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <utmpx.h>

static int last_fd = -1;

static void
utempter_update_utmpx(int type, int fd, const char *host)
{
	struct utmpx ut;
	struct passwd *pw;
	uid_t uid;

	(void) memset(&ut, 0, sizeof (ut));
	ut.ut_type = type;
	ut.ut_pid = getpid();
	ut.ut_session = getsid(0);
	(void) gettimeofday(&ut.ut_tv, NULL);
	if (snprintf(ut.ut_id, sizeof (ut.ut_id), "f%d", fd) >=
	    sizeof (ut.ut_id))
		return;
	uid = getuid();
	if ((pw = getpwuid(uid)) == NULL)
		return;
	(void) strlcpy(ut.ut_user, pw->pw_name, sizeof (ut.ut_user));

	if (type == DEAD_PROCESS) {
		struct utmpx *ut1;
		struct utmpx ut2;

		(void) memset(&ut2, 0, sizeof (ut2));
		ut2.ut_type = USER_PROCESS;
		if (snprintf(ut2.ut_id, sizeof (ut2.ut_id), "f%d", fd) >=
		    sizeof (ut2.ut_id))
			return;
		if ((ut1 = getutxid(&ut2)) == NULL)
			return;
		(void) strlcpy(ut.ut_line, ut1->ut_line, sizeof (ut.ut_line));
	} else {
		char *line = ptsname(fd);
		if (line == NULL)
			return;
		(void) strlcpy(ut.ut_line, line + strlen("/dev/"),
		    sizeof (ut.ut_line));
	}

	if (host != NULL) {
		(void) strlcpy(ut.ut_host, host, sizeof (ut.ut_host));
		ut.ut_syslen = strlen(ut.ut_host) + 1;
	}

	setutxent();
	(void) pututxline(&ut);
	endutxent();
}

int
utempter_add_record(int fd, const char *host)
{
	utempter_update_utmpx(USER_PROCESS, fd, host);
	last_fd = fd;
	return (0);
}

int
utempter_remove_record(int fd)
{
	utempter_update_utmpx(DEAD_PROCESS, fd, NULL);
	if (last_fd == fd)
		last_fd = -1;
	return (0);
}

int
utempter_remove_added_record(void)
{
	if (last_fd < 0)
		return (0);
	utempter_update_utmpx(DEAD_PROCESS, last_fd, NULL);
	last_fd = -1;
	return (0);
}

void
addToUtmp(const char *pty __unused, const char *host, int fd)
{
	(void) utempter_add_record(fd, host);
}

void
removeFromUtmp(void)
{
	(void) utempter_remove_added_record();
}

void
removeLineFromUtmp(const char *pty __unused, int fd)
{
	(void) utempter_remove_record(fd);
}
/*
 * SPDX-License-Identifier: BSD-2-Clause-FreeBSD
 *
 * Copyright (c) 2009 Ed Schouten <ed@FreeBSD.org>
 * 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 AND CONTRIBUTORS ``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 OR CONTRIBUTORS 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.
 */

#ifndef _UTEMPTER_H
#define	_UTEMPTER_H

#ifdef	__cplusplus
extern "C" {
#endif

int	utempter_add_record(int, const char *);
int	utempter_remove_added_record(void);
int	utempter_remove_record(int);
void	addToUtmp(const char *, const char *, int);
void	removeFromUtmp(void);
void	removeLineFromUtmp(const char *, int);

#ifdef	__cplusplus
}
#endif

#endif /* !_UTEMPTER_H */