fix: Подкорректировал работу с исключениями

McEliece_console.py

@@ -1,5 +1,3 @@
-#pyinstaller -F -i "icon.ico" McEliece_console_v2.py
-
 import hashlib
 import getpass
 
@@ -9,8 +7,8 @@ def main():
 def safe_start():
     try:
         start_menu()
-    except:
-        print("\nUnknown error (maybe ctrl+c), emergency exit!")
+    except (KeyboardInterrupt, EOFError):
+        print("\nExiting...")
 
 def start_menu():
     f = True
@@ -29,7 +27,7 @@ def start_menu():
     input()
 
 def menu():
-    import cryptosystem_core_v2 as ME_core
+    import cryptosystem_core as ME_core
     core = ME_core.McEliece_core()
     global_info = "All files are interpreted as raw bytes and must be located in the directory with this executable file.\nDefault filenames with .bin extension: pubkey, privkey_S, privkey_p, plaintext, ciphertext, ciphered_string.\nDon't forget to import (or generate) keys before encryption/decryption and after changing config!\nYou can restore any one key from two another (don't forget to import before).\n"
     info = "Menu numbers: 0 = exit, s = print short info, h = print this info, g = print global info, c = change config;\n1 = generate keys, 10 = unsafe generate keys (seed = hash(password)),\n11 = export pubkey, 12 = export privkey_S, 13 = export privkey_p,\n14 = import pubkey, 15 = import privkey_S, 16 = import privkey_p,\n17 = restore pubkey, 18 = restore privkey_S, 19 = restore privkey_p;\n2 = encrypt,\n21 = encrypt non-default filename, 22 = encrypt string from keyboard, 23 = encrypt hided string.\n3 = decrypt,\n31 = decrypt non-default filename, 32 = decrypt string and show on screen\n"

McEliece_tests.py

@@ -0,0 +1,50 @@
+from cryptosystem_core import *
+
+def main(): #comment "return" for testing
+    core = McEliece_core()
+    n, k = core.get_config()
+    print(n, k)
+    core.change_config(5, 3) #check comments in class implementation to understand possible values
+    n, k = core.get_config()
+    print(n, k)
+    print()
+    core.generate_keys() #true random
+    print(core.get_pubkey())
+    print(core.get_privkey_S())
+    print(core.get_privkey_p())
+    print()
+    core.generate_keys(sum([ord(i) for i in list("password")])) #very simple seed
+    G = core.get_pubkey()
+    S = core.get_privkey_S()
+    p = core.get_privkey_p()
+    print(G)
+    print(S)
+    print(p)
+    print()
+    core.change_config(5, 3) #unset all keys inside core
+    core.set_privkey_S(S)
+    core.set_privkey_p(p)
+    core.restore_pubkey()
+    print(core.get_pubkey() == G)
+    core.change_config(5, 3) #unset all keys inside core
+    core.set_pubkey(G)
+    core.set_privkey_p(p)
+    core.restore_privkey_S()
+    print(core.get_privkey_S() == S)
+    core.change_config(5, 3) #unset all keys inside core
+    core.set_pubkey(G)
+    core.set_privkey_S(S)
+    core.restore_privkey_p()
+    print(core.get_privkey_p() == p)
+    print()
+    for j in range(2):
+        text = [i + 1 for i in range(5)]
+        msg = core.encrypt(text)
+        print(msg)
+        text = core.decrypt(msg)
+        print(text)
+        print()
+    print("All tests are finished!")
+
+if __name__ == "__main__":
+    main()

cryptosystem_core.py

@@ -11,56 +11,6 @@ import numpy as np
 import galois
 import random
 
-#import cryptosystem_core_v2 as ME_core
-#core = ME_core.McEliece_core()
-
-def main(): #comment "return" for testing
-    return
-    core = McEliece_core()
-    n, k = core.get_config()
-    print(n, k)
-    core.change_config(5, 3) #check comments in class implementation to understand possible values
-    n, k = core.get_config()
-    print(n, k)
-    print()
-    core.generate_keys() #true random
-    print(core.get_pubkey())
-    print(core.get_privkey_S())
-    print(core.get_privkey_p())
-    print()
-    core.generate_keys(sum([ord(i) for i in list("password")])) #very simple seed
-    G = core.get_pubkey()
-    S = core.get_privkey_S()
-    p = core.get_privkey_p()
-    print(G)
-    print(S)
-    print(p)
-    print()
-    core.change_config(5, 3) #unset all keys inside core
-    core.set_privkey_S(S)
-    core.set_privkey_p(p)
-    core.restore_pubkey()
-    print(core.get_pubkey() == G)
-    core.change_config(5, 3) #unset all keys inside core
-    core.set_pubkey(G)
-    core.set_privkey_p(p)
-    core.restore_privkey_S()
-    print(core.get_privkey_S() == S)
-    core.change_config(5, 3) #unset all keys inside core
-    core.set_pubkey(G)
-    core.set_privkey_S(S)
-    core.restore_privkey_p()
-    print(core.get_privkey_p() == p)
-    print()
-    for j in range(2):
-        text = [i + 1 for i in range(5)]
-        msg = core.encrypt(text)
-        print(msg)
-        text = core.decrypt(msg)
-        print(text)
-        print()
-    print("All tests are finished!")
-
 class McEliece_core:
     def __init__(self):
         self._order = 256 #p^m = 2**8; encryption of each byte
@@ -80,7 +30,7 @@ class McEliece_core:
                 raise Exception()
             rs = galois.ReedSolomon(n, k, field = self._GF)
         except:
-            raise Exception()
+            raise
         else:
             self._n = n
             self._k = k
@@ -115,7 +65,7 @@ class McEliece_core:
             G = [G[i - self._n : i] for i in range(self._n, self._n * self._k + self._n, self._n)]
             G = self._GF(G)
         except:
-            raise Exception()
+            raise
         else:
             self._G = G
     def set_privkey_S(self, S):
@@ -124,7 +74,7 @@ class McEliece_core:
             S = self._GF(S)
             S_inv = np.linalg.inv(S)
         except:
-            raise Exception()
+            raise
         else:
             self._S = S
             self._S_inv = S_inv
@@ -146,7 +96,7 @@ class McEliece_core:
         try:
             S_inv = np.linalg.inv(S)
         except:
-            raise Exception()
+            raise
         self._S = S
         self._S_inv = S_inv
     def restore_privkey_p(self):
@@ -184,14 +134,14 @@ class McEliece_core:
             out += self._encrypt_one(text)
             return out
         except:
-            raise Exception()
+            raise
     def decrypt(self, msg):
         try:
             msg = [msg[i - self._n : i] for i in range(self._n, len(msg) + self._n, self._n)]
             msg = [self._decrypt_one(self._GF(i)) for i in msg]
             return [i for j in msg for i in j]
         except:
-            raise Exception()
+            raise
 #End of top-level functions, please do NOT use functions below without understanding!
     def _generate_S(self):
         S = self._GF.Random((self._k, self._k))
@@ -249,7 +199,7 @@ class McEliece_core:
         try:
             msg = self._unpad_message(msg)
         except:
-            raise Exception()
+            raise
         return msg
     def _pad_message(self, msg):
         last_value = self._k - (len(msg) % self._k)
@@ -262,6 +212,3 @@ class McEliece_core:
             if msg[-i] != last_value:
                 raise Exception()
         return msg[: -last_value]
-
-if __name__ == "__main__":
-    main()