capability to check for html to avoid shenanigans

pwncollege · Jan 28, 2025 · 88a7beb · 88a7beb
1 parent f508277
commit 88a7beb
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/pwnshop/templates/base/selenium.py b/pwnshop/templates/base/selenium.py
@@ -66,6 +66,11 @@
 
 {% if challenge.print_page %}
 print("Retrieved the following HTML:")
+{% if challenge.ensure_html %}
+if "<html" not in browser.page_source.lower():
+    print("Doesn't look like we retrieved an HTML page...")
+    sys.exit(1)
+{% endif %}
 print(browser.page_source)
 {% endif %}