Avoid crash by passing pointer to RegSetValueEx.
[wine] / dlls / setupapi / virtcopy.c
1 /*
2  * SetupAPI virtual copy operations
3  *
4  * Copyright 2001 Andreas Mohr
5  *
6  * This library is free software; you can redistribute it and/or
7  * modify it under the terms of the GNU Lesser General Public
8  * License as published by the Free Software Foundation; either
9  * version 2.1 of the License, or (at your option) any later version.
10  *
11  * This library is distributed in the hope that it will be useful,
12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
14  * Lesser General Public License for more details.
15  *
16  * You should have received a copy of the GNU Lesser General Public
17  * License along with this library; if not, write to the Free Software
18  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
19  *
20  * FIXME: we now rely on builtin setupapi.dll for dialog resources.
21  *        This is bad ! We ought to have 16bit resource handling working.
22  */
23
24 #include <stdarg.h>
25 #include <string.h>
26 #include "windef.h"
27 #include "winbase.h"
28 #include "winuser.h"
29 #include "winreg.h"
30 #include "wownt32.h"
31 #include "wingdi.h"
32 #include "winnls.h"
33 #include "setupapi.h"
34 #include "setupx16.h"
35 #include "setupapi_private.h"
36 #include "wine/debug.h"
37
38 WINE_DEFAULT_DEBUG_CHANNEL(setupapi);
39
40 static FARPROC16 VCP_Proc = NULL;
41 static LPARAM VCP_MsgRef = 0;
42
43 static BOOL VCP_opened = FALSE;
44
45 static VCPSTATUS vcp_status;
46
47 static HINSTANCE SETUPAPI_hInstance;
48
49 static WORD VCP_Callback( LPVOID obj, UINT16 msg, WPARAM16 wParam, LPARAM lParam, LPARAM lParamRef )
50 {
51     WORD args[8];
52     DWORD ret = OK;
53     if (VCP_Proc)
54     {
55         args[7] = HIWORD(obj);
56         args[6] = LOWORD(obj);
57         args[5] = msg;
58         args[4] = wParam;
59         args[3] = HIWORD(lParam);
60         args[2] = LOWORD(lParam);
61         args[1] = HIWORD(lParamRef);
62         args[0] = LOWORD(lParamRef);
63         WOWCallback16Ex( (DWORD)VCP_Proc, WCB16_PASCAL, sizeof(args), args, &ret );
64     }
65     return (WORD)ret;
66 }
67
68 /****************************** VHSTR management ******************************/
69
70 /*
71  * This is a totally braindead implementation for now;
72  * I don't care about speed at all ! Size and implementation time
73  * is much more important IMHO. I could have created some sophisticated
74  * tree structure, but... what the hell ! :-)
75  */
76 typedef struct {
77     DWORD refcount;
78     LPCSTR pStr;
79 } VHSTR_STRUCT;
80
81 static VHSTR_STRUCT **vhstrlist = NULL;
82 static VHSTR vhstr_alloc = 0;
83
84 #define VALID_VHSTR(x)          ((x < vhstr_alloc) && (vhstrlist[x]) && (vhstrlist[x]->refcount))
85
86 /***********************************************************************
87  *              vsmStringAdd (SETUPX.207)
88  */
89 VHSTR WINAPI vsmStringAdd16(LPCSTR lpszName)
90 {
91     VHSTR n;
92     VHSTR index = 0xffff;
93     HANDLE heap;
94
95     TRACE("add string '%s'\n", lpszName);
96     /* search whether string already inserted */
97     TRACE("searching for existing string...\n");
98     for (n = 0; n < vhstr_alloc; n++)
99     {
100         if ((vhstrlist[n]) && (vhstrlist[n]->refcount))
101         {
102                 TRACE("checking item: %d\n", n);
103             if (!strcmp(vhstrlist[n]->pStr, lpszName))
104             {
105                 TRACE("found\n");
106                 vhstrlist[n]->refcount++;
107                 return n;
108             }
109         }
110     }
111
112     /* hmm, not found yet, let's insert it */
113     TRACE("inserting item\n");
114     for (n = 0; n < vhstr_alloc; n++)
115     {
116         if ((!(vhstrlist[n])) || (!(vhstrlist[n]->refcount)))
117         {
118             index = n;
119             break;
120         }
121     }
122     heap = GetProcessHeap();
123     if (n == vhstr_alloc) /* hmm, no free index found yet */
124     {
125         index = vhstr_alloc;
126         vhstr_alloc += 20;
127
128         if (vhstrlist)
129             vhstrlist = HeapReAlloc(heap, HEAP_ZERO_MEMORY, vhstrlist,
130                                         sizeof(VHSTR_STRUCT *) * vhstr_alloc);
131         else
132             vhstrlist = HeapAlloc(heap, HEAP_ZERO_MEMORY,
133                                         sizeof(VHSTR_STRUCT *) * vhstr_alloc);
134     }
135     if (index == 0xffff)
136         return 0xffff; /* failure */
137     if (!vhstrlist[index])
138         vhstrlist[index] = HeapAlloc(heap, HEAP_ZERO_MEMORY, sizeof(VHSTR_STRUCT));
139     vhstrlist[index]->refcount = 1;
140     vhstrlist[index]->pStr = HeapAlloc(heap, 0, strlen(lpszName)+1);
141     strcpy((LPSTR)vhstrlist[index]->pStr, lpszName);
142     return index;
143 }
144
145 /***********************************************************************
146  *              vsmStringDelete (SETUPX.206)
147  */
148 INT16 WINAPI vsmStringDelete16(VHSTR vhstr)
149 {
150     if (VALID_VHSTR(vhstr))
151     {
152         vhstrlist[vhstr]->refcount--;
153         if (!vhstrlist[vhstr]->refcount)
154         {
155             HeapFree(GetProcessHeap(), 0, (LPSTR)vhstrlist[vhstr]->pStr);
156             vhstrlist[vhstr]->pStr = NULL;
157         }
158         return VCPN_OK;
159     }
160
161     /* string not found */
162     return VCPN_FAIL;
163 }
164
165 /*
166  * vsmStringFind() - not exported from a standard SETUPX.DLL, it seems
167  */
168 VHSTR WINAPI vsmStringFind16(LPCSTR lpszName)
169 {
170     WORD n;
171     for (n = 0; n < vhstr_alloc; n++)
172         if ((vhstrlist[n]) && (vhstrlist[n]->refcount) && (!strcmp(vhstrlist[n]->pStr, lpszName)))
173             return n;
174     return 0xffff;
175 }
176
177 /***********************************************************************
178  *              vsmGetStringName (SETUPX.205)
179  *
180  * Pretty correct, I guess
181  */
182 INT16 WINAPI vsmGetStringName16(VHSTR vhstr, LPSTR lpszBuffer, int cbBuffer)
183 {
184     if (VALID_VHSTR(vhstr))
185     {
186         int len = strlen(vhstrlist[vhstr]->pStr)+1;
187         if (cbBuffer >= len)
188         {
189             if (lpszBuffer)
190                 strcpy(lpszBuffer, vhstrlist[vhstr]->pStr);
191             return len;
192         }
193     }
194     return VCPN_FAIL;
195 }
196
197 /***********************************************************************
198  *              vsmStringCompare (not exported from a standard SETUPX.DLL, it seems)
199  */
200 INT16 WINAPI vsmStringCompare16(VHSTR vhstrA, VHSTR vhstrB)
201 {
202     if ((!VALID_VHSTR(vhstrA)) || (!VALID_VHSTR(vhstrB)))
203         return VCPN_FAIL; /* correct ? */
204     return strcmp(vhstrlist[vhstrA]->pStr, vhstrlist[vhstrB]->pStr);
205 }
206
207 /***********************************************************************
208  *              vsmGetStringRawName (SETUPX.208)
209  */
210 LPCSTR WINAPI vsmGetStringRawName16(VHSTR vhstr)
211 {
212     return (VALID_VHSTR(vhstr)) ? vhstrlist[vhstr]->pStr : NULL;
213 }
214
215
216 /***************************** VIRTNODE management ****************************/
217 static LPVIRTNODE *pvnlist = NULL;
218 static DWORD vn_num = 0;
219 static DWORD vn_last = 0;
220
221 RETERR16 VCP_VirtnodeCreate(LPVCPFILESPEC vfsSrc, LPVCPFILESPEC vfsDst, WORD fl, LPARAM lParam, LPEXPANDVTBL lpExpandVtbl)
222 {
223     HANDLE heap;
224     LPVIRTNODE lpvn;
225     RETERR16 cbres;
226
227     while (vn_last < vn_num)
228     {
229         if (pvnlist[vn_last] == NULL)
230             break;
231         vn_last++;
232     }
233     heap = GetProcessHeap();
234     if (vn_last == vn_num)
235     {
236         vn_num += 20;
237         if (pvnlist)
238             pvnlist = HeapReAlloc(heap, HEAP_ZERO_MEMORY, pvnlist,
239                                 sizeof(LPVIRTNODE *) * vn_num);
240         else
241             pvnlist = HeapAlloc(heap, HEAP_ZERO_MEMORY, 
242                                 sizeof(LPVIRTNODE *) * vn_num);
243     }
244     pvnlist[vn_last] = HeapAlloc(heap, HEAP_ZERO_MEMORY, sizeof(VIRTNODE));
245     lpvn = pvnlist[vn_last];
246     vn_last++;
247
248     lpvn->cbSize = sizeof(VIRTNODE);
249
250     if (vfsSrc)
251         memcpy(&lpvn->vfsSrc, vfsSrc, sizeof(VCPFILESPEC));
252
253     if (vfsDst)
254         memcpy(&lpvn->vfsDst, vfsDst, sizeof(VCPFILESPEC));
255
256     lpvn->fl = fl;
257     lpvn->lParam = lParam;
258     lpvn->lpExpandVtbl = lpExpandVtbl;
259
260     lpvn->vhstrDstFinalName = 0xffff; /* FIXME: what is this ? */
261
262     cbres = VCP_Callback(lpvn, VCPM_NODECREATE, 0, 0, VCP_MsgRef);
263     lpvn->fl |= VFNL_CREATED;
264     cbres = VCP_Callback(lpvn, VCPM_NODEACCEPT, 0, 0, VCP_MsgRef);
265
266     return OK;
267 }
268
269 BOOL VCP_VirtnodeDelete(LPVIRTNODE lpvnDel)
270 {
271     DWORD n;
272     RETERR16 cbres;
273
274     for (n = 0; n < vn_last; n++)
275     {
276         if (pvnlist[n] == lpvnDel)
277         {
278             cbres = VCP_Callback(lpvnDel, VCPM_NODEDESTROY, 0, 0, VCP_MsgRef);
279             HeapFree(GetProcessHeap(), 0, lpvnDel);
280             pvnlist[n] = NULL;
281             return TRUE;
282         }
283     }
284     return FALSE;
285 }
286
287 /***********************************************************************
288  *              VcpOpen (SETUPX.200)
289  *
290  * Sets up a virtual copy operation.
291  * This means that functions such as GenInstall()
292  * create a VIRTNODE struct for every file to be touched in a .INF file
293  * instead of actually touching the file.
294  * The actual copy/move/rename gets started when VcpClose or
295  * VcpFlush is called; several different callbacks are made
296  * (copy, rename, open, close, version conflicts, ...) on every file copied.
297  */
298 RETERR16 WINAPI VcpOpen16(VIFPROC vifproc, LPARAM lparamMsgRef)
299 {
300     TRACE("(%p, %08lx)\n", vifproc, lparamMsgRef);
301     if (VCP_opened)
302         return ERR_VCP_BUSY;
303
304     VCP_Proc = (FARPROC16)vifproc;
305     VCP_MsgRef = lparamMsgRef;
306
307     /* load SETUPAPI needed for dialog resources etc. */
308     SETUPAPI_hInstance = LoadLibraryA("setupapi.dll");
309     if (!SETUPAPI_hInstance)
310     {
311         ERR("Could not load sibling setupapi.dll\n");
312         return ERR_VCP_NOMEM;
313     }
314     VCP_opened = TRUE;
315     return OK;
316 }
317
318 /***********************************************************************
319  *              VcpQueueCopy            [SETUPX.13]
320  *
321  * lpExpandVtbl seems to be deprecated.
322  * fl are the CNFL_xxx and VNFL_xxx flags.
323  * lParam are the VNLP_xxx flags.
324  */
325 RETERR16 WINAPI VcpQueueCopy16(
326         LPCSTR lpszSrcFileName, LPCSTR lpszDstFileName,
327         LPCSTR lpszSrcDir, LPCSTR lpszDstDir,
328         LOGDISKID16 ldidSrc, LOGDISKID16 ldidDst,
329         LPEXPANDVTBL lpExpandVtbl,
330         WORD fl, LPARAM lParam
331 )
332 {
333     VCPFILESPEC vfsSrc, vfsDst;
334
335     if (!VCP_opened)
336         return ERR_VCP_NOTOPEN;
337
338     TRACE("srcdir: %s, srcfile: %s, dstdir: %s, dstfile: %s\n",
339       lpszSrcDir, lpszSrcFileName, lpszDstDir, lpszDstFileName);
340
341     TRACE("ldidSrc == %d, ldidDst == %d\n", ldidSrc, ldidDst);
342
343     vfsSrc.ldid = ldidSrc;
344     vfsSrc.vhstrDir = vsmStringAdd16(lpszSrcDir);
345     vfsSrc.vhstrFileName = vsmStringAdd16(lpszSrcFileName);
346
347     vfsDst.ldid = ldidDst;
348     vfsDst.vhstrDir = vsmStringAdd16(lpszDstDir);
349     vfsDst.vhstrFileName = vsmStringAdd16(lpszDstFileName);
350
351     return VCP_VirtnodeCreate(&vfsSrc, &vfsDst, fl, lParam,
352                     lpExpandVtbl);
353 }
354
355 /***********************************************************************
356  *              VcpQueueDelete          [SETUPX.17]
357  *
358  * Is lParamRef the same as lParam in VcpQueueCopy ?
359  * Damn docu !! Err... which docu ?
360  */
361 RETERR16 WINAPI VcpQueueDelete16(
362         LPCSTR lpszDstFileName,
363         LPCSTR lpszDstDir,
364         LOGDISKID16 ldidDst,
365         LPARAM lParamRef
366 )
367 {
368     VCPFILESPEC vfsDst;
369
370     if (!VCP_opened)
371         return ERR_VCP_NOTOPEN;
372
373     vfsDst.ldid = ldidDst;
374     vfsDst.vhstrDir = vsmStringAdd16(lpszDstDir);
375     vfsDst.vhstrFileName = vsmStringAdd16(lpszDstFileName);
376
377     return VCP_VirtnodeCreate(NULL, &vfsDst, VNFL_DELETE, lParamRef, 0);
378 }
379
380 /***********************************************************************
381  *              VcpQueueRename          [SETUPX.204]
382  *
383  */
384 RETERR16 WINAPI VcpQueueRename16(
385         LPCSTR lpszSrcFileName, LPCSTR lpszDstFileName,
386         LPCSTR lpszSrcDir, LPCSTR lpszDstDir,
387         LOGDISKID16 ldidSrc, LOGDISKID16 ldidDst,
388         LPARAM lParam
389 )
390 {
391     VCPFILESPEC vfsSrc, vfsDst;
392
393     if (!VCP_opened)
394         return ERR_VCP_NOTOPEN;
395
396     vfsSrc.ldid = ldidSrc;
397     vfsSrc.vhstrDir = vsmStringAdd16(lpszSrcDir);
398     vfsSrc.vhstrFileName = vsmStringAdd16(lpszSrcFileName);
399
400     vfsDst.ldid = ldidDst;
401     vfsDst.vhstrDir = vsmStringAdd16(lpszDstDir);
402     vfsDst.vhstrFileName = vsmStringAdd16(lpszDstFileName);
403
404     return VCP_VirtnodeCreate(&vfsSrc, &vfsDst, VNFL_RENAME, lParam,
405                     0);
406 }
407
408 /***********************************************************************
409  *              VcpEnumFiles (SETUPX.@)
410  */
411 INT16 WINAPI VcpEnumFiles(VCPENUMPROC vep, LPARAM lParamRef)
412 {
413     WORD n;
414
415     for (n = 0; n < vn_last; n++)
416         vep(pvnlist[n], lParamRef);
417
418     return 0; /* FIXME: return value ? */
419 }
420
421 /***********************************************************************
422  *              VcpExplain (SETUPX.411)
423  */
424 LPCSTR WINAPI VcpExplain16(LPVIRTNODE lpVn, DWORD dwWhat)
425 {
426     static char buffer[MAX_PATH]; /* FIXME: is this how it's done ? */
427     buffer[0] = '\0';
428     switch (dwWhat)
429     {
430         case VCPEX_SRC_FULL:
431         case VCPEX_DST_FULL:
432             {
433                 LPVCPFILESPEC lpvfs =
434                     (dwWhat == VCPEX_SRC_FULL) ?  &lpVn->vfsSrc : &lpVn->vfsDst;
435
436                 /* if we have an ldid, use it, otherwise use the string */
437                 /* from the vhstrlist array */
438                 if (lpvfs->ldid != 0xffff)
439                   CtlGetLddPath16(lpvfs->ldid, buffer);
440                 else
441                   strcat(buffer, vsmGetStringRawName16(lpvfs->vhstrDir));
442
443                 strcat(buffer, "\\");
444                 strcat(buffer, vsmGetStringRawName16(lpvfs->vhstrFileName));
445             }
446             break;
447         default:
448             FIXME("%ld unimplemented !\n", dwWhat);
449             strcpy(buffer, "Unknown error");
450             break;
451     }
452     return buffer;
453 }
454
455 RETERR16 VCP_CheckPaths(void)
456 {
457     DWORD n;
458     LPVIRTNODE lpvn;
459     RETERR16 cbres;
460
461     cbres = VCP_Callback(&vcp_status, VCPM_VSTATPATHCHECKSTART, 0, 0, VCP_MsgRef);
462     for (n = 0; n < vn_num; n++)
463     {
464         lpvn = pvnlist[n];
465         if (!lpvn) continue;
466         /* FIXME: check paths of all VIRTNODEs here ! */
467         cbres = VCP_Callback(&lpvn->vfsDst, VCPM_CHECKPATH, 0, (DWORD)lpvn, VCP_MsgRef);
468     }
469     cbres = VCP_Callback(&vcp_status, VCPM_VSTATPATHCHECKEND, 0, 0, VCP_MsgRef);
470     return OK;
471 }
472
473 RETERR16 VCP_CopyFiles(void)
474 {
475     char fn_src[MAX_PATH], fn_dst[MAX_PATH];
476     RETERR16 res = OK, cbres;
477     DWORD n;
478     LPVIRTNODE lpvn;
479
480     cbres = VCP_Callback(&vcp_status, VCPM_VSTATCOPYSTART, 0, 0, VCP_MsgRef);
481     for (n = 0; n < vn_num; n++)
482     {
483         lpvn = pvnlist[n];
484         if ((!lpvn) || ((lpvn->fl & VNFL_NODE_TYPE) != VNFL_COPY)) continue;
485         /* FIXME: need to send VCPM_VSTATNEWDISK notification sometimes */
486         strcpy(fn_src, VcpExplain16(lpvn, VCPEX_SRC_FULL));
487         strcpy(fn_dst, VcpExplain16(lpvn, VCPEX_DST_FULL));
488         /* FIXME: what is this VCPM_VSTATWRITE here for ?
489          * I guess it's to signal successful destination file creation */
490         cbres = VCP_Callback(&vcp_status, VCPM_VSTATWRITE, 0, 0, VCP_MsgRef);
491
492         /* FIXME: need to do the file copy in small chunks for notifications */
493         TRACE("copying '%s' to '%s'\n", fn_src, fn_dst);
494         /* perform the file copy */
495         if (!(CopyFileA(fn_src, fn_dst,
496                (lpvn->fl & VNLP_COPYIFEXISTS) ? FALSE : TRUE )))
497         {
498             ERR("error copying, src: %s -> dst: %s\n", fn_src, fn_dst);
499             res = ERR_VCP_IOFAIL;
500         }
501
502         vcp_status.prgFileRead.dwSoFar++;
503         cbres = VCP_Callback(&vcp_status, VCPM_VSTATREAD, 0, 0, VCP_MsgRef);
504         vcp_status.prgFileWrite.dwSoFar++;
505         cbres = VCP_Callback(&vcp_status, VCPM_VSTATWRITE, 0, 0, VCP_MsgRef);
506     }
507
508     cbres = VCP_Callback(&vcp_status, VCPM_VSTATCOPYEND, 0, 0, VCP_MsgRef);
509     return res;
510 }
511
512 /***********************************************************************
513  *              VcpFlush - internal (not exported), but documented
514  *
515  * VNFL_NOW is used for VcpFlush.
516  */
517 RETERR16 VcpFlush16(WORD fl, LPCSTR lpszBackupDest)
518 {
519     return OK;
520 }
521
522 /***********************************************************************
523  *              VcpClose (SETUPX.201)
524  *
525  * Does callbacks (-> vifproc) with VCPM_VSTATCLOSESTART,
526  * VCPM_VSTATCLOSEEND.
527  *
528  * fl gets VCPFL_xxx flags to indicate what to do with the
529  * VIRTNODEs (files to mess with) created by e.g. GenInstall()
530  */
531 RETERR16 WINAPI VcpClose16(WORD fl, LPCSTR lpszBackupDest)
532 {
533     RETERR16 res = OK;
534     WORD cbres = VCPN_PROCEED;
535
536     TRACE("(%04x, '%s')\n", fl, lpszBackupDest);
537
538     /* FIXME: needs to sort virtnodes in case VCPFL_INSPECIFIEDORDER
539      * is not set. This is done by VCP_Callback(VCPM_NODECOMPARE) */
540
541     TRACE("#1\n");
542     memset(&vcp_status, 0, sizeof(VCPSTATUS));
543     /* yes, vcp_status.cbSize is 0 ! */
544     TRACE("#2\n");
545     cbres = VCP_Callback(&vcp_status, VCPM_VSTATCLOSESTART, 0, 0, VCP_MsgRef);
546     TRACE("#3\n");
547
548     res = VCP_CheckPaths();
549     TRACE("#4\n");
550     if (res != OK)
551         return res; /* is this ok ? */
552     VCP_CopyFiles();
553
554     TRACE("#5\n");
555     cbres = VCP_Callback(&vcp_status, VCPM_VSTATCLOSEEND, 0, 0, VCP_MsgRef);
556     TRACE("#6\n");
557     VCP_Proc = NULL;
558     FreeLibrary(SETUPAPI_hInstance);
559     VCP_opened = FALSE;
560     return OK;
561 }
562
563 RETERR16 VCP_RenameFiles(void)
564 {
565     char fn_src[MAX_PATH], fn_dst[MAX_PATH];
566     RETERR16 res = OK, cbres;
567     DWORD n;
568     LPVIRTNODE lpvn;
569
570     cbres = VCP_Callback(&vcp_status, VCPM_VSTATRENAMESTART, 0, 0, VCP_MsgRef);
571     for (n = 0; n < vn_num; n++)
572     {
573         lpvn = pvnlist[n];
574         if ((!lpvn) || ((lpvn->fl & VNFL_NODE_TYPE) != VNFL_RENAME)) continue;
575         strcpy(fn_src, VcpExplain16(lpvn, VCPEX_SRC_FULL));
576         strcpy(fn_dst, VcpExplain16(lpvn, VCPEX_DST_FULL));
577         cbres = VCP_Callback(&lpvn->vfsDst, VCPM_FILEOPENOUT, 0, (LPARAM)lpvn, VCP_MsgRef);
578         if (!(MoveFileExA(fn_src, fn_dst, MOVEFILE_REPLACE_EXISTING)))
579             res = ERR_VCP_IOFAIL;
580         else
581             VCP_VirtnodeDelete(lpvn);
582     }
583     cbres = VCP_Callback(&vcp_status, VCPM_VSTATRENAMEEND, 0, 0, VCP_MsgRef);
584     return res;
585 }
586
587 /***********************************************************************
588  *              vcpDefCallbackProc (SETUPX.202)
589  */
590 RETERR16 WINAPI vcpDefCallbackProc16(LPVOID lpvObj, UINT16 uMsg, WPARAM wParam,
591                                         LPARAM lParam, LPARAM lParamRef)
592 {
593     static int count = 0;
594     if (count < 10)
595         FIXME("(%p, %04x, %04x, %08lx, %08lx) - what to do here ?\n",
596                 lpvObj, uMsg, wParam, lParam, lParamRef);
597     count++;
598     return OK;
599 }
600
601 /********************* point-and-click stuff from here ***********************/
602
603 static HWND hDlgCopy = 0;
604 static HKEY hKeyFiles = 0, hKeyRename = 0, hKeyConflict = 0;
605 static char BackupDir[12];
606
607 static INT_PTR CALLBACK VCP_UI_FileCopyDlgProc(HWND hWndDlg, UINT iMsg, WPARAM wParam, LPARAM lParam)
608 {
609     INT_PTR retval = FALSE;
610
611     if (iMsg == WM_INITDIALOG)
612     {
613         ShowWindow(hWndDlg, SW_SHOWNORMAL);
614         UpdateWindow(hWndDlg);
615         retval = TRUE;
616     }
617     return retval;
618 }
619
620 BOOL VCP_UI_GetDialogTemplate(LPCVOID *template32)
621 {
622     HRSRC hResInfo;
623     HGLOBAL hDlgTmpl32;
624
625     if (!(hResInfo = FindResourceA(SETUPAPI_hInstance, MAKEINTRESOURCEA(COPYFILEDLGORD), (LPSTR)RT_DIALOG)))
626         return FALSE;
627     if (!(hDlgTmpl32 = LoadResource(SETUPAPI_hInstance, hResInfo )) ||
628         !(*template32 = LockResource( hDlgTmpl32 )))
629         return FALSE;
630     return TRUE;
631 }
632
633 static LRESULT WINAPI
634 VCP_UI_FileCopyWndProc (HWND hwnd, UINT uMsg, WPARAM wParam, LPARAM lParam)
635 {
636     if (uMsg != WM_CREATE)
637         return DefWindowProcA (hwnd, uMsg, wParam, lParam);
638
639     switch (uMsg)
640     {
641         case WM_CREATE:
642             return 0;
643         default:
644             FIXME("%04x: unhandled.\n", uMsg);
645     }
646
647     return 0;
648 }
649
650 void VCP_UI_RegisterProgressClass(void)
651 {
652     static BOOL registered = FALSE;
653     WNDCLASSA wndClass;
654
655     if (registered)
656         return;
657
658     registered = TRUE;
659     ZeroMemory (&wndClass, sizeof(WNDCLASSA));
660     wndClass.style         = CS_GLOBALCLASS | CS_DBLCLKS | CS_HREDRAW | CS_VREDRAW;
661     wndClass.lpfnWndProc   = (WNDPROC)VCP_UI_FileCopyWndProc;
662     wndClass.cbClsExtra    = 0;
663     wndClass.cbWndExtra    = 0;
664     wndClass.hCursor       = LoadCursorA (0, (LPSTR)IDC_ARROW);
665     wndClass.hbrBackground = NULL;
666     wndClass.lpszClassName = "setupx_progress";
667
668     RegisterClassA (&wndClass);
669 }
670
671 RETERR16 VCP_UI_NodeCompare(LPVIRTNODE vn1, LPVIRTNODE vn2)
672 {
673     LPCSTR file1, file2;
674     file1 = vsmGetStringRawName16(vn1->vfsSrc.vhstrFileName);
675     file2 = vsmGetStringRawName16(vn2->vfsSrc.vhstrFileName);
676     return (RETERR16)strcmp(file1, file2);
677 }
678
679 RETERR16 VCP_UI_CopyStart(void)
680 {
681     LPCVOID template32;
682     char buf[256]; /* plenty */
683     BOOL dirty;
684     DWORD len;
685
686     /* FIXME: should be registered at DLL startup instead */
687     VCP_UI_RegisterProgressClass();
688     if (!(VCP_UI_GetDialogTemplate(&template32)))
689         return VCPN_FAIL;
690
691     if (vn_num > 10)  /* hack */
692     {
693         hDlgCopy = CreateDialogIndirectParamA(SETUPAPI_hInstance, template32, 0,
694                                               VCP_UI_FileCopyDlgProc, 0);
695         if (!hDlgCopy)
696             return VCPN_FAIL;
697         SetDlgItemTextA(hDlgCopy, SOURCESTRORD, "Scanning ...");
698         SetDlgItemTextA(hDlgCopy, DESTSTRORD, "NOT_IMPLEMENTED_YET");
699     }
700     strcpy(buf, REG_INSTALLEDFILES);
701     if (RegCreateKeyA(HKEY_LOCAL_MACHINE, buf, &hKeyFiles))
702         return VCPN_FAIL;
703     strcat(buf, REGPART_RENAME);
704     if (RegCreateKeyA(HKEY_LOCAL_MACHINE, buf, &hKeyRename))
705         return VCPN_FAIL;
706     if (RegCreateKeyA(HKEY_LOCAL_MACHINE, REG_VERSIONCONFLICT, &hKeyConflict))
707         return VCPN_FAIL;
708     len = 1;
709     if (!(RegQueryValueExA(hKeyConflict, "Dirty", NULL, 0, (LPBYTE)&dirty, &len)))
710     {
711         /* FIXME: what does SETUPX.DLL do in this case ? */
712         MESSAGE("Warning: another program using SETUPX is already running ! Failed.\n");
713         return VCPN_FAIL;
714     }
715     dirty = TRUE;
716     if (RegSetValueExA(hKeyConflict, "Dirty", 0, REG_BINARY, (LPBYTE)&dirty, 1))
717         return VCPN_FAIL;
718     len = 12;
719     if (!(RegQueryValueExA(hKeyConflict, "BackupDirectory", NULL, 0, BackupDir, &len)))
720         strcpy(BackupDir, "VCM");
721
722     /* create C:\WINDOWS\[BackupDir] and set registry key to it */
723     GetWindowsDirectoryA(buf, 256);
724     strcat(buf, "\\");
725     strcat(buf, BackupDir);
726     if (!(CreateDirectoryA(buf, NULL)))
727         return VCPN_FAIL;
728     if (RegSetValueExA(hKeyConflict, "BackupDirectory", 0, REG_SZ, (LPBYTE)buf, strlen(buf)+1))
729         return VCPN_FAIL;
730     RegCloseKey(hKeyConflict);
731
732     return VCPN_OK;
733 }
734
735 /***********************************************************************
736  *              vcpUICallbackProc (SETUPX.213)
737  */
738 RETERR16 WINAPI vcpUICallbackProc16(LPVOID lpvObj, UINT16 uMsg, WPARAM wParam,
739                                         LPARAM lParam, LPARAM lParamRef)
740 {
741     static int count = 0;
742     RETERR16 res = VCPN_OK, cbres;
743
744     if (count < 5)
745         FIXME("(%p, %04x, %04x, %08lx, %08lx) - semi-stub\n",
746                 lpvObj, uMsg, wParam, lParam, lParamRef);
747     count++;
748     switch (uMsg)
749     {
750         /* unused messages, it seems */
751         case VCPM_DISKPREPINFO:
752
753         case VCPM_FILENEEDED:
754
755         case VCPM_NODECREATE:
756         case VCPM_NODEACCEPT:
757
758         case VCPM_VSTATCLOSESTART:
759         case VCPM_VSTATPATHCHECKSTART:
760         case VCPM_VSTATPATHCHECKEND:
761
762         case VCPM_CHECKPATH:
763             break;
764
765         /* the real stuff */
766         case VCPM_NODECOMPARE:
767             res = VCP_UI_NodeCompare((LPVIRTNODE)lpvObj, (LPVIRTNODE)lParam);
768             break;
769         case VCPM_VSTATREAD:
770             break;
771         case VCPM_VSTATWRITE:
772             cbres = VCP_Callback(&vcp_status, VCPM_DISKPREPINFO, 0, 0, VCP_MsgRef);
773             break;
774         case VCPM_VSTATCLOSEEND:
775             RegCloseKey(hKeyFiles);
776             RegCloseKey(hKeyRename);
777             RegDeleteKeyA(HKEY_LOCAL_MACHINE, REG_VERSIONCONFLICT);
778             break;
779         case VCPM_VSTATCOPYSTART:
780             res = VCP_UI_CopyStart();
781             break;
782         case VCPM_VSTATCOPYEND:
783             if (hDlgCopy) DestroyWindow(hDlgCopy);
784             break;
785         default:
786             FIXME("unhandled msg 0x%04x\n", uMsg);
787     }
788     return res;
789 }