geoffrey/steginack
geoffrey/steginack
1
0
Fork 0
mirror of https://github.com/GeoffreyFrogeye/steginack.git synced 2024-11-20 18:06:04 +01:00
Code Issues Releases Wiki Activity

Better argument parsing

Browse source
This commit is contained in:
Geoffrey Frogeye 2015-04-23 11:45:45 +02:00
parent 3f34e955c8
commit be39a7d840
1 changed files with 24 additions and 15 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

39
steginack.py
View file

@ -8,11 +8,11 @@ get_bin = lambda x, n: x >= 0 and str(bin(x))[2:].zfill(n) or "-" + str(bin(x))[
# From http://stackoverflow.com/a/21732313/2766106 # From http://stackoverflow.com/a/21732313/2766106
def hideInFile(infile, hidefile, bits, outfile): def hideInFile(args):
inim = Image.open(infile) inim = Image.open(args.infile)
hideim = Image.open(hidefile) hideim = Image.open(args.hidefile)
assert inim.size == hideim.size, "Both image must be of the same size" assert inim.size == hideim.size, "Both image must be of the same size"
assert 0 <= bits <= 8 assert 0 <= args.bits <= 8
outim = Image.new('RGB', inim.size) outim = Image.new('RGB', inim.size)
eta = ETA(inim.size[0] * inim.size[1]) eta = ETA(inim.size[0] * inim.size[1])
for x in range(inim.size[0]): for x in range(inim.size[0]):
@ -24,7 +24,7 @@ def hideInFile(infile, hidefile, bits, outfile):
inbyt = get_bin(incol[cp], 8) inbyt = get_bin(incol[cp], 8)
hidebyt = get_bin(hidecol[cp], 8) hidebyt = get_bin(hidecol[cp], 8)
outbyt = list(inbyt) outbyt = list(inbyt)
for bit in range(bits): for bit in range(args.bits):
outbyt[-bit - 1] = hidebyt[bit] outbyt[-bit - 1] = hidebyt[bit]
outcol.append(int(''.join(outbyt), 2)) outcol.append(int(''.join(outbyt), 2))
outim.putpixel((x, y), tuple(outcol)) outim.putpixel((x, y), tuple(outcol))
@ -32,11 +32,11 @@ def hideInFile(infile, hidefile, bits, outfile):
eta.done() eta.done()
inim.close() inim.close()
hideim.close() hideim.close()
outim.save(outfile, 'PNG') outim.save(args.outfile, 'PNG')
def inverseFile(infile, outfile): def inverseFile(args):
inim = Image.open(infile) inim = Image.open(args.infile)
outim = Image.new('RGB', inim.size) outim = Image.new('RGB', inim.size)
eta = ETA(inim.size[0] * inim.size[1]) eta = ETA(inim.size[0] * inim.size[1])
for x in range(inim.size[0]): for x in range(inim.size[0]):
@ -53,17 +53,26 @@ def inverseFile(infile, outfile):
eta.print_status(x * inim.size[1] + y) eta.print_status(x * inim.size[1] + y)
eta.done() eta.done()
inim.close() inim.close()
outim.save(outfile, 'PNG') outim.save(args.outfile, 'PNG')
if __name__ == '__main__': if __name__ == '__main__':
parser = argparse.ArgumentParser(description="Hide an image inside another and decode images") parser = argparse.ArgumentParser(description="Hide an image inside another and decode images")
parser.add_argument('infile', metavar='INFILE', type=str, help="Input file") parser.add_argument('infile', metavar='INFILE', type=str, help="Input file")
subparsers = parser.add_subparsers(dest='action')
subparsers.required = True
hidep = subparsers.add_parser('hide')
hidep.add_argument('hidefile', metavar='HIDEFILE', type=str, help="File to hide")
hidep.add_argument('bits', metavar='BITS', type=int, help="Number of bits to use for hiding")
hidep.set_defaults(func=hideInFile)
inversep = subparsers.add_parser('inverse')
inversep.set_defaults(func=inverseFile)
parser.add_argument('outfile', metavar='OUTFILE', type=str, help="Output file") parser.add_argument('outfile', metavar='OUTFILE', type=str, help="Output file")
parser.add_argument('-b', '--bits', type=int, help="Number of bits to use for hiding")
parser.add_argument('-s', '--hide', type=str, help="File to hide")
args = parser.parse_args() args = parser.parse_args()
if args.hide: args.func(args)
hideInFile(args.infile, args.hide, args.bits, args.outfile)
else:
inverseFile(args.infile, args.outfile)