V0.1.6 clean up p2

Author: Vincent-G-Van

AESdecrypt.py

@@ -1,5 +1,6 @@
 from AESdecryptfunc import * #import AESdecryptfunc module to use functions created for this program
 import math #import math module to use function such as ceiling
+import io
 
 #check that script is running with the two text files as the two parameters or else quit
 if len(sys.argv) is not 3:#takes in two arguments for the ciphertext.txt file name and plainhex.txt file name
@@ -9,7 +10,7 @@
 while(len(PassPhrase)!=16):
     print("Enter in the 16 character passphrase to decrypt your text file %s" %sys.argv[1])
     PassPhrase=input()#takes in user input of char, eg. "Iwanttolearnkung"
-    print(len(PassPhrase))
+    #print(len(PassPhrase))
     if(len(PassPhrase)<16):#check if less than 16 characters, if so add one space character until 16 chars
         while(len(PassPhrase)!=16):
             PassPhrase=PassPhrase+" "
@@ -20,16 +21,17 @@
 #open ciphertext.txt file to read and decrypt
 file=open(sys.argv[1], "r")
 message=(file.read())
-print("Inside your ciphertext message is: %s" % message)
+print("Inside your ciphertext message is:\n%s\n" % message)
+file.close()
 
 #set up some parameters
-start=len(message)-32#set starting pointer for the part to decrypt of the ciphertext
-end=len(message)#set ending pointer for the part to decrypt of the plaintex
+start=0#set starting pointer for the part to decrypt of the ciphertext
+end=32#set ending pointer for the part to decrypt of the plaintex
 length=len(message)#check the entire size of the message
 loopmsg=0.00#create a decimal value
 loopmsg=math.ceil(length/32)+1#use formula to figure how long the message is and how many 16 character segmentss must be decrypted
-outputhex=""#setup output message in hex
-asciioutput=""
+outputhex=""#setup output message segment in hex
+asciioutput=""#setup compilation of output message in ascii
 
 #need to setup roundkeys here
 PassPhrase=BitVector(textstring=PassPhrase)
@@ -44,59 +46,62 @@
 roundkey9=findroundkey(roundkey8,9)
 roundkey10=findroundkey(roundkey9,10)
 roundkeys=[roundkey1,roundkey2,roundkey3,roundkey4,roundkey5,roundkey6,roundkey7,roundkey8,roundkey9,roundkey10]
-print("Roundkey 10 is: %s" % roundkeys[9])
+#print("Roundkey 10 is: %s" % roundkeys[9])
+
+FILEOUT = io.open(sys.argv[2], 'w', encoding='utf-8')
+
 # set up the segement message loop parameters
 for y in range(1, loopmsg): # loop to encrypt all segments of the message
     plaintextseg = message[start:end]
-    print("The ciphertext segment to decrypt is: %s" % plaintextseg)
+    #print("The ciphertext segment to decrypt is: %s" % plaintextseg)
     # add round key
     bv1 = BitVector(hexstring=plaintextseg)
     bv2 = BitVector(hexstring=roundkeys[9])
     resultbv = bv1 ^ bv2
     myhexstring = resultbv.get_bitvector_in_hex()
-    print("The output after adding the roundkey 10 is: %s" % myhexstring)
+    #print("The output after adding the roundkey 10 is: %s" % myhexstring)
     #inverse shift row
     myhexstring=invshiftrow(myhexstring)
-    print("The output after invshiftrow is: %s" % myhexstring)
+    #print("The output after invshiftrow is: %s" % myhexstring)
     #inverse subbyte
     myhexstring=invsubbyte(myhexstring)
-    print("The output after invsubbyte is: %s" % myhexstring)
+    #print("The output after invsubbyte is: %s" % myhexstring)
 
     for x in range(8, -1, -1):
-        print("Round: %i" % (x+1))
+        #print("Round: %i" % (x+1))
         # add roundkey for current round
         bv1 = BitVector(hexstring=myhexstring)
         bv2 = BitVector(hexstring=roundkeys[x])
         resultbv = bv1 ^ bv2
         myhexstring = resultbv.get_bitvector_in_hex()
