From d049739cd8d97be10ed8664a81bd1b5d2a73550c Mon Sep 17 00:00:00 2001
From: Rom Walton <rwalton@ssl.berkeley.edu>
Date: Thu, 31 Mar 2011 19:56:47 +0000
Subject: [PATCH] Quick Updates

svn path=/trunk/boinc/; revision=23304
---
 clientscr/screensaver_win.cpp | 14 ++++++--------
 1 file changed, 6 insertions(+), 8 deletions(-)

diff --git a/clientscr/screensaver_win.cpp b/clientscr/screensaver_win.cpp
index 4164e1eac7..ed481c0a1d 100644
--- a/clientscr/screensaver_win.cpp
+++ b/clientscr/screensaver_win.cpp
@@ -77,25 +77,24 @@ INT WINAPI WinMain(
         return retval;
     }
 
+    // Initialize Screensaver
     if (FAILED(hr = BOINCSS.Create(hInstance))) {
         BOINCSS.DisplayErrorMsg(hr);
         WSACleanup();
         return 0;
     }
 
+    // Run Screensaver
     retval = BOINCSS.Run();
     
     // Cleanup any existing screensaver objects and handles
+    BOINCTRACE("WinMain - Cleanup Screensaver Resources\n");
     BOINCSS.Cleanup();
 
     // Cleanup the Windows sockets interface.
+    BOINCTRACE("WinMain - Cleanup Winsock Resources\n");
     WSACleanup();
 
-    // Instruct the OS to terminate the screensaver by any
-    //   means nessassary.
-    fflush(NULL);
-    TerminateProcess(GetCurrentProcess(), retval);
-
     return retval;
 }
 
@@ -361,10 +360,12 @@ HRESULT CScreensaver::Run() {
 // Cleanup anything that needs cleaning.
 //
 HRESULT CScreensaver::Cleanup() {
+    BOINCTRACE("CScreensaver::Cleanup - Cleanup graphics application if running\n");
     if (m_hGraphicsApplication) {
         TerminateProcess(m_hGraphicsApplication, 0);
         m_hGraphicsApplication = NULL;
     }
+    BOINCTRACE("CScreensaver::Cleanup - Cleanup RPC client if running\n");
     if (rpc) {
         rpc->close();
         delete rpc;
@@ -380,11 +381,8 @@ HRESULT CScreensaver::Cleanup() {
 //
 HRESULT CScreensaver::DisplayErrorMsg(HRESULT hr) {
     TCHAR strMsg[512];
-
     GetTextForError(hr, strMsg, 512);
-
     MessageBox(m_hWnd, strMsg, m_strWindowTitle, MB_ICONERROR | MB_OK);
-
     return hr;
 }