← Back to branch summary

~ubuntu-branches/ubuntu/natty/eglibc/natty-security

« back to all changes in this revision

Viewing changes to login/tst-utmp.c

  • Committer: Bazaar Package Importer
  • Author(s): Aurelien Jarno
  • Date: 2009-05-05 09:54:14 UTC
  • Revision ID: james.westby@ubuntu.com-20090505095414-c45qsg9ixjheohru
Import upstream version 2.9

Show diffs side-by-side

added added

removed removed

Lines of Context:
login/tst-utmp.c
 
1
/* Tests for UTMP functions.
 
2
   Copyright (C) 1998, 2001-2003 Free Software Foundation, Inc.
 
3
   This file is part of the GNU C Library.
 
4
   Contributed by Mark Kettenis <kettenis@phys.uva.nl>, 1998.
 
5
 
 
6
   The GNU C Library is free software; you can redistribute it and/or
 
7
   modify it under the terms of the GNU Lesser General Public
 
8
   License as published by the Free Software Foundation; either
 
9
   version 2.1 of the License, or (at your option) any later version.
 
10
 
 
11
   The GNU C Library is distributed in the hope that it will be useful,
 
12
   but WITHOUT ANY WARRANTY; without even the implied warranty of
 
13
   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 
14
   Lesser General Public License for more details.
 
15
 
 
16
   You should have received a copy of the GNU Lesser General Public
 
17
   License along with the GNU C Library; if not, write to the Free
 
18
   Software Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA
 
19
   02111-1307 USA.  */
 
20
 
 
21
#include <errno.h>
 
22
#include <error.h>
 
23
#include <stdlib.h>
 
24
#include <string.h>
 
25
#include <sys/types.h>
 
26
#include <time.h>
 
27
 
 
28
#ifdef UTMPX
 
29
# include <utmpx.h>
 
30
# define utmp utmpx
 
31
# define utmpname utmpxname
 
32
# define setutent setutxent
 
33
# define getutent getutxent
 
34
# define endutent endutxent
 
35
# define getutline getutxline
 
36
# define getutid getutxid
 
37
# define pututline pututxline
 
38
#else
 
39
# include <utmp.h>
 
40
#endif
 
41
 
 
42
 
 
43
#if _HAVE_UT_TYPE || defined UTMPX
 
44
 
 
45
/* Prototype for our test function.  */
 
46
static int do_test (int argc, char *argv[]);
 
47
 
 
48
/* We have a preparation function.  */
 
49
static void do_prepare (int argc, char *argv[]);
 
50
#define PREPARE do_prepare
 
51
 
 
52
/* This defines the `main' function and some more.  */
 
53
#include <test-skeleton.c>
 
54
 
 
55
 
 
56
/* These are for the temporary file we generate.  */
 
57
char *name;
 
58
int fd;
 
59
 
 
60
static void
 
61
do_prepare (int argc, char *argv[])
 
62
{
 
63
  size_t name_len;
 
64
 
 
65
  name_len = strlen (test_dir);
 
66
  name = malloc (name_len + sizeof ("/utmpXXXXXX"));
 
67
  mempcpy (mempcpy (name, test_dir, name_len),
 
68
           "/utmpXXXXXX", sizeof ("/utmpXXXXXX"));
 
69
  add_temp_file (name);
 
70
 
 
71
  /* Open our test file.  */
 
72
  fd = mkstemp (name);
 
73
  if (fd == -1)
 
74
    error (EXIT_FAILURE, errno, "cannot open test file `%s'", name);
 
75
}
 
76
 
 
77
struct utmp entry[] =
 
78
{
 
79
#if _HAVE_UT_TV || defined UTMPX
 
80
#define UT(a)  .ut_tv = { .tv_sec = (a)}
 
81
#else
 
82
#define UT(a)  .ut_time = (a)
 
83
#endif
 
84
 
 
85
  { .ut_type = BOOT_TIME, .ut_pid = 1, UT(1000) },
 
86
  { .ut_type = RUN_LVL, .ut_pid = 1, UT(2000) },
 
87
  { .ut_type = INIT_PROCESS, .ut_pid = 5, .ut_id = "si", UT(3000) },
 
88
  { .ut_type = LOGIN_PROCESS, .ut_pid = 23, .ut_line = "tty1", .ut_id = "1",
 
89
    .ut_user = "LOGIN", UT(4000) },
 
90
  { .ut_type = USER_PROCESS, .ut_pid = 24, .ut_line = "tty2", .ut_id = "2",
 
91
    .ut_user = "albert", UT(8000) },
 
92
  { .ut_type = USER_PROCESS, .ut_pid = 196, .ut_line = "ttyp0", .ut_id = "p0",
 
93
    .ut_user = "niels", UT(10000) },
 
94
  { .ut_type = DEAD_PROCESS, .ut_line = "ttyp1", .ut_id = "p1", UT(16000) },
 
95
  { .ut_type = EMPTY },
 
96
  { .ut_type = EMPTY }
 
97
};
 
