diff --git a/stix2/test/v21/test_malware.py b/stix2/test/v21/test_malware.py index a9158cf..f111826 100644 --- a/stix2/test/v21/test_malware.py +++ b/stix2/test/v21/test_malware.py @@ -193,3 +193,22 @@ def test_malware_non_family_no_name(): "is_family": False, "malware_types": ["something"], }) + + +def test_malware_with_os_refs(): + software = stix2.parse({ + "type": "software", + "name": "SuperOS", + "spec_version": "2.1", + }) + + malware = stix2.parse({ + "type": "malware", + "id": MALWARE_ID, + "spec_version": "2.1", + "is_family": False, + "malware_types": ["something"], + "operating_system_refs": [software], + }) + + assert malware["operating_system_refs"][0] == software["id"] diff --git a/stix2/v21/sdo.py b/stix2/v21/sdo.py index c0bdf30..a431b42 100644 --- a/stix2/v21/sdo.py +++ b/stix2/v21/sdo.py @@ -460,7 +460,7 @@ class Malware(STIXDomainObject): ('kill_chain_phases', ListProperty(KillChainPhase)), ('first_seen', TimestampProperty()), ('last_seen', TimestampProperty()), - ('os_execution_envs', ListProperty(StringProperty)), + ('operating_system_refs', ListProperty(ReferenceProperty(valid_types='software', spec_version='2.1'))), ('architecture_execution_envs', ListProperty(StringProperty)), ('implementation_languages', ListProperty(StringProperty)), ('capabilities', ListProperty(StringProperty)),