C语言-重构string.c系统库函数(2)

一、文章内容介绍

在上篇文章里已经重写了部分string.h 字符串处理的相关函数，这篇文章继续介绍string.h里常用的库函数；虽然也有现成的可以调用，这里重写就相当于锻炼一样，别人能实现的自己也去照着功能写一篇，在没有标准库调用的时候就可以直接拿来使用，如果库函数有BUG或者功能上不满足要求的时候，自己就可以改造一下。重写这些函数的过程中可以充分锻炼自己的编程能力，特别是函数、指针的运用。

上篇文章重写的函数有下面几个：

strcmp  字符串比较
strlen  计算字符串长度
strstr  字符串查找
memcmp  内存比较
strcpy  字符串拷贝
memcpy  内存拷贝

本篇文章介绍下面几个函数:

strdup  字符串拷贝
strchr 字符查找函数 (从头开始)
strrchr 字符查找函数(从尾开始)
strcat strncat  字符串拼接函数
memset 内存初始化函数 (可用于赋初值)

二、重写函数源码

2.1 strdup 字符串拷贝函数

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;
}