这道笔试题竟然运行不出错

#读者提供的面试题

下面这张截图是一个读者在面试的时候遇到的题目，是哪个公司的我就不说出来了，我在微信朋友圈发了这个题目后，有几个好友给我留言说自己也写了这道题。

题目：下面这段代码有什么问题？

#后续

然后我就用这个图片发了朋友圈，很多人也回复了，天资跟我差不多的人呢，都看出来了就是 if 判断的代码需要往上移动下，要不然呢，malloc 后没有释放，就会出现内存泄漏。

这个是基本的，还有一些后续的回复，有几个微信好友回复还是比较不错，在这里分享给大家。

#我写的测试源码

#include "stdio.h"
#include "stdlib.h"
#include "string.h"int swap(void *a,void* b,int size)
{void *p;p = malloc(size);if(size <= 0){printf("error");return -1;}memcpy(p,a,size);memcpy(a,b,size);memcpy(b,p,size);free(p);return 0;
} int main()
{char *a = "12344";char *b = "adbde";swap(a,b,5);printf("a:%s\tb:%s\n",a,b);return 0;
}

程序输出

a:adbde b:12344--------------------------------
Process exited after 0.03298 seconds with return value 0
请按任意键继续. . .

#void* 指针

程序里面有一个 void 指针，这个需要注意下，void指针是可以接收所有类型的指针的，但是其他类型的指针是不能直接接收void指针的。

比如这样

#include "stdio.h"int main()
{ void *p = NULL;char *p1 = NULL;p = p1;return 0;
}

下面这样可能是有问题的

我说是可能有问题是因为GCC、ANSI C 的编译情况是不同的。

#include "stdio.h"int main()
{ void *p = NULL;char *p1 = NULL;p1 = p;return 0;
}

#程序有什么问题？

这段代码有以下几点问题：

1、入参a和b都是void类型的指针，不清楚a和b指向buff的长度是否都为size？size不要设置成int，最好设置成unsigned int，而且a和b入参一定不能代入const xxxxx *类型的实参;

2、进入函数体之后应该首先判断size是否小于等于0、a和b是否为NULL，如果size小于等于0，或者a和b存在NULL，那其他代码将毫无意义。

3、没有if(!p)对malloc进行判断直接使用p，剩余内存较低情况下malloc一片大内存是有可能fail的，直接使用p就会段错，而且不建议malloc。

4、memcpy时因并不明确a b指针指向buff的长度是否都为size，存在内存越界风险。

综上述修改代码为：

int swap(void *a, void *b, uint32_t len_a, uint32_t len_b)
{if ( !a || !b || !len_a || !len_b || len_a!=len_b ) {printf("invalid param! \n");return -1;}if ( a==b ) {printf("the same! \n");return 0;}uint32_t len = len_a;char *p = (char *)malloc(len);if (!p) {printf("malloc fail! \n");return -1;}memcpy(p,a,len);memcpy(a,b,len);memcpy(b,p,len);free(p);return 0;
｝

#参考代码

//我写了一个swap：
//感谢读者LinuxEnginetypedef struct {char byte[4];
} stdb_4;
typedef struct {char byte[8];
} stdb_8;void my_swap(void *p, void *q, size_t p_sz, size_t q_sz)
{char buff[256];if ( !p || !q ||!p_sz || !q_sz || p_sz != q_sz || p==q )return;int sz = p_sz;if ( sz == 4 ) {stdb_4 t  = *(stdb_4 *)p;*(stdb_4 *)p  = *(stdb_4 *)q;*(stdb_4 *)q  = t;return;} else if ( sz == 8 ) {stdb_8 t  = *(stdb_8 *)p;*(stdb_8 *)p  = *(stdb_8 *)q;*(stdb_8 *)q  = t;return;}while ( sz > sizeof(buff) ) {my_swap (p, q, sizeof(buff));p = (char *)p + sizeof(buff);q = (char *)q + sizeof(buff);sz -= sizeof(buff);}memcpy(buff, p,    sz);memcpy(p,    q,    sz);memcpy(q,    buff, sz);
}