Michael Shimniok / SimpleShell

Simple embedded shell with runtime pluggable commands.

Dependents:   DataBus2018

Implements a simple unix-like shell for embedded systems with a pluggable command architecture.

SimpleShell.cpp@33:84dc443909a0, 2018-12-28 (annotated)

Committer:
shimniok
Date:
Fri Dec 28 20:08:22 2018 +0000
Revision:
33:84dc443909a0
Parent:
32:fc09f0eb1e8a
Child:
34:afe994ca0e49
refactor basename to use strrchr()

Who changed what in which revision?

UserRevisionLine numberNew contents of line
shimniok 21:5d7ac1f0b842 1 #include "SimpleShell.h"
shimniok 21:5d7ac1f0b842 2 #include <ctype.h>
shimniok 21:5d7ac1f0b842 3 #include <string>
shimniok 28:753db82debb1 4 #include <list>
shimniok 21:5d7ac1f0b842 5
shimniok 21:5d7ac1f0b842 6 #define ESC 0x1b
shimniok 21:5d7ac1f0b842 7 #define UP 0x41
shimniok 21:5d7ac1f0b842 8 #define DOWN 0x42
shimniok 21:5d7ac1f0b842 9 #define RIGHT 0x43
shimniok 21:5d7ac1f0b842 10 #define LEFT 0x44
shimniok 21:5d7ac1f0b842 11 #define HOME 0x31
shimniok 21:5d7ac1f0b842 12 #define INS 0x32
shimniok 21:5d7ac1f0b842 13 #define DEL 0x33
shimniok 21:5d7ac1f0b842 14 #define PGUP 0x35
shimniok 21:5d7ac1f0b842 15 #define PGDN 0x36
shimniok 21:5d7ac1f0b842 16 #define TAIL 0x7e
shimniok 21:5d7ac1f0b842 17
shimniok 21:5d7ac1f0b842 18
shimniok 29:8d4132274445 19 SimpleShell::SimpleShell(char *home)
shimniok 28:753db82debb1 20 {
shimniok 28:753db82debb1 21 lookupEnd = 0;
shimniok 28:753db82debb1 22
shimniok 29:8d4132274445 23 // Set home directory
shimniok 29:8d4132274445 24 strncpy(_home, canon(home), MAXBUF);
shimniok 29:8d4132274445 25
shimniok 29:8d4132274445 26 // Set current working directory
shimniok 29:8d4132274445 27 strncpy(_cwd, _home, MAXBUF);
shimniok 29:8d4132274445 28
shimniok 28:753db82debb1 29 // Built-in shell commands
shimniok 28:753db82debb1 30 command(callback(this, &SimpleShell::help), "help");
shimniok 28:753db82debb1 31 command(callback(this, &SimpleShell::pwd), "pwd");
shimniok 28:753db82debb1 32 command(callback(this, &SimpleShell::cat), "cat");
shimniok 28:753db82debb1 33 command(callback(this, &SimpleShell::cd), "cd");
shimniok 28:753db82debb1 34 command(callback(this, &SimpleShell::rm), "rm");
shimniok 28:753db82debb1 35 command(callback(this, &SimpleShell::touch), "touch");
shimniok 28:753db82debb1 36 command(callback(this, &SimpleShell::ls), "ls");
shimniok 28:753db82debb1 37 command(callback(this, &SimpleShell::send), "send");
shimniok 28:753db82debb1 38 }
shimniok 28:753db82debb1 39
shimniok 28:753db82debb1 40
shimniok 31:27e8130a0d8f 41 bool SimpleShell::haswildcard(char *s) {
shimniok 31:27e8130a0d8f 42 bool result = \
shimniok 31:27e8130a0d8f 43 strchr(s, '*') != NULL || strchr(s, '?') != NULL ||
shimniok 31:27e8130a0d8f 44 (strchr(s, '[') != NULL && strchr(s, ']') != NULL);
shimniok 31:27e8130a0d8f 45 return result;
shimniok 31:27e8130a0d8f 46 }
shimniok 31:27e8130a0d8f 47
shimniok 31:27e8130a0d8f 48
shimniok 28:753db82debb1 49 /// Path stack representation
shimniok 28:753db82debb1 50 typedef std::list<char*> path_t;
shimniok 28:753db82debb1 51
shimniok 21:5d7ac1f0b842 52 char *SimpleShell::canon(char *path) {
shimniok 28:753db82debb1 53 path_t pstack;
shimniok 28:753db82debb1 54 static char tmp[MAXBUF*2];
shimniok 28:753db82debb1 55 char *e;
shimniok 21:5d7ac1f0b842 56
shimniok 28:753db82debb1 57 // if we're passed empty/null string, just send back cwd so nothing breaks
shimniok 28:753db82debb1 58 if (path == NULL || strlen(path) == 0) {
shimniok 28:753db82debb1 59 strcpy(tmp, _cwd);
shimniok 28:753db82debb1 60 return tmp;
shimniok 28:753db82debb1 61 }
shimniok 28:753db82debb1 62
shimniok 28:753db82debb1 63 // relative path? add current working directory to path stack
shimniok 21:5d7ac1f0b842 64 if (path[0] != '/') {
shimniok 28:753db82debb1 65 strcpy(tmp, _cwd);
shimniok 28:753db82debb1 66 strcat(tmp, "/");
shimniok 28:753db82debb1 67 strcat(tmp, path);
shimniok 28:753db82debb1 68 } else {
shimniok 28:753db82debb1 69 strcpy(tmp, path);
shimniok 21:5d7ac1f0b842 70 }
shimniok 28:753db82debb1 71
shimniok 28:753db82debb1 72 // now canonicalize the path spec
shimniok 28:753db82debb1 73 e = strtok(tmp+1, "/");
shimniok 28:753db82debb1 74 while (e) {
shimniok 28:753db82debb1 75 //printf("e = <%s>\n", e);
shimniok 28:753db82debb1 76 if (strcmp("..", e) == 0) {
shimniok 28:753db82debb1 77 // pop most recent directory
shimniok 28:753db82debb1 78 if (!pstack.empty())
shimniok 28:753db82debb1 79 pstack.pop_back();
shimniok 28:753db82debb1 80 } else if (strcmp(".", e) != 0) {
shimniok 28:753db82debb1 81 // push this dir onto path
shimniok 28:753db82debb1 82 if (strlen(e) > 0)
shimniok 28:753db82debb1 83 pstack.push_back(e);
shimniok 28:753db82debb1 84 }
shimniok 28:753db82debb1 85 e = strtok(NULL, "/");
shimniok 28:753db82debb1 86 }
shimniok 21:5d7ac1f0b842 87
shimniok 28:753db82debb1 88 static std::string result;
shimniok 28:753db82debb1 89 result = "";
shimniok 28:753db82debb1 90
shimniok 28:753db82debb1 91 for (path_t::iterator it = pstack.begin(); it != pstack.end(); it++) {
shimniok 28:753db82debb1 92 result.append("/");
shimniok 28:753db82debb1 93 result.append(*it);
shimniok 28:753db82debb1 94 }
shimniok 28:753db82debb1 95
shimniok 28:753db82debb1 96 // if empty, add a /
shimniok 28:753db82debb1 97 if (result.size() < 2) {
shimniok 28:753db82debb1 98 result.append("/");
shimniok 28:753db82debb1 99 }
shimniok 28:753db82debb1 100
shimniok 28:753db82debb1 101 strcpy(tmp, result.c_str());
shimniok 28:753db82debb1 102
shimniok 28:753db82debb1 103 return tmp;
shimniok 21:5d7ac1f0b842 104 }
shimniok 21:5d7ac1f0b842 105
shimniok 21:5d7ac1f0b842 106
shimniok 23:b1e49cfcaef6 107 char *SimpleShell::basename(char *path) {
shimniok 33:84dc443909a0 108 char *result = strrchr(path, '/');
shimniok 24:004e33abce37 109
shimniok 33:84dc443909a0 110 if (result == NULL) {
shimniok 33:84dc443909a0 111 result = path;
shimniok 33:84dc443909a0 112 } else {
shimniok 33:84dc443909a0 113 if (result[0] == '/') result++;
shimniok 24:004e33abce37 114 }
shimniok 24:004e33abce37 115
shimniok 24:004e33abce37 116 return result;
shimniok 23:b1e49cfcaef6 117 }
shimniok 23:b1e49cfcaef6 118
shimniok 23:b1e49cfcaef6 119
shimniok 21:5d7ac1f0b842 120 void SimpleShell::help(int argc, char **argv)
shimniok 21:5d7ac1f0b842 121 {
shimniok 21:5d7ac1f0b842 122 printf("Available commands: ");
shimniok 21:5d7ac1f0b842 123 for (int i=0; i < lookupEnd; i++) {
shimniok 21:5d7ac1f0b842 124 printf("%s ", lookup[i].command);
shimniok 21:5d7ac1f0b842 125 }
shimniok 21:5d7ac1f0b842 126 printf("\n\n");
shimniok 21:5d7ac1f0b842 127 }
shimniok 21:5d7ac1f0b842 128
shimniok 21:5d7ac1f0b842 129
shimniok 21:5d7ac1f0b842 130 void SimpleShell::cd(int argc, char **argv)
shimniok 21:5d7ac1f0b842 131 {
shimniok 29:8d4132274445 132 if (argc == 1) {
shimniok 29:8d4132274445 133 strncpy(_cwd, _home, MAXBUF);
shimniok 29:8d4132274445 134 } else if (argc == 2) {
shimniok 28:753db82debb1 135 strncpy(_cwd, canon(argv[1]), MAXBUF);
shimniok 21:5d7ac1f0b842 136 } else {
shimniok 21:5d7ac1f0b842 137 puts("usage: cd directory");
shimniok 21:5d7ac1f0b842 138 }
shimniok 21:5d7ac1f0b842 139 return;
shimniok 21:5d7ac1f0b842 140 }
shimniok 21:5d7ac1f0b842 141
shimniok 21:5d7ac1f0b842 142
shimniok 21:5d7ac1f0b842 143 void SimpleShell::pwd(int argc, char **argv)
shimniok 21:5d7ac1f0b842 144 {
shimniok 21:5d7ac1f0b842 145 puts(_cwd);
shimniok 21:5d7ac1f0b842 146 return;
shimniok 21:5d7ac1f0b842 147 }
shimniok 21:5d7ac1f0b842 148
shimniok 21:5d7ac1f0b842 149
shimniok 21:5d7ac1f0b842 150 void SimpleShell::ls(int argc, char **argv)
shimniok 21:5d7ac1f0b842 151 {
shimniok 21:5d7ac1f0b842 152 DIR *d;
shimniok 21:5d7ac1f0b842 153 struct dirent *p;
shimniok 21:5d7ac1f0b842 154 char *path;
shimniok 21:5d7ac1f0b842 155
shimniok 21:5d7ac1f0b842 156 if (argc == 1) {
shimniok 21:5d7ac1f0b842 157 path = _cwd;
shimniok 21:5d7ac1f0b842 158 } else if (argc == 2) {
shimniok 28:753db82debb1 159 path = canon(argv[1]);
shimniok 21:5d7ac1f0b842 160 } else {
shimniok 21:5d7ac1f0b842 161 puts("usage: ls [directory]");
shimniok 21:5d7ac1f0b842 162 return;
shimniok 21:5d7ac1f0b842 163 }
shimniok 21:5d7ac1f0b842 164
shimniok 21:5d7ac1f0b842 165 int cols=0;
shimniok 21:5d7ac1f0b842 166 if ((d = opendir(path)) != NULL) {
shimniok 21:5d7ac1f0b842 167 while ((p = readdir(d)) != NULL) {
shimniok 21:5d7ac1f0b842 168 if (p->d_name && p->d_name[0] != 0xff) {
shimniok 21:5d7ac1f0b842 169 if (cols++ > 3) {
shimniok 21:5d7ac1f0b842 170 putc('\n', stdout);
shimniok 21:5d7ac1f0b842 171 cols = 0;
shimniok 21:5d7ac1f0b842 172 }
shimniok 21:5d7ac1f0b842 173 printf("%-15s ", p->d_name);
shimniok 21:5d7ac1f0b842 174 }
shimniok 21:5d7ac1f0b842 175 }
shimniok 21:5d7ac1f0b842 176 putc('\n', stdout);
shimniok 21:5d7ac1f0b842 177 if (cols < 3)
shimniok 21:5d7ac1f0b842 178 putc('\n', stdout);
shimniok 21:5d7ac1f0b842 179 closedir(d);
shimniok 21:5d7ac1f0b842 180 } else {
shimniok 21:5d7ac1f0b842 181 puts(path);
shimniok 21:5d7ac1f0b842 182 puts(": No such directory\n");
shimniok 21:5d7ac1f0b842 183 }
shimniok 21:5d7ac1f0b842 184
shimniok 21:5d7ac1f0b842 185 return;
shimniok 21:5d7ac1f0b842 186 }
shimniok 21:5d7ac1f0b842 187
shimniok 32:fc09f0eb1e8a 188 #include "fnmatch.h"
shimniok 32:fc09f0eb1e8a 189
shimniok 32:fc09f0eb1e8a 190 list<char*> *scandir(char *pattern, char *path)
shimniok 32:fc09f0eb1e8a 191 {
shimniok 32:fc09f0eb1e8a 192 DIR *d;
shimniok 32:fc09f0eb1e8a 193 list<char*> *listp;
shimniok 32:fc09f0eb1e8a 194
shimniok 32:fc09f0eb1e8a 195 struct dirent *p;
shimniok 32:fc09f0eb1e8a 196 if ((d = opendir(path)) != NULL) {
shimniok 32:fc09f0eb1e8a 197 listp = new list<char*>;
shimniok 32:fc09f0eb1e8a 198 while ((p = readdir(d)) != NULL) {
shimniok 32:fc09f0eb1e8a 199 if (fnmatch(pattern, p->d_name, FNM_PATHNAME|FNM_CASEFOLD) != FNM_NOMATCH) {
shimniok 32:fc09f0eb1e8a 200 char *s = new char[sizeof(p->d_name)];
shimniok 32:fc09f0eb1e8a 201 strcpy(s, p->d_name);
shimniok 32:fc09f0eb1e8a 202 listp->push_back(s);
shimniok 32:fc09f0eb1e8a 203 }
shimniok 32:fc09f0eb1e8a 204 }
shimniok 32:fc09f0eb1e8a 205 closedir(d);
shimniok 32:fc09f0eb1e8a 206 }
shimniok 32:fc09f0eb1e8a 207 return listp;
shimniok 32:fc09f0eb1e8a 208 }
shimniok 32:fc09f0eb1e8a 209
shimniok 21:5d7ac1f0b842 210
shimniok 21:5d7ac1f0b842 211 void SimpleShell::rm(int argc, char **argv)
shimniok 21:5d7ac1f0b842 212 {
shimniok 21:5d7ac1f0b842 213 if (argc >= 2) {
shimniok 21:5d7ac1f0b842 214 for (int i=1; i < argc; i++) {
shimniok 33:84dc443909a0 215 char *arg = canon(argv[i]);
shimniok 33:84dc443909a0 216 char *base = basename(arg);
shimniok 33:84dc443909a0 217
shimniok 33:84dc443909a0 218 printf("arg=<%s>\nbase=<%s>\n", arg, base);
shimniok 33:84dc443909a0 219 return;
shimniok 33:84dc443909a0 220 }
shimniok 33:84dc443909a0 221
shimniok 33:84dc443909a0 222 /*
shimniok 33:84dc443909a0 223 if (haswildcard(arg) {
shimniok 32:fc09f0eb1e8a 224 list<char*> *fl = scandir(argv[i], _cwd);
shimniok 32:fc09f0eb1e8a 225 char *s;
shimniok 32:fc09f0eb1e8a 226 while (!fl->empty()) {
shimniok 32:fc09f0eb1e8a 227 s = fl->front();
shimniok 32:fc09f0eb1e8a 228 printf("<%s>\n", s);
shimniok 32:fc09f0eb1e8a 229 fl->pop_front();
shimniok 32:fc09f0eb1e8a 230 free(s);
shimniok 32:fc09f0eb1e8a 231 }
shimniok 32:fc09f0eb1e8a 232 delete(fl);
shimniok 32:fc09f0eb1e8a 233 } else {
shimniok 32:fc09f0eb1e8a 234 if (remove(canon(argv[i]))) {
shimniok 32:fc09f0eb1e8a 235 printf("%s: cannot remove\n", argv[i]);
shimniok 32:fc09f0eb1e8a 236 }
shimniok 21:5d7ac1f0b842 237 }
shimniok 33:84dc443909a0 238 }*/
shimniok 21:5d7ac1f0b842 239 } else {
shimniok 21:5d7ac1f0b842 240 puts("usage: rm [file1 [file2 ...]]");
shimniok 21:5d7ac1f0b842 241 }
shimniok 21:5d7ac1f0b842 242 }
shimniok 21:5d7ac1f0b842 243
shimniok 21:5d7ac1f0b842 244
shimniok 21:5d7ac1f0b842 245 void SimpleShell::touch(int argc, char **argv)
shimniok 21:5d7ac1f0b842 246 {
shimniok 21:5d7ac1f0b842 247 FILE *fp;
shimniok 21:5d7ac1f0b842 248
shimniok 21:5d7ac1f0b842 249 if (argc >= 2) {
shimniok 21:5d7ac1f0b842 250 for (int i=1; i < argc; i++) {
shimniok 21:5d7ac1f0b842 251 if ((fp = fopen(canon(argv[i]), "w")) != NULL) {
shimniok 21:5d7ac1f0b842 252 fclose(fp);
shimniok 21:5d7ac1f0b842 253 } else {
shimniok 21:5d7ac1f0b842 254 printf("%s: cannot touch\n", argv[1]);
shimniok 21:5d7ac1f0b842 255 }
shimniok 21:5d7ac1f0b842 256 }
shimniok 21:5d7ac1f0b842 257 } else {
shimniok 21:5d7ac1f0b842 258 puts("usage: touch [file1 [file2 ...]]");
shimniok 21:5d7ac1f0b842 259 }
shimniok 21:5d7ac1f0b842 260 }
shimniok 21:5d7ac1f0b842 261
shimniok 21:5d7ac1f0b842 262
shimniok 21:5d7ac1f0b842 263 void SimpleShell::cat(int argc, char **argv)
shimniok 21:5d7ac1f0b842 264 {
shimniok 21:5d7ac1f0b842 265 FILE *fp;
shimniok 21:5d7ac1f0b842 266 //int status=0;
shimniok 21:5d7ac1f0b842 267 char *buf = new char[MAXBUF];
shimniok 21:5d7ac1f0b842 268
shimniok 21:5d7ac1f0b842 269 for (int i=1; i < argc; i++) {
shimniok 21:5d7ac1f0b842 270 //resolveDirectory(path, arg);
shimniok 21:5d7ac1f0b842 271 if ((fp = fopen(canon(argv[i]), "r")) != NULL) {
shimniok 21:5d7ac1f0b842 272 while (!feof(fp)) {
shimniok 21:5d7ac1f0b842 273 fgets(buf, MAXBUF-1, fp);
shimniok 21:5d7ac1f0b842 274 fputs(buf, stdout);
shimniok 21:5d7ac1f0b842 275 }
shimniok 21:5d7ac1f0b842 276 fclose(fp);
shimniok 21:5d7ac1f0b842 277 } else {
shimniok 21:5d7ac1f0b842 278 fputs(argv[i], stdout);
shimniok 21:5d7ac1f0b842 279 fputs(": No such file\n", stdout);
shimniok 21:5d7ac1f0b842 280 //status = 1;
shimniok 21:5d7ac1f0b842 281 }
shimniok 21:5d7ac1f0b842 282 }
shimniok 21:5d7ac1f0b842 283 delete[] buf;
shimniok 21:5d7ac1f0b842 284
shimniok 21:5d7ac1f0b842 285 return;
shimniok 21:5d7ac1f0b842 286 }
shimniok 21:5d7ac1f0b842 287
shimniok 21:5d7ac1f0b842 288
shimniok 23:b1e49cfcaef6 289 void SimpleShell::send(int argc, char **argv)
shimniok 23:b1e49cfcaef6 290 {
shimniok 23:b1e49cfcaef6 291 const char SOF=0x01;
shimniok 23:b1e49cfcaef6 292 const char ETX=0x03;
shimniok 23:b1e49cfcaef6 293 const char EOT=0x04;
shimniok 23:b1e49cfcaef6 294 const char ACK=0x07;
shimniok 23:b1e49cfcaef6 295 const char NACK=0x18;
shimniok 24:004e33abce37 296 char buf[1024];
shimniok 23:b1e49cfcaef6 297 FILE *fp;
shimniok 24:004e33abce37 298 char c;
shimniok 23:b1e49cfcaef6 299
shimniok 23:b1e49cfcaef6 300 int i = 1;
shimniok 23:b1e49cfcaef6 301 if (argc >= 2) {
shimniok 23:b1e49cfcaef6 302 if ((fp = fopen(canon(argv[i]), "r")) == 0) {
shimniok 23:b1e49cfcaef6 303 printf("%s: can't open file\n", basename(argv[i]));
shimniok 23:b1e49cfcaef6 304 } else {
shimniok 24:004e33abce37 305 puts("Ready; initiate receive on client.");
shimniok 24:004e33abce37 306 c = getchar();
shimniok 24:004e33abce37 307 if (c == SOF) {
shimniok 24:004e33abce37 308 puts(basename(argv[i]));
shimniok 24:004e33abce37 309 c = getchar();
shimniok 24:004e33abce37 310 if (c == ACK) {
shimniok 24:004e33abce37 311 while (!feof(fp)) {
shimniok 24:004e33abce37 312 fgets(buf, 1024, fp);
shimniok 24:004e33abce37 313 fputs(buf, stdout);
shimniok 24:004e33abce37 314 }
shimniok 24:004e33abce37 315 putchar(EOT);
shimniok 24:004e33abce37 316 } else if (c == NACK) {
shimniok 24:004e33abce37 317 puts("client error.");
shimniok 24:004e33abce37 318 } else {
shimniok 24:004e33abce37 319 puts("ACK/NACK expected");
shimniok 24:004e33abce37 320 }
shimniok 24:004e33abce37 321 } else if (c == NACK) {
shimniok 24:004e33abce37 322 puts("server cancelled.");
shimniok 24:004e33abce37 323 } else {
shimniok 24:004e33abce37 324 puts("SOF/NACK expected.");
shimniok 24:004e33abce37 325 }
shimniok 24:004e33abce37 326 fclose(fp);
shimniok 23:b1e49cfcaef6 327 }
shimniok 23:b1e49cfcaef6 328 } else {
shimniok 23:b1e49cfcaef6 329 puts("usage: send file1 [file2 ...]");
shimniok 23:b1e49cfcaef6 330 }
shimniok 23:b1e49cfcaef6 331 }
shimniok 23:b1e49cfcaef6 332
shimniok 23:b1e49cfcaef6 333
shimniok 21:5d7ac1f0b842 334 void SimpleShell::run()
shimniok 21:5d7ac1f0b842 335 {
shimniok 21:5d7ac1f0b842 336 bool done=false;
shimniok 21:5d7ac1f0b842 337 callback_t cb;
shimniok 21:5d7ac1f0b842 338 //int status; // TODO implement command status return
shimniok 21:5d7ac1f0b842 339 std::string x;
shimniok 21:5d7ac1f0b842 340
shimniok 21:5d7ac1f0b842 341 printf("Type help for assistance.\n");
shimniok 21:5d7ac1f0b842 342 help(0, NULL);
shimniok 21:5d7ac1f0b842 343 while (!done) {
shimniok 21:5d7ac1f0b842 344 printPrompt();
shimniok 21:5d7ac1f0b842 345 readCommand();
shimniok 21:5d7ac1f0b842 346 if (argv[0]) { // skip blank command
shimniok 21:5d7ac1f0b842 347 if (cb = findCommand()) {
shimniok 21:5d7ac1f0b842 348 cb.call(argc, argv);
shimniok 21:5d7ac1f0b842 349 } else {
shimniok 21:5d7ac1f0b842 350 printf("command <%s> not found\n", argv[0]);
shimniok 21:5d7ac1f0b842 351 }
shimniok 21:5d7ac1f0b842 352 }
shimniok 21:5d7ac1f0b842 353 }
shimniok 21:5d7ac1f0b842 354 puts("exiting shell\n");
shimniok 21:5d7ac1f0b842 355
shimniok 21:5d7ac1f0b842 356 return;
shimniok 21:5d7ac1f0b842 357 }
shimniok 21:5d7ac1f0b842 358
shimniok 21:5d7ac1f0b842 359
shimniok 28:753db82debb1 360 void SimpleShell::command(callback_t cb, char *command)
shimniok 21:5d7ac1f0b842 361 {
shimniok 21:5d7ac1f0b842 362 if (lookupEnd < MAXLOOKUP) {
shimniok 21:5d7ac1f0b842 363 lookup[lookupEnd].cb = cb;
shimniok 21:5d7ac1f0b842 364 lookup[lookupEnd].command = command;
shimniok 21:5d7ac1f0b842 365 lookupEnd++;
shimniok 21:5d7ac1f0b842 366 }
shimniok 21:5d7ac1f0b842 367
shimniok 21:5d7ac1f0b842 368 return;
shimniok 21:5d7ac1f0b842 369 }
shimniok 21:5d7ac1f0b842 370
shimniok 21:5d7ac1f0b842 371
shimniok 21:5d7ac1f0b842 372 SimpleShell::callback_t SimpleShell::findCommand()
shimniok 21:5d7ac1f0b842 373 {
shimniok 21:5d7ac1f0b842 374 SimpleShell::callback_t cb=NULL;
shimniok 21:5d7ac1f0b842 375
shimniok 21:5d7ac1f0b842 376 for (int i=0; i < lookupEnd; i++) {
shimniok 21:5d7ac1f0b842 377 if (strncmp(argv[0], lookup[i].command, MAXBUF) == 0) {
shimniok 21:5d7ac1f0b842 378 cb = lookup[i].cb;
shimniok 21:5d7ac1f0b842 379 break;
shimniok 21:5d7ac1f0b842 380 }
shimniok 21:5d7ac1f0b842 381 }
shimniok 21:5d7ac1f0b842 382
shimniok 21:5d7ac1f0b842 383 return cb;
shimniok 21:5d7ac1f0b842 384 }
shimniok 21:5d7ac1f0b842 385
shimniok 21:5d7ac1f0b842 386
shimniok 21:5d7ac1f0b842 387 void SimpleShell::printPrompt()
shimniok 21:5d7ac1f0b842 388 {
shimniok 21:5d7ac1f0b842 389 fputc('(', stdout);
shimniok 21:5d7ac1f0b842 390 fputs(_cwd, stdout);
shimniok 21:5d7ac1f0b842 391 fputs(")# ", stdout);
shimniok 21:5d7ac1f0b842 392
shimniok 21:5d7ac1f0b842 393 return;
shimniok 21:5d7ac1f0b842 394 }
shimniok 21:5d7ac1f0b842 395
shimniok 21:5d7ac1f0b842 396
shimniok 21:5d7ac1f0b842 397 void SimpleShell::readCommand()
shimniok 21:5d7ac1f0b842 398 {
shimniok 21:5d7ac1f0b842 399 int i=0;
shimniok 21:5d7ac1f0b842 400 char c;
shimniok 21:5d7ac1f0b842 401 bool done = false;
shimniok 21:5d7ac1f0b842 402 static char cmd[MAXBUF];
shimniok 21:5d7ac1f0b842 403
shimniok 21:5d7ac1f0b842 404 memset(cmd, 0, MAXBUF);
shimniok 21:5d7ac1f0b842 405 do {
shimniok 21:5d7ac1f0b842 406 cmd[i] = 0;
shimniok 21:5d7ac1f0b842 407 c = fgetc(stdin);
shimniok 21:5d7ac1f0b842 408 if (c == '\r') { // if return is hit, we're done, don't add \r to cmd
shimniok 21:5d7ac1f0b842 409 done = true;
shimniok 21:5d7ac1f0b842 410 } else if (c == ESC) { // keyboard escape codes (arrow keys, etc)
shimniok 21:5d7ac1f0b842 411 int c2 = getchar();
shimniok 21:5d7ac1f0b842 412 int c3 = getchar();
shimniok 21:5d7ac1f0b842 413
shimniok 21:5d7ac1f0b842 414 if (c2 == 0x4f && c3 == 0x46) {
shimniok 21:5d7ac1f0b842 415 printf("<END>");
shimniok 21:5d7ac1f0b842 416 } else if (c2 == 0x5b) {
shimniok 21:5d7ac1f0b842 417
shimniok 21:5d7ac1f0b842 418 if (c3 == UP) {
shimniok 21:5d7ac1f0b842 419 printf("<UP>");
shimniok 21:5d7ac1f0b842 420 } else if (c3 == DOWN) {
shimniok 21:5d7ac1f0b842 421 printf("<DOWN>");
shimniok 21:5d7ac1f0b842 422 } else if (c3 == RIGHT) {
shimniok 21:5d7ac1f0b842 423 printf("<RIGHT>");
shimniok 21:5d7ac1f0b842 424 } else if (c3 == LEFT) {
shimniok 21:5d7ac1f0b842 425 printf("<LEFT>");
shimniok 21:5d7ac1f0b842 426 } else if (c3 == HOME || c3 == INS || c3 == DEL ||
shimniok 21:5d7ac1f0b842 427 c3 == PGUP || c3 == PGDN) {
shimniok 21:5d7ac1f0b842 428 char c4 = getchar();
shimniok 21:5d7ac1f0b842 429 if (c4 == TAIL) {
shimniok 21:5d7ac1f0b842 430 if (c4 == HOME) {
shimniok 21:5d7ac1f0b842 431 printf("<HOME>");
shimniok 21:5d7ac1f0b842 432 } else if (c4 == INS) {
shimniok 21:5d7ac1f0b842 433 printf("<INS>");
shimniok 21:5d7ac1f0b842 434 } else if (c4 == DEL) {
shimniok 21:5d7ac1f0b842 435 printf("<DEL>");
shimniok 21:5d7ac1f0b842 436 } else if (c4 == PGUP) {
shimniok 21:5d7ac1f0b842 437 printf("<PGUP>");
shimniok 21:5d7ac1f0b842 438 } else if (c4 == PGDN) {
shimniok 21:5d7ac1f0b842 439 printf("<PGDN>");
shimniok 21:5d7ac1f0b842 440 }//if
shimniok 21:5d7ac1f0b842 441 }//if
shimniok 21:5d7ac1f0b842 442 }//if
shimniok 21:5d7ac1f0b842 443 }//if
shimniok 21:5d7ac1f0b842 444 //printf("\n");
shimniok 21:5d7ac1f0b842 445 } else if (i < MAXBUF-1) {
shimniok 21:5d7ac1f0b842 446 if (c == 0x7f || c == '\b') { // backspace or delete
shimniok 21:5d7ac1f0b842 447 if (i > 0) { // if we're at the beginning, do nothing
shimniok 21:5d7ac1f0b842 448 i--;
shimniok 21:5d7ac1f0b842 449 fputs("\b \b", stdout);
shimniok 21:5d7ac1f0b842 450 }
shimniok 21:5d7ac1f0b842 451 } else {
shimniok 21:5d7ac1f0b842 452 if (isprint(c))
shimniok 21:5d7ac1f0b842 453 fputc(c, stdout);
shimniok 21:5d7ac1f0b842 454 cmd[i++] = c;
shimniok 21:5d7ac1f0b842 455 }
shimniok 21:5d7ac1f0b842 456 }
shimniok 21:5d7ac1f0b842 457
shimniok 21:5d7ac1f0b842 458 } while (!done);
shimniok 21:5d7ac1f0b842 459 fputc('\n', stdout);
shimniok 21:5d7ac1f0b842 460
shimniok 21:5d7ac1f0b842 461 // remove leading/trailing whitespace
shimniok 21:5d7ac1f0b842 462 char *s = cmd;
shimniok 21:5d7ac1f0b842 463 while (isspace(*s)) {
shimniok 21:5d7ac1f0b842 464 s++;
shimniok 21:5d7ac1f0b842 465 }
shimniok 21:5d7ac1f0b842 466 for (int j=i; j >= 0 && isspace(cmd[j]); j--) {
shimniok 21:5d7ac1f0b842 467 cmd[j] = '\0';
shimniok 21:5d7ac1f0b842 468 }
shimniok 21:5d7ac1f0b842 469
shimniok 21:5d7ac1f0b842 470 // split into command and arguments
shimniok 21:5d7ac1f0b842 471 argc = 0;
shimniok 21:5d7ac1f0b842 472 char *t;
shimniok 21:5d7ac1f0b842 473
shimniok 21:5d7ac1f0b842 474 for (int i=0; i < MAXARGS; i++) {
shimniok 21:5d7ac1f0b842 475 argv[i] = NULL;
shimniok 21:5d7ac1f0b842 476 }
shimniok 21:5d7ac1f0b842 477 t = strtok(s, " ");
shimniok 30:35522ea06236 478 while (t && argc < MAXARGS) {
shimniok 21:5d7ac1f0b842 479 argv[argc++] = t;
shimniok 21:5d7ac1f0b842 480 t = strtok(NULL, " ");
shimniok 21:5d7ac1f0b842 481 }
shimniok 21:5d7ac1f0b842 482
shimniok 21:5d7ac1f0b842 483 return;
shimniok 21:5d7ac1f0b842 484 }