summaryrefslogtreecommitdiff
path: root/src/user/app/yosh/main.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/user/app/yosh/main.cpp')
-rw-r--r--src/user/app/yosh/main.cpp375
1 files changed, 375 insertions, 0 deletions
diff --git a/src/user/app/yosh/main.cpp b/src/user/app/yosh/main.cpp
new file mode 100644
index 0000000..fe3eeed
--- /dev/null
+++ b/src/user/app/yosh/main.cpp
@@ -0,0 +1,375 @@
+#include <tce/syscall.h>
+#include <tce/vfs.h>
+#include <stdlib.h>
+#include <stdio.h>
+#include <readline.h>
+#include <IO/IOStream.h>
+
+using namespace libc;
+
+FILE cwd_f;
+char *cwd;
+
+void about() {
+ stdio << "Trivial/Computing Environment v0.1.0 - yosh 3.\n";
+}
+
+void help(char *args[]) {
+ if (args[1] == 0) {
+ stdio << "Available commands: about, help, exit, ls, cd, goto.\n";
+ } else if (strcmp(args[1], "about") == 0) {
+ stdio << "Usage:\tabout\nShows some info about yosh. Very usefull.\n";
+ } else if (strcmp(args[1], "help") == 0) {
+ stdio << "Usage:\thelp\n\thelp <command>\n"
+ << "Shows some info about the command you want, or commands in general.\n";
+ } else if (strcmp(args[1], "exit") == 0) {
+ stdio << "Usage:\texit\n"
+ << "Exits the shell.\nWill probably make your system panic if you only have a shell running.\n";
+ } else if (strcmp(args[1], "cd") == 0) {
+ stdio << "Usage:\tcd <location>\nGoes to the location specified.\n";
+ } else if (strcmp(args[1], "ls") == 0) {
+ stdio << "Usage:\tls\n\tls <location>...\nShows the content of the current/specified directory.\n";
+ } else if (strcmp(args[1], "goto") == 0) {
+ stdio << "Usage:\tgoto <vt name>\nSwitchs focus to specified virtual terminal.\n"
+ << "Type `ls /.ui` to see available terminals.\n";
+ } else {
+ stdio.printf("No such command: %s\n", args[1]);
+ }
+}
+
+void simplify_path(char* p) {
+ char *it = p;
+ char *member = it;
+ while (*it != 0) {
+ if (*it == '/') {
+ if (it == member && it != p) {
+ // two consecutive slashes
+ char *i = member;
+ while (1) {
+ i[0] = i[1];
+ if (i[0] == 0) break;
+ i++;
+ }
+ } else {
+ *it = 0;
+ if (strcmp(member, ".") == 0) {
+ char *i = member;
+ while (1) {
+ i[0] = i[2];
+ if (i[0] == 0) break;
+ i++;
+ }
+ } else if (strcmp(member, "..") == 0) {
+ *it = '/';
+ char* start = member - 2;
+ char* next = member + 3;
+ while (start > p && *start != '/') {
+ start--;
+ }
+ start++;
+ it = member = start;
+ while (1) {
+ *start = *next;
+ if (*start == 0) break;
+ start++;
+ next++;
+ }
+ } else {
+ *it = '/';
+ it++;
+ member = it;
+ }
+ }
+ } else {
+ it++;
+ }
+ }
+}
+
+char* path_cat (char* a, char* b, int trailing_slash) {
+ char* ret;
+ if (b[0] == '/') {
+ int lb = strlen(b);
+ ret = (char*)malloc(lb + 2);
+ memcpy(ret, b, lb);
+ if (ret[lb - 1] != '/') {
+ ret[lb] = '/';
+ lb++;
+ }
+ ret[lb] = 0;
+ } else {
+ int la = strlen(a);
+ int lb = strlen(b);
+
+ ret = (char*)malloc(la + lb + 3);
+ memcpy(ret, a, la);
+ if (ret[la-1] != '/') {
+ ret[la] = '/';
+ la++;
+ }
+ memcpy(ret + la, b, lb);
+ if (ret[la + lb - 1] != '/') {
+ ret[la + lb] = '/';
+ lb++;
+ }
+ ret[la + lb] = 0;
+ }
+ simplify_path(ret);
+ if (!trailing_slash) {
+ int l = strlen(ret);
+ if (ret[l-1] == '/') ret[l-1] = 0;
+ }
+ return ret;
+}
+
+void cd(char **args) {
+ if (args[1] == 0) {
+ print("Usage: cd <directory>\n");
+ }
+ if (strcmp(args[1], ".") == 0) return;
+
+ char* newcwd = path_cat(cwd, args[1], 1);
+
+ file_info info;
+ int i = stat(newcwd, &info);
+ if (i == E_NOT_FOUND) {
+ print("No such file or directory.\n");
+ free(newcwd);
+ } else if (i < 0) {
+ stdio.printf("Error stating: %i\n", i);
+ free(newcwd);
+ } else if ((info.type & FT_DIR) == 0) {
+ print("Not a directory.\n");
+ free(newcwd);
+ } else {
+ FILE nf = open(newcwd, 0);
+ if (nf < 0) {
+ stdio.printf("Error opening: %i\n", nf);
+ free(newcwd);
+ } else {
+ free(cwd);
+ cwd = newcwd;
+ close(cwd_f);
+ cwd_f = nf;
+ }
+ }
+}
+
+void ls_dir(FILE f) {
+ file_info info;
+ int i = 0;
+ char buf[256];
+ while (read(f, i, 256, buf) > 0) {
+ i++;
+
+ if (strcmp(buf, ".") == 0 || strcmp(buf, "..") == 0) continue;
+
+ stdio.printf(" %s", buf);
+
+ stat_relative(f, buf, &info);
+ if (info.type & FT_DIR) print("/");
+ print(" \t");
+
+ if (info.type & FT_FILE) print("file ");
+ if (info.type & FT_DIR) print("dir ");
+ if (info.type & FT_SYMLINK) print("symlink ");
+ if (info.type & FT_DEV) print("dev ");
+ if (info.type & FT_TERMINAL) print("term ");
+
+ print("\t");
+ if (info.type & FT_TERMINAL) {
+ stdio.printf("%ix%i", info.size >> 16, info.size & 0xFFFF);
+ } else if ((info.type & FT_DEV) == 0) {
+ stdio.printf("%i", info.size);
+ }
+
+ print("\n");
+ }
+}
+
+void ls(char **args) {
+ if (args[1] == 0) {
+ ls_dir(cwd_f);
+ } else {
+ int i;
+ for (i = 1; args[i] != 0; i++) {
+ stdio.printf("Contents of %s :\n", args[i]);
+ char* d = path_cat(cwd, args[i], 1);
+ file_info info;
+ int e = stat(d, &info);
+ if (e == E_NOT_FOUND) {
+ print(" No such file or directory\n");
+ } else if (e < 0) {
+ stdio.printf(" Error stating: %i\n", e);
+ } else if ((info.type & FT_DIR) == 0) {
+ stdio.printf(" Not a directory.\n");
+ } else {
+ FILE ff = open(d, 0);
+ if (ff < 0) {
+ stdio.printf(" Error opening: %i\n", ff);
+ } else {
+ ls_dir(ff);
+ close(ff);
+ }
+ }
+ free(d);
+ }
+ }
+}
+
+void cat(char **args) {
+ int i;
+ for (i = 1; args[i] != 0; i++) {
+ char *d = path_cat(cwd, args[i], 0);
+ file_info info;
+ int e = stat(d, &info);
+ if (e == E_NOT_FOUND) {
+ stdio.printf("No such file: %s\n", d);
+ } else if (e < 0) {
+ stdio.printf("Error stating %s : %i\n", d, e);
+ } else if ((info.type & FT_FILE) == 0) {
+ stdio.printf("Not a file: %s\n", d);
+ } else {
+ FILE ff = open(d, 0);
+ char* buff = (char*)malloc(info.size);
+ read(ff, 0, info.size, buff);
+ close(ff);
+ write(term, 0, info.size, buff);
+ free(buff);
+ }
+ }
+}
+
+void t_goto(char** args) {
+ if (args[1] == 0) {
+ print("Usage: goto <vt_name>\n");
+ return;
+ }
+ char* p = path_cat("/.ui/", args[1], 0);
+ int i;
+ if ((i = link(p, "/.dev/vgatxt", LM_OUTPUT_TO)) == 0) {
+ link("/.dev/ps2kbd", p, LM_OUTPUT_TO);
+ } else {
+ stdio.printf("Error %i\n", i);
+ }
+}
+
+int Main(char **sh_args) {
+ about();
+
+ cwd = strdup("/");
+ cwd_f = open(cwd, 0);
+
+ char *path = path_cat(sh_args[0], "..", 1);
+
+ int bg_pr[128], bg_pr_c = 0;
+
+ Term *term = stdio.term;
+ if (term == 0) {
+ stdio << "Error: no terminal...\n";
+ return -1;
+ }
+
+ while (1) {
+ stdio.printf("\x1b[33m %s \x1b[1m", cwd);
+
+ char *s = term->readline();
+ stdio.printf("\x1b[0m");
+ if (s == 0) return -1;
+ if (*s == 0) continue;
+
+ char *c_args[16] = {0}, *cmd;
+ c_args[0] = s;
+ int argc = 1;
+ for (cmd = s; *cmd != 0; cmd++) {
+ if (*cmd == ' ') {
+ *cmd = 0;
+ if (cmd[1] != ' ' && cmd[1] != 0) {
+ c_args[argc] = cmd + 1;
+ argc++;
+ if (argc == 15) break;
+ }
+ }
+ }
+
+ if (strcmp(c_args[0], "about") == 0) {
+ about();
+ } else if (strcmp(c_args[0], "help") == 0) {
+ help(c_args);
+ } else if (strcmp(c_args[0], "exit") == 0) {
+ print("Exiting the shell. See you later!\n");
+ break;
+ } else if (strcmp(c_args[0], "cd") == 0) {
+ cd(c_args);
+ } else if (strcmp(c_args[0], "ls") == 0) {
+ ls(c_args);
+ } else if (strcmp(c_args[0], "goto") == 0) {
+ t_goto(c_args);
+ } else if (strcmp(c_args[0], "cat") == 0) {
+ cat(c_args);
+ } else {
+ FILE vt = term->fd;
+ char *t = 0;
+ if (strcmp(c_args[0], "on") == 0) {
+ if (c_args[1] == 0 || c_args[2] == 0) {
+ stdio.printf("Usage:\ton <vt> <command>\n");
+ continue;
+ }
+ t = path_cat("/.ui/", c_args[1], 0);
+ vt = open(t, 0);
+ if (vt < 0 || vt == term->fd) {
+ stdio.printf("Error: cannot open terminal %s (%i)\n", t, vt);
+ free(t);
+ continue;
+ }
+ int i = 0;
+ while (1) {
+ c_args[i] = c_args[i+2];
+ if (c_args[i] == 0) break;
+ i++;
+ }
+ }
+
+ char *c;
+ if (strchr(c_args[0], '/')) {
+ c = path_cat(cwd, c_args[0], 0);
+ } else {
+ c = path_cat(path, c_args[0], 0);
+ }
+ int pid = run(c, c_args + 1, vt);
+ if (pid <= 0) {
+ if (pid == E_NOT_FOUND) {
+ stdio.printf("Error: no such file %s\n", c);
+ } else {
+ stdio.printf("Error %i\n", pid);
+ }
+ } else {
+ if (vt == term->fd) {
+ int ret = waitpid(pid, 1);
+ stdio.printf("(yosh) child (pid %i) exited with status %i\n", pid, ret);
+ } else {
+ stdio.printf("(yosh) running on terminal %s, pid:%i\n", t, pid);
+ bg_pr[bg_pr_c++] = pid;
+ free(t);
+ }
+ }
+ free(c);
+ }
+
+ for (; s < cmd; s++) if (*s == 0) *s = ' ';
+
+ int i = 0;
+ while (i < bg_pr_c) {
+ int ret = waitpid(bg_pr[i], 0);
+ if (ret != E_NOT_FINISHED) {
+ stdio.printf("(yosh) child (pid %i) exited with status %i\n", bg_pr[i], ret);
+ bg_pr_c--;
+ bg_pr[i] = bg_pr[bg_pr_c];
+ } else {
+ i++;
+ }
+ }
+ }
+
+ return 0;
+}