Push 0.10.0
This commit is contained in:
parent
266f3fc76d
commit
12c05afa57
SSH key fingerprint:
SHA256:x4/RtnjUyuHkdvmwNDsWSfcfF1V5PNr3OpriZqOvCX8
25 changed files with 2016 additions and 645 deletions
237
libray/core.py
237
libray/core.py
|
|
@ -1,7 +1,7 @@
|
|||
# -*- coding: utf8 -*-
|
||||
|
||||
# libray - Libre Blu-Ray PS3 ISO Tool
|
||||
# Copyright © 2018 - 2021 Nichlas Severinsen
|
||||
# Copyright © 2018 - 2024 Nichlas Severinsen
|
||||
#
|
||||
# This file is part of libray.
|
||||
#
|
||||
|
|
@ -29,9 +29,11 @@ from bs4 import BeautifulSoup
|
|||
|
||||
|
||||
try:
|
||||
from libray import iso
|
||||
from libray import iso
|
||||
from libray import ird
|
||||
except ImportError:
|
||||
import iso
|
||||
import iso
|
||||
import ird
|
||||
|
||||
# Magic numbers / Constant variables
|
||||
|
||||
|
|
@ -43,15 +45,15 @@ GET_IRD_NET_LOC = 'http://jonnysp.bplaced.net/ird/'
|
|||
|
||||
|
||||
def to_int(data, byteorder='big'):
|
||||
"""Convert bytes to integer"""
|
||||
if isinstance(data, bytes):
|
||||
return int.from_bytes(data, byteorder)
|
||||
"""Convert bytes to integer"""
|
||||
if isinstance(data, bytes):
|
||||
return int.from_bytes(data, byteorder)
|
||||
|
||||
|
||||
def to_bytes(data):
|
||||
"""Convert a string of HEX to bytes"""
|
||||
if isinstance(data, str):
|
||||
return bytes(bytearray.fromhex(data))
|
||||
"""Convert a string of HEX to bytes"""
|
||||
if isinstance(data, str):
|
||||
return bytes(bytearray.fromhex(data))
|
||||
|
||||
|
||||
ISO_SECRET = to_bytes("380bcf0b53455b3c7817ab4fa3ba90ed")
|
||||
|
|
@ -59,166 +61,183 @@ ISO_IV = to_bytes("69474772af6fdab342743aefaa186287")
|
|||
|
||||
|
||||
def size(path):
|
||||
"""Get size of a file or block device in bytes"""
|
||||
pathstat = os.stat(path)
|
||||
"""Get size of a file or block device in bytes"""
|
||||
pathstat = os.stat(path)
|
||||
|
||||
# Check if it's a block device
|
||||
# Check if it's a block device
|
||||
|
||||
if stat.S_ISBLK(pathstat.st_mode):
|
||||
return open(path, 'rb').seek(0, os.SEEK_END)
|
||||
if stat.S_ISBLK(pathstat.st_mode):
|
||||
return open(path, 'rb').seek(0, os.SEEK_END)
|
||||
|
||||
# Otherwise, it's hopefully a file
|
||||
# Otherwise, it's hopefully a file
|
||||
|
||||
return pathstat.st_size
|
||||
return pathstat.st_size
|
||||
|
||||
|
||||
def read_seven_bit_encoded_int(fileobj, order):
|
||||
"""Read an Int32, 7 bits at a time."""
|
||||
# The highest bit of the byte, when on, means to continue reading more bytes.
|
||||
count = 0
|
||||
shift = 0
|
||||
byte = -1
|
||||
while (byte & 0x80) != 0 or byte == -1:
|
||||
# Check for a corrupted stream. Read a max of 5 bytes.
|
||||
if shift == (5 * 7):
|
||||
raise ValueError
|
||||
byte = to_int(fileobj.read(1), order)
|
||||
count |= (byte & 0x7F) << shift
|
||||
shift += 7
|
||||
return count
|
||||
"""Read an Int32, 7 bits at a time."""
|
||||
# The highest bit of the byte, when on, means to continue reading more bytes.
|
||||
count = 0
|
||||
shift = 0
|
||||
byte = -1
|
||||
while (byte & 0x80) != 0 or byte == -1:
|
||||
# Check for a corrupted stream. Read a max of 5 bytes.
|
||||
if shift == (5 * 7):
|
||||
raise ValueError
|
||||
byte = to_int(fileobj.read(1), order)
|
||||
count |= (byte & 0x7F) << shift
|
||||
shift += 7
|
||||
return count
|
||||
|
||||
|
||||
def error(msg):
|
||||
"""Print fatal error message and terminate"""
|
||||
print('[ERROR] %s' % msg)
|
||||
sys.exit(1)
|
||||
"""Print fatal error message and terminate"""
|
||||
print('[ERROR] %s' % msg, file=sys.stderr)
|
||||
sys.exit(1)
|
||||
|
||||
|
||||
def warning(msg, args):
|
||||
"""Print a warning message. Warning messages can be silenced with --quiet"""
|
||||
"""Print a warning message. Warning messages can be silenced with --quiet"""
|
||||
|
||||
if not args.quiet:
|
||||
print('[WARNING] %s. Continuing regardless.' % msg)
|
||||
if not args.quiet:
|
||||
print('[WARNING] %s. Continuing regardless.' % msg, file=sys.stderr)
|
||||
|
||||
|
||||
def vprint(msg, args):
|
||||
"""Vprint, verbose print, can be silenced with --quiet"""
|
||||
"""Vprint, verbose print, can be silenced with --quiet"""
|
||||
|
||||
if not args.quiet:
|
||||
print('[*] ' + msg)
|
||||
if not args.quiet:
|
||||
print('[*] ' + msg)
|
||||
|
||||
|
||||
def download_ird(ird_name):
|
||||
"""Download an .ird from GET_IRD_NET_LOC"""
|
||||
"""Download an .ird from GET_IRD_NET_LOC"""
|
||||
|
||||
# Check if file already exists and skip if it does
|
||||
if os.path.exists(ird_name): # TODO: might want to check that the file is valid first, could do a HEAD agains the url
|
||||
return
|
||||
# Check if file already exists and skip if it does
|
||||
if os.path.exists(ird_name): # TODO: might want to check that the file is valid first, could do a HEAD agains the url
|
||||
return
|
||||
|
||||
ird_link = GET_IRD_NET_LOC + ird_name
|
||||
r = requests.get(ird_link, stream=True)
|
||||
ird_link = GET_IRD_NET_LOC + ird_name
|
||||
r = requests.get(ird_link, stream=True)
|
||||
|
||||
with open(ird_name, 'wb') as ird_file:
|
||||
r.raw.decode_content = True
|
||||
shutil.copyfileobj(r.raw, ird_file)
|
||||
with open(ird_name, 'wb') as ird_file:
|
||||
r.raw.decode_content = True
|
||||
shutil.copyfileobj(r.raw, ird_file)
|
||||
|
||||
|
||||
def ird_by_game_id(game_id):
|
||||
"""Using a game_id, download the responding .ird from ALL_IRD_NET_LOC"""
|
||||
gameid = game_id.replace('-','')
|
||||
try:
|
||||
r = requests.get(ALL_IRD_NET_LOC, headers = {'User-Agent': 'Anonymous (You)' }, timeout=5)
|
||||
except requests.exceptions.ReadTimeout:
|
||||
error('Server timed out, fix your connection or manually specify a key/ird.')
|
||||
soup = BeautifulSoup(r.text, "html.parser")
|
||||
"""Using a game_id, download the responding .ird from ALL_IRD_NET_LOC"""
|
||||
gameid = game_id.replace('-', '')
|
||||
try:
|
||||
r = requests.get(ALL_IRD_NET_LOC, headers={'User-Agent': 'Anonymous (You)'}, timeout=5)
|
||||
except requests.exceptions.ReadTimeout:
|
||||
error('Server timed out, fix your connection or manually specify a key/ird.')
|
||||
soup = BeautifulSoup(r.text, "html.parser")
|
||||
|
||||
ird_name = False
|
||||
for elem in soup.find_all("a"):
|
||||
url = elem.get('href').split('/')[-1].replace('\\"','')
|
||||
if gameid in url:
|
||||
ird_name = url
|
||||
ird_name = False
|
||||
for elem in soup.find_all("a"):
|
||||
url = elem.get('href').split('/')[-1].replace('\\"', '')
|
||||
if gameid in url:
|
||||
ird_name = url
|
||||
|
||||
if not ird_name:
|
||||
error("Unable to download IRD, couldn't find link. You could specify the decryption key with -d if you have it.")
|
||||
if not ird_name:
|
||||
error("Unable to download IRD, couldn't find link. You could specify the decryption key with -d if you have it.")
|
||||
|
||||
download_ird(ird_name)
|
||||
download_ird(ird_name)
|
||||
|
||||
return(ird_name)
|
||||
return (ird_name)
|
||||
|
||||
|
||||
def crc32(filename):
|
||||
"""Calculate crc32 for file"""
|
||||
def crc32(filename, keep_going=[True]):
|
||||
"""Calculate crc32 for file"""
|
||||
|
||||
with open(filename, 'rb') as infile:
|
||||
with open(filename, 'rb') as infile:
|
||||
|
||||
crc32 = 0
|
||||
crc32 = 0
|
||||
|
||||
while True:
|
||||
data = infile.read(65536)
|
||||
if not data:
|
||||
break
|
||||
crc32 = zlib.crc32(data, crc32)
|
||||
while keep_going[0] == True:
|
||||
data = infile.read(65536)
|
||||
if not data:
|
||||
break
|
||||
crc32 = zlib.crc32(data, crc32)
|
||||
|
||||
return "%08X" % (crc32 & 0xFFFFFFFF)
|
||||
if keep_going[0] == False:
|
||||
return None
|
||||
|
||||
return "%08X" % (crc32 & 0xFFFFFFFF)
|
||||
|
||||
|
||||
def serial_country(title):
|
||||
"""Get country from disc serial / productcode / title_id"""
|
||||
"""Get country from disc serial / productcode / title_id"""
|
||||
|
||||
if title[2] == 'A':
|
||||
return 'Asia'
|
||||
if title[2] == 'C':
|
||||
return 'China'
|
||||
if title[2] == 'E':
|
||||
return 'Europe'
|
||||
if title[2] == 'H':
|
||||
return 'Hong Kong'
|
||||
if title[2] == 'J' or title[2] == 'P':
|
||||
return 'Japan'
|
||||
if title[2] == 'K':
|
||||
return 'Korea'
|
||||
if title[2] == 'U':
|
||||
return 'USA'
|
||||
if title[2] == 'A':
|
||||
return 'Asia'
|
||||
if title[2] == 'C':
|
||||
return 'China'
|
||||
if title[2] == 'E':
|
||||
return 'Europe'
|
||||
if title[2] == 'H':
|
||||
return 'Hong Kong'
|
||||
if title[2] == 'J' or title[2] == 'P':
|
||||
return 'Japan'
|
||||
if title[2] == 'K':
|
||||
return 'Korea'
|
||||
if title[2] == 'U' or title[2] == 'T':
|
||||
return 'USA'
|
||||
|
||||
raise ValueError('Unknown country?!')
|
||||
raise ValueError('Unknown country?!')
|
||||
|
||||
|
||||
def multiman_title(title):
|
||||
"""Fix special characters in title for Multiman style"""
|
||||
"""Fix special characters in title for Multiman style"""
|
||||
|
||||
replace = {
|
||||
':': ' -',
|
||||
'/': '-',
|
||||
'™': '',
|
||||
'®': '',
|
||||
}
|
||||
replace = {
|
||||
':': ' -',
|
||||
'/': '-',
|
||||
'™': '',
|
||||
'®': '',
|
||||
}
|
||||
|
||||
for key, val in replace.items():
|
||||
title = title.replace(key, val)
|
||||
for key, val in replace.items():
|
||||
title = title.replace(key, val)
|
||||
|
||||
return title
|
||||
return title
|
||||
|
||||
|
||||
# Main functions
|
||||
|
||||
def info(args):
|
||||
"""Print information about .iso and then quit."""
|
||||
|
||||
if args.iso:
|
||||
input_iso = iso.ISO(args)
|
||||
input_iso.print_info()
|
||||
sys.exit()
|
||||
|
||||
if args.ird:
|
||||
input_ird = ird.IRD(args.ird)
|
||||
input_ird.print_info(regions=True)
|
||||
sys.exit()
|
||||
|
||||
|
||||
def decrypt(args):
|
||||
"""Try to decrypt a given .iso using relevant .ird or encryption key from argparse
|
||||
"""Try to decrypt a given .iso using relevant .ird or encryption key from argparse
|
||||
|
||||
If no .ird is given this will try to automatically download an .ird file with the encryption/decryption key for the given game .iso
|
||||
"""
|
||||
If no .ird is given this will try to automatically download an .ird file with the encryption/decryption key for the given game .iso
|
||||
"""
|
||||
|
||||
input_iso = iso.ISO(args)
|
||||
input_iso = iso.ISO(args)
|
||||
|
||||
input_iso.decrypt(args) # TODO: some of the logic should probably be moved up here instead of residing in the decrypt function
|
||||
# TODO: some of the logic should probably be moved up here instead of residing in the decrypt function
|
||||
input_iso.decrypt(args)
|
||||
|
||||
|
||||
def encrypt(args):
|
||||
"""Try to re-encrypt a decrypted .iso using relevant .ird or encryption key from argparse
|
||||
"""Try to re-encrypt a decrypted .iso using relevant .ird or encryption key from argparse
|
||||
|
||||
If no .ird is given this will try to automatically download an .ird file with the encryption/decryption key for the given game .iso
|
||||
"""
|
||||
If no .ird is given this will try to automatically download an .ird file with the encryption/decryption key for the given game .iso
|
||||
"""
|
||||
|
||||
input_iso = iso.ISO(args)
|
||||
input_iso = iso.ISO(args)
|
||||
|
||||
input_iso.encrypt(args)
|
||||
input_iso.encrypt(args)
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue