Call IShellExecuteHook interface for ShellExecute() calls with ID
[wine/multimedia.git] / dlls / ntdll / path.c
blob1dd7bcc7016c25aaed745972eed0436d1a19e822
1 /*
2 * Ntdll path functions
4 * Copyright 2002, 2003 Alexandre Julliard
5 * Copyright 2003 Eric Pouech
7 * This library is free software; you can redistribute it and/or
8 * modify it under the terms of the GNU Lesser General Public
9 * License as published by the Free Software Foundation; either
10 * version 2.1 of the License, or (at your option) any later version.
12 * This library is distributed in the hope that it will be useful,
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
15 * Lesser General Public License for more details.
17 * You should have received a copy of the GNU Lesser General Public
18 * License along with this library; if not, write to the Free Software
19 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
22 #include "config.h"
24 #include <stdarg.h>
26 #include "windef.h"
27 #include "winbase.h"
28 #include "winreg.h"
29 #include "winternl.h"
30 #include "wine/unicode.h"
31 #include "wine/debug.h"
32 #include "ntdll_misc.h"
34 WINE_DEFAULT_DEBUG_CHANNEL(file);
36 static const WCHAR DeviceRootW[] = {'\\','\\','.','\\',0};
37 static const WCHAR NTDosPrefixW[] = {'\\','?','?','\\',0};
38 static const WCHAR UncPfxW[] = {'U','N','C','\\',0};
40 /* FIXME: hack! */
41 HANDLE (WINAPI *pCreateFileW)( LPCWSTR filename, DWORD access, DWORD sharing,
42 LPSECURITY_ATTRIBUTES sa, DWORD creation,
43 DWORD attributes, HANDLE template );
45 #define IS_SEPARATOR(ch) ((ch) == '\\' || (ch) == '/')
47 /***********************************************************************
48 * RtlDetermineDosPathNameType_U (NTDLL.@)
50 DOS_PATHNAME_TYPE WINAPI RtlDetermineDosPathNameType_U( PCWSTR path )
52 if (IS_SEPARATOR(path[0]))
54 if (!IS_SEPARATOR(path[1])) return ABSOLUTE_PATH; /* "/foo" */
55 if (path[2] != '.') return UNC_PATH; /* "//foo" */
56 if (IS_SEPARATOR(path[3])) return DEVICE_PATH; /* "//./foo" */
57 if (path[3]) return UNC_PATH; /* "//.foo" */
58 return UNC_DOT_PATH; /* "//." */
60 else
62 if (!path[0] || path[1] != ':') return RELATIVE_PATH; /* "foo" */
63 if (IS_SEPARATOR(path[2])) return ABSOLUTE_DRIVE_PATH; /* "c:/foo" */
64 return RELATIVE_DRIVE_PATH; /* "c:foo" */
68 /******************************************************************
69 * RtlDoesFileExists_U
71 * FIXME: should not use CreateFileW
73 BOOLEAN WINAPI RtlDoesFileExists_U(LPCWSTR file_name)
75 HANDLE handle = pCreateFileW( file_name, 0, FILE_SHARE_READ|FILE_SHARE_WRITE,
76 NULL, OPEN_EXISTING, 0, 0 );
77 if (handle == INVALID_HANDLE_VALUE) return FALSE;
78 NtClose( handle );
79 return TRUE;
82 /***********************************************************************
83 * RtlIsDosDeviceName_U (NTDLL.@)
85 * Check if the given DOS path contains a DOS device name.
87 * Returns the length of the device name in the low word and its
88 * position in the high word (both in bytes, not WCHARs), or 0 if no
89 * device name is found.
91 ULONG WINAPI RtlIsDosDeviceName_U( PCWSTR dos_name )
93 static const WCHAR consoleW[] = {'\\','\\','.','\\','C','O','N',0};
94 static const WCHAR auxW[3] = {'A','U','X'};
95 static const WCHAR comW[3] = {'C','O','M'};
96 static const WCHAR conW[3] = {'C','O','N'};
97 static const WCHAR lptW[3] = {'L','P','T'};
98 static const WCHAR nulW[3] = {'N','U','L'};
99 static const WCHAR prnW[3] = {'P','R','N'};
101 const WCHAR *start, *end, *p;
103 switch(RtlDetermineDosPathNameType_U( dos_name ))
105 case INVALID_PATH:
106 case UNC_PATH:
107 return 0;
108 case DEVICE_PATH:
109 if (!strcmpiW( dos_name, consoleW ))
110 return MAKELONG( sizeof(conW), 4 * sizeof(WCHAR) ); /* 4 is length of \\.\ prefix */
111 return 0;
112 default:
113 break;
116 end = dos_name + strlenW(dos_name) - 1;
117 if (end >= dos_name && *end == ':') end--; /* remove trailing ':' */
119 /* find start of file name */
120 for (start = end; start >= dos_name; start--)
122 if (IS_SEPARATOR(start[0])) break;
123 /* check for ':' but ignore if before extension (for things like NUL:.txt) */
124 if (start[0] == ':' && start[1] != '.') break;
126 start++;
128 /* remove extension */
129 if ((p = strchrW( start, '.' )))
131 end = p - 1;
132 if (end >= dos_name && *end == ':') end--; /* remove trailing ':' before extension */
134 else
136 /* no extension, remove trailing spaces */
137 while (end >= dos_name && *end == ' ') end--;
140 /* now we have a potential device name between start and end, check it */
141 switch(end - start + 1)
143 case 3:
144 if (strncmpiW( start, auxW, 3 ) &&
145 strncmpiW( start, conW, 3 ) &&
146 strncmpiW( start, nulW, 3 ) &&
147 strncmpiW( start, prnW, 3 )) break;
148 return MAKELONG( 3 * sizeof(WCHAR), (start - dos_name) * sizeof(WCHAR) );
149 case 4:
150 if (strncmpiW( start, comW, 3 ) && strncmpiW( start, lptW, 3 )) break;
151 if (*end <= '0' || *end > '9') break;
152 return MAKELONG( 4 * sizeof(WCHAR), (start - dos_name) * sizeof(WCHAR) );
153 default: /* can't match anything */
154 break;
156 return 0;
160 /**************************************************************************
161 * RtlDosPathNameToNtPathName_U [NTDLL.@]
163 * dos_path: a DOS path name (fully qualified or not)
164 * ntpath: pointer to a UNICODE_STRING to hold the converted
165 * path name
166 * file_part:will point (in ntpath) to the file part in the path
167 * cd: directory reference (optional)
169 * FIXME:
170 * + fill the cd structure
172 BOOLEAN WINAPI RtlDosPathNameToNtPathName_U(PWSTR dos_path,
173 PUNICODE_STRING ntpath,
174 PWSTR* file_part,
175 CURDIR* cd)
177 static const WCHAR LongFileNamePfxW[4] = {'\\','\\','?','\\'};
178 ULONG sz, ptr_sz, offset;
179 WCHAR local[MAX_PATH];
180 LPWSTR ptr;
182 TRACE("(%s,%p,%p,%p)\n",
183 debugstr_w(dos_path), ntpath, file_part, cd);
185 if (cd)
187 FIXME("Unsupported parameter\n");
188 memset(cd, 0, sizeof(*cd));
191 if (!dos_path || !*dos_path) return FALSE;
193 if (!memcmp(dos_path, LongFileNamePfxW, sizeof(LongFileNamePfxW)))
195 dos_path += sizeof(LongFileNamePfxW) / sizeof(WCHAR);
196 ptr = NULL;
197 ptr_sz = 0;
199 else
201 ptr = local;
202 ptr_sz = sizeof(local);
204 sz = RtlGetFullPathName_U(dos_path, ptr_sz, ptr, file_part);
205 if (sz == 0) return FALSE;
206 if (sz > ptr_sz)
208 ptr = RtlAllocateHeap(GetProcessHeap(), 0, sz);
209 sz = RtlGetFullPathName_U(dos_path, sz, ptr, file_part);
212 ntpath->MaximumLength = sz + (4 /* unc\ */ + 4 /* \??\ */) * sizeof(WCHAR);
213 ntpath->Buffer = RtlAllocateHeap(GetProcessHeap(), 0, ntpath->MaximumLength);
214 if (!ntpath->Buffer)
216 if (ptr != local) RtlFreeHeap(GetProcessHeap(), 0, ptr);
217 return FALSE;
220 strcpyW(ntpath->Buffer, NTDosPrefixW);
221 offset = 0;
222 switch (RtlDetermineDosPathNameType_U(ptr))
224 case UNC_PATH: /* \\foo */
225 if (ptr[2] != '?')
227 offset = 2;
228 strcatW(ntpath->Buffer, UncPfxW);
230 break;
231 case DEVICE_PATH: /* \\.\foo */
232 offset = 4;
233 break;
234 default: break; /* needed to keep gcc quiet */
237 strcatW(ntpath->Buffer, ptr + offset);
238 ntpath->Length = strlenW(ntpath->Buffer) * sizeof(WCHAR);
240 if (file_part && *file_part)
241 *file_part = ntpath->Buffer + ntpath->Length / sizeof(WCHAR) - strlenW(*file_part);
243 /* FIXME: cd filling */
245 if (ptr != local) RtlFreeHeap(GetProcessHeap(), 0, ptr);
246 return TRUE;
249 /******************************************************************
250 * RtlDosSearchPath_U
252 * Searchs a file of name 'name' into a ';' separated list of paths
253 * (stored in paths)
254 * Doesn't seem to search elsewhere than the paths list
255 * Stores the result in buffer (file_part will point to the position
256 * of the file name in the buffer)
257 * FIXME:
258 * - how long shall the paths be ??? (MAX_PATH or larger with \\?\ constructs ???)
260 ULONG WINAPI RtlDosSearchPath_U(LPCWSTR paths, LPCWSTR search, LPCWSTR ext,
261 ULONG buffer_size, LPWSTR buffer,
262 LPWSTR* file_part)
264 DOS_PATHNAME_TYPE type = RtlDetermineDosPathNameType_U(search);
265 ULONG len = 0;
267 if (type == RELATIVE_PATH)
269 ULONG allocated = 0, needed, filelen;
270 WCHAR *name = NULL;
272 filelen = 1 /* for \ */ + strlenW(search) + 1 /* \0 */;
274 /* Windows only checks for '.' without worrying about path components */
275 if (strchrW( search, '.' )) ext = NULL;
276 if (ext != NULL) filelen += strlenW(ext);
278 while (*paths)
280 LPCWSTR ptr;
282 for (needed = 0, ptr = paths; *ptr != 0 && *ptr++ != ';'; needed++);
283 if (needed + filelen > allocated)
285 if (!name) name = RtlAllocateHeap(GetProcessHeap(), 0,
286 (needed + filelen) * sizeof(WCHAR));
287 else
289 WCHAR *newname = RtlReAllocateHeap(GetProcessHeap(), 0, name,
290 (needed + filelen) * sizeof(WCHAR));
291 if (!newname) RtlFreeHeap(GetProcessHeap(), 0, name);
292 name = newname;
294 if (!name) return 0;
295 allocated = needed + filelen;
297 memmove(name, paths, needed * sizeof(WCHAR));
298 /* append '\\' if none is present */
299 if (needed > 0 && name[needed - 1] != '\\') name[needed++] = '\\';
300 strcpyW(&name[needed], search);
301 if (ext) strcatW(&name[needed], ext);
302 if (RtlDoesFileExists_U(name))
304 len = RtlGetFullPathName_U(name, buffer_size, buffer, file_part);
305 break;
307 paths = ptr;
309 RtlFreeHeap(GetProcessHeap(), 0, name);
311 else if (RtlDoesFileExists_U(search))
313 len = RtlGetFullPathName_U(search, buffer_size, buffer, file_part);
316 return len;
320 /******************************************************************
321 * collapse_path
323 * Helper for RtlGetFullPathName_U.
324 * Get rid of . and .. components in the path.
326 static inline void collapse_path( WCHAR *path, UINT mark )
328 WCHAR *p, *next;
330 /* convert every / into a \ */
331 for (p = path; *p; p++) if (*p == '/') *p = '\\';
333 /* collapse duplicate backslashes */
334 next = path + max( 1, mark );
335 for (p = next; *p; p++) if (*p != '\\' || next[-1] != '\\') *next++ = *p;
336 *next = 0;
338 p = path + mark;
339 while (*p)
341 if (*p == '.')
343 switch(p[1])
345 case '\\': /* .\ component */
346 next = p + 2;
347 memmove( p, next, (strlenW(next) + 1) * sizeof(WCHAR) );
348 continue;
349 case 0: /* final . */
350 if (p > path + mark) p--;
351 *p = 0;
352 continue;
353 case '.':
354 if (p[2] == '\\') /* ..\ component */
356 next = p + 3;
357 if (p > path + mark)
359 p--;
360 while (p > path + mark && p[-1] != '\\') p--;
362 memmove( p, next, (strlenW(next) + 1) * sizeof(WCHAR) );
363 continue;
365 else if (!p[2]) /* final .. */
367 if (p > path + mark)
369 p--;
370 while (p > path + mark && p[-1] != '\\') p--;
371 if (p > path + mark) p--;
373 *p = 0;
374 continue;
376 break;
379 /* skip to the next component */
380 while (*p && *p != '\\') p++;
381 if (*p == '\\') p++;
385 /******************************************************************
386 * get_full_path_helper
388 * Helper for RtlGetFullPathName_U
389 * Note: name and buffer are allowed to point to the same memory spot
391 static ULONG get_full_path_helper(LPCWSTR name, LPWSTR buffer, ULONG size)
393 ULONG reqsize = 0, mark = 0, dep = 0, deplen;
394 DOS_PATHNAME_TYPE type;
395 LPWSTR ins_str = NULL;
396 LPCWSTR ptr;
397 const UNICODE_STRING* cd;
398 WCHAR tmp[4];
400 RtlAcquirePebLock();
402 cd = &NtCurrentTeb()->Peb->ProcessParameters->CurrentDirectoryName;
404 switch (type = RtlDetermineDosPathNameType_U(name))
406 case UNC_PATH: /* \\foo */
407 ptr = name + 2;
408 while (*ptr && !IS_SEPARATOR(*ptr)) ptr++; /* share name */
409 while (*ptr && IS_SEPARATOR(*ptr)) ptr++;
410 while (*ptr && !IS_SEPARATOR(*ptr)) ptr++; /* dir name */
411 mark = (ptr - name);
412 break;
414 case DEVICE_PATH: /* \\.\foo */
415 mark = 4;
416 break;
418 case ABSOLUTE_DRIVE_PATH: /* c:\foo */
419 reqsize = sizeof(WCHAR);
420 tmp[0] = toupperW(name[0]);
421 ins_str = tmp;
422 dep = 1;
423 mark = 3;
424 break;
426 case RELATIVE_DRIVE_PATH: /* c:foo */
427 dep = 2;
428 if (toupperW(name[0]) != toupperW(cd->Buffer[0]) || cd->Buffer[1] != ':')
430 UNICODE_STRING var, val;
432 tmp[0] = '=';
433 tmp[1] = name[0];
434 tmp[2] = ':';
435 tmp[3] = '\0';
436 var.Length = 3 * sizeof(WCHAR);
437 var.MaximumLength = 4 * sizeof(WCHAR);
438 var.Buffer = tmp;
439 val.Length = 0;
440 val.MaximumLength = size;
441 val.Buffer = RtlAllocateHeap(GetProcessHeap(), 0, size);
443 switch (RtlQueryEnvironmentVariable_U(NULL, &var, &val))
445 case STATUS_SUCCESS:
446 /* FIXME: Win2k seems to check that the environment variable actually points
447 * to an existing directory. If not, root of the drive is used
448 * (this seems also to be the only spot in RtlGetFullPathName that the
449 * existence of a part of a path is checked)
451 /* fall thru */
452 case STATUS_BUFFER_TOO_SMALL:
453 reqsize = val.Length + sizeof(WCHAR); /* append trailing '\\' */
454 val.Buffer[val.Length / sizeof(WCHAR)] = '\\';
455 ins_str = val.Buffer;
456 break;
457 case STATUS_VARIABLE_NOT_FOUND:
458 reqsize = 3 * sizeof(WCHAR);
459 tmp[0] = name[0];
460 tmp[1] = ':';
461 tmp[2] = '\\';
462 ins_str = tmp;
463 break;
464 default:
465 ERR("Unsupported status code\n");
466 break;
468 mark = 3;
469 break;
471 /* fall through */
473 case RELATIVE_PATH: /* foo */
474 reqsize = cd->Length;
475 ins_str = cd->Buffer;
476 if (cd->Buffer[1] != ':')
478 ptr = strchrW(cd->Buffer + 2, '\\');
479 if (ptr) ptr = strchrW(ptr + 1, '\\');
480 if (!ptr) ptr = cd->Buffer + strlenW(cd->Buffer);
481 mark = ptr - cd->Buffer;
483 else mark = 3;
484 break;
486 case ABSOLUTE_PATH: /* \xxx */
487 if (cd->Buffer[1] == ':')
489 reqsize = 2 * sizeof(WCHAR);
490 tmp[0] = cd->Buffer[0];
491 tmp[1] = ':';
492 ins_str = tmp;
493 mark = 3;
495 else
497 ptr = strchrW(cd->Buffer + 2, '\\');
498 if (ptr) ptr = strchrW(ptr + 1, '\\');
499 if (!ptr) ptr = cd->Buffer + strlenW(cd->Buffer);
500 reqsize = (ptr - cd->Buffer) * sizeof(WCHAR);
501 mark = reqsize / sizeof(WCHAR);
502 ins_str = cd->Buffer;
504 break;
506 case UNC_DOT_PATH: /* \\. */
507 reqsize = 4 * sizeof(WCHAR);
508 dep = 3;
509 tmp[0] = '\\';
510 tmp[1] = '\\';
511 tmp[2] = '.';
512 tmp[3] = '\\';
513 ins_str = tmp;
514 mark = 4;
515 break;
517 case INVALID_PATH:
518 goto done;
521 /* enough space ? */
522 deplen = strlenW(name + dep) * sizeof(WCHAR);
523 if (reqsize + deplen + sizeof(WCHAR) > size)
525 /* not enough space, return need size (including terminating '\0') */
526 reqsize += deplen + sizeof(WCHAR);
527 goto done;
530 memmove(buffer + reqsize / sizeof(WCHAR), name + dep, deplen + sizeof(WCHAR));
531 if (reqsize) memcpy(buffer, ins_str, reqsize);
532 reqsize += deplen;
534 if (ins_str && ins_str != tmp && ins_str != cd->Buffer)
535 RtlFreeHeap(GetProcessHeap(), 0, ins_str);
537 collapse_path( buffer, mark );
538 reqsize = strlenW(buffer) * sizeof(WCHAR);
540 done:
541 RtlReleasePebLock();
542 return reqsize;
545 /******************************************************************
546 * RtlGetFullPathName_U (NTDLL.@)
548 * Returns the number of bytes written to buffer (not including the
549 * terminating NULL) if the function succeeds, or the required number of bytes
550 * (including the terminating NULL) if the buffer is too small.
552 * file_part will point to the filename part inside buffer (except if we use
553 * DOS device name, in which case file_in_buf is NULL)
556 DWORD WINAPI RtlGetFullPathName_U(const WCHAR* name, ULONG size, WCHAR* buffer,
557 WCHAR** file_part)
559 WCHAR* ptr;
560 DWORD dosdev;
561 DWORD reqsize;
563 TRACE("(%s %lu %p %p)\n", debugstr_w(name), size, buffer, file_part);
565 if (!name || !*name) return 0;
567 if (file_part) *file_part = NULL;
569 /* check for DOS device name */
570 dosdev = RtlIsDosDeviceName_U(name);
571 if (dosdev)
573 DWORD offset = HIWORD(dosdev) / sizeof(WCHAR); /* get it in WCHARs, not bytes */
574 DWORD sz = LOWORD(dosdev); /* in bytes */
576 if (8 + sz + 2 > size) return sz + 10;
577 strcpyW(buffer, DeviceRootW);
578 memmove(buffer + 4, name + offset, sz);
579 buffer[4 + sz / sizeof(WCHAR)] = '\0';
580 /* file_part isn't set in this case */
581 return sz + 8;
584 reqsize = get_full_path_helper(name, buffer, size);
585 if (reqsize > size)
587 LPWSTR tmp = RtlAllocateHeap(GetProcessHeap(), 0, reqsize);
588 reqsize = get_full_path_helper(name, tmp, reqsize);
589 if (reqsize > size) /* it may have worked the second time */
591 RtlFreeHeap(GetProcessHeap(), 0, tmp);
592 return reqsize + sizeof(WCHAR);
594 memcpy( buffer, tmp, reqsize + sizeof(WCHAR) );
595 RtlFreeHeap(GetProcessHeap(), 0, tmp);
598 /* find file part */
599 if (file_part && (ptr = strrchrW(buffer, '\\')) != NULL && ptr >= buffer + 2 && *++ptr)
600 *file_part = ptr;
601 return reqsize;
604 /*************************************************************************
605 * RtlGetLongestNtPathLength [NTDLL.@]
607 * Get the longest allowed path length
609 * PARAMS
610 * None.
612 * RETURNS
613 * The longest allowed path length (277 characters under Win2k).
615 DWORD WINAPI RtlGetLongestNtPathLength(void)
617 return 277;
620 /******************************************************************
621 * RtlIsNameLegalDOS8Dot3 (NTDLL.@)
623 * Returns TRUE iff unicode is a valid DOS (8+3) name.
624 * If the name is valid, oem gets filled with the corresponding OEM string
625 * spaces is set to TRUE if unicode contains spaces
627 BOOLEAN WINAPI RtlIsNameLegalDOS8Dot3( const UNICODE_STRING *unicode,
628 OEM_STRING *oem, BOOLEAN *spaces )
630 static const char* illegal = "*?<>|\"+=,;[]:/\\\345";
631 int dot = -1;
632 unsigned int i;
633 char buffer[12];
634 OEM_STRING oem_str;
635 BOOLEAN got_space = FALSE;
637 if (!oem)
639 oem_str.Length = sizeof(buffer);
640 oem_str.MaximumLength = sizeof(buffer);
641 oem_str.Buffer = buffer;
642 oem = &oem_str;
644 if (RtlUpcaseUnicodeStringToCountedOemString( oem, unicode, FALSE ) != STATUS_SUCCESS)
645 return FALSE;
647 if (oem->Length > 12) return FALSE;
649 /* a starting . is invalid, except for . and .. */
650 if (oem->Buffer[0] == '.')
652 if (oem->Length != 1 && (oem->Length != 2 || oem->Buffer[1] != '.')) return FALSE;
653 if (spaces) *spaces = FALSE;
654 return TRUE;
657 for (i = 0; i < oem->Length; i++)
659 switch (oem->Buffer[i])
661 case ' ':
662 /* leading/trailing spaces not allowed */
663 if (!i || i == oem->Length-1 || oem->Buffer[i+1] == '.') return FALSE;
664 got_space = TRUE;
665 break;
666 case '.':
667 if (dot != -1) return FALSE;
668 dot = i;
669 break;
670 default:
671 if (strchr(illegal, oem->Buffer[i])) return FALSE;
672 break;
675 /* check file part is shorter than 8, extension shorter than 3
676 * dot cannot be last in string
678 if (dot == -1)
680 if (oem->Length > 8) return FALSE;
682 else
684 if (dot > 8 || (oem->Length - dot > 4) || dot == oem->Length - 1) return FALSE;
686 if (spaces) *spaces = got_space;
687 return TRUE;
690 /******************************************************************
691 * RtlGetCurrentDirectory_U (NTDLL.@)
694 NTSTATUS WINAPI RtlGetCurrentDirectory_U(ULONG buflen, LPWSTR buf)
696 UNICODE_STRING* us;
697 ULONG len;
699 TRACE("(%lu %p)\n", buflen, buf);
701 RtlAcquirePebLock();
703 us = &NtCurrentTeb()->Peb->ProcessParameters->CurrentDirectoryName;
704 len = us->Length / sizeof(WCHAR);
705 if (us->Buffer[len - 1] == '\\' && us->Buffer[len - 2] != ':')
706 len--;
708 if (buflen / sizeof(WCHAR) > len)
710 memcpy(buf, us->Buffer, len * sizeof(WCHAR));
711 buf[len] = '\0';
713 else
715 len++;
718 RtlReleasePebLock();
720 return len * sizeof(WCHAR);
723 /******************************************************************
724 * RtlSetCurrentDirectory_U (NTDLL.@)
727 NTSTATUS WINAPI RtlSetCurrentDirectory_U(const UNICODE_STRING* dir)
729 UNICODE_STRING* curdir;
730 NTSTATUS nts = STATUS_SUCCESS;
731 ULONG size;
732 PWSTR buf = NULL;
734 TRACE("(%s)\n", debugstr_w(dir->Buffer));
736 RtlAcquirePebLock();
738 curdir = &NtCurrentTeb()->Peb->ProcessParameters->CurrentDirectoryName;
739 size = curdir->MaximumLength;
741 buf = RtlAllocateHeap(GetProcessHeap(), 0, size);
742 if (buf == NULL)
744 nts = STATUS_NO_MEMORY;
745 goto out;
748 size = RtlGetFullPathName_U(dir->Buffer, size, buf, 0);
749 if (!size)
751 nts = STATUS_OBJECT_NAME_INVALID;
752 goto out;
755 switch (RtlDetermineDosPathNameType_U(buf))
757 case ABSOLUTE_DRIVE_PATH:
758 case UNC_PATH:
759 break;
760 default:
761 FIXME("Don't support those cases yes\n");
762 nts = STATUS_NOT_IMPLEMENTED;
763 goto out;
766 /* FIXME: should check that the directory actually exists,
767 * and fill CurrentDirectoryHandle accordingly
770 /* append trailing \ if missing */
771 if (buf[size / sizeof(WCHAR) - 1] != '\\')
773 buf[size / sizeof(WCHAR)] = '\\';
774 buf[size / sizeof(WCHAR) + 1] = '\0';
775 size += sizeof(WCHAR);
778 memmove(curdir->Buffer, buf, size + sizeof(WCHAR));
779 curdir->Length = size;
781 #if 0
782 if (curdir->Buffer[1] == ':')
784 UNICODE_STRING env;
785 WCHAR var[4];
787 var[0] = '=';
788 var[1] = curdir->Buffer[0];
789 var[2] = ':';
790 var[3] = 0;
791 env.Length = 3 * sizeof(WCHAR);
792 env.MaximumLength = 4 * sizeof(WCHAR);
793 env.Buffer = var;
795 RtlSetEnvironmentVariable(NULL, &env, curdir);
797 #endif
799 out:
800 if (buf) RtlFreeHeap(GetProcessHeap(), 0, buf);
802 RtlReleasePebLock();
804 return nts;