From 285299dcf16d23bf7fdcd460cc314e658766a358 Mon Sep 17 00:00:00 2001
From: foyoodo <foyoodo@gmail.com>
Date: Sun, 7 Mar 2021 20:18:09 +0800
Subject: [PATCH] Lab util: find

---
 Makefile    |  1 +
 user/find.c | 71 +++++++++++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 72 insertions(+)
 create mode 100644 user/find.c

diff --git a/Makefile b/Makefile
index ced0582..1aec17a 100644
--- a/Makefile
+++ b/Makefile
@@ -135,6 +135,7 @@ mkfs/mkfs: mkfs/mkfs.c $K/fs.h $K/param.h
 UPROGS=\
 	$U/_cat\
 	$U/_echo\
+	$U/_find\
 	$U/_forktest\
 	$U/_grep\
 	$U/_init\
diff --git a/user/find.c b/user/find.c
new file mode 100644
index 0000000..7f1cdca
--- /dev/null
+++ b/user/find.c
@@ -0,0 +1,71 @@
+#include "kernel/types.h"
+#include "kernel/stat.h"
+#include "kernel/fs.h"
+#include "user/user.h"
+
+void
+find(char *path, char *file)
+{
+  char buf[512], *p;
+  int fd;
+  struct dirent de;
+  struct stat st;
+
+  if ((fd = open(path, 0)) < 0) {
+    fprintf(2, "find: cannot find path %s\n", path);
+    exit(1);
+  }
+
+  if (fstat(fd, &st) < 0) {
+    fprintf(2, "find: cannot stat %s\n", path);
+    close(fd);
+    exit(1);
+  }
+
+  if (st.type != T_DIR) {
+    fprintf(2, "find: %s is not a directory\n", path);
+    close(fd);
+    exit(1);
+  }
+
+  if(strlen(path) + 1 + DIRSIZ + 1 > sizeof buf){
+    fprintf(2, "find: path too long\n");
+    close(fd);
+    exit(1);
+  }
+
+  strcpy(buf, path);
+  p = buf + strlen(buf);
+  *p++ = '/';
+  while (read(fd, &de, sizeof(de)) == sizeof(de)) {
+    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(1, "find: cannot stat %s\n", buf);
+      continue;
+    }
+    if (st.type == T_FILE && !strcmp(de.name, file)) {
+      fprintf(1, "%s\n", buf);
+    } else if (st.type == T_DIR) {
+      find(buf, file);
+    }
+  }
+  close(fd);
+}
+
+int
+main(int argc, char *argv[])
+{
+  if (argc < 2 || argc > 3) {
+    fprintf(2, "Usage: find (path) [file]\n");
+    exit(1);
+  } else if (argc < 3) {
+    find(".", argv[1]);
+  } else {
+    find(argv[1], argv[2]);
+  }
+
+  exit(0);
+}