/mandos/trunk

To get this branch, use:
bzr branch http://bzr.recompile.se/loggerhead/mandos/trunk

« back to all changes in this revision

Viewing changes to plugins.d/usplash.c

  • Committer: Teddy Hogeborn
  • Date: 2008-09-26 21:54:54 UTC
  • Revision ID: teddy@fukt.bsnet.se-20080926215454-7kfrl0gxmisz71gw
* mandos: Remove unused "select" module.  Import "ctypes.util".
  (AvahiError.__init__): Call __init__ of superclass.
  (AvahiService.__init__): Rename keyword arg "type" to "servicetype".
  (AvahiService.rename): Bug fix: use "self.rename_count".
  (Client.__init__): Use "None" instead of "{}" as default value.
  (tcp_handler): Renamed to "TCP_handler".  All callers changed.
  (IPv6_TCPServer.__init__): Don't "return", just call the superclass
                             method.
  (if_nametoindex): Don't import "ctypes.util" here.
  (main): Removed unused variable "main_loop_started".

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
/*  -*- coding: utf-8 -*- */
2
 
/*
3
 
 * Usplash - Read a password from usplash and output it
4
 
 * 
5
 
 * Copyright © 2008,2009 Teddy Hogeborn
6
 
 * Copyright © 2008,2009 Björn Påhlsson
7
 
 * 
8
 
 * This program is free software: you can redistribute it and/or
9
 
 * modify it under the terms of the GNU General Public License as
10
 
 * published by the Free Software Foundation, either version 3 of the
11
 
 * License, or (at your option) any later version.
12
 
 * 
13
 
 * This program is distributed in the hope that it will be useful, but
14
 
 * WITHOUT ANY WARRANTY; without even the implied warranty of
15
 
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
16
 
 * General Public License for more details.
17
 
 * 
18
 
 * You should have received a copy of the GNU General Public License
19
 
 * along with this program.  If not, see
20
 
 * <http://www.gnu.org/licenses/>.
21
 
 * 
22
 
 * Contact the authors at <https://www.fukt.bsnet.se/~belorn/> and
23
 
 * <https://www.fukt.bsnet.se/~teddy/>.
24
 
 */
25
 
 
26
1
#define _GNU_SOURCE             /* asprintf() */
27
2
#include <signal.h>             /* sig_atomic_t, struct sigaction,
28
3
                                   sigemptyset(), sigaddset(), SIGINT,
30
5
                                   SIG_IGN, kill(), SIGKILL */
31
6
#include <stdbool.h>            /* bool, false, true */
32
7
#include <fcntl.h>              /* open(), O_WRONLY, O_RDONLY */
33
 
#include <iso646.h>             /* and, or, not*/
34
8
#include <errno.h>              /* errno, EINTR */
35
 
#include <sys/types.h>          /* size_t, ssize_t, pid_t, DIR, struct
36
 
                                   dirent */
 
9
#include <sys/types.h>          /* size_t, ssize_t, pid_t, DIR,
 
10
                                   struct dirent */
37
11
#include <stddef.h>             /* NULL */
38
12
#include <string.h>             /* strlen(), memcmp() */
39
 
#include <stdio.h>              /* asprintf(), perror(), sscanf() */
 
13
#include <stdio.h>              /* asprintf(), perror() */
40
14
#include <unistd.h>             /* close(), write(), readlink(),
41
15
                                   read(), STDOUT_FILENO, sleep(),
42
16
                                   fork(), setuid(), geteuid(),
43
17
                                   setsid(), chdir(), dup2(),
44
18
                                   STDERR_FILENO, execv() */
45
 
#include <stdlib.h>             /* free(), EXIT_FAILURE, realloc(),
46
 
                                   EXIT_SUCCESS, malloc(), _exit() */
 
19
#include <stdlib.h>             /* free(), EXIT_FAILURE, strtoul(),
 
20
                                   realloc(), EXIT_SUCCESS, malloc(),
 
21
                                   _exit() */
47
22
#include <stdlib.h>             /* getenv() */
48
23
#include <dirent.h>             /* opendir(), readdir(), closedir() */
49
 
#include <inttypes.h>           /* intmax_t, SCNdMAX */
50
 
#include <sys/stat.h>           /* struct stat, lstat(), S_ISLNK */
 
24
 
 
25
 
 
26
 
 
27
#include <iso646.h>             /* and */
 
28
#include <sys/wait.h>           /* waitpid(), WIFEXITED(),
 
29
                                   WEXITSTATUS() */
51
30
 
52
31
sig_atomic_t interrupted_by_signal = 0;
53
32
 
57
36
 
58
37
static bool usplash_write(const char *cmd, const char *arg){
59
38
  /* 
60
 
   * usplash_write("TIMEOUT", "15") will write "TIMEOUT 15\0"
61
 
   * usplash_write("PULSATE", NULL) will write "PULSATE\0"
 
39
   * usplash_write("TIMEOUT", "15"); -> "TIMEOUT 15\0"
 
40
   * usplash_write("PULSATE", NULL); -> "PULSATE\0"
62
41
   * SEE ALSO
63
42
   *         usplash_write(8)
64
43
   */
92
71
  }
93
72
  
94
73
  size_t written = 0;
95
 
  ssize_t sret = 0;
