|
|
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.2 2003/11/24 17:31:27 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 | { |
|
|
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 | /* EXTPROTO */ |
|
|
60 | void |
193 | void |
61 | rxvt_makeutent(pR_ const char *pty, const char *hostname) |
194 | ptytty_unix::login (int cmd_pid, bool login_shell, const char *hostname) |
62 | { |
195 | { |
63 | #ifdef HAVE_STRUCT_UTMP |
196 | const char *pty = name; |
64 | struct utmp *ut = &(R->h->ut); |
197 | |
65 | #endif |
198 | if (!pty || !*pty) |
|
|
199 | return; |
|
|
200 | |
|
|
201 | this->cmd_pid = cmd_pid; |
|
|
202 | this->login_shell = login_shell; |
|
|
203 | |
66 | #ifdef HAVE_STRUCT_UTMPX |
204 | #ifdef HAVE_STRUCT_UTMP |
67 | struct utmpx *utx = &(R->h->utx); |
205 | struct utmp *ut = &this->ut; |
68 | #endif |
206 | #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 |
207 | #ifdef HAVE_STRUCT_UTMPX |
|
|
208 | struct utmpx *utx = &this->utx; |
|
|
209 | #endif |
|
|
210 | int i; |
|
|
211 | struct passwd *pwent = getpwuid (getuid ()); |
|
|
212 | const char *name = (pwent && pwent->pw_name) ? pwent->pw_name : "?"; |
|
|
213 | |
|
|
214 | if (!strncmp (pty, "/dev/", 5)) |
|
|
215 | pty += 5; /* skip /dev/ prefix */ |
|
|
216 | |
|
|
217 | #if defined(HAVE_UTMP_PID) || defined(HAVE_STRUCT_UTMPX) |
|
|
218 | if (!strncmp (pty, "pty", 3) || !strncmp (pty, "tty", 3)) |
|
|
219 | strncpy (ut_id, pty + 3, sizeof (ut_id)); |
|
|
220 | else if (sscanf (pty, "pts/%d", &i) == 1) |
|
|
221 | sprintf (ut_id, "vt%02x", (i & 0xff)); /* sysv naming */ |
|
|
222 | else |
|
|
223 | { |
|
|
224 | ptytty_warn ("can't parse tty name \"%s\", not adding utmp entry.\n", pty); |
|
|
225 | return; |
|
|
226 | } |
|
|
227 | #endif |
|
|
228 | |
|
|
229 | #ifdef HAVE_STRUCT_UTMP |
91 | MEMSET(ut, 0, sizeof(struct utmp)); |
230 | memset (ut, 0, sizeof (struct utmp)); |
92 | # ifdef HAVE_UTMP_PID |
231 | # ifdef HAVE_UTMP_PID |
93 | setutent(); |
232 | setutent (); |
94 | STRNCPY(ut->ut_id, ut_id, sizeof(ut->ut_id)); |
233 | strncpy (ut->ut_id, ut_id, sizeof (ut->ut_id)); |
95 | ut->ut_type = DEAD_PROCESS; |
234 | ut->ut_type = DEAD_PROCESS; |
96 | getutid(ut); /* position to entry in utmp file */ |
235 | getutid (ut); /* position to entry in utmp file */ |
97 | STRNCPY(R->h->ut_id, ut_id, sizeof(R->h->ut_id)); |
|
|
98 | # endif |
|
|
99 | #endif |
236 | # 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 |
237 | #endif |
109 | |
238 | |
110 | #ifdef HAVE_STRUCT_UTMP |
239 | #ifdef HAVE_STRUCT_UTMPX |
|
|
240 | memset (utx, 0, sizeof (struct utmpx)); |
|
|
241 | setutxent (); |
|
|
242 | strncpy (utx->ut_id, ut_id, sizeof (utx->ut_id)); |
|
|
243 | utx->ut_type = DEAD_PROCESS; |
|
|
244 | getutxid (utx); /* position to entry in utmp file */ |
|
|
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)); |
|
|
249 | # ifdef HAVE_UTMP_HOST |
|
|
250 | strncpy (ut->ut_host, hostname, sizeof (ut->ut_host)); |
|
|
251 | # endif |
112 | ut->ut_time = time(NULL); |
252 | ut->ut_time = time (NULL); |
113 | # ifdef HAVE_UTMP_PID |
253 | # ifdef HAVE_UTMP_PID |
114 | STRNCPY(ut->ut_user, (pwent && pwent->pw_name) ? pwent->pw_name : "?", |
254 | strncpy (ut->ut_user, name, sizeof (ut->ut_user)); |
115 | sizeof(ut->ut_user)); |
|
|
116 | STRNCPY(ut->ut_id, ut_id, sizeof(ut->ut_id)); |
255 | strncpy (ut->ut_id, ut_id, sizeof (ut->ut_id)); |
117 | ut->ut_time = time(NULL); |
|
|
118 | ut->ut_pid = R->h->cmd_pid; |
256 | ut->ut_pid = cmd_pid; |
|
|
257 | ut->ut_type = USER_PROCESS; |
|
|
258 | pututline (ut); |
|
|
259 | endutent (); /* close the file */ |
|
|
260 | utmp_pos = 0; |
|
|
261 | # else |
|
|
262 | strncpy (ut->ut_name, name, sizeof (ut->ut_name)); |
|
|
263 | # endif |
|
|
264 | #endif |
|
|
265 | |
|
|
266 | #ifdef HAVE_STRUCT_UTMPX |
|
|
267 | strncpy (utx->ut_line, pty, sizeof (utx->ut_line)); |
|
|
268 | strncpy (utx->ut_user, name, sizeof (utx->ut_user)); |
|
|
269 | strncpy (utx->ut_id, ut_id, sizeof (utx->ut_id)); |
|
|
270 | # if HAVE_UTMPX_SESSION |
|
|
271 | utx->ut_session = getsid (0); |
|
|
272 | # endif |
|
|
273 | utx->ut_tv.tv_sec = time (NULL); |
|
|
274 | utx->ut_tv.tv_usec = 0; |
|
|
275 | utx->ut_pid = cmd_pid; |
119 | # ifdef HAVE_UTMP_HOST |
276 | # ifdef HAVE_UTMPX_HOST |
120 | STRNCPY(ut->ut_host, hostname, sizeof(ut->ut_host)); |
277 | strncpy (utx->ut_host, hostname, sizeof (utx->ut_host)); |
|
|
278 | # if 0 |
|
|
279 | { |
|
|
280 | char *colon; |
|
|
281 | |
|
|
282 | if ((colon = strrchr (ut->ut_host, ':')) != NULL) |
|
|
283 | *colon = '\0'; |
|
|
284 | } |
121 | # endif |
285 | # endif |
122 | ut->ut_type = USER_PROCESS; |
|
|
123 | pututline(ut); |
|
|
124 | endutent(); /* close the file */ |
|
|
125 | R->h->utmp_pos = 0; |
|
|
126 | # else |
|
|
127 | STRNCPY(ut->ut_name, (pwent && pwent->pw_name) ? pwent->pw_name : "?", |
|
|
128 | sizeof(ut->ut_name)); |
|
|
129 | # ifdef HAVE_UTMP_HOST |
|
|
130 | STRNCPY(ut->ut_host, hostname, sizeof(ut->ut_host)); |
|
|
131 | # endif |
286 | # endif |
132 | # endif |
|
|
133 | #endif |
|
|
134 | |
|
|
135 | #ifdef HAVE_STRUCT_UTMPX |
|
|
136 | STRNCPY(utx->ut_line, pty, sizeof(utx->ut_line)); |
|
|
137 | STRNCPY(utx->ut_user, (pwent && pwent->pw_name) ? pwent->pw_name : "?", |
|
|
138 | sizeof(utx->ut_user)); |
|
|
139 | STRNCPY(utx->ut_id, ut_id, sizeof(utx->ut_id)); |
|
|
140 | utx->ut_session = getsid(0); |
|
|
141 | utx->ut_tv.tv_sec = time(NULL); |
|
|
142 | utx->ut_tv.tv_usec = 0; |
|
|
143 | utx->ut_pid = R->h->cmd_pid; |
|
|
144 | # ifdef HAVE_UTMPX_HOST |
|
|
145 | STRNCPY(utx->ut_host, hostname, sizeof(utx->ut_host)); |
|
|
146 | # if 0 |
|
|
147 | { |
|
|
148 | char *colon; |
|
|
149 | |
|
|
150 | if ((colon = STRRCHR(ut->ut_host, ':')) != NULL) |
|
|
151 | *colon = '\0'; |
|
|
152 | } |
|
|
153 | # endif |
|
|
154 | # endif |
|
|
155 | utx->ut_type = USER_PROCESS; |
287 | utx->ut_type = USER_PROCESS; |
156 | pututxline(utx); |
288 | pututxline (utx); |
157 | endutxent(); /* close the file */ |
289 | endutxent (); /* close the file */ |
158 | R->h->utmp_pos = 0; |
290 | utmp_pos = 0; |
159 | #endif |
291 | #endif |
160 | |
292 | |
161 | #if defined(HAVE_STRUCT_UTMP) && !defined(HAVE_UTMP_PID) |
293 | #if defined(HAVE_STRUCT_UTMP) && !defined(HAVE_UTMP_PID) |
162 | { |
294 | { |
163 | int i; |
295 | # if 1 |
164 | # ifdef HAVE_TTYSLOT |
296 | int fdstdin = dup (STDIN_FILENO); |
|
|
297 | dup2 (tty, STDIN_FILENO); |
|
|
298 | |
165 | i = ttyslot(); |
299 | 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)) |
300 | if (write_bsd_utmp (i, ut)) |
180 | i = 0; |
301 | utmp_pos = i; |
181 | R->h->utmp_pos = i; |
302 | |
182 | fclose(fd0); |
303 | dup2 (fdstdin, STDIN_FILENO); |
183 | break; |
304 | close (fdstdin); |
184 | } |
|
|
185 | i++; |
|
|
186 | } |
|
|
187 | fclose(fd0); |
|
|
188 | } |
|
|
189 | # endif |
305 | # endif |
190 | } |
306 | } |
191 | #endif |
307 | #endif |
192 | |
308 | |
193 | #ifdef WTMP_SUPPORT |
309 | #ifdef WTMP_SUPPORT |
194 | # ifdef WTMP_ONLY_ON_LOGIN |
310 | #ifdef LOG_ONLY_ON_LOGIN |
195 | if (R->Options & Opt_loginShell) |
311 | if (login_shell) |
196 | # endif |
312 | #endif |
197 | { |
313 | { |
198 | # ifdef HAVE_STRUCT_UTMP |
314 | # ifdef HAVE_STRUCT_UTMP |
199 | # ifdef HAVE_UPDWTMP |
315 | # ifdef HAVE_UPDWTMP |
200 | updwtmp(RXVT_WTMP_FILE, ut); |
316 | updwtmp (WTMP_FILE, ut); |
201 | # else |
317 | # else |
202 | rxvt_update_wtmp(RXVT_WTMP_FILE, ut); |
318 | update_wtmp (WTMP_FILE, ut); |
203 | # endif |
319 | # endif |
204 | # endif |
320 | # endif |
205 | # ifdef HAVE_STRUCT_UTMPX |
321 | # if defined(HAVE_STRUCT_UTMPX) && defined(HAVE_UPDWTMPX) |
206 | updwtmpx(RXVT_WTMPX_FILE, utx); |
322 | updwtmpx (WTMPX_FILE, utx); |
207 | # endif |
|
|
208 | } |
|
|
209 | #endif |
323 | # endif |
|
|
324 | } |
|
|
325 | #endif |
210 | #if defined(LASTLOG_SUPPORT) && defined(RXVT_LASTLOG_FILE) |
326 | #if defined(LASTLOG_SUPPORT) && defined(LASTLOG_FILE) |
211 | if (R->Options & Opt_loginShell) |
327 | #ifdef LOG_ONLY_ON_LOGIN |
|
|
328 | if (login_shell) |
|
|
329 | #endif |
212 | rxvt_update_lastlog(RXVT_LASTLOG_FILE, pty, hostname); |
330 | update_lastlog (LASTLOG_FILE, pty, hostname); |
213 | #endif |
331 | #endif |
214 | } |
332 | } |
215 | |
333 | |
216 | /* ------------------------------------------------------------------------- */ |
334 | /* ------------------------------------------------------------------------- */ |
217 | /* |
335 | /* |
218 | * remove utmp and wtmp entries |
336 | * remove utmp and wtmp entries |
219 | */ |
337 | */ |
220 | /* EXTPROTO */ |
|
|
221 | void |
338 | void |
222 | rxvt_cleanutent(pR) |
339 | ptytty_unix::logout () |
223 | { |
340 | { |
224 | #ifdef HAVE_STRUCT_UTMP |
341 | if (!cmd_pid) |
225 | struct utmp *ut = &(R->h->ut); |
342 | return; |
226 | #endif |
343 | |
227 | #ifdef HAVE_STRUCT_UTMPX |
344 | #ifdef HAVE_STRUCT_UTMP |
|
|
345 | struct utmp *tmput, *ut = &this->ut; |
|
|
346 | #endif |
|
|
347 | #ifdef HAVE_STRUCT_UTMPX |
228 | struct utmpx *tmputx, *utx = &(R->h->utx); |
348 | struct utmpx *tmputx, *utx = &this->utx; |
229 | #endif |
349 | #endif |
230 | |
350 | |
231 | #ifdef HAVE_STRUCT_UTMP |
351 | #ifdef HAVE_STRUCT_UTMP |
232 | # ifdef HAVE_UTMP_PID |
352 | # ifdef HAVE_UTMP_PID |
233 | MEMSET(ut, 0, sizeof(struct utmp)); |
353 | memset (ut, 0, sizeof (struct utmp)); |
234 | setutent(); |
354 | setutent (); |
235 | STRNCPY(ut->ut_id, R->h->ut_id, sizeof(ut->ut_id)); |
355 | strncpy (ut->ut_id, this->ut_id, sizeof (ut->ut_id)); |
236 | ut->ut_type = USER_PROCESS; |
356 | ut->ut_type = USER_PROCESS; |
237 | { |
|
|
238 | struct utmp *tmput = getutid(ut); |
|
|
239 | |
|
|
240 | if (tmput) /* position to entry in utmp file */ |
357 | if ((tmput = getutid (ut))) /* position to entry in utmp file */ |
241 | ut = tmput; |
358 | ut = tmput; |
242 | } |
|
|
243 | ut->ut_type = DEAD_PROCESS; |
359 | ut->ut_type = DEAD_PROCESS; |
244 | # else |
360 | # else |
245 | MEMSET(ut->ut_name, 0, sizeof(ut->ut_name)); |
361 | memset (ut->ut_name, 0, sizeof (ut->ut_name)); |
246 | # ifdef HAVE_UTMP_HOST |
362 | # ifdef HAVE_UTMP_HOST |
247 | MEMSET(ut->ut_host, 0, sizeof(ut->ut_host)); |
363 | memset (ut->ut_host, 0, sizeof (ut->ut_host)); |
248 | # endif |
364 | # endif |
249 | # endif |
365 | # endif |
250 | ut->ut_time = time(NULL); |
366 | ut->ut_time = time (NULL); |
251 | #endif |
367 | #endif |
252 | |
368 | |
253 | #ifdef HAVE_STRUCT_UTMPX |
369 | #ifdef HAVE_STRUCT_UTMPX |
254 | MEMSET(utx, 0, sizeof(struct utmpx)); |
370 | memset (utx, 0, sizeof (struct utmpx)); |
255 | setutxent(); |
371 | setutxent (); |
256 | STRNCPY(utx->ut_id, R->h->ut_id, sizeof(utx->ut_id)); |
372 | strncpy (utx->ut_id, this->ut_id, sizeof (utx->ut_id)); |
257 | utx->ut_type = USER_PROCESS; |
373 | utx->ut_type = USER_PROCESS; |
258 | if ((tmputx = getutxid(utx))) /* position to entry in utmp file */ |
374 | if ((tmputx = getutxid (utx))) /* position to entry in utmp file */ |
259 | utx = tmputx; |
375 | utx = tmputx; |
260 | utx->ut_type = DEAD_PROCESS; |
376 | utx->ut_type = DEAD_PROCESS; |
|
|
377 | # if HAVE_UTMPX_SESSION |
261 | utx->ut_session = getsid(0); |
378 | utx->ut_session = getsid (0); |
|
|
379 | # endif |
262 | utx->ut_tv.tv_sec = time(NULL); |
380 | utx->ut_tv.tv_sec = time (NULL); |
263 | utx->ut_tv.tv_usec = 0; |
381 | utx->ut_tv.tv_usec = 0; |
264 | #endif |
382 | #endif |
265 | |
383 | |
266 | /* |
384 | /* |
267 | * Write ending wtmp entry |
385 | * Write ending wtmp entry |
268 | */ |
386 | */ |
269 | #ifdef WTMP_SUPPORT |
387 | #ifdef WTMP_SUPPORT |
270 | # ifdef WTMP_ONLY_ON_LOGIN |
388 | #ifdef LOG_ONLY_ON_LOGIN |
271 | if (R->Options & Opt_loginShell) |
389 | if (login_shell) |
272 | # endif |
390 | #endif |
273 | { |
391 | { |
274 | # ifdef HAVE_STRUCT_UTMP |
392 | # ifdef HAVE_STRUCT_UTMP |
275 | # ifdef HAVE_UPDWTMP |
393 | # ifdef HAVE_UPDWTMP |
276 | updwtmp(RXVT_WTMP_FILE, ut); |
394 | updwtmp (WTMP_FILE, ut); |
277 | # else |
395 | # else |
278 | rxvt_update_wtmp(RXVT_WTMP_FILE, ut); |
396 | update_wtmp (WTMP_FILE, ut); |
279 | # endif |
397 | # endif |
280 | # endif |
398 | # endif |
281 | # ifdef HAVE_STRUCT_UTMPX |
399 | # if defined(HAVE_STRUCT_UTMPX) && defined(HAVE_UPDWTMPX) |
282 | updwtmpx(RXVT_WTMPX_FILE, utx); |
400 | updwtmpx (WTMPX_FILE, utx); |
283 | # endif |
|
|
284 | } |
|
|
285 | #endif |
401 | # endif |
|
|
402 | } |
|
|
403 | #endif |
286 | |
404 | |
287 | /* |
405 | /* |
288 | * Write utmp entry |
406 | * Write utmp entry |
289 | */ |
407 | */ |
290 | #ifdef HAVE_STRUCT_UTMP |
408 | #ifdef HAVE_STRUCT_UTMP |
291 | # ifdef HAVE_UTMP_PID |
409 | # ifdef HAVE_UTMP_PID |
292 | if (ut->ut_pid == R->h->cmd_pid) |
410 | if (ut->ut_pid == cmd_pid) |
293 | pututline(ut); |
411 | pututline (ut); |
294 | endutent(); |
412 | endutent (); |
295 | # else |
413 | # else |
296 | MEMSET(ut, 0, sizeof(struct utmp)); |
414 | memset (ut, 0, sizeof (struct utmp)); |
297 | rxvt_write_bsd_utmp(R->h->utmp_pos, ut); |
415 | write_bsd_utmp (utmp_pos, ut); |
298 | # endif |
|
|
299 | #endif |
416 | # endif |
|
|
417 | #endif |
300 | #ifdef HAVE_STRUCT_UTMPX |
418 | #ifdef HAVE_STRUCT_UTMPX |
301 | if (utx->ut_pid == R->h->cmd_pid) |
419 | if (utx->ut_pid == cmd_pid) |
302 | pututxline(utx); |
420 | pututxline (utx); |
303 | endutxent(); |
421 | endutxent (); |
304 | #endif |
422 | #endif |
305 | } |
|
|
306 | |
423 | |
307 | /* ------------------------------------------------------------------------- */ |
424 | 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 | } |
425 | } |
326 | #endif |
|
|
327 | |
426 | |
328 | /* ------------------------------------------------------------------------- */ |
427 | #else |
329 | /* |
|
|
330 | * Update a BSD style wtmp entry |
|
|
331 | */ |
|
|
332 | #if defined(WTMP_SUPPORT) && !defined(HAVE_UPDWTMP) |
|
|
333 | /* INTPROTO */ |
|
|
334 | void |
428 | void |
335 | rxvt_update_wtmp(const char *fname, const struct utmp *putmp) |
429 | ptytty_unix::login (int cmd_pid, bool login_shell, const char *hostname) |
336 | { |
430 | { |
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 | } |
431 | } |
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 |
432 | #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 | |
433 | |
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 */ |
|
|