1/* Copyright (C) 1996-2020 Free Software Foundation, Inc.
2 This file is part of the GNU C Library.
3 Contributed by Ulrich Drepper <drepper@cygnus.com>, 1996.
4
5 The GNU C Library is free software; you can redistribute it and/or
6 modify it under the terms of the GNU Lesser General Public
7 License as published by the Free Software Foundation; either
8 version 2.1 of the License, or (at your option) any later version.
9
10 The GNU C Library is distributed in the hope that it will be useful,
11 but WITHOUT ANY WARRANTY; without even the implied warranty of
12 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
13 Lesser General Public License for more details.
14
15 You should have received a copy of the GNU Lesser General Public
16 License along with the GNU C Library; if not, see
17 <https://www.gnu.org/licenses/>. */
18
19#include <assert.h>
20#include <errno.h>
21#include <limits.h>
22#include <string.h>
23#include <unistd.h>
24#include <stdlib.h>
25#include <utmp.h>
26
27
28/* Return the result of ttyname in the buffer pointed to by TTY, which should
29 be of length BUF_LEN. If it is too long to fit in this buffer, a
30 sufficiently long buffer is allocated using malloc, and returned in TTY.
31 0 is returned upon success, -1 otherwise. */
32static int
33tty_name (int fd, char **tty, size_t buf_len)
34{
35 int rv; /* Return value. 0 = success. */
36 char *buf = *tty; /* Buffer for ttyname, initially the user's. */
37
38 for (;;)
39 {
40 char *new_buf;
41
42 if (buf_len)
43 {
44 rv = ttyname_r (fd, buf, buf_len);
45
46 if (rv != 0 || memchr (buf, '\0', buf_len))
47 /* We either got an error, or we succeeded and the
48 returned name fit in the buffer. */
49 break;
50
51 /* Try again with a longer buffer. */
52 buf_len += buf_len; /* Double it */
53 }
54 else
55 /* No initial buffer; start out by mallocing one. */
56 buf_len = 128; /* First time guess. */
57
58 if (buf != *tty)
59 /* We've already malloced another buffer at least once. */
60 new_buf = realloc (buf, buf_len);
61 else
62 new_buf = malloc (buf_len);
63 if (! new_buf)
64 {
65 rv = -1;
66 __set_errno (ENOMEM);
67 break;
68 }
69 buf = new_buf;
70 }
71
72 if (rv == 0)
73 *tty = buf; /* Return buffer to the user. */
74 else if (buf != *tty)
75 free (buf); /* Free what we malloced when returning an error. */
76
77 return rv;
78}
79
80void
81login (const struct utmp *ut)
82{
83#ifdef PATH_MAX
84 char _tty[PATH_MAX + UT_LINESIZE];
85#else
86 char _tty[512 + UT_LINESIZE];
87#endif
88 char *tty = _tty;
89 int found_tty;
90 const char *ttyp;
91 struct utmp copy = *ut;
92
93 /* Fill in those fields we supply. */
94 copy.ut_type = USER_PROCESS;
95 copy.ut_pid = getpid ();
96
97 /* Seek tty. */
98 found_tty = tty_name (STDIN_FILENO, &tty, sizeof (_tty));
99 if (found_tty < 0)
100 found_tty = tty_name (STDOUT_FILENO, &tty, sizeof (_tty));
101 if (found_tty < 0)
102 found_tty = tty_name (STDERR_FILENO, &tty, sizeof (_tty));
103
104 if (found_tty >= 0)
105 {
106 /* We only want to insert the name of the tty without path.
107 But take care of name like /dev/pts/3. */
108 if (strncmp (tty, "/dev/", 5) == 0)
109 ttyp = tty + 5; /* Skip the "/dev/". */
110 else
111 ttyp = basename (tty);
112
113 /* Position to record for this tty. */
114 strncpy (copy.ut_line, ttyp, UT_LINESIZE);
115
116 /* Tell that we want to use the UTMP file. */
117 if (utmpname (_PATH_UTMP) == 0)
118 {
119 /* Open UTMP file. */
120 setutent ();
121
122 /* Write the entry. */
123 pututline (&copy);
124
125 /* Close UTMP file. */
126 endutent ();
127 }
128
129 if (tty != _tty)
130 free (tty); /* Free buffer malloced by tty_name. */
131 }
132 else
133 /* We provide a default value so that the output does not contain
134 an random bytes in this field. */
135 strncpy (copy.ut_line, "???", UT_LINESIZE);
136
137 /* Update the WTMP file. Here we have to add a new entry. */
138 updwtmp (_PATH_WTMP, &copy);
139}
140