Sanitize valid html in result text content
This inspects the text content of each individual result div and strips out valid 'script' or 'iframe' tags from the result. Closes #1076main
parent
0d013c788f
commit
c36396e9cb
|
@ -3,6 +3,7 @@ from bs4 import BeautifulSoup
|
||||||
from bs4.element import ResultSet, Tag
|
from bs4.element import ResultSet, Tag
|
||||||
from cryptography.fernet import Fernet
|
from cryptography.fernet import Fernet
|
||||||
from flask import render_template
|
from flask import render_template
|
||||||
|
import html
|
||||||
import urllib.parse as urlparse
|
import urllib.parse as urlparse
|
||||||
from urllib.parse import parse_qs
|
from urllib.parse import parse_qs
|
||||||
import re
|
import re
|
||||||
|
@ -160,6 +161,9 @@ class Filter:
|
||||||
self.update_styling()
|
self.update_styling()
|
||||||
self.remove_block_tabs()
|
self.remove_block_tabs()
|
||||||
|
|
||||||
|
for div in self.main_divs:
|
||||||
|
self.sanitize_div(div)
|
||||||
|
|
||||||
for img in [_ for _ in self.soup.find_all('img') if 'src' in _.attrs]:
|
for img in [_ for _ in self.soup.find_all('img') if 'src' in _.attrs]:
|
||||||
self.update_element_src(img, 'image/png')
|
self.update_element_src(img, 'image/png')
|
||||||
|
|
||||||
|
@ -197,6 +201,34 @@ class Filter:
|
||||||
self.remove_site_blocks(self.soup)
|
self.remove_site_blocks(self.soup)
|
||||||
return self.soup
|
return self.soup
|
||||||
|
|
||||||
|
def sanitize_div(self, div) -> None:
|
||||||
|
"""Removes escaped script and iframe tags from results
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
None (The soup object is modified directly)
|
||||||
|
"""
|
||||||
|
if not div:
|
||||||
|
return
|
||||||
|
|
||||||
|
for d in div.find_all('div', recursive=True):
|
||||||
|
d_text = d.find(text=True, recursive=False)
|
||||||
|
|
||||||
|
# Ensure we're working with tags that contain text content
|
||||||
|
if not d_text or not d.string:
|
||||||
|
continue
|
||||||
|
|
||||||
|
d.string = html.unescape(d_text)
|
||||||
|
div_soup = BeautifulSoup(d.string, 'html.parser')
|
||||||
|
|
||||||
|
# Remove all valid script or iframe tags in the div
|
||||||
|
for script in div_soup.find_all('script'):
|
||||||
|
script.decompose()
|
||||||
|
|
||||||
|
for iframe in div_soup.find_all('iframe'):
|
||||||
|
iframe.decompose()
|
||||||
|
|
||||||
|
d.string = str(div_soup)
|
||||||
|
|
||||||
def remove_site_blocks(self, soup) -> None:
|
def remove_site_blocks(self, soup) -> None:
|
||||||
if not self.config.block or not soup.body:
|
if not self.config.block or not soup.body:
|
||||||
return
|
return
|
||||||
|
|
Loading…
Reference in New Issue