def check_md5(rlsm_file, filearchives):
rlsm_file.make_file_tree()
raf_archives = {}
bad_files = []
for f in rlsm_file.file_tree:
if (f.is_archived()):
#print('Checking:', f.path)
if not f.version in raf_archives:
if not f.version in filearchives:
print(f)
continue
raf_archives[f.version] = raf.Raf(filearchives[f.version])
raf_archives[f.version].read()
check = raf_archives[f.version].check_file_md5(f.path, f.md5)
if not check:
print('Bad File:', f.path)
bad_files.append(f)
return bad_files
def extract(base_path, needle):
filearchives = get_filearchives(lol_path)
raf_archives = {}
rlsm_file = rlsm.RLSM(get_last_releasemanifest(lol_path))
files = rlsm_file.match_file(needle)
for i in files:
print('Extracted ' + i.path)
file_dir, file_name = os.path.split(i.path)
try:
os.makedirs(os.path.join(base_path, file_dir))
except:
pass
if not i.version in raf_archives:
if not i.version in filearchives:
print(f)
continue
raf_archives[i.version] = raf.Raf(filearchives[i.version])
raf_archives[i.version].read()
f = raf_archives[i.version].find_file(i.path)
raf_archives[i.version].data_file.extract_file(f.offset, f.size, os.path.join(base_path, i.path))
return 0
def apply_patch(path, rlsm_file, filearchives):
needed_versions = []
print('Unpacking Archives, please wait...')
threads = []
for dirname, dirnames, filenames in os.walk(os.path.join(path, 'DATA')):
for filename in filenames:
file_path = os.path.join(dirname, filename)
raf_path = re.sub(path + '/', '', file_path)
file_info = rlsm_file.find_file(raf_path)
if not file_info:
print("File '" + raf_path + "' not found in releasemanifest, patch could not be applied.")
exit(1)
version_dir = os.path.join(path, 'filearchives', int_to_ver(file_info.version), 'raf_archive')
if not file_info.version in needed_versions:
needed_versions.append(file_info.version)
raf_file = raf.Raf(filearchives[file_info.version])
raf_file.read()
if enable_threading:
t = ThreadedUnpack(raf_file, version_dir)
threads.append(t)
t.start()
else:
raf_file.unpack(version_dir)
#print(file_path)
#print(os.path.join(path, 'filearchives', int_to_ver(file_info.version), raf_path))
[t.join() for t in threads]
print('Saving Backups, please wait...')
timestamp = datetime.datetime.now().strftime('%Y%m%d-%H%M%S')
for version in needed_versions:
backup_dir = os.path.join(bk_dir, timestamp, int_to_ver(version))
file_name = os.path.split(filearchives[version])[1]
backup_file = os.path.join(backup_dir, file_name)
try:
os.makedirs(backup_dir)
except:
pass
shutil.copyfile(filearchives[version], backup_file)
shutil.copyfile(filearchives[version] + '.dat', backup_file + '.dat')
print('Replacing Files, please wait...')
for dirname, dirnames, filenames in os.walk(os.path.join(path, 'DATA')):
for filename in filenames:
file_path = os.path.join(dirname, filename)
raf_path = re.sub(path + '/', '', file_path)
file_info = rlsm_file.find_file(raf_path)
if not file_info:
print("File not found in releasemanifest, patch could not be applied.")
exit(1)
version_dir = os.path.join(path, 'filearchives', int_to_ver(file_info.version), 'raf_archive')
shutil.copyfile(file_path, os.path.join(version_dir, raf_path))
print('Repacking Archives, please wait...')
threads = []
for version in needed_versions:
version_dir = os.path.join(path, 'filearchives', int_to_ver(version))
file_name = os.path.split(filearchives[version])[1]
new_raf = raf.Raf(os.path.join(version_dir, file_name))
if enable_threading:
t = ThreadedPack(new_raf, rlsm_file, os.path.join(version_dir, 'raf_archive'))
threads.append(t)
t.start()
else:
new_raf.make_from_dirtree(os.path.join(version_dir, 'raf_archive'), rlsm_file)
new_raf.save()
[t.join() for t in threads]
print('Checking New Archives, please wait...')
for version in needed_versions:
version_dir = os.path.join(path, 'filearchives', int_to_ver(version))
file_name = os.path.split(filearchives[version])[1]
new_file = os.path.join(version_dir, file_name)
if os.path.getsize(new_file) != os.path.getsize(filearchives[version]):
print(new_file, 'size does not match the original one!')
exit(1)
test_1 = raf.Raf(new_file)
test_2 = raf.Raf(filearchives[version])
test_1.read()
test_2.read()
for i in range(0, len(test_1.files)):
if test_1.files[i].p_hash != test_2.files[i].p_hash:
print('Wrong Hash:', test_1.files[i], test_2.files[i])
exit(1)
if test_1.paths[test_1.files[i].p_index].string != test_2.paths[test_2.files[i].p_index].string:
print('Wrong Path:', test_1.paths[test_1.files[i].p_index].string, test_2.paths[test_2.files[i].p_index].string)
exit(1)
del test_1
del test_2
print('Moving Archives, please wait...')
for version in needed_versions:
version_dir = os.path.join(path, 'filearchives', int_to_ver(version))
file_name = os.path.split(filearchives[version])[1]
new_file = os.path.join(version_dir, file_name)
os.remove(filearchives[version])
os.remove(filearchives[version] + '.dat')
shutil.move(new_file, filearchives[version])
shutil.move(new_file + '.dat', filearchives[version] + '.dat')
print('Cleaning up, please wait...')
shutil.rmtree(tmp_dir)
print('Patch applied Successfully!')
elif sys.argv[1] == 'search':
if len(sys.argv) != 3:
print_usage()
rlsm_file = rlsm.RLSM(get_last_releasemanifest(lol_path))
files = rlsm_file.match_file(sys.argv[2])
for i in files:
print(i)
elif sys.argv[1] == 'extract':
if len(sys.argv) != 3:
print_usage()
base_path = 'extract'
extract(base_path, sys.argv[2])
elif sys.argv[1] == 'unpack':
if len(sys.argv) != 3:
print_usage()
filearchives = get_filearchives(lol_path)
raf_file = None
try:
raf_file = raf.Raf(filearchives[int(sys.argv[2])])
except:
print("Archive file not found.")
exit(1)
raf_file.read()
raf_file.unpack(os.path.join(tmp_dir, "unpack"))
elif sys.argv[1] == 'info':
print("Not Yet Implemented")
else:
print(sys.argv[1], 'is not a valid command.')