even length components and byte length please

2026-04-09 17:38:07 +02:00
parent 596ba4753e
commit c7c965710e
1 changed files with 9 additions and 0 deletions
@@ -19,6 +19,15 @@ def main(argv):
        elif opt in ("-c3","--comp3"):
            comp3 = str(arg)
    
+    if len(comp1) != len(comp2) or str(len(comp1)) != str(len(comp3)):
+        print("All components should be of the same length")
+        sys.exit(2)
+    
+    if len(comp1) not in (16, 32, 64, 128, 256):
+        print("All components should be of byte length")
+        sys.exit(2)
+
+