diff --git a/Makefile b/Makefile index b6d2c19..59d261c 100644 --- a/Makefile +++ b/Makefile @@ -189,6 +189,7 @@ UPROGS=\ $U/_wc\ $U/_zombie\ $U/_sleep\ + $U/_uptime\ $U/_pingpong\ $U/_primes\ $U/_find\ diff --git a/time.txt b/time.txt index 00750ed..7ed6ff8 100644 --- a/time.txt +++ b/time.txt @@ -1 +1 @@ -3 +5 diff --git a/user/find.c b/user/find.c index 1e00f93..ecee4b4 100644 --- a/user/find.c +++ b/user/find.c @@ -3,20 +3,48 @@ #include "kernel/types.h" #include "user/user.h" -void find(char *path, char *filename) { - int fd = open(path, 0); +int match_pattern(const char *name, const char *pattern) { + const char *star = 0; + const char *name_ptr = name; + const char *pattern_ptr = pattern; + + while (1) { + if (*pattern_ptr == '*') { + star = pattern_ptr++; + name_ptr = name; + continue; + } + + if (!*name) + return (!*pattern_ptr || (star && !*++pattern_ptr)); + if (*pattern_ptr == '?' || *pattern_ptr == *name) { + pattern_ptr++; + name++; + continue; + } + + if (star) { + pattern_ptr = star + 1; + name = ++name_ptr; + continue; + } + return 0; + } +} + +void find(char *path, char *pattern) { char buf[512], *p; + int fd; struct dirent de; struct stat st; - if (fd < 0 || strlen(path) + 1 + DIRSIZ + 1 >= sizeof(buf)) { - fprintf(2, "find: Invalid path: %s\n", path); - // close(fd); + if ((fd = open(path, 0)) < 0) { + fprintf(2, "find: cannot open %s\n", path); return; } if (fstat(fd, &st) < 0) { - fprintf(2, "find: Failed to stat %s\n", path); + fprintf(2, "find: cannot stat %s\n", path); close(fd); return; } @@ -27,36 +55,42 @@ void find(char *path, char *filename) { return; } - strncpy(buf, path, strlen(path) + 1); - p = buf + strlen(path); + if (strlen(path) + 1 + DIRSIZ + 1 > sizeof buf) { + fprintf(2, "find: path too long\n"); + close(fd); + return; + } + + strcpy(buf, path); + p = buf + strlen(buf); *p++ = '/'; + while (read(fd, &de, sizeof(de)) == sizeof(de)) { - if (de.inum == 0) - continue; - if (!strcmp(de.name, ".") || !strcmp(de.name, "..")) + if (de.inum == 0 || !strcmp(de.name, ".") || !strcmp(de.name, "..")) continue; memmove(p, de.name, DIRSIZ); p[DIRSIZ] = 0; if (stat(buf, &st) < 0) { - fprintf(2, "find: Failed to stat %s\n", buf); + fprintf(2, "find: cannot stat %s\n", buf); continue; } if (st.type == T_DIR) { - find(buf, filename); + find(buf, pattern); } else if (st.type == T_FILE) { - if (!strcmp(filename, "*") || !strcmp(filename, de.name)) { - printf("%s\n", buf); + if (match_pattern(de.name, pattern)) { + fprintf(1, "%s\n", buf); } } } + close(fd); } int main(int argc, char *argv[]) { if (argc != 3) { - printf("Usage: find path filename\n"); + fprintf(2, "Usage: find \n"); exit(1); } find(argv[1], argv[2]); diff --git a/user/sh.c b/user/sh.c index 836ebcb..492cf0f 100644 --- a/user/sh.c +++ b/user/sh.c @@ -1,15 +1,16 @@ // Shell. +#include "kernel/fcntl.h" +#include "kernel/fs.h" #include "kernel/types.h" #include "user/user.h" -#include "kernel/fcntl.h" // Parsed command representation -#define EXEC 1 +#define EXEC 1 #define REDIR 2 -#define PIPE 3 -#define LIST 4 -#define BACK 5 +#define PIPE 3 +#define LIST 4 +#define BACK 5 #define MAXARGS 10 @@ -49,15 +50,15 @@ struct backcmd { struct cmd *cmd; }; -int fork1(void); // Fork but panics on failure. -void panic(char*); -struct cmd *parsecmd(char*); -void runcmd(struct cmd*) __attribute__((noreturn)); +int fork1(void); // Fork but panics on failure. +void panic(char *); +struct cmd *parsecmd(char *); +void runcmd(struct cmd *) __attribute__((noreturn)); +// void handle_tab_completion(char *buf, int *i, int nbuf); +// 尝试添加tab补全功能 // Execute cmd. Never returns. -void -runcmd(struct cmd *cmd) -{ +void runcmd(struct cmd *cmd) { int p[2]; struct backcmd *bcmd; struct execcmd *ecmd; @@ -65,25 +66,25 @@ runcmd(struct cmd *cmd) struct pipecmd *pcmd; struct redircmd *rcmd; - if(cmd == 0) + if (cmd == 0) exit(1); - switch(cmd->type){ + switch (cmd->type) { default: panic("runcmd"); case EXEC: - ecmd = (struct execcmd*)cmd; - if(ecmd->argv[0] == 0) + ecmd = (struct execcmd *)cmd; + if (ecmd->argv[0] == 0) exit(1); exec(ecmd->argv[0], ecmd->argv); fprintf(2, "exec %s failed\n", ecmd->argv[0]); break; case REDIR: - rcmd = (struct redircmd*)cmd; + rcmd = (struct redircmd *)cmd; close(rcmd->fd); - if(open(rcmd->file, rcmd->mode) < 0){ + if (open(rcmd->file, rcmd->mode) < 0) { fprintf(2, "open %s failed\n", rcmd->file); exit(1); } @@ -91,25 +92,25 @@ runcmd(struct cmd *cmd) break; case LIST: - lcmd = (struct listcmd*)cmd; - if(fork1() == 0) + lcmd = (struct listcmd *)cmd; + if (fork1() == 0) runcmd(lcmd->left); wait(0); runcmd(lcmd->right); break; case PIPE: - pcmd = (struct pipecmd*)cmd; - if(pipe(p) < 0) + pcmd = (struct pipecmd *)cmd; + if (pipe(p) < 0) panic("pipe"); - if(fork1() == 0){ + if (fork1() == 0) { close(1); dup(p[1]); close(p[0]); close(p[1]); runcmd(pcmd->left); } - if(fork1() == 0){ + if (fork1() == 0) { close(0); dup(p[0]); close(p[0]); @@ -123,90 +124,82 @@ runcmd(struct cmd *cmd) break; case BACK: - bcmd = (struct backcmd*)cmd; - if(fork1() == 0) + bcmd = (struct backcmd *)cmd; + if (fork1() == 0) runcmd(bcmd->cmd); break; } exit(0); } -int -getcmd(char *buf, int nbuf) -{ +int getcmd(char *buf, int nbuf) { write(2, "$ ", 2); memset(buf, 0, nbuf); gets(buf, nbuf); - if(buf[0] == 0) // EOF + if (buf[0] == 0) // EOF return -1; return 0; } -int -main(void) -{ +int main(void) { static char buf[100]; int fd; // Ensure that three file descriptors are open. - while((fd = open("console", O_RDWR)) >= 0){ - if(fd >= 3){ + while ((fd = open("console", O_RDWR)) >= 0) { + if (fd >= 3) { close(fd); break; } } // Read and run input commands. - while(getcmd(buf, sizeof(buf)) >= 0){ - if(buf[0] == 'c' && buf[1] == 'd' && buf[2] == ' '){ + while (getcmd(buf, sizeof(buf)) >= 0) { + if (strcmp(buf, "exit\n") == 0) { + exit(0); + } + if (buf[0] == 'c' && buf[1] == 'd' && buf[2] == ' ') { // Chdir must be called by the parent, not the child. - buf[strlen(buf)-1] = 0; // chop \n - if(chdir(buf+3) < 0) - fprintf(2, "cannot cd %s\n", buf+3); + buf[strlen(buf) - 1] = 0; // chop \n + if (chdir(buf + 3) < 0) + fprintf(2, "cannot cd %s\n", buf + 3); continue; } - if(fork1() == 0) + if (fork1() == 0) runcmd(parsecmd(buf)); wait(0); } exit(0); } -void -panic(char *s) -{ +void panic(char *s) { fprintf(2, "%s\n", s); exit(1); } -int -fork1(void) -{ +int fork1(void) { int pid; pid = fork(); - if(pid == -1) + if (pid == -1) panic("fork"); return pid; } -//PAGEBREAK! -// Constructors +// PAGEBREAK! +// Constructors -struct cmd* -execcmd(void) -{ +struct cmd *execcmd(void) { struct execcmd *cmd; cmd = malloc(sizeof(*cmd)); memset(cmd, 0, sizeof(*cmd)); cmd->type = EXEC; - return (struct cmd*)cmd; + return (struct cmd *)cmd; } -struct cmd* -redircmd(struct cmd *subcmd, char *file, char *efile, int mode, int fd) -{ +struct cmd *redircmd(struct cmd *subcmd, char *file, char *efile, int mode, + int fd) { struct redircmd *cmd; cmd = malloc(sizeof(*cmd)); @@ -217,12 +210,10 @@ redircmd(struct cmd *subcmd, char *file, char *efile, int mode, int fd) cmd->efile = efile; cmd->mode = mode; cmd->fd = fd; - return (struct cmd*)cmd; + return (struct cmd *)cmd; } -struct cmd* -pipecmd(struct cmd *left, struct cmd *right) -{ +struct cmd *pipecmd(struct cmd *left, struct cmd *right) { struct pipecmd *cmd; cmd = malloc(sizeof(*cmd)); @@ -230,12 +221,10 @@ pipecmd(struct cmd *left, struct cmd *right) cmd->type = PIPE; cmd->left = left; cmd->right = right; - return (struct cmd*)cmd; + return (struct cmd *)cmd; } -struct cmd* -listcmd(struct cmd *left, struct cmd *right) -{ +struct cmd *listcmd(struct cmd *left, struct cmd *right) { struct listcmd *cmd; cmd = malloc(sizeof(*cmd)); @@ -243,39 +232,35 @@ listcmd(struct cmd *left, struct cmd *right) cmd->type = LIST; cmd->left = left; cmd->right = right; - return (struct cmd*)cmd; + return (struct cmd *)cmd; } -struct cmd* -backcmd(struct cmd *subcmd) -{ +struct cmd *backcmd(struct cmd *subcmd) { struct backcmd *cmd; cmd = malloc(sizeof(*cmd)); memset(cmd, 0, sizeof(*cmd)); cmd->type = BACK; cmd->cmd = subcmd; - return (struct cmd*)cmd; + return (struct cmd *)cmd; } -//PAGEBREAK! -// Parsing +// PAGEBREAK! +// Parsing char whitespace[] = " \t\r\n\v"; char symbols[] = "<|>&;()"; -int -gettoken(char **ps, char *es, char **q, char **eq) -{ +int gettoken(char **ps, char *es, char **q, char **eq) { char *s; int ret; s = *ps; - while(s < es && strchr(whitespace, *s)) + while (s < es && strchr(whitespace, *s)) s++; - if(q) + if (q) *q = s; ret = *s; - switch(*s){ + switch (*s) { case 0: break; case '|': @@ -288,53 +273,49 @@ gettoken(char **ps, char *es, char **q, char **eq) break; case '>': s++; - if(*s == '>'){ + if (*s == '>') { ret = '+'; s++; } break; default: ret = 'a'; - while(s < es && !strchr(whitespace, *s) && !strchr(symbols, *s)) + while (s < es && !strchr(whitespace, *s) && !strchr(symbols, *s)) s++; break; } - if(eq) + if (eq) *eq = s; - while(s < es && strchr(whitespace, *s)) + while (s < es && strchr(whitespace, *s)) s++; *ps = s; return ret; } -int -peek(char **ps, char *es, char *toks) -{ +int peek(char **ps, char *es, char *toks) { char *s; s = *ps; - while(s < es && strchr(whitespace, *s)) + while (s < es && strchr(whitespace, *s)) s++; *ps = s; return *s && strchr(toks, *s); } -struct cmd *parseline(char**, char*); -struct cmd *parsepipe(char**, char*); -struct cmd *parseexec(char**, char*); -struct cmd *nulterminate(struct cmd*); +struct cmd *parseline(char **, char *); +struct cmd *parsepipe(char **, char *); +struct cmd *parseexec(char **, char *); +struct cmd *nulterminate(struct cmd *); -struct cmd* -parsecmd(char *s) -{ +struct cmd *parsecmd(char *s) { char *es; struct cmd *cmd; es = s + strlen(s); cmd = parseline(&s, es); peek(&s, es, ""); - if(s != es){ + if (s != es) { fprintf(2, "leftovers: %s\n", s); panic("syntax"); } @@ -342,102 +323,92 @@ parsecmd(char *s) return cmd; } -struct cmd* -parseline(char **ps, char *es) -{ +struct cmd *parseline(char **ps, char *es) { struct cmd *cmd; cmd = parsepipe(ps, es); - while(peek(ps, es, "&")){ + while (peek(ps, es, "&")) { gettoken(ps, es, 0, 0); cmd = backcmd(cmd); } - if(peek(ps, es, ";")){ + if (peek(ps, es, ";")) { gettoken(ps, es, 0, 0); cmd = listcmd(cmd, parseline(ps, es)); } return cmd; } -struct cmd* -parsepipe(char **ps, char *es) -{ +struct cmd *parsepipe(char **ps, char *es) { struct cmd *cmd; cmd = parseexec(ps, es); - if(peek(ps, es, "|")){ + if (peek(ps, es, "|")) { gettoken(ps, es, 0, 0); cmd = pipecmd(cmd, parsepipe(ps, es)); } return cmd; } -struct cmd* -parseredirs(struct cmd *cmd, char **ps, char *es) -{ +struct cmd *parseredirs(struct cmd *cmd, char **ps, char *es) { int tok; char *q, *eq; - while(peek(ps, es, "<>")){ + while (peek(ps, es, "<>")) { tok = gettoken(ps, es, 0, 0); - if(gettoken(ps, es, &q, &eq) != 'a') + if (gettoken(ps, es, &q, &eq) != 'a') panic("missing file for redirection"); - switch(tok){ + switch (tok) { case '<': cmd = redircmd(cmd, q, eq, O_RDONLY, 0); break; case '>': - cmd = redircmd(cmd, q, eq, O_WRONLY|O_CREATE|O_TRUNC, 1); + cmd = redircmd(cmd, q, eq, O_WRONLY | O_CREATE | O_TRUNC, 1); break; - case '+': // >> - cmd = redircmd(cmd, q, eq, O_WRONLY|O_CREATE, 1); + case '+': // >> + cmd = redircmd(cmd, q, eq, O_WRONLY | O_CREATE, 1); break; } } return cmd; } -struct cmd* -parseblock(char **ps, char *es) -{ +struct cmd *parseblock(char **ps, char *es) { struct cmd *cmd; - if(!peek(ps, es, "(")) + if (!peek(ps, es, "(")) panic("parseblock"); gettoken(ps, es, 0, 0); cmd = parseline(ps, es); - if(!peek(ps, es, ")")) + if (!peek(ps, es, ")")) panic("syntax - missing )"); gettoken(ps, es, 0, 0); cmd = parseredirs(cmd, ps, es); return cmd; } -struct cmd* -parseexec(char **ps, char *es) -{ +struct cmd *parseexec(char **ps, char *es) { char *q, *eq; int tok, argc; struct execcmd *cmd; struct cmd *ret; - if(peek(ps, es, "(")) + if (peek(ps, es, "(")) return parseblock(ps, es); ret = execcmd(); - cmd = (struct execcmd*)ret; + cmd = (struct execcmd *)ret; argc = 0; ret = parseredirs(ret, ps, es); - while(!peek(ps, es, "|)&;")){ - if((tok=gettoken(ps, es, &q, &eq)) == 0) + while (!peek(ps, es, "|)&;")) { + if ((tok = gettoken(ps, es, &q, &eq)) == 0) break; - if(tok != 'a') + if (tok != 'a') panic("syntax"); cmd->argv[argc] = q; cmd->eargv[argc] = eq; argc++; - if(argc >= MAXARGS) + if (argc >= MAXARGS) panic("too many args"); ret = parseredirs(ret, ps, es); } @@ -447,9 +418,7 @@ parseexec(char **ps, char *es) } // NUL-terminate all the counted strings. -struct cmd* -nulterminate(struct cmd *cmd) -{ +struct cmd *nulterminate(struct cmd *cmd) { int i; struct backcmd *bcmd; struct execcmd *ecmd; @@ -457,36 +426,36 @@ nulterminate(struct cmd *cmd) struct pipecmd *pcmd; struct redircmd *rcmd; - if(cmd == 0) + if (cmd == 0) return 0; - switch(cmd->type){ + switch (cmd->type) { case EXEC: - ecmd = (struct execcmd*)cmd; - for(i=0; ecmd->argv[i]; i++) + ecmd = (struct execcmd *)cmd; + for (i = 0; ecmd->argv[i]; i++) *ecmd->eargv[i] = 0; break; case REDIR: - rcmd = (struct redircmd*)cmd; + rcmd = (struct redircmd *)cmd; nulterminate(rcmd->cmd); *rcmd->efile = 0; break; case PIPE: - pcmd = (struct pipecmd*)cmd; + pcmd = (struct pipecmd *)cmd; nulterminate(pcmd->left); nulterminate(pcmd->right); break; case LIST: - lcmd = (struct listcmd*)cmd; + lcmd = (struct listcmd *)cmd; nulterminate(lcmd->left); nulterminate(lcmd->right); break; case BACK: - bcmd = (struct backcmd*)cmd; + bcmd = (struct backcmd *)cmd; nulterminate(bcmd->cmd); break; } diff --git a/user/uptime.c b/user/uptime.c new file mode 100644 index 0000000..1db040e --- /dev/null +++ b/user/uptime.c @@ -0,0 +1,15 @@ +#include "kernel/types.h" +#include "user/user.h" + +int main(int argc, char *argv[]) { + if (argc != 1) { + // 错误输出到stderr(文件描述符2) + fprintf(2, "Usage: uptime\n"); + exit(1); + } + + fprintf(1,"up: %d\n",uptime()); + + exit(0); +} +