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