89 lines
3.2 KiB
Python
89 lines
3.2 KiB
Python
'''
|
|
This script records videos of a particular length and then takes the normal file and encrypts it,
|
|
storing both the encrypted file and the randomly generated iv (and deleting the plaintext video
|
|
file). It is designed to be called every x seconds, where x is the length of the desired video.
|
|
If it is not, then the resulting videos will have gaps between them.
|
|
'''
|
|
|
|
# import things
|
|
from cryptography.fernet import Fernet
|
|
from Crypto.Cipher import PKCS1_OAEP as RSA
|
|
from Crypto.Cipher import AES
|
|
import os
|
|
import glob
|
|
from signal import signal, SIGINT
|
|
from goprocam import GoProCamera, constants
|
|
from base64 import b64decode, b64encode
|
|
|
|
# global vars
|
|
location = "/home/pi/final/videos/" # location of video files, subfolders "plaintext" and "encrypted" assumed to exist
|
|
seconds = 3 # seconds to record video for
|
|
|
|
# helps with video recording
|
|
def handler(s, f):
|
|
gopro.stopWebcam()
|
|
quit()
|
|
|
|
# returns the current file number for naming convention
|
|
def numFiles():
|
|
return str(len([item for item in os.listdir(location+"encrypted/") if os.path.isfile(os.path.join(location+"encrypted/", item))]))
|
|
|
|
|
|
# take video for time seconds, transfer to local storage and delete from gopro
|
|
def takeVideo(time):
|
|
signal(SIGINT, handler)
|
|
gopro = GoProCamera.GoPro(ip_address=GoProCamera.GoPro.getWebcamIP())
|
|
gopro.shoot_video(time)
|
|
gopro.downloadLastMedia(custom_filename=(location+"plaintext/" + numFiles() + ".mp4"))
|
|
gopro.delete("last")
|
|
|
|
# TODO encrypt file data with assumed pgp key (file should already be read, this essentially
|
|
# encrypts the text or contents of the file)
|
|
def encryptPGP(file):
|
|
return file
|
|
#with open("pubkey.pem", "r") as keyfile:
|
|
# key = RSA.importKey(b64decode(str(keyfile)))
|
|
keytext = "-----BEGIN PUBLIC KEY-----\nMIGfMA0GCSqGSIb3DQEBAQUAA4GNADCBiQKBgQDLjon5dPwTgHUr44KndKz0W7bN\nwEqs6IImvnh/X++FD8vpMNPpepgEpZBC52vTyhmsgJYbkNnp8DxSUApE+IL5FlKr\nDCTLp3KLfjD8onjHf7mtD7uMpxWkafmor4BGq05M/QRwttYjaQNBGT3BFYHi5NdX\negPZLolmwrqwndwGAQIDAQAB\n-----END PUBLIC KEY-----"
|
|
key = RSA.importKey(keytext)
|
|
return key.encrypt(file, 'x')[0]
|
|
|
|
# encrypt given (video) file and store both encrypted video and (TODO) encrypted random iv
|
|
def encryptVideo(file):
|
|
# open the file as bytes
|
|
with open(file, "rb") as og:
|
|
message = og.read()
|
|
# generate keys
|
|
key = Fernet.generate_key()
|
|
fernet = Fernet(key)
|
|
# encrypt file
|
|
enc = fernet.encrypt(message)
|
|
# save the encrypted mp4
|
|
with open(location+"encrypted/" + numFiles() + ".mpc", "wb") as encf:
|
|
encf.write(enc)
|
|
# save the key
|
|
with open("keys/" + numFiles() + ".asc", "wb") as keyf:
|
|
keyf.write(encryptPGP(key))
|
|
|
|
# record the video and store in file location
|
|
takeVideo(seconds)
|
|
|
|
# get all video files in folder, encrypt (get all just in case one was missed previously)
|
|
fileset = [file for file in glob.glob(location + "plaintext/*.mp4", recursive=False)]
|
|
|
|
# whether or not to print out current action
|
|
verbose = True
|
|
|
|
for file in fileset:
|
|
if verbose:
|
|
print("Working on file " + file)
|
|
print("Encrypting...")
|
|
encryptVideo(file)
|
|
|
|
if verbose:
|
|
print("Deleting...")
|
|
os.remove(file)
|
|
|
|
if verbose:
|
|
print("Done.")
|
|
|