实现C/C++中的字符串操作函数是很练内功的,别看着这些函数很简单,自己实现起来,还是有许多地方需要考虑的,所以没独立写过的朋友,可以自己敲敲看 .--By Crazybaby
strcpy:
char* cb_strcpy(char* dst, const char* src) { assert((dst!=NULL) && (src!=NULL)); char* retAddr = dst; /**< retAddr is in static , char retAddr[] will in Stack, So... */ while ((*(dst++) = *(src++))!='\0') { ; } return retAddr; }strncpy:
char* cb_strncpy(char* dst, const char* src, size_t size) { assert((dst!=NULL) && (src!=NULL)); char* retAddr = dst; /**< retAddr is in static , char retAddr[] will in Stack, So... */ int i = 0; while (((*(dst++) = *(src++))!='\0') && ((i++) < size)) { ; } *(retAddr+size)='\0'; /**< cut off String */ return retAddr; }这个strncpy实现版本 和 stdlib.h 下的 strncpy 还是有区别的, 比如
CB: strncpy相对于strcpy来说 ,安全性提高了一个等级 . 另外一个要非常注意的地方那个是 strcpy 和 strncpy 都会遇到 '\0' 结束.
另外:当请求的目标地址空间比源字符串空间要小的时候,strncpy 将不再用”\0”来结束字符串。这是一个巨大的隐患,所以还是不安全
memcpy Version 1:
char* cb_memcpyVer1(char* dst, const char* src, size_t size) { assert((dst!=NULL) && (src!=NULL)); char* retAddr = dst; while (size-- > 0) { *(dst++) = *(src++); } return retAddr; }CB: memcpy 和 strncpy 最大的区别是 memcpy不会遇到 '\0' 结束拷贝 .
memcpy Version 2 :
char* cb_memcpyVer2(char* dst, const char* src, size_t size) { assert((dst!=NULL) && (src!=NULL)); char* retAddr = dst; size_t i = 0; /* --- 解决数据区重叠问题 --- */ if ((retAddr>src) && (retAddr<(src+size))) { for (i=size-1; i>=0; i--) { *(dst++) = *(src++); } } else { for (i=0; i<size; i++) { *(dst++) = *(src++); } } *(retAddr+size)='\0'; return retAddr; }CB: memcpyVersion1不能防止overlapping区域的问题, Ver2很好的解决了.
memcpy还有种蛋疼的写法:
void* cb_memcpy(void* dest, const void* src, size_t count) { char* d = (char*)dest; const char* s = (const char*)src; /* --- 每次复制8bit --- */ int n = (count + 7) / 8; switch(count & 7) { case 0: do { *d++ = *s++; case 7: *d++ = *s++; case 6: *d++ = *s++; case 5: *d++ = *s++; case 4: *d++ = *s++; case 3: *d++ = *s++; case 2: *d++ = *s++; case 1: *d++ = *s++; } while (--n > 0); } return dest; }memset:
void* cb_memset(void* buffer, int b, size_t size) { assert(buffer!=NULL); char* retAddr = (char*)buffer; while (size-- > 0) { *(retAddr++) = (char)b; } return retAddr; }Memset使用时误区:
char a[10]; memset(a, 0, sizeof(char)*10); //这个操作完全没必要 因为下面这块内存马上要被使用了。 memcpy(a, "Hello", 5);
strlen:
int cb_strlen(const char* str) { assert(str!=NULL); int len = 0; while (*str!='\0') { /**< '\0', stop */ str++; len++ ; } return len; }Recursive strlen:
int cb_strlen(const char *str) { if ((str == NULL) || (*str == '\0')) { return 0; } else { return cb_strlen(str+1)+1; /**< Rescursive */ } }strcat:
char* cb_strcat(char* dst, char* src) { assert(src!=NULL); char* retAddr = dst; /* --- Find last position --- */ while (*dst++ != '\0') { ; } dst--; while (*dst++ = *src++) { ; } return retAddr; }strcmp:
int cb_strcmp(char* str, char* str2) { assert((str!=NULL) && (str2!=NULL)); char* pStr = str; char* pStr2 = str2; while (*pStr && *pStr2 && (*pStr==*pStr2)) { pStr++; pStr2++; } return (*pStr - *pStr2); /**< 相等则为0 , 前者大于后者大于0, 反之小于0 */ } strncmp:
int cb_strncmp(char* str, char* str2, size_t size) { assert((str!=NULL) && (str2!=NULL)); char* pStr = str; char* pStr2 = str2; while (size-- && *pStr && *pStr2 && (*pStr == *pStr2)) { pStr++; pStr2++; } return (*pStr - *pStr2); }
strchr:
char* cb_strchr(char* str, char c){ assert(str!=NULL); char* retAddr = str; while (*retAddr!=c) { retAddr++; } if (*retAddr == c) { return retAddr; } else { return NULL; } }这些字符串操作函数实现时要注意这些问题 :
1. 需要检查指针的有效性,一般通过直接和NULL进行比较来判断。
2. 函数需要能够进行链式操作,也就是说 char* a = strcpy(b, "Hello");
3. src的值需要加 const
4. strcnpy 和 memcpy 有 '\0' 判断的区别
5. 内存区重叠问题 :
比如这个程序:
int main(void) { char buffer[]="abcdefg"; memcpy(buffer, buffer+2 ,3); //buffer+2(从c开始 长度3个 cde) printf("%s", buffer); }
这个结果为: cdedefg .
再看下面这个程序:
int main(void) { char buffer[]="abcdefg"; memcpy(buffer+2, buffer ,3); //如果你幸运的话 会出现 abcab[d]fg 中括号是随机值 printf("%s", buffer); }这里的memcpy就需要用 memmove的实现方式来代替 。