source: trunk/dll/undel.c@ 1038

Last change on this file since 1038 was 1038, checked in by Gregg Young, 17 years ago

More fortify cleanup
  • Property svn:eol-style set to native
  • Property svn:keywords set to Author Date Id Revision
File size: 12.3 KB
Line 
1
2/***********************************************************************
3
4 $Id: undel.c 1038 2008-07-04 20:33:59Z gyoung $
5
6 Copyright (c) 1993-98 M. Kimes
7 Copyright (c) 2004, 2006 Steven H. Levine
8
9 01 Aug 04 SHL Rework lstrip/rstrip usage
10 24 May 05 SHL Rework Win_Error usage
11 17 Jul 06 SHL Use Runtime_Error
12 29 Jul 06 SHL Use xfgets_bstripcr
13 03 Nov 06 SHL Count thread usage
14 06 Aug 07 GKY Reduce DosSleep times (ticket 148)
15 20 Aug 07 GKY Move #pragma alloc_text to end for OpenWatcom compat
16 29 Feb 08 GKY Use xfree where appropriate
17
18***********************************************************************/
19
20#include <stdlib.h>
21#include <string.h>
22#include <ctype.h>
23#include <process.h> // _beginthread
24
25#define INCL_DOS
26#define INCL_WIN
27#define INCL_LONGLONG
28
29#include "fm3dlg.h"
30#include "fm3str.h"
31#include "errutil.h" // Dos_Error...
32#include "strutil.h" // GetPString
33#include "fm3dll.h"
34#include "fortify.h"
35
36#pragma data_seg(DATA2)
37
38static PSZ pszSrcFile = __FILE__;
39
40struct tempstruct
41{
42 HWND hwnd;
43 CHAR path[CCHMAXPATH];
44 BOOL inclsubdirs;
45};
46
47static VOID FillUndelListThread(VOID * arg)
48{
49 HWND hwnd;
50 CHAR s[CCHMAXPATH * 2];
51 CHAR *path;
52 HAB thab;
53 HMQ thmq;
54 FILE *fp;
55 HFILE oldstdout, newstdout;
56 struct tempstruct *undelinfo;
57 BOOL killme = FALSE;
58 FILESTATUS3 fsa;
59
60 undelinfo = (struct tempstruct *)arg;
61 hwnd = undelinfo->hwnd;
62 path = undelinfo->path;
63 DosError(FERR_DISABLEHARDERR);
64
65# ifdef FORTIFY
66 Fortify_EnterScope();
67# endif
68 thab = WinInitialize(0);
69 thmq = WinCreateMsgQueue(thab, 0);
70 if (thab && thmq) {
71 WinCancelShutdown(thmq, TRUE);
72 IncrThreadUsage();
73 WinSendDlgItemMsg(hwnd, UNDEL_LISTBOX, LM_DELETEALL, MPVOID, MPVOID);
74 unlinkf("%s", "$UDELETE.#$#");
75 fp = xfopen("$UDELETE.#$#", "w", pszSrcFile, __LINE__);
76 if (!fp) {
77 Win_Error(NULLHANDLE, hwnd, pszSrcFile, __LINE__,
78 GetPString(IDS_REDIRECTERRORTEXT));
79 killme = TRUE;
80 goto Abort;
81 }
82 else {
83 newstdout = -1;
84 if (DosDupHandle(fileno(stdout), &newstdout)) {
85 saymsg(MB_CANCEL,
86 hwnd,
87 GetPString(IDS_MAYDAYTEXT), GetPString(IDS_REDIRECTERRORTEXT));
88 fclose(fp);
89 killme = TRUE;
90 goto Abort;
91 }
92 oldstdout = fileno(stdout);
93 DosDupHandle(fileno(fp), &oldstdout);
94 runemf2(SEPARATE | INVISIBLE | WINDOWED | BACKGROUND | WAIT,
95 hwnd, pszSrcFile, __LINE__,
96 NULL,
97 NULL,
98 "UNDELETE.COM %s /L%s",
99 path, (undelinfo->inclsubdirs) ? " /S" : NullStr);
100 oldstdout = fileno(stdout);
101 DosDupHandle(newstdout, &oldstdout);
102 DosClose(newstdout);
103 fclose(fp);
104 }
105 fp = xfopen("$UDELETE.#$#", "r", pszSrcFile, __LINE__);
106 if (fp) {
107 xfgets(s, sizeof(s), fp, pszSrcFile, __LINE__); // Skip 1st line
108 while (!feof(fp)) {
109 strset(s, 0);
110 if (!xfgets_bstripcr(s, CCHMAXPATH + 2, fp, pszSrcFile, __LINE__))
111 break;
112 if (*s) {
113 if (!strnicmp(s, "SYS3194: ", 9)) {
114
115 APIRET temp;
116
117 strcat(s, " ");
118 xfgets(&s[strlen(s)], CCHMAXPATH + 128 - strlen(s), fp,
119 pszSrcFile, __LINE__);
120 fclose(fp);
121 s[CCHMAXPATH + 128] = 0;
122 stripcr(s);
123 rstrip(s);
124 strcat(s, GetPString(IDS_ASKABOUTUNDELETEHELPTEXT));
125 temp = saymsg(MB_YESNOCANCEL | MB_ICONEXCLAMATION,
126 hwnd, GetPString(IDS_ERRORTEXT), "%s", s);
127 if (temp == MBID_YES)
128 runemf2(BACKGROUND | INVISIBLE | SEPARATE | WINDOWED,
129 hwnd, pszSrcFile, __LINE__,
130 NULL, NULL, "%s /C HELP UNDELETE", GetCmdSpec(FALSE));
131 if (temp == MBID_CANCEL)
132 killme = TRUE;
133 goto Abort;
134 }
135 else if (s[1] != ':')
136 continue;
137 else if ((SHORT)
138 WinSendDlgItemMsg(hwnd, UNDEL_LISTBOX, LM_SEARCHSTRING,
139 MPFROM2SHORT(0, LIT_FIRST),
140 MPFROMP(s)) < 0
141 && DosQueryPathInfo(s, FIL_STANDARD, &fsa,
142 (ULONG) sizeof(fsa)))
143 WinSendDlgItemMsg(hwnd, UNDEL_LISTBOX, LM_INSERTITEM,
144 MPFROM2SHORT(LIT_SORTASCENDING, 0), MPFROMP(s));
145 }
146 }
147 fclose(fp);
148 }
149 Abort:
150 ;
151 }
152 DosForceDelete("$UDELETE.#$#");
153 xfree(undelinfo, pszSrcFile, __LINE__);
154 if (thmq) {
155 PostMsg(hwnd, UM_CONTAINER_FILLED, MPVOID, MPVOID);
156 if (killme)
157 PostMsg(hwnd, WM_CLOSE, MPVOID, MPVOID);
158 WinDestroyMsgQueue(thmq);
159 }
160 if (thab) {
161 DecrThreadUsage();
162 WinTerminate(thab);
163 }
164# ifdef FORTIFY
165 Fortify_LeaveScope();
166# endif
167}
168
169MRESULT EXPENTRY UndeleteDlgProc(HWND hwnd, ULONG msg, MPARAM mp1, MPARAM mp2)
170{
171 SHORT sSelect;
172 static BOOL listdone, changed = FALSE, refresh = FALSE;
173 static HPOINTER hptrIcon = (HPOINTER) 0;
174
175 switch (msg) {
176 case WM_INITDLG:
177 listdone = TRUE;
178 if (!mp2 || !*(CHAR *)mp2) {
179 Runtime_Error2(pszSrcFile, __LINE__, IDS_NODATATEXT);
180 WinDismissDlg(hwnd, 0);
181 break;
182 }
183 hptrIcon = WinLoadPointer(HWND_DESKTOP, FM3ModHandle, UNDEL_FRAME);
184 WinDefDlgProc(hwnd, WM_SETICON, MPFROMLONG(hptrIcon), MPVOID);
185 WinSendDlgItemMsg(hwnd, UNDEL_ENTRY, EM_SETTEXTLIMIT,
186 MPFROM2SHORT(CCHMAXPATH, 0), MPVOID);
187 {
188 CHAR s[CCHMAXPATH];
189
190 strcpy(s, (CHAR *)mp2);
191 if (s[strlen(s) - 1] != '\\')
192 strcat(s, "\\");
193 strcat(s, "*");
194 WinSetDlgItemText(hwnd, UNDEL_ENTRY, s);
195 WinCheckButton(hwnd, UNDEL_SUBDIRS, TRUE);
196 FillPathListBox(hwnd, WinWindowFromID(hwnd, UNDEL_DRIVELIST), (HWND) 0,
197 s, TRUE);
198 }
199 PostMsg(hwnd, UM_SETUP, MPVOID, MPVOID);
200 break;
201
202 case UM_SETUP:
203 if (listdone) {
204
205 struct tempstruct *undelinfo;
206 CHAR s[CCHMAXPATH];
207
208 listdone = FALSE;
209 undelinfo = xmallocz(sizeof(struct tempstruct), pszSrcFile, __LINE__);
210 if (!undelinfo) {
211 listdone = TRUE;
212 WinDismissDlg(hwnd, 0);
213 }
214 else {
215 undelinfo->hwnd = hwnd;
216 WinQueryDlgItemText(hwnd,
217 UNDEL_ENTRY,
218 sizeof(undelinfo->path), undelinfo->path);
219 bstrip(undelinfo->path);
220 MakeFullName(undelinfo->path);
221 undelinfo->inclsubdirs = WinQueryButtonCheckstate(hwnd,
222 UNDEL_SUBDIRS);
223 sprintf(s,
224 GetPString(IDS_UNDELETETITLETEXT), toupper(*undelinfo->path));
225 WinSetWindowText(hwnd, s);
226 if (_beginthread(FillUndelListThread, NULL, 65536, (PVOID) undelinfo)
227 == -1) {
228 Runtime_Error(pszSrcFile, __LINE__,
229 GetPString(IDS_COULDNTSTARTTHREADTEXT));
230 xfree(undelinfo, pszSrcFile, __LINE__);
231 listdone = TRUE;
232 WinDismissDlg(hwnd, 0);
233 }
234 else
235 DosSleep(100);//05 Aug 07 GKY 500
236 }
237 refresh = FALSE;
238 }
239 else
240 refresh = TRUE;
241 changed = FALSE;
242 return 0;
243
244 case UM_CONTAINER_FILLED:
245 listdone = TRUE;
246 {
247 CHAR s[33];
248
249 sSelect = (SHORT) WinSendDlgItemMsg(hwnd,
250 UNDEL_LISTBOX,
251 LM_QUERYITEMCOUNT, MPVOID, MPVOID);
252 sprintf(s, "%d", sSelect);
253 WinSetDlgItemText(hwnd, UNDEL_COUNT, s);
254 if (refresh)
255 PostMsg(hwnd, UM_SETUP, MPVOID, MPVOID);
256 refresh = FALSE;
257 }
258 return 0;
259
260 case WM_CONTROL:
261 switch (SHORT1FROMMP(mp1)) {
262 case UNDEL_SUBDIRS:
263 PostMsg(hwnd, UM_SETUP, MPVOID, MPVOID);
264 break;
265
266 case UNDEL_ENTRY:
267 switch (SHORT2FROMMP(mp1)) {
268 case EN_CHANGE:
269 changed = TRUE;
270 break;
271 case EN_KILLFOCUS:
272 if (changed)
273 PostMsg(hwnd, UM_SETUP, MPVOID, MPVOID);
274 break;
275 }
276 break;
277
278 case UNDEL_DRIVELIST:
279 if (!listdone) {
280 Runtime_Error(pszSrcFile, __LINE__, "not listdone");
281 break;
282 }
283 switch (SHORT2FROMMP(mp1)) {
284 case CBN_ENTER:
285 {
286 CHAR s[CCHMAXPATH], drive, *p;
287
288 strset(s, 0);
289 WinQueryDlgItemText(hwnd, UNDEL_DRIVELIST, 3, s);
290 if (!isalpha(*s)) {
291 Runtime_Error2(pszSrcFile, __LINE__, IDS_NODATATEXT);
292 }
293 else {
294 drive = toupper(*s);
295 WinQueryDlgItemText(hwnd, UNDEL_ENTRY, sizeof(s), s);
296 *s = drive;
297 s[1] = ':';
298 s[2] = '\\';
299 p = strrchr(s + 2, '\\');
300 if (p) {
301 p++;
302 if (*p)
303 memmove(s + 3, p, strlen(p) + 1);
304 else {
305 s[3] = '*';
306 s[4] = 0;
307 }
308 }
309 else {
310 s[3] = '*';
311 s[4] = 0;
312 }
313 s[CCHMAXPATH - 1] = 0;
314 WinSetDlgItemText(hwnd, UNDEL_ENTRY, s);
315 PostMsg(hwnd, UM_SETUP, MPVOID, MPVOID);
316 }
317 }
318 break;
319 }
320 break;
321
322 case UNDEL_LISTBOX:
323 switch (SHORT2FROMMP(mp2)) {
324 case LN_ENTER:
325 WinSendDlgItemMsg(hwnd, DID_OK, BM_CLICK, MPFROMSHORT(TRUE), MPVOID);
326 break;
327 }
328 break;
329
330 default:
331 break;
332 }
333 return 0;
334
335 case WM_ADJUSTWINDOWPOS:
336 PostMsg(hwnd, UM_STRETCH, MPVOID, MPVOID);
337 break;
338
339 case UM_STRETCH:
340 {
341 SWP swpC, swp, swpM, swpD, swpL, swpE;
342
343 WinQueryWindowPos(hwnd, &swp);
344 if (!(swp.fl & (SWP_HIDE | SWP_MINIMIZE))) {
345 WinQueryWindowPos(WinWindowFromID(hwnd, UNDEL_LISTBOX), &swpC);
346 WinQueryWindowPos(WinWindowFromID(hwnd, UNDEL_MASKHDR), &swpM);
347 WinQueryWindowPos(WinWindowFromID(hwnd, UNDEL_DRVHDR), &swpD);
348 WinQueryWindowPos(WinWindowFromID(hwnd, UNDEL_DRIVELIST), &swpL);
349 WinQueryWindowPos(WinWindowFromID(hwnd, UNDEL_ENTRY), &swpE);
350 WinSetWindowPos(WinWindowFromID(hwnd, UNDEL_LISTBOX), HWND_TOP,
351 SysVal(SV_CXSIZEBORDER),
352 swpC.y,
353 swp.cx - (SysVal(SV_CXSIZEBORDER) * 2),
354 (swp.cy - swpC.y) - (SysVal(SV_CYTITLEBAR) +
355 SysVal(SV_CYSIZEBORDER) +
356 swpM.cy + 16),
357 SWP_MOVE | SWP_SIZE);
358 WinSetWindowPos(WinWindowFromID(hwnd, UNDEL_MASKHDR), HWND_TOP,
359 swpM.x,
360 (swp.cy - swpM.cy) - (SysVal(SV_CYTITLEBAR) +
361 SysVal(SV_CYSIZEBORDER) + 8),
362 swpM.cx, swpM.cy, SWP_MOVE | SWP_SIZE);
363 WinSetWindowPos(WinWindowFromID(hwnd, UNDEL_DRVHDR), HWND_TOP,
364 swpD.x,
365 (swp.cy - swpM.cy) - (SysVal(SV_CYTITLEBAR) +
366 SysVal(SV_CYSIZEBORDER) + 8),
367 swpD.cx, swpM.cy, SWP_MOVE | SWP_SIZE);
368 WinSetWindowPos(WinWindowFromID(hwnd, UNDEL_DRIVELIST), HWND_TOP,
369 swpL.x,
370 SysVal(SV_CYSIZEBORDER),
371 swpL.cx,
372 swp.cy - (SysVal(SV_CYTITLEBAR) +
373 (SysVal(SV_CYSIZEBORDER) * 2) + 6),
374 SWP_MOVE | SWP_SIZE);
375 WinSetWindowPos(WinWindowFromID(hwnd, UNDEL_ENTRY), HWND_TOP,
376 swpM.x + swpM.cx + 4,
377 (swp.cy - swpM.cy) - (SysVal(SV_CYTITLEBAR) +
378 SysVal(SV_CYSIZEBORDER) + 8),
379 swp.cx - ((swpM.x + swpM.cx + 4) +
380 (SysVal(SV_CXSIZEBORDER) + 8)),
381 swpM.cy + 2, SWP_MOVE | SWP_SIZE);
382 WinInvalidateRect(WinWindowFromID(hwnd, UNDEL_ENTRY), NULL, FALSE);
383 }
384 }
385 return 0;
386
387 case WM_COMMAND:
388 switch (SHORT1FROMMP(mp1)) {
389 case UNDEL_DEL:
390 case DID_OK:
391 if (changed || !listdone) {
392 Runtime_Error(pszSrcFile, __LINE__, "not done");
393 }
394 else {
395 sSelect = (USHORT) WinSendDlgItemMsg(hwnd, UNDEL_LISTBOX,
396 LM_QUERYSELECTION,
397 MPFROMSHORT(LIT_FIRST), MPVOID);
398 if (sSelect >= 0) {
399
400 FILE *fp;
401 CHAR s[CCHMAXPATH + 1];
402
403 DosForceDelete("\\FMUNDEL.CMD");
404 fp = xfopen("\\FMUNDEL.CMD", "w", pszSrcFile, __LINE__);
405 if (fp) {
406 while (sSelect >= 0) {
407 *s = 0;
408 WinSendDlgItemMsg(hwnd, UNDEL_LISTBOX, LM_QUERYITEMTEXT,
409 MPFROM2SHORT(sSelect, CCHMAXPATH),
410 MPFROMP(s));
411 if (SHORT1FROMMP(mp1) == UNDEL_DEL)
412 WinSendDlgItemMsg(hwnd, UNDEL_LISTBOX, LM_DELETEITEM,
413 MPFROM2SHORT(sSelect, 0), MPVOID);
414 if (*s) {
415 if (SHORT1FROMMP(mp1) == DID_OK)
416 fprintf(fp,
417 "IF NOT EXIST \"%s\" UNDELETE.COM \"%s\" /A\n",
418 s, s);
419 else
420 fprintf(fp, "UNDELETE.COM \"%s\" /F /A\n", s);
421 }
422 sSelect = (USHORT) WinSendDlgItemMsg(hwnd,
423 UNDEL_LISTBOX,
424 LM_QUERYSELECTION,
425 (SHORT1FROMMP(mp1) ==
426 DID_OK) ?
427 MPFROMSHORT(sSelect) :
428 MPFROMSHORT(LIT_FIRST),
429 MPVOID);
430 }
431 fprintf(fp, "DEL \\FMUNDEL.CMD /F\n");
432 fclose(fp);
433 runemf2(WINDOWED | BACKGROUND | SEPARATE | INVISIBLE,
434 hwnd, pszSrcFile, __LINE__,
435 NULL, NULL, "%s /C \\FMUNDEL.CMD", GetCmdSpec(FALSE));
436 }
437 }
438 if (SHORT1FROMMP(mp1) == DID_OK)
439 WinDismissDlg(hwnd, 0);
440 {
441 CHAR s[33];
442
443 sSelect = (SHORT) WinSendDlgItemMsg(hwnd,
444 UNDEL_LISTBOX,
445 LM_QUERYITEMCOUNT,
446 MPVOID, MPVOID);
447 sprintf(s, "%d", sSelect);
448 WinSetDlgItemText(hwnd, UNDEL_COUNT, s);
449 }
450 }
451 break;
452
453 case DID_CANCEL:
454 if (!listdone)
455 Runtime_Error(pszSrcFile, __LINE__, "is busy");
456 else
457 WinDismissDlg(hwnd, 0);
458 break;
459
460 case IDM_HELP:
461 saymsg(MB_ENTER | MB_ICONASTERISK,
462 hwnd,
463 GetPString(IDS_UNDELETEHELPTITLETEXT),
464 GetPString(IDS_UNDELETEHELPTEXT));
465 break;
466 }
467 return 0;
468
469 case WM_CLOSE:
470 if (!listdone) {
471 Runtime_Error(pszSrcFile, __LINE__, "not listdone");
472 return 0;
473 }
474 break;
475
476 case WM_DESTROY:
477 if (hptrIcon)
478 WinDestroyPointer(hptrIcon);
479 hptrIcon = (HPOINTER) 0;
480 break;
481 }
482 return WinDefDlgProc(hwnd, msg, mp1, mp2);
483}
484
485#pragma alloc_text(UNDELETE,FillUndelListThread,UndeleteDlgProc)
Note: See TracBrowser for help on using the repository browser.