96
74
  while(not interrupted_by_signal and written < cmd_line_len){
97
 
    sret = write(fifo_fd, cmd_line + written,
98
 
                 cmd_line_len - written);
99
 
    if(sret == -1){
 
75
    ret = write(fifo_fd, cmd_line + written,
 
76
                cmd_line_len - written);
 
77
    if(ret == -1){
100
78
      if(errno != EINTR or interrupted_by_signal){
101
79
        int e = errno;
102
80
        close(fifo_fd);
107
85
        continue;
108
86
      }
109
87
    }
110
 
    written += (size_t)sret;
 
88
    written += (size_t)ret;
111
89
  }
112
90
  free(cmd_line_alloc);
113
91
  do{
170
148
    for(struct dirent *proc_ent = readdir(proc_dir);
171
149
        proc_ent != NULL;
172
150
        proc_ent = readdir(proc_dir)){
173
 
      pid_t pid;
174
 
      {
175
 
        intmax_t tmpmax;
176
 
        int numchars;
177
 
        ret = sscanf(proc_ent->d_name, "%" SCNdMAX "%n", &tmpmax,
178
 
                     &numchars);
179
 
        if(ret < 1 or tmpmax != (pid_t)tmpmax
180
 
           or proc_ent->d_name[numchars] != '\0'){
181
 
          /* Not a process */
182
 
          continue;
183
 
        }
184
 
        pid = (pid_t)tmpmax;
 
151
      pid_t pid = (pid_t) strtoul(proc_ent->d_name, NULL, 10);
 
152
      if(pid == 0){
 
153
        /* Not a process */
 
154
        continue;
185
155
      }
186
156
      /* Find the executable name by doing readlink() on the
187
157
         /proc/<pid>/exe link */
188
158
      char exe_target[sizeof(usplash_name)];
189
159
      {
190
 
        /* create file name string */
191
160
        char *exe_link;
192
161
        ret = asprintf(&exe_link, "/proc/%s/exe", proc_ent->d_name);
193
162
        if(ret == -1){
196
165
          closedir(proc_dir);
197
166
          return EXIT_FAILURE;
198
167
        }
199
 
        
200
 
        /* Check that it refers to a symlink owned by root:root */
201
 
        struct stat exe_stat;
202
 
        ret = lstat(exe_link, &exe_stat);
203
 
        if(ret == -1){
204
 
          if(errno == ENOENT){
205
 
            free(exe_link);
206
 
            continue;
207
 
          }
208
 
          perror("lstat");
209
 
          free(exe_link);
210
 
          free(prompt);
211
 
          closedir(proc_dir);
212
 
          return EXIT_FAILURE;
213
 
        }
214
 
        if(not S_ISLNK(exe_stat.st_mode)
215
 
           or exe_stat.st_uid != 0
216
 
           or exe_stat.st_gid != 0){
217
 
          free(exe_link);
218
 
          continue;
219
 
        }
220
 
        
221
168
        sret = readlink(exe_link, exe_target, sizeof(exe_target));
222
169
        free(exe_link);
 
170
        if(sret == -1){
 
171
          continue;
 
172
        }
223
173
      }
224
174
      if((sret == ((ssize_t)sizeof(exe_target)-1))
225
175
         and (memcmp(usplash_name, exe_target,
303
253
      free(prompt);
304
254
      return EXIT_FAILURE;
305
255
    }
306
 
    if(old_action.sa_handler != SIG_IGN){
 
256
    if (old_action.sa_handler != SIG_IGN){
307
257
      ret = sigaction(SIGINT, &new_action, NULL);
308
258
      if(ret == -1){
309
259
        perror("sigaction");
317
267
      free(prompt);
318
268
      return EXIT_FAILURE;
319
269
    }
320
 
    if(old_action.sa_handler != SIG_IGN){
 
270
    if (old_action.sa_handler != SIG_IGN){
321
271
      ret = sigaction(SIGHUP, &new_action, NULL);
322
272
      if(ret == -1){
323
273
        perror("sigaction");
331
281
      free(prompt);
332
282
      return EXIT_FAILURE;
333
283
    }
334
 
    if(old_action.sa_handler != SIG_IGN){
 
284
    if (old_action.sa_handler != SIG_IGN){
335
285
      ret = sigaction(SIGTERM, &new_action, NULL);
336
286
      if(ret == -1){
337
287
        perror("sigaction");
457
407
      return EXIT_SUCCESS;
458
408
    }
459
409
    break;                      /* Big */
460
 
  }                             /* end of non-loop while() */
 
410
  }
461
411
  
462
412
  /* If we got here, an error or interrupt must have happened */
463
413
  
464
 
  /* Create argc and argv for new usplash*/
465
414
  int cmdline_argc = 0;
466
415
  char **cmdline_argv = malloc(sizeof(char *));
 
416
  /* Create argv and argc for new usplash*/
467
417
  {
468
418
    size_t position = 0;
469
419
    while(position < cmdline_len){
483
433
    cmdline_argv[cmdline_argc] = NULL;
484
434
  }
485
435
  /* Kill old usplash */
486
 
  kill(usplash_pid, SIGTERM);
487
 
  sleep(2);
 
436
    kill(usplash_pid, SIGTERM);
 
437
    sleep(2);
488
438
  while(kill(usplash_pid, 0) == 0){
489
439
    kill(usplash_pid, SIGKILL);
490
440
    sleep(1);
494
444
    /* Child; will become new usplash process */
495
445
    
496
446
    /* Make the effective user ID (root) the only user ID instead of
497
 
       the real user ID (_mandos) */
 
447
       the real user ID (mandos) */
498
448
    ret = setuid(geteuid());
499
 
    if(ret == -1){
 
449
    if (ret == -1){
500
450
      perror("setuid");
501
451
    }
502
452
    
512
462
    }
513
463
    
514
464
    execv(usplash_name, cmdline_argv);
515
 
    if(not interrupted_by_signal){
516
 
      perror("execv");
517
 
    }
 
465
    perror("execv");
518
466
    free(cmdline);
519
467
    free(cmdline_argv);
520
468
    _exit(EXIT_FAILURE);