File size: 3,255 Bytes
d29aad7 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 |
from PIL import Image
def encrypt(text, secret_key, img):
image = Image.fromarray(img)
pixels = list(image.getdata())
binary_text = ''
for char in text:
binary_text+=format(ord(char), '08b')
binary_secret = ''
for char in secret_key:
binary_secret+=format(ord(char), '08b')
# print(binary_text)
binary_text = binary_demorf(binary_text, binary_secret)
binary_text = binary_text + binary_secret
# print(binary_text)
i = 0
print("number of bits we can store (binary_text + binary_secret): ",3*len(pixels))
new_pixels = []
for rgb in pixels:
tcr = []
for clr in rgb:
if i < len(binary_text):
if binary_text[i] == '0':
if clr % 2 != 0:
clr -= 1
else:
if clr % 2 == 0:
clr += 1
i += 1
tcr += [clr]
new_pixels += [tuple(tcr)]
new_image = Image.new(image.mode, image.size)
new_image.putdata(new_pixels)
return new_image
def decrypt(secret_key, img):
image = Image.fromarray(img)
pixels = list(image.getdata())
binary_secret = ""
for(char) in secret_key:
binary_secret += format(ord(char), '08b')
binary_text = ""
for rgb in pixels:
for(clr) in rgb:
if clr % 2 == 0:
binary_text += '0'
else:
binary_text += '1'
# print(binary_secret)
text_pos = binary_text.split(binary_secret)
if len(text_pos) == 1:
return "No text found"
else :
bt = text_pos[0];
# print(bt)
bt = binary_morf(bt, binary_secret)
# print(bt)
text = ""
for i in range(0, len(bt), 8):
text += chr(int(bt[i:i+8], 2))
return text
def binary_demorf(binary_text,binary_secret):
if len(binary_text) < len(binary_secret):
return binary_text
text = ""
cidx =0
idx = 0
while idx < len(binary_text):
ch = binary_text[idx]
if cidx < len(binary_secret):
if binary_secret[cidx] == ch:
if cidx == len(binary_secret)-1:
cidx = 0
if ch == '0':
text += '1'
else:
text += '0'
else:
text += ch
cidx += 1
idx += 1
else:
if(binary_secret[0]==ch):
cidx=1
else:
cidx=0
text += ch
idx+=1
else:
text += ch
idx+=1
return text
def binary_morf(binary_text,binary_secret):
if len(binary_text) < len(binary_secret):
return binary_text
t = binary_secret[:-1] # secret without the last digit
l = binary_text.split(t) # split text with that
# print(l)
result = ''
i = 0;
while i < len(l):
cr = ""
if(i!=0):
cr = l[i][1:]
else:
cr = l[i]
if i != len(l)-1:
result += cr + binary_secret;
else:
result += cr
i+=1
return result |