编写shell (一)

编写shell

刺猬@http://blog.youkuaiyun.com/littlehedgehog

 

 

 

 

 

编写shell的第一步: 让shell可以执行程序

 

只有两个要点需要注意，第一，unix/linux提供相应的执行函数exec，我们只需要将我们输入的命令行分解成字符串数组即可。
比如: ls -al  那么我们只需要将这个字符串拆成argv[0]="ls",argv[1]="-al" 即可。 第二点，需要用fork新建一个进程，在子进程中执行我们的命令，即是execvp(argv[0],argv)

 

代码参照<understanding unix/linux programming>

 

1. #include <stdio.h>
2. #include <unistd.h>
3. #include <string.h>
4. #include <signal.h>
6. #define BUFSIZE 256
8. #define PROMPT "kiss winona $:" 
10. void setup()
11. {
12.     signal(SIGINT,SIG_IGN);
13.     signal(SIGQUIT,SIG_IGN);
14. }
16. void fatal(char *err,int n)
17. {
18.     perror(err);
19.     exit(n);
20. }
21. /* 下面是封装malloc 和 remalloc*/
22. void * emalloc(size_t n)
23. {
24.     void *ret_p=NULL;
25.     if((ret_p=malloc(n))!=NULL)
26.         return ret_p;
27.     fatal("Out of memory! ",-1);
28. }
30. void *erealloc(void *p,size_t n)
31. {
32.     void *ret_p=NULL;
33.     if((ret_p=realloc(ret_p,n))!=NULL)
34.         return ret_p;
35.     fatal("Realloc failed! ",-1);
36. }
38. char * makestr(char *p)
39. {
40.     int len=strlen(p);
41.     char *ret=emalloc(len+1);
42.     strncpy(ret,p,len);
43.     ret[len]='/0';
44.     return ret;
45. }
47. /* 我们需要把一个命令行 转化为字符串数组 */
48. char ** splitline(char *line)
49. {
50.     char *p;
51.     char **argv;
52.     int argc=0;
53.     if(!line)
54.         return NULL;
55.     argv=emalloc(BUFSIZE);
57.     p=strtok(line," ");         //目前只是处理了空格
58.     while(p)
59.     {
60.         argv[argc++]=makestr(p);
61.         p=strtok(NULL," ");
62.     }
63.     argv[argc]=0;
64.     return argv;
65. }
67. /* 从标准输入获取命令字符串  注意我们用了个动态空间装载*/
68. char * get_cmd(char *prompt,FILE *fp)
69. {
70.     char c;
71.     char *buffer;
72.     int pos=0,bufsize=0;    
73.     printf("%s ",prompt);
75.     if(!fp)
76.         return NULL;
77.     while((c=getc(fp))!=EOF)
78.     {
79.         if(pos+1>=bufsize)  
80.         {
81.             if(!bufsize)    
82.                 buffer=emalloc(BUFSIZE);    
83.             else            
84.                 buffer=erealloc(buffer,bufsize+BUFSIZE);
85.             bufsize+=BUFSIZE;   
86.         }
87.         if(c=='/n')
88.             break;
89.         buffer[pos++]=c;
90.     }
91.     buffer[pos]='/0';
92.     return buffer;
93. }
95. /* 小心内存泄露 */
96. void freelist(char **argv)
97. {
98.     while(*argv)
99.         free(*argv++);
100.     free(argv);
101. }
103. /* 主要函数: 原理简单 父进程fork一个子进程并等待  子进程执行命令 */
104. int execute(char **argv)
105. {
106.     int pid,info=-1;
107.     
108.     if(!argv[0])
109.         return 0;
110.     if((pid=fork())==-1)
111.         fatal("Fork error ",1);
112.     else if(!pid)   
113.     {
114.         signal(SIGINT,SIG_DFL);
115.         signal(SIGQUIT,SIG_DFL);
116.         execvp(argv[0],argv);
117.         fatal("command not found ",1);
118.     }
119.     else
120.     {
121.         if(wait(&info)==-1)
122.             fatal("wait error ",1);
123.         return info;
124.     }
125. }
128. int main()
129. {
130.     char *cmd=NULL,*prompt=NULL,**argv=NULL;        
131.     prompt=PROMPT;
132.     setup();
133.     while((cmd=get_cmd(prompt,stdin))!=NULL)
134.     {
135.         if((argv=splitline(cmd))!=NULL)
136.         {
137.             execute(argv);
138.             free(argv);
139.         }
140.         free(cmd);      //注意cmd也要释放
141.     }
142.     return 0;
143. }