98
int num_entries = sizeof entry / sizeof (struct utmp);
 
99
 
 
100
time_t entry_time = 20000;
 
101
pid_t entry_pid = 234;
 
102
 
 
103
static int
 
104
do_init (void)
 
105
{
 
106
  int n;
 
107
 
 
108
  setutent ();
 
109
 
 
110
  for (n = 0; n < num_entries; n++)
 
111
    {
 
112
      if (pututline (&entry[n]) == NULL)
 
113
        {
 
114
          error (0, errno, "cannot write UTMP entry");
 
115
          return 1;
 
116
        }
 
117
    }
 
118
 
 
119
  endutent ();
 
120
 
 
121
  return 0;
 
122
}
 
123
 
 
124
 
 
125
static int
 
126
do_check (void)
 
127
{
 
128
  struct utmp *ut;
 
129
  int n;
 
130
 
 
131
  setutent ();
 
132
 
 
133
  n = 0;
 
134
  while ((ut = getutent ()))
 
135
    {
 
136
      if (n < num_entries &&
 
137
          memcmp (ut, &entry[n], sizeof (struct utmp)))
 
138
        {
 
139
          error (0, 0, "UTMP entry does not match");
 
140
          return 1;
 
141
        }
 
142
 
 
143
      n++;
 
144
    }
 
145
 
 
146
  if (n != num_entries)
 
147
    {
 
148
      error (0, 0, "number of UTMP entries is incorrect");
 
149
      return 1;
 
150
    }
 
151
 
 
152
  endutent ();
 
153
 
 
154
  return 0;
 
155
}
 
156
 
 
157
static int
 
158
simulate_login (const char *line, const char *user)
 
159
{
 
160
  int n;
 
161
 
 
162
  for (n = 0; n < num_entries; n++)
 
163
    {
 
164
      if (strcmp (line, entry[n].ut_line) == 0 ||
 
165
          entry[n].ut_type == DEAD_PROCESS)
 
166
        {
 
167
          if (entry[n].ut_pid == DEAD_PROCESS)
 
168
            entry[n].ut_pid = (entry_pid += 27);
 
169
          entry[n].ut_type = USER_PROCESS;
 
170
          strncpy (entry[n].ut_user, user, sizeof (entry[n].ut_user));
 
171
#if _HAVE_UT_TV - 0 || defined UTMPX
 
172
          entry[n].ut_tv.tv_sec = (entry_time += 1000);
 
173
#else
 
174
          entry[n].ut_time = (entry_time += 1000);
 
175
#endif
 
176
          setutent ();
 
177
 
 
178
          if (pututline (&entry[n]) == NULL)
 
179
            {
 
180
              error (0, errno, "cannot write UTMP entry");
 
181
              return 1;
 
182
            }
 
183
 
 
184
          endutent ();
 
185
 
 
186
          return 0;
 
187
        }
 
188
    }
 
189
 
 
190
  error (0, 0, "no entries available");
 
191
  return 1;
 
192
}
 
193
 
 
194
static int
 
195
simulate_logout (const char *line)
 
196
{
 
197
  int n;
 
198
 
 
199
  for (n = 0; n < num_entries; n++)
 
200
    {
 
201
      if (strcmp (line, entry[n].ut_line) == 0)
 
202
        {
 
203
          entry[n].ut_type = DEAD_PROCESS;
 
204
          strncpy (entry[n].ut_user, "", sizeof (entry[n].ut_user));
 
205
#if _HAVE_UT_TV - 0 || defined UTMPX
 
206
          entry[n].ut_tv.tv_sec = (entry_time += 1000);
 
207
#else
 
208
          entry[n].ut_time = (entry_time += 1000);
 
209
#endif
 
210
          setutent ();
 
211
 
 
212
          if (pututline (&entry[n]) == NULL)
 
213
            {
 
214
              error (0, errno, "cannot write UTMP entry");
 
215
              return 1;
 
216
            }
 
217
 
 
218
          endutent ();
 
219
 
 
220
          return 0;
 
221
        }
 
222
    }
 
223
 
 
224
  error (0, 0, "no entry found for `%s'", line);
 
225
  return 1;
 
226
}
 
227
 
 
228
static int
 
229
check_login (const char *line)
 
