Merge pull request #705 from CybercentreCanada/update/class-access

Update/class access
CybercentreCanada · Apr 10, 2024 · ce92aab · ce92aab
2 parents 46fb77f + 6eeb3ae
commit ce92aab
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/jsjaws.py b/jsjaws.py
@@ -2224,7 +2224,7 @@ def _set_element_attribute_script(attr_id: str, attr_val: str, random_element_va
         if '"' in attr_val:
             attr_val = attr_val.replace('"', '\\"')
 
-        if "\\" in attr_val:
+        if "\\" in attr_val and "url(" not in attr_val:
             attr_val = attr_val.replace("\\", "\\\\")
 
         return f'{random_element_varname}.setAttribute("{attr_id}", "{attr_val}");\n'

diff --git a/tools/malwarejail/env/web/browser.js b/tools/malwarejail/env/web/browser.js
@@ -143,6 +143,11 @@ $ = function (thing) {
         let attribute = split_selector[1];
         let element = document.getElementById(id);
         return element;
+    } else if (typeof thing === "string" && thing.indexOf(".") === 0) {
+        // This is a getter for the class!
+        let class_name = thing.replace(".", "");
+        let element = document.getElementById(class_name);
+        return element;
     }
     return document.getElementById(thing);
 };