string.cpp 3.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194
  1. #include <string.h>
  2. #include <errno.h>
  3. #include <stdio.h>
  4. #include <signal.h>
  5. #include <assert.h>
  6. extern "C" {
  7. void bzero(void* dest, size_t n)
  8. {
  9. memset(dest, 0, n);
  10. }
  11. void bcopy(const void* src, void* dest, size_t n)
  12. {
  13. memmove(dest, src, n);
  14. }
  15. void* memset(void* dest, int c, size_t n)
  16. {
  17. uint8_t* bdest = (uint8_t*)dest;
  18. for (; n; --n)
  19. *(bdest++) = c;
  20. return dest;
  21. }
  22. size_t strspn(const char* s, const char* accept)
  23. {
  24. const char* p = s;
  25. cont:
  26. char ch = *p++;
  27. char ac;
  28. for (const char* ap = accept; (ac = *ap++) != '\0';) {
  29. if (ac == ch)
  30. goto cont;
  31. }
  32. return p - 1 - s;
  33. }
  34. size_t strcspn(const char* s, const char* reject)
  35. {
  36. for (auto* p = s;;) {
  37. char c = *p++;
  38. auto* rp = reject;
  39. char rc;
  40. do {
  41. if ((rc = *rp++) == c)
  42. return p - 1 - s;
  43. } while(rc);
  44. }
  45. }
  46. size_t strlen(const char* str)
  47. {
  48. size_t len = 0;
  49. while (*(str++))
  50. ++len;
  51. return len;
  52. }
  53. int strcmp(const char* s1, const char* s2)
  54. {
  55. while (*s1 == *s2++)
  56. if (*s1++ == 0)
  57. return 0;
  58. return *(const unsigned char*)s1 - *(const unsigned char*)--s2;
  59. }
  60. int strncmp(const char* s1, const char* s2, size_t n)
  61. {
  62. if (!n)
  63. return 0;
  64. do {
  65. if (*s1 != *s2++)
  66. return *(const unsigned char*)s1 - *(const unsigned char*)--s2;
  67. if (*s1++ == 0)
  68. break;
  69. } while (--n);
  70. return 0;
  71. }
  72. int memcmp(const void* v1, const void* v2, size_t n)
  73. {
  74. auto* s1 = (const uint8_t*)v1;
  75. auto* s2 = (const uint8_t*)v2;
  76. while (n-- > 0) {
  77. if (*s1++ != *s2++)
  78. return s1[-1] < s2[-1] ? -1 : 1;
  79. }
  80. return 0;
  81. }
  82. void memcpy(void* dest, const void* src, size_t n)
  83. {
  84. auto* bdest = (unsigned char*)dest;
  85. auto* bsrc = (const unsigned char*)src;
  86. for (; n; --n)
  87. *(bdest++) = *(bsrc++);
  88. }
  89. void memmove(void* dest, const void* src, size_t n)
  90. {
  91. if (dest < src)
  92. return memcpy(dest, src, n);
  93. // FIXME: Implement backwards copy.
  94. assert(false);
  95. }
  96. char* strcpy(char* dest, const char *src)
  97. {
  98. char* originalDest = dest;
  99. while ((*dest++ = *src++) != '\0');
  100. return originalDest;
  101. }
  102. char* strncpy(char* dest, const char* src, size_t n)
  103. {
  104. size_t i;
  105. for (i = 0; i < n && src[i] != '\0'; ++i)
  106. dest[i] = src[i];
  107. for ( ; i < n; ++i)
  108. dest[i] = '\0';
  109. return dest;
  110. }
  111. char* strchr(const char* str, int c)
  112. {
  113. char ch = c;
  114. for (;; ++str) {
  115. if (*str == ch)
  116. return const_cast<char*>(str);
  117. if (!*str)
  118. return nullptr;
  119. }
  120. }
  121. char* strrchr(const char* str, int ch)
  122. {
  123. char *last = nullptr;
  124. char c;
  125. for (; (c = *str); ++str) {
  126. if (c == ch)
  127. last = const_cast<char*>(str);
  128. }
  129. return last;
  130. }
  131. char* strcat(char *dest, const char *src)
  132. {
  133. size_t destLength = strlen(dest);
  134. size_t i;
  135. for (i = 0 ; src[i] != '\0' ; i++)
  136. dest[destLength + i] = src[i];
  137. dest[destLength + i] = '\0';
  138. return dest;
  139. }
  140. char* strncat(char *dest, const char *src, size_t n)
  141. {
  142. size_t destLength = strlen(dest);
  143. size_t i;
  144. for (i = 0 ; i < n && src[i] != '\0' ; i++)
  145. dest[destLength + i] = src[i];
  146. dest[destLength + i] = '\0';
  147. return dest;
  148. }
  149. const char* sys_errlist[] = {
  150. #undef __ERROR
  151. #define __ERROR(a, b) b,
  152. __ENUMERATE_ALL_ERRORS
  153. #undef __ERROR
  154. };
  155. int sys_nerr = __errno_count;
  156. char* strerror(int errnum)
  157. {
  158. if (errnum >= __errno_count) {
  159. printf("strerror() missing string for errnum=%d\n", errnum);
  160. return const_cast<char*>("Unknown error");
  161. }
  162. return const_cast<char*>(sys_errlist[errnum]);
  163. }
  164. char* strsignal(int signum)
  165. {
  166. if (signum >= __signal_count) {
  167. printf("strsignal() missing string for signum=%d\n", signum);
  168. return const_cast<char*>("Unknown signal");
  169. }
  170. return const_cast<char*>(sys_siglist[signum]);
  171. }
  172. }