走进C标准库(6)——"string.h"中函数的实现memchr

深入解析memchr函数及其实现优化

转载于 2014-04-30 12:45:00 发布 · 73 阅读

0 ·

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/shijiezhenmei/p/3701090.html

我写的memchr：

1 void *memchr(const void *buf, char ch, unsigned count){
2      unsigned int cnt = 0;
3      while(*(buf++) != ch && cnt <= count){cnt++;}
4      if(cnt > count)
5         return NULL;
6      else
7         return buf;
8 }

红色部分报错。

该错误为为ANSIC中认定的错误，是因为它坚持：进行算法操作的指针必须是确定知道其指向数据类型大小的。

但是GNU则不这么认定，它指定void * 的算法操作与char * 一致。

在实际的程序设计中，为迎合ANSI标准，并提高程序的可移植性，我们可以对void指针先进行强制类型转换为特定类型的指针。

则上述代码修改为：

1 void *memchr(const void *buf, char ch, unsigned count){
2      unsigned int cnt = 0;
3      while(*((char *)buf++) != ch && cnt <= count){cnt++;}
4      if(cnt > count)
5         return NULL;
6      else
7         return buf;
8  }

使用测试程序测试：

1 int main(void)
2 {
3     char *s = "hello world";
4     char *ptr = memchr(s,'w',10);
5     printf("%x\n%x\n%c",s,ptr,*ptr);
6     return 0;
7 }

输出结果为：

403000
403007
o

有错，理论上*ptr应该为w

循环条件上多加了1

最终的程序：

1 void *memchr(const void *buf, char ch, unsigned count){
2     unsigned int cnt = 0;
3     while(*((char *)buf++) != ch && cnt <= count){cnt++;}
4     if(cnt > count)
5        return NULL;
6     else
7        return (char *)buf - 1;
8 }

microsoft visual C 中的memchr的实现为：

 1 void * __cdecl memchr (
 2         const void * buf,
 3         int chr,
 4         size_t cnt
 5         )
 6 {
 7         while ( cnt && (*(unsigned char *)buf != (unsigned char)chr) ) {
 8                 buf = (unsigned char *)buf + 1;
 9                 cnt--;
10         }
11 
12         return(cnt ? (void *)buf : NULL);
13 }