編寫shell (一)

編寫shell

刺蝟@http://blog.csdn.net/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. }