Linux_C smsh1

这是一个模拟shell端的程序。

使用了execvp,fork,wait,malloc,realloc以及strtok()函数。

smsh.h

char* next_cmd();
char** splitline(char* );
void freelist(char **);
int execute(char** );

smsh1.c

#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <signal.h>
#define DFL_PROMPT ">:"
int main(){
  char * cmdline, *prompt, **arglist;
  int i;
  void setup();
  prompt = DFL_PROMPT;
  setup();
  while((cmdline=next_cmd(prompt, stdin)) != NULL ){
    if((arglist=splitline(cmdline))!=NULL){
      for(i=0; i<2; i++)
    printf("%s  ", arglist[i]);
      printf("
");
      printf("will execute.
");
      execute(arglist);
      freelist(arglist);
    }
    free(cmdline);
  }
  
  return 0;
}
void setup(){
  /*
   * purpose: initialize shell
   * 在shell中忽略信号SIGINT&SIGQUIT, 但是在子进程中恢复对
   * SIGINT&SIGQUIT的默认操作，允许用户通过按表示结束多文件多Ctrl-D来退出
   */
  signal(SIGINT, SIG_IGN);
  signal(SIGQUIT, SIG_IGN);
}

splitline.c

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include "smsh.h"

char* next_cmd(char* prompt, FILE *fp){
  /*
   * purpose: read next command line from fp
   * return: dynamically allocated string holding command line
   * note: allocates space in BUFSIZ chunks.其中BUFSIZ是库函数中define过了的
   */
  char* buf;
  int bufspace = 0;
  int pos = 0;
  int c;
  //printf("%s", prompt);
  while((c=getc(fp)) != EOF) {
    if(pos+1>=bufspace) {
      if(bufspace==0)
    buf = malloc(BUFSIZ);
      else 
    buf = realloc(buf, bufspace+BUFSIZ);
      bufspace+=BUFSIZ;
    }
    if(c=='
')
      break;
    buf[pos++]=c;
  }
  if(c==EOF || pos==0)
    return NULL;
  buf[pos]=0;
  //printf("u cin is end.
");
  return buf;
}

char** splitline(char* cmdline){
  /*
   * purpose: split a line 
   * note:注意：在最后一次给arglist赋值时，也就是strtok()返回为NULL的时候，也需要给
   *            arglist[i]=malloc(sizeof(char*));一下，即使最后赋的为NULL也需要分配一个指针空间给它。
   */
  char **arglist;
  char *delim=" ";
  char *cmdbuf;
  int i=1;
  cmdbuf=strtok(cmdline,delim);
  //printf("cmdbuf0: %s
", cmdbuf);
  arglist = malloc(sizeof(char*));
  arglist[0]=malloc(strlen(cmdbuf)*sizeof(char));
  strcpy(arglist[0], cmdbuf);
  while((cmdbuf=strtok(NULL, delim))) {
    arglist = realloc(arglist, (1+i)*sizeof(char*));
    arglist[i]=malloc(strlen(cmdbuf)*sizeof(char));
    strcpy(arglist[i], cmdbuf);
    //printf("cmdbuf%d: %s
",i,cmdbuf);
    i++;
  }
  arglist[i]=malloc(sizeof(char*));
  arglist[i]=NULL;
  return arglist;
}

execute.c

/* execute.c- code used by small shell to execute commands
 */
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <signal.h>
#include <sys/wait.h>

int execute(char** arglist){
  /*
   *purpose: run a program passing it arguments;
   *return: status returned via wait, or -1 on error
   *errors: -1 on fork() on wait() errors
   */
  int pid;
  int child_info = -1;
  printf("now its go into the function execute
");
  if((pid=fork()) == -1)
    perror("fork");
  else if(pid == 0) {
    signal(SIGINT, SIG_DFL);
    signal(SIGQUIT, SIG_DFL);
    execvp(arglist[0], arglist);
    perror("cannot execute command");
    exit(1);
  }
  else {
    if(wait(&child_info) == -1)
      perror("wait");
  }
  return child_info;
}
void freelist(char** list) {
  /*
   *purpose: free the list returned by splitline
   *returns: nothing
   *actoin: free all strings in list and then free the list
   */
  char **cp=list;
  while(*cp)
    free(*cp++);
  free(list);
}

 关于为什么要开一个进程给execvp()使用，是因为unix运行一个程序;1.将指定的程序复制到调用它的进程。2.将指定的字符串数组作为argv[]传给这个程序。3.运行这个程序。

如果不开进程就来执行execvp()，就会运行完一个命令后就会结束退出。这是因为execvp()用命令指定的程序代码覆盖了shell的程序代码，然后在命令指定的程序结束之后就退出。