Function crypt

src/python/icrypt.py:1–47

()

def crypt():
    import OpenImageIO as o
    import array
    import random
    import pickle

    path = raw_input("Enter OIIO plugin path: ")
    in_file = raw_input("Enter the name of the file for encryption: ")
    out_file = raw_input("Enter the name of the resulting file: ")
    key_path = raw_input("Enter the name of the file in which to store the key: ")

    # open the input file
    spec = o.ImageSpec()
    pic = o.ImageInput.create(in_file, path)
    pic.open(in_file, spec)
    desc = spec.format

    # Create a couple of arrays where we'll store the data.
    # They all need to be the same size, and we'll fill them with dummy data for now.
    # We'll read the original pixel values in arr
    arr = array.array("B", "\0" * spec.image_bytes())
    # the values we'll write to the encrypted file
    new_values = arr[:]
    length = range(len(new_values))
    print "Working, please wait..."
    pic.read_image(desc, arr)

    # save the state of the random number generator so we can use it 
    # to decode the image
    state = random.getstate()

    # generate random values, add them to the original values.
    # Do % 256 so nothing overflows
    for i in length:
        rand_val = random.randint(0, 255)
        new_values[i] = (arr[i] + rand_val) % 256
        
    # write new values to the output file, close everything
    out = o.ImageOutput.create(out_file, path)
    out.open(out_file, spec, False)
    out.write_image(desc, new_values)
    out.close()
    # save the state of the RNG - that's the key for decryption
    f = open(key_path, "w")
    pickle.dump(state, f)
    f.close()
    return True


def decrypt():