Программа стеганографии - преобразование python 2 в 3, синтаксическая ошибка в: base64.b64decode ("". Join (chars)) - PullRequest
Новая
       85

Программа стеганографии - преобразование python 2 в 3, синтаксическая ошибка в: base64.b64decode ("". Join (chars))

0 голосов
/ 07 мая 2020

У меня проблема с синтаксисом в последней части программы steg. Я попытался преобразовать python 2 версию (рабочего кода) в python 3, и это последняя его часть: 

flag = base64.b64decode("".join(chars)) <- error
print(flag)

Программа 1. шифрует сообщение в Last Significiant Бит изображения как сохраняет его как новое изображение. Затем 2. расшифровывает сообщение, которое хранится во «флаге», и печатает его. * может ли ошибка быть вызвана неправильным типом ввода?: 

message = input("Your message: ")

НИЖЕ: ОТКРЫТИЕ ПРОГРАММЫ 

    #coding: utf-8
import base64
from PIL import Image


image = Image.open("after.png")

extracted = ''

pixels = image.load()
#Iterating in 1st row
for x in range(0,image.width):
    r,g,b = pixels[x,0]
    # Storing LSB of each color
    extracted += bin(r)[-1]
    extracted += bin(g)[-1]
    extracted += bin(b)[-1]

chars = []
for i in range(len(extracted)/8):
    byte = extracted[i*8:(i+1)*8]
    chars.append(chr(int(''.join([str(bit) for bit in byte]), 2)))

flag = base64.b64decode(''.join(chars))
print flag

НИЖЕ: СКРЫТИЕ ПРОГРАММЫ: 

    import bitarray
import base64
from PIL import Image
with Image.open('before.png') as im:
    pixels=im.load()

message = input("Your message: ")

encoded_message = base64.b64encode(message.encode('utf-8'))

#Convert the message into an array of bits
ba = bitarray.bitarray()
ba.frombytes(encoded_message)
bit_array = [int(i) for i in ba]

#Duplicate the original picture
im = Image.open("before.png")
im.save("after.png")

im = Image.open("after.png")
width, height = im.size
pixels = im.load()

#Hide message in the first row
i = 0
for x in range(0,width):
    r,g,b = pixels[x,0]
    #print("[+] Pixel : [%d,%d]"%(x,0))
    #print("[+] \tBefore : (%d,%d,%d)"%(r,g,b))
    #Default values in case no bit has to be modified
    new_bit_red_pixel = 255
    new_bit_green_pixel = 255
    new_bit_blue_pixel = 255

    if i<len(bit_array):
        #Red pixel
        r_bit = bin(r)
        r_last_bit = int(r_bit[-1])
        r_new_last_bit = r_last_bit & bit_array[i]
        new_bit_red_pixel = int(r_bit[:-1]+str(r_new_last_bit),2)
        i += 1

    if i<len(bit_array):
        #Green pixel
        g_bit = bin(g)
        g_last_bit = int(g_bit[-1])
        g_new_last_bit = g_last_bit & bit_array[i]
        new_bit_green_pixel = int(g_bit[:-1]+str(g_new_last_bit),2)
        i += 1

    if i<len(bit_array):
        #Blue pixel
        b_bit = bin(b)
        b_last_bit = int(b_bit[-1])
        b_new_last_bit = b_last_bit & bit_array[i]
        new_bit_blue_pixel = int(b_bit[:-1]+str(b_new_last_bit),2)
        i += 1

    pixels[x,0] = (new_bit_red_pixel,new_bit_green_pixel,new_bit_blue_pixel)
    #print("[+] \tAfter: (%d,%d,%d)"%(new_bit_red_pixel,new_bit_green_pixel,new_bit_blue_pixel))
    im.save('after.png')

ошибка 

ValueError: string argument should contain only ASCII characters

1 Ответ

0 голосов
/ 07 мая 2020

help для base64.b64decode говорит: 

b64decode(s, altchars=None, validate=False)
    Decode the Base64 encoded bytes-like object or ASCII string s.
    ...

Учитывая, что в Python 2 были «нормальные» str s и unicode- str s (с префиксом u), Предлагаю внимательнее присмотреться к тому, что производят "".join(chars). Он содержит только символы ASCII? Я предлагаю добавить: 

print("Codes:",[ord(c) for c in chars])

непосредственно перед: 

flag = base64.b64decode("".join(chars))

Если внутри кодов будет число> 127, это означает, что он может не работать, поскольку он подходит только для чистого ASCII str с.

...