-        print("The output after adding the roundkey %i is: %s" %((x+1),myhexstring))
+        #print("The output after adding the roundkey %i is: %s" %((x+1),myhexstring))
         # mix column
         bv3 = BitVector(hexstring=myhexstring)
         myhexstring=invmixcolumn(bv3)
-        print("The output after invmixcolumn %i is: %s" % ((x + 1), myhexstring))
+        #print("The output after invmixcolumn %i is: %s" % ((x + 1), myhexstring))
         # shift rows
         myhexstring = invshiftrow(myhexstring)
-        print("The output after invshiftrow is: %s" % myhexstring)
+        #print("The output after invshiftrow is: %s" % myhexstring)
         # sub byte
         myhexstring = invsubbyte(myhexstring)
-        print("The output after invsubbyte is: %s" % myhexstring)
+        #print("The output after invsubbyte is: %s" % myhexstring)
 
     #add initial round key
     bv1 = BitVector(hexstring=myhexstring)
     bv2 = PassPhrase
     resultbv = bv1 ^ bv2
     myhexstring = resultbv.get_bitvector_in_hex()
-    print("The output after adding the initial roundkey is: %s" % myhexstring)
+    #print("The output after adding the initial roundkey is: %s" % myhexstring)
 
-    start = start - 32 #increment start pointer
-    end = end - 32 #increment end pointer
+    start = start + 32 #increment start pointer
+    end = end + 32 #increment end pointer
 
-    outputhex = myhexstring+outputhex
+    outputhex = BitVector(hexstring=myhexstring)
+    asciioutput = outputhex.get_bitvector_in_ascii()
+    FILEOUT.write(asciioutput)
 
-print(myhexstring)
-outputhex = BitVector(hexstring=outputhex)
-asciioutput = outputhex.get_bitvector_in_ascii()
-print("The decrypted message for the entire ciphertext is: %s" % asciioutput)
-FILEOUT = open(sys.argv[2], 'w')
-FILEOUT.write(asciioutput)
 FILEOUT.close()
+
+file2=io.open(sys.argv[2], "r", encoding='utf-8')
+print("The decrypted message for the entire ciphertext is:\n%s\n" % file2.read())
+file2.close()

AESdecryptfunc.py

@@ -69,7 +69,7 @@ def subbyte(myhexstring):
 
     lookuptable=[part0,part1,part2,part3,part4,part5,part6,part7,part8,part9,part10,part11,part12,part13,part14,part15]
 
-    print("The string size is ", len(myhexstring), " and the loop will run", math.ceil(len(myhexstring)/2), " times." )
+    #print("The string size is ", len(myhexstring), " and the loop will run", math.ceil(len(myhexstring)/2), " times." )
     for loop in range(0, math.ceil(len(myhexstring)/2) ):
         x = ""
         y = ""
@@ -205,7 +205,7 @@ def invsubbyte(myhexstring):
 
     lookuptable=[part0,part1,part2,part3,part4,part5,part6,part7,part8,part9,part10,part11,part12,part13,part14,part15]
 
-    print("The string size is ", len(myhexstring), " and the loop will run", math.ceil(len(myhexstring)/2), " times." )
+    #print("The string size is ", len(myhexstring), " and the loop will run", math.ceil(len(myhexstring)/2), " times." )
     for loop in range(0, math.ceil(len(myhexstring)/2) ):
         x = ""
         y = ""

AESencrypt.py

@@ -10,7 +10,7 @@
 while(len(PassPhrase)!=16):
     print("Enter in the 16 character passphrase to encrypt your text file %s" %sys.argv[1])
     PassPhrase=input()#takes in user input of char, eg. "Iwanttolearnkung"
-    print(len(PassPhrase))
+    #print(len(PassPhrase))
     if(len(PassPhrase)<16):#check if less than 16 characters, if so add one space character until 16 chars
         while(len(PassPhrase)!=16):
             PassPhrase=PassPhrase+" "
