mirror of
https://github.com/heqin-zhu/algorithm.git
synced 2024-03-22 13:30:46 +08:00
70 lines
2.1 KiB
Python
70 lines
2.1 KiB
Python
|
from PIL import Image
|
|||
|
|
from skimage import color
|
|||
|
|
import numpy as np
|
|||
|
|
import matplotlib.pyplot as plt
|
|||
|
|
import math
|
|||
|
|
import argparse
|
|||
|
|
|
|||
|
|
|
|||
|
|
parser = argparse.ArgumentParser()
|
|||
|
|
parser.add_argument('picture')
|
|||
|
|
parser.add_argument('-f','file')
|
|||
|
|
parser.add_argument('-s','string')
|
|||
|
|
|
|||
|
|
args = parser.parse_args()
|
|||
|
|
|
|||
|
|
PIC = args.picture
|
|||
|
|
FILE = args.file
|
|||
|
|
STR = args.string
|
|||
|
|
|
|||
|
|
|
|||
|
|
class steganography:
|
|||
|
|
def __init__(self,picture):
|
|||
|
|
self.pic = poicture
|
|||
|
|
|
|||
|
|
def handlePixel(self,target,attach):
|
|||
|
|
'''对一个像素进行隐写 ,attach是要处理的数值，允许在0~9'''
|
|||
|
|
a,b,c = target
|
|||
|
|
pa,pb,pc = attach
|
|||
|
|
return a%10+pa,b%10+pb,c%10+pc
|
|||
|
|
def d2n(self,n,base=16):
|
|||
|
|
''' num of base_n(n<36) to decimal'''
|
|||
|
|
dic = {i:chr(i+ord('0')) for i in range(10)}
|
|||
|
|
if base>10:
|
|||
|
|
dic.update({i+10:chr(i+ord('A')) for i in range(26)})
|
|||
|
|
rst=[]
|
|||
|
|
while n!=0:
|
|||
|
|
i=int(n/base)
|
|||
|
|
rst.append(dic[n-i*base])
|
|||
|
|
n=i
|
|||
|
|
return ''.join(rst[::-1])
|
|||
|
|
def changeBody(self,path):
|
|||
|
|
self.pic = path
|
|||
|
|
def encryptPic(self,content,base =8):
|
|||
|
|
'''将bytes内容content隐写到self.picture中 base can be different value, default 8'''
|
|||
|
|
body = np .array (Image.open(self.pic))
|
|||
|
|
attach = np.array(content)
|
|||
|
|
r,c, _ = body.shape
|
|||
|
|
ar,ac,_ = attach.shape
|
|||
|
|
|
|||
|
|
raise Exception('信息量太大，不能隐写在此图片中，换张更大的图片')
|
|||
|
|
def encryptStr(self,content):
|
|||
|
|
body = np.array (Image.open(self.pic))
|
|||
|
|
r,c,d = body.shape
|
|||
|
|
btay = bytearray(content)
|
|||
|
|
length = len(btay)
|
|||
|
|
if length*8 > (r-1)*c:raise Exception('信息量太大，不能隐写在此图片中，换张更大的图片')
|
|||
|
|
def getContent(self,file=None,s=None):
|
|||
|
|
'''get the bytes , file is prior to str'''
|
|||
|
|
byte = b''
|
|||
|
|
if not file:
|
|||
|
|
if not s:
|
|||
|
|
s = input('输入要隐写的信息')
|
|||
|
|
byte = encode(STR,'utf-8')
|
|||
|
|
else:byte = open(FILE,'wb').read()
|
|||
|
|
return byte
|
|||
|
|
|
|||
|
|
if __name__ =='__main__':
|
|||
|
|
|
|||
|
|
PIC
|