txt2stix is a Python script that is designed to identify and extract IoCs and TTPs from text files, identify the relationships between them, convert them to STIX 2.1 objects, and output as a STIX 2.1 bundle.
The general design goal of txt2stix was to keep it flexible, but simple, so that new extractions could be added or modified over time.
In short txt2stix;
- takes a txt file input
- extracts observables for enabled extractions (ai, pattern, or lookup)
- converts extracted observables to STIX 2.1 objects
- generates the relationships between extracted observables (ai, standard)
- converts extracted relationships to STIX 2.1 SRO objects
- outputs a STIX 2.1 bundle
Install the required dependencies using:
# clone the latest code
git clone https://github.com/muchdogesec/txt2stix
cd txt2stix
# create a venv
python3 -m venv txt2stix-venv
source txt2stix-venv/bin/activate
# install requirements
pip3 install .
txt2stix has various settings that are defined in an .env
file.
To create a template for the file:
cp .env.example .env
To see more information about how to set the variables, and what they do, read the .env.markdown
file.
python3 txt2stix.py \
--relationship_mode MODE \
--input_file FILE.txt \
...
The following arguments are available:
--input_file
(REQUIRED): the file to be converted. Must be.txt
--name
(REQUIRED): name of file, max 72 chars. Will be used in the STIX Report Object created.--report_id
(OPTIONAL): Sometimes it is required to control the id of thereport
object generated. You can therefore pass a valid UUIDv4 in this field to be assigned to the report. e.g. passing2611965-930e-43db-8b95-30a1e119d7e2
would create a STIX object idreport--2611965-930e-43db-8b95-30a1e119d7e2
. If this argument is not passed, the UUID will be randomly generated.--tlp_level
(OPTIONAL): Options areclear
,green
,amber
,amber_strict
,red
. Default if not passed, isclear
.--confidence
(OPTIONAL): value between 0-100. Default if not passed is null.--labels
(OPTIONAL): comma seperated list of labels. Case-insensitive (will all be converted to lower-case). Alloweda-z
,0-9
. e.g.label1,label2
would create 2 labels.--created
(OPTIONAL): by default all objectcreated
times will take the time the script was run. If you want to explicitly set these times you can do so using this flag. Pass the value in the formatYYYY-MM-DDTHH:MM:SS.sssZ
e.g.2020-01-01T00:00:00.000Z
--use_identity
(OPTIONAL): can pass a full STIX 2.1 identity object (make sure to properly escape). Will be validated by the STIX2 library.--external_refs
(OPTIONAL): txt2stix will automatically populate theexternal_references
of the report object it creates for the input. You can use this value to add additional objects toexternal_references
. Note, you can only addsource_name
andexternal_id
values currently. Pass assource_name=external_id
. e.g.--external_refs txt2stix=demo1 source=id
would create the following objects under theexternal_references
property:{"source_name":"txt2stix","external_id":"demo1"},{"source_name":"source","external_id":"id"}
How the extractions are performed
--use_extractions
(REQUIRED): if you only want to use certain extraction types, you can pass their slug found in eitherincludes/ai/config.yaml
,includes/lookup/config.yaml
includes/pattern/config.yaml
(e.g.pattern_ipv4_address_only
). Default if not passed, no extractions applied.- Important: if using any AI extractions, you must set an OpenAI API key in your
.env
file - Important: if you are using any MITRE ATT&CK, CAPEC, CWE, ATLAS or Location extractions you must set
CTIBUTLER
or NVD CPE or CVE extractions you must setVULMATCH
settings in your.env
file
- Important: if using any AI extractions, you must set an OpenAI API key in your
--relationship_mode
(REQUIRED): either.ai
: AI provider must be enabled. extractions performed by either regex or AI for extractions user selected. Rich relationships created from AI provider from extractions.standard
: extractions performed by either regex or AI (AI provider must be enabled) for extractions user selected. Basic relationships created from extractions back to master Report object generated.
--ignore_image_refs
(defaulttrue
): images references in documents don't usually need extracting. e.g.<img src="https://example.com/image.png" alt="something">
you would not want domain or file extractions extractingexample.com
andimage.png
. Hence these are ignored by default (they are removed from text sent to extraction). Note, only theimg src
is ignored, all other values e.g.alt
are considered. If you want extractions to consider this data, set it tofalse
--ignore_link_refs
(defaulttrue
): link references in documents don't usually need extracting e.g.<a href="https://example.com/link.html" title="something">Bad Actor</a>
you would only wantBad actor
to be considered for extraction. Hence these part of the link are ignored by default (they are removed from text sent to extraction). Note, only thea href
is ignored, all other values e.g.title
are considered. Setting this tofalse
will also include everything inside the link tag (e.g.example.com
would extract as a domain)
If any AI extractions, or AI relationship mode is set, you must set the following accordingly
--ai_settings_extractions
:- defines the
provider:model
to be used for extractions. You can supply more than one provider. Seperate with a space (e.g.gpt-4o claude-3-opus-latest
) If more than one provider passed, txt2stix will take extractions from all models, de-dupelicate them, and them package them in the output. Currently supports: - See
tests/manual-tests/cases-ai-extraction-type.md
for some examples
- defines the
--ai_settings_relationships
:- similar to
ai_settings_extractions
but defines the model used to generate relationships. Only one model can be provided. Passed in same format asai_settings_extractions
- See
tests/manual-tests/cases-ai-relationships.md
for some examples
- similar to
It is very likely you'll want to extend txt2stix to include new extractions to;
- Add a new lookup extraction: add your lookup to
includes/lookups
as a.txt
file. Lookups should be a list of items seperated by new lines to be searched for in documents. Once this is added, updateincludes/extractions/lookup/config.yaml
with a new record pointing to your lookup. You can now use this lookup time at script run-time. - Add a new AI extraction: Edit
includes/extractions/ai/config.yaml
with a new record for your extraction. You can craft the prompt used in the config to control how the LLM performs the extraction.
Currently it is not possible to easily add any other types of extractions (without modifying the logic at a code level).
If you would like to understand how txt2stix works in more detail, please refer to the documentation in /docs/README.md
.
This documentation is paticularly helpful to read for those of you wanting to add your own custom extractions.
- Python Validators
- STIX 2: APIs for serializing and de-serializing STIX2 JSON content
- STIX 2 Pattern Validator: a tool for checking the syntax of the Cyber Threat Intelligence (CTI) STIX Pattern expressions
- MISP Warning Lists: Warning lists to inform users of MISP about potential false-positives or other information in indicators
- STIX Viewer: Quickly load bundles produced from your report