@@ -21,7 +21,8 @@
 #open plaintext.txt file to read and encrypt
 file=open(sys.argv[1], "r")
 message=(file.read())
-print("Inside your plaintext message is: %s" % message)
+print("Inside your plaintext message is:\n%s\n" % message)
+file.close()
 
 #set up some parameters
 start=0#set starting pointer for the part to encrypt of the plaintext
@@ -45,6 +46,8 @@
 roundkey10=findroundkey(roundkey9,10)
 roundkeys=[roundkey1,roundkey2,roundkey3,roundkey4,roundkey5,roundkey6,roundkey7,roundkey8,roundkey9,roundkey10]
 
+FILEOUT = open(sys.argv[2], 'w')
+
 # set up the segement message loop parameters
 for y in range(1, loopmsg): # loop to encrypt all segments of the message
     if(end+16<length): #if the end pointer is less than the size of the message, then set the segment to be 16 characters
@@ -56,70 +59,71 @@
             plaintextseg=plaintextseg.get_bitvector_in_hex()+"00"
             plaintextseg=BitVector(hexstring=plaintextseg).get_bitvector_in_ascii()
     #add round key zero/ find round key one
-    print("The round key string is : %s" % PassPhrase)
-    print("The part of the message to be encrypted is : %s" % plaintextseg)
+    #print("The round key string is : %s" % PassPhrase)
+    #print("The part of the message to be encrypted is : %s" % plaintextseg)
     bv1 = BitVector(textstring=plaintextseg)
-    print("The plaintext message in hex is : %s" % bv1.get_bitvector_in_hex())
+    #print("The plaintext message in hex is : %s" % bv1.get_bitvector_in_hex())
     bv2 = PassPhrase
-    print("The password in hex/ roundkey zero is : %s" % bv2.get_bitvector_in_hex())
+    #print("The password in hex/ roundkey zero is : %s" % bv2.get_bitvector_in_hex())
     resultbv=bv1^bv2
     myhexstring = resultbv.get_bitvector_in_hex()
-    print("The initial adding round key output is : %s" % myhexstring)
+    #print("The initial adding round key output is : %s" % myhexstring)
 
     for x in range(1, 10):  # loop through 9 rounds
         # sub byte
-        print("Round: %i" % x)
+        #print("Round: %i" % x)
         myhexstring = resultbv.get_bitvector_in_hex()
         temp1=subbyte(myhexstring)
-        print("The subbyte output is: %s" % temp1)
+        #print("The subbyte output is: %s" % temp1)
 
         # shift rows
         temp2=shiftrow(temp1)
-        print("The shiftrow output is: %s" % temp2)
+        #print("The shiftrow output is: %s" % temp2)
 
         # mix column
         bv3 = BitVector(hexstring=temp2)
         newbvashex=mixcolumn(bv3)
         newbv=BitVector(hexstring=newbvashex)
-        print("The Mixcolumn Output is: %s" % newbvashex)
+        #print("The Mixcolumn Output is: %s" % newbvashex)
 
         #add roundkey for current round
         bv1 = BitVector(bitlist=newbv)
         bv2 = BitVector(hexstring=roundkeys[x-1])
         resultbv = bv1 ^ bv2
         myhexresult = resultbv.get_bitvector_in_hex()
-        print("The round key %i is " % (x) + roundkeys[x-1])
+        #print("The round key %i is " % (x) + roundkeys[x-1])
 
-        print("The output after adding the round key: %s" % myhexresult)
+        #print("The output after adding the round key: %s" % myhexresult)
 
     #start round 10
     # sub byte round 10
-    print("Round 10:")
+    #print("Round 10:")
     myhexstring = resultbv.get_bitvector_in_hex()
     temp1=subbyte(myhexstring)
-    print("The subbyte output of round 10 is: %s" % temp1)
+    #print("The subbyte output of round 10 is: %s" % temp1)
 
     # shift rows round 10
     temp2=shiftrow(temp1)
