summarylogtreecommitdiffstats
path: root/unlocker.py
diff options
context:
space:
mode:
authorJean-Marc Lenoir2019-10-05 11:29:34 +0200
committerJean-Marc Lenoir2019-10-05 11:29:34 +0200
commit256e517b3cbfcf7753befd48614e99fe36bc0477 (patch)
tree22e27a731ac04837e7bd6defba85cbca2a851ca0 /unlocker.py
parent55714fed2a5f3c4af7a2a6481a858ca0a2d4fd1c (diff)
downloadaur-256e517b3cbfcf7753befd48614e99fe36bc0477.tar.gz
Cleanup
Diffstat (limited to 'unlocker.py')
-rw-r--r--unlocker.py400
1 files changed, 0 insertions, 400 deletions
diff --git a/unlocker.py b/unlocker.py
deleted file mode 100644
index 0f9d3f66eeba..000000000000
--- a/unlocker.py
+++ /dev/null
@@ -1,400 +0,0 @@
-#!/usr/bin/env python
-"""
-The MIT License (MIT)
-
-Copyright (c) 2014-2018 Dave Parsons & Sam Bingner
-
-Permission is hereby granted, free of charge, to any person obtaining a copy
-of this software and associated documentation files (the 'Software'), to deal
-in the Software without restriction, including without limitation the rights
-to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
-copies of the Software, and to permit persons to whom the Software is
-furnished to do so, subject to the following conditions:
-
-The above copyright notice and this permission notice shall be included in
-all copies or substantial portions of the Software.
-
-THE SOFTWARE IS PROVIDED 'AS IS', WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
-IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
-FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
-AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
-LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
-OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
-THE SOFTWARE.
-
-vSMC Header Structure
-Offset Length Struct Type Description
-----------------------------------------
-0x00/00 0x08/08 Q ptr Offset to key table
-0x08/08 0x04/4 I int Number of private keys
-0x0C/12 0x04/4 I int Number of public keys
-
-vSMC Key Data Structure
-Offset Length Struct Type Description
-----------------------------------------
-0x00/00 0x04/04 4s int Key name (byte reversed e.g. #KEY is YEK#)
-0x04/04 0x01/01 B byte Length of returned data
-0x05/05 0x04/04 4s int Data type (byte reversed e.g. ui32 is 23iu)
-0x09/09 0x01/01 B byte Flag R/W
-0x0A/10 0x06/06 6x byte Padding
-0x10/16 0x08/08 Q ptr Internal VMware routine
-0x18/24 0x30/48 48B byte Data
-"""
-
-from __future__ import print_function
-import codecs
-import os
-import re
-import struct
-import sys
-
-if sys.version_info < (2, 7):
- sys.stderr.write('You need Python 2.7 or later\n')
- sys.exit(1)
-
-# Setup imports depending on whether IronPython or CPython
-if sys.platform == 'win32' \
- or sys.platform == 'cli':
- # noinspection PyUnresolvedReferences
- if sys.version_info > (3, 0):
- from winreg import *
- else:
- from _winreg import *
-
-
-def bytetohex(data):
- if sys.version_info > (3, 0):
- # Python 3 code in this block
- return "".join("{:02X} ".format(c) for c in data)
- else:
- # Python 2 code in this block
- return "".join("{:02X} ".format(ord(c)) for c in data)
-
-
-def joinpath(folder, filename):
- return os.path.join(folder, filename)
-
-
-def printkey(i, offset, smc_key, smc_data):
- print(str(i + 1).zfill(3)
- + ' ' + hex(offset)
- + ' ' + smc_key[0][::-1].decode('UTF-8')
- + ' ' + str(smc_key[1]).zfill(2)
- + ' ' + smc_key[2][::-1].replace(b'\x00', b' ').decode('UTF-8')
- + ' ' + '{0:#0{1}x}'.format(smc_key[3], 4)
- + ' ' + hex(smc_key[4])
- + ' ' + bytetohex(smc_data))
-
-
-def set_bit(value, bit):
- return value | (1 << bit)
-
-
-def clear_bit(value, bit):
- return value & ~(1 << bit)
-
-
-def test_bit(value, bit):
- return value & bit
-
-
-E_CLASS64 = 2
-E_SHT_RELA = 4
-
-
-def patchelf(f, oldoffset, newoffset):
- f.seek(0)
- magic = f.read(4)
- if not magic == b'\x7fELF':
- raise Exception('Magic number does not match')
-
- ei_class = struct.unpack('=B', f.read(1))[0]
- if ei_class != E_CLASS64:
- raise Exception('Not 64bit elf header: ' + ei_class)
-
- f.seek(40)
- e_shoff = struct.unpack('=Q', f.read(8))[0]
- f.seek(58)
- e_shentsize = struct.unpack('=H', f.read(2))[0]
- e_shnum = struct.unpack('=H', f.read(2))[0]
- e_shstrndx = struct.unpack('=H', f.read(2))[0]
-
- print('e_shoff: 0x{:x} e_shentsize: 0x{:x} e_shnum:0x{:x} e_shstrndx:0x{:x}'.format(e_shoff, e_shentsize,
- e_shnum, e_shstrndx))
-
- for i in range(0, e_shnum):
- f.seek(e_shoff + i * e_shentsize)
- e_sh = struct.unpack('=LLQQQQLLQQ', f.read(e_shentsize))
- # e_sh_name = e_sh[0]
- e_sh_type = e_sh[1]
- e_sh_offset = e_sh[4]
- e_sh_size = e_sh[5]
- e_sh_entsize = e_sh[9]
- if e_sh_type == E_SHT_RELA:
- e_sh_nument = int(e_sh_size / e_sh_entsize)
- # print 'RELA at 0x{:x} with {:d} entries'.format(e_sh_offset, e_sh_nument)
- for j in range(0, e_sh_nument):
- f.seek(e_sh_offset + e_sh_entsize * j)
- rela = struct.unpack('=QQq', f.read(e_sh_entsize))
- r_offset = rela[0]
- r_info = rela[1]
- r_addend = rela[2]
- if r_addend == oldoffset:
- r_addend = newoffset
- f.seek(e_sh_offset + e_sh_entsize * j)
- f.write(struct.pack('=QQq', r_offset, r_info, r_addend))
- print('Relocation modified at: ' + hex(e_sh_offset + e_sh_entsize * j))
-
-
-def patchkeys(f, key):
- # Setup struct pack string
- key_pack = '=4sB4sB6xQ'
- # smc_old_memptr = 0
- smc_new_memptr = 0
-
- # Do Until OSK1 read
- i = 0
- while True:
-
- # Read key into struct str and data byte str
- offset = key + (i * 72)
- f.seek(offset)
- smc_key = struct.unpack(key_pack, f.read(24))
- smc_data = f.read(smc_key[1])
-
- # Reset pointer to beginning of key entry
- f.seek(offset)
-
- if smc_key[0] == b'SKL+':
- # Use the +LKS data routine for OSK0/1
- smc_new_memptr = smc_key[4]
- print('+LKS Key: ')
- printkey(i, offset, smc_key, smc_data)
-
- elif smc_key[0] == b'0KSO':
- # Write new data routine pointer from +LKS
- print('OSK0 Key Before:')
- printkey(i, offset, smc_key, smc_data)
- # smc_old_memptr = smc_key[4]
- f.seek(offset)
- f.write(struct.pack(key_pack, smc_key[0], smc_key[1], smc_key[2], smc_key[3], smc_new_memptr))
- f.flush()
-
- # Write new data for key
- f.seek(offset + 24)
- smc_new_data = codecs.encode('bheuneqjbexolgurfrjbeqfthneqrqcy', 'rot_13')
- f.write(smc_new_data.encode('UTF-8'))
- f.flush()
-
- # Re-read and print key
- f.seek(offset)
- smc_key = struct.unpack(key_pack, f.read(24))
- smc_data = f.read(smc_key[1])
- print('OSK0 Key After:')
- printkey(i, offset, smc_key, smc_data)
-
- elif smc_key[0] == b'1KSO':
- # Write new data routine pointer from +LKS
- print('OSK1 Key Before:')
- printkey(i, offset, smc_key, smc_data)
- smc_old_memptr = smc_key[4]
- f.seek(offset)
- f.write(struct.pack(key_pack, smc_key[0], smc_key[1], smc_key[2], smc_key[3], smc_new_memptr))
- f.flush()
-
- # Write new data for key
- f.seek(offset + 24)
- smc_new_data = codecs.encode('rnfrqbagfgrny(p)NccyrPbzchgreVap', 'rot_13')
- f.write(smc_new_data.encode('UTF-8'))
- f.flush()
-
- # Re-read and print key
- f.seek(offset)
- smc_key = struct.unpack(key_pack, f.read(24))
- smc_data = f.read(smc_key[1])
- print('OSK1 Key After:')
- printkey(i, offset, smc_key, smc_data)
-
- # Finished so get out of loop
- break
-
- else:
- pass
-
- i += 1
- return smc_old_memptr, smc_new_memptr
-
-
-def patchsmc(name, sharedobj):
- with open(name, 'r+b') as f:
-
- smc_old_memptr = 0
- smc_new_memptr = 0
-
- # Read file into string variable
- vmx = f.read()
-
- print('File: ' + name + '\n')
-
- # Setup hex string for vSMC headers
- # These are the private and public key counts
- smc_header_v0 = b'\xF2\x00\x00\x00\xF0\x00\x00\x00'
- smc_header_v1 = b'\xB4\x01\x00\x00\xB0\x01\x00\x00'
-
- # Setup hex string for #KEY key
- key_key = b'\x59\x45\x4B\x23\x04\x32\x33\x69\x75'
-
- # Setup hex string for $Adr key
- adr_key = b'\x72\x64\x41\x24\x04\x32\x33\x69\x75'
-
- # Find the vSMC headers
- smc_header_v0_offset = vmx.find(smc_header_v0) - 8
- smc_header_v1_offset = vmx.find(smc_header_v1) - 8
-
- # Find '#KEY' keys
- smc_key0 = vmx.find(key_key)
- smc_key1 = vmx.rfind(key_key)
-
- # Find '$Adr' key only V1 table
- smc_adr = vmx.find(adr_key)
-
- # Print vSMC0 tables and keys
- print('appleSMCTableV0 (smc.version = "0")')
- print('appleSMCTableV0 Address : ' + hex(smc_header_v0_offset))
- print('appleSMCTableV0 Private Key #: 0xF2/242')
- print('appleSMCTableV0 Public Key #: 0xF0/240')
-
- if (smc_adr - smc_key0) != 72:
- print('appleSMCTableV0 Table : ' + hex(smc_key0))
- smc_old_memptr, smc_new_memptr = patchkeys(f, smc_key0)
- elif (smc_adr - smc_key1) != 72:
- print('appleSMCTableV0 Table : ' + hex(smc_key1))
- smc_old_memptr, smc_new_memptr = patchkeys(f, smc_key1)
-
- print()
-
- # Print vSMC1 tables and keys
- print('appleSMCTableV1 (smc.version = "1")')
- print('appleSMCTableV1 Address : ' + hex(smc_header_v1_offset))
- print('appleSMCTableV1 Private Key #: 0x01B4/436')
- print('appleSMCTableV1 Public Key #: 0x01B0/432')
-
- if (smc_adr - smc_key0) == 72:
- print('appleSMCTableV1 Table : ' + hex(smc_key0))
- smc_old_memptr, smc_new_memptr = patchkeys(f, smc_key0)
- elif (smc_adr - smc_key1) == 72:
- print('appleSMCTableV1 Table : ' + hex(smc_key1))
- smc_old_memptr, smc_new_memptr = patchkeys(f, smc_key1)
-
- print()
-
- # Find matching RELA record in .rela.dyn in ESXi ELF files
- # This is temporary code until proper ELF parsing written
- if sharedobj:
- print('Modifying RELA records from: ' + hex(smc_old_memptr) + ' to ' + hex(smc_new_memptr))
- patchelf(f, smc_old_memptr, smc_new_memptr)
-
- # Tidy up
- f.flush()
- f.close()
-
-
-def patchbase(name):
- # Patch file
- print('GOS Patching: ' + name)
- f = open(name, 'r+b')
-
- # Entry to search for in GOS table
- # Should work for Workstation 12-15...
- darwin = re.compile(
- b'\x10\x00\x00\x00[\x10|\x20]\x00\x00\x00[\x01|\x02]\x00\x00\x00\x00\x00\x00\x00'
- b'\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00')
-
- # Read file into string variable
- base = f.read()
-
- # Loop through each entry and set top bit
- # 0xBE --> 0xBF (WKS 12)
- # 0x3E --> 0x3F (WKS 14)
- for m in darwin.finditer(base):
- offset = m.start()
- f.seek(offset + 32)
- flag = ord(f.read(1))
- flag = set_bit(flag, 0)
- flag = chr(flag)
- f.seek(offset + 32)
- f.write(flag)
- print('GOS Patched flag @: ' + hex(offset))
-
- # Tidy up
- f.flush()
- f.close()
- print('GOS Patched: ' + name)
-
-
-def patchvmkctl(name):
- # Patch file
- print('smcPresent Patching: ' + name)
- f = open(name, 'r+b')
-
- # Read file into string variable
- vmkctl = f.read()
- applesmc = vmkctl.find(b'applesmc')
- f.seek(applesmc)
- f.write(b'vmkernel')
-
- # Tidy up
- f.flush()
- f.close()
- print('smcPresent Patched: ' + name)
-
-
-# noinspection PyUnresolvedReferences
-def main():
- # Work around absent Platform module on VMkernel
- if os.name == 'nt' or os.name == 'cli':
- osname = 'windows'
- else:
- osname = os.uname()[0].lower()
-
- # vmwarebase = ''
- vmx_so = False
-
- # Setup default paths
- if osname == 'linux':
- vmx_path = '/usr/lib/vmware/bin/'
- vmx = joinpath(vmx_path, 'vmware-vmx')
- vmx_debug = joinpath(vmx_path, 'vmware-vmx-debug')
- vmx_stats = joinpath(vmx_path, 'vmware-vmx-stats')
- if os.path.isfile('/usr/lib/vmware/lib/libvmwarebase.so/libvmwarebase.so'):
- vmx_so = True
- vmwarebase = '/usr/lib/vmware/lib/libvmwarebase.so/libvmwarebase.so'
- else:
- vmwarebase = '/usr/lib/vmware/lib/libvmwarebase.so.0/libvmwarebase.so.0'
-
- elif osname == 'windows':
- reg = ConnectRegistry(None, HKEY_LOCAL_MACHINE)
- key = OpenKey(reg, r'SOFTWARE\Wow6432Node\VMware, Inc.\VMware Workstation')
- vmwarebase_path = QueryValueEx(key, 'InstallPath')[0]
- vmx_path = QueryValueEx(key, 'InstallPath64')[0]
- vmx = joinpath(vmx_path, 'vmware-vmx.exe')
- vmx_debug = joinpath(vmx_path, 'vmware-vmx-debug.exe')
- vmx_stats = joinpath(vmx_path, 'vmware-vmx-stats.exe')
- vmwarebase = joinpath(vmwarebase_path, 'vmwarebase.dll')
-
- else:
- print('Unknown Operating System: ' + osname)
- return
-
- # Patch the vmx executables skipping stats version for Player
- patchsmc(vmx, vmx_so)
- patchsmc(vmx_debug, vmx_so)
- if os.path.isfile(vmx_stats):
- patchsmc(vmx_stats, vmx_so)
-
- # Patch vmwarebase for Workstation and Player
- patchbase(vmwarebase)
-
-
-if __name__ == '__main__':
- main()