|
|
1 | // This file is part of libptytty. Do not make local modifications. |
|
|
2 | // http://software.schmorp.de/pkg/libptytty |
|
|
3 | |
1 | /*--------------------------------*-C-*---------------------------------* |
4 | /*----------------------------------------------------------------------* |
2 | * File: logging.C |
5 | * File: logging.C |
3 | *----------------------------------------------------------------------* |
6 | *----------------------------------------------------------------------* |
4 | * |
7 | * |
5 | * All portions of code are copyright by their respective author/s. |
8 | * All portions of code are copyright by their respective author/s. |
6 | * Copyright (c) 1992 John Bovey <jdb@ukc.ac.uk> |
9 | * Copyright (c) 1992 John Bovey <jdb@ukc.ac.uk> |
… | |
… | |
13 | * Copyright (c) 1998-2001 Geoff Wing <gcw@pobox.com> |
16 | * Copyright (c) 1998-2001 Geoff Wing <gcw@pobox.com> |
14 | * - extensive modifications |
17 | * - extensive modifications |
15 | * Copyright (c) 1999 D J Hawkey Jr <hawkeyd@visi.com> |
18 | * Copyright (c) 1999 D J Hawkey Jr <hawkeyd@visi.com> |
16 | * - lastlog support |
19 | * - lastlog support |
17 | * Copyright (c) 2004-2006 Marc Lehmann <pcg@goof.com> |
20 | * Copyright (c) 2004-2006 Marc Lehmann <pcg@goof.com> |
|
|
21 | * Copyright (c) 2006 Emanuele Giaquinta <e.giaquinta@glauco.it> |
18 | * |
22 | * |
19 | * This program is free software; you can redistribute it and/or modify |
23 | * This program is free software; you can redistribute it and/or modify |
20 | * it under the terms of the GNU General Public License as published by |
24 | * it under the terms of the GNU General Public License as published by |
21 | * the Free Software Foundation; either version 2 of the License, or |
25 | * the Free Software Foundation; either version 2 of the License, or |
22 | * (at your option) any later version. |
26 | * (at your option) any later version. |
… | |
… | |
30 | * along with this program; if not, write to the Free Software |
34 | * along with this program; if not, write to the Free Software |
31 | * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA. |
35 | * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA. |
32 | *----------------------------------------------------------------------*/ |
36 | *----------------------------------------------------------------------*/ |
33 | |
37 | |
34 | #include "../config.h" |
38 | #include "../config.h" |
35 | #include "rxvt.h" |
39 | |
36 | #include "ptytty.h" |
40 | #include "ptytty.h" |
37 | |
41 | |
38 | #if UTMP_SUPPORT |
42 | #if UTMP_SUPPORT |
39 | |
43 | |
40 | #if HAVE_STRUCT_UTMP |
44 | #include <cstdio> |
41 | static int rxvt_write_bsd_utmp (int utmp_pos, struct utmp *wu); |
45 | #include <cstring> |
42 | static void rxvt_update_wtmp (const char *fname, const struct utmp *putmp); |
|
|
43 | #endif |
|
|
44 | |
46 | |
45 | static void rxvt_update_lastlog (const char *fname, const char *pty, const char *host); |
47 | #include <sys/types.h> |
|
|
48 | #include <sys/stat.h> |
|
|
49 | #include <fcntl.h> |
|
|
50 | #include <unistd.h> |
|
|
51 | #include <time.h> |
|
|
52 | #include <errno.h> |
46 | |
53 | |
47 | /* |
54 | /* |
48 | * BSD style utmp entry |
55 | * BSD style utmp entry |
49 | * ut_line, ut_name, ut_host, ut_time |
56 | * ut_line, ut_name, ut_host, ut_time |
50 | * SYSV style utmp (and utmpx) entry |
57 | * SYSV style utmp (and utmpx) entry |
51 | * ut_user, ut_id, ut_line, ut_pid, ut_type, ut_exit, ut_time |
58 | * ut_user, ut_id, ut_line, ut_pid, ut_type, ut_exit, ut_time |
52 | */ |
59 | */ |
53 | |
60 | |
54 | /* ------------------------------------------------------------------------- */ |
61 | /* ------------------------------------------------------------------------- */ |
|
|
62 | /* |
|
|
63 | * Write a BSD style utmp entry |
|
|
64 | */ |
|
|
65 | #if defined(HAVE_STRUCT_UTMP) && !defined(HAVE_UTMP_PID) |
|
|
66 | static int |
|
|
67 | write_bsd_utmp (int utmp_pos, struct utmp *wu) |
|
|
68 | { |
|
|
69 | int fd; |
|
|
70 | |
|
|
71 | if (utmp_pos <= 0 || (fd = open (UTMP_FILE, O_WRONLY)) == -1) |
|
|
72 | return 0; |
|
|
73 | |
|
|
74 | if (lseek (fd, (off_t) (utmp_pos * sizeof (struct utmp)), SEEK_SET) != -1) |
|
|
75 | write (fd, wu, sizeof (struct utmp)); |
|
|
76 | close (fd); |
|
|
77 | return 1; |
|
|
78 | } |
|
|
79 | #endif |
|
|
80 | |
|
|
81 | /* ------------------------------------------------------------------------- */ |
|
|
82 | /* |
|
|
83 | * Update a BSD style wtmp entry |
|
|
84 | */ |
|
|
85 | #if defined(WTMP_SUPPORT) && !defined(HAVE_UPDWTMP) && defined(HAVE_STRUCT_UTMP) |
|
|
86 | static void |
|
|
87 | update_wtmp (const char *fname, const struct utmp *putmp) |
|
|
88 | { |
|
|
89 | int fd, gotlock, retry; |
|
|
90 | struct flock lck; /* fcntl locking scheme */ |
|
|
91 | struct stat sbuf; |
|
|
92 | |
|
|
93 | if ((fd = open (fname, O_WRONLY | O_APPEND, 0)) < 0) |
|
|
94 | return; |
|
|
95 | |
|
|
96 | lck.l_whence = SEEK_END; /* start lock at current eof */ |
|
|
97 | lck.l_len = 0; /* end at ``largest possible eof'' */ |
|
|
98 | lck.l_start = 0; |
|
|
99 | lck.l_type = F_WRLCK; /* we want a write lock */ |
|
|
100 | |
|
|
101 | /* attempt lock with F_SETLK; F_SETLKW would cause a deadlock! */ |
|
|
102 | for (retry = 10, gotlock = 0; retry--;) |
|
|
103 | if (fcntl (fd, F_SETLK, &lck) != -1) |
|
|
104 | { |
|
|
105 | gotlock = 1; |
|
|
106 | break; |
|
|
107 | } |
|
|
108 | else if (errno != EAGAIN && errno != EACCES) |
|
|
109 | break; |
|
|
110 | if (!gotlock) |
|
|
111 | { |
|
|
112 | /* give it up */ |
|
|
113 | close (fd); |
|
|
114 | return; |
|
|
115 | } |
|
|
116 | if (fstat (fd, &sbuf) == 0) |
|
|
117 | if (write (fd, putmp, sizeof (struct utmp)) != sizeof (struct utmp)) |
|
|
118 | ftruncate (fd, sbuf.st_size); /* remove bad writes */ |
|
|
119 | |
|
|
120 | lck.l_type = F_UNLCK; /* unlocking the file */ |
|
|
121 | fcntl (fd, F_SETLK, &lck); |
|
|
122 | close (fd); |
|
|
123 | } |
|
|
124 | #endif |
|
|
125 | |
|
|
126 | /* ------------------------------------------------------------------------- */ |
|
|
127 | #ifdef LASTLOG_SUPPORT |
|
|
128 | static void |
|
|
129 | update_lastlog (const char *fname, const char *pty, const char *host) |
|
|
130 | { |
|
|
131 | # if defined(HAVE_STRUCT_LASTLOGX) && defined(HAVE_UPDLASTLOGX) |
|
|
132 | struct lastlogx llx; |
|
|
133 | # endif |
|
|
134 | # ifdef HAVE_STRUCT_LASTLOG |
|
|
135 | int fd; |
|
|
136 | struct lastlog ll; |
|
|
137 | char lastlogfile[256]; |
|
|
138 | struct passwd *pwent; |
|
|
139 | struct stat st; |
|
|
140 | # endif |
|
|
141 | |
|
|
142 | # if defined(HAVE_STRUCT_LASTLOGX) && defined(HAVE_UPDLASTLOGX) |
|
|
143 | memset (&llx, 0, sizeof (llx)); |
|
|
144 | llx.ll_tv.tv_sec = time (NULL); |
|
|
145 | llx.ll_tv.tv_usec = 0; |
|
|
146 | strncpy (llx.ll_line, pty, sizeof (llx.ll_line)); |
|
|
147 | strncpy (llx.ll_host, host, sizeof (llx.ll_host)); |
|
|
148 | updlastlogx (LASTLOGX_FILE, getuid (), &llx); |
|
|
149 | # endif |
|
|
150 | |
|
|
151 | # ifdef HAVE_STRUCT_LASTLOG |
|
|
152 | pwent = getpwuid (getuid ()); |
|
|
153 | if (!pwent) |
|
|
154 | { |
|
|
155 | ptytty_warn ("no entry in password file, not updating lastlog.\n", 0); |
|
|
156 | return; |
|
|
157 | } |
|
|
158 | |
|
|
159 | memset (&ll, 0, sizeof (ll)); |
|
|
160 | ll.ll_time = time (NULL); |
|
|
161 | strncpy (ll.ll_line, pty, sizeof (ll.ll_line)); |
|
|
162 | strncpy (ll.ll_host, host, sizeof (ll.ll_host)); |
|
|
163 | if (stat (fname, &st) != 0) |
|
|
164 | return; |
|
|
165 | if (S_ISDIR (st.st_mode)) |
|
|
166 | { |
|
|
167 | sprintf (lastlogfile, "%.*s/%.*s", |
|
|
168 | (int)(sizeof (lastlogfile) - sizeof (pwent->pw_name) - 2), fname, |
|
|
169 | (int)sizeof (pwent->pw_name), |
|
|
170 | (!pwent->pw_name || pwent->pw_name[0] == '\0') ? "unknown" |
|
|
171 | : pwent->pw_name); |
|
|
172 | if ((fd = open (lastlogfile, O_WRONLY | O_CREAT, 0644)) >= 0) |
|
|
173 | { |
|
|
174 | write (fd, &ll, sizeof (ll)); |
|
|
175 | close (fd); |
|
|
176 | } |
|
|
177 | } |
|
|
178 | else if (S_ISREG (st.st_mode)) |
|
|
179 | if ((fd = open (fname, O_RDWR)) != -1) |
|
|
180 | { |
|
|
181 | if (lseek (fd, (off_t) ((long)pwent->pw_uid * sizeof (ll)), |
|
|
182 | SEEK_SET) != -1) |
|
|
183 | write (fd, &ll, sizeof (ll)); |
|
|
184 | close (fd); |
|
|
185 | } |
|
|
186 | # endif /* HAVE_STRUCT_LASTLOG */ |
|
|
187 | } |
|
|
188 | #endif /* LASTLOG_SUPPORT */ |
|
|
189 | |
|
|
190 | /* ------------------------------------------------------------------------- */ |
55 | |
191 | |
56 | /* |
192 | /* |
57 | * make and write utmp and wtmp entries |
193 | * make and write utmp and wtmp entries |
58 | */ |
194 | */ |
59 | void |
195 | void |
60 | rxvt_ptytty_unix::login (int cmd_pid, bool login_shell, const char *hostname) |
196 | ptytty_unix::login (int cmd_pid, bool login_shell, const char *hostname) |
61 | { |
197 | { |
62 | const char *pty = name; |
198 | const char *pty = name; |
63 | |
199 | |
64 | if (!pty || !*pty) |
200 | if (!pty || !*pty) |
65 | return; |
201 | return; |
… | |
… | |
73 | #ifdef HAVE_STRUCT_UTMPX |
209 | #ifdef HAVE_STRUCT_UTMPX |
74 | struct utmpx *utx = &this->utx; |
210 | struct utmpx *utx = &this->utx; |
75 | #endif |
211 | #endif |
76 | int i; |
212 | int i; |
77 | struct passwd *pwent = getpwuid (getuid ()); |
213 | struct passwd *pwent = getpwuid (getuid ()); |
|
|
214 | const char *name = (pwent && pwent->pw_name) ? pwent->pw_name : "?"; |
78 | |
215 | |
79 | if (!strncmp (pty, "/dev/", 5)) |
216 | if (!strncmp (pty, "/dev/", 5)) |
80 | pty += 5; /* skip /dev/ prefix */ |
217 | pty += 5; /* skip /dev/ prefix */ |
81 | |
218 | |
82 | #if defined(HAVE_UTMP_PID) || defined(HAVE_STRUCT_UTMPX) |
219 | #if defined(HAVE_UTMP_PID) || defined(HAVE_STRUCT_UTMPX) |
83 | if (!strncmp (pty, "pty", 3) || !strncmp (pty, "tty", 3)) |
220 | if (!strncmp (pty, "pty", 3) || !strncmp (pty, "tty", 3)) |
84 | strncpy (ut_id, pty + 3, sizeof (ut_id)); |
221 | strncpy (ut_id, pty + 3, sizeof (ut_id)); |
85 | else if (sscanf (pty, "pts/%d", &i) == 1) |
222 | else if (sscanf (pty, "pts/%d", &i) == 1) |
86 | sprintf (ut_id, "vt%02x", (i & 0xff)); /* sysv naming */ |
223 | sprintf (ut_id, "vt%02x", (i & 0xff)); /* sysv naming */ |
87 | else if (strncmp (pty, "pty", 3) && strncmp (pty, "tty", 3)) |
224 | else |
88 | { |
225 | { |
89 | rxvt_warn ("can't parse tty name \"%s\", not adding utmp entry.\n", pty); |
226 | ptytty_warn ("can't parse tty name \"%s\", not adding utmp entry.\n", pty); |
90 | return; |
227 | return; |
91 | } |
228 | } |
92 | #endif |
229 | #endif |
93 | |
230 | |
94 | #ifdef HAVE_STRUCT_UTMP |
231 | #ifdef HAVE_STRUCT_UTMP |
… | |
… | |
96 | # ifdef HAVE_UTMP_PID |
233 | # ifdef HAVE_UTMP_PID |
97 | setutent (); |
234 | setutent (); |
98 | strncpy (ut->ut_id, ut_id, sizeof (ut->ut_id)); |
235 | strncpy (ut->ut_id, ut_id, sizeof (ut->ut_id)); |
99 | ut->ut_type = DEAD_PROCESS; |
236 | ut->ut_type = DEAD_PROCESS; |
100 | getutid (ut); /* position to entry in utmp file */ |
237 | getutid (ut); /* position to entry in utmp file */ |
101 | strncpy (this->ut_id, ut_id, sizeof (this->ut_id)); |
|
|
102 | # endif |
238 | # endif |
103 | #endif |
239 | #endif |
104 | |
240 | |
105 | #ifdef HAVE_STRUCT_UTMPX |
241 | #ifdef HAVE_STRUCT_UTMPX |
106 | memset (utx, 0, sizeof (struct utmpx)); |
242 | memset (utx, 0, sizeof (struct utmpx)); |
107 | setutxent (); |
243 | setutxent (); |
108 | strncpy (utx->ut_id, ut_id, sizeof (utx->ut_id)); |
244 | strncpy (utx->ut_id, ut_id, sizeof (utx->ut_id)); |
109 | utx->ut_type = DEAD_PROCESS; |
245 | utx->ut_type = DEAD_PROCESS; |
110 | getutxid (utx); /* position to entry in utmp file */ |
246 | getutxid (utx); /* position to entry in utmp file */ |
111 | strncpy (this->ut_id, ut_id, sizeof (this->ut_id)); |
|
|
112 | #endif |
247 | #endif |
113 | |
248 | |
114 | #ifdef HAVE_STRUCT_UTMP |
249 | #ifdef HAVE_STRUCT_UTMP |
115 | strncpy (ut->ut_line, pty, sizeof (ut->ut_line)); |
250 | strncpy (ut->ut_line, pty, sizeof (ut->ut_line)); |
|
|
251 | # ifdef HAVE_UTMP_HOST |
|
|
252 | strncpy (ut->ut_host, hostname, sizeof (ut->ut_host)); |
|
|
253 | # endif |
116 | ut->ut_time = time (NULL); |
254 | ut->ut_time = time (NULL); |
117 | # ifdef HAVE_UTMP_PID |
255 | # ifdef HAVE_UTMP_PID |
118 | strncpy (ut->ut_user, (pwent && pwent->pw_name) ? pwent->pw_name : "?", |
256 | strncpy (ut->ut_user, name, sizeof (ut->ut_user)); |
119 | sizeof (ut->ut_user)); |
|
|
120 | strncpy (ut->ut_id, ut_id, sizeof (ut->ut_id)); |
257 | strncpy (ut->ut_id, ut_id, sizeof (ut->ut_id)); |
121 | ut->ut_time = time (NULL); |
|
|
122 | ut->ut_pid = cmd_pid; |
258 | ut->ut_pid = cmd_pid; |
123 | # ifdef HAVE_UTMP_HOST |
|
|
124 | strncpy (ut->ut_host, hostname, sizeof (ut->ut_host)); |
|
|
125 | # endif |
|
|
126 | ut->ut_type = USER_PROCESS; |
259 | ut->ut_type = USER_PROCESS; |
127 | pututline (ut); |
260 | pututline (ut); |
128 | endutent (); /* close the file */ |
261 | endutent (); /* close the file */ |
129 | utmp_pos = 0; |
262 | utmp_pos = 0; |
130 | # else |
263 | # else |
131 | strncpy (ut->ut_name, (pwent && pwent->pw_name) ? pwent->pw_name : "?", |
|
|
132 | sizeof (ut->ut_name)); |
|
|
133 | # ifdef HAVE_UTMP_HOST |
|
|
134 | strncpy (ut->ut_host, hostname, sizeof (ut->ut_host)); |
264 | strncpy (ut->ut_name, name, sizeof (ut->ut_name)); |
135 | # endif |
|
|
136 | # endif |
265 | # endif |
137 | #endif |
266 | #endif |
138 | |
267 | |
139 | #ifdef HAVE_STRUCT_UTMPX |
268 | #ifdef HAVE_STRUCT_UTMPX |
140 | strncpy (utx->ut_line, pty, sizeof (utx->ut_line)); |
269 | strncpy (utx->ut_line, pty, sizeof (utx->ut_line)); |
141 | strncpy (utx->ut_user, (pwent && pwent->pw_name) ? pwent->pw_name : "?", |
270 | strncpy (utx->ut_user, name, sizeof (utx->ut_user)); |
142 | sizeof (utx->ut_user)); |
|
|
143 | strncpy (utx->ut_id, ut_id, sizeof (utx->ut_id)); |
271 | strncpy (utx->ut_id, ut_id, sizeof (utx->ut_id)); |
144 | # if HAVE_UTMPX_SESSION |
272 | # if HAVE_UTMPX_SESSION |
145 | utx->ut_session = getsid (0); |
273 | utx->ut_session = getsid (0); |
146 | # endif |
274 | # endif |
147 | utx->ut_tv.tv_sec = time (NULL); |
275 | utx->ut_tv.tv_sec = time (NULL); |
… | |
… | |
163 | endutxent (); /* close the file */ |
291 | endutxent (); /* close the file */ |
164 | utmp_pos = 0; |
292 | utmp_pos = 0; |
165 | #endif |
293 | #endif |
166 | |
294 | |
167 | #if defined(HAVE_STRUCT_UTMP) && !defined(HAVE_UTMP_PID) |
295 | #if defined(HAVE_STRUCT_UTMP) && !defined(HAVE_UTMP_PID) |
168 | |
|
|
169 | { |
296 | { |
170 | # ifdef HAVE_TTYSLOT |
297 | # if 1 |
|
|
298 | int fdstdin = dup (STDIN_FILENO); |
|
|
299 | dup2 (tty, STDIN_FILENO); |
|
|
300 | |
171 | i = ttyslot (); |
301 | i = ttyslot (); |
172 | if (rxvt_write_bsd_utmp (i, ut)) |
302 | if (write_bsd_utmp (i, ut)) |
173 | utmp_pos = i; |
303 | utmp_pos = i; |
174 | # else |
|
|
175 | FILE *fd0; |
|
|
176 | |
304 | |
177 | if ((fd0 = fopen (TTYTAB_FILENAME, "r")) != NULL) |
305 | dup2 (fdstdin, STDIN_FILENO); |
178 | { |
306 | close (fdstdin); |
179 | char buf[256], name[256]; |
|
|
180 | |
|
|
181 | buf[sizeof (buf) - 1] = '\0'; |
|
|
182 | for (i = 1; (fgets (buf, sizeof (buf) - 1, fd0) != NULL);) |
|
|
183 | { |
|
|
184 | if (*buf == '#' || sscanf (buf, "%s", name) != 1) |
|
|
185 | continue; |
|
|
186 | if (!strcmp (ut->ut_line, name)) |
|
|
187 | { |
|
|
188 | if (!rxvt_write_bsd_utmp (i, ut)) |
|
|
189 | i = 0; |
|
|
190 | utmp_pos = i; |
|
|
191 | fclose (fd0); |
|
|
192 | break; |
|
|
193 | } |
|
|
194 | i++; |
|
|
195 | } |
|
|
196 | fclose (fd0); |
|
|
197 | } |
|
|
198 | # endif |
307 | # endif |
199 | } |
308 | } |
200 | #endif |
309 | #endif |
201 | |
310 | |
202 | #ifdef WTMP_SUPPORT |
311 | #ifdef WTMP_SUPPORT |
203 | # ifdef WTMP_ONLY_ON_LOGIN |
312 | #ifdef LOG_ONLY_ON_LOGIN |
204 | if (login_shell) |
313 | if (login_shell) |
205 | # endif |
314 | #endif |
206 | { |
315 | { |
207 | # ifdef HAVE_STRUCT_UTMP |
316 | # ifdef HAVE_STRUCT_UTMP |
208 | # ifdef HAVE_UPDWTMP |
317 | # ifdef HAVE_UPDWTMP |
209 | updwtmp (RXVT_WTMP_FILE, ut); |
318 | updwtmp (WTMP_FILE, ut); |
210 | # else |
319 | # else |
211 | rxvt_update_wtmp (RXVT_WTMP_FILE, ut); |
320 | update_wtmp (WTMP_FILE, ut); |
212 | # endif |
321 | # endif |
213 | # endif |
322 | # endif |
214 | # if defined(HAVE_STRUCT_UTMPX) && defined(HAVE_UPDWTMPX) |
323 | # if defined(HAVE_STRUCT_UTMPX) && defined(HAVE_UPDWTMPX) |
215 | updwtmpx (RXVT_WTMPX_FILE, utx); |
324 | updwtmpx (WTMPX_FILE, utx); |
216 | # endif |
|
|
217 | } |
|
|
218 | #endif |
325 | # endif |
|
|
326 | } |
|
|
327 | #endif |
219 | #if defined(LASTLOG_SUPPORT) && defined(RXVT_LASTLOG_FILE) |
328 | #if defined(LASTLOG_SUPPORT) && defined(LASTLOG_FILE) |
|
|
329 | #ifdef LOG_ONLY_ON_LOGIN |
220 | if (login_shell) |
330 | if (login_shell) |
|
|
331 | #endif |
221 | rxvt_update_lastlog (RXVT_LASTLOG_FILE, pty, hostname); |
332 | update_lastlog (LASTLOG_FILE, pty, hostname); |
222 | #endif |
333 | #endif |
223 | } |
334 | } |
224 | |
335 | |
225 | /* ------------------------------------------------------------------------- */ |
336 | /* ------------------------------------------------------------------------- */ |
226 | /* |
337 | /* |
227 | * remove utmp and wtmp entries |
338 | * remove utmp and wtmp entries |
228 | */ |
339 | */ |
229 | void |
340 | void |
230 | rxvt_ptytty_unix::logout () |
341 | ptytty_unix::logout () |
231 | { |
342 | { |
232 | if (!cmd_pid) |
343 | if (!cmd_pid) |
233 | return; |
344 | return; |
234 | |
345 | |
235 | #ifdef HAVE_STRUCT_UTMP |
346 | #ifdef HAVE_STRUCT_UTMP |
… | |
… | |
274 | |
385 | |
275 | /* |
386 | /* |
276 | * Write ending wtmp entry |
387 | * Write ending wtmp entry |
277 | */ |
388 | */ |
278 | #ifdef WTMP_SUPPORT |
389 | #ifdef WTMP_SUPPORT |
279 | # ifdef WTMP_ONLY_ON_LOGIN |
390 | #ifdef LOG_ONLY_ON_LOGIN |
280 | if (login_shell) |
391 | if (login_shell) |
281 | # endif |
392 | #endif |
282 | { |
393 | { |
283 | # ifdef HAVE_STRUCT_UTMP |
394 | # ifdef HAVE_STRUCT_UTMP |
284 | # ifdef HAVE_UPDWTMP |
395 | # ifdef HAVE_UPDWTMP |
285 | updwtmp (RXVT_WTMP_FILE, ut); |
396 | updwtmp (WTMP_FILE, ut); |
286 | # else |
397 | # else |
287 | rxvt_update_wtmp (RXVT_WTMP_FILE, ut); |
398 | update_wtmp (WTMP_FILE, ut); |
288 | # endif |
399 | # endif |
289 | # endif |
400 | # endif |
290 | # if defined(HAVE_STRUCT_UTMPX) && defined(HAVE_UPDWTMPX) |
401 | # if defined(HAVE_STRUCT_UTMPX) && defined(HAVE_UPDWTMPX) |
291 | updwtmpx (RXVT_WTMPX_FILE, utx); |
402 | updwtmpx (WTMPX_FILE, utx); |
292 | # endif |
403 | # endif |
293 | } |
404 | } |
294 | #endif |
405 | #endif |
295 | |
406 | |
296 | /* |
407 | /* |
… | |
… | |
301 | if (ut->ut_pid == cmd_pid) |
412 | if (ut->ut_pid == cmd_pid) |
302 | pututline (ut); |
413 | pututline (ut); |
303 | endutent (); |
414 | endutent (); |
304 | # else |
415 | # else |
305 | memset (ut, 0, sizeof (struct utmp)); |
416 | memset (ut, 0, sizeof (struct utmp)); |
306 | rxvt_write_bsd_utmp (utmp_pos, ut); |
417 | write_bsd_utmp (utmp_pos, ut); |
307 | # endif |
418 | # endif |
308 | #endif |
419 | #endif |
309 | #ifdef HAVE_STRUCT_UTMPX |
420 | #ifdef HAVE_STRUCT_UTMPX |
310 | if (utx->ut_pid == cmd_pid) |
421 | if (utx->ut_pid == cmd_pid) |
311 | pututxline (utx); |
422 | pututxline (utx); |
… | |
… | |
313 | #endif |
424 | #endif |
314 | |
425 | |
315 | cmd_pid = 0; |
426 | cmd_pid = 0; |
316 | } |
427 | } |
317 | |
428 | |
318 | /* ------------------------------------------------------------------------- */ |
429 | #else |
319 | /* |
430 | void |
320 | * Write a BSD style utmp entry |
431 | ptytty_unix::login (int cmd_pid, bool login_shell, const char *hostname) |
321 | */ |
|
|
322 | #if defined(HAVE_STRUCT_UTMP) && !defined(HAVE_UTMP_PID) |
|
|
323 | static int |
|
|
324 | rxvt_write_bsd_utmp (int utmp_pos, struct utmp *wu) |
|
|
325 | { |
432 | { |
326 | int fd; |
|
|
327 | |
|
|
328 | if (utmp_pos <= 0 || (fd = open (RXVT_UTMP_FILE, O_WRONLY)) == -1) |
|
|
329 | return 0; |
|
|
330 | |
|
|
331 | if (lseek (fd, (off_t) (utmp_pos * sizeof (struct utmp)), SEEK_SET) != -1) |
|
|
332 | write (fd, wu, sizeof (struct utmp)); |
|
|
333 | close (fd); |
|
|
334 | return 1; |
|
|
335 | } |
433 | } |
336 | #endif |
|
|
337 | |
|
|
338 | /* ------------------------------------------------------------------------- */ |
|
|
339 | /* |
|
|
340 | * Update a BSD style wtmp entry |
|
|
341 | */ |
|
|
342 | #if defined(WTMP_SUPPORT) && !defined(HAVE_UPDWTMP) && defined(HAVE_STRUCT_UTMP) |
|
|
343 | static void |
|
|
344 | rxvt_update_wtmp (const char *fname, const struct utmp *putmp) |
|
|
345 | { |
|
|
346 | int fd, gotlock, retry; |
|
|
347 | struct flock lck; /* fcntl locking scheme */ |
|
|
348 | struct stat sbuf; |
|
|
349 | |
|
|
350 | if ((fd = open (fname, O_WRONLY | O_APPEND, 0)) < 0) |
|
|
351 | return; |
|
|
352 | |
|
|
353 | lck.l_whence = SEEK_END; /* start lock at current eof */ |
|
|
354 | lck.l_len = 0; /* end at ``largest possible eof'' */ |
|
|
355 | lck.l_start = 0; |
|
|
356 | lck.l_type = F_WRLCK; /* we want a write lock */ |
|
|
357 | |
|
|
358 | /* attempt lock with F_SETLK; F_SETLKW would cause a deadlock! */ |
|
|
359 | for (retry = 10, gotlock = 0; retry--;) |
|
|
360 | if (fcntl (fd, F_SETLK, &lck) != -1) |
|
|
361 | { |
|
|
362 | gotlock = 1; |
|
|
363 | break; |
|
|
364 | } |
|
|
365 | else if (errno != EAGAIN && errno != EACCES) |
|
|
366 | break; |
|
|
367 | if (!gotlock) |
|
|
368 | { /* give it up */ |
|
|
369 | close (fd); |
|
|
370 | return; |
|
|
371 | } |
|
|
372 | if (fstat (fd, &sbuf) == 0) |
|
|
373 | if (write (fd, putmp, sizeof (struct utmp)) != sizeof (struct utmp)) |
|
|
374 | ftruncate (fd, sbuf.st_size); /* remove bad writes */ |
|
|
375 | |
|
|
376 | lck.l_type = F_UNLCK; /* unlocking the file */ |
|
|
377 | fcntl (fd, F_SETLK, &lck); |
|
|
378 | close (fd); |
|
|
379 | } |
|
|
380 | #endif |
|
|
381 | |
|
|
382 | /* ------------------------------------------------------------------------- */ |
|
|
383 | #ifdef LASTLOG_SUPPORT |
|
|
384 | static void |
|
|
385 | rxvt_update_lastlog (const char *fname, const char *pty, const char *host) |
|
|
386 | { |
|
|
387 | # ifdef HAVE_STRUCT_LASTLOGX |
|
|
388 | struct lastlogx llx; |
|
|
389 | # endif |
434 | #endif |
390 | # ifdef HAVE_STRUCT_LASTLOG |
|
|
391 | int fd; |
|
|
392 | struct lastlog ll; |
|
|
393 | # ifdef LASTLOG_IS_DIR |
|
|
394 | char lastlogfile[256]; |
|
|
395 | # endif |
|
|
396 | struct passwd *pwent; |
|
|
397 | # endif |
|
|
398 | |
435 | |
399 | # ifdef HAVE_STRUCT_LASTLOGX |
|
|
400 | memset (&llx, 0, sizeof (llx)); |
|
|
401 | llx.ll_tv.tv_sec = time (NULL); |
|
|
402 | llx.ll_tv.tv_usec = 0; |
|
|
403 | strncpy (llx.ll_line, pty, sizeof (llx.ll_line)); |
|
|
404 | strncpy (llx.ll_host, host, sizeof (llx.ll_host)); |
|
|
405 | updlastlogx (RXVT_LASTLOGX_FILE, getuid (), &llx); |
|
|
406 | # endif |
|
|
407 | |
|
|
408 | # ifdef HAVE_STRUCT_LASTLOG |
|
|
409 | pwent = getpwuid (getuid ()); |
|
|
410 | if (!pwent) |
|
|
411 | { |
|
|
412 | rxvt_warn ("no entry in password file, not updating lastlog.\n"); |
|
|
413 | return; |
|
|
414 | } |
|
|
415 | |
|
|
416 | memset (&ll, 0, sizeof (ll)); |
|
|
417 | ll.ll_time = time (NULL); |
|
|
418 | strncpy (ll.ll_line, pty, sizeof (ll.ll_line)); |
|
|
419 | strncpy (ll.ll_host, host, sizeof (ll.ll_host)); |
|
|
420 | # ifdef LASTLOG_IS_DIR |
|
|
421 | sprintf (lastlogfile, "%.*s/%.*s", |
|
|
422 | sizeof (lastlogfile) - sizeof (pwent->pw_name) - 2, fname, |
|
|
423 | sizeof (pwent->pw_name), |
|
|
424 | (!pwent->pw_name || pwent->pw_name[0] == '\0') ? "unknown" |
|
|
425 | : pwent->pw_name); |
|
|
426 | if ((fd = open (lastlogfile, O_WRONLY | O_CREAT, 0644)) >= 0) |
|
|
427 | { |
|
|
428 | write (fd, &ll, sizeof (ll)); |
|
|
429 | close (fd); |
|
|
430 | } |
|
|
431 | # else |
|
|
432 | if ((fd = open (fname, O_RDWR)) != -1) |
|
|
433 | { |
|
|
434 | if (lseek (fd, (off_t) ((long)pwent->pw_uid * sizeof (ll)), |
|
|
435 | SEEK_SET) != -1) |
|
|
436 | write (fd, &ll, sizeof (ll)); |
|
|
437 | close (fd); |
|
|
438 | } |
|
|
439 | # endif /* LASTLOG_IS_DIR */ |
|
|
440 | # endif /* HAVE_STRUCT_LASTLOG */ |
|
|
441 | } |
|
|
442 | #endif /* LASTLOG_SUPPORT */ |
|
|
443 | /* ------------------------------------------------------------------------- */ |
|
|
444 | |
|
|
445 | #endif /* UTMP_SUPPORT */ |
|
|