fix up threat type in threats by context

huaijinli · Nov 10, 2021 · 5106b37 · 5106b37
1 parent 42fa3a4
commit 5106b37
Showing 1 changed file with 53 additions and 0 deletions.
diff --git a/td.desktop/core/services/threatengine.js b/td.desktop/core/services/threatengine.js
@@ -44,6 +44,59 @@ function threatengine() {
     return results.events.map(function(result) {
       /* fix up for STRIDE / CIA / LINDDUN */
       result.params.modelType = modelType;
+      switch (result.params.type) {
+        case 'Linkability':
+          if (modelType == 'STRIDE') {
+            result.params.type = 'Information disclosure';
+          } else if (modelType == 'CIA') {
+            result.params.type = 'Confidentiality';
+          }
+          break;
+        case 'Identifiability':
+          if (modelType == 'STRIDE') {
+            result.params.type = 'Spoofing';
+          } else if (modelType == 'CIA') {
+            result.params.type = 'Confidentiality';
+          }
+          break;
+        case 'Repudiation':
+          if (modelType == 'LINDDUN') {
+              result.params.type = 'Non-repudiation';
+          } else if (modelType == 'CIA') {
+            result.params.type = 'Integrity';
+          }
+          break;
+        case 'Detectability':
+          if (modelType == 'STRIDE') {
+            result.params.type = 'Information disclosure';
+          } else if (modelType == 'CIA') {
+            result.params.type = 'Confidentiality';
+          }
+          break;
+        case 'Disclosure':
+          if (modelType == 'LINDDUN') {
+              result.params.type = 'Disclosure of information';
+          } else if (modelType == 'STRIDE') {
+              result.params.type = 'Information disclosure';
+          } else if (modelType == 'CIA') {
+            result.params.type = 'Confidentiality';
+          }
+          break;
+        case 'Unawareness':
+          if (modelType == 'STRIDE') {
+            result.params.type = 'Elevation of privilege';
+          } else if (modelType == 'CIA') {
+            result.params.type = 'Integrity';
+          }
+          break;
+        case 'Tampering':
+          if (modelType == 'LINDDUN') {
+            result.params.type = 'Non-compliance';
+          } else if (modelType == 'CIA') {
+            result.params.type = 'Integrity';
+          }
+          break;
+      }
       return result.params;
     });
   }