CodeQL自动化代码审计初探篇

CodeQL 介绍

CodeQL 是一款功能强大的静态代码分析工具，旨在帮助开发人员和安全研究人员自动查找代码错误、检查代码质量以及识别漏洞，并协助手动代码审查。目前支持的语言包括 C/C++、C#、Go、Java、Kotlin、JavaScript、Python、Ruby、TypeScript 和 Swift。

它的前身是牛津大学衍生公司 Semmle 基于牛津大学团队十多年的研究成果推出的一个在线代码扫描平台 LGTM.com，GitHub 于 2019 年收购了 Semmle 公司，以利用该公司的 CodeQL 分析技术对自家数量庞大的代码库进行安全扫描，并于 2020 年底对外开放了 CodeQL，但未开源。

CodeQL 通过创建程序的结构化信息集合数据库，对代码进行语义分析，这意味着它不仅可以进行简单的模式匹配，还能够理解代码中的行为和关系。同时，CodeQL 还提供一种表达式、声明式逻辑查询语言 QL，可用于编写复杂的、自定义的代码查询，类似处理数据一样去处理代码，从而通过数据流和污点分析来发现代码库中特定类型的漏洞或安全风险。

环境准备

在使用 CodeQL 前，先下载安装 CodeQL CLI 命令行工具、CodeQL 标准库以及 Visual Studio Code CodeQL 扩展，当然 Visual Studio Code 也是不可少的，因为后续的 CodeQL 查询是在 Visual Studio Code 中进行的。除此之外，还要准备一个带有丰富漏洞的靶场环境。如果系统是 ARM 架构的 macOS，则还需要额外安装依赖。

额外依赖安装

对于 ARM 架构的 macOS，需额外安装两个依赖，通过如下命令安装。虽然 CodeQL 目前支持 ARM 架构的 macOS，但对于 ARM 架构的 Linux 和 Windows 系统，截止本文发布之时还未得到 CodeQL 的支持。

1	xcode-select --install && softwareupdate --install-rosetta

CodeQL CLI

CodeQL CLI 是一个独立的命令行工具，其主要目的是生成代码库的数据库表示形式，即 CodeQL 数据库，当数据库生成后，就可通过交互式查询来查询它，或者运行一系列查询以生成一组 SARIF 格式的结果。

通过其 GitHub Releases 页（https://github.com/github/codeql-cli-binaries/releases ）下载最新版本的压缩包，下载后将其解压。同时，在家目录下创建一个 CodeQL 文件夹，将刚才解压后的整个文件夹放置其中。

1 2	mkdir ~/CodeQL/ mv ~/Downloads/codeql ~/CodeQL

以及配置环境变量并重载。

1 2	echo '\n# CodeQL\nexport PATH="$HOME/CodeQL/codeql:$PATH"' >> ~/.zshrc source ~/.zshrc

现在，在系统任何地方都可以直接运行 codeql 命令，若出现如下结果就意味着 CodeQL CLI 配置成功。

codeql
Usage: codeql <command> <argument>...
Create and query CodeQL databases, or work with the QL language.

GitHub makes this program freely available for the analysis of open-source software and certain other uses, but it
is not itself free software. Type codeql --license to see the license terms.

......

在将来，建议保持 codeql-cli-binaries 为最新版本。

CodeQL 查询库

GitHub 中的 codeql 仓库（https://github.com/github/codeql ）包含所有受支持语言的 CodeQL 分析所需的查询和库，可将其直接下载至本地文件夹以供使用。

不过，GitHub 提供了一种在 VS Code 中更方便地使用查询库的方式。使用如下命令递归克隆 vscode-codeql-starter 仓库至本地~/CodeQL 文件夹，这同时也会将 codeql 仓库克隆至 ql 文件夹中。

1	cd ~/CodeQL && git clone --recursive https://github.com/github/vscode-codeql-starter

进入 vscode-codeql-starter 中，可以看到 ql 文件夹包含各种语言的查询库。在将来，建议不定期地执行git submodule update --remote命令以更新查询库。

cd vscode-codeql-starter && ls ql
BUILD.bazel         MODULE.bazel          config       docs        python
CODEOWNERS          README.md             conftest.py  go          ql
CODE_OF_CONDUCT.md  WORKSPACE.bazel       cpp          java        ruby
CONTRIBUTING.md     change-notes          csharp       javascript  shared
LICENSE             codeql-workspace.yml  defs.bzl     misc        swift

将 vscode-codeql-starter 克隆至本地后，每次都可以使用 VS Code 将其作为一个工作区打开。

未来自己编写的 CodeQL 查询，都可以存放在这些目录中。

在vscode-codeql-starter/ql/java/ql/src/Security/CWE/文件夹中以.ql 作为后缀名的就是 GitHub 官方维护的 Java 程序常见漏洞的 CodeQL 查询，可直接拿来使用。

