实现 libc 部分函数 strlen strcat 等

memchr_strlen_diy.c

@@ -1,11 +1,24 @@
 #include <stdio.h>
+
+/**
+ * 自行实现strlen函数
+ * @param s 字符串指针
+ * @return 字符串的长度
+ */
 size_t strlen_diy(const char *s) {
   const char *a = s;
-  for (; *s; s++)
+  while (*s++)
     ;
-  return s - a;
+  return s - a - 1;
 }
 
+/**
+ * 自行实现memchr函数
+ * @param src 源数据指针
+ * @param c 要查找的字符
+ * @param n 检查的字符数
+ * @return 找到字符的指针，或NULL
+ */
 void *memchr_diy(const void *src, int c, size_t n) {
   const unsigned char *s = src;
   c = (unsigned char)c;
@@ -16,22 +29,61 @@ void *memchr_diy(const void *src, int c, size_t n) {
   return n ? (void *)s : 0;
 }
 
+/**
+ * 自行实现strnlen函数
+ * @param s 字符串指针
+ * @param n 最大检查长度
+ * @return 字符串的实际长度，或n
+ */
 size_t strnlen_diy(const char *s, size_t n) {
   const char *p = memchr_diy(s, 0, n);
   return p ? p - s : n;
 }
 
+/**
+ * 自行实现strcpy函数
+ * @param d 目标字符串指针
+ * @param s 源字符串指针
+ * @return 目标字符串指针
+ */
+char *strcpy_diy(char *d, const char *s) {
+  char *p = d;
+  // for (; (*d = *s); s++, d++) ;
+  while ((*d++ = *s++))
+    ;
+  return p;
+}
+
+/**
+ * 自行实现strcat函数
+ * @param d 目标字符串指针
+ * @param s 源字符串指针
+ * @return 目标字符串指针
+ */
+char *strcat_diy(char *d, const char *s) {
+  strcpy_diy(d + strlen_diy(d), s);
+  return d;
+}
+
 int main() {
-  char str[] =
-      "Locate character in block of memory\n"
-      "Searches within the first num bytes of the block of memory pointed by "
-      "ptr for the first occurrence of value (interpreted as an unsigned "
-      "char), and returns a pointer to it.";
+  char str[] = "Hello World -> (strlen_diy memchr_diy strcpy_diy strcat_diy)";
+  // 使用memchr_diy查找字符'('的位置
   puts(memchr_diy(str, '(', strlen_diy(str)));
 
-  int len = strnlen_diy(str, 512);
-
+  // 使用strlen_diy计算字符串长度
+  int len = strlen_diy(str);
   printf("len: %d\n", len);
 
+  // 使用strnlen_diy计算字符串长度，限制最大检查长度为100
+  len = strnlen_diy(str, 100);
+  printf("len: %d\n", len); 
+
+  // 使用strcpy_diy复制字符串
+  char *s = strcpy_diy(str, "Hello C/C++ LIBC:");
+  printf("%s\n", s);
+
+  // 使用strcat_diy追加字符串
+  s = strcat_diy(str, "-> strcat_diy");
+  printf("%s\n", s);
   return 0;
-}
+}