d9fcec3fd68b020d1ccfd6fa5bee0a70a845cabd
[privoxy.git] / win32.c
1 const char win32_rcs[] = "$Id: win32.c,v 1.7 2002/03/24 12:03:47 jongfoster Exp $";
2 /*********************************************************************
3  *
4  * File        :  $Source: /cvsroot/ijbswa/current/win32.c,v $
5  *
6  * Purpose     :  Win32 User Interface initialization and message loop
7  *
8  * Copyright   :  Written by and Copyright (C) 2001-2002 members of
9  *                the Privoxy team.  http://www.privoxy.org/
10  *
11  *                Written by and Copyright (C) 1999 Adam Lock
12  *                <locka@iol.ie>
13  *
14  *                This program is free software; you can redistribute it 
15  *                and/or modify it under the terms of the GNU General
16  *                Public License as published by the Free Software
17  *                Foundation; either version 2 of the License, or (at
18  *                your option) any later version.
19  *
20  *                This program is distributed in the hope that it will
21  *                be useful, but WITHOUT ANY WARRANTY; without even the
22  *                implied warranty of MERCHANTABILITY or FITNESS FOR A
23  *                PARTICULAR PURPOSE.  See the GNU General Public
24  *                License for more details.
25  *
26  *                The GNU General Public License should be included with
27  *                this file.  If not, you can view it at
28  *                http://www.gnu.org/copyleft/gpl.html
29  *                or write to the Free Software Foundation, Inc., 59
30  *                Temple Place - Suite 330, Boston, MA  02111-1307, USA.
31  *
32  * Revisions   :
33  *    $Log: win32.c,v $
34  *    Revision 1.7  2002/03/24 12:03:47  jongfoster
35  *    Name change
36  *
37  *    Revision 1.6  2002/03/16 21:53:28  jongfoster
38  *    VC++ Heap debug option
39  *
40  *    Revision 1.5  2002/03/04 23:47:30  jongfoster
41  *    - Rewritten, simpler command-line pre-parser
42  *    - not using raise(SIGINT) any more
43  *
44  *    Revision 1.4  2001/11/30 21:29:33  jongfoster
45  *    Fixing a warning
46  *
47  *    Revision 1.3  2001/11/16 00:46:31  jongfoster
48  *    Fixing compiler warnings
49  *
50  *    Revision 1.2  2001/07/29 19:32:00  jongfoster
51  *    Renaming _main() [mingw32 only] to real_main(), for ANSI compliance.
52  *
53  *    Revision 1.1.1.1  2001/05/15 13:59:08  oes
54  *    Initial import of version 2.9.3 source tree
55  *
56  *
57  *********************************************************************/
58 \f
59
60 #include "config.h"
61
62 #ifdef _WIN32
63
64 #include <stdio.h>
65
66 #include "project.h"
67 #include "jcc.h"
68 #include "miscutil.h"
69
70 /* Uncomment this if you want to build Win32 as a console app */
71 /* #define _WIN_CONSOLE */
72
73 #include <windows.h>
74
75 #include <stdarg.h>
76 #include <process.h>
77
78 #if defined(_WIN32) && defined(_MSC_VER) && defined(_DEBUG)
79 /* Visual C++ Heap debugging */
80 #include <crtdbg.h>
81 #endif /* defined(_WIN32) && defined(_MSC_VER) && defined(_DEBUG) */
82
83 #include "win32.h"
84
85 const char win32_h_rcs[] = WIN32_H_VERSION;
86
87 const char win32_blurb[] =
88 "Privoxy version " VERSION " for Windows\n"
89 "Copyright (C) 2000-2002 by members of the Privoxy Team\n"
90 "Copyright (C) 1997-8 by Junkbusters Corp.\n"
91 "This is free software; it may be used and copied under the\n"
92 "GNU General Public License: http://www.gnu.org/copyleft/gpl.html .\n"
93 "This program comes with ABSOLUTELY NO WARRANTY OF ANY KIND.\n"
94 "\n"
95 "For information about how to to configure the proxy and your browser, see\n"
96 "        " HOME_PAGE_URL "\n"
97 "\n";
98
99 #ifdef _WIN_CONSOLE
100
101 int hideConsole     = 0;
102
103 #else
104
105 HINSTANCE g_hInstance;
106 int g_nCmdShow;
107
108 static void  __cdecl UserInterfaceThread(void *);
109
110 #endif
111
112
113 /*********************************************************************
114  *
115  * Function    :  WinMain
116  *
117  * Description :  M$ Windows "main" routine:
118  *                parse the `lpCmdLine' param into main's argc and argv variables,
119  *                start the user interface thread (for the systray window), and
120  *                call main (i.e. patch execution into normal startup).
121  *
122  * Parameters  :
123  *          1  :  hInstance = instance handle of this execution
124  *          2  :  hPrevInstance = instance handle of previous execution
125  *          3  :  lpCmdLine = command line string which started us
126  *          4  :  nCmdShow = window show value (MIN, MAX, NORMAL, etc...)
127  *
128  * Returns     :  `main' never returns, so WinMain will also never return.
129  *
130  *********************************************************************/
131 int WINAPI WinMain(HINSTANCE hInstance, HINSTANCE hPrevInstance, LPSTR lpCmdLine, int nCmdShow)
132 {
133    int i;
134    int res;
135    int argc = 1;
136    const char *argv[3];
137    char szModule[MAX_PATH+1];
138 #ifndef _WIN_CONSOLE
139    HANDLE hInitCompleteEvent = NULL;
140 #endif
141
142
143 #if defined(_WIN32) && defined(_MSC_VER) && defined(_DEBUG)
144 #if 0
145    /* Visual C++ Heap debugging */
146
147    /* Get current flag*/
148    int tmpFlag = _CrtSetDbgFlag( _CRTDBG_REPORT_FLAG );
149
150    /* Turn on leak-checking bit */
151    tmpFlag |= _CRTDBG_ALLOC_MEM_DF | _CRTDBG_LEAK_CHECK_DF | _CRTDBG_CHECK_ALWAYS_DF;
152
153    /* Turn off CRT block checking bit */
154    tmpFlag &= ~(_CRTDBG_CHECK_CRT_DF | _CRTDBG_DELAY_FREE_MEM_DF);
155
156    /* Set flag to the new value */
157    _CrtSetDbgFlag( tmpFlag );
158 #endif
159 #endif /* defined(_WIN32) && defined(_MSC_VER) && defined(_DEBUG) */
160
161    /*
162     * Cheat in parsing the command line.  We only ever have at most one
163     * paramater, which may optionally be specified inside double quotes.
164     */
165
166    if (lpCmdLine != NULL)
167    {
168       /* Make writable copy */
169       lpCmdLine = strdup(lpCmdLine);
170    }
171    if (lpCmdLine != NULL)
172    {
173       chomp(lpCmdLine);
174       i = strlen(lpCmdLine);
175       if ((i >= 2) && (lpCmdLine[0] == '\"') && (lpCmdLine[i - 1] == '\"'))
176       {
177          lpCmdLine[i - 1] = '\0';
178          lpCmdLine++;
179       }
180       if (lpCmdLine[0] == '\0')
181       {
182          lpCmdLine = NULL;
183       }
184    }
185
186    GetModuleFileName(hInstance, szModule, MAX_PATH);
187    argv[0] = szModule;
188    argv[1] = lpCmdLine;
189    argv[2] = NULL;
190    argc = ((lpCmdLine != NULL) ? 2 : 1);
191
192 #ifndef _WIN_CONSOLE
193    /* Create a user-interface thread and wait for it to initialise */
194    hInitCompleteEvent = CreateEvent(NULL, TRUE, FALSE, NULL);
195    g_hInstance = hInstance;
196    g_nCmdShow = nCmdShow;
197    _beginthread(UserInterfaceThread, 0, &hInitCompleteEvent);
198    WaitForSingleObject(hInitCompleteEvent, INFINITE);
199    DeleteObject(hInitCompleteEvent);
200 #endif
201
202 #ifdef __MINGW32__
203    res = real_main( argc, argv );
204 #else
205    res = main( argc, argv );
206 #endif
207
208    return res;
209
210 }
211
212 #endif
213
214 /*********************************************************************
215  *
216  * Function    :  InitWin32
217  *
218  * Description :  Initialise windows, setting up the console or windows as appropriate.
219  *
220  * Parameters  :  None
221  *
222  * Returns     :  N/A
223  *
224  *********************************************************************/
225 void InitWin32(void)
226 {
227    WORD wVersionRequested;
228    WSADATA wsaData;
229
230 #ifdef _WIN_CONSOLE
231    SetProcessShutdownParameters(0x100, SHUTDOWN_NORETRY);
232    if (hideConsole)
233    {
234       FreeConsole();
235    }
236 #endif
237    wVersionRequested = MAKEWORD(2, 0);
238    if (WSAStartup(wVersionRequested, &wsaData) != 0)
239    {
240 #ifndef _WIN_CONSOLE
241       MessageBox(NULL, "Cannot initialize WinSock library", "Privoxy Error", 
242          MB_OK | MB_ICONERROR | MB_TASKMODAL | MB_SETFOREGROUND | MB_TOPMOST);  
243 #endif
244       exit(1);
245    }
246
247 }
248
249
250 #ifndef _WIN_CONSOLE
251 #include <signal.h>
252 #include <assert.h>
253
254 #include "win32.h"
255 #include "w32log.h"
256
257
258 /*********************************************************************
259  *
260  * Function    :  UserInterfaceThread
261  *
262  * Description :  User interface thread.  WinMain will wait for us to set
263  *                the hInitCompleteEvent before patching over to `main'.
264  *                This ensures the systray window is active before beginning
265  *                operations.
266  *
267  * Parameters  :
268  *          1  :  pData = pointer to `hInitCompleteEvent'.
269  *
270  * Returns     :  N/A
271  *
272  *********************************************************************/
273 static void __cdecl UserInterfaceThread(void *pData)
274 {
275    MSG msg;
276    HANDLE hInitCompleteEvent = *((HANDLE *) pData);
277
278    /* Initialise */
279    InitLogWindow();
280    SetEvent(hInitCompleteEvent);
281
282    /* Enter a message processing loop */
283    while (GetMessage(&msg, (HWND) NULL, 0, 0))
284    {
285       TranslateMessage(&msg);
286       DispatchMessage(&msg);
287    }
288
289    /* Cleanup */
290    TermLogWindow();
291
292    /* Time to die... */
293    exit(0);
294
295 }
296
297
298 #endif
299
300
301 /*
302   Local Variables:
303   tab-width: 3
304   end:
305 */