files/directory.c

   1 /*
   2  * DOS directories functions
   3  *
   4  * Copyright 1995 Alexandre Julliard
   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
  21 #include "config.h"
  22
  23 #include <ctype.h>
  24 #include <stdlib.h>
  25 #include <stdio.h>
  26 #include <string.h>
  27 #include <sys/types.h>
  28 #include <sys/stat.h>
  29 #ifdef HAVE_UNISTD_H
  30 # include <unistd.h>
  31 #endif
  32 #include <errno.h>
  33 #ifdef HAVE_SYS_ERRNO_H
  34 #include <sys/errno.h>
  35 #endif
  36
  37 #include "winbase.h"
  38 #include "wine/winbase16.h"
  39 #include "windef.h"
  40 #include "wingdi.h"
  41 #include "wine/winuser16.h"
  42 #include "winerror.h"
  43 #include "winternl.h"
  44 #include "wine/unicode.h"
  45 #include "drive.h"
  46 #include "file.h"
  47 #include "heap.h"
  48 #include "msdos.h"
  49 #include "wine/debug.h"
  50
  51 WINE_DEFAULT_DEBUG_CHANNEL(dosfs);
  52 WINE_DECLARE_DEBUG_CHANNEL(file);
  53
  54 static DOS_FULL_NAME DIR_Windows;
  55 static DOS_FULL_NAME DIR_System;
  56
  57 static const WCHAR wineW[] = {'w','i','n','e',0};
  58
  59 /***********************************************************************
  60  *           FILE_contains_pathW
  61  */
  62 inline static int FILE_contains_pathW (LPCWSTR name)
  63 {
  64     return ((*name && (name[1] == ':')) ||
  65             strchrW (name, '/') || strchrW (name, '\\'));
  66 }
  67
  68 /***********************************************************************
  69  *           DIR_GetPath
  70  *
  71  * Get a path name from the wine.ini file and make sure it is valid.
  72  */
  73 static int DIR_GetPath( LPCWSTR keyname, LPCWSTR defval, DOS_FULL_NAME *full_name,
  74                         LPWSTR longname, INT longname_len, BOOL warn )
  75 {
  76     WCHAR path[MAX_PATHNAME_LEN];
  77     BY_HANDLE_FILE_INFORMATION info;
  78     const char *mess = "does not exist";
  79
  80     PROFILE_GetWineIniString( wineW, keyname, defval, path, MAX_PATHNAME_LEN );
  81
  82     if (!DOSFS_GetFullName( path, TRUE, full_name ) ||
  83         (!FILE_Stat( full_name->long_name, &info ) && (mess=strerror(errno)))||
  84         (!(info.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) && (mess="not a directory")) ||
  85         (!(GetLongPathNameW(full_name->short_name, longname, longname_len))) )
  86     {
  87         if (warn)
  88             MESSAGE("Invalid path %s for %s directory: %s\n",
  89                     debugstr_w(path), debugstr_w(keyname), mess);
  90         return 0;
  91     }
  92     return 1;
  93 }
  94
  95
  96 /***********************************************************************
  97  *           DIR_Init
  98  */
  99 int DIR_Init(void)
 100 {
 101     char path[MAX_PATHNAME_LEN];
 102     WCHAR longpath[MAX_PATHNAME_LEN];
 103     DOS_FULL_NAME tmp_dir, profile_dir;
 104     int drive;
 105     const char *cwd;
 106     static const WCHAR windowsW[] = {'w','i','n','d','o','w','s',0};
 107     static const WCHAR systemW[] = {'s','y','s','t','e','m',0};
 108     static const WCHAR tempW[] = {'t','e','m','p',0};
 109     static const WCHAR profileW[] = {'p','r','o','f','i','l','e',0};
 110     static const WCHAR windows_dirW[] = {'c',':','\\','w','i','n','d','o','w','s',0};
 111     static const WCHAR system_dirW[] = {'c',':','\\','w','i','n','d','o','w','s','\\','s','y','s','t','e','m',0};
 112     static const WCHAR pathW[] = {'p','a','t','h',0};
 113     static const WCHAR path_dirW[] = {'c',':','\\','w','i','n','d','o','w','s',';',
 114                                       'c',':','\\','w','i','n','d','o','w','s','\\','s','y','s','t','e','m',0};
 115     static const WCHAR path_capsW[] = {'P','A','T','H',0};
 116     static const WCHAR temp_capsW[] = {'T','E','M','P',0};
 117     static const WCHAR tmp_capsW[] = {'T','M','P',0};
 118     static const WCHAR windirW[] = {'w','i','n','d','i','r',0};
 119     static const WCHAR winsysdirW[] = {'w','i','n','s','y','s','d','i','r',0};
 120     static const WCHAR userprofileW[] = {'U','S','E','R','P','R','O','F','I','L','E',0};
 121     static const WCHAR systemrootW[] = {'S','Y','S','T','E','M','R','O','O','T',0};
 122     static const WCHAR empty_strW[] = { 0 };
 123
 124     if (!getcwd( path, MAX_PATHNAME_LEN ))
 125     {
 126         perror( "Could not get current directory" );
 127         return 0;
 128     }
 129     cwd = path;
 130     if ((drive = DRIVE_FindDriveRoot( &cwd )) == -1)
 131     {
 132         MESSAGE("Warning: could not find wine config [Drive x] entry "
 133             "for current working directory %s; "
 134             "starting in windows directory.\n", cwd );
 135     }
 136     else
 137     {
 138         WCHAR szdrive[3]={drive+'A',':',0};
 139         MultiByteToWideChar(DRIVE_GetCodepage(drive), 0, cwd, -1, longpath, MAX_PATHNAME_LEN);
 140         DRIVE_SetCurrentDrive( drive );
 141         DRIVE_Chdir( drive, longpath );
 142         if(GetDriveTypeW(szdrive)==DRIVE_CDROM)
 143             chdir("/"); /* change to root directory so as not to lock cdroms */
 144     }
 145
 146     if (!(DIR_GetPath( windowsW, windows_dirW, &DIR_Windows, longpath, MAX_PATHNAME_LEN, TRUE )) ||
 147         !(DIR_GetPath( systemW, system_dirW, &DIR_System, longpath, MAX_PATHNAME_LEN, TRUE )) ||
 148         !(DIR_GetPath( tempW, windows_dirW, &tmp_dir, longpath, MAX_PATHNAME_LEN, TRUE )))
 149     {
 150         PROFILE_UsageWineIni();
 151         return 0;
 152     }
 153     if (-1 == access( tmp_dir.long_name, W_OK ))
 154     {
 155         if (errno==EACCES)
 156         {
 157                 MESSAGE("Warning: the temporary directory '%s' (specified in wine configuration file) is not writeable.\n", tmp_dir.long_name);
 158                 PROFILE_UsageWineIni();
 159         }
 160         else
 161                 MESSAGE("Warning: access to temporary directory '%s' failed (%s).\n",
 162                     tmp_dir.long_name, strerror(errno));
 163     }
 164
 165     if (drive == -1)
 166     {
 167         drive = DIR_Windows.drive;
 168         DRIVE_SetCurrentDrive( drive );
 169         DRIVE_Chdir( drive, DIR_Windows.short_name + 2 );
 170     }
 171
 172     /* Set the environment variables */
 173
 174     /* set PATH only if not set already */
 175     if (!GetEnvironmentVariableW( path_capsW, longpath, MAX_PATHNAME_LEN ))
 176     {
 177         PROFILE_GetWineIniString(wineW, pathW, path_dirW, longpath, MAX_PATHNAME_LEN);
 178         if (strchrW(longpath, '/'))
 179         {
 180             MESSAGE("Fix your wine config to use DOS drive syntax in [wine] 'Path=' statement! (no '/' allowed)\n");
 181             PROFILE_UsageWineIni();
 182             ExitProcess(1);
 183         }
 184         SetEnvironmentVariableW( path_capsW, longpath );
 185     }
 186
 187     SetEnvironmentVariableW( temp_capsW, tmp_dir.short_name );
 188     SetEnvironmentVariableW( tmp_capsW, tmp_dir.short_name );
 189     SetEnvironmentVariableW( windirW, DIR_Windows.short_name );
 190     SetEnvironmentVariableW( winsysdirW, DIR_System.short_name );
 191
 192     /* set COMSPEC only if it doesn't exist already */
 193     if (!GetEnvironmentVariableA( "COMSPEC", NULL, 0 ))
 194         SetEnvironmentVariableA( "COMSPEC", "c:\\command.com" );
 195
 196     TRACE("WindowsDir = %s (%s)\n",
 197           debugstr_w(DIR_Windows.short_name), DIR_Windows.long_name );
 198     TRACE("SystemDir  = %s (%s)\n",
 199           debugstr_w(DIR_System.short_name), DIR_System.long_name );
 200     TRACE("TempDir    = %s (%s)\n",
 201           debugstr_w(tmp_dir.short_name), tmp_dir.long_name );
 202     TRACE("Path       = %s\n", debugstr_w(longpath) );
 203     TRACE("Cwd        = %c:\\%s\n",
 204           'A' + drive, debugstr_w(DRIVE_GetDosCwd(drive)) );
 205
 206     if (DIR_GetPath( profileW, empty_strW, &profile_dir, longpath, MAX_PATHNAME_LEN, FALSE ))
 207     {
 208         TRACE("USERPROFILE= %s\n", debugstr_w(longpath) );
 209         SetEnvironmentVariableW( userprofileW, longpath );
 210     }
 211
 212     TRACE("SYSTEMROOT = %s\n", debugstr_w(DIR_Windows.short_name) );
 213     SetEnvironmentVariableW( systemrootW, DIR_Windows.short_name );
 214
 215     return 1;
 216 }
 217
 218
 219 /***********************************************************************
 220  *           GetTempPathA   (KERNEL32.@)
 221  */
 222 UINT WINAPI GetTempPathA( UINT count, LPSTR path )
 223 {
 224     WCHAR pathW[MAX_PATH];
 225     UINT ret;
 226
 227     ret = GetTempPathW(MAX_PATH, pathW);
 228
 229     if (!ret)
 230         return 0;
 231
 232     if (ret > MAX_PATH)
 233     {
 234         SetLastError(ERROR_FILENAME_EXCED_RANGE);
 235         return 0;
 236     }
 237
 238     ret = WideCharToMultiByte(CP_ACP, 0, pathW, -1, NULL, 0, NULL, NULL);
 239     if (ret <= count)
 240     {
 241         WideCharToMultiByte(CP_ACP, 0, pathW, -1, path, count, NULL, NULL);
 242         ret--; /* length without 0 */
 243     }
 244     return ret;
 245 }
 246
 247
 248 /***********************************************************************
 249  *           GetTempPathW   (KERNEL32.@)
 250  */
 251 UINT WINAPI GetTempPathW( UINT count, LPWSTR path )
 252 {
 253     static const WCHAR tmp[]  = { 'T', 'M', 'P', 0 };
 254     static const WCHAR temp[] = { 'T', 'E', 'M', 'P', 0 };
 255     WCHAR tmp_path[MAX_PATH];
 256     UINT ret;
 257
 258     TRACE("%u,%p\n", count, path);
 259
 260     if (!(ret = GetEnvironmentVariableW( tmp, tmp_path, MAX_PATH )))
 261         if (!(ret = GetEnvironmentVariableW( temp, tmp_path, MAX_PATH )))
 262             if (!(ret = GetCurrentDirectoryW( MAX_PATH, tmp_path )))
 263                 return 0;
 264
 265     if (ret > MAX_PATH)
 266     {
 267         SetLastError(ERROR_FILENAME_EXCED_RANGE);
 268         return 0;
 269     }
 270
 271     ret = GetFullPathNameW(tmp_path, MAX_PATH, tmp_path, NULL);
 272     if (!ret) return 0;
 273
 274     if (ret > MAX_PATH - 2)
 275     {
 276         SetLastError(ERROR_FILENAME_EXCED_RANGE);
 277         return 0;
 278     }
 279
 280     if (tmp_path[ret-1] != '\\')
 281     {
 282         tmp_path[ret++] = '\\';
 283         tmp_path[ret]   = '\0';
 284     }
 285
 286     ret++; /* add space for terminating 0 */
 287
 288     if (count)
 289     {
 290         lstrcpynW(path, tmp_path, count);
 291         if (count >= ret)
 292             ret--; /* return length without 0 */
 293         else if (count < 4)
 294             path[0] = 0; /* avoid returning ambiguous "X:" */
 295     }
 296
 297     TRACE("returning %u, %s\n", ret, debugstr_w(path));
 298     return ret;
 299 }
 300
 301
 302 /***********************************************************************
 303  *           DIR_GetWindowsUnixDir
 304  */
 305 UINT DIR_GetWindowsUnixDir( LPSTR path, UINT count )
 306 {
 307     if (path) lstrcpynA( path, DIR_Windows.long_name, count );
 308     return strlen( DIR_Windows.long_name );
 309 }
 310
 311
 312 /***********************************************************************
 313  *           DIR_GetSystemUnixDir
 314  */
 315 UINT DIR_GetSystemUnixDir( LPSTR path, UINT count )
 316 {
 317     if (path) lstrcpynA( path, DIR_System.long_name, count );
 318     return strlen( DIR_System.long_name );
 319 }
 320
 321
 322 /***********************************************************************
 323  *           GetTempDrive   (KERNEL.92)
 324  * A closer look at krnl386.exe shows what the SDK doesn't mention:
 325  *
 326  * returns:
 327  *   AL: driveletter
 328  *   AH: ':'            - yes, some kernel code even does stosw with
 329  *                            the returned AX.
 330  *   DX: 1 for success
 331  */
 332 UINT WINAPI GetTempDrive( BYTE ignored )
 333 {
 334     char *buffer;
 335     BYTE ret;
 336     UINT len = GetTempPathA( 0, NULL );
 337
 338     if (!(buffer = HeapAlloc( GetProcessHeap(), 0, len + 1 )) )
 339         ret = DRIVE_GetCurrentDrive() + 'A';
 340     else
 341     {
 342         /* FIXME: apparently Windows does something with the ignored byte */
 343         if (!GetTempPathA( len, buffer )) buffer[0] = 'C';
 344         ret = toupper(buffer[0]);
 345         HeapFree( GetProcessHeap(), 0, buffer );
 346     }
 347     return MAKELONG( ret | (':' << 8), 1 );
 348 }
 349
 350
 351 /***********************************************************************
 352  *           GetWindowsDirectory   (KERNEL.134)
 353  */
 354 UINT16 WINAPI GetWindowsDirectory16( LPSTR path, UINT16 count )
 355 {
 356     return (UINT16)GetWindowsDirectoryA( path, count );
 357 }
 358
 359
 360 /***********************************************************************
 361  *           GetWindowsDirectoryW   (KERNEL32.@)
 362  *
 363  * See comment for GetWindowsDirectoryA.
 364  */
 365 UINT WINAPI GetWindowsDirectoryW( LPWSTR path, UINT count )
 366 {
 367     UINT len = strlenW( DIR_Windows.short_name ) + 1;
 368     if (path && count >= len)
 369     {
 370         strcpyW( path, DIR_Windows.short_name );
 371         len--;
 372     }
 373     return len;
 374 }
 375
 376
 377 /***********************************************************************
 378  *           GetWindowsDirectoryA   (KERNEL32.@)
 379  *
 380  * Return value:
 381  * If buffer is large enough to hold full path and terminating '\0' character
 382  * function copies path to buffer and returns length of the path without '\0'.
 383  * Otherwise function returns required size including '\0' character and
 384  * does not touch the buffer.
 385  */
 386 UINT WINAPI GetWindowsDirectoryA( LPSTR path, UINT count )
 387 {
 388     UINT len = WideCharToMultiByte( CP_ACP, 0, DIR_Windows.short_name, -1, NULL, 0, NULL, NULL );
 389     if (path && count >= len)
 390     {
 391         WideCharToMultiByte( CP_ACP, 0, DIR_Windows.short_name, -1, path, count, NULL, NULL );
 392         len--;
 393     }
 394     return len;
 395 }
 396
 397
 398 /***********************************************************************
 399  *           GetSystemWindowsDirectoryA   (KERNEL32.@) W2K, TS4.0SP4
 400  */
 401 UINT WINAPI GetSystemWindowsDirectoryA( LPSTR path, UINT count )
 402 {
 403     return GetWindowsDirectoryA( path, count );
 404 }
 405
 406
 407 /***********************************************************************
 408  *           GetSystemWindowsDirectoryW   (KERNEL32.@) W2K, TS4.0SP4
 409  */
 410 UINT WINAPI GetSystemWindowsDirectoryW( LPWSTR path, UINT count )
 411 {
 412     return GetWindowsDirectoryW( path, count );
 413 }
 414
 415
 416 /***********************************************************************
 417  *           GetSystemDirectory   (KERNEL.135)
 418  */
 419 UINT16 WINAPI GetSystemDirectory16( LPSTR path, UINT16 count )
 420 {
 421     return (UINT16)GetSystemDirectoryA( path, count );
 422 }
 423
 424
 425 /***********************************************************************
 426  *           GetSystemDirectoryW   (KERNEL32.@)
 427  *
 428  * See comment for GetWindowsDirectoryA.
 429  */
 430 UINT WINAPI GetSystemDirectoryW( LPWSTR path, UINT count )
 431 {
 432     UINT len = strlenW( DIR_System.short_name ) + 1;
 433     if (path && count >= len)
 434     {
 435         strcpyW( path, DIR_System.short_name );
 436         len--;
 437     }
 438     return len;
 439 }
 440
 441
 442 /***********************************************************************
 443  *           GetSystemDirectoryA   (KERNEL32.@)
 444  *
 445  * See comment for GetWindowsDirectoryA.
 446  */
 447 UINT WINAPI GetSystemDirectoryA( LPSTR path, UINT count )
 448 {
 449     UINT len = WideCharToMultiByte( CP_ACP, 0, DIR_System.short_name, -1, NULL, 0, NULL, NULL );
 450     if (path && count >= len)
 451     {
 452         WideCharToMultiByte( CP_ACP, 0, DIR_System.short_name, -1, path, count, NULL, NULL );
 453         len--;
 454     }
 455     return len;
 456 }
 457
 458
 459 /***********************************************************************
 460  *           CreateDirectory   (KERNEL.144)
 461  */
 462 BOOL16 WINAPI CreateDirectory16( LPCSTR path, LPVOID dummy )
 463 {
 464     TRACE_(file)("(%s,%p)\n", path, dummy );
 465     return (BOOL16)CreateDirectoryA( path, NULL );
 466 }
 467
 468
 469 /***********************************************************************
 470  *           CreateDirectoryW   (KERNEL32.@)
 471  * RETURNS:
 472  *      TRUE : success
 473  *      FALSE : failure
 474  *              ERROR_DISK_FULL:        on full disk
 475  *              ERROR_ALREADY_EXISTS:   if directory name exists (even as file)
 476  *              ERROR_ACCESS_DENIED:    on permission problems
 477  *              ERROR_FILENAME_EXCED_RANGE: too long filename(s)
 478  */
 479 BOOL WINAPI CreateDirectoryW( LPCWSTR path,
 480                                   LPSECURITY_ATTRIBUTES lpsecattribs )
 481 {
 482     DOS_FULL_NAME full_name;
 483
 484     if (!path || !*path)
 485     {
 486         SetLastError(ERROR_PATH_NOT_FOUND);
 487         return FALSE;
 488     }
 489
 490     TRACE_(file)("(%s,%p)\n", debugstr_w(path), lpsecattribs );
 491
 492     if (DOSFS_GetDevice( path ))
 493     {
 494         TRACE_(file)("cannot use device %s!\n", debugstr_w(path));
 495         SetLastError( ERROR_ACCESS_DENIED );
 496         return FALSE;
 497     }
 498     if (!DOSFS_GetFullName( path, FALSE, &full_name )) return 0;
 499     if (mkdir( full_name.long_name, 0777 ) == -1) {
 500         WARN_(file)("Error '%s' trying to create directory '%s'\n", strerror(errno), full_name.long_name);
 501         /* the FILE_SetDosError() generated error codes don't match the
 502          * CreateDirectory ones for some errnos */
 503         switch (errno) {
 504         case EEXIST:
 505         {
 506             if (!strcmp(DRIVE_GetRoot(full_name.drive), full_name.long_name))
 507                 SetLastError(ERROR_ACCESS_DENIED);
 508             else
 509                 SetLastError(ERROR_ALREADY_EXISTS);
 510             break;
 511         }
 512         case ENOSPC: SetLastError(ERROR_DISK_FULL); break;
 513         default: FILE_SetDosError();break;
 514         }
 515         return FALSE;
 516     }
 517     return TRUE;
 518 }
 519
 520
 521 /***********************************************************************
 522  *           CreateDirectoryA   (KERNEL32.@)
 523  */
 524 BOOL WINAPI CreateDirectoryA( LPCSTR path,
 525                                   LPSECURITY_ATTRIBUTES lpsecattribs )
 526 {
 527     UNICODE_STRING pathW;
 528     BOOL ret = FALSE;
 529
 530     if (!path || !*path)
 531     {
 532         SetLastError(ERROR_PATH_NOT_FOUND);
 533         return FALSE;
 534     }
 535
 536     if (RtlCreateUnicodeStringFromAsciiz(&pathW, path))
 537     {
 538         ret = CreateDirectoryW(pathW.Buffer, lpsecattribs);
 539         RtlFreeUnicodeString(&pathW);
 540     }
 541     else
 542         SetLastError(ERROR_NOT_ENOUGH_MEMORY);
 543     return ret;
 544 }
 545
 546
 547 /***********************************************************************
 548  *           CreateDirectoryExA   (KERNEL32.@)
 549  */
 550 BOOL WINAPI CreateDirectoryExA( LPCSTR template, LPCSTR path,
 551                                     LPSECURITY_ATTRIBUTES lpsecattribs)
 552 {
 553     return CreateDirectoryA(path,lpsecattribs);
 554 }
 555
 556
 557 /***********************************************************************
 558  *           CreateDirectoryExW   (KERNEL32.@)
 559  */
 560 BOOL WINAPI CreateDirectoryExW( LPCWSTR template, LPCWSTR path,
 561                                     LPSECURITY_ATTRIBUTES lpsecattribs)
 562 {
 563     return CreateDirectoryW(path,lpsecattribs);
 564 }
 565
 566
 567 /***********************************************************************
 568  *           RemoveDirectory   (KERNEL.145)
 569  */
 570 BOOL16 WINAPI RemoveDirectory16( LPCSTR path )
 571 {
 572     return (BOOL16)RemoveDirectoryA( path );
 573 }
 574
 575
 576 /***********************************************************************
 577  *           RemoveDirectoryW   (KERNEL32.@)
 578  */
 579 BOOL WINAPI RemoveDirectoryW( LPCWSTR path )
 580 {
 581     DOS_FULL_NAME full_name;
 582
 583     if (!path)
 584     {
 585         SetLastError(ERROR_INVALID_PARAMETER);
 586         return FALSE;
 587     }
 588
 589     TRACE_(file)("%s\n", debugstr_w(path));
 590
 591     if (DOSFS_GetDevice( path ))
 592     {
 593         TRACE_(file)("cannot remove device %s!\n", debugstr_w(path));
 594         SetLastError( ERROR_FILE_NOT_FOUND );
 595         return FALSE;
 596     }
 597     if (!DOSFS_GetFullName( path, TRUE, &full_name )) return FALSE;
 598     if (rmdir( full_name.long_name ) == -1)
 599     {
 600         FILE_SetDosError();
 601         return FALSE;
 602     }
 603     return TRUE;
 604 }
 605
 606
 607 /***********************************************************************
 608  *           RemoveDirectoryA   (KERNEL32.@)
 609  */
 610 BOOL WINAPI RemoveDirectoryA( LPCSTR path )
 611 {
 612     UNICODE_STRING pathW;
 613     BOOL ret = FALSE;
 614
 615     if (!path)
 616     {
 617         SetLastError(ERROR_INVALID_PARAMETER);
 618         return FALSE;
 619     }
 620
 621     if (RtlCreateUnicodeStringFromAsciiz(&pathW, path))
 622     {
 623         ret = RemoveDirectoryW(pathW.Buffer);
 624         RtlFreeUnicodeString(&pathW);
 625     }
 626     else
 627         SetLastError(ERROR_NOT_ENOUGH_MEMORY);
 628     return ret;
 629 }
 630
 631
 632 /***********************************************************************
 633  *           DIR_TryPath
 634  *
 635  * Helper function for DIR_SearchPath.
 636  */
 637 static BOOL DIR_TryPath( const DOS_FULL_NAME *dir, LPCWSTR name,
 638                            DOS_FULL_NAME *full_name )
 639 {
 640     LPSTR p_l = full_name->long_name + strlen(dir->long_name) + 1;
 641     LPWSTR p_s = full_name->short_name + strlenW(dir->short_name) + 1;
 642
 643     if ((p_s >= full_name->short_name + sizeof(full_name->short_name)/sizeof(full_name->short_name[0]) - 14) ||
 644         (p_l >= full_name->long_name + sizeof(full_name->long_name) - 1))
 645     {
 646         SetLastError( ERROR_PATH_NOT_FOUND );
 647         return FALSE;
 648     }
 649     if (!DOSFS_FindUnixName( dir, name, p_l,
 650                    sizeof(full_name->long_name) - (p_l - full_name->long_name),
 651                    p_s, !(DRIVE_GetFlags(dir->drive) & DRIVE_CASE_SENSITIVE) ))
 652         return FALSE;
 653
 654     full_name->drive = dir->drive;
 655     strcpy( full_name->long_name, dir->long_name );
 656     p_l[-1] = '/';
 657     strcpyW( full_name->short_name, dir->short_name );
 658     p_s[-1] = '\\';
 659     return TRUE;
 660 }
 661
 662 static BOOL DIR_SearchSemicolonedPaths(LPCWSTR name, DOS_FULL_NAME *full_name, LPWSTR pathlist)
 663 {
 664     LPWSTR next, buffer = NULL;
 665     INT len = strlenW(name), newlen, currlen = 0;
 666     BOOL ret = FALSE;
 667
 668     next = pathlist;
 669     while (!ret && next)
 670     {
 671         static const WCHAR bkslashW[] = {'\\',0};
 672         LPWSTR cur = next;
 673         while (*cur == ';') cur++;
 674         if (!*cur) break;
 675         next = strchrW( cur, ';' );
 676         if (next) *next++ = '\0';
 677         newlen = strlenW(cur) + len + 2;
 678
 679         if (newlen > currlen)
 680         {
 681             if (!(buffer = HeapReAlloc( GetProcessHeap(), 0, buffer, newlen * sizeof(WCHAR))))
 682                 goto done;
 683             currlen = newlen;
 684         }
 685
 686         strcpyW( buffer, cur );
 687         strcatW( buffer, bkslashW );
 688         strcatW( buffer, name );
 689         ret = DOSFS_GetFullName( buffer, TRUE, full_name );
 690     }
 691 done:
 692     HeapFree( GetProcessHeap(), 0, buffer );
 693     return ret;
 694 }
 695
 696
 697 /***********************************************************************
 698  *           DIR_TryEnvironmentPath
 699  *
 700  * Helper function for DIR_SearchPath.
 701  * Search in the specified path, or in $PATH if NULL.
 702  */
 703 static BOOL DIR_TryEnvironmentPath( LPCWSTR name, DOS_FULL_NAME *full_name, LPCWSTR envpath )
 704 {
 705     LPWSTR path;
 706     BOOL ret = FALSE;
 707     DWORD size;
 708     static const WCHAR pathW[] = {'P','A','T','H',0};
 709
 710     size = envpath ? strlenW(envpath)+1 : GetEnvironmentVariableW( pathW, NULL, 0 );
 711     if (!size) return FALSE;
 712     if (!(path = HeapAlloc( GetProcessHeap(), 0, size * sizeof(WCHAR) ))) return FALSE;
 713     if (envpath) strcpyW( path, envpath );
 714     else if (!GetEnvironmentVariableW( pathW, path, size )) goto done;
 715
 716     ret = DIR_SearchSemicolonedPaths(name, full_name, path);
 717
 718 done:
 719     HeapFree( GetProcessHeap(), 0, path );
 720     return ret;
 721 }
 722
 723
 724 /***********************************************************************
 725  *           DIR_TryModulePath
 726  *
 727  * Helper function for DIR_SearchPath.
 728  */
 729 static BOOL DIR_TryModulePath( LPCWSTR name, DOS_FULL_NAME *full_name, BOOL win32 )
 730 {
 731     WCHAR bufferW[MAX_PATH];
 732     LPWSTR p;
 733
 734     if (!win32)
 735     {
 736         char buffer[OFS_MAXPATHNAME];
 737         if (!GetCurrentTask()) return FALSE;
 738         if (!GetModuleFileName16( GetCurrentTask(), buffer, sizeof(buffer) ))
 739             return FALSE;
 740         MultiByteToWideChar(CP_ACP, 0, buffer, -1, bufferW, MAX_PATH);
 741     } else {
 742         if (!GetModuleFileNameW( 0, bufferW, MAX_PATH ) )
 743             return FALSE;
 744     }
 745     if (!(p = strrchrW( bufferW, '\\' ))) return FALSE;
 746     if (MAX_PATH - (++p - bufferW) <= strlenW(name)) return FALSE;
 747     strcpyW( p, name );
 748     return DOSFS_GetFullName( bufferW, TRUE, full_name );
 749 }
 750
 751
 752 /***********************************************************************
 753  *           DIR_TryAppPath
 754  *
 755  * Helper function for DIR_SearchPath.
 756  */
 757 static BOOL DIR_TryAppPath( LPCWSTR name, DOS_FULL_NAME *full_name )
 758 {
 759     HKEY hkAppPaths = 0, hkApp = 0;
 760     WCHAR buffer[MAX_PATHNAME_LEN], *lpAppPaths;
 761     LPWSTR lpFileName;
 762     BOOL res = FALSE;
 763     DWORD count;
 764     OBJECT_ATTRIBUTES attr;
 765     UNICODE_STRING nameW;
 766     KEY_VALUE_PARTIAL_INFORMATION *info;
 767     static const WCHAR PathW[] = {'P','a','t','h',0};
 768     static const WCHAR AppPathsW[] = {'M','a','c','h','i','n','e','\\',
 769                                       'S','o','f','t','w','a','r','e','\\',
 770                                       'M','i','c','r','o','s','o','f','t','\\',
 771                                       'W','i','n','d','o','w','s','\\',
 772                                       'C','u','r','r','e','n','t','V','e','r','s','i','o','n','\\',
 773                                       'A','p','p',' ','P','a','t','h','s',0};
 774
 775     attr.Length = sizeof(attr);
 776     attr.RootDirectory = 0;
 777     attr.ObjectName = &nameW;
 778     attr.Attributes = 0;
 779     attr.SecurityDescriptor = NULL;
 780     attr.SecurityQualityOfService = NULL;
 781     RtlInitUnicodeString( &nameW, AppPathsW );
 782     if (NtOpenKey( &hkAppPaths, KEY_ALL_ACCESS, &attr ) != STATUS_SUCCESS) return FALSE;
 783
 784     if (!GetModuleFileNameW(0, buffer, MAX_PATHNAME_LEN))
 785     {
 786         WARN("huh, module not found ??\n");
 787         goto end;
 788     }
 789     lpFileName = strrchrW(buffer, '\\');
 790     if (!lpFileName) lpFileName = buffer;
 791     else lpFileName++; /* skip '\\' */
 792
 793     attr.RootDirectory = hkAppPaths;
 794     RtlInitUnicodeString( &nameW, lpFileName );
 795     if (NtOpenKey( &hkApp, KEY_ALL_ACCESS, &attr ) != STATUS_SUCCESS) goto end;
 796
 797     RtlInitUnicodeString( &nameW, PathW );
 798     if (NtQueryValueKey( hkApp, &nameW, KeyValuePartialInformation,
 799                          buffer, sizeof(buffer)-sizeof(WCHAR), &count )) goto end;
 800     info = (KEY_VALUE_PARTIAL_INFORMATION *)buffer;
 801     lpAppPaths = (WCHAR *)info->Data;
 802     lpAppPaths[info->DataLength/sizeof(WCHAR)] = 0;
 803     res = DIR_SearchSemicolonedPaths(name, full_name, lpAppPaths);
 804 end:
 805     if (hkApp) NtClose(hkApp);
 806     if (hkAppPaths) NtClose(hkAppPaths);
 807     return res;
 808 }
 809
 810 /***********************************************************************
 811  *           DIR_SearchPath
 812  *
 813  * Implementation of SearchPathA. 'win32' specifies whether the search
 814  * order is Win16 (module path last) or Win32 (module path first).
 815  *
 816  * FIXME: should return long path names.
 817  */
 818 DWORD DIR_SearchPath( LPCWSTR path, LPCWSTR name, LPCWSTR ext,
 819                       DOS_FULL_NAME *full_name, BOOL win32 )
 820 {
 821     LPCWSTR p;
 822     LPWSTR tmp = NULL;
 823     BOOL ret = TRUE;
 824
 825     /* First check the supplied parameters */
 826
 827     p = strrchrW( name, '.' );
 828     if (p && !strchrW( p, '/' ) && !strchrW( p, '\\' ))
 829         ext = NULL;  /* Ignore the specified extension */
 830     if (FILE_contains_pathW (name))
 831         path = NULL;  /* Ignore path if name already contains a path */
 832     if (path && !*path) path = NULL;  /* Ignore empty path */
 833
 834     /* Allocate a buffer for the file name and extension */
 835
 836     if (ext)
 837     {
 838         DWORD len = strlenW(name) + strlenW(ext);
 839         if (!(tmp = HeapAlloc( GetProcessHeap(), 0, (len + 1) * sizeof(WCHAR) )))
 840         {
 841             SetLastError( ERROR_OUTOFMEMORY );
 842             return 0;
 843         }
 844         strcpyW( tmp, name );
 845         strcatW( tmp, ext );
 846         name = tmp;
 847     }
 848
 849     /* If the name contains an explicit path, everything's easy */
 850
 851     if (FILE_contains_pathW(name))
 852     {
 853         ret = DOSFS_GetFullName( name, TRUE, full_name );
 854         goto done;
 855     }
 856
 857     /* Search in the specified path */
 858
 859     if (path)
 860     {
 861         ret = DIR_TryEnvironmentPath( name, full_name, path );
 862         goto done;
 863     }
 864
 865     /* Try the path of the current executable (for Win32 search order) */
 866
 867     if (win32 && DIR_TryModulePath( name, full_name, win32 )) goto done;
 868
 869     /* Try the current directory */
 870
 871     if (DOSFS_GetFullName( name, TRUE, full_name )) goto done;
 872
 873     /* Try the Windows system directory */
 874
 875     if (DIR_TryPath( &DIR_System, name, full_name ))
 876         goto done;
 877
 878     /* Try the Windows directory */
 879
 880     if (DIR_TryPath( &DIR_Windows, name, full_name ))
 881         goto done;
 882
 883     /* Try the path of the current executable (for Win16 search order) */
 884
 885     if (!win32 && DIR_TryModulePath( name, full_name, win32 )) goto done;
 886
 887     /* Try the "App Paths" entry if existing (undocumented ??) */
 888     if (DIR_TryAppPath(name, full_name))
 889         goto done;
 890
 891     /* Try all directories in path */
 892
 893     ret = DIR_TryEnvironmentPath( name, full_name, NULL );
 894
 895 done:
 896     if (tmp) HeapFree( GetProcessHeap(), 0, tmp );
 897     return ret;
 898 }
 899
 900
 901 /***********************************************************************
 902  * SearchPathW [KERNEL32.@]
 903  *
 904  * Searches for a specified file in the search path.
 905  *
 906  * PARAMS
 907  *    path      [I] Path to search
 908  *    name      [I] Filename to search for.
 909  *    ext       [I] File extension to append to file name. The first
 910  *                  character must be a period. This parameter is
 911  *                  specified only if the filename given does not
 912  *                  contain an extension.
 913  *    buflen    [I] size of buffer, in characters
 914  *    buffer    [O] buffer for found filename
 915  *    lastpart  [O] address of pointer to last used character in
 916  *                  buffer (the final '\')
 917  *
 918  * RETURNS
 919  *    Success: length of string copied into buffer, not including
 920  *             terminating null character. If the filename found is
 921  *             longer than the length of the buffer, the length of the
 922  *             filename is returned.
 923  *    Failure: Zero
 924  *
 925  * NOTES
 926  *    If the file is not found, calls SetLastError(ERROR_FILE_NOT_FOUND)
 927  *    (tested on NT 4.0)
 928  */
 929 DWORD WINAPI SearchPathW( LPCWSTR path, LPCWSTR name, LPCWSTR ext, DWORD buflen,
 930                           LPWSTR buffer, LPWSTR *lastpart )
 931 {
 932     LPSTR res;
 933     DOS_FULL_NAME full_name;
 934
 935     if (!DIR_SearchPath( path, name, ext, &full_name, TRUE ))
 936     {
 937         SetLastError(ERROR_FILE_NOT_FOUND);
 938         return 0;
 939     }
 940
 941 TRACE("found %s %s\n", full_name.long_name, debugstr_w(full_name.short_name));
 942 TRACE("drive %c: root %s\n", 'A' + full_name.drive, DRIVE_GetRoot(full_name.drive));
 943
 944     lstrcpynW( buffer, full_name.short_name, buflen );
 945     res = full_name.long_name +
 946               strlen(DRIVE_GetRoot( full_name.drive ));
 947     while (*res == '/') res++;
 948     if (buflen)
 949     {
 950         LPWSTR p;
 951         if (buflen > 3)
 952         {
 953             MultiByteToWideChar(DRIVE_GetCodepage(full_name.drive), 0,
 954                                 res, -1, buffer + 3, buflen - 3);
 955             buffer[buflen - 1] = 0;
 956         }
 957         for (p = buffer; *p; p++) if (*p == '/') *p = '\\';
 958         if (lastpart) *lastpart = strrchrW( buffer, '\\' ) + 1;
 959     }
 960     TRACE("Returning %s\n", debugstr_w(buffer) );
 961     return strlenW(buffer);
 962 }
 963
 964
 965 /***********************************************************************
 966  *           SearchPathA   (KERNEL32.@)
 967  */
 968 DWORD WINAPI SearchPathA( LPCSTR path, LPCSTR name, LPCSTR ext,
 969                           DWORD buflen, LPSTR buffer, LPSTR *lastpart )
 970 {
 971     UNICODE_STRING pathW, nameW, extW;
 972     WCHAR bufferW[MAX_PATH];
 973     DWORD ret, retW;
 974
 975     if (path) RtlCreateUnicodeStringFromAsciiz(&pathW, path);
 976     else pathW.Buffer = NULL;
 977     if (name) RtlCreateUnicodeStringFromAsciiz(&nameW, name);
 978     else nameW.Buffer = NULL;
 979     if (ext) RtlCreateUnicodeStringFromAsciiz(&extW, ext);
 980     else extW.Buffer = NULL;
 981
 982     retW = SearchPathW(pathW.Buffer, nameW.Buffer, extW.Buffer, MAX_PATH, bufferW, NULL);
 983
 984     if (!retW)
 985         ret = 0;
 986     else if (retW > MAX_PATH)
 987     {
 988         SetLastError(ERROR_FILENAME_EXCED_RANGE);
 989         ret = 0;
 990     }
 991     else
 992     {
 993         ret = WideCharToMultiByte(CP_ACP, 0, bufferW, -1, NULL, 0, NULL, NULL);
 994         if (buflen >= ret)
 995         {
 996             WideCharToMultiByte(CP_ACP, 0, bufferW, -1, buffer, buflen, NULL, NULL);
 997             ret--; /* length without 0 */
 998             if (lastpart) *lastpart = strrchr(buffer, '\\') + 1;
 999         }
1000     }
1001
1002     RtlFreeUnicodeString(&pathW);
1003     RtlFreeUnicodeString(&nameW);
1004     RtlFreeUnicodeString(&extW);
1005     return ret;
1006 }
1007
1008
1009 /***********************************************************************
1010  *           search_alternate_path
1011  *
1012  *
1013  * FIXME: should return long path names.?
1014  */
1015 static BOOL search_alternate_path(LPCWSTR dll_path, LPCWSTR name, LPCWSTR ext,
1016                                   DOS_FULL_NAME *full_name)
1017 {
1018     LPCWSTR p;
1019     LPWSTR tmp = NULL;
1020     BOOL ret = TRUE;
1021
1022     /* First check the supplied parameters */
1023
1024     p = strrchrW( name, '.' );
1025     if (p && !strchrW( p, '/' ) && !strchrW( p, '\\' ))
1026         ext = NULL;  /* Ignore the specified extension */
1027
1028     /* Allocate a buffer for the file name and extension */
1029
1030     if (ext)
1031     {
1032         DWORD len = strlenW(name) + strlenW(ext);
1033         if (!(tmp = HeapAlloc( GetProcessHeap(), 0, (len + 1) * sizeof(WCHAR) )))
1034         {
1035             SetLastError( ERROR_OUTOFMEMORY );
1036             return 0;
1037         }
1038         strcpyW( tmp, name );
1039         strcatW( tmp, ext );
1040         name = tmp;
1041     }
1042
1043     if (DIR_TryEnvironmentPath (name, full_name, dll_path))
1044         ;
1045     else if (DOSFS_GetFullName (name, TRUE, full_name)) /* current dir */
1046         ;
1047     else if (DIR_TryPath (&DIR_System, name, full_name)) /* System dir */
1048         ;
1049     else if (DIR_TryPath (&DIR_Windows, name, full_name)) /* Windows dir */
1050         ;
1051     else
1052         ret = DIR_TryEnvironmentPath( name, full_name, NULL );
1053
1054     if (tmp) HeapFree( GetProcessHeap(), 0, tmp );
1055     return ret;
1056 }
1057
1058
1059 /***********************************************************************
1060  * DIR_SearchAlternatePath
1061  *
1062  * Searches for a specified file in the search path.
1063  *
1064  * PARAMS
1065  *    dll_path  [I] Path to search
1066  *    name      [I] Filename to search for.
1067  *    ext       [I] File extension to append to file name. The first
1068  *                  character must be a period. This parameter is
1069  *                  specified only if the filename given does not
1070  *                  contain an extension.
1071  *    buflen    [I] size of buffer, in characters
1072  *    buffer    [O] buffer for found filename
1073  *    lastpart  [O] address of pointer to last used character in
1074  *                  buffer (the final '\') (May be NULL)
1075  *
1076  * RETURNS
1077  *    Success: length of string copied into buffer, not including
1078  *             terminating null character. If the filename found is
1079  *             longer than the length of the buffer, the length of the
1080  *             filename is returned.
1081  *    Failure: Zero
1082  *
1083  * NOTES
1084  *    If the file is not found, calls SetLastError(ERROR_FILE_NOT_FOUND)
1085  *
1086  * FIXME: convert to unicode
1087  */
1088 DWORD DIR_SearchAlternatePath( LPCSTR dll_path, LPCSTR name, LPCSTR ext,
1089                                DWORD buflen, LPSTR buffer, LPSTR *lastpart )
1090 {
1091     LPSTR p;
1092     DOS_FULL_NAME full_name;
1093     DWORD ret = 0;
1094     UNICODE_STRING dll_pathW, nameW, extW;
1095
1096     if (dll_path) RtlCreateUnicodeStringFromAsciiz(&dll_pathW, dll_path);
1097     else dll_pathW.Buffer = NULL;
1098     if (name) RtlCreateUnicodeStringFromAsciiz(&nameW, name);
1099     else nameW.Buffer = NULL;
1100     if (ext) RtlCreateUnicodeStringFromAsciiz(&extW, ext);
1101     else extW.Buffer = NULL;
1102
1103     if (search_alternate_path( dll_pathW.Buffer, nameW.Buffer, extW.Buffer, &full_name))
1104     {
1105         ret = WideCharToMultiByte(CP_ACP, 0, full_name.short_name, -1, NULL, 0, NULL, NULL);
1106         if (buflen >= ret)
1107         {
1108             WideCharToMultiByte(CP_ACP, 0, full_name.short_name, -1, buffer, buflen, NULL, NULL);
1109             for (p = buffer; *p; p++) if (*p == '/') *p = '\\';
1110             if (lastpart) *lastpart = strrchr( buffer, '\\' ) + 1;
1111             ret--; /* length without 0 */
1112         }
1113     }
1114     else
1115         SetLastError(ERROR_FILE_NOT_FOUND);
1116
1117     RtlFreeUnicodeString(&dll_pathW);
1118     RtlFreeUnicodeString(&nameW);
1119     RtlFreeUnicodeString(&extW);
1120
1121     TRACE("Returning %ld\n", ret );
1122     return ret;
1123 }