├── .gitignore ├── LICENSE ├── README.md ├── configs └── dwk.yaml ├── data └── sensiAPI.txt ├── env.sh ├── mypy.ini ├── requirements.txt ├── run.sh └── src ├── data_generator.py ├── datas ├── datamodules.py ├── datasets.py ├── graphs.py └── samples.py ├── detect.py ├── evaluate.py ├── joern └── joern-parse.py ├── metrics.py ├── models ├── modules │ ├── common_layers.py │ └── gnns.py └── vd.py ├── preprocess ├── dataset_generator.py ├── symbolizer.py └── word_embedding.py ├── run.py ├── train.py ├── utils.py └── vocabulary.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jumormt/DeepWukong/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jumormt/DeepWukong/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jumormt/DeepWukong/HEAD/README.md -------------------------------------------------------------------------------- /configs/dwk.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jumormt/DeepWukong/HEAD/configs/dwk.yaml -------------------------------------------------------------------------------- /data/sensiAPI.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jumormt/DeepWukong/HEAD/data/sensiAPI.txt -------------------------------------------------------------------------------- /env.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jumormt/DeepWukong/HEAD/env.sh -------------------------------------------------------------------------------- /mypy.ini: -------------------------------------------------------------------------------- 1 | [mypy] 2 | ignore_missing_imports = True 3 | -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jumormt/DeepWukong/HEAD/requirements.txt -------------------------------------------------------------------------------- /run.sh: -------------------------------------------------------------------------------- 1 | PYTHONPATH="." python src/run.py -------------------------------------------------------------------------------- /src/data_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jumormt/DeepWukong/HEAD/src/data_generator.py -------------------------------------------------------------------------------- /src/datas/datamodules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jumormt/DeepWukong/HEAD/src/datas/datamodules.py -------------------------------------------------------------------------------- /src/datas/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jumormt/DeepWukong/HEAD/src/datas/datasets.py -------------------------------------------------------------------------------- /src/datas/graphs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jumormt/DeepWukong/HEAD/src/datas/graphs.py -------------------------------------------------------------------------------- /src/datas/samples.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jumormt/DeepWukong/HEAD/src/datas/samples.py -------------------------------------------------------------------------------- /src/detect.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jumormt/DeepWukong/HEAD/src/detect.py -------------------------------------------------------------------------------- /src/evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jumormt/DeepWukong/HEAD/src/evaluate.py -------------------------------------------------------------------------------- /src/joern/joern-parse.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jumormt/DeepWukong/HEAD/src/joern/joern-parse.py -------------------------------------------------------------------------------- /src/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jumormt/DeepWukong/HEAD/src/metrics.py -------------------------------------------------------------------------------- /src/models/modules/common_layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jumormt/DeepWukong/HEAD/src/models/modules/common_layers.py -------------------------------------------------------------------------------- /src/models/modules/gnns.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jumormt/DeepWukong/HEAD/src/models/modules/gnns.py -------------------------------------------------------------------------------- /src/models/vd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jumormt/DeepWukong/HEAD/src/models/vd.py -------------------------------------------------------------------------------- /src/preprocess/dataset_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jumormt/DeepWukong/HEAD/src/preprocess/dataset_generator.py -------------------------------------------------------------------------------- /src/preprocess/symbolizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jumormt/DeepWukong/HEAD/src/preprocess/symbolizer.py -------------------------------------------------------------------------------- /src/preprocess/word_embedding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jumormt/DeepWukong/HEAD/src/preprocess/word_embedding.py -------------------------------------------------------------------------------- /src/run.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jumormt/DeepWukong/HEAD/src/run.py -------------------------------------------------------------------------------- /src/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jumormt/DeepWukong/HEAD/src/train.py -------------------------------------------------------------------------------- /src/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jumormt/DeepWukong/HEAD/src/utils.py -------------------------------------------------------------------------------- /src/vocabulary.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jumormt/DeepWukong/HEAD/src/vocabulary.py --------------------------------------------------------------------------------