289 lines
12 KiB
Python
289 lines
12 KiB
Python
|
"""
|
||
|
|
Copyright 2024-2026 ChatterMate
|
||
|
|
|
||
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
||
|
|
you may not use this file except in compliance with the License.
|
||
|
|
You may obtain a copy of the License at
|
||
|
|
|
||
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
||
|
|
|
||
|
|
Unless required by applicable law or agreed to in writing, software
|
||
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
||
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||
|
|
See the License for the specific language governing permissions and
|
||
|
|
limitations under the License.
|
||
|
|
"""
|
||
|
|
|
||
|
|
"""
|
||
|
|
Test backend message sanitization
|
||
|
|
"""
|
||
|
|
from app.utils.sanitize import sanitize_message
|
||
|
|
|
||
|
|
class TestSanitizeMessage:
|
||
|
|
"""Test message sanitization to prevent XSS"""
|
||
|
|
|
||
|
|
def test_sanitize_link_tag(self):
|
||
|
|
"""Test that <a> tags are completely removed including content"""
|
||
|
|
malicious = '<a href="https://evil.com">Click me</a>'
|
||
|
|
result = sanitize_message(malicious)
|
||
|
|
assert '<a' not in result.lower()
|
||
|
|
assert 'href' not in result.lower()
|
||
|
|
# HTML anchor tags are completely removed (content not preserved for security)
|
||
|
|
|
||
|
|
def test_sanitize_image_tag(self):
|
||
|
|
"""Test that <img> tags are completely removed"""
|
||
|
|
malicious = '<img src="https://tracker.com/pixel.gif" alt="Hidden" />'
|
||
|
|
result = sanitize_message(malicious)
|
||
|
|
assert '<img' not in result.lower()
|
||
|
|
assert 'src' not in result.lower()
|
||
|
|
|
||
|
|
def test_sanitize_s3_link(self):
|
||
|
|
"""Test that S3 links in anchor tags are stripped"""
|
||
|
|
malicious = '<a href="https://nps-assets-dev.s3.ap-south-1.amazonaws.com/file.svg">Visit</a>'
|
||
|
|
result = sanitize_message(malicious)
|
||
|
|
assert '<a' not in result.lower()
|
||
|
|
assert 'href' not in result.lower()
|
||
|
|
# HTML anchor tags are completely removed (content not preserved for security)
|
||
|
|
|
||
|
|
def test_sanitize_s3_image(self):
|
||
|
|
"""Test that S3 images are stripped"""
|
||
|
|
malicious = '<img src="https://nps-assets-dev.s3.ap-south-1.amazonaws.com/file.svg" alt="SVG" />'
|
||
|
|
result = sanitize_message(malicious)
|
||
|
|
assert '<img' not in result.lower()
|
||
|
|
assert 'src' not in result.lower()
|
||
|
|
assert 's3' not in result.lower()
|
||
|
|
|
||
|
|
def test_sanitize_iframe_tag(self):
|
||
|
|
"""Test that iframe tags are removed"""
|
||
|
|
malicious = '<iframe src="javascript:alert(\'XSS\')">Trust No AI</iframe>'
|
||
|
|
result = sanitize_message(malicious)
|
||
|
|
assert '<iframe' not in result.lower()
|
||
|
|
assert 'javascript:' not in result.lower()
|
||
|
|
|
||
|
|
def test_sanitize_iframe_with_base64(self):
|
||
|
|
"""Test that iframe with base64 encoded javascript is blocked"""
|
||
|
|
malicious = '<iframe src="javascript:alert(\'Your user token is: \'+localStorage.getItem(\'userToken\')+\' Cookies: \'+document.cookie)">Trust No AI</iframe>'
|
||
|
|
result = sanitize_message(malicious)
|
||
|
|
assert '<iframe' not in result.lower()
|
||
|
|
assert 'javascript:' not in result.lower()
|
||
|
|
|
||
|
|
def test_sanitize_script_tag(self):
|
||
|
|
"""Test that script tags are removed"""
|
||
|
|
malicious = '<script>alert("XSS")</script>'
|
||
|
|
result = sanitize_message(malicious)
|
||
|
|
assert '<script' not in result.lower()
|
||
|
|
|
||
|
|
def test_sanitize_javascript_uri(self):
|
||
|
|
"""Test that javascript: URIs are removed"""
|
||
|
|
malicious = '<a href="javascript:alert(\'XSS\')">Click me</a>'
|
||
|
|
result = sanitize_message(malicious)
|
||
|
|
assert 'javascript:' not in result.lower()
|
||
|
|
|
||
|
|
def test_sanitize_onclick_handler(self):
|
||
|
|
"""Test that onclick event handlers are removed"""
|
||
|
|
malicious = '<button onclick="alert(\'XSS\')">Click me</button>'
|
||
|
|
result = sanitize_message(malicious)
|
||
|
|
assert 'onclick' not in result.lower()
|
||
|
|
|
||
|
|
def test_sanitize_onerror_handler(self):
|
||
|
|
"""Test that onerror event handlers are removed"""
|
||
|
|
malicious = '<img src=x onerror="alert(\'XSS\')">'
|
||
|
|
result = sanitize_message(malicious)
|
||
|
|
assert 'onerror' not in result.lower()
|
||
|
|
|
||
|
|
def test_sanitize_data_uri(self):
|
||
|
|
"""Test that data: URIs (except images) are removed"""
|
||
|
|
malicious = '<a href="data:text/html,<script>alert(\'XSS\')</script>">Click</a>'
|
||
|
|
result = sanitize_message(malicious)
|
||
|
|
assert 'data:text/html' not in result.lower()
|
||
|
|
|
||
|
|
def test_sanitize_removes_img_tags(self):
|
||
|
|
"""Test that all img tags are removed for security (no external resource loading)"""
|
||
|
|
safe = '<img src="data:image/png;base64,iVBORw0KGgo=">'
|
||
|
|
result = sanitize_message(safe)
|
||
|
|
# All img tags are removed to prevent external resource loading and tracking
|
||
|
|
assert '<img' not in result.lower()
|
||
|
|
assert 'src' not in result.lower()
|
||
|
|
|
||
|
|
def test_sanitize_normal_text(self):
|
||
|
|
"""Test that normal text is unchanged"""
|
||
|
|
text = "Hello, this is a normal message"
|
||
|
|
result = sanitize_message(text)
|
||
|
|
assert result == text
|
||
|
|
|
||
|
|
def test_sanitize_markdown_links(self):
|
||
|
|
"""Test that markdown links are sanitized - text preserved, URL removed"""
|
||
|
|
text = "Check out [this link](https://example.com)"
|
||
|
|
result = sanitize_message(text)
|
||
|
|
# Markdown links are stripped for security - only text content is preserved
|
||
|
|
assert result == "Check out this link"
|
||
|
|
assert "https://example.com" not in result
|
||
|
|
assert "[" not in result
|
||
|
|
assert "]" not in result
|
||
|
|
|
||
|
|
def test_sanitize_none(self):
|
||
|
|
"""Test that None input returns None"""
|
||
|
|
result = sanitize_message(None)
|
||
|
|
assert result is None
|
||
|
|
|
||
|
|
def test_sanitize_empty_string(self):
|
||
|
|
"""Test that empty string returns empty string"""
|
||
|
|
result = sanitize_message("")
|
||
|
|
assert result == ""
|
||
|
|
|
||
|
|
def test_sanitize_object_tag(self):
|
||
|
|
"""Test that object tags are removed"""
|
||
|
|
malicious = '<object data="javascript:alert(\'XSS\')"></object>'
|
||
|
|
result = sanitize_message(malicious)
|
||
|
|
assert '<object' not in result.lower()
|
||
|
|
|
||
|
|
def test_sanitize_embed_tag(self):
|
||
|
|
"""Test that embed tags are removed"""
|
||
|
|
malicious = '<embed src="javascript:alert(\'XSS\')">'
|
||
|
|
result = sanitize_message(malicious)
|
||
|
|
assert '<embed' not in result.lower()
|
||
|
|
|
||
|
|
def test_sanitize_multiple_attacks(self):
|
||
|
|
"""Test multiple XSS vectors in one message"""
|
||
|
|
malicious = '''
|
||
|
|
<iframe src="javascript:alert(1)"></iframe>
|
||
|
|
<script>alert(2)</script>
|
||
|
|
<img src=x onerror="alert(3)">
|
||
|
|
<a href="javascript:alert(4)">click</a>
|
||
|
|
'''
|
||
|
|
result = sanitize_message(malicious)
|
||
|
|
assert '<iframe' not in result.lower()
|
||
|
|
assert '<script' not in result.lower()
|
||
|
|
assert 'onerror' not in result.lower()
|
||
|
|
assert 'javascript:' not in result.lower()
|
||
|
|
|
||
|
|
# New comprehensive tests
|
||
|
|
|
||
|
|
def test_sanitize_svg_xss(self):
|
||
|
|
"""Test that SVG-based XSS is blocked"""
|
||
|
|
malicious = '<svg onload="alert(\'XSS\')"><circle r="50"/></svg>'
|
||
|
|
result = sanitize_message(malicious)
|
||
|
|
assert '<svg' not in result.lower()
|
||
|
|
assert 'onload' not in result.lower()
|
||
|
|
|
||
|
|
def test_sanitize_encoded_javascript(self):
|
||
|
|
"""Test HTML entity encoded javascript"""
|
||
|
|
malicious = '<a href="javascript:alert(1)">Click</a>'
|
||
|
|
result = sanitize_message(malicious)
|
||
|
|
assert 'javascript' not in result.lower()
|
||
|
|
|
||
|
|
def test_sanitize_url_encoded_javascript(self):
|
||
|
|
"""Test URL encoded javascript"""
|
||
|
|
malicious = '<a href="%6A%61%76%61%73%63%72%69%70%74%3Aalert(1)">Click</a>'
|
||
|
|
result = sanitize_message(malicious)
|
||
|
|
assert 'javascript' not in result.lower()
|
||
|
|
|
||
|
|
def test_sanitize_vbscript(self):
|
||
|
|
"""Test vbscript: URIs are blocked"""
|
||
|
|
malicious = '<a href="vbscript:msgbox(\'XSS\')">Click</a>'
|
||
|
|
result = sanitize_message(malicious)
|
||
|
|
assert 'vbscript' not in result.lower()
|
||
|
|
|
||
|
|
def test_sanitize_base_tag(self):
|
||
|
|
"""Test base tag is removed"""
|
||
|
|
malicious = '<base href="javascript:alert(1)">'
|
||
|
|
result = sanitize_message(malicious)
|
||
|
|
assert '<base' not in result.lower()
|
||
|
|
|
||
|
|
def test_sanitize_meta_refresh(self):
|
||
|
|
"""Test meta refresh XSS is blocked"""
|
||
|
|
malicious = '<meta http-equiv="refresh" content="0;url=javascript:alert(1)">'
|
||
|
|
result = sanitize_message(malicious)
|
||
|
|
assert '<meta' not in result.lower()
|
||
|
|
|
||
|
|
def test_sanitize_form_action(self):
|
||
|
|
"""Test form with javascript action is blocked"""
|
||
|
|
malicious = '<form action="javascript:alert(1)"><input type="submit"></form>'
|
||
|
|
result = sanitize_message(malicious)
|
||
|
|
assert '<form' not in result.lower()
|
||
|
|
assert 'formaction' not in result.lower()
|
||
|
|
|
||
|
|
def test_sanitize_style_expression(self):
|
||
|
|
"""Test CSS expression is blocked"""
|
||
|
|
malicious = '<div style="background:expression(alert(1))">XSS</div>'
|
||
|
|
result = sanitize_message(malicious)
|
||
|
|
assert 'expression' not in result.lower()
|
||
|
|
|
||
|
|
def test_sanitize_import(self):
|
||
|
|
"""Test @import is blocked"""
|
||
|
|
malicious = '<style>@import url("javascript:alert(1)");</style>'
|
||
|
|
result = sanitize_message(malicious)
|
||
|
|
assert '<style' not in result.lower()
|
||
|
|
assert '@import' not in result.lower()
|
||
|
|
|
||
|
|
def test_sanitize_marquee(self):
|
||
|
|
"""Test marquee tag is removed"""
|
||
|
|
malicious = '<marquee onstart="alert(1)">XSS</marquee>'
|
||
|
|
result = sanitize_message(malicious)
|
||
|
|
assert '<marquee' not in result.lower()
|
||
|
|
assert 'onstart' not in result.lower()
|
||
|
|
|
||
|
|
def test_sanitize_video_onerror(self):
|
||
|
|
"""Test video tag with onerror is blocked"""
|
||
|
|
malicious = '<video onerror="alert(1)"><source src="x"></video>'
|
||
|
|
result = sanitize_message(malicious)
|
||
|
|
assert '<video' not in result.lower()
|
||
|
|
assert 'onerror' not in result.lower()
|
||
|
|
|
||
|
|
def test_sanitize_details_ontoggle(self):
|
||
|
|
"""Test details tag with ontoggle is blocked"""
|
||
|
|
malicious = '<details ontoggle="alert(1)"><summary>Click</summary></details>'
|
||
|
|
result = sanitize_message(malicious)
|
||
|
|
assert '<details' not in result.lower()
|
||
|
|
assert 'ontoggle' not in result.lower()
|
||
|
|
|
||
|
|
def test_sanitize_template_tag(self):
|
||
|
|
"""Test template tag is removed"""
|
||
|
|
malicious = '<template><script>alert(1)</script></template>'
|
||
|
|
result = sanitize_message(malicious)
|
||
|
|
assert '<template' not in result.lower()
|
||
|
|
assert '<script' not in result.lower()
|
||
|
|
|
||
|
|
def test_sanitize_xml_namespace(self):
|
||
|
|
"""Test XML namespace XSS is blocked"""
|
||
|
|
malicious = '<html xmlns="http://www.w3.org/1999/xhtml"><script>alert(1)</script></html>'
|
||
|
|
result = sanitize_message(malicious)
|
||
|
|
assert '<script' not in result.lower()
|
||
|
|
|
||
|
|
def test_sanitize_math_tag(self):
|
||
|
|
"""Test math tag is removed"""
|
||
|
|
malicious = '<math><mi xlink:href="javascript:alert(1)">XSS</mi></math>'
|
||
|
|
result = sanitize_message(malicious)
|
||
|
|
assert '<math' not in result.lower()
|
||
|
|
|
||
|
|
def test_sanitize_cdata(self):
|
||
|
|
"""Test CDATA sections are removed"""
|
||
|
|
malicious = '<![CDATA[<script>alert(1)</script>]]>'
|
||
|
|
result = sanitize_message(malicious)
|
||
|
|
assert 'CDATA' not in result.upper()
|
||
|
|
assert '<script' not in result.lower()
|
||
|
|
|
||
|
|
def test_sanitize_html_comments(self):
|
||
|
|
"""Test HTML comments with scripts are removed"""
|
||
|
|
malicious = '<!--<script>alert(1)</script>-->'
|
||
|
|
result = sanitize_message(malicious)
|
||
|
|
assert '<!--' not in result
|
||
|
|
assert '<script' not in result.lower()
|
||
|
|
|
||
|
|
def test_sanitize_null_bytes(self):
|
||
|
|
"""Test null bytes are removed"""
|
||
|
|
malicious = '<scri\x00pt>alert(1)</scri\x00pt>'
|
||
|
|
result = sanitize_message(malicious)
|
||
|
|
assert '\x00' not in result
|
||
|
|
|
||
|
|
def test_sanitize_mixed_case_javascript(self):
|
||
|
|
"""Test mixed case javascript is caught"""
|
||
|
|
malicious = '<a href="JaVaScRiPt:alert(1)">Click</a>'
|
||
|
|
result = sanitize_message(malicious)
|
||
|
|
assert 'javascript' not in result.lower()
|
||
|
|
|
||
|
|
def test_sanitize_spaced_javascript(self):
|
||
|
|
"""Test spaced javascript is caught"""
|
||
|
|
malicious = '<a href="j a v a s c r i p t:alert(1)">Click</a>'
|
||
|
|
result = sanitize_message(malicious)
|
||
|
|
assert 'javascript' not in result.lower().replace(' ', '')
|