コード例 #1
0
ファイル: shellp.c プロジェクト: shiift/CSE4095
void freeCommand(Command* c)
{
   if (c->_com) free(c->_com);
   if (c->_input) free(c->_input);
   if (c->_output) free(c->_output);
   for(int i=0;i < c->_nba;i++)
      if (c->_args[i]) free(c->_args[i]);
   free(c->_args);
   for(int j=0;j<c->_nbs;j++)
      freeStage(c->_stages[j]);
   if (c->_stages) free(c->_stages);
   free(c);
}
コード例 #2
0
ファイル: mush.c プロジェクト: Ryuho/CPE357
/** The main thing.
 * @param argc count of command-line tokens.
 * @param argv array of command-line tokens.
 * @return 0 on success, 1-255 on failure.
 */
int main(int argc, char* argv[]){
   struct stat stattemp;
   int fd;
   bool done = FALSE;
   char* commandLine;
   int* status;
   stageStruct** pStages;
   int** pipefd = NULL;
   
   numOfPid = 0;
   pid = NULL;
   running = NULL;
   status = NULL;
   
   struct sigaction my_act;    //new signal response
   struct sigaction old_act;   //old signal response
    
   //signal stuff
   if(sigemptyset(&(my_act.sa_mask)) == -1){
        fprintf(stderr, "mush: sigemptyset\n");
        _exit(EXIT_FAILURE);
    }
    /*behavior modification of signals*/
    my_act.sa_flags = 0;

    my_act.sa_handler = sighandler;
    if(sigaction(SIGINT, &my_act, &old_act) == -1){
        fprintf(stderr, "mush: sigaction\n");
        _exit(EXIT_FAILURE);
    }

   //handles arguments
   if(argc >= 3){
      fprintf(stderr, "Usage: mush [file]\n");
   }
   else if(argc == 2){
      fd = open(argv[1], O_RDONLY);
      if(fd == -1){
         fprintf(stderr,"error on open(%s, O_RDONLY)\n",argv[1]);
      }
      if(dup2(fd, STDIN_FILENO) == -1){
         fprintf(stderr,"error on dup2(fd, STDIN_FILENO)\n");
         _exit(EXIT_FAILURE);
      }
      close(fd);
   }
   
   //THE while loop
   while(!done){
      //prompt
      if(isatty(fileno(stdin)) && isatty(fileno(stdout))){
         printf("8-P ");
         fflush(stdout);
      }
      
      //get the command
      commandLine = getLine(stdin);
      

      //exit if EOF is sent
      if(feof(stdin)){
         break;
      }
      
      if((commandLine == NULL) || (strlen(commandLine) == 0)){
         continue;
      }
      
      //parse the command into stage structs
      pStages = parseline(commandLine);
      
      
      if(pStages == NULL){
         continue;
      }
      
      
      
      
      //checking to see if it's a cd command
      if(strcmp(pStages[0]->argvv[0], "cd") == 0){
         if(pStages[0]->argcc == 1){
            if(chdir(getenv("HOME")) == -1){
               fprintf(stderr,"chdir failed (chdir(HOME))\n");
            }
         }
         else{
            if(chdir(pStages[0]->argvv[1]) == -1){
               fprintf(stderr,"chdir failed\n");
            }
         }
         freeStage(pStages,NULL);
         continue;
      }
      
      //copy the number of stages
      numOfPid = pStages[0]->totalStageNumber;
      //start mallocing everything
      

      pid = (pid_t*)malloc(sizeof(pid_t)*numOfPid);
      

      running = (bool*)malloc(sizeof(bool)*numOfPid);

      status = (int*)malloc(sizeof(int)*numOfPid);

      pipefd = (int**)malloc(sizeof(int*)*(numOfPid));

      
      //mallocing each file descriptors, in and out
      for(int i = 0; i < numOfPid; i++){
         pipefd[i] = (int*)malloc(sizeof(int)*2);
      }
      
      //setting each of the running state to false
      for(int i = 0; i < numOfPid; i++){
         running[i] = FALSE;
      }
      
      for(int i = 0; i < numOfPid; i++){
         if(pipe(pipefd[i]) == -1){
               fprintf(stderr,"mush: pipe errer");
               close(pipefd[i][0]);
               close(pipefd[i][1]);
               continue;
         }
      }
      
      
      //special case for first pipe in
      if(pStages[0]->input != NULL){
         pipefd[0][0] = open(pStages[0]->input, O_RDONLY);
         fprintf(stderr,"pipefd[0][0] == %d\n",pipefd[0][0]);
         if(pipefd[0][0] == -1){
            fprintf(stderr,"mush: %s: No such file or directory\n",pStages[0]->input);
            freeStuff(status,pipefd,running,pid,numOfPid);
            freeStage(pStages,NULL);
            continue;
         }
      }
      else{
         pipefd[0][0] = STDIN_FILENO;
      }
      //special case for last pipe for out
      if(pStages[numOfPid-1]->output != NULL){
         if((stat(pStages[numOfPid-1]->output, &stattemp) == 0 && !(S_ISDIR(stattemp.st_mode))) || errno == ENOENT){
            pipefd[numOfPid-1][1] = open(pStages[numOfPid-1]->output, (O_WRONLY|O_CREAT|O_TRUNC), S_IRUSR|S_IWUSR);
            fprintf(stderr,"pipefd[numOfPid-1][1] == %d\n",pipefd[numOfPid-1][1]);
            if(pipefd[numOfPid-1][1] == -1){
               fprintf(stderr,"file open failed with : %s",pStages[numOfPid-1]->output);
               close(pipefd[numOfPid-1][0]);
               freeStuff(status,pipefd,running,pid,numOfPid);
               freeStage(pStages,NULL);
               continue;
            }
         }
         else{
            //file is a directory
            fprintf(stderr,"This is a directory\n");
            close(pipefd[numOfPid-1][0]);
            freeStuff(status,pipefd,running,pid,numOfPid);
            freeStage(pStages,NULL);
            continue;
         }
      }
      else{
         pipefd[numOfPid-1][1] = STDOUT_FILENO;
      }
      
      
      for(int i = 0; i < numOfPid; i++){

         //now start duping and closing the file descriptors
         running[i] = TRUE;
         pid[i] = fork();
         //something went wrong
         if(pid[i] == -1){
            fprintf(stderr,"mush: fork error\n");
            close(pipefd[i][0]);
            close(pipefd[i][1]);
            freeStuff(status,pipefd,running,pid,numOfPid);
            freeStage(pStages,NULL);
            continue;
         }
         //this is the child 
         else if(pid[i] == 0){
            //make mushit killable by SIGINT
            if(sigaction(SIGINT, &old_act, NULL) == -1){
            }
            
            
            if(i == 0 && numOfPid == 1){
            }
            
            
            if(numOfPid != 1){
               dup2(pipefd[i][1],STDIN_FILENO);
               if(i == (numOfPid-1)){
                  //fprintf(stderr,"STDOUT_FILENO=%d\n",STDOUT_FILENO);
                  //dup2(1,STDOUT_FILENO);
               }
               else{
                  dup2(pipefd[i+1][0],STDOUT_FILENO);
               }
            }
            
            
            if(execvp(pStages[i]->argvv[0], pStages[i]->argvv) == -1){
               fprintf(stderr,"execvp failed with : %s\n",pStages[i]->argvv[0]);
               freeStuff(status,pipefd,running,pid,numOfPid);
               freeStage(pStages,NULL);
               _exit(EXIT_FAILURE);
            }
            fprintf(stderr,"------------------\n");
         }
         

         //close if not std
         if(i != 0){
            close(pipefd[i][0]);
         }
         if(i != numOfPid-1){
            close(pipefd[i][1]);
         }
         
         
      }
      //end of the for loop

      //and now we wait for all the child to come back
      for(int i = 0; i < numOfPid; i++){
         if(waitpid(pid[i], &status[i], 0) == -1){
            if(WIFSIGNALED(status[i])){
               if(WTERMSIG(status[i]) != SIGINT){
                  //fprintf(stderr, "mush: waitpid signaled error %d\n", WTERMSIG(status[i]));
               }
            }
         }
         running[i] = FALSE;
      }
      
      freeStage(pStages,NULL);
      freeStuff(status,pipefd,running,pid,numOfPid);
   }
   //the very long while loop ends here
   
   free(commandLine);
   fprintf(stdout,"Good Bye.\n");
   return EXIT_SUCCESS;
}