summaryrefslogtreecommitdiff
path: root/migrationanalysis/src/msokill/msokill.cpp
blob: 0b2e9376807689254623f68f83b084cddafb1be0 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
/*************************************************************************
 *
 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
 * 
 * Copyright 2008 by Sun Microsystems, Inc.
 *
 * OpenOffice.org - a multi-platform office productivity suite
 *
 * $RCSfile: msokill.cpp,v $
 *
 * This file is part of OpenOffice.org.
 *
 * OpenOffice.org is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Lesser General Public License version 3
 * only, as published by the Free Software Foundation.
 *
 * OpenOffice.org is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Lesser General Public License version 3 for more details
 * (a copy is included in the LICENSE file that accompanied this code).
 *
 * You should have received a copy of the GNU Lesser General Public License
 * version 3 along with OpenOffice.org.  If not, see
 * <http://www.openoffice.org/license.html>
 * for a copy of the LGPLv3 License.
 *
 ************************************************************************/

/*
 * Description: Put MSO in a state where it can be closed using
 *              automation or kill it completely
 */

#include "stdafx.h"
#include <stdio.h>


void KillOffice();
BOOL KillAppFromWindow(HWND hWnd, char *appName);
BOOL CloseActiveDialogs();
void printUsage();

//Callbacks used in closing
BOOL CALLBACK CloseOfficeDlgProc(HWND hwndChild, LPARAM lParam);
BOOL CALLBACK CountOfficeDlgProc(HWND hwndChild, LPARAM lParam);

//Global counters for number of windows found
int gWDDlgCount = 0;
int gXLDlgCount = 0;
int gPPDlgCount = 0;

//Dialog window class names for excel, powerpoint and word
//These are "Best guess" dialog names
const char *pWordDlg2k  = "bosa_sdm_Microsoft Word 9.0";
const char *pWordDlg2k3 = "bosa_sdm_Microsoft Office Word";
const char *pXLDlg2k    = "bosa_sdm_XL9";
const char *pPPDlg2k    = "#32770";
const char *pXLDlg2k3   = "bosa_sdm_XL9";
const char *pPPDlg2k3   = "#32770";
const char *pGenMSODlg  = "bosa_sdm_Mso96";
//consider adding - bosa_sdm_Mso96

//Command Line Argument constants
const char *ARG_HELP  = "--help";
const char *ARG_KILL  = "--kill";
const char *ARG_CLOSE = "--close";

//Window class names for MSO apps - if we need to look at other office instances
//then this list would need to be expanded
#define NUM_WINDOWCLASSNAMES 4
char *wndClassName[NUM_WINDOWCLASSNAMES] = {"OpusApp", "XLMAIN", "PP9FrameClass", "PP10FrameClass"};

int main(int argc, char* argv[])
{
    if (argc < 2) {
        printUsage();
        return 0;
    }

    if (strcmpi(argv[1], ARG_HELP) == 0) {
        printUsage();
        return 0;
    }

    if (strcmpi(argv[1], ARG_KILL) == 0) {
        KillOffice();
        return 0;
    }

    if (strcmpi(argv[1], ARG_CLOSE) == 0) {
        CloseActiveDialogs();
        return 0;
    }

    return 0;
}

/*--------------------------------------------------------------
  Find the MSO window if it is available and explictly kill it
  MSO apps in this case are Excel, Word and PP
  Use FindWindow Win32 API to detect if they are available

  -------------------------------------------------------------*/
void KillOffice() {
    HWND hWnd;

    for (int i=0;i<NUM_WINDOWCLASSNAMES;i++) {
        int j = 0;
        while (((hWnd = FindWindow(wndClassName[i], NULL )) != NULL) && (j < 10)) {
            KillAppFromWindow(hWnd, wndClassName[i]);
            j++;
        }
    }
}

/*--------------------------------------------------------------
  Using window handle, get process handle and try to kill the
  app. This may not be successful if you do not have enough
  privileges to kill the app.

  --------------------------------------------------------------*/
