code cleaning

3 files changed, 8 insertions, 10 deletions

diff --git a/plugins/DbChecker/src/main.cpp b/plugins/DbChecker/src/main.cpp
index 18ee5cc29b..89b5633957 100644
--- a/plugins/DbChecker/src/main.cpp
+++ b/plugins/DbChecker/src/main.cpp
@@ -64,13 +64,14 @@ static INT_PTR ServiceMode(WPARAM, LPARAM)
 	bServiceMode = true;
 	
 	opts.db = db_get_current();
+	db_setCurrent(nullptr);
 	wcsncpy_s(opts.filename, VARSW(L"%miranda_userdata%\\%miranda_profilename%.dat"), _TRUNCATE);
 
 	opts.dbChecker = opts.db->GetChecker();
 	if (opts.dbChecker == nullptr)
 		return SERVICE_FAILED;
 
-	DialogBox(g_plugin.getInst(), MAKEINTRESOURCE(IDD_WIZARD), nullptr, ProgressDlgProc);
+	DialogBox(g_plugin.getInst(), MAKEINTRESOURCE(IDD_WIZARD), nullptr, WizardDlgProc);
 	return (bLaunchMiranda) ? SERVICE_CONTINUE : SERVICE_FAILED;
 }
 
diff --git a/plugins/DbChecker/src/selectdb.cpp b/plugins/DbChecker/src/selectdb.cpp
index 66df5f7480..162a49af7c 100644
--- a/plugins/DbChecker/src/selectdb.cpp
+++ b/plugins/DbChecker/src/selectdb.cpp
@@ -166,6 +166,10 @@ INT_PTR CALLBACK SelectDbDlgProc(HWND hdlg, UINT message, WPARAM wParam, LPARAM
 	switch (message) {
 	case WM_INITDIALOG:
 		TranslateDialogDefault(hdlg);
+		if (bServiceMode) {
+			OpenDatabase(hdlg);
+			return TRUE;
+		}
 		{
 			HIMAGELIST hIml = ImageList_Create(GetSystemMetrics(SM_CXSMICON), GetSystemMetrics(SM_CYSMICON), ILC_COLOR32 | ILC_MASK, 3, 3);
 			ImageList_AddIcon(hIml, LoadIcon(g_plugin.getInst(), MAKEINTRESOURCE(IDI_PROFILEGREEN)));
diff --git a/plugins/DbChecker/src/worker.cpp b/plugins/DbChecker/src/worker.cpp
index 7911691107..379a702613 100644
--- a/plugins/DbChecker/src/worker.cpp
+++ b/plugins/DbChecker/src/worker.cpp
@@ -29,19 +29,17 @@ static void Finalize()
 
 void __cdecl WorkerThread(void *)
 {
-	int task, firstTime;
 	time_t ts = time(nullptr);
 
 	AddToStatus(STATUS_MESSAGE, TranslateT("Database worker thread activated"));
 
 	DWORD sp = 0;
-	firstTime = 0;
 
 	DBCHeckCallback callback;
 	callback.pfnAddLogMessage = AddToStatus;
 	opts.dbChecker->Start(&callback);
 
-	for (task = 0;;) {
+	for (int task = 0;; task++) {
 		if (callback.spaceProcessed / (callback.spaceUsed / 1000 + 1) > sp) {
 			sp = callback.spaceProcessed / (callback.spaceUsed / 1000 + 1);
 			SetProgressBar(sp);
@@ -52,8 +50,7 @@ void __cdecl WorkerThread(void *)
 			break;
 		}
 
-		int ret = opts.dbChecker->CheckDb(task, firstTime);
-		firstTime = 0;
+		int ret = opts.dbChecker->CheckDb(task);
 		if (ret == ERROR_OUT_OF_PAPER) {
 			Finalize();
 			AddToStatus(STATUS_MESSAGE, TranslateT("Elapsed time: %d sec"), time(nullptr) - ts);
@@ -63,10 +60,6 @@ void __cdecl WorkerThread(void *)
 				AddToStatus(STATUS_SUCCESS, TranslateT("All tasks completed successfully"));
 			break;
 		}
-		else if (ret == ERROR_NO_MORE_ITEMS) {
-			task++;
-			firstTime = 1;
-		}
 		else if (ret != ERROR_SUCCESS)
 			break;
 	}