/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 plugin-runner.c

  • Committer: Teddy Hogeborn
  • Date: 2008-08-24 23:33:02 UTC
  • mfrom: (24.1.65 mandos)
  • Revision ID: teddy@fukt.bsnet.se-20080824233302-kfu4g2pkozg7xrmz
* plugin-runner.c (process): Set "completed" and "status" as volatile.
  (handle_sigchld): Bug fix: reset proc to process_list every time.

Show diffs side-by-side

added added

removed removed

Lines of Context:
plugin-runner.c
56
56
#include <argp.h>               /* struct argp_option, struct
57
57
                                   argp_state, struct argp,
58
58
                                   argp_parse(), ARGP_ERR_UNKNOWN,
59
 
                                   ARGP_KEY_END, ARGP_KEY_ARG, error_t */
 
59
                                   ARGP_KEY_END, ARGP_KEY_ARG,
 
60
                                   error_t */
60
61
#include <signal.h>             /* struct sigaction, sigemptyset(),
61
62
                                   sigaddset(), sigaction(),
62
63
                                   sigprocmask(), SIG_BLOCK, SIGCHLD,
78
79
  size_t buffer_size;
79
80
  size_t buffer_length;
80
81
  bool eof;
81
 
  bool completed;
82
 
  int status;
 
82
  volatile bool completed;
 
83
  volatile int status;
83
84
  struct process *next;
84
85
} process;
85
86
 
199
200
 
200
201
process *process_list = NULL;
201
202
 
202
 
/* Mark a process as completed when it exits, and save its exit
 
203
/* Mark processes as completed when they exit, and save their exit
203
204
   status. */
204
205
void handle_sigchld(__attribute__((unused)) int sig){
205
 
  process *proc = process_list;
206
 
  int status;
207
 
  pid_t pid = wait(&status);
208
 
  if(pid == -1){
209
 
    perror("wait");
210
 
    return;
211
 
  }
212
 
  while(proc != NULL and proc->pid != pid){
213
 
    proc = proc->next;
214
 
  }
215
 
  if(proc == NULL){
216
 
    /* Process not found in process list */
217
 
    return;
218
 
  }
219
 
  proc->status = status;
220
 
  proc->completed = true;
 
206
  while(true){
 
207
    process *proc = process_list;
 
208
    int status;
 
209
    pid_t pid = waitpid(-1, &status, WNOHANG);
 
210
    if(pid == 0){
 
211
      /* Only still running child processes */
 
212
      break;
 
213
    }
 
214
    if(pid == -1){
 
215
      if (errno != ECHILD){
 
216
        perror("waitpid");
 
217
      }
 
218
      /* No child processes */
 
219
      break;
 
220
    }
 
221
 
 
222
    /* A child exited, find it in process_list */
 
223
    while(proc != NULL and proc->pid != pid){
 
224
      proc = proc->next;
 
225
    }
 
226
    if(proc == NULL){
 
227
      /* Process not found in process list */
 
228
      continue;
 
229
    }
 
230
    proc->status = status;
 
231
    proc->completed = true;
 
232
  }
221
233
}
222
234
 
223
235
bool print_out_password(const char *buffer, size_t length){
242
254
    if(argv == NULL){
243
255
      return NULL;
244
256
    }
245
 
    argv[0] = NULL;     /* Will be set to argv[0] in main before parsing */
 
257
    argv[0] = NULL;     /* Will be set to argv[0] in main before
 
258
                           parsing */
246
259
    argv[1] = NULL;
247
260
  }
248
261
  *argc += 1;
501
514
 
502
515
  if(custom_argv != NULL){
503
516
    custom_argv[0] = argv[0];
504
 
    ret = argp_parse (&argp, custom_argc, custom_argv, 0, 0, &plugin_list);
 
517
    ret = argp_parse (&argp, custom_argc, custom_argv, 0, 0,
 
518
                      &plugin_list);
505
519
    if (ret == ARGP_ERR_UNKNOWN){
506
520
      fprintf(stderr, "Unknown error while parsing arguments\n");
507
521
      exitstatus = EXIT_FAILURE;
835
849
          /* Remove the plugin */
836
850
          FD_CLR(proc->fd, &rfds_all);
837
851
          /* Block signal while modifying process_list */
838
 
          ret = sigprocmask (SIG_BLOCK, &sigchld_action.sa_mask, NULL);
 
852
          ret = sigprocmask(SIG_BLOCK, &sigchld_action.sa_mask, NULL);
839
853
          if(ret < 0){
840
854
            perror("sigprocmask");
841
855
            exitstatus = EXIT_FAILURE;
869
883
        }
870
884
        /* This process exited nicely, so print its buffer */
871
885
 
872
 
        bool bret = print_out_password(proc->buffer, proc->buffer_length);
 
886
        bool bret = print_out_password(proc->buffer,
 
887
                                       proc->buffer_length);
873
888
        if(not bret){
874
889
          perror("print_out_password");
875
890
          exitstatus = EXIT_FAILURE;
912
927
 fallback:
913
928
  
914
929
  if(process_list == NULL or exitstatus != EXIT_SUCCESS){
915
 
    /* Fallback if all plugins failed, none are found or an error occured */
 
930
    /* Fallback if all plugins failed, none are found or an error
 
931
       occured */
916
932
    bool bret;
917
933
    fprintf(stderr, "Going to fallback mode using getpass(3)\n");
918
934
    char *passwordbuffer = getpass("Password: ");