BOOL KillAppFromWindow(
    HWND hWnd,
    char *
#ifdef _DEBUG
    appName
#endif
)
{
    BOOL bRet = TRUE;

    if(hWnd == NULL) {
        //The app doesn't appear to be running
#ifdef _DEBUG
        printf("App %s: window not found.\n,", appName);
#endif
        bRet = FALSE;
    } else {        
        DWORD pid;  // Variable to hold the process ID.
        DWORD dThread;  // Variable to hold (unused) thread ID.
        dThread = GetWindowThreadProcessId(hWnd, &pid);
        HANDLE hProcess; // Handle to existing process
        
        hProcess = OpenProcess(SYNCHRONIZE | PROCESS_ALL_ACCESS, TRUE, pid);
        if (hProcess == NULL) {
#ifdef _DEBUG
            printf("App %s : Failed to get process handle",appName);
#endif
            bRet = FALSE;
        } else {
            if (!TerminateProcess(hProcess, 0)) {
                LPTSTR lpMsgBuf;
                FormatMessage( 
                    FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_SYSTEM | FORMAT_MESSAGE_IGNORE_INSERTS,
                    NULL, GetLastError(), MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT),
                    (LPTSTR) &lpMsgBuf, 0, NULL );
                printf("%s\n", lpMsgBuf);
                LocalFree( lpMsgBuf );
                bRet = FALSE;
            }
#ifdef _DEBUG
            else {
                printf("Kill %s appears to be successful.\n", appName);
            }
#endif
        }
    }
    return bRet;
}

/*--------------------------------------------------------------
  Close the dialogs if possible based on their window class
  Use the EnumChildWindows win32 api for this
  --------------------------------------------------------------*/
BOOL CloseActiveDialogs() {
    char buff[1024];

    gWDDlgCount = 0;
    gXLDlgCount = 0;
    gPPDlgCount = 0;

    EnumChildWindows(GetDesktopWindow(), CloseOfficeDlgProc, (LPARAM) 0);
    sprintf(buff, "Word: %d\tExcel: %d\tPP: %d", gWDDlgCount, gXLDlgCount, gPPDlgCount);
    return TRUE;
}

/*--------------------------------------------------------------
  Callback for EnumChildWindows that sends close message to 
  any dialogs that match window class of MSO dialogs

  --------------------------------------------------------------*/
BOOL CALLBACK CloseOfficeDlgProc(HWND hwndChild, LPARAM) 
{ 
    //bosa_sdm_Microsoft Word 9.0
    //bosa_sdm_XL9
    //#32770 (Dialog)

     char szBuff[4096];
    if (GetClassName(hwndChild, szBuff, 4096) == 0) {
        
    } else {
        if ((strcmpi(szBuff, pWordDlg2k) == 0) || (strcmpi(szBuff, pWordDlg2k3) == 0)) {
            gWDDlgCount++;
            SendMessage(hwndChild, WM_CLOSE, 0, 0);
        }
        if (strcmpi(szBuff, pXLDlg2k) == 0) {
            gXLDlgCount++;
            SendMessage(hwndChild, WM_CLOSE, 0, 0);
        }
        if (strcmpi(szBuff, pPPDlg2k) == 0) {
            gPPDlgCount++;
            SendMessage(hwndChild, WM_CLOSE, 0, 0);
        }
        if (strcmpi(szBuff, pGenMSODlg) == 0) {
            SendMessage(hwndChild, WM_CLOSE, 0, 0);
        }
    }
 
    return TRUE; 
} 


/*--------------------------------------------------------------
  Callback for EnumChildWindows that counts numnnber of  
  dialogs that match window class of MSO dialogs

  --------------------------------------------------------------*/
BOOL CALLBACK CountOfficeDlgProc(HWND hwndChild, LPARAM) 
{
     char szBuff[4096];
    if (GetClassName(hwndChild, szBuff, 4096) == 0) {
        
    } else {
        if ((strcmpi(szBuff, pWordDlg2k) == 0) || (strcmpi(szBuff, pWordDlg2k3) == 0)) {
            gWDDlgCount++;
        }
        if (strcmpi(szBuff, pXLDlg2k) == 0) {
            gXLDlgCount++;
        }
        if (strcmpi(szBuff, pPPDlg2k) == 0) {
            gPPDlgCount++;
        }
    }

    return TRUE; 
} 

/*--------------------------------------------------------------
  Simple usage message...
 
  -------------------------------------------------------------*/
void printUsage() {
    printf("Recovery Assistant Utility - try and put MSO apps in a recoverable state\n");
    printf("Copyright Sun Microsystems 2008\n");
    printf("Options:\n");
    printf("   --help : This message\n");
    printf("   --close: Attempt to close any open dialogs owned by \n");
    printf("            MSO apps so Application.Quit() can succeed\n");
    printf("   --kill : Kill any open MSO apps. Use with caution and only as a last resort\n\n");
}