verify.py

# -*- coding: utf-8 -*-

#this file is a lobotomy of squirell.py, taken from NSCB.
#i basically removed entire parts of its code until i was left only with a verification routine, which i patched up a bit because code quality in the original squirrel.py is horrendous.
#usage: .venv/bin/python verif.py "file"



import argparse
import sys
import os

os.chdir(os.path.dirname(os.path.abspath(__file__)))
sys.path.insert(0, 'lib')
try:
	sys.path.insert(0, 'private')
except:pass
import Config
import Status
import Fs


def verify(file) -> bool:
	try:
		parser = argparse.ArgumentParser()
		parser.add_argument('file',nargs='*')

		Status.start()

		filename = file
		dir=os.path.dirname(os.path.abspath(filename))
		tmpfolder =os.path.join(dir,'tmp')
		feed=''
		buffer = 65536
		if filename.lower().endswith(('.nsp','.nsz')):
			f = Fs.Nsp(filename)
		elif filename.lower().endswith('.xci'):
			f = Fs.factory(filename)
			f.open(filename, 'rb')
		elif filename.lower().endswith('.xcz'):
			f = Fs.Xci(filename)
		else:
			raise Exception("{1} does not have the appropriate extension".format(filename))


		check,feed=f.verify()

		if check == False:
			f.flush()
			f.close()
			return False

		verdict,headerlist,feed=f.verify_sig(feed,tmpfolder)
		if verdict == False:
			f.flush()
			f.close()
			return False
		print(filename)
		if filename.endswith('.nsz'):
			verdict,feed=f.nsz_hasher(buffer,headerlist,verdict,feed)
		elif filename.endswith('.xcz'):
			verdict,feed=f.xcz_hasher(buffer,headerlist,verdict,feed)
		else:
			verdict,feed=f.verify_hash_nca(buffer,headerlist,verdict,feed)
		f.flush()
		f.close()



		Status.close()

		return verdict
	except (KeyboardInterrupt, Exception) as e:
		print(f'Error occured on verify of {os.path.basename(file)}: {e}.')
		Config.isRunning = False
		Status.close()
		return False