重写C语言string.c的部分函数

Posted 2021-12-08 DS小龙哥

一、文章内容介绍

在上篇文章里已经重写了部分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;
}