source: mainline/kernel/generic/src/console/kconsole.c@ 82bb9c1

lfn serial ticket/834-toolchain-update topic/msim-upgrade topic/simplify-dev-export
Last change on this file since 82bb9c1 was 06b785f, checked in by Jiri Svoboda <jirik.svoboda@…>, 17 years ago

strlen() → str_size().
  • Property mode set to 100644
File size: 15.8 KB
Line 
1/*
2 * Copyright (c) 2005 Jakub Jermar
3 * All rights reserved.
4 *
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
7 * are met:
8 *
9 * - Redistributions of source code must retain the above copyright
10 * notice, this list of conditions and the following disclaimer.
11 * - Redistributions in binary form must reproduce the above copyright
12 * notice, this list of conditions and the following disclaimer in the
13 * documentation and/or other materials provided with the distribution.
14 * - The name of the author may not be used to endorse or promote products
15 * derived from this software without specific prior written permission.
16 *
17 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
18 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
19 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
20 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
21 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
22 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
23 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
24 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
25 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
26 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
27 */
28
29/** @addtogroup genericconsole
30 * @{
31 */
32
33/**
34 * @file kconsole.c
35 * @brief Kernel console.
36 *
37 * This file contains kernel thread managing the kernel console.
38 */
39
40#include <console/kconsole.h>
41#include <console/console.h>
42#include <console/chardev.h>
43#include <console/cmd.h>
44#include <print.h>
45#include <panic.h>
46#include <arch/types.h>
47#include <adt/list.h>
48#include <arch.h>
49#include <macros.h>
50#include <debug.h>
51#include <func.h>
52#include <string.h>
53#include <macros.h>
54#include <sysinfo/sysinfo.h>
55#include <ddi/device.h>
56#include <symtab.h>
57#include <errno.h>
58#include <putchar.h>
59
60/** Simple kernel console.
61 *
62 * The console is realized by kernel thread kconsole.
63 * It doesn't understand any useful command on its own,
64 * but makes it possible for other kernel subsystems to
65 * register their own commands.
66 */
67
68/** Locking.
69 *
70 * There is a list of cmd_info_t structures. This list
71 * is protected by cmd_lock spinlock. Note that specially
72 * the link elements of cmd_info_t are protected by
73 * this lock.
74 *
75 * Each cmd_info_t also has its own lock, which protects
76 * all elements thereof except the link element.
77 *
78 * cmd_lock must be acquired before any cmd_info lock.
79 * When locking two cmd info structures, structure with
80 * lower address must be locked first.
81 */
82
83SPINLOCK_INITIALIZE(cmd_lock); /**< Lock protecting command list. */
84LIST_INITIALIZE(cmd_head); /**< Command list. */
85
86static cmd_info_t *parse_cmdline(char *cmdline, size_t len);
87static bool parse_argument(char *cmdline, size_t len, index_t *start,
88 index_t *end);
89static char history[KCONSOLE_HISTORY][MAX_CMDLINE] = {};
90
91/** Initialize kconsole data structures
92 *
93 * This is the most basic initialization, almost no
94 * other kernel subsystem is ready yet.
95 *
96 */
97void kconsole_init(void)
98{
99 unsigned int i;
100
101 cmd_init();
102 for (i = 0; i < KCONSOLE_HISTORY; i++)
103 history[i][0] = '\0';
104}
105
106/** Register kconsole command.
107 *
108 * @param cmd Structure describing the command.
109 *
110 * @return 0 on failure, 1 on success.
111 */
112int cmd_register(cmd_info_t *cmd)
113{
114 link_t *cur;
115
116 spinlock_lock(&cmd_lock);
117
118 /*
119 * Make sure the command is not already listed.
120 */
121 for (cur = cmd_head.next; cur != &cmd_head; cur = cur->next) {
122 cmd_info_t *hlp;
123
124 hlp = list_get_instance(cur, cmd_info_t, link);
125
126 if (hlp == cmd) {
127 /* The command is already there. */
128 spinlock_unlock(&cmd_lock);
129 return 0;
130 }
131
132 /* Avoid deadlock. */
133 if (hlp < cmd) {
134 spinlock_lock(&hlp->lock);
135 spinlock_lock(&cmd->lock);
136 } else {
137 spinlock_lock(&cmd->lock);
138 spinlock_lock(&hlp->lock);
139 }
140 if ((strncmp(hlp->name, cmd->name, max(str_size(cmd->name),
141 str_size(hlp->name))) == 0)) {
142 /* The command is already there. */
143 spinlock_unlock(&hlp->lock);
144 spinlock_unlock(&cmd->lock);
145 spinlock_unlock(&cmd_lock);
146 return 0;
147 }
148
149 spinlock_unlock(&hlp->lock);
150 spinlock_unlock(&cmd->lock);
151 }
152
153 /*
154 * Now the command can be added.
155 */
156 list_append(&cmd->link, &cmd_head);
157
158 spinlock_unlock(&cmd_lock);
159 return 1;
160}
161
162/** Print count times a character */
163static void rdln_print_c(wchar_t ch, count_t count)
164{
165 count_t i;
166 for (i = 0; i < count; i++)
167 putchar(ch);
168}
169
170/** Insert character to string */
171static void insert_char(char *str, char ch, int pos)
172{
173 int i;
174
175 for (i = str_size(str); i > pos; i--)
176 str[i] = str[i - 1];
177 str[pos] = ch;
178}
179
180/** Try to find a command beginning with prefix */
181static const char *cmdtab_search_one(const char *name,link_t **startpos)
182{
183 size_t namelen = str_size(name);
184 const char *curname;
185
186 spinlock_lock(&cmd_lock);
187
188 if (!*startpos)
189 *startpos = cmd_head.next;
190
191 for (; *startpos != &cmd_head; *startpos = (*startpos)->next) {
192 cmd_info_t *hlp;
193 hlp = list_get_instance(*startpos, cmd_info_t, link);
194
195 curname = hlp->name;
196 if (str_size(curname) < namelen)
197 continue;
198 if (strncmp(curname, name, namelen) == 0) {
199 spinlock_unlock(&cmd_lock);
200 return curname+namelen;
201 }
202 }
203 spinlock_unlock(&cmd_lock);
204 return NULL;
205}
206
207
208/** Command completion of the commands
209 *
210 * @param name - string to match, changed to hint on exit
211 * @return number of found matches
212 */
213static int cmdtab_compl(char *name)
214{
215 static char output[/*MAX_SYMBOL_NAME*/128 + 1];
216 link_t *startpos = NULL;
217 const char *foundtxt;
218 int found = 0;
219 int i;
220
221 output[0] = '\0';
222 while ((foundtxt = cmdtab_search_one(name, &startpos))) {
223 startpos = startpos->next;
224 if (!found)
225 strncpy(output, foundtxt, str_size(foundtxt) + 1);
226 else {
227 for (i = 0; output[i] && foundtxt[i] &&
228 output[i] == foundtxt[i]; i++)
229 ;
230 output[i] = '\0';
231 }
232 found++;
233 }
234 if (!found)
235 return 0;
236
237 if (found > 1 && !str_size(output)) {
238 printf("\n");
239 startpos = NULL;
240 while ((foundtxt = cmdtab_search_one(name, &startpos))) {
241 cmd_info_t *hlp;
242 hlp = list_get_instance(startpos, cmd_info_t, link);
243 printf("%s - %s\n", hlp->name, hlp->description);
244 startpos = startpos->next;
245 }
246 }
247 strncpy(name, output, 128/*MAX_SYMBOL_NAME*/);
248 return found;
249}
250
251static char *clever_readline(const char *prompt, indev_t *input)
252{
253 static int histposition = 0;
254
255 static char tmp[MAX_CMDLINE + 1];
256 int curlen = 0, position = 0;
257 char *current = history[histposition];
258 int i;
259 char mod; /* Command Modifier */
260 char c;
261
262 printf("%s> ", prompt);
263 while (1) {
264 c = _getc(input);
265 if (c == '\n') {
266 putchar(c);
267 break;
268 }
269 if (c == '\b') { /* Backspace */
270 if (position == 0)
271 continue;
272 for (i = position; i < curlen; i++)
273 current[i - 1] = current[i];
274 curlen--;
275 position--;
276 putchar('\b');
277 for (i = position; i < curlen; i++)
278 putchar(current[i]);
279 putchar(' ');
280 rdln_print_c('\b', curlen - position + 1);
281 continue;
282 }
283 if (c == '\t') { /* Tabulator */
284 int found;
285
286 /* Move to the end of the word */
287 for (; position < curlen && current[position] != ' ';
288 position++)
289 putchar(current[position]);
290 /* Copy to tmp last word */
291 for (i = position - 1; i >= 0 && current[i] != ' '; i--)
292 ;
293 /* If word begins with * or &, skip it */
294 if (tmp[0] == '*' || tmp[0] == '&')
295 for (i = 1; tmp[i]; i++)
296 tmp[i - 1] = tmp[i];
297 i++; /* I is at the start of the word */
298 strncpy(tmp, current + i, position - i + 1);
299
300 if (i == 0) { /* Command completion */
301 found = cmdtab_compl(tmp);
302 } else { /* Symtab completion */
303 found = symtab_compl(tmp);
304 }
305
306 if (found == 0)
307 continue;
308 for (i = 0; tmp[i] && curlen < MAX_CMDLINE;
309 i++, curlen++)
310 insert_char(current, tmp[i], i + position);
311
312 if (str_size(tmp) || found == 1) { /* If we have a hint */
313 for (i = position; i < curlen; i++)
314 putchar(current[i]);
315 position += str_size(tmp);
316 /* Add space to end */
317 if (found == 1 && position == curlen &&
318 curlen < MAX_CMDLINE) {
319 current[position] = ' ';
320 curlen++;
321 position++;
322 putchar(' ');
323 }
324 } else { /* No hint, table was printed */
325 printf("%s> ", prompt);
326 for (i = 0; i < curlen; i++)
327 putchar(current[i]);
328 position += str_size(tmp);
329 }
330 rdln_print_c('\b', curlen - position);
331 continue;
332 }
333 if (c == 0x1b) { /* Special command */
334 mod = _getc(input);
335 c = _getc(input);
336
337 if (mod != 0x5b && mod != 0x4f)
338 continue;
339
340 if (c == 0x33 && _getc(input) == 0x7e) {
341 /* Delete */
342 if (position == curlen)
343 continue;
344 for (i = position + 1; i < curlen; i++) {
345 putchar(current[i]);
346 current[i - 1] = current[i];
347 }
348 putchar(' ');
349 rdln_print_c('\b', curlen - position);
350 curlen--;
351 } else if (c == 0x48) { /* Home */
352 rdln_print_c('\b', position);
353 position = 0;
354 } else if (c == 0x46) { /* End */
355 for (i = position; i < curlen; i++)
356 putchar(current[i]);
357 position = curlen;
358 } else if (c == 0x44) { /* Left */
359 if (position > 0) {
360 putchar('\b');
361 position--;
362 }
363 continue;
364 } else if (c == 0x43) { /* Right */
365 if (position < curlen) {
366 putchar(current[position]);
367 position++;
368 }
369 continue;
370 } else if (c == 0x41 || c == 0x42) {
371 /* Up, down */
372 rdln_print_c('\b', position);
373 rdln_print_c(' ', curlen);
374 rdln_print_c('\b', curlen);
375 if (c == 0x41) /* Up */
376 histposition--;
377 else
378 histposition++;
379 if (histposition < 0) {
380 histposition = KCONSOLE_HISTORY - 1;
381 } else {
382 histposition =
383 histposition % KCONSOLE_HISTORY;
384 }
385 current = history[histposition];
386 printf("%s", current);
387 curlen = str_size(current);
388 position = curlen;
389 continue;
390 }
391 continue;
392 }
393 if (curlen >= MAX_CMDLINE)
394 continue;
395
396 insert_char(current, c, position);
397
398 curlen++;
399 for (i = position; i < curlen; i++)
400 putchar(current[i]);
401 position++;
402 rdln_print_c('\b',curlen - position);
403 }
404 if (curlen) {
405 histposition++;
406 histposition = histposition % KCONSOLE_HISTORY;
407 }
408 current[curlen] = '\0';
409 return current;
410}
411
412bool kconsole_check_poll(void)
413{
414 return check_poll(stdin);
415}
416
417/** Kernel console prompt.
418 *
419 * @param prompt Kernel console prompt (e.g kconsole/panic).
420 * @param msg Message to display in the beginning.
421 * @param kcon Wait for keypress to show the prompt
422 * and never exit.
423 *
424 */
425void kconsole(char *prompt, char *msg, bool kcon)
426{
427 cmd_info_t *cmd_info;
428 count_t len;
429 char *cmdline;
430
431 if (!stdin) {
432 LOG("No stdin for kernel console");
433 return;
434 }
435
436 if (msg)
437 printf("%s", msg);
438
439 if (kcon)
440 _getc(stdin);
441 else
442 printf("Type \"exit\" to leave the console.\n");
443
444 while (true) {
445 cmdline = clever_readline((char *) prompt, stdin);
446 len = str_size(cmdline);
447 if (!len)
448 continue;
449
450 if ((!kcon) && (len == 4) && (strncmp(cmdline, "exit", 4) == 0))
451 break;
452
453 cmd_info = parse_cmdline(cmdline, len);
454 if (!cmd_info)
455 continue;
456
457 (void) cmd_info->func(cmd_info->argv);
458 }
459}
460
461/** Kernel console managing thread.
462 *
463 */
464void kconsole_thread(void *data)
465{
466 kconsole("kconsole", "Kernel console ready (press any key to activate)\n", true);
467}
468
469static int parse_int_arg(char *text, size_t len, unative_t *result)
470{
471 uintptr_t symaddr;
472 bool isaddr = false;
473 bool isptr = false;
474 int rc;
475
476 static char symname[MAX_SYMBOL_NAME];
477
478 /* If we get a name, try to find it in symbol table */
479 if (text[0] == '&') {
480 isaddr = true;
481 text++;
482 len--;
483 } else if (text[0] == '*') {
484 isptr = true;
485 text++;
486 len--;
487 }
488 if (text[0] < '0' || text[0] > '9') {
489 strncpy(symname, text, min(len + 1, MAX_SYMBOL_NAME));
490 rc = symtab_addr_lookup(symname, &symaddr);
491 switch (rc) {
492 case ENOENT:
493 printf("Symbol %s not found.\n", symname);
494 return -1;
495 case EOVERFLOW:
496 printf("Duplicate symbol %s.\n", symname);
497 symtab_print_search(symname);
498 return -1;
499 default:
500 printf("No symbol information available.\n");
501 return -1;
502 }
503
504 if (isaddr)
505 *result = (unative_t)symaddr;
506 else if (isptr)
507 *result = **((unative_t **)symaddr);
508 else
509 *result = *((unative_t *)symaddr);
510 } else { /* It's a number - convert it */
511 *result = atoi(text);
512 if (isptr)
513 *result = *((unative_t *)*result);
514 }
515
516 return 0;
517}
518
519/** Parse command line.
520 *
521 * @param cmdline Command line as read from input device.
522 * @param len Command line length.
523 *
524 * @return Structure describing the command.
525 */
526cmd_info_t *parse_cmdline(char *cmdline, size_t len)
527{
528 index_t start = 0, end = 0;
529 cmd_info_t *cmd = NULL;
530 link_t *cur;
531 count_t i;
532 int error = 0;
533
534 if (!parse_argument(cmdline, len, &start, &end)) {
535 /* Command line did not contain alphanumeric word. */
536 return NULL;
537 }
538
539 spinlock_lock(&cmd_lock);
540
541 for (cur = cmd_head.next; cur != &cmd_head; cur = cur->next) {
542 cmd_info_t *hlp;
543
544 hlp = list_get_instance(cur, cmd_info_t, link);
545 spinlock_lock(&hlp->lock);
546
547 if (strncmp(hlp->name, &cmdline[start], max(str_size(hlp->name),
548 end - start + 1)) == 0) {
549 cmd = hlp;
550 break;
551 }
552
553 spinlock_unlock(&hlp->lock);
554 }
555
556 spinlock_unlock(&cmd_lock);
557
558 if (!cmd) {
559 /* Unknown command. */
560 printf("Unknown command.\n");
561 return NULL;
562 }
563
564 /* cmd == hlp is locked */
565
566 /*
567 * The command line must be further analyzed and
568 * the parameters therefrom must be matched and
569 * converted to those specified in the cmd info
570 * structure.
571 */
572
573 for (i = 0; i < cmd->argc; i++) {
574 char *buf;
575 start = end + 1;
576 if (!parse_argument(cmdline, len, &start, &end)) {
577 printf("Too few arguments.\n");
578 spinlock_unlock(&cmd->lock);
579 return NULL;
580 }
581
582 error = 0;
583 switch (cmd->argv[i].type) {
584 case ARG_TYPE_STRING:
585 buf = (char *) cmd->argv[i].buffer;
586 strncpy(buf, (const char *) &cmdline[start],
587 min((end - start) + 2, cmd->argv[i].len));
588 buf[min((end - start) + 1, cmd->argv[i].len - 1)] =
589 '\0';
590 break;
591 case ARG_TYPE_INT:
592 if (parse_int_arg(cmdline + start, end - start + 1,
593 &cmd->argv[i].intval))
594 error = 1;
595 break;
596 case ARG_TYPE_VAR:
597 if (start != end && cmdline[start] == '"' &&
598 cmdline[end] == '"') {
599 buf = (char *) cmd->argv[i].buffer;
600 strncpy(buf, (const char *) &cmdline[start + 1],
601 min((end-start), cmd->argv[i].len));
602 buf[min((end - start), cmd->argv[i].len - 1)] =
603 '\0';
604 cmd->argv[i].intval = (unative_t) buf;
605 cmd->argv[i].vartype = ARG_TYPE_STRING;
606 } else if (!parse_int_arg(cmdline + start,
607 end - start + 1, &cmd->argv[i].intval)) {
608 cmd->argv[i].vartype = ARG_TYPE_INT;
609 } else {
610 printf("Unrecognized variable argument.\n");
611 error = 1;
612 }
613 break;
614 case ARG_TYPE_INVALID:
615 default:
616 printf("invalid argument type\n");
617 error = 1;
618 break;
619 }
620 }
621
622 if (error) {
623 spinlock_unlock(&cmd->lock);
624 return NULL;
625 }
626
627 start = end + 1;
628 if (parse_argument(cmdline, len, &start, &end)) {
629 printf("Too many arguments.\n");
630 spinlock_unlock(&cmd->lock);
631 return NULL;
632 }
633
634 spinlock_unlock(&cmd->lock);
635 return cmd;
636}
637
638/** Parse argument.
639 *
640 * Find start and end positions of command line argument.
641 *
642 * @param cmdline Command line as read from the input device.
643 * @param len Number of characters in cmdline.
644 * @param start On entry, 'start' contains pointer to the index
645 * of first unprocessed character of cmdline.
646 * On successful exit, it marks beginning of the next argument.
647 * @param end Undefined on entry. On exit, 'end' points to the last character
648 * of the next argument.
649 *
650 * @return false on failure, true on success.
651 */
652bool parse_argument(char *cmdline, size_t len, index_t *start, index_t *end)
653{
654 index_t i;
655 bool found_start = false;
656
657 ASSERT(start != NULL);
658 ASSERT(end != NULL);
659
660 for (i = *start; i < len; i++) {
661 if (!found_start) {
662 if (isspace(cmdline[i]))
663 (*start)++;
664 else
665 found_start = true;
666 } else {
667 if (isspace(cmdline[i]))
668 break;
669 }
670 }
671 *end = i - 1;
672
673 return found_start;
674}
675
676/** @}
677 */
Note: See TracBrowser for help on using the repository browser.