From 373bf41ff3d2710e66cda16e4b05d7a01728236c Mon Sep 17 00:00:00 2001
From: oepen <oepen@96b47cad-a561-4643-ad3b-153ac7d7599c>
Date: Thu, 1 Sep 2011 14:04:36 +0000
Subject: [PATCH] - Remember the password when loading a card from disk and
 reuse it when saving the card. - Added a bit of documentation

git-svn-id: https://vsmartcard.svn.sourceforge.net/svnroot/vsmartcard@512 96b47cad-a561-4643-ad3b-153ac7d7599c
---
 .../vpicc/virtualsmartcard/CardGenerator.py   | 25 +++++++++++--------
 1 file changed, 15 insertions(+), 10 deletions(-)

diff --git a/virtualsmartcard/src/vpicc/virtualsmartcard/CardGenerator.py b/virtualsmartcard/src/vpicc/virtualsmartcard/CardGenerator.py
index ca33c66..9e314fc 100644
--- a/virtualsmartcard/src/vpicc/virtualsmartcard/CardGenerator.py
+++ b/virtualsmartcard/src/vpicc/virtualsmartcard/CardGenerator.py
@@ -48,6 +48,7 @@ class CardGenerator(object):
         self.type = card_type
         self.mf = mf
         self.sam = sam
+        self.password = None
     
     def __generate_iso_card(self):
         default_pin = "1234"
@@ -62,6 +63,10 @@ class CardGenerator(object):
         self.sam.set_MF(self.mf)
            
     def __generate_ePass(self):
+        """Generate the MF and SAM of an ICAO passport. This method is 
+        responsible for generating the filesystem and filling it with content.
+        Therefore it must interact with the user by prompting for the MRZ and
+        optionally for the path to a photo."""
         from PIL import Image
             
         #TODO: Sanity checks
@@ -168,28 +173,28 @@ class CardGenerator(object):
             self.sam = sam
         
     
-    def loadCard(self, filename, password=None):
+    def loadCard(self, filename):
         """Load a card from disk"""
         db = anydbm.open(filename, 'r')
         
-        if password is None:
-            password = getpass.getpass("Please enter your password:")
+        if self.password is None:
+            self.password = getpass.getpass("Please enter your password:")
         
-        serializedMF = read_protected_string(db["mf"], password)
-        serializedSAM = read_protected_string(db["sam"], password)
+        serializedMF = read_protected_string(db["mf"], self.password)
+        serializedSAM = read_protected_string(db["sam"], self.password)
         self.sam = loads(serializedSAM)
         self.mf = loads(serializedMF)
         self.type = db["type"]
             
-    def saveCard(self, filename, password=None):
+    def saveCard(self, filename):
         """Save the currently running card to disk"""
-        if password is None:
+        if self.password is None:
             passwd1 = getpass.getpass("Please enter your password:")
             passwd2 = getpass.getpass("Please retype your password:")
             if (passwd1 != passwd2):
                 raise ValueError, "Passwords did not match. Will now exit"
             else:
-                password = passwd1
+                self.password = passwd1
         
         if self.mf == None or self.sam == None:
             raise ValueError, "Card Generator wasn't set up properly" +\
@@ -197,8 +202,8 @@ class CardGenerator(object):
         
         mf_string = dumps(self.mf)
         sam_string = dumps(self.sam)
-        protectedMF = protect_string(mf_string, password)
-        protectedSAM = protect_string(sam_string, password)
+        protectedMF = protect_string(mf_string, self.password)
+        protectedSAM = protect_string(sam_string, self.password)
         
         db = anydbm.open(filename, 'c')
         db["mf"] = protectedMF