-    print("The output after shiftrow is: %s" % temp2)
+    #print("The output after shiftrow is: %s" % temp2)
 
     # addround key round 10
     newbv = BitVector(hexstring=temp2)
     bv1 = BitVector(bitlist=newbv)
     bv2 = BitVector(hexstring=roundkeys[9])
     resultbv = bv1 ^ bv2
     myhexstring = resultbv.get_bitvector_in_hex()
-    print("The output after adding the roundkey is: %s" % myhexstring)
-
+    #print("The output after adding the roundkey is: %s" % myhexstring)
     #set encrypted hex segement of message to output string
     outputhextemp = resultbv.get_hex_string_from_bitvector()
-    print("The outputhex value for this part of the message is: %s" % outputhextemp)
-    outputhex = outputhex + outputhextemp
+    #print("The outputhex value for this part of the message is: %s" % outputhextemp)
+    FILEOUT.write(outputhextemp)
     start = start + 16 #increment start pointer
     end = end + 16 #increment end pointer
 
-#  encrypted output hex string to specified cipherhex file
-print("The outputhex value for the entire message is: %s" % outputhex)
-FILEOUT = open(sys.argv[2], 'w')
-FILEOUT.write(outputhex)
+# encrypted output hex string to specified cipherhex file
 FILEOUT.close()
+
+file2=open(sys.argv[2], "r")
+print("The output hex value for the entire message is:\n%s\n" % file2.read())
+file2.close()
+

AESencryptfunc.py

@@ -26,7 +26,7 @@ def subbyte(myhexstring):
 
     lookuptable=[part0,part1,part2,part3,part4,part5,part6,part7,part8,part9,part10,part11,part12,part13,part14,part15]
 
-    print("The string size is ", len(myhexstring), " and the loop will run", math.ceil(len(myhexstring)/2), " times." )
+    #print("The string size is ", len(myhexstring), " and the loop will run", math.ceil(len(myhexstring)/2), " times." )
     for loop in range(0, math.ceil(len(myhexstring)/2) ):
         x = ""
         y = ""

message.txt

@@ -1,10 +1,8 @@
-hi wats up this is a test to see if it works. this message will be quite long. its only to be sure that its works completely
-i really want it to work.
-if not
-idk wat ill do
->0.0>
-this is my lifes work
-shaking intensifies
-hands sweaty
-knees weak
-Panic
+my name is katia
+and i like
+big dog noses
+and i cannot lie
+not other dogs can deny
+when a dog walks in
+with itty bitty waist
+and round nose in my face

output.txt

@@ -1 +1 @@
-8d7595f9499e787e998d39df57beaaadf34ac83183b19613835a63d0887b4b17c58351051594ee15c7bf388cbd601970d3ee27dbc480a910ff5ba8dfe540093bc426cae5cf357b3f72ecd7dcf12e4c3307f869f0ba5a3bf82951e5f1220cab1303e9ea3b91b7eae74bfa371992ae542eb5ceed62aaef6f0a6635868b2d38eb1c22dde4b6072f25d3030e11296201cfb5c429dea8d23347f262bcf8b106a0f7ff7e20a529e99cbbfdca4855f8f7564d4d00fe8c169d72ba0fda130f81b86132c86e30182357e8a3ff404b01b0592fbc283e20900ea1223d36ff2eeeaa2d1c947b84bcf01111cabb358012d42e133b6624c1a267f513b0241a2ac5c1b07feff227
+fae009acc4b24b17e5fdc13935ba6bb78b3ae40d6149c773505180aa1db892324810562d531c6b79f33e81843ed2d5311031a7cf1b69af856968ac4cb7c5d31963b2534759f5246f2537bd7ad145aa73e77c2a7eddf95c122cfec2e1fa9a46d982f2d543296edd343e16de093ef464112d8fd82191cdff98742d9016af73adc4be20026de125f00dffc0c61a9ca84a913dfb5cc7db95b3cdba09edba104bfad4