util.py 文件源码

def prepare_working_dir(directory_path, purge=False):
    folders = ["/findings/", "/corpus", "/findings/panic", "/findings/kasan", "/findings/timeout", "/rbuf", "/evaluation"]

    if purge:
        if os.path.isdir(directory_path):
            for root, dirs, files in os.walk(directory_path, topdown=False):
                for name in files:
                    os.remove(os.path.join(root, name))
                for name in dirs:
                    os.rmdir(os.path.join(root, name))
        if os.path.exists("/dev/shm/kafl_filter0"):
            os.remove("/dev/shm/kafl_filter0")
        if os.path.exists("/dev/shm/kafl_tfilter"):
            os.remove("/dev/shm/kafl_tfilter")

    if len(os.listdir(directory_path)) == 0:
        for folder in folders:
            os.makedirs(directory_path + folder)
    else:
        for folder in folders:
            if not os.path.isdir(directory_path + folder):
                return False
    return True