mirrors/legendary
1
0
Fork 0
mirror of https://github.com/derrod/legendary.git synced 2024-12-22 01:45:28 +00:00
Code Issues Projects Releases Wiki Activity

[core] Verify CDN manifest hash

Browse source
This commit is contained in:
derrod 2022-01-10 17:00:45 +01:00
parent cf95da395c
commit 742d3a3b05
1 changed files with 9 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
legendary/core.py
View file

@ -10,6 +10,7 @@ from base64 import b64decode
from collections import defaultdict from collections import defaultdict
from concurrent.futures import ThreadPoolExecutor from concurrent.futures import ThreadPoolExecutor
from datetime import timezone from datetime import timezone
from hashlib import sha1
from locale import getdefaultlocale from locale import getdefaultlocale
from multiprocessing import Queue from multiprocessing import Queue
from platform import system from platform import system
@ -1173,6 +1174,7 @@ class LegendaryCore:
if len(m_api_r['elements']) > 1: if len(m_api_r['elements']) > 1:
raise ValueError('Manifest response has more than one element!') raise ValueError('Manifest response has more than one element!')
manifest_hash = m_api_r['elements'][0]['hash']
base_urls = [] base_urls = []
manifest_urls = [] manifest_urls = []
for manifest in m_api_r['elements'][0]['manifests']: for manifest in m_api_r['elements'][0]['manifests']:
@ -1186,13 +1188,18 @@ class LegendaryCore:
else: else:
manifest_urls.append(manifest['uri']) manifest_urls.append(manifest['uri'])
return manifest_urls, base_urls return manifest_urls, base_urls, manifest_hash
def get_cdn_manifest(self, game, platform='Windows'): def get_cdn_manifest(self, game, platform='Windows'):
manifest_urls, base_urls = self.get_cdn_urls(game, platform) manifest_urls, base_urls, manifest_hash = self.get_cdn_urls(game, platform)
self.log.debug(f'Downloading manifest from {manifest_urls[0]} ...') self.log.debug(f'Downloading manifest from {manifest_urls[0]} ...')
r = self.egs.unauth_session.get(manifest_urls[0]) r = self.egs.unauth_session.get(manifest_urls[0])
r.raise_for_status() r.raise_for_status()
manifest_bytes = r.content
if sha1(manifest_bytes).hexdigest() != manifest_hash:
raise ValueError('Manifest sha hash mismatch!')
return r.content, base_urls return r.content, base_urls
def get_uri_manifest(self, uri): def get_uri_manifest(self, uri):