NetBSD/lib/libc/gen/utmpx.c

493 lines
10 KiB
C

/* $NetBSD: utmpx.c,v 1.21 2003/09/06 16:42:10 wiz Exp $ */
/*-
* Copyright (c) 2002 The NetBSD Foundation, Inc.
* All rights reserved.
*
* This code is derived from software contributed to The NetBSD Foundation
* by Christos Zoulas.
*
* 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.
* 3. All advertising materials mentioning features or use of this software
* must display the following acknowledgement:
* This product includes software developed by the NetBSD
* Foundation, Inc. and its contributors.
* 4. Neither the name of The NetBSD Foundation nor the names of its
* contributors may be used to endorse or promote products derived
* from this software without specific prior written permission.
*
* THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. 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 FOUNDATION 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/cdefs.h>
#if defined(LIBC_SCCS) && !defined(lint)
__RCSID("$NetBSD: utmpx.c,v 1.21 2003/09/06 16:42:10 wiz Exp $");
#endif /* LIBC_SCCS and not lint */
#include "namespace.h"
#include <sys/types.h>
#include <sys/param.h>
#include <sys/socket.h>
#include <sys/stat.h>
#include <sys/time.h>
#include <sys/wait.h>
#include <assert.h>
#include <db.h>
#include <errno.h>
#include <fcntl.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <utmp.h>
/* don't define earlier, has side effects in fcntl.h */
#define __LIBC12_SOURCE__
#include <utmpx.h>
#include <vis.h>
__warn_references(getlastlogx,
"warning: reference to compatibility getlastlogx(); include <utmpx.h> for correct reference")
__warn_references(lastlogxname,
"warning: reference to deprecated lastlogxname()")
static FILE *fp;
static int readonly = 0;
static struct utmpx ut;
static char utfile[MAXPATHLEN] = _PATH_UTMPX;
static char llfile[MAXPATHLEN] = _PATH_LASTLOGX;
static struct utmpx *utmp_update(const struct utmpx *);
static const char vers[] = "utmpx-1.00";
void
setutxent()
{
(void)memset(&ut, 0, sizeof(ut));
if (fp == NULL)
return;
(void)fseeko(fp, (off_t)sizeof(ut), SEEK_SET);
}
void
endutxent()
{
(void)memset(&ut, 0, sizeof(ut));
if (fp != NULL) {
(void)fclose(fp);
fp = NULL;
readonly = 0;
}
}
struct utmpx *
getutxent()
{
if (fp == NULL) {
struct stat st;
if ((fp = fopen(utfile, "r+")) == NULL)
if ((fp = fopen(utfile, "w+")) == NULL) {
if ((fp = fopen(utfile, "r")) == NULL)
goto fail;
else
readonly = 1;
}
/* get file size in order to check if new file */
if (fstat(fileno(fp), &st) == -1)
goto failclose;
if (st.st_size == 0) {
/* new file, add signature record */
(void)memset(&ut, 0, sizeof(ut));
ut.ut_type = SIGNATURE;
(void)memcpy(ut.ut_user, vers, sizeof(vers));
if (fwrite(&ut, sizeof(ut), 1, fp) != 1)
goto failclose;
} else {
/* old file, read signature record */
if (fread(&ut, sizeof(ut), 1, fp) != 1)
goto failclose;
if (memcmp(ut.ut_user, vers, sizeof(vers)) != 0 ||
ut.ut_type != SIGNATURE)
goto failclose;
}
}
if (fread(&ut, sizeof(ut), 1, fp) != 1)
goto fail;
return &ut;
failclose:
(void)fclose(fp);
fail:
(void)memset(&ut, 0, sizeof(ut));
return NULL;
}
struct utmpx *
getutxid(const struct utmpx *utx)
{
_DIAGASSERT(utx != NULL);
if (utx->ut_type == EMPTY)
return NULL;
do {
if (ut.ut_type == EMPTY)
continue;
switch (utx->ut_type) {
case EMPTY:
return NULL;
case RUN_LVL:
case BOOT_TIME:
case OLD_TIME:
case NEW_TIME:
if (ut.ut_type == utx->ut_type)
return &ut;
break;
case INIT_PROCESS:
case LOGIN_PROCESS:
case USER_PROCESS:
case DEAD_PROCESS:
switch (ut.ut_type) {
case INIT_PROCESS:
case LOGIN_PROCESS:
case USER_PROCESS:
case DEAD_PROCESS:
if (memcmp(ut.ut_id, utx->ut_id,
sizeof(ut.ut_id)) == 0)
return &ut;
break;
default:
break;
}
break;
default:
return NULL;
}
} while (getutxent() != NULL);
return NULL;
}
struct utmpx *
getutxline(const struct utmpx *utx)
{
_DIAGASSERT(utx != NULL);
do {
switch (ut.ut_type) {
case EMPTY:
break;
case LOGIN_PROCESS:
case USER_PROCESS:
if (strncmp(ut.ut_line, utx->ut_line,
sizeof(ut.ut_line)) == 0)
return &ut;
break;
default:
break;
}
} while (getutxent() != NULL);
return NULL;
}
struct utmpx *
pututxline(const struct utmpx *utx)
{
struct utmpx temp, *u = NULL;
int gotlock = 0;
_DIAGASSERT(utx != NULL);
if (utx == NULL)
return NULL;
if (strcmp(_PATH_UTMPX, utfile) == 0)
if ((fp != NULL && readonly) || (fp == NULL && geteuid() != 0))
return utmp_update(utx);
(void)memcpy(&temp, utx, sizeof(temp));
if (fp == NULL) {
(void)getutxent();
if (fp == NULL || readonly)
return NULL;
}
if (getutxid(&temp) == NULL) {
setutxent();
if (getutxid(&temp) == NULL) {
if (lockf(fileno(fp), F_LOCK, (off_t)0) == -1)
return NULL;
gotlock++;
if (fseeko(fp, (off_t)0, SEEK_END) == -1)
goto fail;
}
}
if (!gotlock) {
/* we are not appending */
if (fseeko(fp, -(off_t)sizeof(ut), SEEK_CUR) == -1)
return NULL;
}
if (fwrite(&temp, sizeof (temp), 1, fp) != 1)
goto fail;
if (fflush(fp) == -1)
goto fail;
u = memcpy(&ut, &temp, sizeof(ut));
fail:
if (gotlock) {
if (lockf(fileno(fp), F_ULOCK, (off_t)0) == -1)
return NULL;
}
return u;
}
static struct utmpx *
utmp_update(const struct utmpx *utx)
{
char buf[sizeof(*utx) * 4 + 1];
pid_t pid;
int status;
_DIAGASSERT(utx != NULL);
(void)strvisx(buf, (const char *)(const void *)utx, sizeof(*utx),
VIS_WHITE);
switch (pid = fork()) {
case 0:
(void)execl(_PATH_UTMP_UPDATE,
strrchr(_PATH_UTMP_UPDATE, '/') + 1, buf, NULL);
exit(1);
/*NOTREACHED*/
case -1:
return NULL;
default:
if (waitpid(pid, &status, 0) == -1)
return NULL;
if (WIFEXITED(status) && WEXITSTATUS(status) == 0)
return memcpy(&ut, utx, sizeof(ut));
return NULL;
}
}
/*
* The following are extensions and not part of the X/Open spec.
*/
int
updwtmpx(const char *file, const struct utmpx *utx)
{
int fd;
int saved_errno;
_DIAGASSERT(file != NULL);
_DIAGASSERT(utx != NULL);
fd = open(file, O_WRONLY|O_APPEND|O_SHLOCK);
if (fd == -1) {
if ((fd = open(file, O_CREAT|O_WRONLY|O_EXLOCK, 0644)) == -1)
return -1;
(void)memset(&ut, 0, sizeof(ut));
ut.ut_type = SIGNATURE;
(void)memcpy(ut.ut_user, vers, sizeof(vers));
if (write(fd, &ut, sizeof(ut)) == -1)
goto failed;
}
if (write(fd, utx, sizeof(*utx)) == -1)
goto failed;
if (close(fd) == -1)
return -1;
return 0;
failed:
saved_errno = errno;
(void) close(fd);
errno = saved_errno;
return -1;
}
int
utmpxname(const char *fname)
{
size_t len;
_DIAGASSERT(fname != NULL);
len = strlen(fname);
if (len >= sizeof(utfile))
return 0;
/* must end in x! */
if (fname[len - 1] != 'x')
return 0;
(void)strlcpy(utfile, fname, sizeof(utfile));
endutxent();
return 1;
}
void
getutmp(const struct utmpx *ux, struct utmp *u)
{
_DIAGASSERT(ux != NULL);
_DIAGASSERT(u != NULL);
(void)memcpy(u->ut_name, ux->ut_name, sizeof(u->ut_name));
(void)memcpy(u->ut_line, ux->ut_line, sizeof(u->ut_line));
(void)memcpy(u->ut_host, ux->ut_host, sizeof(u->ut_host));
u->ut_time = ux->ut_tv.tv_sec;
}
void
getutmpx(const struct utmp *u, struct utmpx *ux)
{
_DIAGASSERT(ux != NULL);
_DIAGASSERT(u != NULL);
(void)memcpy(ux->ut_name, u->ut_name, sizeof(u->ut_name));
(void)memcpy(ux->ut_line, u->ut_line, sizeof(u->ut_line));
(void)memcpy(ux->ut_host, u->ut_host, sizeof(u->ut_host));
ux->ut_tv.tv_sec = u->ut_time;
ux->ut_tv.tv_usec = 0;
(void)memset(&ux->ut_ss, 0, sizeof(ux->ut_ss));
ux->ut_pid = 0;
ux->ut_type = USER_PROCESS;
ux->ut_session = 0;
ux->ut_exit.e_termination = 0;
ux->ut_exit.e_exit = 0;
}
int
lastlogxname(const char *fname)
{
size_t len;
_DIAGASSERT(fname != NULL);
len = strlen(fname);
if (len >= sizeof(llfile))
return 0;
/* must end in x! */
if (fname[len - 1] != 'x')
return 0;
(void)strlcpy(llfile, fname, sizeof(llfile));
return 1;
}
struct lastlogx *
getlastlogx(uid_t uid, struct lastlogx *ll)
{
return __getlastlogx13(_PATH_LASTLOGX, uid, ll);
}
struct lastlogx *
__getlastlogx13(const char *fname, uid_t uid, struct lastlogx *ll)
{
DBT key, data;
DB *db;
_DIAGASSERT(fname != NULL);
_DIAGASSERT(ll != NULL);
db = dbopen(fname, O_RDONLY|O_SHLOCK, 0, DB_HASH, NULL);
if (db == NULL)
return NULL;
key.data = &uid;
key.size = sizeof(uid);
if ((db->get)(db, &key, &data, 0) != 0)
goto error;
if (data.size != sizeof(*ll)) {
errno = EFTYPE;
goto error;
}
if (ll == NULL)
if ((ll = malloc(sizeof(*ll))) == NULL)
goto done;
(void)memcpy(ll, data.data, sizeof(*ll));
goto done;
error:
ll = NULL;
done:
(db->close)(db);
return ll;
}
int
updlastlogx(const char *fname, uid_t uid, struct lastlogx *ll)
{
DBT key, data;
int error = 0;
DB *db;
_DIAGASSERT(fname != NULL);
_DIAGASSERT(ll != NULL);
db = dbopen(fname, O_RDWR|O_CREAT|O_EXLOCK, 0, DB_HASH, NULL);
if (db == NULL)
return -1;
key.data = &uid;
key.size = sizeof(uid);
data.data = ll;
data.size = sizeof(*ll);
if ((db->put)(db, &key, &data, 0) != 0)
error = -1;
(db->close)(db);
return error;
}