mirror of https://github.com/CIRCL/lookyloo
271 lines
14 KiB
Python
271 lines
14 KiB
Python
#!/usr/bin/env python3
|
|
# -*- coding: utf-8 -*-
|
|
|
|
from typing import Dict, Any, Optional, List, Union
|
|
from datetime import date
|
|
import hashlib
|
|
import json
|
|
from pathlib import Path
|
|
import time
|
|
|
|
|
|
from .helpers import get_homedir
|
|
from .exceptions import ConfigError
|
|
|
|
import vt # type: ignore
|
|
from pysanejs import SaneJS
|
|
from pyeupi import PyEUPI
|
|
|
|
|
|
class SaneJavaScript():
|
|
|
|
skip_lookup: Dict[str, str] = {
|
|
"717ea0ff7f3f624c268eccb244e24ec1305ab21557abb3d6f1a7e183ff68a2d28f13d1d2af926c9ef6d1fb16dd8cbe34cd98cacf79091dddc7874dcee21ecfdc": "This is a 1*1 pixel GIF",
|
|
"e508d5d17e94d14b126164082342a9ca4774f404e87a3dd56c26812493ee18d9c3d6daacca979134a94a003066aca24116de874596d00d1e52130c1283d54209": "This is a 1*1 pixel GIF",
|
|
"2d073e10ae40fde434eb31cbedd581a35cd763e51fb7048b88caa5f949b1e6105e37a228c235bc8976e8db58ed22149cfccf83b40ce93a28390566a28975744a": "This is a 1*1 pixel GIF",
|
|
"84e24a70b78e9de9c9d0dfeb49f3f4247dbc1c715d8844471ee40669270682e199d48f5fbec62bd984c9c0270534b407c4d2561dd6c05adec3c83c1534f32d5c": "This is a 1*1 pixel GIF",
|
|
"d5da26b5d496edb0221df1a4057a8b0285d15592a8f8dc7016a294df37ed335f3fde6a2252962e0df38b62847f8b771463a0124ef3f84299f262ed9d9d3cee4c": "This is a 1*1 pixel GIF",
|
|
"f7a5f748f4c0d3096a3ca972886fe9a9dff5dce7792779ec6ffc42fa880b3815e2e4c3bdea452352f3844b81864c9bfb7861f66ac961cfa66cb9cb4febe568e8": "This is a 1*1 pixel GIF",
|
|
"b2ca25a3311dc42942e046eb1a27038b71d689925b7d6b3ebb4d7cd2c7b9a0c7de3d10175790ac060dc3f8acf3c1708c336626be06879097f4d0ecaa7f567041": "This is a 1*1 pixel GIF",
|
|
"b8d82d64ec656c63570b82215564929adad167e61643fd72283b94f3e448ef8ab0ad42202f3537a0da89960bbdc69498608fc6ec89502c6c338b6226c8bf5e14": "This is a 1*1 pixel GIF",
|
|
"2991c3aa1ba61a62c1cccd990c0679a1fb8dccd547d153ec0920b91a75ba20820de1d1c206f66d083bf2585d35050f0a39cd7a3e11c03882dafec907d27a0180": "This is a 1*1 pixel GIF",
|
|
"b1a6cfa7b21dbb0b281d241af609f3ba7f3a63e5668095bba912bf7cfd7f0320baf7c3b0bfabd0f8609448f39902baeb145ba7a2d8177fe22a6fcea03dd29be1": "This is a 1*1 pixel GIF",
|
|
"ebfe0c0df4bcc167d5cb6ebdd379f9083df62bef63a23818e1c6adf0f64b65467ea58b7cd4d03cf0a1b1a2b07fb7b969bf35f25f1f8538cc65cf3eebdf8a0910": "This is a 1*1 pixel GIF",
|
|
"1d68b92e8d822fe82dc7563edd7b37f3418a02a89f1a9f0454cca664c2fc2565235e0d85540ff9be0b20175be3f5b7b4eae1175067465d5cca13486aab4c582c": "This is a 1*1 pixel GIF",
|
|
"ac44da7f455bfae52b883639964276026fb259320902aa813d0333e021c356a7b3e3537b297f9a2158e588c302987ce0854866c039d1bb0ffb27f67560739db2": "This is a 1*1 pixel GIF",
|
|
"921944dc10fbfb6224d69f0b3ac050f4790310fd1bcac3b87c96512ad5ed9a268824f3f5180563d372642071b4704c979d209baf40bc0b1c9a714769aba7dfc7": "This is a 1*1 pixel GIF",
|
|
"89dfc38ec77cf258362e4db7c8203cae8a02c0fe4f99265b0539ec4f810c84f8451e22c9bef1ebc59b4089af7e93e378e053c542a5967ec4912d4c1fc5de22f0": "This is a 1*1 pixel GIF",
|
|
"280ea4383ee6b37051d91c5af30a5ce72aa4439340fc6d31a4fbe7ba8a8156eb7893891d5b2371b9fc4934a78f08de3d57e5b63fa9d279a317dcbefb8a07a6b0": "This is a 1*1 pixel GIF",
|
|
"3844065e1dd778a05e8cc39901fbf3191ded380d594359df137901ec56ca52e03d57eb60acc2421a0ee74f0733bbb5d781b7744685c26fb013a236f49b02fed3": "This is a 1*1 pixel GIF",
|
|
"bd9ab35dde3a5242b04c159187732e13b0a6da50ddcff7015dfb78cdd68743e191eaf5cddedd49bef7d2d5a642c217272a40e5ba603fe24ca676a53f8c417c5d": "This is a 1*1 pixel GIF",
|
|
"d052ecec2839340876eb57247cfc2e777dd7f2e868dc37cd3f3f740c8deb94917a0c9f2a4fc8229987a0b91b04726de2d1e9f6bcbe3f9bef0e4b7e0d7f65ea12": "This is a 1*1 pixel GIF",
|
|
"8717074ddf1198d27b9918132a550cb4ba343794cc3d304a793f9d78c9ff6c4929927b414141d40b6f6ad296725520f4c63edeb660ed530267766c2ab74ee4a9": "This is a 1*1 pixel GIF",
|
|
"6834f1548f26b94357fcc3312a3491e8c87080a84f678f990beb2c745899a01e239964521e64a534d7d5554222f728af966ec6ec8291bc64d2005861bcfd78ec": "This is a 1*1 pixel GIF",
|
|
"3be8176915593e79bc280d08984a16c29c495bc53be9b439276094b8dcd3764a3c72a046106a06b958e08e67451fe02743175c621a1faa261fe7a9691cc77141": "This is a 1*1 pixel GIF",
|
|
"826225fc21717d8861a05b9d2f959539aad2d2b131b2afed75d88fbca535e1b0d5a0da8ac69713a0876a0d467848a37a0a7f926aeafad8cf28201382d16466ab": "This is a 1*1 pixel GIF",
|
|
"202612457d9042fe853daab3ddcc1f0f960c5ffdbe8462fa435713e4d1d85ff0c3f197daf8dba15bda9f5266d7e1f9ecaeee045cbc156a4892d2f931fe6fa1bb": "This is a 1*1 pixel GIF",
|
|
"b82c6aa1ae927ade5fadbbab478cfaef26d21c1ac441f48e69cfc04cdb779b1e46d7668b4368b933213276068e52f9060228907720492a70fd9bc897191ee77c": "This is a 1*1 pixel GIF",
|
|
"763de1053a56a94eef4f72044adb2aa370b98ffa6e0add0b1cead7ee27da519e223921c681ae1db3311273f45d0dd3dc022d102d42ce210c90cb3e761b178438": "This is a 1*1 pixel GIF",
|
|
"69e2da5cdc318fc237eaa243b6ea7ecc83b68dbdea8478dc69154abdda86ecb4e16c35891cc1facb3ce7e0cf19d5abf189c50f59c769777706f4558f6442abbc": "This is a 1*1 pixel GIF",
|
|
"16dd1560fdd43c3eee7bcf622d940be93e7e74dee90286da37992d69cea844130911b97f41c71f8287b54f00bd3a388191112f490470cf27c374d524f49ba516": "This is a 1*1 pixel GIF",
|
|
"01211111688dc2007519ff56603fbe345d057337b911c829aaee97b8d02e7d885e7a2c2d51730f54a04aebc1821897c8041f15e216f1c973ed313087fa91a3fb": "This is a 1*1 pixel GIF",
|
|
"71db01662075fac031dea18b2c766826c77dbab01400a8642cdc7059394841d5df9020076554c3beca6f808187d42e1a1acc98fad9a0e1ad32ae869145f53746": "This is a 1*1 pixel GIF",
|
|
"49b8daf1f5ba868bc8c6b224c787a75025ca36513ef8633d1d8f34e48ee0b578f466fcc104a7bed553404ddc5f9faff3fef5f894b31cd57f32245e550fad656a": "This is a 1*1 pixel GIF",
|
|
"c57ebbadcf59f982ba28da35fdbd5e5369a8500a2e1edad0dc9c9174de6fd99f437953732e545b95d3de5943c61077b6b949c989f49553ff2e483f68fcc30641": "This is a 1*1 pixel GIF",
|
|
"c87bf81fd70cf6434ca3a6c05ad6e9bd3f1d96f77dddad8d45ee043b126b2cb07a5cf23b4137b9d8462cd8a9adf2b463ab6de2b38c93db72d2d511ca60e3b57e": "This is a 1*1 pixel GIF",
|
|
"fd8b021f0236e487bfee13bf8f0ae98760abc492f7ca3023e292631979e135cb4ccb0c89b6234971b060ad72c0ca4474cbb5092c6c7a3255d81a54a36277b486": "This is a 1*1 pixel GIF",
|
|
"235479f42cbbe0a4b0100167fece0d14c9b47d272b3ba8322bcfe8539f055bf31d500e7b2995cc968ebf73034e039f59c5f0f9410428663034bf119d74b5672c": "This is a 1*1 pixel GIF",
|
|
"a85e09c3b5dbb560f4e03ba880047dbc8b4999a64c1f54fbfbca17ee0bcbed3bc6708d699190b56668e464a59358d6b534c3963a1329ba01db21075ef5bedace": "This is a 1*1 pixel GIF",
|
|
"27656d6106a6da0c84174ba7a6307e6f1c4b3f2cc085c8466b6a25d54331035dabc7081aac208d960d8d37c5577547628c0d1c4b77bb4cf254c71859673feec1": "This is a 1*1 pixel GIF",
|
|
# "": "This is a 1*1 pixel GIF",
|
|
"f1c33e72643ce366fd578e3b5d393799e8c9ea27b180987826af43b4fc00b65a4eaae5e6426a23448956fee99e3108c6a86f32fb4896c156e24af0571a11c498": "This is a 1*1 pixel PNG",
|
|
"dc7c40381b3d22919e32c1b700ccb77b1b0aea2690642d01c1ac802561e135c01d5a4d2a0ea18efc0ec3362e8c549814a10a23563f1f56bd62aee0ced7e2bd99": "This is a 1*1 pixel PNG",
|
|
"c2c239cb5cdd0b670780ad6414ef6be9ccd4c21ce46bb93d1fa3120ac812f1679445162978c3df05cb2e1582a1844cc4c41cf74960b8fdae3123999c5d2176cc": "This is a 1*1 pixel PNG",
|
|
"6ad523f5b65487369d305613366b9f68dcdeee225291766e3b25faf45439ca069f614030c08ca54c714fdbf7a944fac489b1515a8bf9e0d3191e1bcbbfe6a9df": "This is a 1*1 pixel PNG",
|
|
# "": "This is a 1*1 pixel PNG",
|
|
"cf83e1357eefb8bdf1542850d66d8007d620e4050b5715dc83f4a921d36ce9ce47d0d13c5d85f2b0ff8318d2877eec2f63b931bd47417a81a538327af927da3e": "This is an empty file"
|
|
}
|
|
|
|
def __init__(self, config: Dict[str, Any]):
|
|
if not ('enabled' in config or config['enabled']):
|
|
self.available = False
|
|
return
|
|
self.client = SaneJS()
|
|
if not self.client.is_up:
|
|
self.available = False
|
|
return
|
|
self.available = True
|
|
self.storage_dir = get_homedir() / 'sanejs'
|
|
self.storage_dir.mkdir(parents=True, exist_ok=True)
|
|
|
|
def hashes_lookup(self, sha512: Union[List[str], str], force: bool=False) -> Dict[str, Any]:
|
|
if isinstance(sha512, str):
|
|
hashes = [sha512]
|
|
else:
|
|
hashes = sha512
|
|
|
|
today_dir = self.storage_dir / date.today().isoformat()
|
|
today_dir.mkdir(parents=True, exist_ok=True)
|
|
sanejs_unknowns = today_dir / 'unknown'
|
|
unknown_hashes = []
|
|
if sanejs_unknowns.exists():
|
|
with sanejs_unknowns.open() as f:
|
|
unknown_hashes = [line.strip() for line in f.readlines()]
|
|
|
|
to_return: Dict[str, Union[str, List[str]]] = {h: details for h, details in self.skip_lookup.items() if h in sha512}
|
|
|
|
to_lookup = [h for h in hashes if h not in self.skip_lookup]
|
|
if not force:
|
|
to_lookup = [h for h in to_lookup if (h not in unknown_hashes
|
|
and not (today_dir / h).exists())]
|
|
for h in to_lookup:
|
|
response = self.client.sha512(h)
|
|
if 'error' in response:
|
|
# Server not ready
|
|
break
|
|
if 'response' in response and response['response']:
|
|
cached_path = today_dir / h
|
|
with cached_path.open('w') as f:
|
|
json.dump(response['response'], f)
|
|
to_return[h] = response['response']
|
|
else:
|
|
unknown_hashes.append(h)
|
|
|
|
for h in hashes:
|
|
cached_path = today_dir / h
|
|
if h in unknown_hashes or h in to_return:
|
|
continue
|
|
elif cached_path.exists():
|
|
with cached_path.open() as f:
|
|
to_return[h] = json.load(f)
|
|
|
|
return to_return
|
|
|
|
|
|
class PhishingInitiative():
|
|
|
|
def __init__(self, config: Dict[str, Any]):
|
|
if 'apikey' not in config:
|
|
self.available = False
|
|
return
|
|
|
|
self.available = True
|
|
self.autosubmit = False
|
|
self.client = PyEUPI(config['apikey'])
|
|
if config.get('autosubmit'):
|
|
self.autosubmit = True
|
|
self.storage_dir_eupi = get_homedir() / 'eupi'
|
|
self.storage_dir_eupi.mkdir(parents=True, exist_ok=True)
|
|
|
|
def __get_cache_directory(self, url: str) -> Path:
|
|
m = hashlib.md5()
|
|
m.update(url.encode())
|
|
return self.storage_dir_eupi / m.hexdigest()
|
|
|
|
def get_url_lookup(self, url: str) -> Optional[Dict[str, Any]]:
|
|
url_storage_dir = self.__get_cache_directory(url)
|
|
if not url_storage_dir.exists():
|
|
return None
|
|
cached_entries = sorted(url_storage_dir.glob('*'), reverse=True)
|
|
if not cached_entries:
|
|
return None
|
|
|
|
with cached_entries[0].open() as f:
|
|
return json.load(f)
|
|
|
|
def url_lookup(self, url: str, force: bool=False) -> None:
|
|
'''Lookup an URL on Phishing Initiative
|
|
Note: force means 2 things:
|
|
* (re)scan of the URL
|
|
* re fetch the object from Phishing Initiative even if we already did it today
|
|
|
|
Note: the URL will only be sent for scan if autosubmit is set to true in the config
|
|
'''
|
|
if not self.available:
|
|
raise ConfigError('PhishingInitiative not available, probably no API key')
|
|
|
|
url_storage_dir = self.__get_cache_directory(url)
|
|
url_storage_dir.mkdir(parents=True, exist_ok=True)
|
|
pi_file = url_storage_dir / date.today().isoformat()
|
|
|
|
scan_requested = False
|
|
if self.autosubmit and force:
|
|
self.client.post_submission(url, comment='Received on Lookyloo')
|
|
scan_requested = True
|
|
|
|
if not force and pi_file.exists():
|
|
return
|
|
|
|
for i in range(3):
|
|
url_information = self.client.lookup(url)
|
|
if not url_information['results']:
|
|
# No results, that should not happen (?)
|
|
break
|
|
if url_information['results'][0]['tag'] == -1:
|
|
# Not submitted
|
|
if not self.autosubmit:
|
|
break
|
|
if not scan_requested:
|
|
self.client.post_submission(url, comment='Received on Lookyloo')
|
|
scan_requested = True
|
|
time.sleep(1)
|
|
else:
|
|
with pi_file.open('w') as _f:
|
|
json.dump(url_information, _f)
|
|
break
|
|
|
|
|
|
class VirusTotal():
|
|
|
|
def __init__(self, config: Dict[str, Any]):
|
|
if 'apikey' not in config:
|
|
self.available = False
|
|
return
|
|
|
|
self.available = True
|
|
self.autosubmit = False
|
|
self.client = vt.Client(config['apikey'])
|
|
if config.get('autosubmit'):
|
|
self.autosubmit = True
|
|
self.storage_dir_vt = get_homedir() / 'vt_url'
|
|
self.storage_dir_vt.mkdir(parents=True, exist_ok=True)
|
|
|
|
def __del__(self) -> None:
|
|
if hasattr(self, 'client'):
|
|
self.client.close()
|
|
|
|
def __get_cache_directory(self, url: str) -> Path:
|
|
url_id = vt.url_id(url)
|
|
m = hashlib.md5()
|
|
m.update(url_id.encode())
|
|
return self.storage_dir_vt / m.hexdigest()
|
|
|
|
def get_url_lookup(self, url: str) -> Optional[Dict[str, Any]]:
|
|
url_storage_dir = self.__get_cache_directory(url)
|
|
if not url_storage_dir.exists():
|
|
return None
|
|
cached_entries = sorted(url_storage_dir.glob('*'), reverse=True)
|
|
if not cached_entries:
|
|
return None
|
|
|
|
with cached_entries[0].open() as f:
|
|
return json.load(f)
|
|
|
|
def url_lookup(self, url: str, force: bool=False) -> None:
|
|
'''Lookup an URL on VT
|
|
Note: force means 2 things:
|
|
* (re)scan of the URL
|
|
* re fetch the object from VT even if we already did it today
|
|
|
|
Note: the URL will only be sent for scan if autosubmit is set to true in the config
|
|
'''
|
|
if not self.available:
|
|
raise ConfigError('VirusTotal not available, probably no API key')
|
|
|
|
url_id = vt.url_id(url)
|
|
url_storage_dir = self.__get_cache_directory(url)
|
|
url_storage_dir.mkdir(parents=True, exist_ok=True)
|
|
vt_file = url_storage_dir / date.today().isoformat()
|
|
|
|
scan_requested = False
|
|
if self.autosubmit and force:
|
|
self.client.scan_url(url)
|
|
scan_requested = True
|
|
|
|
if not force and vt_file.exists():
|
|
return
|
|
|
|
for i in range(3):
|
|
try:
|
|
url_information = self.client.get_object(f"/urls/{url_id}")
|
|
with vt_file.open('w') as _f:
|
|
json.dump(url_information.to_dict(), _f)
|
|
break
|
|
except vt.APIError as e:
|
|
if not self.autosubmit:
|
|
break
|
|
if not scan_requested and e.code == 'NotFoundError':
|
|
self.client.scan_url(url)
|
|
scan_requested = True
|
|
time.sleep(5)
|