一、文章内容介绍
在上篇文章里已经重写了部分string.h 字符串处理的相关函数,这篇文章继续介绍string.h里常用的库函数;虽然也有现成的可以调用,这里重写就相当于锻炼一样,别人能实现的自己也去照着功能写一篇,在没有标准库调用的时候就可以直接拿来使用,如果库函数有BUG或者功能上不满足要求的时候,自己就可以改造一下。重写这些函数的过程中可以充分锻炼自己的编程能力,特别是函数、指针的运用。
上篇文章重写的函数有下面几个:
strcmp 字符串比较 strlen 计算字符串长度 strstr 字符串查找 memcmp 内存比较 strcpy 字符串拷贝 memcpy 内存拷贝
本篇文章介绍下面几个函数:
strdup 字符串拷贝 strchr 字符查找函数 (从头开始) strrchr 字符查找函数(从尾开始) strcat strncat 字符串拼接函数 memset 内存初始化函数 (可用于赋初值)
二、重写函数源码
strdup是字符串拷贝函数,一般要配合free使用,因为内部会申请空间存放新的字符串。也就是返回的地址用完之后需要free释放掉。
char * strdup(const char *s) { char *new; if ((s == NULL) || ((new = malloc (strlen(s) + 1)) == NULL) ) { return NULL; } strcpy (new, s); return new; }
2.2 strchr 字符查找函数(从头开始)
在源字符串中查找第一个出现指定字符的地方,返回这个字符在字符串里的地址。
char * strchr(const char * s, int c) { for(; *s != (char) c; ++s) if (*s == '\0') return NULL; return (char *) s; }
2.3 strrchr 字符查找函数(从尾开始)
在源字符串中查找最后一次出现指定字符的地方,返回这个字符在字符串里的地址。
char * strrchr(const char * s, int c) { const char *p = s + strlen(s); do { if (*p == (char)c) return (char *)p; } while (--p >= s); return NULL; }
2.4 strcat 字符串拼接函数
strcat 用于字符串拼接,有两种原型。带n的可以指定拷贝的长度。
char * strcat(char * dest, const char * src) { char *tmp = dest; while (*dest) dest++; while ((*dest++ = *src++) != '\0') ; return tmp; } char * strncat(char *dest, const char *src, size_t count) { char *tmp = dest; if (count) { while (*dest) dest++; while ((*dest++ = *src++)) { if (--count == 0) { *dest = '\0'; break; } } } return tmp; }
2.5 memset 内存初始化函数 (可用于赋初值)
memset 可以给指定的内存按顺序赋值,一般用于清空内存空间或者初始化、赋初值等。
void * memset(void * s,int c,size_t count) { unsigned long *sl = (unsigned long *) s; unsigned long cl = 0; char *s8; int i; /* do it one word at a time (32 bits or 64 bits) while possible */ if ( ((ulong)s & (sizeof(*sl) - 1)) == 0) { for (i = 0; i < sizeof(*sl); i++) { cl <<= 8; cl |= c & 0xff; } while (count >= sizeof(*sl)) { *sl++ = cl; count -= sizeof(*sl); } } /* fill 8 bits at a time */ s8 = (char *)sl; while (count--) *s8++ = c; return s; }