更直观地，可在如下网站（https://codeql.github.com/codeql-query-help/java-cwe/ ）看到受支持的全部 CWE（通用弱点枚举）。

VS Code 扩展

对于 VS Code 相关扩展的安装，可直接在 VS Code 的扩展中搜索 CodeQL 关键词或通过如下链接进行安装。

https://marketplace.visualstudio.com/items?itemName=GitHub.vscode-codeql

除 CodeQL 扩展外，建议再安装一个 SARIF Viewer 扩展，用于查看 SARIF 日志文件。SARIF 全称静态分析结果交换格式，是一种定义输出文件格式的 OASIS 标准，SARIF 标准用于简化静态分析工具分享其结果的方式。

https://marketplace.visualstudio.com/items?itemName=MS-SarifVSCode.sarif-viewer

漏洞靶场环境

对于漏洞靶场环境，本文将会采用 Java Sec Code 这一含有丰富漏洞类型的靶场环境，在~/CodeQL 下创建 vulns 文件夹，将这个靶场克隆至其中。

1 2	cd ~/CodeQL && mkdir vulns && cd vulns git clone https://github.com/JoyChou93/java-sec-code/

数据库创建

CodeQL 数据库是代码库的关系表示，其中包含不同源代码元素（如类和函数）的信息，在创建数据库的过程中会提取代码库中每一个源文件的单一关系表示，并将每个元素放入单独的数据表中。

对于编译型语言，提取是通过监控正常的编译过程来实现的，每次调用编译器处理源文件时，都会复制一份该文件，并收集源代码的所有相关信息，包括抽象语法树的语法数据，以及名称绑定和类型信息的语义数据。

由于 Java 是编译型语言，CodeQL 需要调用所需的构建系统来生成数据库，因此构建方法须提供给 CodeQL，也就是-c接收的参数值mvn clean install。而对于解释型语言，如 JavaScript、Python 则无需添加该参数。最终通过如下命令来创建数据库。

1	codeql database create ./jscdb -s ./java-sec-code -l java -c "mvn clean install -DskipTests"

数据库分析

利用 CLI 分析数据库

在数据库生成后，就可利用先前安装的 CodeQL CLI 工具来运行一系列查询，以生成一组 SARIF 格式或 CSV 格式的结果。

首先是 SARIF 格式，命令如下，等待分析完毕，便会在当前目录下生成 jsc-scan-result.sarif 文件。

1	codeql database analyze jscdb --format=sarif-latest --output=jsc-scan-result.sarif

接着，通过 VS Code 打开 vscode-codeql-starter 工作区，并在 CodeQL 插件中添加 jscdb 数据库文件夹。

然后，再用 VS Code 打开 jsc-scan-result.sarif 文件。如下图所见，此前安装的 SARIF Viewer 扩展起到作用了，这样分析 SARIF 文件就会变得很方便。

漏洞代码如下。

@PostMapping("/upload")
public String singleFileUpload(@RequestParam("file") MultipartFile file,
                               RedirectAttributes redirectAttributes) {
    if (file.isEmpty()) {
        // 赋值给uploadStatus.html里的动态参数message
        redirectAttributes.addFlashAttribute("message", "Please select a file to upload");
        return "redirect:/file/status";
    }

    try {
        // Get the file and save it somewhere
        byte[] bytes = file.getBytes();
        Path path = Paths.get(UPLOADED_FOLDER + file.getOriginalFilename());
        Files.write(path, bytes);

        redirectAttributes.addFlashAttribute("message",
                "You successfully uploaded '" + UPLOADED_FOLDER + file.getOriginalFilename() + "'");

    } catch (IOException e) {
        redirectAttributes.addFlashAttribute("message", "upload failed");
        logger.error(e.toString());
    }

    return "redirect:/file/status";
}

除 SARIF 格式外，还可将分析结果导出成 CSV 格式的文件，利用如下命令。

1	codeql database analyze jscdb --format=csv --output=jsc-scan-result.csv

在 Excel 中打开 jsc-scan-result.csv 文件，可看到如下结果，但这种方式就不如上面的那么直观。

在 VS Code 中查询分析

除了利用 CLI 工具对数据库进行分析外，还可在 VS Code 中执行 CodeQL 查询分析。例如如下，运行 Log4J LDAP JNDI 注入查询。

@RestController
public class Log4j {

    private static final Logger logger = LogManager.getLogger("Log4j");

    /**
     * http://localhost:8080/log4j?token=${jndi:ldap://127.0.0.1:1389/0iun75}
     * Default: error/fatal/off
     * Fix: Update log4j to lastet version.
     */
    @RequestMapping(value = "/log4j")
    public String log4j(String token) {
        logger.error(token);
        return token;
    }

    public static void main(String[] args) {
        String poc = "${jndi:ldap://127.0.0.1:1389/0iun75}";
        logger.error(poc);
    }

}