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