py: added asym en/decryption, fine tuned test script

bindings/py/test.py

@@ -1,30 +1,89 @@
 #!/usr/local/bin/python
 
+from sys import argv, stdout
 from pypcp import *
 from pprint import pprint
 
-# import secret key
-zbobsec = open("../../tests/key-bobby-sec", "r").read()
-bobsec = Key(encoded=zbobsec, passphrase="b")
-#bobsec.dump()
+orig = "hello world"
 
-# import public key
-zalipub = open("../../tests/key-alicia-pub", "r").read()
-alipub = PublicKey(encoded=zalipub)
-#alipub.dump()
 
-# always required
-ctx = Context()
+def importkey(filename, passwd=None, secret=True, public=False):
+    raw = open(filename, "r").read()
+    if secret and not public:
+        key = Key(encoded=raw, passphrase=passwd)
+        return key
+    else:
+        key = PublicKey(encoded=raw)
+        return key
+
+def asym():
+    # import keys
+    bobSec = importkey("../../tests/key-bobby-sec", "b")
+    bobPub = importkey("../../tests/key-bobby-pub", public=True)
+    aliSec = importkey("../../tests/key-alicia-sec", "a")
+    aliPub = importkey("../../tests/key-alicia-pub", public=True)
+
+    # one context for each
+    ctxA = Context()
+    ctxB = Context()
+
+    # prepare ctx' for crypto
+    ctxA.addkey(aliSec)
+    ctxA.recipients(bobPub)
+
+    ctxB.addkey(bobSec)
+    ctxB.recipients(aliPub)
+
+    # Alice encrypt => Bob
+    # FIXME: if passed as is, then it's empty later on
+    encrypted = ctxA.encrypt(source="%s" % orig, armor=True)
+    #print "encrypted:\n%s" % encrypted
+
+    # Bob decrypt from Alice
+    clear = ctxB.decrypt(source=encrypted)
+    #print "clear: %s" % clear
+
+    if clear == orig:
+        return True
+    else:
+        return False
+
+
+def sym():
+    # always required
+    ctx = Context()
+
+    # symmetric encryption (self mode)
+    encrypted = ctx.encrypt(source="%s" % orig, passphrase="x", armor=True)
+    
+    # decrypt
+    clear = ctx.decrypt(source=encrypted, passphrase="x")
+    
+    if clear == orig:
+        return True
+    else:
+        return False
+
+
+
+def defun(name):
+    if not globals()[name]():
+        print "%s: failed" % name
+        exit(1)
+    else:
+        print "%s: ok" % name
+        return True
+
+
+if len(argv) == 2:
+    if defun(argv[1]):
+        exit(0)
+else:
+    # execute all
+    for func in ["asym", "sym"]:
+        defun(func)
+    exit(0)
+
 
-# symmetric encryption (self mode)
-sencrypted = ctx.encrypt(string="hello world", passphrase="x", armor=True)
-print sencrypted
 
-# asymmetric encryption bob => alice
-ctx.addkey(bobsec)
-ctx.recipients(alipub)
-aencrypted = ctx.encrypt(string="hello world", armor=True, sign=False)
-#print aencrypted
 
-sclear = ctx.decrypt(string=sencrypted, passphrase="x")
-print sclear