Fix crash if lpuCurDirLen or lpuDestDirLen==0.
[wine] / dlls / version / install.c
1 /* 
2  * Implementation of VERSION.DLL - File Installer routines
3  * 
4  * Copyright 1996,1997 Marcus Meissner
5  * Copyright 1997 David Cuthbert
6  */
7
8 #include <stdlib.h>
9 #include <string.h>
10
11 #include "windef.h"
12 #include "winver.h"
13 #include "wine/winestring.h"
14 #include "winerror.h"
15 #include "heap.h"
16 #include "ver.h"
17 #include "lzexpand.h"
18 #include "xmalloc.h"
19 #include "debug.h"
20
21
22 /******************************************************************************
23  *
24  *   void  ver_dstring(
25  *      char const * prologue,
26  *      char const * teststring,
27  *      char const * epilogue )
28  *
29  *   This function will print via dprintf[_]ver to stddeb the prologue string,
30  *   followed by the address of teststring and the string it contains if
31  *   teststring is non-null or "(null)" otherwise, and then the epilogue
32  *   string followed by a new line.
33  *
34  *   Revision history
35  *      30-May-1997 Dave Cuthbert (dacut@ece.cmu.edu)
36  *         Original implementation as dprintf[_]ver_string
37  *      05-Jul-1997 Dave Cuthbert (dacut@ece.cmu.edu)
38  *         Fixed problem that caused bug with tools/make_debug -- renaming
39  *         this function should fix the problem.
40  *      15-Feb-1998 Dimitrie Paun (dimi@cs.toronto.edu)
41  *         Modified it to make it print the message using only one
42  *         dprintf[_]ver call.
43  *
44  *****************************************************************************/
45
46 static void  ver_dstring(
47     char const * prologue,
48     char const * teststring,
49     char const * epilogue )
50 {
51     TRACE(ver, "%s %p (\"%s\") %s\n", prologue,
52                 (void const *) teststring,
53                 teststring ? teststring : "(null)",
54                 epilogue);
55 }
56
57 /******************************************************************************
58  *
59  *   int  testFileExistence(
60  *      char const * path,
61  *      char const * file )
62  *
63  *   Tests whether a given path/file combination exists.  If the file does
64  *   not exist, the return value is zero.  If it does exist, the return
65  *   value is non-zero.
66  *
67  *   Revision history
68  *      30-May-1997 Dave Cuthbert (dacut@ece.cmu.edu)
69  *         Original implementation
70  *
71  *****************************************************************************/
72
73 static int  testFileExistence(
74    char const * path,
75    char const * file )
76 {
77     char  filename[1024];
78     int  filenamelen;
79     OFSTRUCT  fileinfo;
80     int  retval;
81
82     fileinfo.cBytes = sizeof(OFSTRUCT);
83
84     strcpy(filename, path);
85     filenamelen = strlen(filename);
86
87     /* Add a trailing \ if necessary */
88     if(filenamelen) {
89         if(filename[filenamelen - 1] != '\\')
90             strcat(filename, "\\");
91     }
92     else /* specify the current directory */
93         strcpy(filename, ".\\");
94
95     /* Create the full pathname */
96     strcat(filename, file);
97
98     if(OpenFile(filename, &fileinfo, OF_EXIST) == HFILE_ERROR)
99         retval = 0;
100     else
101         retval = 1;
102
103     return  retval;
104 }
105
106 /******************************************************************************
107  *
108  *   int  testFileExclusiveExistence(
109  *      char const * path,
110  *      char const * file )
111  *
112  *   Tests whether a given path/file combination exists and ensures that no
113  *   other programs have handles to the given file.  If the file does not
114  *   exist or is open, the return value is zero.  If it does exist, the
115  *   return value is non-zero.
116  *
117  *   Revision history
118  *      30-May-1997 Dave Cuthbert (dacut@ece.cmu.edu)
119  *         Original implementation
120  *
121  *****************************************************************************/
122
123 static int  testFileExclusiveExistence(
124    char const * path,
125    char const * file )
126 {
127     char  filename[1024];
128     int  filenamelen;
129     OFSTRUCT  fileinfo;
130     int  retval;
131
132     fileinfo.cBytes = sizeof(OFSTRUCT);
133
134     strcpy(filename, path);
135     filenamelen = strlen(filename);
136
137     /* Add a trailing \ if necessary */
138     if(filenamelen) {
139         if(filename[filenamelen - 1] != '\\')
140             strcat(filename, "\\");
141     }
142     else /* specify the current directory */
143         strcpy(filename, ".\\");
144
145     /* Create the full pathname */
146     strcat(filename, file);
147
148     if(OpenFile(filename, &fileinfo, OF_EXIST | OF_SHARE_EXCLUSIVE) ==
149        HFILE_ERROR)
150         retval = 0;
151     else
152         retval = 1;
153
154     return retval;
155 }
156
157 /*****************************************************************************
158  *
159  *   VerFindFile() [VER.8]
160  *   Determines where to install a file based on whether it locates another
161  *   version of the file in the system.  The values VerFindFile returns are
162  *   used in a subsequent call to the VerInstallFile function.
163  *
164  *   Revision history:
165  *      30-May-1997   Dave Cuthbert (dacut@ece.cmu.edu)
166  *         Reimplementation of VerFindFile from original stub.
167  *
168  ****************************************************************************/
169
170 /* VerFindFile32A                                               [VERSION.5] */
171 DWORD WINAPI VerFindFileA(
172     UINT flags,
173     LPCSTR lpszFilename,
174     LPCSTR lpszWinDir,
175     LPCSTR lpszAppDir,
176     LPSTR lpszCurDir,
177     UINT *lpuCurDirLen,
178     LPSTR lpszDestDir,
179     UINT *lpuDestDirLen )
180 {
181     DWORD  retval;
182     char  curDir[256];
183     char  destDir[256];
184     unsigned int  curDirSizeReq;
185     unsigned int  destDirSizeReq;
186
187     retval = 0;
188
189     /* Print out debugging information */
190     TRACE(ver, "called with parameters:\n"
191                  "\tflags = %x", flags);
192     if(flags & VFFF_ISSHAREDFILE)
193         TRACE(ver, " (VFFF_ISSHAREDFILE)\n");
194     else
195         TRACE(ver, "\n");
196
197     ver_dstring("\tlpszFilename = ", lpszFilename, "");
198     ver_dstring("\tlpszWinDir = ", lpszWinDir, "");
199     ver_dstring("\tlpszAppDir = ", lpszAppDir, "");
200
201     TRACE(ver, "\tlpszCurDir = %p\n", lpszCurDir);
202     if(lpuCurDirLen)
203         TRACE(ver, "\tlpuCurDirLen = %p (%u)\n",
204                     lpuCurDirLen, *lpuCurDirLen);
205     else
206         TRACE(ver, "\tlpuCurDirLen = (null)\n");
207
208     TRACE(ver, "\tlpszDestDir = %p\n", lpszDestDir);
209     if(lpuDestDirLen)
210         TRACE(ver, "\tlpuDestDirLen = %p (%u)\n",
211                     lpuDestDirLen, *lpuDestDirLen);
212
213     /* Figure out where the file should go; shared files default to the
214        system directory */
215
216     strcpy(curDir, "");
217     strcpy(destDir, "");
218
219     if(flags & VFFF_ISSHAREDFILE) {
220         GetSystemDirectoryA(destDir, 256);
221
222         /* Were we given a filename?  If so, try to find the file. */
223         if(lpszFilename) {
224             if(testFileExistence(destDir, lpszFilename)) {
225                 strcpy(curDir, destDir);
226
227                 if(!testFileExclusiveExistence(destDir, lpszFilename))
228                     retval |= VFF_FILEINUSE;
229             }
230             else if(lpszAppDir && testFileExistence(lpszAppDir,
231                                                     lpszFilename)) {
232                 strcpy(curDir, lpszAppDir);
233                 retval |= VFF_CURNEDEST;
234
235                 if(!testFileExclusiveExistence(lpszAppDir, lpszFilename))
236                     retval |= VFF_FILEINUSE;
237             }
238         }
239     }
240     else if(!(flags & VFFF_ISSHAREDFILE)) { /* not a shared file */
241         if(lpszAppDir) {
242             char  systemDir[256];
243             GetSystemDirectoryA(systemDir, 256);
244
245             strcpy(destDir, lpszAppDir);
246
247             if(lpszFilename) {
248                 if(testFileExistence(lpszAppDir, lpszFilename)) {
249                     strcpy(curDir, lpszAppDir);
250
251                     if(!testFileExclusiveExistence(lpszAppDir, lpszFilename))
252                         retval |= VFF_FILEINUSE;
253                 }
254                 else if(testFileExistence(systemDir, lpszFilename)) {
255                     strcpy(curDir, systemDir);
256                     retval |= VFF_CURNEDEST;
257
258                     if(!testFileExclusiveExistence(systemDir, lpszFilename))
259                         retval |= VFF_FILEINUSE;
260                 }
261             }
262         }
263     }
264
265     curDirSizeReq = strlen(curDir) + 1;
266     destDirSizeReq = strlen(destDir) + 1;
267
268
269
270     /* Make sure that the pointers to the size of the buffers are
271        valid; if not, do NOTHING with that buffer.  If that pointer
272        is valid, then make sure that the buffer pointer is valid, too! */
273
274     if(lpuDestDirLen && lpszDestDir) {
275         if(*lpuDestDirLen < destDirSizeReq) {
276             retval |= VFF_BUFFTOOSMALL;
277             if (*lpuDestDirLen) {
278             strncpy(lpszDestDir, destDir, *lpuDestDirLen - 1);
279             lpszDestDir[*lpuDestDirLen - 1] = '\0';
280         }
281         }
282         else
283             strcpy(lpszDestDir, destDir);
284
285         *lpuDestDirLen = destDirSizeReq;
286     }
287     
288     if(lpuCurDirLen && lpszCurDir) {
289         if(*lpuCurDirLen < curDirSizeReq) {
290             retval |= VFF_BUFFTOOSMALL;
291             if (*lpuCurDirLen) {
292             strncpy(lpszCurDir, curDir, *lpuCurDirLen - 1);
293             lpszCurDir[*lpuCurDirLen - 1] = '\0';
294             }
295         }
296         else
297             strcpy(lpszCurDir, curDir);
298
299         *lpuCurDirLen = curDirSizeReq;
300     }
301
302     TRACE(ver, "ret = %lu (%s%s%s)\n", retval,
303                  (retval & VFF_CURNEDEST) ? "VFF_CURNEDEST " : "",
304                  (retval & VFF_FILEINUSE) ? "VFF_FILEINUSE " : "",
305                  (retval & VFF_BUFFTOOSMALL) ? "VFF_BUFFTOOSMALL " : "");
306
307     ver_dstring("\t(Exit) lpszCurDir = ", lpszCurDir, "");
308     if(lpuCurDirLen)
309         TRACE(ver, "\t(Exit) lpuCurDirLen = %p (%u)\n",
310                     lpuCurDirLen, *lpuCurDirLen);
311     else
312         TRACE(ver, "\t(Exit) lpuCurDirLen = (null)\n");
313
314     ver_dstring("\t(Exit) lpszDestDir = ", lpszDestDir, "");
315     if(lpuDestDirLen)
316         TRACE(ver, "\t(Exit) lpuDestDirLen = %p (%u)\n",
317                     lpuDestDirLen, *lpuDestDirLen);
318
319     return retval;
320 }
321
322 /* VerFindFile32W                                               [VERSION.6] */
323 DWORD WINAPI VerFindFileW(
324         UINT flags,LPCWSTR filename,LPCWSTR windir,LPCWSTR appdir,
325         LPWSTR curdir,UINT *pcurdirlen,LPWSTR destdir,UINT *pdestdirlen )
326 {
327     UINT curdirlen, destdirlen;
328     LPSTR wfn,wwd,wad,wdd,wcd;
329     DWORD ret;
330
331     wfn = HEAP_strdupWtoA( GetProcessHeap(), 0, filename );
332     wwd = HEAP_strdupWtoA( GetProcessHeap(), 0, windir );
333     wad = HEAP_strdupWtoA( GetProcessHeap(), 0, appdir );
334     wcd = HeapAlloc( GetProcessHeap(), 0, *pcurdirlen );
335     wdd = HeapAlloc( GetProcessHeap(), 0, *pdestdirlen );
336     ret = VerFindFileA(flags,wfn,wwd,wad,wcd,&curdirlen,wdd,&destdirlen);
337     lstrcpynAtoW(curdir,wcd,*pcurdirlen);
338     lstrcpynAtoW(destdir,wdd,*pdestdirlen);
339     *pcurdirlen = strlen(wcd);
340     *pdestdirlen = strlen(wdd);
341     HeapFree( GetProcessHeap(), 0, wfn );
342     HeapFree( GetProcessHeap(), 0, wwd );
343     HeapFree( GetProcessHeap(), 0, wad );
344     HeapFree( GetProcessHeap(), 0, wcd );
345     HeapFree( GetProcessHeap(), 0, wdd );
346     return ret;
347 }
348
349 static LPBYTE
350 _fetch_versioninfo(LPSTR fn,VS_FIXEDFILEINFO **vffi) {
351     DWORD       alloclen;
352     LPBYTE      buf;
353     DWORD       ret;
354
355     alloclen = 1000;
356     buf= xmalloc(alloclen);
357     while (1) {
358         ret = GetFileVersionInfoA(fn,0,alloclen,buf);
359         if (!ret) {
360             free(buf);
361             return 0;
362         }
363         if (alloclen<*(WORD*)buf) {
364             free(buf);
365             alloclen = *(WORD*)buf;
366             buf = xmalloc(alloclen);
367         } else {
368             *vffi = (VS_FIXEDFILEINFO*)(buf+0x14);
369             if ((*vffi)->dwSignature == 0x004f0049) /* hack to detect unicode */
370                 *vffi = (VS_FIXEDFILEINFO*)(buf+0x28);
371             if ((*vffi)->dwSignature != VS_FFI_SIGNATURE)
372                 WARN(ver,"Bad VS_FIXEDFILEINFO signature 0x%08lx\n",(*vffi)->dwSignature);
373             return buf;
374         }
375     }
376 }
377
378 static DWORD
379 _error2vif(DWORD error) {
380     switch (error) {
381     case ERROR_ACCESS_DENIED:
382         return VIF_ACCESSVIOLATION;
383     case ERROR_SHARING_VIOLATION:
384         return VIF_SHARINGVIOLATION;
385     default:
386         return 0;
387     }
388 }
389
390
391 /******************************************************************************
392  * VerInstallFile32A [VERSION.7]
393  */
394 DWORD WINAPI VerInstallFileA(
395         UINT flags,LPCSTR srcfilename,LPCSTR destfilename,LPCSTR srcdir,
396         LPCSTR destdir,LPCSTR curdir,LPSTR tmpfile,UINT *tmpfilelen )
397 {
398     LPCSTR pdest;
399     char        destfn[260],tmpfn[260],srcfn[260];
400     HFILE       hfsrc,hfdst;
401     DWORD       attr,ret,xret,tmplast;
402     LPBYTE      buf1,buf2;
403     OFSTRUCT    ofs;
404
405     TRACE(ver,"(%x,%s,%s,%s,%s,%s,%p,%d)\n",
406             flags,srcfilename,destfilename,srcdir,destdir,curdir,tmpfile,*tmpfilelen
407     );
408     xret = 0;
409     sprintf(srcfn,"%s\\%s",srcdir,srcfilename);
410     if (!destdir || !*destdir) pdest = srcdir;
411     else pdest = destdir;
412     sprintf(destfn,"%s\\%s",pdest,destfilename);
413     hfsrc=LZOpenFileA(srcfn,&ofs,OF_READ);
414     if (hfsrc==HFILE_ERROR)
415         return VIF_CANNOTREADSRC;
416     sprintf(tmpfn,"%s\\%s",pdest,destfilename);
417     tmplast=strlen(pdest)+1;
418     attr = GetFileAttributesA(tmpfn);
419     if (attr!=-1) {
420         if (attr & FILE_ATTRIBUTE_READONLY) {
421             LZClose(hfsrc);
422             return VIF_WRITEPROT;
423         }
424         /* FIXME: check if file currently in use and return VIF_FILEINUSE */
425     }
426     attr = -1;
427     if (flags & VIFF_FORCEINSTALL) {
428         if (tmpfile[0]) {
429             sprintf(tmpfn,"%s\\%s",pdest,tmpfile);
430             tmplast = strlen(pdest)+1;
431             attr = GetFileAttributesA(tmpfn);
432             /* if it exists, it has been copied by the call before.
433              * we jump over the copy part... 
434              */
435         }
436     }
437     if (attr == -1) {
438         char    *s;
439
440         GetTempFileNameA(pdest,"ver",0,tmpfn); /* should not fail ... */
441         s=strrchr(tmpfn,'\\');
442         if (s)
443             tmplast = s-tmpfn;
444         else
445             tmplast = 0;
446         hfdst = OpenFile(tmpfn,&ofs,OF_CREATE);
447         if (hfdst == HFILE_ERROR) {
448             LZClose(hfsrc);
449             return VIF_CANNOTCREATE; /* | translated dos error */
450         }
451         ret = LZCopy(hfsrc,hfdst);
452         _lclose(hfdst);
453         if (((long) ret) < 0) {
454             /* translate LZ errors into VIF_xxx */
455             switch (ret) {
456             case LZERROR_BADINHANDLE:
457             case LZERROR_READ:
458             case LZERROR_BADVALUE:
459             case LZERROR_UNKNOWNALG:
460                 ret = VIF_CANNOTREADSRC;
461                 break;
462             case LZERROR_BADOUTHANDLE:
463             case LZERROR_WRITE:
464                 ret = VIF_OUTOFMEMORY; /* FIXME: correct? */
465                 break;
466             case LZERROR_GLOBALLOC:
467             case LZERROR_GLOBLOCK:
468                 ret = VIF_OUTOFSPACE;
469                 break;
470             default: /* unknown error, should not happen */
471                 ret = 0;
472                 break;
473             }
474             if (ret) {
475                 LZClose(hfsrc);
476                 return ret;
477             }
478         }
479     }
480     xret = 0;
481     if (!(flags & VIFF_FORCEINSTALL)) {
482         VS_FIXEDFILEINFO *destvffi,*tmpvffi;
483         buf1 = _fetch_versioninfo(destfn,&destvffi);
484         if (buf1) {
485             buf2 = _fetch_versioninfo(tmpfn,&tmpvffi);
486             if (buf2) {
487                 char    *tbuf1,*tbuf2;
488                 UINT    len1,len2;
489
490                 len1=len2=40;
491
492                 /* compare file versions */
493                 if ((destvffi->dwFileVersionMS > tmpvffi->dwFileVersionMS)||
494                     ((destvffi->dwFileVersionMS==tmpvffi->dwFileVersionMS)&&
495                      (destvffi->dwFileVersionLS > tmpvffi->dwFileVersionLS)
496                     )
497                 )
498                     xret |= VIF_MISMATCH|VIF_SRCOLD;
499                 /* compare filetypes and filesubtypes */
500                 if ((destvffi->dwFileType!=tmpvffi->dwFileType) ||
501                     (destvffi->dwFileSubtype!=tmpvffi->dwFileSubtype)
502                 )
503                     xret |= VIF_MISMATCH|VIF_DIFFTYPE;
504                 if (VerQueryValueA(buf1,"\\VarFileInfo\\Translation",(LPVOID*)&tbuf1,&len1) &&
505                     VerQueryValueA(buf2,"\\VarFileInfo\\Translation",(LPVOID*)&tbuf2,&len2)
506                 ) {
507                     /* irgendwas mit tbuf1 und tbuf2 machen 
508                      * generiert DIFFLANG|MISMATCH
509                      */
510                 }
511                 free(buf2);
512             } else
513                 xret=VIF_MISMATCH|VIF_SRCOLD;
514             free(buf1);
515         }
516     }
517     if (xret) {
518         if (*tmpfilelen<strlen(tmpfn+tmplast)) {
519             xret|=VIF_BUFFTOOSMALL;
520             DeleteFileA(tmpfn);
521         } else {
522             strcpy(tmpfile,tmpfn+tmplast);
523             *tmpfilelen = strlen(tmpfn+tmplast)+1;
524             xret|=VIF_TEMPFILE;
525         }
526     } else {
527         if (-1!=GetFileAttributesA(destfn))
528             if (!DeleteFileA(destfn)) {
529                 xret|=_error2vif(GetLastError())|VIF_CANNOTDELETE;
530                 DeleteFileA(tmpfn);
531                 LZClose(hfsrc);
532                 return xret;
533             }
534         if ((!(flags & VIFF_DONTDELETEOLD))     && 
535             curdir                              && 
536             *curdir                             &&
537             lstrcmpiA(curdir,pdest)
538         ) {
539             char curfn[260];
540
541             sprintf(curfn,"%s\\%s",curdir,destfilename);
542             if (-1!=GetFileAttributesA(curfn)) {
543                 /* FIXME: check if in use ... if it is, VIF_CANNOTDELETECUR */
544                 if (!DeleteFileA(curfn))
545                     xret|=_error2vif(GetLastError())|VIF_CANNOTDELETECUR;
546             }
547         }
548         if (!MoveFileA(tmpfn,destfn)) {
549             xret|=_error2vif(GetLastError())|VIF_CANNOTRENAME;
550             DeleteFileA(tmpfn);
551         }
552     }
553     LZClose(hfsrc);
554     return xret;
555 }
556
557
558 /* VerInstallFile32W                            [VERSION.8] */
559 DWORD WINAPI VerInstallFileW(
560         UINT flags,LPCWSTR srcfilename,LPCWSTR destfilename,LPCWSTR srcdir,
561         LPCWSTR destdir,LPCWSTR curdir,LPWSTR tmpfile,UINT *tmpfilelen )
562 {
563     LPSTR wsrcf,wsrcd,wdestf,wdestd,wtmpf,wcurd;
564     DWORD ret;
565
566     wsrcf  = HEAP_strdupWtoA( GetProcessHeap(), 0, srcfilename );
567     wsrcd  = HEAP_strdupWtoA( GetProcessHeap(), 0, srcdir );
568     wdestf = HEAP_strdupWtoA( GetProcessHeap(), 0, destfilename );
569     wdestd = HEAP_strdupWtoA( GetProcessHeap(), 0, destdir );
570     wtmpf  = HEAP_strdupWtoA( GetProcessHeap(), 0, tmpfile );
571     wcurd  = HEAP_strdupWtoA( GetProcessHeap(), 0, curdir );
572     ret = VerInstallFileA(flags,wsrcf,wdestf,wsrcd,wdestd,wcurd,wtmpf,tmpfilelen);
573     if (!ret)
574         lstrcpynAtoW(tmpfile,wtmpf,*tmpfilelen);
575     HeapFree( GetProcessHeap(), 0, wsrcf );
576     HeapFree( GetProcessHeap(), 0, wsrcd );
577     HeapFree( GetProcessHeap(), 0, wdestf );
578     HeapFree( GetProcessHeap(), 0, wdestd );
579     HeapFree( GetProcessHeap(), 0, wtmpf );
580     if (wcurd) 
581         HeapFree( GetProcessHeap(), 0, wcurd );
582     return ret;
583 }
584