printf.c 1.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103
  1. #include "types.h"
  2. #include "stat.h"
  3. #include "user.h"
  4. static void
  5. putc(int fd, char c)
  6. {
  7. write(fd, &c, 1);
  8. }
  9. u32 div(u32 n, u32 d) // long division
  10. {
  11. u32 q=0, r=0;
  12. int i;
  13. for(i=31;i>=0;i--){
  14. r = r << 1;
  15. r = r | ((n >> i) & 1);
  16. if(r >= d) {
  17. r = r - d;
  18. q = q | (1 << i);
  19. }
  20. }
  21. return q;
  22. }
  23. static void
  24. printint(int fd, int xx, int base, int sgn)
  25. {
  26. static char digits[] = "0123456789ABCDEF";
  27. char buf[16];
  28. int i, neg;
  29. uint x, y, b;
  30. neg = 0;
  31. if(sgn && xx < 0){
  32. neg = 1;
  33. x = -xx;
  34. } else {
  35. x = xx;
  36. }
  37. b = base;
  38. i = 0;
  39. do{
  40. y = div(x, b);
  41. buf[i++] = digits[x - y * b];
  42. }while((x = y) != 0);
  43. if(neg)
  44. buf[i++] = '-';
  45. while(--i >= 0)
  46. putc(fd, buf[i]);
  47. }
  48. // Print to the given fd. Only understands %d, %x, %p, %s.
  49. void
  50. printf(int fd, char *fmt, ...)
  51. {
  52. char *s;
  53. int c, i, state;
  54. uint *ap;
  55. state = 0;
  56. ap = (uint*)(void*)&fmt + 1;
  57. for(i = 0; fmt[i]; i++){
  58. c = fmt[i] & 0xff;
  59. if(state == 0){
  60. if(c == '%'){
  61. state = '%';
  62. } else {
  63. putc(fd, c);
  64. }
  65. } else if(state == '%'){
  66. if(c == 'd'){
  67. printint(fd, *ap, 10, 1);
  68. ap++;
  69. } else if(c == 'x' || c == 'p'){
  70. printint(fd, *ap, 16, 0);
  71. ap++;
  72. } else if(c == 's'){
  73. s = (char*)*ap;
  74. ap++;
  75. if(s == 0)
  76. s = "(null)";
  77. while(*s != 0){
  78. putc(fd, *s);
  79. s++;
  80. }
  81. } else if(c == 'c'){
  82. putc(fd, *ap);
  83. ap++;
  84. } else if(c == '%'){
  85. putc(fd, c);
  86. } else {
  87. // Unknown % sequence. Print it to draw attention.
  88. putc(fd, '%');
  89. putc(fd, c);
  90. }
  91. state = 0;
  92. }
  93. }
  94. }