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@32:fc09f0eb1e8a, 2018-12-28 (annotated)

Committer:
shimniok
Date:
Fri Dec 28 19:54:48 2018 +0000
Revision:
32:fc09f0eb1e8a
Parent:
31:27e8130a0d8f
Child:
33:84dc443909a0
try out wildcard file match for rm

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 24:004e33abce37 108 char *result = path + strlen(path);
shimniok 24:004e33abce37 109
shimniok 24:004e33abce37 110 while (result != path) {
shimniok 24:004e33abce37 111 if (*result == '/') {
shimniok 24:004e33abce37 112 result++;
shimniok 24:004e33abce37 113 break;
shimniok 24:004e33abce37 114 }
shimniok 24:004e33abce37 115 result--;
shimniok 24:004e33abce37 116 }
shimniok 24:004e33abce37 117
shimniok 24:004e33abce37 118 return result;
shimniok 23:b1e49cfcaef6 119 }
shimniok 23:b1e49cfcaef6 120
shimniok 23:b1e49cfcaef6 121
shimniok 21:5d7ac1f0b842 122 void SimpleShell::help(int argc, char **argv)
shimniok 21:5d7ac1f0b842 123 {
shimniok 21:5d7ac1f0b842 124 printf("Available commands: ");
shimniok 21:5d7ac1f0b842 125 for (int i=0; i < lookupEnd; i++) {
shimniok 21:5d7ac1f0b842 126 printf("%s ", lookup[i].command);
shimniok 21:5d7ac1f0b842 127 }
shimniok 21:5d7ac1f0b842 128 printf("\n\n");
shimniok 21:5d7ac1f0b842 129 }
shimniok 21:5d7ac1f0b842 130
shimniok 21:5d7ac1f0b842 131
shimniok 21:5d7ac1f0b842 132 void SimpleShell::cd(int argc, char **argv)
shimniok 21:5d7ac1f0b842 133 {
shimniok 29:8d4132274445 134 if (argc == 1) {
shimniok 29:8d4132274445 135 strncpy(_cwd, _home, MAXBUF);
shimniok 29:8d4132274445 136 } else if (argc == 2) {
shimniok 28:753db82debb1 137 strncpy(_cwd, canon(argv[1]), MAXBUF);
shimniok 21:5d7ac1f0b842 138 } else {
shimniok 21:5d7ac1f0b842 139 puts("usage: cd directory");
shimniok 21:5d7ac1f0b842 140 }
shimniok 21:5d7ac1f0b842 141 return;
shimniok 21:5d7ac1f0b842 142 }
shimniok 21:5d7ac1f0b842 143
shimniok 21:5d7ac1f0b842 144
shimniok 21:5d7ac1f0b842 145 void SimpleShell::pwd(int argc, char **argv)
shimniok 21:5d7ac1f0b842 146 {
shimniok 21:5d7ac1f0b842 147 puts(_cwd);
shimniok 21:5d7ac1f0b842 148 return;
shimniok 21:5d7ac1f0b842 149 }
shimniok 21:5d7ac1f0b842 150
shimniok 21:5d7ac1f0b842 151
shimniok 21:5d7ac1f0b842 152 void SimpleShell::ls(int argc, char **argv)
shimniok 21:5d7ac1f0b842 153 {
shimniok 21:5d7ac1f0b842 154 DIR *d;
shimniok 21:5d7ac1f0b842 155 struct dirent *p;
shimniok 21:5d7ac1f0b842 156 char *path;
shimniok 21:5d7ac1f0b842 157
shimniok 21:5d7ac1f0b842 158 if (argc == 1) {
shimniok 21:5d7ac1f0b842 159 path = _cwd;
shimniok 21:5d7ac1f0b842 160 } else if (argc == 2) {
shimniok 28:753db82debb1 161 path = canon(argv[1]);
shimniok 21:5d7ac1f0b842 162 } else {
shimniok 21:5d7ac1f0b842 163 puts("usage: ls [directory]");
shimniok 21:5d7ac1f0b842 164 return;
shimniok 21:5d7ac1f0b842 165 }
shimniok 21:5d7ac1f0b842 166
shimniok 21:5d7ac1f0b842 167 int cols=0;
shimniok 21:5d7ac1f0b842 168 if ((d = opendir(path)) != NULL) {
shimniok 21:5d7ac1f0b842 169 while ((p = readdir(d)) != NULL) {
shimniok 21:5d7ac1f0b842 170 if (p->d_name && p->d_name[0] != 0xff) {
shimniok 21:5d7ac1f0b842 171 if (cols++ > 3) {
shimniok 21:5d7ac1f0b842 172 putc('\n', stdout);
shimniok 21:5d7ac1f0b842 173 cols = 0;
shimniok 21:5d7ac1f0b842 174 }
shimniok 21:5d7ac1f0b842 175 printf("%-15s ", p->d_name);
shimniok 21:5d7ac1f0b842 176 }
shimniok 21:5d7ac1f0b842 177 }
shimniok 21:5d7ac1f0b842 178 putc('\n', stdout);
shimniok 21:5d7ac1f0b842 179 if (cols < 3)
shimniok 21:5d7ac1f0b842 180 putc('\n', stdout);
shimniok 21:5d7ac1f0b842 181 closedir(d);
shimniok 21:5d7ac1f0b842 182 } else {
shimniok 21:5d7ac1f0b842 183 puts(path);
shimniok 21:5d7ac1f0b842 184 puts(": No such directory\n");
shimniok 21:5d7ac1f0b842 185 }
shimniok 21:5d7ac1f0b842 186
shimniok 21:5d7ac1f0b842 187 return;
shimniok 21:5d7ac1f0b842 188 }
shimniok 21:5d7ac1f0b842 189
shimniok 32:fc09f0eb1e8a 190 #include "fnmatch.h"
shimniok 32:fc09f0eb1e8a 191
shimniok 32:fc09f0eb1e8a 192 list<char*> *scandir(char *pattern, char *path)
shimniok 32:fc09f0eb1e8a 193 {
shimniok 32:fc09f0eb1e8a 194 DIR *d;
shimniok 32:fc09f0eb1e8a 195 list<char*> *listp;
shimniok 32:fc09f0eb1e8a 196
shimniok 32:fc09f0eb1e8a 197 struct dirent *p;
shimniok 32:fc09f0eb1e8a 198 if ((d = opendir(path)) != NULL) {
shimniok 32:fc09f0eb1e8a 199 listp = new list<char*>;
shimniok 32:fc09f0eb1e8a 200 while ((p = readdir(d)) != NULL) {
shimniok 32:fc09f0eb1e8a 201 if (fnmatch(pattern, p->d_name, FNM_PATHNAME|FNM_CASEFOLD) != FNM_NOMATCH) {
shimniok 32:fc09f0eb1e8a 202 char *s = new char[sizeof(p->d_name)];
shimniok 32:fc09f0eb1e8a 203 strcpy(s, p->d_name);
shimniok 32:fc09f0eb1e8a 204 listp->push_back(s);
shimniok 32:fc09f0eb1e8a 205 }
shimniok 32:fc09f0eb1e8a 206 }
shimniok 32:fc09f0eb1e8a 207 closedir(d);
shimniok 32:fc09f0eb1e8a 208 }
shimniok 32:fc09f0eb1e8a 209 return listp;
shimniok 32:fc09f0eb1e8a 210 }
shimniok 32:fc09f0eb1e8a 211
shimniok 21:5d7ac1f0b842 212
shimniok 21:5d7ac1f0b842 213 void SimpleShell::rm(int argc, char **argv)
shimniok 21:5d7ac1f0b842 214 {
shimniok 21:5d7ac1f0b842 215 if (argc >= 2) {
shimniok 21:5d7ac1f0b842 216 for (int i=1; i < argc; i++) {
shimniok 32:fc09f0eb1e8a 217 if (haswildcard(argv[i])) {
shimniok 32:fc09f0eb1e8a 218 list<char*> *fl = scandir(argv[i], _cwd);
shimniok 32:fc09f0eb1e8a 219 char *s;
shimniok 32:fc09f0eb1e8a 220 while (!fl->empty()) {
shimniok 32:fc09f0eb1e8a 221 s = fl->front();
shimniok 32:fc09f0eb1e8a 222 printf("<%s>\n", s);
shimniok 32:fc09f0eb1e8a 223 fl->pop_front();
shimniok 32:fc09f0eb1e8a 224 free(s);
shimniok 32:fc09f0eb1e8a 225 }
shimniok 32:fc09f0eb1e8a 226 delete(fl);
shimniok 32:fc09f0eb1e8a 227 } else {
shimniok 32:fc09f0eb1e8a 228 if (remove(canon(argv[i]))) {
shimniok 32:fc09f0eb1e8a 229 printf("%s: cannot remove\n", argv[i]);
shimniok 32:fc09f0eb1e8a 230 }
shimniok 21:5d7ac1f0b842 231 }
shimniok 21:5d7ac1f0b842 232 }
shimniok 21:5d7ac1f0b842 233 } else {
shimniok 21:5d7ac1f0b842 234 puts("usage: rm [file1 [file2 ...]]");
shimniok 21:5d7ac1f0b842 235 }
shimniok 21:5d7ac1f0b842 236 }
shimniok 21:5d7ac1f0b842 237
shimniok 21:5d7ac1f0b842 238
shimniok 21:5d7ac1f0b842 239 void SimpleShell::touch(int argc, char **argv)
shimniok 21:5d7ac1f0b842 240 {
shimniok 21:5d7ac1f0b842 241 FILE *fp;
shimniok 21:5d7ac1f0b842 242
shimniok 21:5d7ac1f0b842 243 if (argc >= 2) {
shimniok 21:5d7ac1f0b842 244 for (int i=1; i < argc; i++) {
shimniok 21:5d7ac1f0b842 245 if ((fp = fopen(canon(argv[i]), "w")) != NULL) {
shimniok 21:5d7ac1f0b842 246 fclose(fp);
shimniok 21:5d7ac1f0b842 247 } else {
shimniok 21:5d7ac1f0b842 248 printf("%s: cannot touch\n", argv[1]);
shimniok 21:5d7ac1f0b842 249 }
shimniok 21:5d7ac1f0b842 250 }
shimniok 21:5d7ac1f0b842 251 } else {
shimniok 21:5d7ac1f0b842 252 puts("usage: touch [file1 [file2 ...]]");
shimniok 21:5d7ac1f0b842 253 }
shimniok 21:5d7ac1f0b842 254 }
shimniok 21:5d7ac1f0b842 255
shimniok 21:5d7ac1f0b842 256
shimniok 21:5d7ac1f0b842 257 void SimpleShell::cat(int argc, char **argv)
shimniok 21:5d7ac1f0b842 258 {
shimniok 21:5d7ac1f0b842 259 FILE *fp;
shimniok 21:5d7ac1f0b842 260 //int status=0;
shimniok 21:5d7ac1f0b842 261 char *buf = new char[MAXBUF];
shimniok 21:5d7ac1f0b842 262
shimniok 21:5d7ac1f0b842 263 for (int i=1; i < argc; i++) {
shimniok 21:5d7ac1f0b842 264 //resolveDirectory(path, arg);
shimniok 21:5d7ac1f0b842 265 if ((fp = fopen(canon(argv[i]), "r")) != NULL) {
shimniok 21:5d7ac1f0b842 266 while (!feof(fp)) {
shimniok 21:5d7ac1f0b842 267 fgets(buf, MAXBUF-1, fp);
shimniok 21:5d7ac1f0b842 268 fputs(buf, stdout);
shimniok 21:5d7ac1f0b842 269 }
shimniok 21:5d7ac1f0b842 270 fclose(fp);
shimniok 21:5d7ac1f0b842 271 } else {
shimniok 21:5d7ac1f0b842 272 fputs(argv[i], stdout);
shimniok 21:5d7ac1f0b842 273 fputs(": No such file\n", stdout);
shimniok 21:5d7ac1f0b842 274 //status = 1;
shimniok 21:5d7ac1f0b842 275 }
shimniok 21:5d7ac1f0b842 276 }
shimniok 21:5d7ac1f0b842 277 delete[] buf;
shimniok 21:5d7ac1f0b842 278
shimniok 21:5d7ac1f0b842 279 return;
shimniok 21:5d7ac1f0b842 280 }
shimniok 21:5d7ac1f0b842 281
shimniok 21:5d7ac1f0b842 282
shimniok 23:b1e49cfcaef6 283 void SimpleShell::send(int argc, char **argv)
shimniok 23:b1e49cfcaef6 284 {
shimniok 23:b1e49cfcaef6 285 const char SOF=0x01;
shimniok 23:b1e49cfcaef6 286 const char ETX=0x03;
shimniok 23:b1e49cfcaef6 287 const char EOT=0x04;
shimniok 23:b1e49cfcaef6 288 const char ACK=0x07;
shimniok 23:b1e49cfcaef6 289 const char NACK=0x18;
shimniok 24:004e33abce37 290 char buf[1024];
shimniok 23:b1e49cfcaef6 291 FILE *fp;
shimniok 24:004e33abce37 292 char c;
shimniok 23:b1e49cfcaef6 293
shimniok 23:b1e49cfcaef6 294 int i = 1;
shimniok 23:b1e49cfcaef6 295 if (argc >= 2) {
shimniok 23:b1e49cfcaef6 296 if ((fp = fopen(canon(argv[i]), "r")) == 0) {
shimniok 23:b1e49cfcaef6 297 printf("%s: can't open file\n", basename(argv[i]));
shimniok 23:b1e49cfcaef6 298 } else {
shimniok 24:004e33abce37 299 puts("Ready; initiate receive on client.");
shimniok 24:004e33abce37 300 c = getchar();
shimniok 24:004e33abce37 301 if (c == SOF) {
shimniok 24:004e33abce37 302 puts(basename(argv[i]));
shimniok 24:004e33abce37 303 c = getchar();
shimniok 24:004e33abce37 304 if (c == ACK) {
shimniok 24:004e33abce37 305 while (!feof(fp)) {
shimniok 24:004e33abce37 306 fgets(buf, 1024, fp);
shimniok 24:004e33abce37 307 fputs(buf, stdout);
shimniok 24:004e33abce37 308 }
shimniok 24:004e33abce37 309 putchar(EOT);
shimniok 24:004e33abce37 310 } else if (c == NACK) {
shimniok 24:004e33abce37 311 puts("client error.");
shimniok 24:004e33abce37 312 } else {
shimniok 24:004e33abce37 313 puts("ACK/NACK expected");
shimniok 24:004e33abce37 314 }
shimniok 24:004e33abce37 315 } else if (c == NACK) {
shimniok 24:004e33abce37 316 puts("server cancelled.");
shimniok 24:004e33abce37 317 } else {
shimniok 24:004e33abce37 318 puts("SOF/NACK expected.");
shimniok 24:004e33abce37 319 }
shimniok 24:004e33abce37 320 fclose(fp);
shimniok 23:b1e49cfcaef6 321 }
shimniok 23:b1e49cfcaef6 322 } else {
shimniok 23:b1e49cfcaef6 323 puts("usage: send file1 [file2 ...]");
shimniok 23:b1e49cfcaef6 324 }
shimniok 23:b1e49cfcaef6 325 }
shimniok 23:b1e49cfcaef6 326
shimniok 23:b1e49cfcaef6 327
shimniok 21:5d7ac1f0b842 328 void SimpleShell::run()
shimniok 21:5d7ac1f0b842 329 {
shimniok 21:5d7ac1f0b842 330 bool done=false;
shimniok 21:5d7ac1f0b842 331 callback_t cb;
shimniok 21:5d7ac1f0b842 332 //int status; // TODO implement command status return
shimniok 21:5d7ac1f0b842 333 std::string x;
shimniok 21:5d7ac1f0b842 334
shimniok 21:5d7ac1f0b842 335 printf("Type help for assistance.\n");
shimniok 21:5d7ac1f0b842 336 help(0, NULL);
shimniok 21:5d7ac1f0b842 337 while (!done) {
shimniok 21:5d7ac1f0b842 338 printPrompt();
shimniok 21:5d7ac1f0b842 339 readCommand();
shimniok 21:5d7ac1f0b842 340 if (argv[0]) { // skip blank command
shimniok 21:5d7ac1f0b842 341 if (cb = findCommand()) {
shimniok 21:5d7ac1f0b842 342 cb.call(argc, argv);
shimniok 21:5d7ac1f0b842 343 } else {
shimniok 21:5d7ac1f0b842 344 printf("command <%s> not found\n", argv[0]);
shimniok 21:5d7ac1f0b842 345 }
shimniok 21:5d7ac1f0b842 346 }
shimniok 21:5d7ac1f0b842 347 }
shimniok 21:5d7ac1f0b842 348 puts("exiting shell\n");
shimniok 21:5d7ac1f0b842 349
shimniok 21:5d7ac1f0b842 350 return;
shimniok 21:5d7ac1f0b842 351 }
shimniok 21:5d7ac1f0b842 352
shimniok 21:5d7ac1f0b842 353
shimniok 28:753db82debb1 354 void SimpleShell::command(callback_t cb, char *command)
shimniok 21:5d7ac1f0b842 355 {
shimniok 21:5d7ac1f0b842 356 if (lookupEnd < MAXLOOKUP) {
shimniok 21:5d7ac1f0b842 357 lookup[lookupEnd].cb = cb;
shimniok 21:5d7ac1f0b842 358 lookup[lookupEnd].command = command;
shimniok 21:5d7ac1f0b842 359 lookupEnd++;
shimniok 21:5d7ac1f0b842 360 }
shimniok 21:5d7ac1f0b842 361
shimniok 21:5d7ac1f0b842 362 return;
shimniok 21:5d7ac1f0b842 363 }
shimniok 21:5d7ac1f0b842 364
shimniok 21:5d7ac1f0b842 365
shimniok 21:5d7ac1f0b842 366 SimpleShell::callback_t SimpleShell::findCommand()
shimniok 21:5d7ac1f0b842 367 {
shimniok 21:5d7ac1f0b842 368 SimpleShell::callback_t cb=NULL;
shimniok 21:5d7ac1f0b842 369
shimniok 21:5d7ac1f0b842 370 for (int i=0; i < lookupEnd; i++) {
shimniok 21:5d7ac1f0b842 371 if (strncmp(argv[0], lookup[i].command, MAXBUF) == 0) {
shimniok 21:5d7ac1f0b842 372 cb = lookup[i].cb;
shimniok 21:5d7ac1f0b842 373 break;
shimniok 21:5d7ac1f0b842 374 }
shimniok 21:5d7ac1f0b842 375 }
shimniok 21:5d7ac1f0b842 376
shimniok 21:5d7ac1f0b842 377 return cb;
shimniok 21:5d7ac1f0b842 378 }
shimniok 21:5d7ac1f0b842 379
shimniok 21:5d7ac1f0b842 380
shimniok 21:5d7ac1f0b842 381 void SimpleShell::printPrompt()
shimniok 21:5d7ac1f0b842 382 {
shimniok 21:5d7ac1f0b842 383 fputc('(', stdout);
shimniok 21:5d7ac1f0b842 384 fputs(_cwd, stdout);
shimniok 21:5d7ac1f0b842 385 fputs(")# ", stdout);
shimniok 21:5d7ac1f0b842 386
shimniok 21:5d7ac1f0b842 387 return;
shimniok 21:5d7ac1f0b842 388 }
shimniok 21:5d7ac1f0b842 389
shimniok 21:5d7ac1f0b842 390
shimniok 21:5d7ac1f0b842 391 void SimpleShell::readCommand()
shimniok 21:5d7ac1f0b842 392 {
shimniok 21:5d7ac1f0b842 393 int i=0;
shimniok 21:5d7ac1f0b842 394 char c;
shimniok 21:5d7ac1f0b842 395 bool done = false;
shimniok 21:5d7ac1f0b842 396 static char cmd[MAXBUF];
shimniok 21:5d7ac1f0b842 397
shimniok 21:5d7ac1f0b842 398 memset(cmd, 0, MAXBUF);
shimniok 21:5d7ac1f0b842 399 do {
shimniok 21:5d7ac1f0b842 400 cmd[i] = 0;
shimniok 21:5d7ac1f0b842 401 c = fgetc(stdin);
shimniok 21:5d7ac1f0b842 402 if (c == '\r') { // if return is hit, we're done, don't add \r to cmd
shimniok 21:5d7ac1f0b842 403 done = true;
shimniok 21:5d7ac1f0b842 404 } else if (c == ESC) { // keyboard escape codes (arrow keys, etc)
shimniok 21:5d7ac1f0b842 405 int c2 = getchar();
shimniok 21:5d7ac1f0b842 406 int c3 = getchar();
shimniok 21:5d7ac1f0b842 407
shimniok 21:5d7ac1f0b842 408 if (c2 == 0x4f && c3 == 0x46) {
shimniok 21:5d7ac1f0b842 409 printf("<END>");
shimniok 21:5d7ac1f0b842 410 } else if (c2 == 0x5b) {
shimniok 21:5d7ac1f0b842 411
shimniok 21:5d7ac1f0b842 412 if (c3 == UP) {
shimniok 21:5d7ac1f0b842 413 printf("<UP>");
shimniok 21:5d7ac1f0b842 414 } else if (c3 == DOWN) {
shimniok 21:5d7ac1f0b842 415 printf("<DOWN>");
shimniok 21:5d7ac1f0b842 416 } else if (c3 == RIGHT) {
shimniok 21:5d7ac1f0b842 417 printf("<RIGHT>");
shimniok 21:5d7ac1f0b842 418 } else if (c3 == LEFT) {
shimniok 21:5d7ac1f0b842 419 printf("<LEFT>");
shimniok 21:5d7ac1f0b842 420 } else if (c3 == HOME || c3 == INS || c3 == DEL ||
shimniok 21:5d7ac1f0b842 421 c3 == PGUP || c3 == PGDN) {
shimniok 21:5d7ac1f0b842 422 char c4 = getchar();
shimniok 21:5d7ac1f0b842 423 if (c4 == TAIL) {
shimniok 21:5d7ac1f0b842 424 if (c4 == HOME) {
shimniok 21:5d7ac1f0b842 425 printf("<HOME>");
shimniok 21:5d7ac1f0b842 426 } else if (c4 == INS) {
shimniok 21:5d7ac1f0b842 427 printf("<INS>");
shimniok 21:5d7ac1f0b842 428 } else if (c4 == DEL) {
shimniok 21:5d7ac1f0b842 429 printf("<DEL>");
shimniok 21:5d7ac1f0b842 430 } else if (c4 == PGUP) {
shimniok 21:5d7ac1f0b842 431 printf("<PGUP>");
shimniok 21:5d7ac1f0b842 432 } else if (c4 == PGDN) {
shimniok 21:5d7ac1f0b842 433 printf("<PGDN>");
shimniok 21:5d7ac1f0b842 434 }//if
shimniok 21:5d7ac1f0b842 435 }//if
shimniok 21:5d7ac1f0b842 436 }//if
shimniok 21:5d7ac1f0b842 437 }//if
shimniok 21:5d7ac1f0b842 438 //printf("\n");
shimniok 21:5d7ac1f0b842 439 } else if (i < MAXBUF-1) {
shimniok 21:5d7ac1f0b842 440 if (c == 0x7f || c == '\b') { // backspace or delete
shimniok 21:5d7ac1f0b842 441 if (i > 0) { // if we're at the beginning, do nothing
shimniok 21:5d7ac1f0b842 442 i--;
shimniok 21:5d7ac1f0b842 443 fputs("\b \b", stdout);
shimniok 21:5d7ac1f0b842 444 }
shimniok 21:5d7ac1f0b842 445 } else {
shimniok 21:5d7ac1f0b842 446 if (isprint(c))
shimniok 21:5d7ac1f0b842 447 fputc(c, stdout);
shimniok 21:5d7ac1f0b842 448 cmd[i++] = c;
shimniok 21:5d7ac1f0b842 449 }
shimniok 21:5d7ac1f0b842 450 }
shimniok 21:5d7ac1f0b842 451
shimniok 21:5d7ac1f0b842 452 } while (!done);
shimniok 21:5d7ac1f0b842 453 fputc('\n', stdout);
shimniok 21:5d7ac1f0b842 454
shimniok 21:5d7ac1f0b842 455 // remove leading/trailing whitespace
shimniok 21:5d7ac1f0b842 456 char *s = cmd;
shimniok 21:5d7ac1f0b842 457 while (isspace(*s)) {
shimniok 21:5d7ac1f0b842 458 s++;
shimniok 21:5d7ac1f0b842 459 }
shimniok 21:5d7ac1f0b842 460 for (int j=i; j >= 0 && isspace(cmd[j]); j--) {
shimniok 21:5d7ac1f0b842 461 cmd[j] = '\0';
shimniok 21:5d7ac1f0b842 462 }
shimniok 21:5d7ac1f0b842 463
shimniok 21:5d7ac1f0b842 464 // split into command and arguments
shimniok 21:5d7ac1f0b842 465 argc = 0;
shimniok 21:5d7ac1f0b842 466 char *t;
shimniok 21:5d7ac1f0b842 467
shimniok 21:5d7ac1f0b842 468 for (int i=0; i < MAXARGS; i++) {
shimniok 21:5d7ac1f0b842 469 argv[i] = NULL;
shimniok 21:5d7ac1f0b842 470 }
shimniok 21:5d7ac1f0b842 471 t = strtok(s, " ");
shimniok 30:35522ea06236 472 while (t && argc < MAXARGS) {
shimniok 21:5d7ac1f0b842 473 argv[argc++] = t;
shimniok 21:5d7ac1f0b842 474 t = strtok(NULL, " ");
shimniok 21:5d7ac1f0b842 475 }
shimniok 21:5d7ac1f0b842 476
shimniok 21:5d7ac1f0b842 477 return;
shimniok 21:5d7ac1f0b842 478 }