small steps at a time

pdf_generator/templates/new_template/main.tex

@@ -40,10 +40,11 @@
 %----------------------------------------------------------------------------------------
 
 \aboutme{
-PKCS8 Private Key \\
-SSL Certificate \\
-SSH-RSA Private Key Block \\
-Generic Public Key
+\BLOCK{if 'crypto_material' in analysis}
+	\BLOCK{for selected_summary in analysis['crypto_material']['summary']}
+		\VAR{selected_summary | filter_chars} \\
+	\BLOCK{endfor}
+\BLOCK{endif}
 } % To have no About Me section, just remove all the text and leave \aboutme{}
 
 %----------------------------------------------------------------------------------------
@@ -51,8 +52,12 @@ Generic Public Key
 %----------------------------------------------------------------------------------------
 
 % Skill bar section, each skill must have a value between 0 an 6 (float)
-\skills{{NX/5.96},{Canary/0.01},{PIE/3.82},{RELRO/0.13}}
-
+\skills{
+    \BLOCK{for selected_summary in analysis['exploit_mitigations']['summary']}
+        {\VAR{ selected_summary | filter_chars}/6},
+    \BLOCK{endfor}
+    {}
+}
 %------------------------------------------------
 
 % Skill text section, each skill must have a value between 0 an 6
@@ -68,10 +73,12 @@ Generic Public Key
 %----------------------------------------------------------------------------------------
 
 \section{Binwalk}
+\BLOCK{if analysis['entropy_analysis_graph']}
 
-\subsection{Entropy Graph}
-
-\includegraphics[width = \textwidth]{/tmp/interface/data/entropy_analysis_graph.png}
+	\subsection{Entropy Graph}
+	%\includegraphics[width = \textwidth]{/tmp/interface/data/entropy_analysis_graph.png}
+	\includegraphics[width = \textwidth]{\VAR{analysis['entropy_analysis_graph'] | base64_to_png('entropy_analysis_graph', tmp_dir)}}
+\BLOCK{endif}
 
 %\begin{twenty} % Environment for a list with descriptions
 %	\twentyitem{since 1865}{Ph.D. {\normalfont candidate in Computer Science}}{Wonderland}{\emph{A Quantified Theory of Social Cohesion.}}

pdf_generator/tex_generation/template_engine.py

@@ -13,7 +13,7 @@ MAIN_TEMPLATE = 'main.tex'
 META_TEMPLATE = 'meta.tex'
 CUSTOM_TEMPLATE_CLASS = 'twentysecondcv.cls'
 PLUGIN_TEMPLATE_BLUEPRINT = '{}.tex'
-LOGO_FILE = '/new_template/fact.png'
+LOGO_FILE = 'fact.png'
 
 
 def render_number_as_size(number, verbose=True):
@@ -45,6 +45,7 @@ def replace_special_characters(data):
     latex_character_escapes = OrderedDict()
     latex_character_escapes['\\'] = ''
     latex_character_escapes['\''] = ''
+    latex_character_escapes['/'] = ' '
     latex_character_escapes['$'] = '\\$'
     latex_character_escapes['('] = '$($'
     latex_character_escapes[')'] = '$)$'
@@ -101,6 +102,21 @@ def item_contains_string(item, string):
     return string in item
 
 
+def count_mitigations(exploit_mitigations):
+    count = 0
+    if 'Canary' in exploit_mitigations['summary']:
+        for selected_summary in exploit_mitigations:
+            if 'Canary' in selected_summary:
+                count += len(selected_summary)
+    elif 'NX' in exploit_mitigations['summary']:
+        pass
+    elif 'RELRO' in exploit_mitigations['summary']:
+        pass
+    elif 'PIE' in exploit_mitigations['summary']:
+        pass
+    return count
+
+
 def create_jinja_environment(templates_to_use='default'):
     template_directory = Path(Path(__file__).parent.parent, 'templates', templates_to_use)
     environment = jinja2.Environment(