1.gets()函数
问:请找出下面代码里的问题:
- #include<stdio.h>
- int main(void)
- {
- char buff[10];
- memset(buff,0,sizeof(buff));
-
- gets(buff);
-
- printf("\n The buffer entered is [%s]\n",buff);
-
- return 0;
- }
答:上面代码里的问题在于函数gets()的使用,这个函数从stdin接收一个字符串而不检查它所复制的缓存的容积,这可能会导致缓存溢出。这里推荐使用标准函数fgets()代替。
2.strcpy()函数
问:下面是一个简单的密码保护功能,你能在不知道密码的情况下将其破解吗?
- #include<stdio.h>
-
- int main(int argc, char *argv[])
- {
- int flag = 0;
- char passwd[10];
-
- memset(passwd,0,sizeof(passwd));
-
- strcpy(passwd, argv[1]);
-
- if(0 == strcmp("LinuxGeek", passwd))
- {
- flag = 1;
- }
-
- if(flag)
- {
- printf("\n Password cracked \n");
- }
- else
- {
- printf("\n Incorrect passwd \n");
-
- }
- return 0;
- }
答:破解上述加密的关键在于利用攻破strcpy()函数的漏洞。所以用户在向“passwd”缓存输入随机密码的时候并没有提前检查“passwd”的容量是否足够。所以,如果用户输入一个足够造成缓存溢出并且重写“flag”变量默认值所存在位置的内存的长“密码”,即使这个密码无法通过验证,flag验证位也变成了非零,也就可以获得被保护的数据了。例如:
- $ ./psswd aaaaaaaaaaaaa
-
- Password cracked
虽然上面的密码并不正确,但我们仍然可以通过缓存溢出绕开密码安全保护。
要避免这样的问题,建议使用 strncpy()函数。
作者注:最近的编译器会在内部检测栈溢出的可能,所以这样往栈里存储变量很难出现栈溢出。在我的gcc里默认就是这样,所以我不得不使用编译命令‘-fno-stack-protector’来实现上述方案。
3.main()的返回类型
问:下面的代码能 编译通过吗?如果能,它有什么潜在的问题吗?
- #include<stdio.h>
-
- void main(void)
- {
- char *ptr = (char*)malloc(10);
-
- if(NULL == ptr)
- {
- printf("\n Malloc failed \n");
- return;
- }
- else
- {
-
- free(ptr);
- }
-
- return;
- }
答:因为main()方法的返回类型,这段代码的错误在大多数编译器里会被当作警告。main()的返回类型应该是“int”而不是“void”。因为“int”返回类型会让程序返回状态值。这点非常重要,特别当程序是作为依赖于程序成功运行的脚本的一部分运行时。
4.内存泄露
问:下面的代码会导致内存泄漏吗?
- #include<stdio.h>
-
- void main(void)
- {
- char *ptr = (char*)malloc(10);
-
- if(NULL == ptr)
- {
- printf("\n Malloc failed \n");
- return;
- }
- else
- {
-
- }
-
- return;
- }
答:尽管上面的代码并没有释放分配给“ptr”的内存,但并不会在程序退出后导致内存泄漏。在程序结束后,所有这个程序分配的内存都会自动被处理掉。但如果上面的代码处于一个“while循环”中,那将会导致严重的内存泄漏问题!
提示:如果你想知道更多关于内存泄漏的知识和内存泄漏检测工具,可以来看看我们在Valgrind上的文章。
5.free()函数
问:下面的程序会在用户输入'freeze'的时候出问题,而'zebra'则不会,为什么?
- #include<stdio.h>
-
- int main(int argc, char *argv[])
- {
- char *ptr = (char*)malloc(10);
-
- if(NULL == ptr)
- {
- printf("\n Malloc failed \n");
- return -1;
- }
- else if(argc == 1)
- {
- printf("\n Usage \n");
- }
- else
- {
- memset(ptr, 0, 10);
-
- strncpy(ptr, argv[1], 9);
-
- while(*ptr != 'z')
- {
- if(*ptr == '')
- break;
- else
- ptr++;
- }
-
- if(*ptr == 'z')
- {
- printf("\n String contains 'z'\n");
-
- }
-
- free(ptr);
- }
-
- return 0;
- }
答:这里的问题在于,代码会(通过增加“ptr”)修改while循环里“ptr”存储的地址。当输入“zebra”时,while循环会在执行前被终止,因此传给free()的变量就是传给malloc()的地址。但在“freeze”时,“ptr”存储的地址会在while循环里被修改,因此导致传给free()的地址出错,也就导致了seg-fault或者崩溃。
6.使用_exit退出
问:在下面的代码中,atexit()并没有被调用,为什么?
- #include<stdio.h>
-
- void func(void)
- {
- printf("\n Cleanup function called \n");
- return;
- }
-
- int main(void)
- {
- int i = 0;
-
- atexit(func);
-
- for(;i<0xffffff;i++);
-
- _exit(0);
- }
这是因为_exit()函数的使用,该函数并没有调用atexit()等函数清理。如果使用atexit()就应当使用exit()或者“return”与之相配合。