2015-04-23 09:06:09 +02:00
|
|
|
#!/usr/bin/env python3
|
|
|
|
|
|
|
|
from PIL import Image
|
|
|
|
import argparse
|
2015-04-23 09:22:24 +02:00
|
|
|
from eta import ETA
|
|
|
|
|
|
|
|
get_bin = lambda x, n: x >= 0 and str(bin(x))[2:].zfill(n) or "-" + str(bin(x))[3:].zfill(n)
|
|
|
|
# From http://stackoverflow.com/a/21732313/2766106
|
2015-04-23 09:06:09 +02:00
|
|
|
|
|
|
|
|
|
|
|
def hideInFile(infile, hidefile, bits, outfile):
|
|
|
|
inim = Image.open(infile)
|
|
|
|
hideim = Image.open(hidefile)
|
|
|
|
assert inim.size == hideim.size, "Both image must be of the same size"
|
2015-04-23 09:22:24 +02:00
|
|
|
assert 0 <= bits <= 8
|
2015-04-23 09:06:09 +02:00
|
|
|
outim = Image.new('RGB', inim.size)
|
2015-04-23 09:22:24 +02:00
|
|
|
eta = ETA(inim.size[0] * inim.size[1])
|
|
|
|
for x in range(inim.size[0]):
|
|
|
|
for y in range(inim.size[1]):
|
|
|
|
incol = inim.getpixel((x, y))
|
|
|
|
hidecol = inim.getpixel((x, y))
|
|
|
|
outcol = []
|
|
|
|
for cp in range(len(incol)):
|
|
|
|
inbyt = get_bin(incol[cp], 8)
|
|
|
|
hidebyt = get_bin(hidecol[cp], 8)
|
|
|
|
outbyt = list(inbyt)
|
|
|
|
for bit in range(bits):
|
|
|
|
outbyt[-bit - 1] = inbyt[1]
|
|
|
|
outcol.append(int(''.join(outbyt), 2))
|
|
|
|
outim.putpixel((x, y), tuple(outcol))
|
|
|
|
eta.print_status(x * inim.size[1] + y)
|
|
|
|
eta.done()
|
2015-04-23 09:06:09 +02:00
|
|
|
inim.close()
|
|
|
|
hideim.close()
|
|
|
|
outim.save(outfile, 'PNG')
|
|
|
|
|
2015-04-23 09:22:24 +02:00
|
|
|
|
2015-04-23 09:06:09 +02:00
|
|
|
def inverseFile(infile, bits, outfile):
|
|
|
|
inim = Image.open(infile)
|
|
|
|
outim = Image.new('RGB', inim.size)
|
2015-04-23 09:22:24 +02:00
|
|
|
assert 0 <= bits <= 8
|
2015-04-23 09:06:09 +02:00
|
|
|
inim.close()
|
|
|
|
outim.save(outfile, 'PNG')
|
|
|
|
|
|
|
|
|
|
|
|
if __name__ == '__main__':
|
|
|
|
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('bits', metavar='BITS', type=int, help="Number of bits to use")
|
|
|
|
parser.add_argument('outfile', metavar='OUTFILE', type=str, help="Output file")
|
|
|
|
parser.add_argument('-s', '--hide', type=str, help="File to hide")
|
|
|
|
args = parser.parse_args()
|
|
|
|
if args.hide:
|
|
|
|
hideInFile(args.infile, args.hide, args.bits, args.outfile)
|
|
|
|
else:
|
|
|
|
inverseFile(args.infile, args.bits, args.outfile)
|