├── LICENSE ├── README.md ├── baselines.zip ├── configs.json ├── configs.py ├── data ├── cpg │ ├── 0_cpg.pkl │ └── 1_cpg.pkl ├── input │ ├── 0_cpg_input.pkl │ └── 1_cpg_input.pkl ├── model │ └── mydevignModel.pth └── tokens │ ├── 0_cpg_tokens.pkl │ └── 1_cpg_tokens.pkl ├── fine-tune.py ├── input ├── 0_cpg_input.pkl └── 1_cpg_input.pkl ├── joern ├── .gitignore └── graph-for-funcs.sc ├── models ├── LMGNN.py └── layers.py ├── run.py ├── test.py └── utils ├── __init__.py ├── __pycache__ └── __init__.cpython-38.pyc ├── data ├── __init__.py ├── __pycache__ │ ├── __init__.cpython-38.pyc │ └── datamanager.cpython-38.pyc └── datamanager.py ├── functions ├── __init__.py ├── __pycache__ │ ├── __init__.cpython-38.pyc │ ├── cpg.cpython-38.pyc │ ├── digraph.cpython-38.pyc │ ├── input_dataset.cpython-38.pyc │ ├── log.cpython-38.pyc │ └── parse.cpython-38.pyc ├── cpg.py ├── cpg │ ├── __init__.py │ ├── __pycache__ │ │ ├── __init__.cpython-38.pyc │ │ ├── ast.cpython-38.pyc │ │ ├── edge.cpython-38.pyc │ │ ├── function.cpython-38.pyc │ │ ├── node.cpython-38.pyc │ │ └── properties.cpython-38.pyc │ ├── ast.py │ ├── edge.py │ ├── function.py │ ├── node.py │ └── properties.py ├── digraph.py ├── input_dataset.py ├── log.py └── parse.py └── process ├── README.md ├── __init__.py ├── cpg_client_wrapper.py ├── cpg_generator.py └── embeddings.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vul-LMGNN/vul-LMGGNN/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vul-LMGNN/vul-LMGGNN/HEAD/README.md -------------------------------------------------------------------------------- /baselines.zip: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vul-LMGNN/vul-LMGGNN/HEAD/baselines.zip -------------------------------------------------------------------------------- /configs.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vul-LMGNN/vul-LMGGNN/HEAD/configs.json -------------------------------------------------------------------------------- /configs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vul-LMGNN/vul-LMGGNN/HEAD/configs.py -------------------------------------------------------------------------------- /data/cpg/0_cpg.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vul-LMGNN/vul-LMGGNN/HEAD/data/cpg/0_cpg.pkl -------------------------------------------------------------------------------- /data/cpg/1_cpg.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vul-LMGNN/vul-LMGGNN/HEAD/data/cpg/1_cpg.pkl -------------------------------------------------------------------------------- /data/input/0_cpg_input.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vul-LMGNN/vul-LMGGNN/HEAD/data/input/0_cpg_input.pkl -------------------------------------------------------------------------------- /data/input/1_cpg_input.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vul-LMGNN/vul-LMGGNN/HEAD/data/input/1_cpg_input.pkl -------------------------------------------------------------------------------- /data/model/mydevignModel.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vul-LMGNN/vul-LMGGNN/HEAD/data/model/mydevignModel.pth -------------------------------------------------------------------------------- /data/tokens/0_cpg_tokens.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vul-LMGNN/vul-LMGGNN/HEAD/data/tokens/0_cpg_tokens.pkl -------------------------------------------------------------------------------- /data/tokens/1_cpg_tokens.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vul-LMGNN/vul-LMGGNN/HEAD/data/tokens/1_cpg_tokens.pkl -------------------------------------------------------------------------------- /fine-tune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vul-LMGNN/vul-LMGGNN/HEAD/fine-tune.py -------------------------------------------------------------------------------- /input/0_cpg_input.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vul-LMGNN/vul-LMGGNN/HEAD/input/0_cpg_input.pkl -------------------------------------------------------------------------------- /input/1_cpg_input.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vul-LMGNN/vul-LMGGNN/HEAD/input/1_cpg_input.pkl -------------------------------------------------------------------------------- /joern/.gitignore: -------------------------------------------------------------------------------- 1 | joern-cli* -------------------------------------------------------------------------------- /joern/graph-for-funcs.sc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vul-LMGNN/vul-LMGGNN/HEAD/joern/graph-for-funcs.sc -------------------------------------------------------------------------------- /models/LMGNN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vul-LMGNN/vul-LMGGNN/HEAD/models/LMGNN.py -------------------------------------------------------------------------------- /models/layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vul-LMGNN/vul-LMGGNN/HEAD/models/layers.py -------------------------------------------------------------------------------- /run.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vul-LMGNN/vul-LMGGNN/HEAD/run.py -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vul-LMGNN/vul-LMGGNN/HEAD/test.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/__pycache__/__init__.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vul-LMGNN/vul-LMGGNN/HEAD/utils/__pycache__/__init__.cpython-38.pyc -------------------------------------------------------------------------------- /utils/data/__init__.py: -------------------------------------------------------------------------------- 1 | from .datamanager import * 2 | -------------------------------------------------------------------------------- /utils/data/__pycache__/__init__.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vul-LMGNN/vul-LMGGNN/HEAD/utils/data/__pycache__/__init__.cpython-38.pyc -------------------------------------------------------------------------------- /utils/data/__pycache__/datamanager.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vul-LMGNN/vul-LMGGNN/HEAD/utils/data/__pycache__/datamanager.cpython-38.pyc -------------------------------------------------------------------------------- /utils/data/datamanager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vul-LMGNN/vul-LMGGNN/HEAD/utils/data/datamanager.py -------------------------------------------------------------------------------- /utils/functions/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vul-LMGNN/vul-LMGGNN/HEAD/utils/functions/__init__.py -------------------------------------------------------------------------------- /utils/functions/__pycache__/__init__.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vul-LMGNN/vul-LMGGNN/HEAD/utils/functions/__pycache__/__init__.cpython-38.pyc -------------------------------------------------------------------------------- /utils/functions/__pycache__/cpg.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vul-LMGNN/vul-LMGGNN/HEAD/utils/functions/__pycache__/cpg.cpython-38.pyc -------------------------------------------------------------------------------- /utils/functions/__pycache__/digraph.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vul-LMGNN/vul-LMGGNN/HEAD/utils/functions/__pycache__/digraph.cpython-38.pyc -------------------------------------------------------------------------------- /utils/functions/__pycache__/input_dataset.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vul-LMGNN/vul-LMGGNN/HEAD/utils/functions/__pycache__/input_dataset.cpython-38.pyc -------------------------------------------------------------------------------- /utils/functions/__pycache__/log.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vul-LMGNN/vul-LMGGNN/HEAD/utils/functions/__pycache__/log.cpython-38.pyc -------------------------------------------------------------------------------- /utils/functions/__pycache__/parse.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vul-LMGNN/vul-LMGGNN/HEAD/utils/functions/__pycache__/parse.cpython-38.pyc -------------------------------------------------------------------------------- /utils/functions/cpg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vul-LMGNN/vul-LMGGNN/HEAD/utils/functions/cpg.py -------------------------------------------------------------------------------- /utils/functions/cpg/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vul-LMGNN/vul-LMGGNN/HEAD/utils/functions/cpg/__init__.py -------------------------------------------------------------------------------- /utils/functions/cpg/__pycache__/__init__.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vul-LMGNN/vul-LMGGNN/HEAD/utils/functions/cpg/__pycache__/__init__.cpython-38.pyc -------------------------------------------------------------------------------- /utils/functions/cpg/__pycache__/ast.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vul-LMGNN/vul-LMGGNN/HEAD/utils/functions/cpg/__pycache__/ast.cpython-38.pyc -------------------------------------------------------------------------------- /utils/functions/cpg/__pycache__/edge.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vul-LMGNN/vul-LMGGNN/HEAD/utils/functions/cpg/__pycache__/edge.cpython-38.pyc -------------------------------------------------------------------------------- /utils/functions/cpg/__pycache__/function.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vul-LMGNN/vul-LMGGNN/HEAD/utils/functions/cpg/__pycache__/function.cpython-38.pyc -------------------------------------------------------------------------------- /utils/functions/cpg/__pycache__/node.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vul-LMGNN/vul-LMGGNN/HEAD/utils/functions/cpg/__pycache__/node.cpython-38.pyc -------------------------------------------------------------------------------- /utils/functions/cpg/__pycache__/properties.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vul-LMGNN/vul-LMGGNN/HEAD/utils/functions/cpg/__pycache__/properties.cpython-38.pyc -------------------------------------------------------------------------------- /utils/functions/cpg/ast.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vul-LMGNN/vul-LMGGNN/HEAD/utils/functions/cpg/ast.py -------------------------------------------------------------------------------- /utils/functions/cpg/edge.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vul-LMGNN/vul-LMGGNN/HEAD/utils/functions/cpg/edge.py -------------------------------------------------------------------------------- /utils/functions/cpg/function.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vul-LMGNN/vul-LMGGNN/HEAD/utils/functions/cpg/function.py -------------------------------------------------------------------------------- /utils/functions/cpg/node.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vul-LMGNN/vul-LMGGNN/HEAD/utils/functions/cpg/node.py -------------------------------------------------------------------------------- /utils/functions/cpg/properties.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vul-LMGNN/vul-LMGGNN/HEAD/utils/functions/cpg/properties.py -------------------------------------------------------------------------------- /utils/functions/digraph.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vul-LMGNN/vul-LMGGNN/HEAD/utils/functions/digraph.py -------------------------------------------------------------------------------- /utils/functions/input_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vul-LMGNN/vul-LMGGNN/HEAD/utils/functions/input_dataset.py -------------------------------------------------------------------------------- /utils/functions/log.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vul-LMGNN/vul-LMGGNN/HEAD/utils/functions/log.py -------------------------------------------------------------------------------- /utils/functions/parse.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vul-LMGNN/vul-LMGGNN/HEAD/utils/functions/parse.py -------------------------------------------------------------------------------- /utils/process/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vul-LMGNN/vul-LMGGNN/HEAD/utils/process/README.md -------------------------------------------------------------------------------- /utils/process/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vul-LMGNN/vul-LMGGNN/HEAD/utils/process/__init__.py -------------------------------------------------------------------------------- /utils/process/cpg_client_wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vul-LMGNN/vul-LMGGNN/HEAD/utils/process/cpg_client_wrapper.py -------------------------------------------------------------------------------- /utils/process/cpg_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vul-LMGNN/vul-LMGGNN/HEAD/utils/process/cpg_generator.py -------------------------------------------------------------------------------- /utils/process/embeddings.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vul-LMGNN/vul-LMGGNN/HEAD/utils/process/embeddings.py --------------------------------------------------------------------------------