前言:
之前已经写过两篇针对tmin、cmin的修改,现在再改afl-fuzz.c已经轻车熟路了,对整个afl的源码结构也比较了解了。
afl-tmin:https://www.cnblogs.com/wayne-tao/p/11964565.html
afl-cmin:https://www.cnblogs.com/wayne-tao/p/11971922.html
此番修改是因为,对AFL关于 out_dir 文件夹的细节处理不是很满意,AFL对 out_dir 的处理是这样的:
- 检测 out_dir 文件的合法性,不仅检测路径合法,还检测是否可用、当前状态是否可以用于处理;
- 检查文件夹里面的内容 ,如果是之前跑的结果,检查里面的信息是不是 valuable,如果有价值,就会暂停,程序停止;
- 经过前两步的处理,保证可以对out_dir进行简单操作了,cleanup文件夹里的信息,开始使用out_dir。
源码分析:
【一】需求分析
当运行时间 过长时,比如这样,虽然没有crash,但是运行了很长时间,也经过了许多变异:
当停下之后,如果继续进行 afl 操作的话,会出现如下问题:
在PROGRAM ABORT处,可看到,程序停止的位置,现在的想法是:
当发现 valuable out info 的时候,对文件夹进行暂存操作,让程序继续进行,当然这里的操作跟 cmin 的处理如出一辙,先重命名文件夹,而后新建与原来同名的文件夹作为 out_dir。
【二】代码设计
通过FATAL()和输出内容很容易找到程序停止所在位置:
进行如下修改:(其中 old_out_dir 是在开始声明的跟 out_dir 相同类型的变量)
因为源码修改较多,所以,修改后的源码都会放在git上:https://github.com/WayneDevMaze/Chinese_noted_AFL
此处代码稍微解释一下:
1.这里的old_out_dir是之前声明过的跟out_dir同类型变量,然后 if 里面的 alloc_printf 是afl自己弄的一个拼接字符串的东西;
2.rename操作,重命名文件夹;
3.mkdir新建文件夹,用的是旧的名字;
4.如果最开始的拼接失败,可能是out_dir本身就有问题,停止程序;
ps.
OKF()//成功情况的输出提示 FATAL()//失败情况的输出提示,并且输出结束后,还会exit(1)
代码原文:
//发现文件夹有价值,又不舍得删,就备份一下,跟cmin思路相同 // if (old_out_dir = alloc_printf("%s_old", out_dir)) { rename(out_dir, old_out_dir); mkdir(out_dir, 0700); OKF("Success to create old_file_dir and move valuable files in it."); } else{ OKF("Fail to move file!!!"); FATAL("At-risk data found in ‘%s‘", out_dir); }
【三】运行效果
依旧用之前会停止的那个指令,发现程序继续运行,并且给与提示了:
而且存放out的那个文件夹,确实也暂存了:
总体来说,思路还是很清楚的,也不太麻烦,如果对 Linux C 有了解,还是很容易实现的,之后还会继续对AFL的功能进行自己的修改。
Reference
【1】LinuxC 文件操作指南:linuxC文件以及目录操作函数 - yongfengnice - 博客园
【2】Linux C文件夹重命名:linux c 重命名文件和文件夹_he_fa的专栏-CSDN博客
【3】uint8_t意味着什么:uint8_t / uint16_t / uint32_t /uint64_t数据类型详解 - Z--Y - 博客园
【4】我自己git的源码地址:https://github.com/WayneDevMaze/Chinese_noted_AFL
原文地址:https://www.cnblogs.com/wayne-tao/p/12129385.html