ControlFlag基于機器學(xué)習(xí)的代碼檢測工具

ControlFlag 是一個自監(jiān)督的特殊模式檢測系統(tǒng)，它通過從開源存儲庫（在 GitHub 和其他版本控制系統(tǒng)上）挖掘這些模式來學(xué)習(xí)高級編程語言（如 C/C++）控制結(jié)構(gòu)中出現(xiàn)的典型模式。然后應(yīng)用學(xué)習(xí)到的模式來檢測用戶代碼中的異常模式。

簡要技術(shù)說明

ControlFlag 的模式異常檢測系統(tǒng)可用于解決各種問題，例如排版錯誤檢測、標(biāo)記缺失的 NULL 檢查等等。

下圖顯示了 ControlFlag 的兩個主要階段：（1）模式挖掘階段，以及（2）異常模式掃描階段。模式挖掘階段是一個“訓(xùn)練階段”，它在用戶提供的 GitHub 倉庫中挖掘典型模式，然后從挖掘的模式中構(gòu)建決策樹。另一方面，掃描階段應(yīng)用挖掘的模式來標(biāo)記用戶指定的目標(biāo)倉庫中的異常表達(dá)式。

目錄結(jié)構(gòu)（不斷發(fā)展）

• src: 用于排版錯誤檢測系統(tǒng)的 ControlFlag 的源代碼
• scripts: 用于模式挖掘和掃描異常的腳本
• quick_start：運行快速啟動測試的腳本
• github：用于下載 GitHub 倉庫的腳本和數(shù)據(jù)。它還包含預(yù)處理的訓(xùn)練數(shù)據(jù)，其中包含使用 C 作為主要語言從 6000 個 GitHub 倉庫中挖掘的模式。
• tests: 單元測試

安裝

ControlFlag 可以在 Linux 和 MacOS 上構(gòu)建。

要求

• CMake 3.4.3 或以上
• C++17 兼容編譯器
• Tree-sitter?解析器（作為 cmake 的一部分自動下載）
• GNU parallel（可選，如果您想生成自己的訓(xùn)練數(shù)據(jù)）

在基于 Linux 的系統(tǒng)上測試構(gòu)建配置

• CentOS-7.6/Ubuntu-20.04 with g++-v10.2.0 for x86_64

在 MacOS 上測試構(gòu)建配置

• MacOS Mojave v10.14.6 with clang-1001.0.46.4 (Apple LLVM version 10.0.1) for x86_64（從命令行工具包獲得）

構(gòu)建

$ cd controlflag
$ cmake .
$ make -j
$ make test

所有測試都make test應(yīng)該通過，但目前 Verilog 的測試由于版本不匹配問題而失敗。Verilog 支持是 WIP。

?

?