230
{
 
231
  struct utmp *up;
 
232
  struct utmp ut;
 
233
  int n;
 
234
 
 
235
  setutent ();
 
236
 
 
237
  strcpy (ut.ut_line, line);
 
238
  up = getutline (&ut);
 
239
  if (up == NULL)
 
240
    {
 
241
      error (0, errno, "cannot get entry for line `%s'", line);
 
242
      return 1;
 
243
    }
 
244
 
 
245
  endutent ();
 
246
 
 
247
  for (n = 0; n < num_entries; n++)
 
248
    {
 
249
      if (strcmp (line, entry[n].ut_line) == 0)
 
250
        {
 
251
          if (memcmp (up, &entry[n], sizeof (struct utmp)))
 
252
            {
 
253
              error (0, 0, "UTMP entry does not match");
 
254
              return 1;
 
255
            }
 
256
 
 
257
          return 0;
 
258
        }
 
259
    }
 
260
 
 
261
  error (0, 0, "bogus entry for line `%s'", line);
 
262
  return 1;
 
263
}
 
264
 
 
265
static int
 
266
check_logout (const char *line)
 
267
{
 
268
  struct utmp ut;
 
269
 
 
270
  setutent ();
 
271
 
 
272
  strcpy (ut.ut_line, line);
 
273
  if (getutline (&ut) != NULL)
 
274
    {
 
275
      error (0, 0, "bogus login entry for `%s'", line);
 
276
      return 1;
 
277
    }
 
278
 
 
279
  endutent ();
 
280
 
 
281
  return 0;
 
282
}
 
283
 
 
284
static int
 
285
check_id (const char *id)
 
286
{
 
287
  struct utmp *up;
 
288
  struct utmp ut;
 
289
  int n;
 
290
 
 
291
  setutent ();
 
292
 
 
293
  ut.ut_type = USER_PROCESS;
 
294
  strcpy (ut.ut_id, id);
 
295
  up = getutid (&ut);
 
296
  if (up == NULL)
 
297
    {
 
298
      error (0, errno, "cannot get entry for ID `%s'", id);
 
299
      return 1;
 
300
    }
 
301
 
 
302
  endutent ();
 
303
 
 
304
  for (n = 0; n < num_entries; n++)
 
305
    {
 
306
      if (strcmp (id, entry[n].ut_id) == 0)
 
307
        {
 
308
          if (memcmp (up, &entry[n], sizeof (struct utmp)))
 
309
            {
 
310
              error (0, 0, "UTMP entry does not match");
 
311
              return 1;
 
312
            }
 
313
 
 
314
          return 0;
 
315
        }
 
316
    }
 
317
 
 
318
  error (0, 0, "bogus entry for ID `%s'", id);
 
319
  return 1;
 
320
}
 
321
 
 
322
static int
 
323
check_type (int type)
 
324
{
 
325
  struct utmp *up;
 
326
  struct utmp ut;
 
327
  int n;
 
328
 
 
329
  setutent ();
 
330
 
 
331
  ut.ut_type = type;
 
332
  up = getutid (&ut);
 
333
  if (up == NULL)
 
334
    {
 
335
      error (0, errno, "cannot get entry for type `%d'", type);
 
336
      return 1;
 
337
    }
 
338
 
 
339
  endutent ();
 
340
 
 
341
  for (n = 0; n < num_entries; n++)
 
342
    {
 
343
      if (type == entry[n].ut_type)
 
344
        {
 
345
          if (memcmp (up, &entry[n], sizeof (struct utmp)))
 
346
            {
 
347
              error (0, 0, "UTMP entry does not match");
 
348
              return 1;
 
349
            }
 
350
 
 
351
          return 0;
 
352
        }
 
353
    }
 
354
 
 
355
  error (0, 0, "bogus entry for type `%d'", type);
 
356
  return 1;
 
357
}
 
358
 
 
359
static int
 
360
do_test (int argc, char *argv[])
 
361
{
 
362
  int result = 0;
 
363
 
 
364
  utmpname (name);
 
365
 
 
366
  result |= do_init ();
 
367
  result |= do_check ();
 
368
 
 
369
  result |= simulate_login ("tty1", "erwin");
 
370
  result |= do_check ();
 
371
 
 
372
  result |= simulate_login ("ttyp1", "paul");
 
373
  result |= do_check ();
 
374
 
 
375
  result |= simulate_logout ("tty2");
 
376
  result |= do_check ();
 
377
 
 
378
  result |= simulate_logout ("ttyp0");
 
379
  result |= do_check ();
 
380
 
 
381
  result |= simulate_login ("ttyp2", "richard");
 
382
  result |= do_check ();
 
383
 
 
384
  result |= check_login ("tty1");
 
385
  result |= check_logout ("ttyp0");
 
386
  result |= check_id ("p1");
 
387
  result |= check_id ("2");
 
388
  result |= check_id ("si");
 
389
  result |= check_type (BOOT_TIME);
 
390
  result |= check_type (RUN_LVL);
 
391
 
 
392
  return result;
 
393
}
 
394
 
 
395
#else
 
396
 
 
397
/* No field 'ut_type' in struct utmp.  */
 
398
int
 
399
main ()
 
400
{
 
401
  return 0;
 
402
}
 
403
 
 
404
#endif