xref: /freebsd/lib/libc/gen/utxdb.c (revision 984485a02eb3e63b4170dd911b72de38b35b2289)
1 /*-
2  * Copyright (c) 2010 Ed Schouten <ed@FreeBSD.org>
3  * All rights reserved.
4  *
5  * Redistribution and use in source and binary forms, with or without
6  * modification, are permitted provided that the following conditions
7  * are met:
8  * 1. Redistributions of source code must retain the above copyright
9  *    notice, this list of conditions and the following disclaimer.
10  * 2. Redistributions in binary form must reproduce the above copyright
11  *    notice, this list of conditions and the following disclaimer in the
12  *    documentation and/or other materials provided with the distribution.
13  *
14  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
15  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
16  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
17  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
18  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
19  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
20  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
21  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
22  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
23  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
24  * SUCH DAMAGE.
25  */
26 
27 #include <sys/cdefs.h>
28 __FBSDID("$FreeBSD$");
29 
30 #include "namespace.h"
31 #include <sys/endian.h>
32 #include <sys/param.h>
33 #include <stdlib.h>
34 #include <string.h>
35 #include <utmpx.h>
36 #include "utxdb.h"
37 #include "un-namespace.h"
38 
39 #define	UTOF_STRING(ut, fu, field) do { \
40 	strncpy((fu)->fu_ ## field, (ut)->ut_ ## field,		\
41 	    MIN(sizeof (fu)->fu_ ## field, sizeof (ut)->ut_ ## field));	\
42 } while (0)
43 #define	UTOF_ID(ut, fu) do { \
44 	memcpy((fu)->fu_id, (ut)->ut_id,				\
45 	    MIN(sizeof (fu)->fu_id, sizeof (ut)->ut_id));		\
46 } while (0)
47 #define	UTOF_PID(ut, fu) do { \
48 	(fu)->fu_pid = htobe32((ut)->ut_pid);				\
49 } while (0)
50 #define	UTOF_TYPE(ut, fu) do { \
51 	(fu)->fu_type = (ut)->ut_type;					\
52 } while (0)
53 #define	UTOF_TV(ut, fu) do { \
54 	(fu)->fu_tv = htobe64((uint64_t)(ut)->ut_tv.tv_sec * 1000000 +	\
55 	    (uint64_t)(ut)->ut_tv.tv_usec);				\
56 } while (0)
57 
58 void
59 utx_to_futx(const struct utmpx *ut, struct futx *fu)
60 {
61 
62 	memset(fu, 0, sizeof *fu);
63 
64 	switch (ut->ut_type) {
65 	case BOOT_TIME:
66 	case OLD_TIME:
67 	case NEW_TIME:
68 	/* Extension: shutdown time. */
69 	case SHUTDOWN_TIME:
70 		break;
71 	case USER_PROCESS:
72 		UTOF_ID(ut, fu);
73 		UTOF_STRING(ut, fu, user);
74 		UTOF_STRING(ut, fu, line);
75 		/* Extension: host name. */
76 		UTOF_STRING(ut, fu, host);
77 		UTOF_PID(ut, fu);
78 		break;
79 	case INIT_PROCESS:
80 		UTOF_ID(ut, fu);
81 		UTOF_PID(ut, fu);
82 		break;
83 	case LOGIN_PROCESS:
84 		UTOF_ID(ut, fu);
85 		UTOF_STRING(ut, fu, user);
86 		/* XXX: bug in the specification? Needed for getutxline(). */
87 		UTOF_STRING(ut, fu, line);
88 		UTOF_PID(ut, fu);
89 		break;
90 	case DEAD_PROCESS:
91 		UTOF_ID(ut, fu);
92 		UTOF_PID(ut, fu);
93 		break;
94 	default:
95 		fu->fu_type = EMPTY;
96 		return;
97 	}
98 
99 	UTOF_TYPE(ut, fu);
100 	UTOF_TV(ut, fu);
101 }
102 
103 #define	FTOU_STRING(fu, ut, field) do { \
104 	strncpy((ut)->ut_ ## field, (fu)->fu_ ## field,		\
105 	    MIN(sizeof (ut)->ut_ ## field - 1, sizeof (fu)->fu_ ## field)); \
106 } while (0)
107 #define	FTOU_ID(fu, ut) do { \
108 	memcpy((ut)->ut_id, (fu)->fu_id,				\
109 	    MIN(sizeof (ut)->ut_id, sizeof (fu)->fu_id));		\
110 } while (0)
111 #define	FTOU_PID(fu, ut) do { \
112 	(ut)->ut_pid = be32toh((fu)->fu_pid);				\
113 } while (0)
114 #define	FTOU_TYPE(fu, ut) do { \
115 	(ut)->ut_type = (fu)->fu_type;					\
116 } while (0)
117 #define	FTOU_TV(fu, ut) do { \
118 	uint64_t t;							\
119 	t = be64toh((fu)->fu_tv);					\
120 	(ut)->ut_tv.tv_sec = t / 1000000;				\
121 	(ut)->ut_tv.tv_usec = t % 1000000;				\
122 } while (0)
123 
124 struct utmpx *
125 futx_to_utx(const struct futx *fu)
126 {
127 	static struct utmpx *ut;
128 
129 	if (ut == NULL) {
130 		ut = calloc(1, sizeof *ut);
131 		if (ut == NULL)
132 			return (NULL);
133 	} else {
134 		memset(ut, 0, sizeof *ut);
135 	}
136 
137 	switch (fu->fu_type) {
138 	case BOOT_TIME:
139 	case OLD_TIME:
140 	case NEW_TIME:
141 	/* Extension: shutdown time. */
142 	case SHUTDOWN_TIME:
143 		break;
144 	case USER_PROCESS:
145 		FTOU_ID(fu, ut);
146 		FTOU_STRING(fu, ut, user);
147 		FTOU_STRING(fu, ut, line);
148 		/* Extension: host name. */
149 		FTOU_STRING(fu, ut, host);
150 		FTOU_PID(fu, ut);
151 		break;
152 	case INIT_PROCESS:
153 		FTOU_ID(fu, ut);
154 		FTOU_PID(fu, ut);
155 		break;
156 	case LOGIN_PROCESS:
157 		FTOU_ID(fu, ut);
158 		FTOU_STRING(fu, ut, user);
159 		/* XXX: bug in the specification? Needed for getutxline(). */
160 		FTOU_STRING(fu, ut, line);
161 		FTOU_PID(fu, ut);
162 		break;
163 	case DEAD_PROCESS:
164 		FTOU_ID(fu, ut);
165 		FTOU_PID(fu, ut);
166 		break;
167 	default:
168 		ut->ut_type = EMPTY;
169 		return (ut);
170 	}
171 
172 	FTOU_TYPE(fu, ut);
173 	FTOU_TV(fu, ut);
174 	return (ut);
175 }
176