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