3 files changed, 67 insertions, 12 deletions

diff --git a/pwmanager/pwmanager/pwmdoc.cpp b/pwmanager/pwmanager/pwmdoc.cpp
index e9906a4..f4a1636 100644
--- a/pwmanager/pwmanager/pwmdoc.cpp
+++ b/pwmanager/pwmanager/pwmdoc.cpp
@@ -11,7 +11,7 @@
 
 /***************************************************************************
  * copyright (C) 2004 by Ulf Schenk
- * This file is originaly based on version 2.0 of pwmanager
+ * This file is originaly based on version 1.1 of pwmanager
  * and was modified to run on embedded devices that run microkde
  *
  * $Id$
@@ -330,20 +330,56 @@ PwMDoc::~PwMDoc()
 PwMerror PwMDoc::saveDoc(char compress, const QString *file)
 {
 	PwMerror ret, e;
+	string serialized;
+	QFile f;
+	QString tmpFileMoved(QString::null);
+	bool wasDeepLocked;
+	QString savedFilename(filename);
+
 	if (!file) {
 		if (filename == "")
 			return e_filename;
-	} else {
+		if (isDeepLocked()) {
+		  /* We don't need to save any data.
+		   * It's already all on disk, because
+		   * we are deeplocked.
+		   */
+		  unsetDocStatFlag(DOC_STAT_DISK_DIRTY);
+		  ret = e_success;
+		  goto out;
+		}
+ 	} else {
 		if (*file == "" && filename == "")
 			return e_filename;
 		if (*file != "")
 			filename = *file;
 	}
 
-	bool wasDeepLocked = isDeepLocked();
+	wasDeepLocked = isDeepLocked();
 	if (wasDeepLocked) {
-		if (deepLock(false) != e_success)
-			return e_noPw;
+	  /* We are deeplocked. That means all data is already
+	   * on disk. BUT we need to do saving procedure,
+	   * because *file != savedFilename.
+	   * Additionally we need to tempoarly restore
+	   * the old "filename", because deepLock() references it.
+	   */
+	  QString newFilename(filename);
+	  filename = savedFilename;
+	  getDataChangedLock();
+	  e = deepLock(false);
+	  putDataChangedLock();
+	  filename = newFilename;
+	  switch (e) {
+	     case e_success:
+	       break;
+	     case e_wrongPw:
+	     case e_noPw:
+	       emitDataChanged(this);
+	       return e;
+	     default:
+	       emitDataChanged(this);
+	       return e_openFile;
+	  }
 	}
 
 	if (!isPwAvailable()) {
@@ -389,7 +425,6 @@ PwMerror PwMDoc::saveDoc(char compress, const QString *file)
 		if (!backupFile(filename))
 			return e_fileBackup;
 	}
-	QString tmpFileMoved(QString::null);
 	if (QFile::exists(filename)) {
 		/* Move the existing file to some tmp file.
 		 * When saving file succeeds, delete tmp file. Otherwise
@@ -408,8 +443,7 @@ PwMerror PwMDoc::saveDoc(char compress, const QString *file)
 				  + " failed!");
 		}
 	}
-	QFile f(filename);
-	string serialized;
+	f.setName(filename);
 	if (!f.open(IO_ReadWrite)) {
 		ret = e_openFile;
 		goto out_moveback;
@@ -476,8 +510,16 @@ PwMerror PwMDoc::saveDoc(char compress, const QString *file)
 		printWarn(string("chmod failed: ") + strerror(errno));
 	}
 	openDocList.edit(this, getTitle().latin1());
-	if (wasDeepLocked)
-		deepLock(true);
+	if (wasDeepLocked) {
+	  /* Do _not_ save the data with the deepLock()
+	   * call, because this will recurse
+	   * into saveDoc()
+	   */
+	  deepLock(true, false);
+	  /* We don't check return value here, because
+	   * it won't fail. See NOTE in deepLock()
+	   */
+	}	
 	if (tmpFileMoved != QString::null) {
 		// now remove the moved file.
 		if (!QFile::remove(tmpFileMoved)) {
@@ -1769,6 +1811,9 @@ bool PwMDoc::unlockAll_tempoary(bool revert)
 PwMerror PwMDoc::deepLock(bool lock, bool saveToFile)
 {
 	PwMerror ret;
+	/* NOTE: saveDoc() depends on this function to return
+	 *       e_success if saveToFile == false
+	 */
 
 	if (lock) {
 		if (isDeepLocked())
diff --git a/pwmanager/pwmanager/pwmdoc.h b/pwmanager/pwmanager/pwmdoc.h
index 535fb92..a6e5f58 100644
--- a/pwmanager/pwmanager/pwmdoc.h
+++ b/pwmanager/pwmanager/pwmdoc.h
@@ -11,7 +11,7 @@
 
 /***************************************************************************
  * copyright (C) 2004 by Ulf Schenk
- * This file is originaly based on version 2.0 of pwmanager
+ * This file is originaly based on version 1.1 of pwmanager
  * and was modified to run on embedded devices that run microkde
  *
  * $Id$
diff --git a/pwmanager/pwmanager/pwmdocui.cpp b/pwmanager/pwmanager/pwmdocui.cpp
index 7b8e0ee..6ddb6f5 100644
--- a/pwmanager/pwmanager/pwmdocui.cpp
+++ b/pwmanager/pwmanager/pwmdocui.cpp
@@ -273,6 +273,11 @@ bool PwMDocUi::saveDocUi(PwMDoc *doc)
 				   i18n("backup failed"));
 		doc->timer()->putLock(DocTimer::id_autoLockTimer);
 		return false;
+	} else if (ret == e_noPw ||
+		   ret == e_wrongPw ||
+		   ret == e_openFile) {
+	            doc->timer()->putLock(DocTimer::id_autoLockTimer);
+		    return false;
 	} else if (ret != e_success) {
 		KMessageBox::error(currentView,
 				   i18n("Error: Couldn't write to file.\n"
@@ -318,7 +323,12 @@ bool PwMDocUi::saveAsDocUi(PwMDoc *doc)
 		fn += ".pwm";
 
 	PwMerror ret = doc->saveDoc(conf()->confGlobCompression(), &fn);
-	if (ret != e_success) {
+	if (ret == e_noPw ||
+	    ret == e_wrongPw ||
+	    ret == e_openFile) {
+	  doc->timer()->putLock(DocTimer::id_autoLockTimer);
+	  return false;
+	} else if (ret != e_success) {
 		KMessageBox::error(currentView,
 				   i18n("Error: Couldn't write to file.\n"
 					"Please check if you have permission to\n"