我试图在c中编写一个shell程序。 该程序需要fork函数创建多个进程,并能够在一个命令中打印多个输出行。 例如,像linux终端一样,如果输入是" ls; ps; pwd;",输出应该是这样的。
$./shell
shell> ls ; ps ; pwd ;
(ls output)
(ps output)
(pwd output)
它应该能够打开一个文件并显示命令列表和该文件包含的输出。(批处理模式,我猜?)
让我们说这些命令列表在批处理文件中。
batch
1 ls
2 ps
3 ls ; pwd ; ps
输出
$./shell batch
shell> ls
shell> (ls output)
shell> ps
shell> (ps output)
shell> ls ; pwd ; ps
shell> (ls output)
(ps output)
(pwd output)
这是我写的代码
#include <stdio.h>
#include <unistd.h>
#include <stdlib.h>
#include <sys/wait.h>
#include <string.h>
#include <fcntl.h>
void command(char* myargs[][10], char* buffer);
int tokenizing(char* myargs[][10], char* buffer);
int main(int argc, char* argv[]) {
int fd;
char buffer[200];
char* myargs[10][10];
char* token;
if(argc >= 2) {
if((fd=open(argv[1], O_RDONLY)) == -1)
printf("cannot open file\n");
else {
read(fd, buffer, 200);
printf("%s\n", buffer);
token = strtok(buffer, "\n");
while(token != NULL) {
printf("%s\n", token);
command(myargs, token);
token = strtok(NULL, "\n");
}
return 0;
}
}
while(1) {
printf("prompt> ");
if(fgets(buffer, 200, stdin) == NULL||
strcmp(buffer, "quit\n") == 0)
break;
command(myargs, buffer);
}
return 0;
}
void command(char* myargs[][10], char* buffer) {
int rc = fork();
if(rc < 0) {
fprintf(stderr, "fork failed\n");
} else if(rc == 0) {
int n = tokenizing(myargs, buffer);
for(int i = 0 ; i < n; i++) {
int rc2 = fork();
if(rc2 < 0) {
fprintf(stderr, "for failed\n");
} else if(rc2 == 0) {
execvp(myargs[i][0], myargs[i]);
printf("%s: command not found\n", myargs[i][0]);
exit(0);
} else {
wait(NULL);
}
}
exit(0);
}
else {
wait(NULL);
}
}
int tokenizing(char* myargs[][10], char* buffer) {
int i = 0;
int j = 0;
int k = 0;
char* token;
char* subCommand[10];
token = strtok(buffer, ";\n");
while(token != NULL) {
subCommand[k] = token;
k++;
token = strtok(NULL, ";\n");
}
for(int i = 0; i < k; i++) {
token = strtok(subCommand[i], " \n");
while(token != NULL) {
myargs[i][j] = token;
j++;
token = strtok(NULL, " \n");
}
myargs[i][j] = NULL;
j=0;
}
}
此代码工作正常,但有一些问题。当这段代码与批处理文件一起运行时,我会遇到一些错误。
当程序执行时,我认为输出应该像上面的图像文件一样 - 据我所知。
但是程序经常出现一些奇怪的命令行,我甚至都没有打字。这些结果只会发生。
此外,如果你看到&#39; ps&#39;列表,你可以看到两个shell程序正在运行。
你能帮助我解决这些问题吗?
答案 0 :(得分:0)
'字符串'需要一个NUL终结符。在未安全NUL终止的char数组上调用printf(“%s ....”)和strtok()等strng函数会导致未定义的行为。
read()返回一个值。您可以使用它将终结器加载到“缓冲区”中。为了超级安全,您应该尝试只读取[缓冲区大小-1]字符,以确保终结器始终有足够的空间,例如:
buffer[read(fd, buffer, 199)]='\0';