#!/usr/bin/python
|
|
import os
|
import subprocess
|
|
def filechecksum(aaxfile):
|
ret = subprocess.Popen(["ffprobe", "-v", "info", "-hide_banner", os.path.abspath(aaxfile)], stderr=subprocess.PIPE)
|
grep = subprocess.Popen(["grep", "checksum"], stdin=ret.stderr, stdout=subprocess.PIPE)
|
awk = subprocess.Popen(["awk", " { print $8 } "], stdin=grep.stdout, stdout=subprocess.PIPE)
|
hashsum,out = awk.communicate()
|
hashsum = hashsum.strip('\n')
|
return hashsum
|
|
|
def writeauthcode(hashsum):
|
try:
|
file = open(".authcode", "a")
|
file.write(hashsum + os.linesep)
|
except (IOError, OSError) as e:
|
print "Can't write to .authcode"
|
return 1
|
finally:
|
file.close()
|
return 0
|
|
|
def readauthcode():
|
try:
|
if not os.path.isfile(".authcode"):
|
open (".authcode", "w").close()
|
with open(".authcode") as f:
|
hashes = f.read().split()
|
except (IOError, OSError) as e:
|
print "Can't read in .authcode"
|
return 1
|
finally:
|
f.close()
|
return hashes
|
|
|
def newauthcode(checksum):
|
rcwd = os.path.dirname(os.path.realpath(__file__))
|
command = "./rcrack *.rtc -h "
|
command += checksum
|
cmd = subprocess.Popen(command, cwd=rcwd, stdout=subprocess.PIPE, shell=True)
|
grep = subprocess.Popen(["grep","-a", "hex"], stdin=cmd.stdout,stdout=subprocess.PIPE)
|
key,stderr = grep.communicate()
|
key = key.split()
|
return key[2].split(':')[1]
|
|
def checkagainstexisting(abytes,aaxfile):
|
ret = subprocess.Popen(["ffprobe","-activation_bytes", abytes ,"-hide_banner" , os.path.abspath(aaxfile)], stderr=subprocess.PIPE)
|
grep = subprocess.Popen(["grep", "mismatch", "-c"], stdin=ret.stderr, stdout=subprocess.PIPE)
|
works,out = grep.communicate()
|
return works.strip('\n')
|
|
|
def getcorrectkey(aaxfile):
|
authcodes = readauthcode()
|
abyte = ''
|
for x in authcodes:
|
if (checkagainstexisting(x,aaxfile) == '0'):
|
abyte = x
|
return abyte
|
if (abyte == ''):
|
csum = filechecksum(aaxfile)
|
abyte = newauthcode(csum)
|
writeauthcode(abyte)
|
return abyte
|
|
|
if __name__ == "__main__":
|
import sys
|
if len(sys.argv) >= 2:
|
args = sys.argv[1]
|
else:
|
print "I need an file name to lookup"
|
sys.exit(1)
|
checksum = filechecksum(args)
|
out = newauthcode(checksum.strip())
|
if out is not None:
|
print sys.argv[1], "Auth code is:", out
|
else:
|
print sys.argv[1], "no Authcode found"
|
