i386/ucla/src/lab5/lib/printf.c

   1 // Implementation of cprintf console output for user environments,
   2 // based on printfmt() and the sys_cputs() system call.
   3 //
   4 // cprintf is a debugging statement, not a generic output statement.
   5 // It is very important that it always go to the console, especially when
   6 // debugging file descriptor code!
   7
   8 #include <inc/types.h>
   9 #include <inc/stdio.h>
  10 #include <inc/stdarg.h>
  11 #include <inc/lib.h>
  12
  13
  14 // Collect up to 256 characters into a buffer
  15 // and perform ONE system call to print all of them,
  16 // in order to make the lines output to the console atomic
  17 // and prevent interrupts from causing context switches
  18 // in the middle of a console output line and such.
  19 struct printbuf {
  20         int idx;        // current buffer index
  21         int cnt;        // total bytes printed so far
  22         char buf[256];
  23 };
  24
  25
  26 static void
  27 putch(int ch, void *ptr)
  28 {
  29         struct printbuf *b = (struct printbuf *) ptr;
  30         b->buf[b->idx++] = ch;
  31         if (b->idx == 256-1) {
  32                 sys_cputs(b->buf, b->idx);
  33                 b->idx = 0;
  34         }
  35         b->cnt++;
  36 }
  37
  38 int
  39 vcprintf(const char *fmt, va_list ap)
  40 {
  41         struct printbuf b;
  42
  43         b.idx = 0;
  44         b.cnt = 0;
  45         vprintfmt(putch, &b, fmt, ap);
  46         sys_cputs(b.buf, b.idx);
  47
  48         return b.cnt;
  49 }
  50
  51 int
  52 cprintf(const char *fmt, ...)
  53 {
  54         va_list ap;
  55         int cnt;
  56
  57         va_start(ap, fmt);
  58         cnt = vcprintf(fmt, ap);
  59         va_end(ap);
  60
  61         return cnt;
  62 }
  63