init

cronjob.py

@@ -0,0 +1,88 @@
+'''
+  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.")
+