PSM PKG Backup tool

psmbackup.py

#!/usr/bin/env python3

import sys
from xml.dom import minidom
from functools import partial
import grequests
import hashlib
import hmac


PSM_PREFIXES = ['NPOA','NPNA','NPPA', 'NPQA']
PSM_PKG_BASE_URL = 'http://zeus.dl.playstation.net/psm/np/{prefix}/{prefix}{psmid:05d}_00_{hmac}/{version}/{prefix}{psmid:05d}_00.pkg'
PSM_VERSION_BASE_URL = 'http://zeus.dl.playstation.net/psm/np/{prefix}/{prefix}{psmid:05d}_00_{hmac}/version.xml'
HMAC_SECRET = bytearray.fromhex('5AE4E16B214290E14366E5B653C4E3C3E69E4956510EADD66ACB37A077E0686E086F8BAB5030E3D82407F01B676CB8037DF20D1420C08A91A2141A3FE5DC063C')

def eprint(*args, **kwargs):
    print(*args, file=sys.stderr, **kwargs)

def get_hmac(psmid):
    psmid = psmid + '_00'
    hm = hmac.new(HMAC_SECRET, psmid.encode('utf-8'), hashlib.sha1)
    return hm.hexdigest()[0:8].upper()

def psmNumToStr(psmNum):
    return str(psmNum).zfill(5)

def errHandler(req, excp):
    eprint(req, excp)

def respCallback(prefix, psmnum, hm, res, **kwargs):
    if res.status_code != 200:
        eprint('Weird, prefix: {} id:{} hmac:{} got status: {}'.format(prefix, psmnum, hm, res))
        return
    xmldoc = minidom.parseString(res.text)
    version = xmldoc.getElementsByTagName('appVersion')[0].firstChild.nodeValue
    print(PSM_PKG_BASE_URL.format(prefix=prefix, psmid=psmnum, hmac=hm, version=version))
    

def guessGen():
    for prefix in PSM_PREFIXES:
        for psmnum in range(0, 400):
            fullId = prefix + psmNumToStr(psmnum)
            hm = get_hmac(fullId)
            version_url = PSM_VERSION_BASE_URL.format(prefix=prefix, psmid=psmnum, hmac=hm)
            responseCb = partial(respCallback, prefix, psmnum, hm)
            yield grequests.get(version_url, hooks=dict(response=responseCb))

def guessPsmList():
    grequests.map(guessGen(), False, 5, errHandler)

if __name__ == '__main__':
    if sys.argv[1] == 'psmlist':
        guessPsmList()