其他分享
首页 > 其他分享> > 【AFL(七)】afl-fuzz.c小改——输出文件夹暂存

【AFL(七)】afl-fuzz.c小改——输出文件夹暂存

作者:互联网

前言:

之前已经写过两篇针对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 的处理是这样的:

  1. 检测 out_dir 文件的合法性,不仅检测路径合法,还检测是否可用、当前状态是否可以用于处理;
  2. 检查文件夹里面的内容 ,如果是之前跑的结果,检查里面的信息是不是 valuable,如果有价值,就会暂停,程序停止;
  3. 经过前两步的处理,保证可以对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

标签:AFL,afl,old,文件夹,小改,dir,out
来源: https://www.cnblogs.com/wayne-tao/p/12129385.html