ole32: Fix some leaks (coverity).
[wine] / programs / notepad / main.c
1 /*
2  *  Notepad
3  *
4  *  Copyright 2000 Mike McCormack <Mike_McCormack@looksmart.com.au>
5  *  Copyright 1997,98 Marcel Baur <mbaur@g26.ethz.ch>
6  *  Copyright 2002 Sylvain Petreolle <spetreolle@yahoo.fr>
7  *  Copyright 2002 Andriy Palamarchuk
8  *
9  * This library is free software; you can redistribute it and/or
10  * modify it under the terms of the GNU Lesser General Public
11  * License as published by the Free Software Foundation; either
12  * version 2.1 of the License, or (at your option) any later version.
13  *
14  * This library is distributed in the hope that it will be useful,
15  * but WITHOUT ANY WARRANTY; without even the implied warranty of
16  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
17  * Lesser General Public License for more details.
18  *
19  * You should have received a copy of the GNU Lesser General Public
20  * License along with this library; if not, write to the Free Software
21  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
22  *
23  */
24
25 #include <stdio.h>
26 #include <windows.h>
27 #include <commdlg.h>
28 #include <shellapi.h>
29 #include <shlwapi.h>
30
31 #include "main.h"
32 #include "dialog.h"
33 #include "notepad_res.h"
34 #include "wine/unicode.h"
35
36 NOTEPAD_GLOBALS Globals;
37 static ATOM aFINDMSGSTRING;
38 static RECT main_rect;
39
40 static const WCHAR notepad_reg_key[] = {'S','o','f','t','w','a','r','e','\\',
41                                         'M','i','c','r','o','s','o','f','t','\\','N','o','t','e','p','a','d','\0'};
42 static const WCHAR value_fWrap[]            = {'f','W','r','a','p','\0'};
43 static const WCHAR value_iPointSize[]       = {'i','P','o','i','n','t','S','i','z','e','\0'};
44 static const WCHAR value_iWindowPosDX[]     = {'i','W','i','n','d','o','w','P','o','s','D','X','\0'};
45 static const WCHAR value_iWindowPosDY[]     = {'i','W','i','n','d','o','w','P','o','s','D','Y','\0'};
46 static const WCHAR value_iWindowPosX[]      = {'i','W','i','n','d','o','w','P','o','s','X','\0'};
47 static const WCHAR value_iWindowPosY[]      = {'i','W','i','n','d','o','w','P','o','s','Y','\0'};
48 static const WCHAR value_lfCharSet[]        = {'l','f','C','h','a','r','S','e','t','\0'};
49 static const WCHAR value_lfClipPrecision[]  = {'l','f','C','l','i','p','P','r','e','c','i','s','i','o','n','\0'};
50 static const WCHAR value_lfEscapement[]     = {'l','f','E','s','c','a','p','e','m','e','n','t','\0'};
51 static const WCHAR value_lfItalic[]         = {'l','f','I','t','a','l','i','c','\0'};
52 static const WCHAR value_lfOrientation[]    = {'l','f','O','r','i','e','n','t','a','t','i','o','n','\0'};
53 static const WCHAR value_lfOutPrecision[]   = {'l','f','O','u','t','P','r','e','c','i','s','i','o','n','\0'};
54 static const WCHAR value_lfPitchAndFamily[] = {'l','f','P','i','t','c','h','A','n','d','F','a','m','i','l','y','\0'};
55 static const WCHAR value_lfQuality[]        = {'l','f','Q','u','a','l','i','t','y','\0'};
56 static const WCHAR value_lfStrikeOut[]      = {'l','f','S','t','r','i','k','e','O','u','t','\0'};
57 static const WCHAR value_lfUnderline[]      = {'l','f','U','n','d','e','r','l','i','n','e','\0'};
58 static const WCHAR value_lfWeight[]         = {'l','f','W','e','i','g','h','t','\0'};
59 static const WCHAR value_lfFaceName[]       = {'l','f','F','a','c','e','N','a','m','e','\0'};
60 static const WCHAR value_iMarginTop[]       = {'i','M','a','r','g','i','n','T','o','p','\0'};
61 static const WCHAR value_iMarginBottom[]    = {'i','M','a','r','g','i','n','B','o','t','t','o','m','\0'};
62 static const WCHAR value_iMarginLeft[]      = {'i','M','a','r','g','i','n','L','e','f','t','\0'};
63 static const WCHAR value_iMarginRight[]     = {'i','M','a','r','g','i','n','R','i','g','h','t','\0'};
64 static const WCHAR value_szHeader[]         = {'s','z','H','e','a','d','e','r','\0'};
65 static const WCHAR value_szFooter[]         = {'s','z','T','r','a','i','l','e','r','\0'};
66
67 /***********************************************************************
68  *
69  *           SetFileNameAndEncoding
70  *
71  *  Sets global file name and encoding (which is used to preselect original
72  *  encoding in Save As dialog, and when saving without using the Save As
73  *  dialog).
74  */
75 VOID SetFileNameAndEncoding(LPCWSTR szFileName, ENCODING enc)
76 {
77     lstrcpyW(Globals.szFileName, szFileName);
78     Globals.szFileTitle[0] = 0;
79     GetFileTitleW(szFileName, Globals.szFileTitle, sizeof(Globals.szFileTitle));
80     Globals.encFile = enc;
81 }
82
83 /******************************************************************************
84  *      get_dpi
85  *
86  * Get the dpi from registry HKCC\Software\Fonts\LogPixels.
87  */
88 DWORD get_dpi(void)
89 {
90     static const WCHAR dpi_key_name[] = {'S','o','f','t','w','a','r','e','\\','F','o','n','t','s','\0'};
91     static const WCHAR dpi_value_name[] = {'L','o','g','P','i','x','e','l','s','\0'};
92     DWORD dpi = 96;
93     HKEY hkey;
94
95     if (RegOpenKeyW(HKEY_CURRENT_CONFIG, dpi_key_name, &hkey) == ERROR_SUCCESS)
96     {
97         DWORD type, size, new_dpi;
98
99         size = sizeof(new_dpi);
100         if(RegQueryValueExW(hkey, dpi_value_name, NULL, &type, (LPBYTE)&new_dpi, &size) == ERROR_SUCCESS)
101         {
102             if(type == REG_DWORD && new_dpi != 0)
103                 dpi = new_dpi;
104         }
105         RegCloseKey(hkey);
106     }
107     return dpi;
108 }
109
110 /***********************************************************************
111  *
112  *           NOTEPAD_SaveSettingToRegistry
113  *
114  *  Save setting to registry HKCU\Software\Microsoft\Notepad.
115  */
116 static VOID NOTEPAD_SaveSettingToRegistry(void)
117 {
118     HKEY hkey;
119     DWORD disp;
120
121     if(RegCreateKeyExW(HKEY_CURRENT_USER, notepad_reg_key, 0, NULL,
122                 REG_OPTION_NON_VOLATILE, KEY_ALL_ACCESS, NULL, &hkey, &disp) == ERROR_SUCCESS)
123     {
124         DWORD data;
125         WINDOWPLACEMENT wndpl;
126
127         wndpl.length = sizeof(WINDOWPLACEMENT);
128         GetWindowPlacement(Globals.hMainWnd, &wndpl);
129         main_rect = wndpl.rcNormalPosition;
130
131 #define SET_NOTEPAD_REG(hkey, value_name, value_data) do { DWORD data = value_data; RegSetValueExW(hkey, value_name, 0, REG_DWORD, (LPBYTE)&data, sizeof(DWORD)); }while(0)
132         SET_NOTEPAD_REG(hkey, value_fWrap,            Globals.bWrapLongLines);
133         SET_NOTEPAD_REG(hkey, value_iWindowPosX,      main_rect.left);
134         SET_NOTEPAD_REG(hkey, value_iWindowPosY,      main_rect.top);
135         SET_NOTEPAD_REG(hkey, value_iWindowPosDX,     main_rect.right - main_rect.left);
136         SET_NOTEPAD_REG(hkey, value_iWindowPosDY,     main_rect.bottom - main_rect.top);
137         SET_NOTEPAD_REG(hkey, value_lfCharSet,        Globals.lfFont.lfCharSet);
138         SET_NOTEPAD_REG(hkey, value_lfClipPrecision,  Globals.lfFont.lfClipPrecision);
139         SET_NOTEPAD_REG(hkey, value_lfEscapement,     Globals.lfFont.lfEscapement);
140         SET_NOTEPAD_REG(hkey, value_lfItalic,         Globals.lfFont.lfItalic);
141         SET_NOTEPAD_REG(hkey, value_lfOrientation,    Globals.lfFont.lfOrientation);
142         SET_NOTEPAD_REG(hkey, value_lfOutPrecision,   Globals.lfFont.lfOutPrecision);
143         SET_NOTEPAD_REG(hkey, value_lfPitchAndFamily, Globals.lfFont.lfPitchAndFamily);
144         SET_NOTEPAD_REG(hkey, value_lfQuality,        Globals.lfFont.lfQuality);
145         SET_NOTEPAD_REG(hkey, value_lfStrikeOut,      Globals.lfFont.lfStrikeOut);
146         SET_NOTEPAD_REG(hkey, value_lfUnderline,      Globals.lfFont.lfUnderline);
147         SET_NOTEPAD_REG(hkey, value_lfWeight,         Globals.lfFont.lfWeight);
148         SET_NOTEPAD_REG(hkey, value_iMarginTop,       Globals.iMarginTop);
149         SET_NOTEPAD_REG(hkey, value_iMarginBottom,    Globals.iMarginBottom);
150         SET_NOTEPAD_REG(hkey, value_iMarginLeft,      Globals.iMarginLeft);
151         SET_NOTEPAD_REG(hkey, value_iMarginRight,     Globals.iMarginRight);
152 #undef SET_NOTEPAD_REG
153
154         /* Store the current value as 10 * twips */
155         data = MulDiv(abs(Globals.lfFont.lfHeight), 720 , get_dpi());
156         RegSetValueExW(hkey, value_iPointSize, 0, REG_DWORD, (LPBYTE)&data, sizeof(DWORD));
157
158         RegSetValueExW(hkey, value_lfFaceName, 0, REG_SZ, (LPBYTE)&Globals.lfFont.lfFaceName,
159                       lstrlenW(Globals.lfFont.lfFaceName) * sizeof(Globals.lfFont.lfFaceName[0]));
160
161         RegSetValueExW(hkey, value_szHeader, 0, REG_SZ, (LPBYTE)&Globals.szHeader,
162                       lstrlenW(Globals.szHeader) * sizeof(Globals.szHeader[0]));
163
164         RegSetValueExW(hkey, value_szFooter, 0, REG_SZ, (LPBYTE)&Globals.szFooter,
165                       lstrlenW(Globals.szFooter) * sizeof(Globals.szFooter[0]));
166
167         RegCloseKey(hkey);
168     }
169 }
170
171 /***********************************************************************
172  *
173  *           NOTEPAD_LoadSettingFromRegistry
174  *
175  *  Load setting from registry HKCU\Software\Microsoft\Notepad.
176  */
177 static VOID NOTEPAD_LoadSettingFromRegistry(void)
178 {
179     static const WCHAR systemW[] = { 'S','y','s','t','e','m','\0' };
180     HKEY hkey;
181     INT base_length, dx, dy;
182
183     base_length = (GetSystemMetrics(SM_CXSCREEN) > GetSystemMetrics(SM_CYSCREEN))?
184         GetSystemMetrics(SM_CYSCREEN) : GetSystemMetrics(SM_CXSCREEN);
185
186     dx = base_length * .95;
187     dy = dx * 3 / 4;
188     SetRect( &main_rect, 0, 0, dx, dy );
189
190     Globals.bWrapLongLines  = TRUE;
191     Globals.iMarginTop = 2500;
192     Globals.iMarginBottom = 2500;
193     Globals.iMarginLeft = 2000;
194     Globals.iMarginRight = 2000;
195     
196     Globals.lfFont.lfHeight         = -12;
197     Globals.lfFont.lfWidth          = 0;
198     Globals.lfFont.lfEscapement     = 0;
199     Globals.lfFont.lfOrientation    = 0;
200     Globals.lfFont.lfWeight         = FW_REGULAR;
201     Globals.lfFont.lfItalic         = FALSE;
202     Globals.lfFont.lfUnderline      = FALSE;
203     Globals.lfFont.lfStrikeOut      = FALSE;
204     Globals.lfFont.lfCharSet        = DEFAULT_CHARSET;
205     Globals.lfFont.lfOutPrecision   = OUT_DEFAULT_PRECIS;
206     Globals.lfFont.lfClipPrecision  = CLIP_DEFAULT_PRECIS;
207     Globals.lfFont.lfQuality        = DEFAULT_QUALITY;
208     Globals.lfFont.lfPitchAndFamily = FIXED_PITCH | FF_DONTCARE;
209     lstrcpyW(Globals.lfFont.lfFaceName, systemW);
210
211     LoadStringW(Globals.hInstance, STRING_PAGESETUP_HEADERVALUE,
212                 Globals.szHeader, ARRAY_SIZE(Globals.szHeader));
213     LoadStringW(Globals.hInstance, STRING_PAGESETUP_FOOTERVALUE,
214                 Globals.szFooter, ARRAY_SIZE(Globals.szFooter));
215
216     if(RegOpenKeyW(HKEY_CURRENT_USER, notepad_reg_key, &hkey) == ERROR_SUCCESS)
217     {
218         WORD  data_helper[MAX_PATH];
219         DWORD type, data, size;
220
221 #define QUERY_NOTEPAD_REG(hkey, value_name, ret) do { DWORD type, data; DWORD size = sizeof(DWORD); if(RegQueryValueExW(hkey, value_name, 0, &type, (LPBYTE)&data, &size) == ERROR_SUCCESS) if(type == REG_DWORD) ret = data; } while(0)
222         QUERY_NOTEPAD_REG(hkey, value_fWrap,            Globals.bWrapLongLines);
223         QUERY_NOTEPAD_REG(hkey, value_iWindowPosX,      main_rect.left);
224         QUERY_NOTEPAD_REG(hkey, value_iWindowPosY,      main_rect.top);
225         QUERY_NOTEPAD_REG(hkey, value_iWindowPosDX,     dx);
226         QUERY_NOTEPAD_REG(hkey, value_iWindowPosDY,     dy);
227         QUERY_NOTEPAD_REG(hkey, value_lfCharSet,        Globals.lfFont.lfCharSet);
228         QUERY_NOTEPAD_REG(hkey, value_lfClipPrecision,  Globals.lfFont.lfClipPrecision);
229         QUERY_NOTEPAD_REG(hkey, value_lfEscapement,     Globals.lfFont.lfEscapement);
230         QUERY_NOTEPAD_REG(hkey, value_lfItalic,         Globals.lfFont.lfItalic);
231         QUERY_NOTEPAD_REG(hkey, value_lfOrientation,    Globals.lfFont.lfOrientation);
232         QUERY_NOTEPAD_REG(hkey, value_lfOutPrecision,   Globals.lfFont.lfOutPrecision);
233         QUERY_NOTEPAD_REG(hkey, value_lfPitchAndFamily, Globals.lfFont.lfPitchAndFamily);
234         QUERY_NOTEPAD_REG(hkey, value_lfQuality,        Globals.lfFont.lfQuality);
235         QUERY_NOTEPAD_REG(hkey, value_lfStrikeOut,      Globals.lfFont.lfStrikeOut);
236         QUERY_NOTEPAD_REG(hkey, value_lfUnderline,      Globals.lfFont.lfUnderline);
237         QUERY_NOTEPAD_REG(hkey, value_lfWeight,         Globals.lfFont.lfWeight);
238         QUERY_NOTEPAD_REG(hkey, value_iMarginTop,       Globals.iMarginTop);
239         QUERY_NOTEPAD_REG(hkey, value_iMarginBottom,    Globals.iMarginBottom);
240         QUERY_NOTEPAD_REG(hkey, value_iMarginLeft,      Globals.iMarginLeft);
241         QUERY_NOTEPAD_REG(hkey, value_iMarginRight,     Globals.iMarginRight);
242 #undef QUERY_NOTEPAD_REG
243
244         main_rect.right = main_rect.left + dx;
245         main_rect.bottom = main_rect.top + dy;
246
247         size = sizeof(DWORD);
248         if(RegQueryValueExW(hkey, value_iPointSize, 0, &type, (LPBYTE)&data, &size) == ERROR_SUCCESS)
249             if(type == REG_DWORD)
250                 /* The value is stored as 10 * twips */
251                 Globals.lfFont.lfHeight = -MulDiv(abs(data), get_dpi(), 720);
252
253         size = sizeof(Globals.lfFont.lfFaceName);
254         if(RegQueryValueExW(hkey, value_lfFaceName, 0, &type, (LPBYTE)&data_helper, &size) == ERROR_SUCCESS)
255             if(type == REG_SZ)
256                 lstrcpyW(Globals.lfFont.lfFaceName, data_helper);
257
258         size = sizeof(Globals.szHeader);
259         if(RegQueryValueExW(hkey, value_szHeader, 0, &type, (LPBYTE)&data_helper, &size) == ERROR_SUCCESS)
260             if(type == REG_SZ)
261                 lstrcpyW(Globals.szHeader, data_helper);
262
263         size = sizeof(Globals.szFooter);
264         if(RegQueryValueExW(hkey, value_szFooter, 0, &type, (LPBYTE)&data_helper, &size) == ERROR_SUCCESS)
265             if(type == REG_SZ)
266                 lstrcpyW(Globals.szFooter, data_helper);
267         RegCloseKey(hkey);
268     }
269 }
270
271 /***********************************************************************
272  *
273  *           NOTEPAD_MenuCommand
274  *
275  *  All handling of main menu events
276  */
277 static int NOTEPAD_MenuCommand(WPARAM wParam)
278 {
279     switch (wParam)
280     {
281     case CMD_NEW:               DIALOG_FileNew(); break;
282     case CMD_OPEN:              DIALOG_FileOpen(); break;
283     case CMD_SAVE:              DIALOG_FileSave(); break;
284     case CMD_SAVE_AS:           DIALOG_FileSaveAs(); break;
285     case CMD_PRINT:             DIALOG_FilePrint(); break;
286     case CMD_PAGE_SETUP:        DIALOG_FilePageSetup(); break;
287     case CMD_PRINTER_SETUP:     DIALOG_FilePrinterSetup();break;
288     case CMD_EXIT:              DIALOG_FileExit(); break;
289
290     case CMD_UNDO:             DIALOG_EditUndo(); break;
291     case CMD_CUT:              DIALOG_EditCut(); break;
292     case CMD_COPY:             DIALOG_EditCopy(); break;
293     case CMD_PASTE:            DIALOG_EditPaste(); break;
294     case CMD_DELETE:           DIALOG_EditDelete(); break;
295     case CMD_SELECT_ALL:       DIALOG_EditSelectAll(); break;
296     case CMD_TIME_DATE:        DIALOG_EditTimeDate();break;
297
298     case CMD_SEARCH:           DIALOG_Search(); break;
299     case CMD_SEARCH_NEXT:      DIALOG_SearchNext(); break;
300     case CMD_REPLACE:          DIALOG_Replace(); break;
301                                
302     case CMD_WRAP:             DIALOG_EditWrap(); break;
303     case CMD_FONT:             DIALOG_SelectFont(); break;
304
305     case CMD_HELP_CONTENTS:    DIALOG_HelpContents(); break;
306     case CMD_HELP_ABOUT_NOTEPAD: DIALOG_HelpAboutNotepad(); break;
307
308     default:
309         break;
310     }
311    return 0;
312 }
313
314 /***********************************************************************
315  * Data Initialization
316  */
317 static VOID NOTEPAD_InitData(VOID)
318 {
319     LPWSTR p = Globals.szFilter;
320     static const WCHAR txt_files[] = { '*','.','t','x','t',0 };
321     static const WCHAR all_files[] = { '*','.','*',0 };
322
323     LoadStringW(Globals.hInstance, STRING_TEXT_FILES_TXT, p, MAX_STRING_LEN);
324     p += lstrlenW(p) + 1;
325     lstrcpyW(p, txt_files);
326     p += lstrlenW(p) + 1;
327     LoadStringW(Globals.hInstance, STRING_ALL_FILES, p, MAX_STRING_LEN);
328     p += lstrlenW(p) + 1;
329     lstrcpyW(p, all_files);
330     p += lstrlenW(p) + 1;
331     *p = '\0';
332     Globals.hDevMode = NULL;
333     Globals.hDevNames = NULL;
334
335     CheckMenuItem(GetMenu(Globals.hMainWnd), CMD_WRAP,
336             MF_BYCOMMAND | (Globals.bWrapLongLines ? MF_CHECKED : MF_UNCHECKED));
337 }
338
339 /***********************************************************************
340  * Enable/disable items on the menu based on control state
341  */
342 static VOID NOTEPAD_InitMenuPopup(HMENU menu, int index)
343 {
344     int enable;
345
346     EnableMenuItem(menu, CMD_UNDO,
347         SendMessageW(Globals.hEdit, EM_CANUNDO, 0, 0) ? MF_ENABLED : MF_GRAYED);
348     EnableMenuItem(menu, CMD_PASTE,
349         IsClipboardFormatAvailable(CF_TEXT) ? MF_ENABLED : MF_GRAYED);
350     enable = SendMessageW(Globals.hEdit, EM_GETSEL, 0, 0);
351     enable = (HIWORD(enable) == LOWORD(enable)) ? MF_GRAYED : MF_ENABLED;
352     EnableMenuItem(menu, CMD_CUT, enable);
353     EnableMenuItem(menu, CMD_COPY, enable);
354     EnableMenuItem(menu, CMD_DELETE, enable);
355     
356     EnableMenuItem(menu, CMD_SELECT_ALL,
357         GetWindowTextLengthW(Globals.hEdit) ? MF_ENABLED : MF_GRAYED);
358 }
359
360 static LPWSTR NOTEPAD_StrRStr(LPWSTR pszSource, LPWSTR pszLast, LPWSTR pszSrch)
361 {
362     int len = lstrlenW(pszSrch);
363     pszLast--;
364     while (pszLast >= pszSource)
365     {
366         if (StrCmpNW(pszLast, pszSrch, len) == 0)
367             return pszLast;
368         pszLast--;
369     }
370     return NULL;
371 }
372
373 /***********************************************************************
374  * The user activated the Find dialog
375  */
376 void NOTEPAD_DoFind(FINDREPLACEW *fr)
377 {
378     LPWSTR content;
379     LPWSTR found;
380     int len = lstrlenW(fr->lpstrFindWhat);
381     int fileLen;
382     DWORD pos;
383
384     fileLen = GetWindowTextLengthW(Globals.hEdit) + 1;
385     content = HeapAlloc(GetProcessHeap(), 0, fileLen * sizeof(WCHAR));
386     if (!content) return;
387     GetWindowTextW(Globals.hEdit, content, fileLen);
388
389     SendMessageW(Globals.hEdit, EM_GETSEL, 0, (LPARAM)&pos);
390     switch (fr->Flags & (FR_DOWN|FR_MATCHCASE))
391     {
392         case 0:
393             found = StrRStrIW(content, content+pos-len, fr->lpstrFindWhat);
394             break;
395         case FR_DOWN:
396             found = StrStrIW(content+pos, fr->lpstrFindWhat);
397             break;
398         case FR_MATCHCASE:
399             found = NOTEPAD_StrRStr(content, content+pos-len, fr->lpstrFindWhat);
400             break;
401         case FR_DOWN|FR_MATCHCASE:
402             found = StrStrW(content+pos, fr->lpstrFindWhat);
403             break;
404         default:    /* shouldn't happen */
405             return;
406     }
407     HeapFree(GetProcessHeap(), 0, content);
408
409     if (found == NULL)
410     {
411         DIALOG_StringMsgBox(Globals.hFindReplaceDlg, STRING_NOTFOUND, fr->lpstrFindWhat,
412             MB_ICONINFORMATION|MB_OK);
413         return;
414     }
415
416     SendMessageW(Globals.hEdit, EM_SETSEL, found - content, found - content + len);
417 }
418
419 static void NOTEPAD_DoReplace(FINDREPLACEW *fr)
420 {
421     LPWSTR content;
422     int len = lstrlenW(fr->lpstrFindWhat);
423     int fileLen;
424     DWORD pos;
425     DWORD pos_start;
426
427     fileLen = GetWindowTextLengthW(Globals.hEdit) + 1;
428     content = HeapAlloc(GetProcessHeap(), 0, fileLen * sizeof(WCHAR));
429     if (!content) return;
430     GetWindowTextW(Globals.hEdit, content, fileLen);
431
432     SendMessageW(Globals.hEdit, EM_GETSEL, (WPARAM)&pos_start, (LPARAM)&pos);
433     switch (fr->Flags & (FR_DOWN|FR_MATCHCASE))
434     {
435         case FR_DOWN:
436             if ( pos-pos_start == len && StrCmpNIW(fr->lpstrFindWhat, content+pos_start, len) == 0)
437                 SendMessageW(Globals.hEdit, EM_REPLACESEL, TRUE, (LPARAM)fr->lpstrReplaceWith);
438             break;
439         case FR_DOWN|FR_MATCHCASE:
440             if ( pos-pos_start == len && StrCmpNW(fr->lpstrFindWhat, content+pos_start, len) == 0)
441                 SendMessageW(Globals.hEdit, EM_REPLACESEL, TRUE, (LPARAM)fr->lpstrReplaceWith);
442             break;
443         default:    /* shouldn't happen */
444             return;
445     }
446     HeapFree(GetProcessHeap(), 0, content);
447
448     NOTEPAD_DoFind(fr);
449 }
450
451 static void NOTEPAD_DoReplaceAll(FINDREPLACEW *fr)
452 {
453     LPWSTR content;
454     LPWSTR found;
455     int len = lstrlenW(fr->lpstrFindWhat);
456     int fileLen;
457     DWORD pos;
458
459     SendMessageW(Globals.hEdit, EM_SETSEL, 0, 0);
460     while(TRUE){
461         fileLen = GetWindowTextLengthW(Globals.hEdit) + 1;
462         content = HeapAlloc(GetProcessHeap(), 0, fileLen * sizeof(WCHAR));
463         if (!content) return;
464         GetWindowTextW(Globals.hEdit, content, fileLen);
465
466         SendMessageW(Globals.hEdit, EM_GETSEL, 0, (LPARAM)&pos);
467         switch (fr->Flags & (FR_DOWN|FR_MATCHCASE))
468         {
469             case FR_DOWN:
470                 found = StrStrIW(content+pos, fr->lpstrFindWhat);
471                 break;
472             case FR_DOWN|FR_MATCHCASE:
473                 found = StrStrW(content+pos, fr->lpstrFindWhat);
474                 break;
475             default:    /* shouldn't happen */
476                 return;
477         }
478         HeapFree(GetProcessHeap(), 0, content);
479
480         if(found == NULL)
481         {
482             SendMessageW(Globals.hEdit, EM_SETSEL, 0, 0);
483             return;
484         }
485         SendMessageW(Globals.hEdit, EM_SETSEL, found - content, found - content + len);
486         SendMessageW(Globals.hEdit, EM_REPLACESEL, TRUE, (LPARAM)fr->lpstrReplaceWith);
487     }
488 }
489
490 /***********************************************************************
491  *
492  *           NOTEPAD_WndProc
493  */
494 static LRESULT WINAPI NOTEPAD_WndProc(HWND hWnd, UINT msg, WPARAM wParam,
495                                LPARAM lParam)
496 {
497     if (msg == aFINDMSGSTRING)      /* not a constant so can't be used in switch */
498     {
499         FINDREPLACEW *fr = (FINDREPLACEW *)lParam;
500
501         if (fr->Flags & FR_DIALOGTERM)
502             Globals.hFindReplaceDlg = NULL;
503         if (fr->Flags & FR_FINDNEXT)
504         {
505             Globals.lastFind = *fr;
506             NOTEPAD_DoFind(fr);
507         }
508         if (fr->Flags & FR_REPLACE)
509         {
510             Globals.lastFind = *fr;
511             NOTEPAD_DoReplace(fr);
512         }
513         if (fr->Flags & FR_REPLACEALL)
514         {
515             Globals.lastFind = *fr;
516             NOTEPAD_DoReplaceAll(fr);
517         }
518         return 0;
519     }
520     
521     switch (msg) {
522
523     case WM_CREATE:
524     {
525         static const WCHAR editW[] = { 'e','d','i','t',0 };
526         DWORD dwStyle = WS_CHILD | WS_VISIBLE | WS_BORDER | WS_VSCROLL |
527                         ES_AUTOVSCROLL | ES_MULTILINE | ES_NOHIDESEL;
528         RECT rc;
529         GetClientRect(hWnd, &rc);
530
531         if (!Globals.bWrapLongLines) dwStyle |= WS_HSCROLL | ES_AUTOHSCROLL;
532
533         Globals.hEdit = CreateWindowExW(WS_EX_CLIENTEDGE, editW, NULL,
534                              dwStyle, 0, 0, rc.right, rc.bottom, hWnd,
535                              NULL, Globals.hInstance, NULL);
536
537         Globals.hFont = CreateFontIndirectW(&Globals.lfFont);
538         SendMessageW(Globals.hEdit, WM_SETFONT, (WPARAM)Globals.hFont, FALSE);
539         SendMessageW(Globals.hEdit, EM_LIMITTEXT, 0, 0);
540         break;
541     }
542
543     case WM_COMMAND:
544         NOTEPAD_MenuCommand(LOWORD(wParam));
545         break;
546
547     case WM_DESTROYCLIPBOARD:
548         /*MessageBoxW(Globals.hMainWnd, "Empty clipboard", "Debug", MB_ICONEXCLAMATION);*/
549         break;
550
551     case WM_CLOSE:
552         if (DoCloseFile()) {
553             DestroyWindow(hWnd);
554         }
555         break;
556
557     case WM_QUERYENDSESSION:
558         if (DoCloseFile()) {
559             return 1;
560         }
561         break;
562
563     case WM_DESTROY:
564         NOTEPAD_SaveSettingToRegistry();
565
566         PostQuitMessage(0);
567         break;
568
569     case WM_SIZE:
570         SetWindowPos(Globals.hEdit, NULL, 0, 0, LOWORD(lParam), HIWORD(lParam),
571                      SWP_NOOWNERZORDER | SWP_NOZORDER);
572         break;
573
574     case WM_SETFOCUS:
575         SetFocus(Globals.hEdit);
576         break;
577
578     case WM_DROPFILES:
579     {
580         WCHAR szFileName[MAX_PATH];
581         HANDLE hDrop = (HANDLE) wParam;
582
583         DragQueryFileW(hDrop, 0, szFileName, ARRAY_SIZE(szFileName));
584         DragFinish(hDrop);
585         DoOpenFile(szFileName, ENCODING_AUTO);
586         break;
587     }
588     
589     case WM_INITMENUPOPUP:
590         NOTEPAD_InitMenuPopup((HMENU)wParam, lParam);
591         break;
592
593     default:
594         return DefWindowProcW(hWnd, msg, wParam, lParam);
595     }
596     return 0;
597 }
598
599 static int AlertFileDoesNotExist(LPCWSTR szFileName)
600 {
601    int nResult;
602    WCHAR szMessage[MAX_STRING_LEN];
603    WCHAR szResource[MAX_STRING_LEN];
604
605    LoadStringW(Globals.hInstance, STRING_DOESNOTEXIST, szResource, ARRAY_SIZE(szResource));
606    wsprintfW(szMessage, szResource, szFileName);
607
608    LoadStringW(Globals.hInstance, STRING_ERROR, szResource, ARRAY_SIZE(szResource));
609
610    nResult = MessageBoxW(Globals.hMainWnd, szMessage, szResource,
611                          MB_ICONEXCLAMATION | MB_YESNOCANCEL);
612
613    return(nResult);
614 }
615
616 static void HandleCommandLine(LPWSTR cmdline)
617 {
618     WCHAR delimiter;
619     int opt_print=0;
620     
621     /* skip white space */
622     while (*cmdline == ' ') cmdline++;
623
624     /* skip executable name */
625     delimiter = (*cmdline == '"' ? '"' : ' ');
626
627     if (*cmdline == delimiter) cmdline++;
628
629     while (*cmdline && *cmdline != delimiter) cmdline++;
630
631     if (*cmdline == delimiter) cmdline++;
632
633     while (*cmdline == ' ' || *cmdline == '-' || *cmdline == '/')
634     {
635         WCHAR option;
636
637         if (*cmdline++ == ' ') continue;
638
639         option = *cmdline;
640         if (option) cmdline++;
641         while (*cmdline == ' ') cmdline++;
642
643         switch(option)
644         {
645             case 'p':
646             case 'P':
647                 opt_print=1;
648                 break;
649         }
650     }
651
652     if (*cmdline)
653     {
654         /* file name is passed in the command line */
655         LPCWSTR file_name;
656         BOOL file_exists;
657         WCHAR buf[MAX_PATH];
658
659         if (cmdline[0] == '"')
660         {
661             WCHAR* wc;
662             cmdline++;
663             wc=cmdline;
664             /* Note: Double-quotes are not allowed in Windows filenames */
665             while (*wc && *wc != '"') wc++;
666             /* On Windows notepad ignores further arguments too */
667             *wc = 0;
668         }
669
670         if (FileExists(cmdline))
671         {
672             file_exists = TRUE;
673             file_name = cmdline;
674         }
675         else
676         {
677             static const WCHAR txtW[] = { '.','t','x','t',0 };
678
679             /* try to find file with ".txt" extension */
680             if (strchrW(PathFindFileNameW(cmdline), '.'))
681             {
682                 file_exists = FALSE;
683                 file_name = cmdline;
684             }
685             else
686             {
687                 lstrcpynW(buf, cmdline, MAX_PATH - lstrlenW(txtW) - 1);
688                 lstrcatW(buf, txtW);
689                 file_name = buf;
690                 file_exists = FileExists(buf);
691             }
692         }
693
694         if (file_exists)
695         {
696             DoOpenFile(file_name, ENCODING_AUTO);
697             InvalidateRect(Globals.hMainWnd, NULL, FALSE);
698             if (opt_print)
699                 DIALOG_FilePrint();
700         }
701         else
702         {
703             switch (AlertFileDoesNotExist(file_name)) {
704             case IDYES:
705                 SetFileNameAndEncoding(file_name, ENCODING_ANSI);
706                 UpdateWindowCaption();
707                 break;
708
709             case IDNO:
710                 break;
711
712             case IDCANCEL:
713                 DestroyWindow(Globals.hMainWnd);
714                 break;
715             }
716         }
717      }
718 }
719
720 /***********************************************************************
721  *
722  *           WinMain
723  */
724 int PASCAL WinMain(HINSTANCE hInstance, HINSTANCE prev, LPSTR cmdline, int show)
725 {
726     MSG msg;
727     HACCEL hAccel;
728     WNDCLASSEXW class;
729     HMONITOR monitor;
730     MONITORINFO info;
731     INT x, y;
732     static const WCHAR className[] = {'N','o','t','e','p','a','d',0};
733     static const WCHAR winName[]   = {'N','o','t','e','p','a','d',0};
734
735     aFINDMSGSTRING = RegisterWindowMessageW(FINDMSGSTRINGW);
736
737     ZeroMemory(&Globals, sizeof(Globals));
738     Globals.hInstance       = hInstance;
739     NOTEPAD_LoadSettingFromRegistry();
740
741     ZeroMemory(&class, sizeof(class));
742     class.cbSize        = sizeof(class);
743     class.lpfnWndProc   = NOTEPAD_WndProc;
744     class.hInstance     = Globals.hInstance;
745     class.hIcon         = LoadIconW(Globals.hInstance, MAKEINTRESOURCEW(IDI_NOTEPAD));
746     class.hIconSm       = LoadImageW(Globals.hInstance, MAKEINTRESOURCEW(IDI_NOTEPAD), IMAGE_ICON,
747                                      GetSystemMetrics(SM_CXSMICON), GetSystemMetrics(SM_CYSMICON),
748                                      LR_SHARED);
749     class.hCursor       = LoadCursorW(0, (LPCWSTR)IDC_ARROW);
750     class.hbrBackground = (HBRUSH)(COLOR_WINDOW + 1);
751     class.lpszMenuName  = MAKEINTRESOURCEW(MAIN_MENU);
752     class.lpszClassName = className;
753
754     if (!RegisterClassExW(&class)) return FALSE;
755
756     /* Setup windows */
757
758     monitor = MonitorFromRect( &main_rect, MONITOR_DEFAULTTOPRIMARY );
759     info.cbSize = sizeof(info);
760     GetMonitorInfoW( monitor, &info );
761
762     x = main_rect.left;
763     y = main_rect.top;
764     if (main_rect.left >= info.rcWork.right ||
765         main_rect.top >= info.rcWork.bottom ||
766         main_rect.right < info.rcWork.left ||
767         main_rect.bottom < info.rcWork.top)
768         x = y = CW_USEDEFAULT;
769
770     Globals.hMainWnd =
771         CreateWindowW(className, winName, WS_OVERLAPPEDWINDOW, x, y,
772                       main_rect.right - main_rect.left, main_rect.bottom - main_rect.top,
773                       NULL, NULL, Globals.hInstance, NULL);
774     if (!Globals.hMainWnd)
775     {
776         ShowLastError();
777         ExitProcess(1);
778     }
779
780     NOTEPAD_InitData();
781     DIALOG_FileNew();
782
783     ShowWindow(Globals.hMainWnd, show);
784     UpdateWindow(Globals.hMainWnd);
785     DragAcceptFiles(Globals.hMainWnd, TRUE);
786
787     HandleCommandLine(GetCommandLineW());
788
789     hAccel = LoadAcceleratorsW(hInstance, MAKEINTRESOURCEW(ID_ACCEL));
790
791     while (GetMessageW(&msg, 0, 0, 0))
792     {
793         if (!TranslateAcceleratorW(Globals.hMainWnd, hAccel, &msg) && !IsDialogMessageW(Globals.hFindReplaceDlg, &msg))
794         {
795             TranslateMessage(&msg);
796             DispatchMessageW(&msg);
797         }
798     }
799     return msg.wParam;
800 }