def test_OTP_Extra(self):
for i in range(MAX_COVER_NOISE):
# Encode with more JT65 msgs than necessary
jt65msgs = [
"KB2BBC KA1AAB DD44", "KA1AAB KB2BBC DD44", "KB2BBC KA1AAB DD44",
"KA1AAB KB2BBC DD44", "KB2BBC KA1AAB DD44", "KA1AAB KB2BBC DD44"]
jt65data = jts.jt65encodemessages(jt65msgs, False)
stegmsg = "BEACON FTW AND DEF CON 22"
key = ''.join(random.choice(string.ascii_letters + string.digits)
for n in xrange(random.randint(10, 30)))
hidekey = jts.getnoisekey(key)
cipherdata = jts.createciphermsgs(
len(jt65data), stegmsg, "OTP", "I LOVE SECURITY AND STUFF", "", "", False)
finalmsgs = jts.steginject(jt65data, i, cipherdata, hidekey, False)
# Decode
finalmsgscopy = copy.deepcopy(finalmsgs)
jt65stegmsgs = []
decodedjt65msgs = jts.decodemessages(finalmsgs, False)
for i in range(len(finalmsgscopy)):
if jts.validatesteg(decodedjt65msgs[i], finalmsgscopy[i], hidekey, STEG_DETECTION_ERROR_THRESHOLD, False):
jt65stegmsgs.append(finalmsgscopy[i])
stegdata = jts.retrievesteg(jt65stegmsgs, hidekey, False)
resultstegmsg = jts.deciphersteg(
stegdata, "OTP", "I LOVE SECURITY AND STUFF", "", False)
self.assertEqual(len(decodedjt65msgs), len(jt65msgs))
for i in range(len(jt65msgs)):
self.assertEqual(
jt65msgs[i].rstrip(), decodedjt65msgs[i].rstrip())
self.assertEqual(stegmsg.rstrip(), resultstegmsg.rstrip())
def performwavdecode(filename, stegcollection):
# Decode a single wav file
containssteg = False
messages = jt65sound.inputwavfile(filename, verbose=args.verbose)
for currentmsg in messages:
symbols, confidence, msg, s2db, freq, a1, a2 = currentmsg
numpymsg = np.array(symbols)
jt65data = []
jt65data.append(numpymsg)
jt65datacopy = copy.deepcopy(
jt65data
) # Necessary on some version of Python due to 'unprepmsg' not preserving list
# Retrieve JT65 valid messages
jt65msgs = jts.decodemessages(jt65data, args.verbose)
if STEG_ENABLED:
containssteg = jts.validatesteg(jt65msgs[0], symbols, hidekey,
STEG_DETECTION_ERROR_THRESHOLD,
args.verbose)
if containssteg:
# Retrieve steg message
stegdata = jts.retrievesteg(
jt65datacopy, hidekey, args.verbose,
True) # Unprep the steg to get actual bytes
stegcollection.append(stegdata[0])
# Determine if we have a steg result
containsstegresult, stegstatus, resetcollection, stegmsg = getstegresult(
stegcollection, args.cipher, args.key, args.aesmode,
args.verbose)
if resetcollection:
stegcollection[:] = [
] # Reset the steg collection for the next incoming message
# Print result
for index, value in enumerate(jt65msgs):
print "\nDecoded JT65 message " + str(
index
) + " : " + colorama.Fore.BLUE + value + colorama.Fore.RESET
if containssteg and containsstegresult:
print "\nHidden message : " + colorama.Fore.RED + stegmsg + colorama.Fore.RESET
elif containssteg:
print "\n" + colorama.Fore.RED + "Steg detected! " + colorama.Fore.YELLOW + stegstatus + colorama.Fore.RESET
def performwavdecode(filename, stegcollection):
# Decode a single wav file
containssteg = False
messages = jt65sound.inputwavfile(filename, verbose=args.verbose)
for currentmsg in messages:
symbols, confidence, msg, s2db, freq, a1, a2 = currentmsg
numpymsg = np.array(symbols)
jt65data = []
jt65data.append(numpymsg)
jt65datacopy = copy.deepcopy(
jt65data
) # Necessary on some version of Python due to 'unprepmsg' not preserving list
# Retrieve JT65 valid messages
jt65msgs = jts.decodemessages(jt65data, args.verbose)
if STEG_ENABLED:
containssteg = jts.validatesteg(jt65msgs[0], symbols, hidekey, STEG_DETECTION_ERROR_THRESHOLD, args.verbose)
if containssteg:
# Retrieve steg message
stegdata = jts.retrievesteg(
jt65datacopy, hidekey, args.verbose, True
) # Unprep the steg to get actual bytes
stegcollection.append(stegdata[0])
# Determine if we have a steg result
containsstegresult, stegstatus, resetcollection, stegmsg = getstegresult(
stegcollection, args.cipher, args.key, args.aesmode, args.verbose
)
if resetcollection:
stegcollection[:] = [] # Reset the steg collection for the next incoming message
# Print result
for index, value in enumerate(jt65msgs):
print "\nDecoded JT65 message " + str(index) + " : " + colorama.Fore.BLUE + value + colorama.Fore.RESET
if containssteg and containsstegresult:
print "\nHidden message : " + colorama.Fore.RED + stegmsg + colorama.Fore.RESET
elif containssteg:
print "\n" + colorama.Fore.RED + "Steg detected! " + colorama.Fore.YELLOW + stegstatus + colorama.Fore.RESET
def test_OTP_Extra(self):
for i in range(MAX_COVER_NOISE):
# Encode with more JT65 msgs than necessary
jt65msgs = [
"KB2BBC KA1AAB DD44", "KA1AAB KB2BBC DD44",
"KB2BBC KA1AAB DD44", "KA1AAB KB2BBC DD44",
"KB2BBC KA1AAB DD44", "KA1AAB KB2BBC DD44"
]
jt65data = jts.jt65encodemessages(jt65msgs, False)
stegmsg = "BEACON FTW AND DEF CON 22"
key = ''.join(
random.choice(string.ascii_letters + string.digits)
for n in xrange(random.randint(10, 30)))
hidekey = jts.getnoisekey(key)
cipherdata = jts.createciphermsgs(len(jt65data), stegmsg, "OTP",
"I LOVE SECURITY AND STUFF", "",
"", False)
finalmsgs = jts.steginject(jt65data, i, cipherdata, hidekey, False)
# Decode
finalmsgscopy = copy.deepcopy(finalmsgs)
jt65stegmsgs = []
decodedjt65msgs = jts.decodemessages(finalmsgs, False)
for i in range(len(finalmsgscopy)):
if jts.validatesteg(decodedjt65msgs[i], finalmsgscopy[i],
hidekey, STEG_DETECTION_ERROR_THRESHOLD,
False):
jt65stegmsgs.append(finalmsgscopy[i])
stegdata = jts.retrievesteg(jt65stegmsgs, hidekey, False)
resultstegmsg = jts.deciphersteg(stegdata, "OTP",
"I LOVE SECURITY AND STUFF", "",
False)
self.assertEqual(len(decodedjt65msgs), len(jt65msgs))
for i in range(len(jt65msgs)):
self.assertEqual(jt65msgs[i].rstrip(),
decodedjt65msgs[i].rstrip())
self.assertEqual(stegmsg.rstrip(), resultstegmsg.rstrip())
# Decode
elif args.decode:
stegpresent = False
# Process input to JT numpy arrays
jt65data = processinput(args.stdin, args.wavin, args.verbose)
jt65datacopy = copy.deepcopy(jt65data)
jt65stegmsgs = []
# Retrieve JT65 valid messages
jt65msgs = jts.decodemessages(jt65data, args.verbose)
if STEG_ENABLED:
for i in range(len(jt65datacopy)):
if jts.validatesteg(jt65msgs[i], jt65datacopy[i], hidekey, STEG_DETECTION_ERROR_THRESHOLD, args.verbose):
jt65stegmsgs.append(jt65datacopy[i])
stegpresent = True
# Retrieve steg message
stegdata = jts.retrievesteg(jt65stegmsgs, hidekey, args.verbose)
# Decipher steg message
stegmsg = jts.deciphersteg(stegdata, args.cipher, args.key, args.aesmode, args.verbose)
# Print result
for index, value in enumerate(jt65msgs):
print "\nDecoded JT65 message " + str(index) + " : " + colorama.Fore.BLUE + value + colorama.Fore.RESET
if stegpresent:
print "\nHidden message : " + colorama.Fore.RED + stegmsg + colorama.Fore.RESET
# Decode
elif args.decode:
stegpresent = False
# Process input to JT numpy arrays
jt65data = processinput(args.stdin, args.wavin, args.verbose)
jt65datacopy = copy.deepcopy(jt65data)
jt65stegmsgs = []
# Retrieve JT65 valid messages
jt65msgs = jts.decodemessages(jt65data, args.verbose)
if STEG_ENABLED:
for i in range(len(jt65datacopy)):
if jts.validatesteg(jt65msgs[i], jt65datacopy[i], hidekey,
STEG_DETECTION_ERROR_THRESHOLD, args.verbose):
jt65stegmsgs.append(jt65datacopy[i])
stegpresent = True
# Retrieve steg message
stegdata = jts.retrievesteg(jt65stegmsgs, hidekey, args.verbose)
# Decipher steg message
stegmsg = jts.deciphersteg(stegdata, args.cipher, args.key, args.aesmode,
args.verbose)
# Print result
for index, value in enumerate(jt65msgs):
print "\nDecoded JT65 message " + str(
index) + " : " + colorama.Fore.BLUE + value + colorama.Fore.RESET
if stegpresent:
