add authZ to reports

anmalkov · Jan 9, 2024 · b967b34 · b967b34
1 parent 88e3e67
commit b967b34
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/src/Crisp.Core/Helpers/OpenXmlHelper.cs b/src/Crisp.Core/Helpers/OpenXmlHelper.cs
@@ -2,6 +2,7 @@
 using DocumentFormat.OpenXml;
 using DocumentFormat.OpenXml.Packaging;
 using DocumentFormat.OpenXml.Wordprocessing;
+using System.Runtime.InteropServices;
 using System.Text.RegularExpressions;
 using A = DocumentFormat.OpenXml.Drawing;
 using DW = DocumentFormat.OpenXml.Drawing.Wordprocessing;
@@ -55,6 +56,7 @@ public static void AddDataflowAttributes(Stream stream, IEnumerable<DataflowAttr
                     new TableCell(new Paragraph(new Run(new Text(attribute.Transport)))),
                     new TableCell(new Paragraph(new Run(new Text(attribute.DataClassification)))),
                     new TableCell(new Paragraph(new Run(new Text(attribute.Authentication)))),
+                    new TableCell(new Paragraph(new Run(new Text(attribute.Authorization)))),
                     new TableCell(new Paragraph(new Run(new Text(attribute.Notes))))
                 })
             );
@@ -435,6 +437,7 @@ private static IEnumerable<Paragraph> GetParagraphsFromMarkdown(string markdown)
 
     private static Paragraph? FindParagraph(Body body, string text)
     {
+        var p = body.Descendants<Paragraph>();
         return body.Descendants<Paragraph>().Where(p => p.Descendants<Run>().Any(r => r.Descendants<Text>().Any(t => t.Text.ToLower() == text.ToLower()))).FirstOrDefault();
     }
 

diff --git a/src/Crisp.Core/Services/ThreatModelsService.cs b/src/Crisp.Core/Services/ThreatModelsService.cs
@@ -299,7 +299,7 @@ private static string GenerateDataflowAttributeSection(ThreatModel threatModel)
         var section = new StringBuilder();
         foreach (var a in threatModel.DataflowAttributes)
         {
-            section.AppendLine($"| {a.Number.Trim()} | {a.Transport.Trim()} | {a.DataClassification.Trim()} | {a.Authentication.Trim()} | {a.Notes.Trim()} |");
+            section.AppendLine($"| {a.Number.Trim()} | {a.Transport.Trim()} | {a.DataClassification.Trim()} | {a.Authentication.Trim()} | {a.Authorization.Trim()} | {a.Notes.Trim()} |");
         }
         return section.ToString().TrimEnd(Environment.NewLine.ToCharArray());
     }