2013-02-13 12 views
9

Ho un compito su cui sto lavorando e ho difficoltà a completarlo. L'idea è di scrivere un programma if.c che esegua un programma e, se riesce, esegue il secondo programma. Dovrei sopprimere l'output standard del primo programma e l'output standard di unsupress per il secondo. Ricevo il messaggio di errore su più test. Ad esempio: "./if echo no then echo yes" restituisce "echo: errore di scrittura: descrittore di file errato". Ho provato a trovare quello che sto facendo male online ma senza fortuna.In C, come faccio a reindirizzare STDOUT_FILENO in/dev/null usando dup2 e poi reindirizzare nuovamente al suo valore originale in seguito?

Ecco il mio codice:

#include <fcntl.h> 
#include <sys/wait.h> 
#include <stdio.h> 
#include "tlpi_hdr.h" 

int main(int argc, char *argv[]) 
{ 
    if(argc < 4){ 
     fprintf(stderr,"Incorrect number of arguments.\n"); 
     exit(EXIT_FAILURE); 
    } 

    int thenArg = 0; 
    char then[4]; 
    strcpy(then,"then"); 
    for(int x=1; x<argc; x++){ 
     if(strncmp(argv[x], then, 4) == 0) thenArg = x; 
    } 

    if(thenArg == 0){ 
     fprintf(stderr,"No 'then' argument found.\n"); 
     exit(EXIT_FAILURE); 
    } 

    int save_out = dup(STDOUT_FILENO); 
    if(save_out == -1){ 
     fprintf(stderr,"Error in dup(STDOUT_FILENO)\n"); 
     exit(EXIT_FAILURE); 
    } 

    int devNull = open("/dev/null",0); 
    if(devNull == -1){ 
     fprintf(stderr,"Error in open('/dev/null',0)\n"); 
     exit(EXIT_FAILURE); 
    } 

    int dup2Result = dup2(devNull, STDOUT_FILENO); 
    if(dup2Result == -1) { 
     fprintf(stderr,"Error in dup2(devNull, STDOUT_FILENO)\n"); 
     exit(EXIT_FAILURE); 
    } 

    int program1argLocation = 1; 
    int program2argLocation = thenArg + 1; 
    int program1argCount = thenArg-1; 
    int program2argCount = argc-(program2argLocation); 
    char *program1args[program1argCount+1]; 
    char *program2args[program2argCount+1]; 

    for(int i=0; i<program1argCount; i++){ 
     program1args[i]=argv[program1argLocation + i]; 
    } 
    program1args[program1argCount] = NULL; 
    for(int i=0; i<program2argCount; i++){ 
     program2args[i]=argv[program2argLocation + i]; 
    } 
    program2args[program2argCount] = NULL; 

    pid_t pid = fork(); 
    int child_status; 
    switch (pid) { 
    case -1: 
     fprintf(stderr,"Fork failed\n"); 
     exit(EXIT_FAILURE); 

    case 0: //child 
     //child will run program 1 
     if(execvp(program1args[0],&program1args[0]) == -1){ 
      fprintf(stderr,"Program 1 Failed.\n"); 
      exit(EXIT_FAILURE); 
     } 

    default: //parent 
     //parent will run program2 
     pid = wait(&child_status); 

     if(WEXITSTATUS(child_status) == 0){ 
      dup2(save_out, STDOUT_FILENO); 

      int prog2status = execvp(program2args[0],&program2args[0]); 
      if(prog2status == -1) { 
       fprintf(stderr,"Program 2 failed.\n"); 
       exit(EXIT_FAILURE); 
      } 
     } 
    } 

} 
+0

'char poi [ 4]; strcpy (then, "then"); 'è un buffer overflow. Perché non usi semplicemente 'if (0 == strcmp (argv [x]," then "))' o 'if (0 == strncmp (argv [x]," then ", 4))' invece? – nneonneo

+0

Penso di averlo provato la prima volta che l'ho implementato e mi sono imbattuto in una sorta di errore. Mi ricorderò di questo e mi riferirò ad esso la prossima volta che incontrerò la stessa cosa. Grazie! Non è abituato a C, quindi qualsiasi input come questo è grandioso. – Frank

risposta

15

Il tuo errore è qui:

int devNull = open("/dev/null",0); 

Per utilizzare devNull come STDOUT_FILENO, deve essere aperto per la scrittura:

int devNull = open("/dev/null", O_WRONLY); 
+0

Grazie! Ho effettivamente trovato la stessa soluzione pochi istanti prima che tu rispondessi. Ho provato a forzarti, ma a quanto pare non posso farlo, ancora. Sono entusiasta di utilizzare l'overflow dello stack più in futuro! Questa è una risorsa incredibile. – Frank