Major refactor. Release 0.0.1

- Now possible to give .iso and .ird and get a decrypted output.iso

libray/core.py

@@ -0,0 +1,82 @@
+# -*- coding: utf8 -*-
+
+# 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/>.
+
+import os
+import sys
+
+try:
+  from libray import iso
+except ImportError:
+  import iso
+
+# Magic numbers / Constant variables
+
+SECTOR = 2048
+
+# Utility functions
+
+def to_int(data, order='big'):
+  if isinstance(data, bytes):
+    return int.from_bytes(data, order)
+
+
+def to_bytes(data):
+  if isinstance(data, str):
+    return bytes(bytearray.fromhex(data))
+
+
+ISO_SECRET = to_bytes("380bcf0b53455b3c7817ab4fa3ba90ed")
+ISO_IV = to_bytes("69474772af6fdab342743aefaa186287")
+
+
+def filesize(filename):
+  return os.stat(filename).st_size
+
+
+def read_seven_bit_encoded_int(fileobj, order):
+  # Read out an Int32 7 bits at a time. The high 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('ERROR: %s' % msg)
+  sys.exit(1)
+
+# Main functions
+
+def decrypt(args):
+
+  input_iso = iso.ISO(args)
+
+  input_iso.decrypt(args)
+
+  
+  

libray/ird.py

@@ -0,0 +1,136 @@
+# -*- coding: utf8 -*-
+
+# 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/>.
+
+import os
+import sys
+import gzip
+import shutil
+
+try:
+  from libray import core
+except ImportError:
+  import core
+
+
+class IRD:
+
+
+  ORDER = 'little'
+  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)
+    with open(self.TEMP_FILE, 'rb') as input_ird:
+      if input_ird.read(4) != self.MAGIC_STRING:
+        core.error("Either not an IRD file, corruped IRD file, or unknown IRD format")
+
+      self.version = core.to_int(input_ird.read(1), self.ORDER)
+      self.game_id = input_ird.read(9)
+      name_length = core.read_seven_bit_encoded_int(input_ird, self.ORDER)
+      self.game_name = input_ird.read(name_length).decode('utf8')
+      self.update_version = input_ird.read(4)
+      self.game_version = input_ird.read(5)
+      self.app_version = input_ird.read(5)
+
+      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):
+        self.region_hashes.append(input_ird.read(16))
+
+      self.file_count = core.to_int(input_ird.read(4), self.ORDER)
+      self.file_hashes = []
+      for i in range(0, self.file_count):
+        key = core.to_int(input_ird.read(8), self.ORDER)
+        val = input_ird.read(16)
+        self.file_hashes.append({'key': key, 'val': val})
+
+      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)
+      self.data2 = input_ird.read(16)
+
+      if self.version < 9:
+        self.pic = input_ird.read(115)
+
+      if self.version < 7:
+        self.uid = core.to_int(input_ird.read(4), self.ORDER)
+
+      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: 
+      if input_ird.read(4) != self.MAGIC_STRING:
+        uncompress = True
+        
+    if uncompress:
+      with gzip.open(filename, 'rb') as gzfile:
+        with open(self.TEMP_FILE, 'wb') as tmpfile:
+          tmpfile.write(gzfile.read())
+    else:
+      shutil.copyfile(filename, self.TEMP_FILE)
+
+  
+
+  def print_info(self):
+    print('Info from IRD:')
+    print('Version: %s' % self.version)
+    print('Game ID: %s' % self.game_id)
+    print('Game Name: %s' % self.game_name)
+    print('Update Version: %s' % self.update_version)
+    print('Game Version: %s' % self.game_version)
+    print('App Version: %s' % self.app_version)
+    print('Region Count: %s' % self.region_count)
+    print('File Count: %s' % self.file_count)
+    print('Data1: %s' % self.data1.hex())
+    print('Data2: %s' % self.data2.hex())
+
+

libray/iso.py

@@ -0,0 +1,106 @@
+# -*- coding: utf8 -*-
+
+# 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/>.
+
+from Crypto.Cipher import AES
+
+try:
+  from libray import core
+  from libray import ird
+except ImportError:
+  import core
+  import ird
+
+
+class ISO:
+
+  
+  NUM_INFO_BYTES = 4
+
+
+  def __init__(self, args):
+    with open(args.iso, 'rb') as input_iso:
+      self.number_of_regions = core.to_int(input_iso.read(self.NUM_INFO_BYTES))
+      unused_bytes = input_iso.read(self.NUM_INFO_BYTES) # Yeah, I don't know either.
+
+      self.regions = self.read_regions(input_iso, args.iso)
+
+    if args.verbose:
+      self.print_info()
+
+    self.ird = ird.IRD(args)
+
+    cipher = AES.new(core.ISO_SECRET, AES.MODE_CBC, core.ISO_IV)
+    self.disc_key = cipher.encrypt(self.ird.data1)
+
+
+  def decrypt(self, args):
+
+    print('Decrypting with disc key: %s' % self.disc_key.hex())
+
+    with open(args.iso, 'rb') as input_iso:
+      with open(args.output, 'wb') as output_iso:
+        for i, region in enumerate(self.regions):
+          input_iso.seek(region['start'])
+
+          if not region['enc']:
+            while input_iso.tell() < region['end']:
+              data = input_iso.read(core.SECTOR)
+              output_iso.write(data)
+            continue
+          else:
+            while input_iso.tell() < region['end']:
+              num = input_iso.tell() // 2048
+              iv = bytearray([0 for i in range(0,16)])
+              for j in range(0,16):
+                iv[16 - j - 1] = (num & 0xFF)
+                num >>= 8
+            
+              data = input_iso.read(core.SECTOR)
+             
+              cipher = AES.new(self.disc_key, AES.MODE_CBC, bytes(iv))
+              decrypted = cipher.decrypt(data)
+            
+              output_iso.write(decrypted)
+
+
+  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, 
+        'enc': encrypted
+      })
+      input_iso.seek(input_iso.tell() - self.NUM_INFO_BYTES)
+      encrypted = not encrypted
+    regions[-1]['end'] = core.filesize(filename)
+
+    return regions
+
+
+  def print_info(self):
+    print('Info from ISO:')
+    print('Regions: %s (%s)' % (self.number_of_regions, self.number_of_regions*2) )
+    for i, region in enumerate(self.regions):
+      print(i, region)
+
+    

libray/libray

@@ -0,0 +1 @@
+libray.py

libray/libray.py

@@ -0,0 +1,44 @@
+#!/usr/bin/env python3
+# -*- coding: utf8 -*-
+
+# 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/>.
+
+
+import argparse
+
+try:
+  from libray import core
+except ImportError:
+  import core
+
+
+if __name__ == '__main__':
+
+  # Parse command line arguments with argpase
+  parser = argparse.ArgumentParser(description='A Libre (FLOSS) Python application for unencrypting, extracting, repackaging, and encrypting PS3 ISOs')
+  parser.add_argument('-v', '--verbose', help="Increase verbosity", action='count')
+  parser.add_argument('-o', '--output', dest='output', type=str, help="Output filename", default='output.iso')
+  required = parser.add_argument_group('required arguments')
+  required.add_argument('-i', '--iso', dest='iso', type=str, help="Path to .iso file", required=True)
+  required.add_argument('-k', '--ird', dest='ird', type=str, help="Path to .ird file", required=True)
+  args = parser.parse_args()
+
+  core.decrypt(args)
+
+