Oracle/LibRay
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Some tidying

Browse source
This commit is contained in:
Nichlas Severinsen 2019-05-16 10:37:28 +02:00
parent e45367a875
commit 56e9beaa20
8 changed files with 87 additions and 47 deletions
Download patch file Download diff file Expand all files Collapse all files

16
libray/core.py
View file

@ -2,19 +2,19 @@
# libray - Libre Blu-Ray PS3 ISO Tool
# Copyright (C) 2018 Nichlas Severinsen
#
#
# This file is part of libray.
#
#
# libray is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
#
# libray is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
#
# You should have received a copy of the GNU General Public License
# along with libray. If not, see <https://www.gnu.org/licenses/>.
@ -102,12 +102,12 @@ def ird_by_game_id(game_id):
ird_name = url
if not ird_name:
error("Unable to download IRD, couldn't find link")
error("Unable to download IRD, couldn't find link")
download_ird(ird_name)
return(ird_name)
# Main functions
@ -118,5 +118,5 @@ def decrypt(args):
input_iso.decrypt(args)

30
libray/ird.py
View file

@ -2,19 +2,19 @@
# libray - Libre Blu-Ray PS3 ISO Tool
# Copyright (C) 2018 Nichlas Severinsen
#
#
# This file is part of libray.
#
#
# libray is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
#
# libray is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
#
# You should have received a copy of the GNU General Public License
# along with libray. If not, see <https://www.gnu.org/licenses/>.
@ -37,9 +37,9 @@ class IRD:
TEMP_FILE = 'ird'
MAGIC_STRING = b"3IRD"
def __init__(self, args):
self.uncompress(args.ird) # TODO: Try/Except
self.size = core.filesize(self.TEMP_FILE)
@ -57,12 +57,12 @@ class IRD:
if self.version == 7:
self.identifier = input_ird.read(4)
header_length = (core.to_int(input_ird.read(4), self.ORDER))
self.header = input_ird.read(header_length)
footer_length = (core.to_int(input_ird.read(4), self.ORDER))
self.footer = input_ird.read(footer_length)
self.region_count = core.to_int(input_ird.read(1), self.ORDER)
self.region_hashes = []
for i in range(0, self.region_count):
@ -77,7 +77,7 @@ class IRD:
if self.version >= 9:
self.pic = input_ird.read(115)
unused_bytes = input_ird.read(4) # Yeah, I don't know either.
self.data1 = input_ird.read(16)
@ -91,24 +91,24 @@ class IRD:
if args.verbose:
self.print_info()
os.remove(self.TEMP_FILE)
def get_if_exists(self, input_ird):
starting_address = input_ird.tell()
length = core.read_seven_bit_encoded_int(input_ird, self.ORDER)
print(length)
if length:
return input_ird.read(length)
input_ird.seek(starting_address)
return None
def uncompress(self, filename):
uncompress = False
with open(filename, 'rb') as input_ird:
with open(filename, 'rb') as input_ird:
if input_ird.read(4) != self.MAGIC_STRING:
uncompress = True
@ -119,7 +119,7 @@ class IRD:
else:
shutil.copyfile(filename, self.TEMP_FILE)
def print_info(self):
print('Info from IRD:')

27
libray/iso.py
View file

@ -2,19 +2,19 @@
# libray - Libre Blu-Ray PS3 ISO Tool
# Copyright (C) 2018 Nichlas Severinsen
#
#
# This file is part of libray.
#
#
# libray is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
#
# libray is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
#
# You should have received a copy of the GNU General Public License
# along with libray. If not, see <https://www.gnu.org/licenses/>.
@ -33,7 +33,7 @@ except ImportError:
class ISO:
NUM_INFO_BYTES = 4
@ -76,7 +76,7 @@ class ISO:
with open(args.output, 'wb') as output_iso:
pbar = tqdm(total= (self.size // 2048) - 4 )
for i, region in enumerate(self.regions):
input_iso.seek(region['start'])
@ -96,28 +96,28 @@ class ISO:
for j in range(0,16):
iv[16 - j - 1] = (num & 0xFF)
num >>= 8
data = input_iso.read(core.SECTOR)
if not data:
core.warning("Trying to read past the end of the file")
break
pbar.update(1)
cipher = AES.new(self.disc_key, AES.MODE_CBC, bytes(iv))
decrypted = cipher.decrypt(data)
output_iso.write(decrypted)
pbar.close()
def read_regions(self, input_iso, filename):
regions = []
encrypted = False
for i in range(0, self.number_of_regions*2):
regions.append({
'start': core.to_int(input_iso.read(self.NUM_INFO_BYTES))*core.SECTOR,
'end': core.to_int(input_iso.read(self.NUM_INFO_BYTES))*core.SECTOR,
'start': core.to_int(input_iso.read(self.NUM_INFO_BYTES))*core.SECTOR,
'end': core.to_int(input_iso.read(self.NUM_INFO_BYTES))*core.SECTOR,
'enc': encrypted
})
input_iso.seek(input_iso.tell() - self.NUM_INFO_BYTES)
@ -133,4 +133,3 @@ class ISO:
for i, region in enumerate(self.regions):
print(i, region)

9
libray/libray.py
View file

@ -3,19 +3,19 @@
# libray - Libre Blu-Ray PS3 ISO Tool
# Copyright (C) 2018 Nichlas Severinsen
#
#
# This file is part of libray.
#
#
# libray is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
#
# libray is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
#
# You should have received a copy of the GNU General Public License
# along with libray. If not, see <https://www.gnu.org/licenses/>.
@ -41,4 +41,3 @@ if __name__ == '__main__':
core.decrypt(args)