1 /*-
2 * SPDX-License-Identifier: BSD-2-Clause
3 *
4 * Copyright (c) 2009 Ed Schouten <ed@FreeBSD.org>
5 * All rights reserved.
6 *
7 * Redistribution and use in source and binary forms, with or without
8 * modification, are permitted provided that the following conditions
9 * are met:
10 * 1. Redistributions of source code must retain the above copyright
11 * notice, this list of conditions and the following disclaimer.
12 * 2. Redistributions in binary form must reproduce the above copyright
13 * notice, this list of conditions and the following disclaimer in the
14 * documentation and/or other materials provided with the distribution.
15 *
16 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
17 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
18 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
19 * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
20 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
21 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
22 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
23 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
24 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
25 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
26 * SUCH DAMAGE.
27 */
28
29 #include <sys/param.h>
30 #include <sys/time.h>
31 #include <paths.h>
32 #include <sha.h>
33 #include <string.h>
34 #include <unistd.h>
35 #include <utmpx.h>
36 #include "ulog.h"
37
38 static void
ulog_fill(struct utmpx * utx,const char * line)39 ulog_fill(struct utmpx *utx, const char *line)
40 {
41 SHA_CTX c;
42 char id[SHA_DIGEST_LENGTH];
43
44 /* Remove /dev/ component. */
45 if (strncmp(line, _PATH_DEV, sizeof _PATH_DEV - 1) == 0)
46 line += sizeof _PATH_DEV - 1;
47
48 memset(utx, 0, sizeof *utx);
49
50 utx->ut_pid = getpid();
51 gettimeofday(&utx->ut_tv, NULL);
52 strlcpy(utx->ut_line, line, sizeof utx->ut_line);
53
54 SHA1_Init(&c);
55 SHA1_Update(&c, "libulog", 7);
56 SHA1_Update(&c, utx->ut_line, sizeof utx->ut_line);
57 SHA1_Final(id, &c);
58
59 memcpy(utx->ut_id, id, MIN(sizeof utx->ut_id, sizeof id));
60 }
61
62 void
ulog_login(const char * line,const char * user,const char * host)63 ulog_login(const char *line, const char *user, const char *host)
64 {
65 struct utmpx utx;
66
67 ulog_fill(&utx, line);
68 utx.ut_type = USER_PROCESS;
69 strlcpy(utx.ut_user, user, sizeof utx.ut_user);
70 if (host != NULL)
71 strlcpy(utx.ut_host, host, sizeof utx.ut_host);
72 pututxline(&utx);
73 }
74
75 void
ulog_logout(const char * line)76 ulog_logout(const char *line)
77 {
78 struct utmpx utx;
79
80 ulog_fill(&utx, line);
81 utx.ut_type = DEAD_PROCESS;
82 pututxline(&utx);
83 }
84