#安全学术圈 84 个 #论文笔记 192 个

原文标题：ENRE: A Tool Framework for Extensible eNtity Relation Extraction
原文作者：Wuxia Jin, Yuanfang Cai, Rick Kazman, et al.
原文链接：https://ieeexplore.ieee.org/abstract/document/8802634
发表会议：2019 IEEE/ACM 41st International Conference on Software Engineering: Companion Proceedings (ICSE-Companion)
笔记作者：senu11@SecQuan
笔记小编：ourren@SecQuan

1.目的

本节通过论文中的插图来引入作者此篇文章解决的问题。

如上图所示，函数 train1() 和 train2() 都调用了方法 run()，该方法属于类 FirstAgent 或 SecondAgent，作者使用商业逆向工程工具Understand来提取依赖项，发现它忽略了从 train2() 到 FirstAgent 或 SecondAgent 的关系，因为它无法解析是哪个函数运行的类（只能在运行时动态确定）。也即静态分析会忽略实体间潜在的关系。

另外一个问题就是现有的工具不能支持多种编程语言来提取实体间的依赖关系。

为此，作者提出了一个框架ENRE，可以从多种语言中提取代码实体之间的依赖关系。目前，该框架可以从Java、python、c++、JS/TS、golang中提取代码实体间的依赖关系。

项目地址：https://github.com/jinwuxia/ENRE

新搬迁至：https://github.com/xjtu-enre

2.ENRE框架

本节先介绍作者是如何定义与统一实体关系表示的，然后再介绍框架。

2.1.实体关系表示统一

提取实体间的依赖关系，需要首先定义实体、实体间的关系。

1）实体。实体是具有给定名称或标识符的对象。文章中定义了5种基本类型的实体：Package、File、Class、Function、Variable。

2）实体关系。实体关系即实体之间的关系。举个简单的例子：当实体 i 依赖于实体 j 来完成其功能时，实体 i 到实体 j 之间存在关系。文章中定义了 8 种类型的关系：① Import，② Extend (e.g. Implement, Inherit), ③ Call, ④ Set (a function modifies a variable), ⑤ Use (a function reads or uses a variable), ⑥ Parameter Type (a function takes a class type as a parameter), ⑦ Return Type (a function returns a class type), ⑧ Dynamic relation。关系具有如下属性：sourceId、destId、type、weight。

Dynamic relation，即动态关系，动态关系是只能在运行时精确确定的关系。静态关系和动态关系的区别主要表现在调用关系上。在判定调用类型时，文章为调用关系定义了两种动态依赖关系：

a）Internal Case，即内部调用。

b）External Case，即外部调用。如本文开头的贴图中例子演示，可以将对象作为参数传递而调用。

2.2.ENRE框架

ENRE框架如下所示，用户可根据需要自行扩展相关组件。ENRE以源代码作为输入，并输出代码之间的实体依赖关系，输出格式支持JSON、XML、DOT，用户也可以自行扩展输出其它格式。此工具框架基于Pipe-And-Filter模式，其中每个组件都可以被替换掉而不影响其他组件的工作。此工具由三部分组成，下面详细介绍。

1）Entity Generator，实体生成器。此部分将源代码转换为实体树，作为一种中间表示，包括两个部分：

a) Parser，解析器。文章使用 Antlr 作为底层解析器，对源代码进行词法和句法分析。它使用给定语言的语法规则生成解析树。

b) EntityTree Builder，实体树生成器。这部分遍历解析树来构建实体树。实体树中的每个节点都是一个具体实体，扩展了 UERR 中定义的基本实体。每个节点可以有一个父节点和多个子节点。实体树中的节点用关系相关的语句进行注释，如“import”、“function call”。该部分取决于何种语言，分析师需要从对应的语言的解析树中识别和实现具体实体。

2）Relation Generator，关系生成器。该组件使用两个组件从实体树中提取和分析关系：

a) PriRelation Extractor。该组件通过分析实体树中的注释来提取实体关系。在处理完所有的注解后，生成所有的Primitive Relations，这些关系是代码中的直接关系。该组件也取决于是何种语言，因为不同的语言在名称范围和解析方面具有不同的语义。

b) HiRelation Analyzer。根据层次范围，该组件处理Primitive Relations并生成层次关系，例如哪些包包含哪些文件，哪些文件包含哪些功能等。

3）Dependency Exporter，依赖导出器。这将从前两个组件中提取的关系转换为指定格式，也包括两个组件：a）Formator。该组件定义了数据模型来wrap实体关系。数据模型可以使用流行的数据模型，例如 GraphViz DOT ，亦或者自定义模型。

b) writer。该组件将实体和关系格式化输出为 .CSV、.XML 或 .JSON 等文件。

要为给定的编程语言创建引擎（文章定义从 ENRE 扩展的依赖项提取程序为 ENRE 引擎），用户只需扩展 EntityTree Builder 和 PriRelation Extractor 来适应不同的实体和关系类型。