fixed payload detection; text_prep changes

eyaler · Aug 10, 2022 · 0bc8b4d · 0bc8b4d
1 parent 23345f7
commit 0bc8b4d
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/example.py b/example.py
@@ -8,7 +8,7 @@
 
 books = 30123, 2600
 mtf_variants = 0, 80
-output_folder = 'output'
+output_folder = 'output_oct08'
 
 
 for book, mtf in zip(books, mtf_variants):

diff --git a/ztml/validation.py b/ztml/validation.py
@@ -182,7 +182,9 @@ def validate_files(filenames: Mapping[str, str],
                 if ext == 'html' and label != 'base64_html':
                     with open(filename, 'rb') as f:
                         script = f.read()
-                        script = script.replace(max(regex.finditer(webify.get_literals_regex(payload_var).encode(), script), key=lambda m: len(m[0]), default=b'')[0].split(b'`')[1], b'')
+                        if label == 'crenc_html':
+                            print('hi')
+                        script = script.replace(max(regex.finditer(webify.get_literals_regex(payload_var).encode(), script), key=lambda m: len(m[0]), default=b'')[0].split(b'`', 1)[1].rsplit(b'`', 1)[0], b'')
                     stats.append(f'code: {len(script):,} B = {round(len(script) / 1024, 1):,} kB')
                 stats = ' '.join(stats)
                 if stats: