@mitre-attack/attack-data-model
Version:
A TypeScript API for the MITRE ATT&CK data model
1,334 lines (1,300 loc) • 111 kB
JavaScript
"use strict";
var __create = Object.create;
var __defProp = Object.defineProperty;
var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
var __getOwnPropNames = Object.getOwnPropertyNames;
var __getProtoOf = Object.getPrototypeOf;
var __hasOwnProp = Object.prototype.hasOwnProperty;
var __export = (target, all) => {
for (var name in all)
__defProp(target, name, { get: all[name], enumerable: true });
};
var __copyProps = (to, from, except, desc) => {
if (from && typeof from === "object" || typeof from === "function") {
for (let key of __getOwnPropNames(from))
if (!__hasOwnProp.call(to, key) && key !== except)
__defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
}
return to;
};
var __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__getProtoOf(mod)) : {}, __copyProps(
// If the importer is in node compatibility mode or this is not an ESM
// file that has been converted to a CommonJS file using a Babel-
// compatible transform (i.e. "__esModule" has not been set), then set
// "default" to the CommonJS "module.exports" for node compatibility.
isNodeMode || !mod || !mod.__esModule ? __defProp(target, "default", { value: mod, enumerable: true }) : target,
mod
));
var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
// src/index.ts
var index_exports = {};
__export(index_exports, {
AssetImpl: () => AssetImpl,
AttackBaseImpl: () => AttackBaseImpl,
AttackDataModel: () => AttackDataModel,
CampaignImpl: () => CampaignImpl2,
CollectionImpl: () => CollectionImpl,
DataComponentImpl: () => DataComponentImpl,
DataSourceImpl: () => DataSourceImpl,
DataSourceRegistration: () => DataSourceRegistration,
GroupImpl: () => GroupImpl,
IdentityImpl: () => IdentityImpl,
ImplementationLanguagesOpenVocabulary: () => ImplementationLanguagesOpenVocabulary,
MalwareCapabilitiesOpenVocabulary: () => MalwareCapabilitiesOpenVocabulary,
MalwareImpl: () => MalwareImpl,
MalwareTypeOpenVocabulary: () => MalwareTypeOpenVocabulary,
MarkingDefinitionImpl: () => MarkingDefinitionImpl,
MatrixImpl: () => MatrixImpl,
MitigationImpl: () => MitigationImpl,
ProcessorArchitecturesOpenVocabulary: () => ProcessorArchitecturesOpenVocabulary,
TacticImpl: () => TacticImpl,
TechniqueImpl: () => TechniqueImpl2,
ToolImpl: () => ToolImpl,
ToolTypesOpenVocabulary: () => ToolTypesOpenVocabulary,
aliasesSchema: () => aliasesSchema,
assetSchema: () => assetSchema,
attackBaseDomainObjectSchema: () => attackBaseDomainObjectSchema,
attackBaseMetaObjectSchema: () => attackBaseMetaObjectSchema,
attackBaseRelationshipObjectSchema: () => attackBaseRelationshipObjectSchema,
attackDomainSchema: () => attackDomainSchema,
attackMotivationOpenVocabulary: () => attackMotivationOpenVocabulary,
attackObjectsSchema: () => attackObjectsSchema,
attackResourceLevelOpenVocabulary: () => attackResourceLevelOpenVocabulary,
baseMarkingDefinitionSchema: () => baseMarkingDefinitionSchema,
campaignSchema: () => campaignSchema,
collectionSchema: () => collectionSchema,
createAttackExternalReferencesSchema: () => createAttackExternalReferencesSchema,
createAttackIdInExternalReferencesRefinement: () => createAttackIdInExternalReferencesRefinement,
createCitationsRefinement: () => createCitationsRefinement,
createEnterpriseOnlyPropertiesRefinement: () => createEnterpriseOnlyPropertiesRefinement,
createFirstAliasRefinement: () => createFirstAliasRefinement,
createFirstBundleObjectRefinement: () => createFirstBundleObjectRefinement,
createFirstXMitreAliasRefinement: () => createFirstXMitreAliasRefinement,
createMobileOnlyPropertiesRefinement: () => createMobileOnlyPropertiesRefinement,
createMultiStixTypeValidator: () => createMultiStixTypeValidator,
createOldMitreAttackIdSchema: () => createOldMitreAttackIdSchema,
createStixIdValidator: () => createStixIdValidator,
createStixTypeValidator: () => createStixTypeValidator,
dataComponentSchema: () => dataComponentSchema,
dataSourceSchema: () => dataSourceSchema,
descriptionSchema: () => descriptionSchema,
extensibleAssetSchema: () => extensibleAssetSchema,
extensibleCampaignSchema: () => extensibleCampaignSchema,
extensibleCollectionSchema: () => extensibleCollectionSchema,
extensibleDataComponentSchema: () => extensibleDataComponentSchema,
extensibleGroupSchema: () => extensibleGroupSchema,
extensibleIdentitySchema: () => extensibleIdentitySchema,
extensibleMalwareSchema: () => extensibleMalwareSchema,
extensibleMatrixSchema: () => extensibleMatrixSchema,
extensibleMitigationSchema: () => extensibleMitigationSchema,
extensibleSoftwareSchema: () => extensibleSoftwareSchema,
extensibleStixBundleSchema: () => extensibleStixBundleSchema,
extensibleTacticSchema: () => extensibleTacticSchema,
extensibleTechniqueSchema: () => extensibleTechniqueSchema,
extensibleToolSchema: () => extensibleToolSchema,
extensionSchema: () => extensionSchema,
extensionsSchema: () => extensionsSchema,
externalReferenceSchema: () => externalReferenceSchema,
externalReferencesSchema: () => externalReferencesSchema,
getDataSources: () => getDataSources,
getMitigations: () => getMitigations,
getPlatforms: () => getPlatforms,
getSubTechniques: () => getSubTechniques,
getTactics: () => getTactics,
granularMarkingSchema: () => granularMarkingSchema,
groupSchema: () => groupSchema,
identityClassOpenVocabulary: () => identityClassOpenVocabulary,
identitySchema: () => identitySchema,
industrySectorOpenVocabulary: () => industrySectorOpenVocabulary,
invalidRelationships: () => invalidRelationships,
isValidRelationship: () => isValidRelationship,
killChainNameSchema: () => killChainNameSchema,
killChainPhaseSchema: () => killChainPhaseSchema,
loadDataModel: () => loadDataModel,
malwareSchema: () => malwareSchema,
markingDefinitionSchema: () => markingDefinitionSchema,
matrixSchema: () => matrixSchema,
mitigationSchema: () => mitigationSchema,
nameSchema: () => nameSchema,
objectMarkingRefsSchema: () => objectMarkingRefsSchema,
objectVersionReferenceSchema: () => objectVersionReferenceSchema,
registerDataSource: () => registerDataSource,
relatedAssetSchema: () => relatedAssetSchema,
relatedAssetsSchema: () => relatedAssetsSchema,
relationshipSchema: () => relationshipSchema,
relationshipTypeSchema: () => relationshipTypeSchema,
softwareSchema: () => softwareSchema,
statementMarkingObjectSchema: () => statementMarkingObjectSchema,
stixArtifactType: () => stixArtifactType,
stixBundleSchema: () => stixBundleSchema,
stixCreatedByRefSchema: () => stixCreatedByRefSchema,
stixCreatedTimestampSchema: () => stixCreatedTimestampSchema,
stixDomainObjectSchema: () => stixDomainObjectSchema,
stixFileType: () => stixFileType,
stixIdentifierSchema: () => stixIdentifierSchema,
stixModifiedTimestampSchema: () => stixModifiedTimestampSchema,
stixRelationshipObjectSchema: () => stixRelationshipObjectSchema,
stixSpecVersionSchema: () => stixSpecVersionSchema,
stixTimestampSchema: () => stixTimestampSchema,
stixTypeSchema: () => stixTypeSchema,
stixTypeToTypeName: () => stixTypeToTypeName,
tacticSchema: () => tacticSchema,
techniqueSchema: () => techniqueSchema,
tlpAmberSchema: () => tlpAmberSchema,
tlpGreenSchema: () => tlpGreenSchema,
tlpMarkingDefinitionSchema: () => tlpMarkingDefinitionSchema,
tlpMarkingObjectSchema: () => tlpMarkingObjectSchema,
tlpRedSchema: () => tlpRedSchema,
tlpWhiteSchema: () => tlpWhiteSchema,
toolSchema: () => toolSchema,
validRelationshipObjectTypes: () => validRelationshipObjectTypes,
versionSchema: () => versionSchema,
xMitreAttackSpecVersionSchema: () => xMitreAttackSpecVersionSchema,
xMitreCollectionLayersSchema: () => xMitreCollectionLayersSchema,
xMitreContentsSchema: () => xMitreContentsSchema,
xMitreContributorsSchema: () => xMitreContributorsSchema,
xMitreDataSourceRefSchema: () => xMitreDataSourceRefSchema,
xMitreDataSourceSchema: () => xMitreDataSourceSchema,
xMitreDataSourcesSchema: () => xMitreDataSourcesSchema,
xMitreDefenseBypassesSchema: () => xMitreDefenseBypassesSchema,
xMitreDeprecatedSchema: () => xMitreDeprecatedSchema,
xMitreDetectionSchema: () => xMitreDetectionSchema,
xMitreDomainsSchema: () => xMitreDomainsSchema,
xMitreEffectivePermissionsSchema: () => xMitreEffectivePermissionsSchema,
xMitreFirstSeenCitationSchema: () => xMitreFirstSeenCitationSchema,
xMitreIdentity: () => xMitreIdentity,
xMitreIsSubtechniqueSchema: () => xMitreIsSubtechniqueSchema,
xMitreLastSeenCitationSchema: () => xMitreLastSeenCitationSchema,
xMitreModifiedByRefSchema: () => xMitreModifiedByRefSchema,
xMitreOldAttackIdSchema: () => xMitreOldAttackIdSchema,
xMitrePermissionsRequiredSchema: () => xMitrePermissionsRequiredSchema,
xMitrePlatformsSchema: () => xMitrePlatformsSchema,
xMitreRemoteSupportSchema: () => xMitreRemoteSupportSchema,
xMitreSectorsSchema: () => xMitreSectorsSchema,
xMitreShortNameSchema: () => xMitreShortNameSchema,
xMitreSystemRequirementsSchema: () => xMitreSystemRequirementsSchema,
xMitreTacticRefsSchema: () => xMitreTacticRefsSchema,
xMitreTacticTypeSchema: () => xMitreTacticTypeSchema,
xMitreVersionSchema: () => xMitreVersionSchema
});
module.exports = __toCommonJS(index_exports);
// src/classes/common/attack-object.impl.ts
var AttackBaseImpl = class {
/**
* Sets the object that revokes the current object.
* @param obj - The object that revokes this object.
*/
setRevokedBy(obj) {
this.revokedBy = obj;
}
/**
* Returns the object that revoked this object.
*/
getRevokedBy() {
return this.revokedBy;
}
};
// src/classes/sdo/asset.impl.ts
var AssetImpl = class extends AttackBaseImpl {
constructor(asset) {
super();
this.asset = asset;
// Custom properties prefixed with `_` to avoid conflicts
this._techniques = [];
Object.assign(this, asset);
}
addTechnique(technique) {
this._techniques.push(technique);
}
// Custom method or property
getDisplayName() {
return `${this.asset.name} - Asset`;
}
};
// src/classes/sdo/tactic.impl.ts
var TacticImpl = class extends AttackBaseImpl {
constructor(tactic) {
super();
this.tactic = tactic;
Object.assign(this, tactic);
}
};
// src/classes/sdo/mitigation.impl.ts
var MitigationImpl = class extends AttackBaseImpl {
constructor(mitigation) {
super();
this.mitigation = mitigation;
Object.assign(this, mitigation);
}
};
// src/classes/sdo/data-source.impl.ts
var DataSourceImpl = class extends AttackBaseImpl {
constructor(dataSource) {
super();
this.dataSource = dataSource;
Object.assign(this, dataSource);
}
};
// src/classes/sdo/data-component.impl.ts
var DataComponentImpl = class extends AttackBaseImpl {
constructor(dataComponent) {
super();
this.dataComponent = dataComponent;
this._detectedTechniques = [];
Object.assign(this, dataComponent);
}
// Add a technique detected by this data component
addDetectedTechnique(technique) {
this._detectedTechniques.push(technique);
}
// Getters
getDetectedTechniques() {
return this._detectedTechniques;
}
};
// src/classes/sdo/technique.impl.ts
var TechniqueImpl2 = class extends AttackBaseImpl {
constructor(technique) {
super();
this.technique = technique;
this._subTechniques = [];
this._tactics = [];
this._mitigations = [];
this._dataSources = [];
this._relatedTechniques = [];
this._targetAssets = [];
this._detectingDataComponents = [];
Object.assign(this, technique);
}
setParent(parent) {
this._parentTechnique = parent;
}
addSubTechnique(subTechnique) {
this._subTechniques.push(subTechnique);
}
addTactic(tactic) {
this._tactics.push(tactic);
}
addMitigation(mitigation) {
this._mitigations.push(mitigation);
}
addDataSource(dataSource) {
this._dataSources.push(dataSource);
}
addRelatedTechnique(technique) {
this._relatedTechniques.push(technique);
}
addTargetAsset(asset) {
this._targetAssets.push(asset);
}
addDetectingDataComponent(dataComponent) {
this._detectingDataComponents.push(dataComponent);
}
// Getters
getSubTechniques() {
return this._subTechniques;
}
getTactics() {
return this._tactics;
}
getMitigations() {
return this._mitigations;
}
getDataSources() {
return this._dataSources;
}
getParentTechnique() {
return this._parentTechnique;
}
getRelatedTechniques() {
return this._relatedTechniques;
}
getTargetAssets() {
return this._targetAssets;
}
getDetectingDataComponents() {
return this._detectingDataComponents;
}
};
// src/classes/sdo/malware.impl.ts
var MalwareImpl = class extends AttackBaseImpl {
constructor(malware) {
super();
this.malware = malware;
this._techniques = [];
Object.assign(this, malware);
}
// Add a technique used by the malware
addTechnique(technique) {
this._techniques.push(technique);
}
// Getters
getTechniques() {
return this._techniques;
}
};
// src/classes/sdo/tool.impl.ts
var ToolImpl = class extends AttackBaseImpl {
constructor(tool) {
super();
this.tool = tool;
this._techniques = [];
Object.assign(this, tool);
}
// Add a technique used by the tool
addTechnique(technique) {
this._techniques.push(technique);
}
// Getters
getTechniques() {
return this._techniques;
}
};
// src/classes/sdo/group.impl.ts
var GroupImpl = class extends AttackBaseImpl {
constructor(group) {
super();
this.group = group;
this._techniques = [];
this._software = [];
this._attributedCampaigns = [];
Object.assign(this, group);
}
// Add a technique used by the group
addTechnique(technique) {
this._techniques.push(technique);
}
// Add software used by the group
addSoftware(software) {
this._software.push(software);
}
addAttributedCampaign(campaign) {
this._attributedCampaigns.push(campaign);
}
// Getters
getTechniques() {
return this._techniques;
}
getSoftware() {
return this._software;
}
getAttributedCampaigns() {
return this._attributedCampaigns;
}
};
// src/classes/sdo/campaign.impl.ts
var CampaignImpl2 = class extends AttackBaseImpl {
constructor(campaign) {
super();
this.campaign = campaign;
this._techniques = [];
this._software = [];
Object.assign(this, campaign);
}
// Add a technique used by the campaign
addTechnique(technique) {
this._techniques.push(technique);
}
// Add software used by the campaign
addSoftware(software) {
this._software.push(software);
}
// Set the group this campaign is attributed to
setAttributedTo(group) {
this._attributedTo = group;
}
// Getters
getTechniques() {
return this._techniques;
}
getSoftware() {
return this._software;
}
getAttributedTo() {
return this._attributedTo;
}
};
// src/classes/sdo/collection.impl.ts
var CollectionImpl = class extends AttackBaseImpl {
constructor(collection) {
super();
this.collection = collection;
Object.assign(this, collection);
}
};
// src/classes/sdo/identity.impl.ts
var IdentityImpl = class extends AttackBaseImpl {
constructor(identity) {
super();
this.identity = identity;
Object.assign(this, identity);
}
};
// src/classes/sdo/matrix.impl.ts
var MatrixImpl = class extends AttackBaseImpl {
constructor(matrix) {
super();
this.matrix = matrix;
Object.assign(this, matrix);
}
};
// src/classes/smo/marking-definition.impl.ts
var MarkingDefinitionImpl = class extends AttackBaseImpl {
constructor(markingDefinition) {
super();
this.markingDefinition = markingDefinition;
Object.assign(this, markingDefinition);
}
};
// src/classes/sro/relationship.impl.ts
var RelationshipImpl = class extends AttackBaseImpl {
constructor(relationship) {
super();
this.relationship = relationship;
Object.assign(this, relationship);
}
};
// src/classes/attack-data-model.ts
var AttackDataModel = class {
constructor(uuid, attackObjects) {
this.uuid = uuid;
this.attackObjects = attackObjects;
this.techniques = [];
this.campaigns = [];
this.mitigations = [];
this.identities = [];
this.groups = [];
this.malware = [];
this.tools = [];
this.markingDefinitions = [];
this.dataComponents = [];
this.dataSources = [];
this.tactics = [];
this.assets = [];
this.matrices = [];
this.collections = [];
this.relationships = [];
this.populateData();
}
/**
* Returns the unique identifier for this data source/model.
* @returns string - Returns the unique identifier for this data source/model
*/
getUuid() {
return this.uuid;
}
/**
* Returns a list of ATT&CK objects that have been parsed by Zod schemas. These objects are not TS classes, but are plain JS objects. They do not contain relationship mappings.
* @returns AttackObject[] - a list of ATT&CK objects that have been parsed by Zod schemas. These objects are not TS classes, but are plain JS objects. They do not contain relationship mappings.
*/
getAttackObjects() {
return this.attackObjects;
}
/**
* Populates the class properties (e.g., techniques, groups, etc.) from the parsed objects array.
*/
populateData() {
const objectMap = /* @__PURE__ */ new Map();
this.attackObjects.forEach((object) => {
switch (object.type) {
// ASSET
case "x-mitre-asset": {
const asset = new AssetImpl(object);
this.assets.push(asset);
objectMap.set(object.id, asset);
break;
}
// CAMPAIGN
case "campaign": {
const campaign = new CampaignImpl2(object);
this.campaigns.push(campaign);
objectMap.set(object.id, campaign);
break;
}
// COLLECTION
case "x-mitre-collection": {
const collection = new CollectionImpl(object);
this.collections.push(collection);
objectMap.set(object.id, collection);
break;
}
// DATA COMPONENT
case "x-mitre-data-component": {
const dataComponent = new DataComponentImpl(object);
this.dataComponents.push(dataComponent);
objectMap.set(object.id, dataComponent);
break;
}
// DATA SOURCE
case "x-mitre-data-source": {
const dataSource = new DataSourceImpl(object);
this.dataSources.push(dataSource);
objectMap.set(object.id, dataSource);
break;
}
// GROUP
case "intrusion-set": {
const group = new GroupImpl(object);
this.groups.push(group);
objectMap.set(object.id, group);
break;
}
// IDENTITY
case "identity": {
const identity = new IdentityImpl(object);
this.identities.push(identity);
objectMap.set(object.id, identity);
break;
}
// MALWARE
case "malware": {
const malware = new MalwareImpl(object);
this.malware.push(malware);
objectMap.set(object.id, malware);
break;
}
// MATRIX
case "x-mitre-matrix": {
const matrix = new MatrixImpl(object);
this.matrices.push(matrix);
objectMap.set(object.id, matrix);
break;
}
// MITIGATION
case "course-of-action": {
const mitigation = new MitigationImpl(object);
this.mitigations.push(mitigation);
objectMap.set(object.id, mitigation);
break;
}
// TACTIC
case "x-mitre-tactic": {
const tactic = new TacticImpl(object);
this.tactics.push(tactic);
objectMap.set(object.id, tactic);
break;
}
// TECHNIQUE
case "attack-pattern": {
const technique = new TechniqueImpl2(object);
this.techniques.push(technique);
objectMap.set(object.id, technique);
break;
}
// TOOL
case "tool": {
const tool = new ToolImpl(object);
this.tools.push(tool);
objectMap.set(object.id, tool);
break;
}
// MARKING DEFINITION
case "marking-definition": {
const markingDefinition = new MarkingDefinitionImpl(object);
this.markingDefinitions.push(markingDefinition);
objectMap.set(object.id, markingDefinition);
break;
}
// RELATIONSHIP
case "relationship": {
const relationship = new RelationshipImpl(object);
this.relationships.push(relationship);
objectMap.set(object.id, relationship);
break;
}
}
});
this.initializeRelationships(objectMap);
}
/**
* Initializes relationships between objects, such as sub-techniques, tactics, mitigations, and more.
*/
initializeRelationships(objectMap) {
this.relationships.forEach((relationship) => {
const sourceObj = objectMap.get(relationship.source_ref);
const targetObj = objectMap.get(relationship.target_ref);
if (sourceObj && targetObj) {
switch (relationship.relationship_type) {
case "subtechnique-of":
if (sourceObj instanceof TechniqueImpl2 && targetObj instanceof TechniqueImpl2) {
sourceObj.setParent(targetObj);
targetObj.addSubTechnique(sourceObj);
}
break;
case "uses":
if (sourceObj instanceof GroupImpl && targetObj instanceof TechniqueImpl2) {
sourceObj.addTechnique(targetObj);
} else if (sourceObj instanceof CampaignImpl2 && targetObj instanceof TechniqueImpl2) {
sourceObj.addTechnique(targetObj);
} else if (sourceObj instanceof MalwareImpl && targetObj instanceof TechniqueImpl2) {
sourceObj.addTechnique(targetObj);
} else if (sourceObj instanceof ToolImpl && targetObj instanceof TechniqueImpl2) {
sourceObj.addTechnique(targetObj);
} else if (sourceObj instanceof GroupImpl && (targetObj instanceof MalwareImpl || targetObj instanceof ToolImpl)) {
sourceObj.addSoftware(targetObj);
} else if (sourceObj instanceof CampaignImpl2 && (targetObj instanceof MalwareImpl || targetObj instanceof ToolImpl)) {
sourceObj.addSoftware(targetObj);
}
break;
case "mitigates":
if (sourceObj instanceof MitigationImpl && targetObj instanceof TechniqueImpl2) {
targetObj.addMitigation(sourceObj);
}
break;
case "detects":
if (sourceObj instanceof DataComponentImpl && targetObj instanceof TechniqueImpl2) {
sourceObj.addDetectedTechnique(targetObj);
targetObj.addDetectingDataComponent(sourceObj);
}
break;
case "targets":
if (sourceObj instanceof TechniqueImpl2 && targetObj instanceof AssetImpl) {
sourceObj.addTargetAsset(targetObj);
}
break;
case "attributed-to":
if (sourceObj instanceof CampaignImpl2 && targetObj instanceof GroupImpl) {
sourceObj.setAttributedTo(targetObj);
targetObj.addAttributedCampaign(sourceObj);
}
break;
case "revoked-by":
if (sourceObj.constructor.name === targetObj.constructor.name) {
sourceObj.setRevokedBy(targetObj);
}
break;
default:
break;
}
}
});
}
// Other methods to query objects, get by ID, etc. (unchanged from previous version)
};
// src/classes/utils.ts
function getSubTechniques(technique, relationships, attackObjects) {
return relationships.filter((rel) => rel.relationship_type === "subtechnique-of" && rel.source_ref === technique.id).map((rel) => {
const subTech = attackObjects.find((obj) => obj.id === rel.target_ref);
return subTech;
}).filter((subTech) => subTech !== void 0);
}
function getTactics(technique, relationships, attackObjects) {
const killChainPhaseIds = technique.kill_chain_phases?.map((phase) => phase.phase_name) ?? [];
return attackObjects.filter((obj) => obj.type === "x-mitre-tactic" && killChainPhaseIds.includes(obj.id)).map((obj) => new TacticImpl(obj));
}
function getPlatforms(technique) {
return technique.x_mitre_platforms ?? [];
}
function getMitigations(technique, relationships, attackObjects) {
return relationships.filter((rel) => rel.relationship_type === "mitigates" && rel.target_ref === technique.id).map((rel) => {
const mitigation = attackObjects.find((obj) => obj.id === rel.source_ref);
if (mitigation && mitigation.type === "course-of-action") {
return new MitigationImpl(mitigation);
}
return null;
}).filter((mitigation) => mitigation !== null);
}
function getDataSources(technique, relationships, attackObjects) {
return relationships.filter((rel) => rel.relationship_type === "detects" && rel.target_ref === technique.id).map((rel) => {
const dataSource = attackObjects.find((obj) => obj.id === rel.source_ref);
if (dataSource && dataSource.type === "x-mitre-data-source") {
return new DataSourceImpl(dataSource);
}
return null;
}).filter((dataSource) => dataSource !== null);
}
// src/data-sources/fetch-attack-versions.ts
function normalizeVersion(version) {
return version.replace(/^v/, "");
}
async function fetchAttackVersions() {
const url = "https://api.github.com/repos/mitre-attack/attack-stix-data/releases";
const response = await fetch(url, {
headers: {
Accept: "application/vnd.github+json",
"X-GitHub-Api-Version": "2022-11-28"
}
});
if (!response.ok) {
throw new Error(`HTTP error! status: ${response.status}`);
}
const releases = await response.json();
const versions = releases.map((release) => normalizeVersion(release.tag_name)).sort((a, b) => {
const [aMajor, aMinor] = a.split(".").map(Number);
const [bMajor, bMinor] = b.split(".").map(Number);
if (bMajor !== aMajor) return bMajor - aMajor;
return bMinor - aMinor;
});
return versions;
}
// src/data-sources/data-source-registration.ts
var DataSourceRegistration = class {
/**
* Creates a new DataSourceRegistration instance.
* @param options - The data source options to register.
*/
constructor(options) {
this.options = options;
this.validateOptions();
}
/**
* Validates the data source options to ensure the correct fields are provided for each source type.
* @throws An error if validation fails.
*/
async validateOptions() {
const { source, parsingMode } = this.options;
if (parsingMode && !["strict", "relaxed"].includes(parsingMode)) {
throw new Error(`Invalid parsingMode: ${parsingMode}. Expected 'strict' or 'relaxed'.`);
}
switch (source) {
case "attack": {
await this.validateAttackOptions();
break;
}
case "file": {
this.validateFileOptions();
break;
}
case "url":
case "taxii": {
throw new Error(`The ${source} source is not implemented yet.`);
}
default: {
throw new Error(`Unsupported data source type: ${source}`);
}
}
}
/**
* Validates options specific to the 'attack' source type.
* @throws An error if validation fails.
*/
async validateAttackOptions() {
const { domain, version } = this.options;
if (!domain || !Object.values(attackDomainSchema.enum).includes(domain)) {
throw new Error(
`Invalid domain provided for 'attack' source. Expected one of: ${Object.values(
attackDomainSchema.enum
).join(", ")}`
);
}
if (version) {
const supportedVersions = await fetchAttackVersions();
const normalizedVersion = version.replace(/^v/, "");
if (!supportedVersions.includes(normalizedVersion)) {
throw new Error(
`Invalid version: ${version}. Supported versions are: ${supportedVersions.join(", ")}`
);
}
}
}
/**
* Validates options specific to the 'file' source type.
* @throws An error if validation fails.
*/
validateFileOptions() {
const { path } = this.options;
if (!path) {
throw new Error("The 'file' source requires a 'path' field to specify the file location.");
}
}
};
// src/schemas/common/attack-base-object.ts
var import_zod9 = require("zod");
// src/schemas/common/stix-core.ts
var import_zod8 = require("zod");
// src/schemas/common/stix-identifier.ts
var import_zod2 = require("zod");
// src/schemas/common/stix-type.ts
var import_zod = require("zod");
var stixTypeToTypeName = {
"attack-pattern": "Technique",
bundle: "StixBundle",
campaign: "Campaign",
"course-of-action": "Mitigation",
identity: "Identity",
"intrusion-set": "Group",
malware: "Malware",
tool: "Tool",
"marking-definition": "MarkingDefinition",
"x-mitre-data-component": "DataComponent",
"x-mitre-data-source": "DataSource",
"x-mitre-tactic": "Tactic",
"x-mitre-asset": "Asset",
"x-mitre-matrix": "Matrix",
"x-mitre-collection": "Collection",
relationship: "Relationship",
file: "",
// not used in ATT&CK but used in sample_refs for Malware
artifact: ""
// not used in ATT&CK but used in sample_refs for Malware
// 'observed-data': 'ObservedData', // not used in ATT&CK
// 'report': 'Report', // not used in ATT&CK
// 'threat-actor': 'ThreatActor', // not used in ATT&CK
// 'vulnerability': 'Vulnerability', // not used in ATT&CK
};
var supportedStixTypes = [
"attack-pattern",
"bundle",
"campaign",
"course-of-action",
"identity",
"intrusion-set",
"malware",
"tool",
"marking-definition",
"x-mitre-data-component",
"x-mitre-data-source",
"x-mitre-tactic",
"x-mitre-asset",
"x-mitre-matrix",
"x-mitre-collection",
"relationship",
"file",
// not used in ATT&CK but used in sample_refs for Malware
"artifact"
// not used in ATT&CK but used in sample_refs for Malware
// "indicator", // not used in ATT&CK
// "observed-data", // not used in ATT&CK
// "report", // not used in ATT&CK
// "threat-actor", // not used in ATT&CK
// "vulnerability", // not used in ATT&CK
];
var stixTypeSchema = import_zod.z.enum(supportedStixTypes, {
errorMap: (issue, ctx) => {
if (issue.code === "invalid_enum_value") {
const received = typeof ctx.data === "string" ? ctx.data : String(ctx.data);
return {
message: `Invalid STIX type '${received}'. Expected one of the supported STIX types.`
};
}
return { message: ctx.defaultError };
}
}).describe(
"The type property identifies the type of STIX Object (SDO, Relationship Object, etc). The value of the type field MUST be one of the types defined by a STIX Object (e.g., indicator)."
);
function createStixTypeValidator(stixType) {
const objectName = stixTypeToTypeName[stixType];
return import_zod.z.string().superRefine((val, ctx) => {
if (val !== stixType) {
ctx.addIssue({
code: import_zod.z.ZodIssueCode.custom,
message: `Invalid 'type' property. Expected '${stixType}' for ${objectName} object, but received '${val}'.`
});
return import_zod.z.NEVER;
}
});
}
function createMultiStixTypeValidator(stixTypes) {
const objectNames = stixTypes.map((type) => stixTypeToTypeName[type]).join(" or ");
const typeList = stixTypes.map((t) => `'${t}'`).join(" or ");
return import_zod.z.string().superRefine((val, ctx) => {
if (!stixTypes.includes(val)) {
ctx.addIssue({
code: import_zod.z.ZodIssueCode.custom,
message: `Invalid 'type' property. Expected ${typeList} for ${objectNames} object, but received '${val}'.`
});
return import_zod.z.NEVER;
}
});
}
// src/schemas/common/stix-identifier.ts
var isValidUuid = (uuid) => {
return /^[0-9a-f]{8}-[0-9a-f]{4}-4[0-9a-f]{3}-[89ab][0-9a-f]{3}-[0-9a-f]{12}$/i.test(uuid);
};
var createStixIdError = (id, errorType) => {
const parts = id.split("--");
const stixType = parts.length > 0 ? parts[0] : "";
const typeName = stixType in stixTypeToTypeName ? stixTypeToTypeName[stixType] : "STIX";
let message;
switch (errorType) {
case "format":
message = `Invalid STIX Identifier for ${typeName} object: must comply with format 'type--UUIDv4'`;
break;
case "type":
message = `Invalid STIX Identifier for ${typeName} object: contains invalid STIX type '${stixType}'`;
break;
case "uuid":
message = `Invalid STIX Identifier for ${typeName} object: contains invalid UUIDv4 format`;
break;
}
return {
code: import_zod2.z.ZodIssueCode.custom,
message,
path: ["id"]
};
};
var stixIdentifierSchema = import_zod2.z.string().refine(
(val) => {
if (typeof val !== "string") return false;
if (!val.includes("--")) return false;
const [type, uuid] = val.split("--");
const isValidType = stixTypeSchema.safeParse(type).success;
const isValidUuidValue = isValidUuid(uuid);
return isValidType && isValidUuidValue;
},
(val) => {
if (typeof val !== "string") {
return createStixIdError(String(val), "format");
}
if (!val.includes("--")) {
return createStixIdError(val, "format");
}
const [type, uuid] = val.split("--");
if (!stixTypeSchema.safeParse(type).success) {
return createStixIdError(val, "type");
}
if (!isValidUuid(uuid)) {
return createStixIdError(val, "uuid");
}
return createStixIdError(val, "format");
}
).describe(
"Represents identifiers across the CTI specifications. The format consists of the name of the top-level object being identified, followed by two dashes (--), followed by a UUIDv4."
);
function createStixIdValidator(expectedType) {
const typeName = stixTypeToTypeName[expectedType] || expectedType;
return stixIdentifierSchema.refine(
(val) => val.startsWith(`${expectedType}--`),
{
message: `Invalid STIX Identifier for ${typeName}: must start with '${expectedType}--'`,
path: ["id"]
}
);
}
// src/schemas/common/stix-spec-version.ts
var import_zod3 = require("zod");
var specVersionDescription = [
"The version of the STIX specification used to represent this object.",
"The value of this property MUST be 2.1 for STIX Objects defined according to this specification.",
"If objects are found where this property is not present, the implicit value for all STIX Objects other than SCOs is 2.0.",
"Since SCOs are now top-level objects in STIX 2.1, the default value for SCOs is 2.1."
].join(" ");
var stixSpecVersionSchema = import_zod3.z.enum(["2.0", "2.1"]).describe(specVersionDescription);
// src/schemas/common/stix-timestamp.ts
var import_zod4 = require("zod");
var RFC3339_REGEX = /^(\d{4})-(\d{2})-(\d{2})T(\d{2}):(\d{2}):(\d{2}(?:\.\d+)?)Z$/;
var StixTimestampError = {
InvalidFormat: {
code: import_zod4.z.ZodIssueCode.custom,
message: "Invalid STIX timestamp format: must be an RFC3339 timestamp with a timezone specification of 'Z'."
}
};
var isValidRFC3339 = (timestamp) => {
if (!RFC3339_REGEX.test(timestamp)) return false;
const date = new Date(timestamp);
return !isNaN(date.getTime());
};
var stixTimestampSchema = import_zod4.z.custom(
(val) => {
if (val instanceof Date) {
return isValidRFC3339(val.toISOString());
}
if (typeof val === "string") {
return isValidRFC3339(val);
}
return false;
},
{
message: StixTimestampError.InvalidFormat.message
}
).describe(
"Represents timestamps across the CTI specifications. The format is an RFC3339 timestamp, with a required timezone specification of 'Z'."
);
var stixCreatedTimestampSchema = stixTimestampSchema.brand("StixCreatedTimestamp");
var stixModifiedTimestampSchema = stixTimestampSchema.brand("StixModifiedTimestamp");
// src/schemas/common/misc.ts
var import_zod6 = require("zod");
// src/schemas/common/attack-id.ts
var import_zod5 = require("zod");
var stixTypeToAttackIdMapping = {
"x-mitre-tactic": "tactic",
"attack-pattern": "technique",
// Note: subtechniques are also attack-patterns, but need separate handling
"intrusion-set": "group",
malware: "software",
tool: "software",
"course-of-action": "mitigation",
"x-mitre-asset": "asset",
"x-mitre-data-source": "data-source",
campaign: "campaign"
};
var attackIdPatterns = {
tactic: /^TA\d{4}$/,
technique: /^T\d{4}$/,
subtechnique: /^T\d{4}\.\d{3}$/,
group: /^G\d{4}$/,
software: /^S\d{4}$/,
mitigation: /^M\d{4}$/,
asset: /^A\d{4}$/,
"data-source": /^DS\d{4}$/,
campaign: /^C\d{4}$/
};
// src/schemas/common/misc.ts
var externalReferenceSchema = import_zod6.z.object({
source_name: import_zod6.z.string({
required_error: "Source name is required.",
invalid_type_error: "Source name must be a string."
}),
description: import_zod6.z.string({ invalid_type_error: "Description must be a string." }).optional(),
url: import_zod6.z.string({ invalid_type_error: "URL must be a string." }).url({ message: "Invalid URL format. Please provide a valid URL." }).optional(),
external_id: import_zod6.z.string({ invalid_type_error: "External ID must be a string." }).optional()
});
var externalReferencesSchema = import_zod6.z.array(externalReferenceSchema).min(1, "At least one external reference is required when 'external_references' is defined.").describe("A list of external references which refers to non-STIX information.");
var createAttackExternalReferencesSchema = (stixType) => {
return import_zod6.z.array(externalReferenceSchema).min(1, "At least one external reference is required").refine((refs) => !!refs[0]?.external_id, {
message: "ATT&CK ID must be defined in the first external_references entry.",
path: [0, "external_id"]
}).refine(
(refs) => {
if (!refs[0]?.external_id) return true;
const format = stixTypeToAttackIdMapping[stixType];
return attackIdPatterns[format].test(refs[0].external_id);
},
{
message: `The first external_reference must match the ATT&CK ID format ${getFormatExample(stixType)}.`,
path: [0, "external_id"]
}
).describe("A list of external references with the first containing a valid ATT&CK ID");
};
function getFormatExample(stixType) {
switch (stixType) {
case "x-mitre-tactic":
return "TA####";
case "attack-pattern":
return "T#### or T####.###";
case "intrusion-set":
return "G####";
case "malware":
case "tool":
return "S####";
case "course-of-action":
return "M####";
case "x-mitre-data-source":
return "DS####";
case "x-mitre-asset":
return "A####";
case "campaign":
return "C####";
default:
return "";
}
}
var stixCreatedByRefSchema = createStixIdValidator("identity").describe(
"The created_by_ref property specifies the id property of the identity object that describes the entity that created this object. If this attribute is omitted, the source of this information is undefined. This may be used by object creators who wish to remain anonymous."
);
var granularMarkingSchema = import_zod6.z.object({
marking_ref: stixIdentifierSchema,
selectors: import_zod6.z.array(import_zod6.z.string())
});
var extensionSchema = import_zod6.z.object({
extension_type: import_zod6.z.string(),
extension_properties: import_zod6.z.record(import_zod6.z.unknown())
});
var extensionsSchema = import_zod6.z.record(import_zod6.z.union([extensionSchema, import_zod6.z.record(import_zod6.z.unknown())])).describe("Specifies any extensions of the object, as a dictionary.");
// src/schemas/common/common-properties.ts
var import_zod7 = require("zod");
var versionSchema = import_zod7.z.string().regex(/^\d+\.\d+$/, "Version must be in the format 'major.minor'").default("2.1").describe(
"Represents the version of the object in a 'major.minor' format, where both 'major' and 'minor' are integers. This versioning follows semantic versioning principles but excludes the patch number. The version number is incremented by ATT&CK when the content of the object is updated. This property does not apply to relationship objects."
);
var nameSchema = import_zod7.z.string().min(1, "Name must not be empty").describe("The name of the object.");
var descriptionSchema = import_zod7.z.string().describe("A description of the object.");
var aliasesSchema = import_zod7.z.array(import_zod7.z.string(), {
invalid_type_error: "Aliases must be an array of strings."
}).describe(
"Alternative names used to identify this object. The first alias must match the object's name."
);
var majorMinorVersionRegex = /^(\d{1,3})\.(\d{1,3})$/;
var xMitreVersionSchema = import_zod7.z.custom().refine(
(value) => {
if (!majorMinorVersionRegex.test(value)) return false;
const [major, minor] = value.split(".").map(Number);
return Number.isInteger(major) && Number.isInteger(minor) && major >= 0 && major <= 99 && minor >= 0 && minor <= 99;
},
{
message: "The version must be in the format 'M.N' where M and N are integers between 0 and 99"
}
).describe(
"Represents the version of the object in a 'major.minor' format, where both 'major' and 'minor' are integers between 0 and 99. This versioning follows semantic versioning principles but excludes the patch number. The version number is incremented by ATT&CK when the content of the object is updated. This property does not apply to relationship objects."
);
var majorMinorPatchVersionRegex = /^[0-9]\.[0-9]\.[0-9]$/;
var xMitreAttackSpecVersionSchema = import_zod7.z.string().refine(
(value) => {
if (!majorMinorPatchVersionRegex.test(value)) return false;
const [major, minor, patch] = value.split(".").map(Number);
return major >= 0 && major <= 9 && minor >= 0 && minor <= 9 && patch >= 0 && patch <= 9;
},
{
message: "Must be in the format 'M.N.P' where M, N, and P are single-digit integers (0-9)"
}
).describe(
"The version of the ATT&CK spec used by the object. This field helps consuming software determine if the data format is supported. If the field is not present on an object, the spec version will be assumed to be 2.0.0. Refer to the ATT&CK CHANGELOG for all supported versions."
);
var oldAttackIdRegex = /^MOB-(M|S)\d{4}$/;
function createOldMitreAttackIdSchema(stixType) {
const baseSchema = import_zod7.z.string().describe("Old ATT&CK IDs that may have been associated with this object");
switch (stixType) {
case "malware":
case "tool":
return baseSchema.refine(
(value) => {
return /^MOB-S\d{4}$/.test(value);
},
{
message: `x_mitre_old_attack_id for ${stixType} need to be in the format MOB-S####`
}
);
case "course-of-action":
return baseSchema.refine(
(value) => {
return /^MOB-M\d{4}$/.test(value);
},
{
message: `x_mitre_old_attack_id for ${stixType} need to be in the format MOB-M####`
}
);
default:
throw new Error(`Unsupported STIX type: ${stixType}`);
}
}
var xMitreOldAttackIdSchema = import_zod7.z.string().refine(
(value) => {
return oldAttackIdRegex.test(value);
},
{
message: "Must be in the format 'MOB-X0000' where X is either 'M' or 'S', followed by exactly four digits"
}
).describe("Old ATT&CK IDs that may have been associated with this object");
var attackDomainSchema = import_zod7.z.enum(["enterprise-attack", "mobile-attack", "ics-attack"]);
var xMitreDomainsSchema = import_zod7.z.array(attackDomainSchema).min(1, {
message: "At least one MITRE ATT&CK domain must be specified."
}).describe("The technology domains to which the ATT&CK object belongs.");
var xMitreDeprecatedSchema = import_zod7.z.boolean({
invalid_type_error: "x_mitre_deprecated must be a boolean."
}).describe("Indicates whether the object has been deprecated.");
var supportedMitrePlatforms = [
"Field Controller/RTU/PLC/IED",
"Network Devices",
"Data Historian",
"Google Workspace",
"Office Suite",
"ESXi",
"Identity Provider",
"Containers",
"Azure AD",
"Engineering Workstation",
"Control Server",
"Human-Machine Interface",
"Windows",
"Linux",
"IaaS",
"None",
"iOS",
"PRE",
"SaaS",
"Input/Output Server",
"macOS",
"Android",
"Safety Instrumented System/Protection Relay",
"Embedded"
];
var xMitrePlatformsSchema = import_zod7.z.array(import_zod7.z.enum(supportedMitrePlatforms), {
invalid_type_error: "x_mitre_platforms must be an array of strings.",
message: "x_mitre_platforms may only contain values from the following list: " + supportedMitrePlatforms.join(", ")
}).min(1).refine((items) => new Set(items).size === items.length, {
message: "Platforms must be unique (no duplicates allowed)."
}).describe("List of platforms that apply to the object.");
var objectMarkingRefsSchema = import_zod7.z.array(stixIdentifierSchema).superRefine((val, ctx) => {
val.forEach((identifier, index) => {
if (!identifier.startsWith("marking-definition--")) {
ctx.addIssue({
code: import_zod7.z.ZodIssueCode.custom,
message: `All identifiers must start with 'marking-definition--'. Invalid identifier at index ${index}.`,
path: [index]
});
}
});
}).describe("The list of marking-definition objects to be applied to this object.");
var xMitreContributorsSchema = import_zod7.z.array(import_zod7.z.string()).describe(
"People and organizations who have contributed to the object. Not found on relationship objects."
);
var xMitreIdentity = "identity--c78cb6e5-0c4b-4611-8297-d1b8b55e40b5";
var xMitreModifiedByRefSchema = stixIdentifierSchema.refine((val) => val == xMitreIdentity).describe(
"The STIX ID of the MITRE identity object. Used to track the identity of the MITRE organization, which created the current version of the object. Previous versions of the object may have been created by other individuals or organizations."
);
var killChainNameSchema = import_zod7.z.enum([
"mitre-attack",
"mitre-mobile-attack",
"mitre-ics-attack"
]);
var killChainPhaseSchema = import_zod7.z.object({
phase_name: import_zod7.z.string({
required_error: "Phase name is required.",
invalid_type_error: "Phase name must be a string."
}).describe(
"The name of the phase in the kill chain. The value of this property SHOULD be all lowercase and SHOULD use hyphens instead of spaces or underscores as word separators."
).refine(
(value) => {
const isLowercase = value === value.toLowerCase();
const usesHyphens = !value.includes(" ") && !value.includes("_");
return isLowercase && usesHyphens;
},
{
message: "Phase name should be all lowercase and use hyphens instead of spaces or underscores."
}
),
kill_chain_name: killChainNameSchema
}).strict();
// src/schemas/common/stix-core.ts
var stixBaseObjectSchema = import_zod8.z.object({
id: stixIdentifierSchema.describe(
"The id property universally and uniquely identifies this object."
),
type: stixTypeSchema,
spec_version: stixSpecVersionSchema.describe(
"The version of the STIX specification used to represent this object."
),
created: stixCreatedTimestampSchema.describe(
"The created property represents the time at which the first version of this object was created. The timstamp value MUST be precise to the nearest millisecond."
),
modified: stixModifiedTimestampSchema.describe(
"The modified property represents the time that this particular version of the object was modified. The timstamp value MUST be precise to the nearest millisecond."
),
created_by_ref: stixCreatedByRefSchema.optional(),
labels: import_zod8.z.array(import_zod8.z.string()).describe("The labels property specifies a set of terms used to describe this object.").optional(),
revoked: import_zod8.z.boolean().describe("The revoked property indicates whether the object has been revoked.").optional(),
confidence: import_zod8.z.number().describe("Identifies the confidence that the creator has in the correctness of their data.").int().min(1).max(99).optional().refine((val) => val === void 0 || val > 0 && val < 100, {
message: "Confidence must be between 1 and 99 inclusive."
}).optional(),
lang: import_zod8.z.string().describe("Identifies the language of the text content in this object.").optional(),
external_references: externalReferencesSchema.optional(),
object_marking_refs: objectMarkingRefsSchema.optional(),
granular_markings: import_zod8.z.array(granularMarkingSchema).describe("The set of granular markings that apply to this object.").optional(),
extensions: extensionsSchema.optional()
}).required({
id: true,
type: true,
spec_version: true,
created: true,
modified: true
}).strict();
var stixDomainObjectSchema = stixBaseObjectSchema.extend({});
var stixRelationshipObjectSchema = stixBaseObjectSchema.extend({});
// src/schemas/common/attack-base-object.ts
var attackBaseObjectSchema = stixDomainObjectSchema.extend({
name: nameSchema,
/**
* Required on all ATT&CK schemas except:
* - Marking Definition
*/
x_mitre_attack_spec_version: xMitreAttackSpecVersionSchema,
/**
* Required on all ATT&CK schemas except:
* - Marking Definition
* - Identity
* - Relationship
*/
x_mitre_version: xMitreVersionSchema,
x_mitre_old_attack_id: xMitreOldAttackIdSchema.optional(),
x_mitre_deprecated: xMitreDeprecatedSchema.optional()
});
var attackBaseDomainObject