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