1 files changed, 105 insertions, 0 deletions

diff --git a/pwmanager/pwmanager/selftest.cpp b/pwmanager/pwmanager/selftest.cpp
new file mode 100644
index 0000000..45179d8
--- a/dev/null
+++ b/pwmanager/pwmanager/selftest.cpp
@@ -0,0 +1,105 @@
+/***************************************************************************
+ *                                                                         *
+ *   copyright (C) 2004 by Michael Buesch                                  *
+ *   email: mbuesch@freenet.de                                             *
+ *                                                                         *
+ *   This program is free software; you can redistribute it and/or modify  *
+ *   it under the terms of the GNU General Public License version 2        *
+ *   as published by the Free Software Foundation.                         *
+ *                                                                         *
+ ***************************************************************************/
+
+#include "selftest.h"
+#include "pwmexception.h"
+#include "sha1.h"
+#include "blowfish.h"
+#include "base64.h"
+
+#include <pthread.h>
+
+
+static SelfTest *st = 0;
+static pthread_mutex_t st_mutex = PTHREAD_MUTEX_INITIALIZER;
+
+
+SelfTest::SelfTest()
+{
+        connect(&schedTimer, SIGNAL(timeout()),
+                this, SLOT(doSelfTest()));
+}
+
+SelfTest::~SelfTest()
+{
+}
+
+void SelfTest::run()
+{
+        //printDebug("running self-test...");
+        if (unlikely(!Sha1::selfTest())) {
+                failed("SHA1");
+                return;
+        }
+        if (unlikely(!Blowfish::selfTest())) {
+                failed("BLOWFISH");
+                return;
+        }
+        if (unlikely(!Base64::selfTest())) {
+                failed("BASE64");
+                return;
+        }
+        //printDebug("self-test done.");
+        if (!pthread_mutex_trylock(&st_mutex)) {
+                /* only cancel (delete) this thread, if there's currently
+                 * noone else doing this job. Otherwise we will deadlock.
+                 * we use a timer here, to do some kind of context-switch.
+                 */
+                QTimer::singleShot(0, st, SLOT(doCancel()));
+        }
+}
+
+void SelfTest::failed(const char *algo)
+{
+        string msg("\n\n\n"
+                   "PwManager FATAL ERROR:\n"
+                   "Self-test for algorithm \"");
+        msg += algo;
+        msg += "\" failed!\n\n";
+        msg += "It's likely to be either a PwManager BUG or "
+               "bad hardware in your machine. Please contact the "
+               "maintainer of PwManager for this issue.";
+        std::cerr << msg << std::endl;
+        ::exit(1);
+}
+
+void SelfTest::schedule()
+{
+        if (pthread_mutex_lock(&st_mutex))
+                return;
+        if (st)
+                goto out;
+        st = new SelfTest;
+        st->startTimer();
+out:
+        pthread_mutex_unlock(&st_mutex);
+}
+
+void SelfTest::cancel()
+{
+        if (pthread_mutex_lock(&st_mutex))
+                return;
+        if (!st) {
+                pthread_mutex_unlock(&st_mutex);
+                return;
+        }
+        st->doCancel();
+}
+
+void SelfTest::doCancel()
+{
+        st->stopTimer();
+        st->wait();
+        delete_and_null(st);
+        pthread_mutex_unlock(&st_mutex);
+}
+
+#include "selftest.moc"