str&mem系列函数的实现

常见的str及mem系列函数的实现

/*由src所指内存区域复制count个字节到dst所指内存区域,如果目标区域和源区域有重叠的话
memmove能够保证源串在被覆盖之前将重叠区域的字节拷贝到目标区域中，但复制后源内容会被更改。
但是当目标区域与源区域没有重叠则和memcpy函数功能相同*/
void* my_memmove(void* dst, const void* src, size_t count)
{
	if((NULL == dst) || (NULL == src))
	{
		printf("Invalid parameter!\n");
		return dst;
	}
	
	char* psrc = (char*)src;
	char* pdst = (char*)dst;
	
	/*从前往后*/
	if((pdst <= psrc) || (pdst >= psrc+count))
	{
		while(count--)
		{
			*pdst++ = *psrc++;
		}
	}
	else
	{
		pdst += count - 1;
		psrc += count - 1;
		while(count--)
		{
			*pdst-- = *psrc--;
		}
	}
	
	return dst;	
}


/*由src所指内存区域复制count个字节到dst所指内存区域中
说明：src和dst所指内存区域不能重叠*/
void* my_memcpy(void* dst, const void* src, size_t count)
{
	if((NULL == dst) || (NULL == src))
	{
		printf("Invalid parameter!\n");
		return dst;
	}
	
	char *psrc = (char*)src;
	char *pdst = (char*)dst;
	
	/*从前往后*/
	if((pdst <= psrc) || (pdst >= psrc+count))
	{
		while(count--)
		{
			*pdst++ = *psrc++;
		}
	}
	else /*从后往前*/
	{
		pdst += count-1;
		psrc += count-1;
		while(count--)
		{
				*pdst-- = *psrc--;
		}
	}
	return dst;
}

/*将buffer所指向内存的前n个字节的内容全部替换为val制定的ASCII值，并返回bufffer*/
void* my_memset(void* buffer, int val, size_t n)
{
	if(NULL == buffer)
	{
		printf("Invalid parametera!\n");
		return buffer;
	}
	
	char* tmp = (char*)buffer;
	while(n--)
	{
		*tmp++ = (char)val;
	}
	return buffer;
}


/*将从src地址开始且含有\0结束符的字符串复制到以dst开始的地址空间
说明：src和dst所指内存区域不可重叠，且dst必须有足够空间容纳src的字符串*/
char* my_strcpy(char* dst, const char* src)
{
	if((NULL == src) || (NULL == dst))
	{
		printf("Invalid parameter!\n");
		return dst;
	}
	
	char* tmp = dst;
	while((*tmp ++ = *src++) != '\0');
	
	return dst;
}

/*将src地址开始的前n个字节复制到dst所指空间
说明：当 src 的长度小于 n 时，dest 的剩余部分将用空字节填充。*/
char* my_strncpy(char* dst, const char* src, size_t n)
{
	if((NULL == src) || (NULL == dst))
	{
		printf("Invalid parameter!\n");
		return dst;
	}
	char*tmp = dst;
	
	while((n) && ((*tmp++ = *src++) != '\0'))
	{
		n--;
	}
	
	while(n--)
	{
		*tmp++ = '\0';
	}
	
	return dst;	
}

/*比较两个字符串的大小，
若str1 == str2，返回0，
若str1 > str2,返回正数
若str1 < str2，返回负数*/
int my_strcmp(const char* str1, const char* str2)
{
	if((NULL == str1) || (NULL == str2))
	{
		printf("Invalid parameter!\n");
		return 0;
	}
	
	while((*str1 != '\0') && (*str2 != '\0') && (*str1 == *str2))
	{
		str1++;
		str2++;
	}
	
	return *str1 - *str2;
}

/*把 src 所指向的字符串追加到 dst 所指向的字符串的结尾*/
char* my_strcat(char *dst, const char* src)
{
	if((NULL == dst) || (NULL == src))
	{
		printf("Invalid parameter!\n");
		return dst;
	}
	char* tmp = dst + strlen(dst);
	
	while((*tmp++ = *src++) != '\0');
	
	return dst;
}

/*把src所指字符串的前n个字符添加到dst所指字符串的结尾处，
并覆盖dest所指字符串结尾的'\0'，从而实现字符串的连接*/
char* my_strncat(char* dst, const char* src, size_t n)
{
	if((NULL == src) || (NULL == dst))
	{
		printf("Invalid parameter!\n");
		return dst;
	}
	
	char *tmp = dst + strlen(dst);
	while((n--) && (*tmp++ = *src++) != '\0');
	
	*tmp = '\0';
	
	return dst;
}


/*计算字符串 str 的长度，直到'\0'，但不包括'\0'*/
size_t my_strlen(const char* str)
{
	if(NULL == str)
	{
		printf("Invalid parameter!\n");
		return 0;
	}
	
	size_t len = 0;
	while(*str++ != '\0')
	{
		++len;
	}
	return len;
}