Zero entries were skipped, fix from Adrian Goalby
[p5sagit/p5-mst-13.2.git] / win32 / vdir.h
CommitLineData
7766f137 1/* vdir.h
2 *
3 * (c) 1999 Microsoft Corporation. All rights reserved.
4 * Portions (c) 1999 ActiveState Tool Corp, http://www.ActiveState.com/
5 *
6 * You may distribute under the terms of either the GNU General Public
7 * License or the Artistic License, as specified in the README file.
8 */
9
10#ifndef ___VDir_H___
11#define ___VDir_H___
12
93153446 13/*
14 * Allow one slot for each possible drive letter
15 * and one additional slot for a UNC name
16 */
17const int driveCount = ('Z'-'A')+1+1;
7766f137 18
19class VDir
20{
21public:
f7aeb604 22 VDir(int bManageDir = 1);
7766f137 23 ~VDir() {};
24
25 void Init(VDir* pDir, VMem *pMem);
26 void SetDefaultA(char const *pDefault);
27 void SetDefaultW(WCHAR const *pDefault);
28 char* MapPathA(const char *pInName);
29 WCHAR* MapPathW(const WCHAR *pInName);
30 int SetCurrentDirectoryA(char *lpBuffer);
31 int SetCurrentDirectoryW(WCHAR *lpBuffer);
7766f137 32 inline int GetDefault(void) { return nDefault; };
33
34 inline char* GetCurrentDirectoryA(int dwBufSize, char *lpBuffer)
35 {
36 char* ptr = dirTableA[nDefault];
37 while (dwBufSize--)
38 {
39 if ((*lpBuffer++ = *ptr++) == '\0')
40 break;
41 }
42 return lpBuffer;
43 };
44 inline WCHAR* GetCurrentDirectoryW(int dwBufSize, WCHAR *lpBuffer)
45 {
46 WCHAR* ptr = dirTableW[nDefault];
47 while (dwBufSize--)
48 {
49 if ((*lpBuffer++ = *ptr++) == '\0')
50 break;
51 }
52 return lpBuffer;
53 };
54
55
56 DWORD CalculateEnvironmentSpace(void);
57 LPSTR BuildEnvironmentSpace(LPSTR lpStr);
58
59protected:
60 int SetDirA(char const *pPath, int index);
61 void FromEnvA(char *pEnv, int index);
62 inline const char *GetDefaultDirA(void)
63 {
64 return dirTableA[nDefault];
65 };
66
67 inline void SetDefaultDirA(char const *pPath, int index)
68 {
69 SetDirA(pPath, index);
70 nDefault = index;
71 };
72 int SetDirW(WCHAR const *pPath, int index);
73 inline const WCHAR *GetDefaultDirW(void)
74 {
75 return dirTableW[nDefault];
76 };
77
78 inline void SetDefaultDirW(WCHAR const *pPath, int index)
79 {
80 SetDirW(pPath, index);
81 nDefault = index;
82 };
e2694c95 83 inline const char *GetDirA(int index)
84 {
85 char *ptr = dirTableA[index];
86 if (!ptr) {
87 /* simulate the existance of this drive */
88 ptr = szLocalBufferA;
89 ptr[0] = 'A' + index;
90 ptr[1] = ':';
91 ptr[2] = '\\';
92 ptr[3] = 0;
93 }
94 return ptr;
95 };
96 inline const WCHAR *GetDirW(int index)
97 {
98 WCHAR *ptr = dirTableW[index];
99 if (!ptr) {
100 /* simulate the existance of this drive */
101 ptr = szLocalBufferW;
102 ptr[0] = 'A' + index;
103 ptr[1] = ':';
104 ptr[2] = '\\';
105 ptr[3] = 0;
106 }
107 return ptr;
108 };
7766f137 109
110 inline int DriveIndex(char chr)
111 {
93153446 112 if (chr == '\\' || chr == '/')
113 return ('Z'-'A')+1;
7766f137 114 return (chr | 0x20)-'a';
115 };
116
117 VMem *pMem;
f7aeb604 118 int nDefault, bManageDirectory;
7766f137 119 char *dirTableA[driveCount];
120 char szLocalBufferA[MAX_PATH+1];
121 WCHAR *dirTableW[driveCount];
122 WCHAR szLocalBufferW[MAX_PATH+1];
123};
124
125
f7aeb604 126VDir::VDir(int bManageDir /* = 1 */)
7766f137 127{
128 nDefault = 0;
f7aeb604 129 bManageDirectory = bManageDir;
7766f137 130 memset(dirTableA, 0, sizeof(dirTableA));
131 memset(dirTableW, 0, sizeof(dirTableW));
132}
133
134void VDir::Init(VDir* pDir, VMem *p)
135{
136 int index;
137 DWORD driveBits;
f7aeb604 138 int nSave;
7766f137 139 char szBuffer[MAX_PATH*driveCount];
140
141 pMem = p;
142 if (pDir) {
143 for (index = 0; index < driveCount; ++index) {
144 SetDirW(pDir->GetDirW(index), index);
145 }
146 nDefault = pDir->GetDefault();
147 }
148 else {
f7aeb604 149 nSave = bManageDirectory;
150 bManageDirectory = 0;
7766f137 151 driveBits = GetLogicalDrives();
152 if (GetLogicalDriveStrings(sizeof(szBuffer), szBuffer)) {
153 char* pEnv = GetEnvironmentStrings();
154 char* ptr = szBuffer;
155 for (index = 0; index < driveCount; ++index) {
156 if (driveBits & (1<<index)) {
157 ptr += SetDirA(ptr, index) + 1;
158 FromEnvA(pEnv, index);
159 }
160 }
161 FreeEnvironmentStrings(pEnv);
162 }
163 SetDefaultA(".");
f7aeb604 164 bManageDirectory = nSave;
7766f137 165 }
166}
167
168int VDir::SetDirA(char const *pPath, int index)
169{
170 char chr, *ptr;
171 int length = 0;
172 WCHAR wBuffer[MAX_PATH+1];
173 if (index < driveCount && pPath != NULL) {
174 length = strlen(pPath);
175 pMem->Free(dirTableA[index]);
176 ptr = dirTableA[index] = (char*)pMem->Malloc(length+2);
177 if (ptr != NULL) {
178 strcpy(ptr, pPath);
179 ptr += length-1;
180 chr = *ptr++;
181 if (chr != '\\' && chr != '/') {
182 *ptr++ = '\\';
183 *ptr = '\0';
184 }
185 MultiByteToWideChar(CP_ACP, 0, dirTableA[index], -1,
186 wBuffer, (sizeof(wBuffer)/sizeof(WCHAR)));
187 length = wcslen(wBuffer);
188 pMem->Free(dirTableW[index]);
189 dirTableW[index] = (WCHAR*)pMem->Malloc((length+1)*2);
190 if (dirTableW[index] != NULL) {
191 wcscpy(dirTableW[index], wBuffer);
192 }
193 }
194 }
f7aeb604 195
196 if(bManageDirectory)
197 ::SetCurrentDirectoryA(pPath);
198
7766f137 199 return length;
200}
201
202void VDir::FromEnvA(char *pEnv, int index)
203{ /* gets the directory for index from the environment variable. */
204 while (*pEnv != '\0') {
205 if ((pEnv[0] == '=') && (DriveIndex(pEnv[1]) == index)) {
206 SetDirA(&pEnv[4], index);
207 break;
208 }
209 else
210 pEnv += strlen(pEnv)+1;
211 }
212}
213
214void VDir::SetDefaultA(char const *pDefault)
215{
216 char szBuffer[MAX_PATH+1];
217 char *pPtr;
218
219 if (GetFullPathNameA(pDefault, sizeof(szBuffer), szBuffer, &pPtr)) {
220 if (*pDefault != '.' && pPtr != NULL)
221 *pPtr = '\0';
222
223 SetDefaultDirA(szBuffer, DriveIndex(szBuffer[0]));
224 }
225}
226
227int VDir::SetDirW(WCHAR const *pPath, int index)
228{
229 WCHAR chr, *ptr;
230 char szBuffer[MAX_PATH+1];
231 int length = 0;
232 if (index < driveCount && pPath != NULL) {
233 length = wcslen(pPath);
234 pMem->Free(dirTableW[index]);
235 ptr = dirTableW[index] = (WCHAR*)pMem->Malloc((length+2)*2);
236 if (ptr != NULL) {
237 wcscpy(ptr, pPath);
238 ptr += length-1;
239 chr = *ptr++;
240 if (chr != '\\' && chr != '/') {
241 *ptr++ = '\\';
242 *ptr = '\0';
243 }
244 WideCharToMultiByte(CP_ACP, 0, dirTableW[index], -1, szBuffer, sizeof(szBuffer), NULL, NULL);
245 length = strlen(szBuffer);
246 pMem->Free(dirTableA[index]);
247 dirTableA[index] = (char*)pMem->Malloc(length+1);
248 if (dirTableA[index] != NULL) {
249 strcpy(dirTableA[index], szBuffer);
250 }
251 }
252 }
f7aeb604 253
254 if(bManageDirectory)
255 ::SetCurrentDirectoryW(pPath);
256
7766f137 257 return length;
258}
259
260void VDir::SetDefaultW(WCHAR const *pDefault)
261{
262 WCHAR szBuffer[MAX_PATH+1];
263 WCHAR *pPtr;
264
265 if (GetFullPathNameW(pDefault, (sizeof(szBuffer)/sizeof(WCHAR)), szBuffer, &pPtr)) {
266 if (*pDefault != '.' && pPtr != NULL)
267 *pPtr = '\0';
268
269 SetDefaultDirW(szBuffer, DriveIndex((char)szBuffer[0]));
270 }
271}
272
273inline BOOL IsPathSep(char ch)
274{
275 return (ch == '\\' || ch == '/');
276}
277
278inline void DoGetFullPathNameA(char* lpBuffer, DWORD dwSize, char* Dest)
279{
280 char *pPtr;
281
282 /*
283 * On WinNT GetFullPathName does not fail, (or at least always
284 * succeeds when the drive is valid) WinNT does set *Dest to Nullch
285 * On Win98 GetFullPathName will set last error if it fails, but
286 * does not touch *Dest
287 */
288 *Dest = '\0';
289 GetFullPathNameA(lpBuffer, dwSize, Dest, &pPtr);
290}
291
e2694c95 292inline bool IsSpecialFileName(const char* pName)
293{
294 /* specical file names are devices that the system can open
295 * these include AUX, CON, NUL, PRN, COMx, LPTx, CLOCK$, CONIN$, CONOUT$
296 * (x is a single digit, and names are case-insensitive)
297 */
298 char ch = (pName[0] & ~0x20);
299 switch (ch)
300 {
301 case 'A': /* AUX */
302 if (((pName[1] & ~0x20) == 'U')
303 && ((pName[2] & ~0x20) == 'X')
304 && !pName[3])
305 return true;
306 break;
307 case 'C': /* CLOCK$, COMx, CON, CONIN$ CONOUT$ */
308 ch = (pName[1] & ~0x20);
309 switch (ch)
310 {
311 case 'L': /* CLOCK$ */
312 if (((pName[2] & ~0x20) == 'O')
313 && ((pName[3] & ~0x20) == 'C')
314 && ((pName[4] & ~0x20) == 'K')
315 && (pName[5] == '$')
316 && !pName[6])
317 return true;
318 break;
319 case 'O': /* COMx, CON, CONIN$ CONOUT$ */
320 if ((pName[2] & ~0x20) == 'M') {
321 if ((pName[3] >= '1') && (pName[3] <= '9')
322 && !pName[4])
323 return true;
324 }
325 else if ((pName[2] & ~0x20) == 'N') {
326 if (!pName[3])
327 return true;
328 else if ((pName[3] & ~0x20) == 'I') {
329 if (((pName[4] & ~0x20) == 'N')
330 && (pName[5] == '$')
331 && !pName[6])
332 return true;
333 }
334 else if ((pName[3] & ~0x20) == 'O') {
335 if (((pName[4] & ~0x20) == 'U')
336 && ((pName[5] & ~0x20) == 'T')
337 && (pName[6] == '$')
338 && !pName[7])
339 return true;
340 }
341 }
342 break;
343 }
344 break;
345 case 'L': /* LPTx */
346 if (((pName[1] & ~0x20) == 'U')
347 && ((pName[2] & ~0x20) == 'X')
348 && (pName[3] >= '1') && (pName[3] <= '9')
349 && !pName[4])
350 return true;
351 break;
352 case 'N': /* NUL */
353 if (((pName[1] & ~0x20) == 'U')
354 && ((pName[2] & ~0x20) == 'L')
355 && !pName[3])
356 return true;
357 break;
358 case 'P': /* PRN */
359 if (((pName[1] & ~0x20) == 'R')
360 && ((pName[2] & ~0x20) == 'N')
361 && !pName[3])
362 return true;
363 break;
364 }
365 return false;
366}
367
7766f137 368char *VDir::MapPathA(const char *pInName)
369{ /*
370 * possiblities -- relative path or absolute path with or without drive letter
371 * OR UNC name
372 */
373 char szBuffer[(MAX_PATH+1)*2];
374 char szlBuf[MAX_PATH+1];
93153446 375 int length = strlen(pInName);
376
377 if (!length)
378 return (char*)pInName;
7766f137 379
93153446 380 if (length > MAX_PATH) {
7766f137 381 strncpy(szlBuf, pInName, MAX_PATH);
382 if (IsPathSep(pInName[0]) && !IsPathSep(pInName[1])) {
383 /* absolute path - reduce length by 2 for drive specifier */
384 szlBuf[MAX_PATH-2] = '\0';
385 }
386 else
387 szlBuf[MAX_PATH] = '\0';
388 pInName = szlBuf;
389 }
390 /* strlen(pInName) is now <= MAX_PATH */
391
392 if (pInName[1] == ':') {
393 /* has drive letter */
394 if (IsPathSep(pInName[2])) {
395 /* absolute with drive letter */
396 strcpy(szLocalBufferA, pInName);
397 }
398 else {
399 /* relative path with drive letter */
400 strcpy(szBuffer, GetDirA(DriveIndex(*pInName)));
401 strcat(szBuffer, &pInName[2]);
402 if(strlen(szBuffer) > MAX_PATH)
403 szBuffer[MAX_PATH] = '\0';
404
405 DoGetFullPathNameA(szBuffer, sizeof(szLocalBufferA), szLocalBufferA);
406 }
407 }
408 else {
409 /* no drive letter */
410 if (IsPathSep(pInName[1]) && IsPathSep(pInName[0])) {
411 /* UNC name */
412 strcpy(szLocalBufferA, pInName);
413 }
414 else {
415 strcpy(szBuffer, GetDefaultDirA());
416 if (IsPathSep(pInName[0])) {
417 /* absolute path */
418 szLocalBufferA[0] = szBuffer[0];
419 szLocalBufferA[1] = szBuffer[1];
420 strcpy(&szLocalBufferA[2], pInName);
421 }
422 else {
423 /* relative path */
e2694c95 424 if (IsSpecialFileName(pInName)) {
425 return (char*)pInName;
426 }
427 else {
428 strcat(szBuffer, pInName);
429 if (strlen(szBuffer) > MAX_PATH)
430 szBuffer[MAX_PATH] = '\0';
7766f137 431
e2694c95 432 DoGetFullPathNameA(szBuffer, sizeof(szLocalBufferA), szLocalBufferA);
433 }
7766f137 434 }
435 }
436 }
437
438 return szLocalBufferA;
439}
440
441int VDir::SetCurrentDirectoryA(char *lpBuffer)
442{
93153446 443 char *pPtr;
fd9459bc 444 int length, nRet = -1;
7766f137 445
93153446 446 pPtr = MapPathA(lpBuffer);
447 length = strlen(pPtr);
448 if(length > 3 && IsPathSep(pPtr[length-1])) {
449 /* don't remove the trailing slash from 'x:\' */
450 pPtr[length-1] = '\0';
fd9459bc 451 }
7766f137 452
93153446 453 DWORD r = GetFileAttributesA(pPtr);
454 if ((r != 0xffffffff) && (r & FILE_ATTRIBUTE_DIRECTORY))
455 {
456 SetDefaultDirA(pPtr, DriveIndex(pPtr[0]));
7766f137 457 nRet = 0;
458 }
93153446 459
7766f137 460 return nRet;
461}
462
463DWORD VDir::CalculateEnvironmentSpace(void)
464{ /* the current directory environment strings are stored as '=d=d:\path' */
465 int index;
466 DWORD dwSize = 0;
467 for (index = 0; index < driveCount; ++index) {
468 if (dirTableA[index] != NULL) {
469 dwSize += strlen(dirTableA[index]) + 4; /* add 1 for trailing NULL and 3 for '=d=' */
470 }
471 }
472 return dwSize;
473}
474
475LPSTR VDir::BuildEnvironmentSpace(LPSTR lpStr)
476{ /* store the current directory environment strings as '=d=d:\path' */
477 int index;
478 LPSTR lpDirStr;
479 for (index = 0; index < driveCount; ++index) {
480 lpDirStr = dirTableA[index];
481 if (lpDirStr != NULL) {
482 lpStr[0] = '=';
483 lpStr[1] = lpDirStr[0];
484 lpStr[2] = '=';
485 strcpy(&lpStr[3], lpDirStr);
486 lpStr += strlen(lpDirStr) + 4; /* add 1 for trailing NULL and 3 for '=d=' */
487 }
488 }
489 return lpStr;
490}
491
492inline BOOL IsPathSep(WCHAR ch)
493{
494 return (ch == '\\' || ch == '/');
495}
496
497inline void DoGetFullPathNameW(WCHAR* lpBuffer, DWORD dwSize, WCHAR* Dest)
498{
499 WCHAR *pPtr;
500
501 /*
502 * On WinNT GetFullPathName does not fail, (or at least always
503 * succeeds when the drive is valid) WinNT does set *Dest to Nullch
504 * On Win98 GetFullPathName will set last error if it fails, but
505 * does not touch *Dest
506 */
507 *Dest = '\0';
508 GetFullPathNameW(lpBuffer, dwSize, Dest, &pPtr);
509}
510
e2694c95 511inline bool IsSpecialFileName(const WCHAR* pName)
512{
513 /* specical file names are devices that the system can open
514 * these include AUX, CON, NUL, PRN, COMx, LPTx, CLOCK$, CONIN$, CONOUT$
515 * (x is a single digit, and names are case-insensitive)
516 */
517 WCHAR ch = (pName[0] & ~0x20);
518 switch (ch)
519 {
520 case 'A': /* AUX */
521 if (((pName[1] & ~0x20) == 'U')
522 && ((pName[2] & ~0x20) == 'X')
523 && !pName[3])
524 return true;
525 break;
526 case 'C': /* CLOCK$, COMx, CON, CONIN$ CONOUT$ */
527 ch = (pName[1] & ~0x20);
528 switch (ch)
529 {
530 case 'L': /* CLOCK$ */
531 if (((pName[2] & ~0x20) == 'O')
532 && ((pName[3] & ~0x20) == 'C')
533 && ((pName[4] & ~0x20) == 'K')
534 && (pName[5] == '$')
535 && !pName[6])
536 return true;
537 break;
538 case 'O': /* COMx, CON, CONIN$ CONOUT$ */
539 if ((pName[2] & ~0x20) == 'M') {
540 if ((pName[3] >= '1') && (pName[3] <= '9')
541 && !pName[4])
542 return true;
543 }
544 else if ((pName[2] & ~0x20) == 'N') {
545 if (!pName[3])
546 return true;
547 else if ((pName[3] & ~0x20) == 'I') {
548 if (((pName[4] & ~0x20) == 'N')
549 && (pName[5] == '$')
550 && !pName[6])
551 return true;
552 }
553 else if ((pName[3] & ~0x20) == 'O') {
554 if (((pName[4] & ~0x20) == 'U')
555 && ((pName[5] & ~0x20) == 'T')
556 && (pName[6] == '$')
557 && !pName[7])
558 return true;
559 }
560 }
561 break;
562 }
563 break;
564 case 'L': /* LPTx */
565 if (((pName[1] & ~0x20) == 'U')
566 && ((pName[2] & ~0x20) == 'X')
567 && (pName[3] >= '1') && (pName[3] <= '9')
568 && !pName[4])
569 return true;
570 break;
571 case 'N': /* NUL */
572 if (((pName[1] & ~0x20) == 'U')
573 && ((pName[2] & ~0x20) == 'L')
574 && !pName[3])
575 return true;
576 break;
577 case 'P': /* PRN */
578 if (((pName[1] & ~0x20) == 'R')
579 && ((pName[2] & ~0x20) == 'N')
580 && !pName[3])
581 return true;
582 break;
583 }
584 return false;
585}
586
7766f137 587WCHAR* VDir::MapPathW(const WCHAR *pInName)
588{ /*
589 * possiblities -- relative path or absolute path with or without drive letter
590 * OR UNC name
591 */
592 WCHAR szBuffer[(MAX_PATH+1)*2];
593 WCHAR szlBuf[MAX_PATH+1];
93153446 594 int length = wcslen(pInName);
7766f137 595
93153446 596 if (!length)
597 return (WCHAR*)pInName;
598
599 if (length > MAX_PATH) {
7766f137 600 wcsncpy(szlBuf, pInName, MAX_PATH);
601 if (IsPathSep(pInName[0]) && !IsPathSep(pInName[1])) {
602 /* absolute path - reduce length by 2 for drive specifier */
603 szlBuf[MAX_PATH-2] = '\0';
604 }
605 else
606 szlBuf[MAX_PATH] = '\0';
607 pInName = szlBuf;
608 }
609 /* strlen(pInName) is now <= MAX_PATH */
610
611 if (pInName[1] == ':') {
612 /* has drive letter */
613 if (IsPathSep(pInName[2])) {
614 /* absolute with drive letter */
615 wcscpy(szLocalBufferW, pInName);
616 }
617 else {
618 /* relative path with drive letter */
619 wcscpy(szBuffer, GetDirW(DriveIndex((char)*pInName)));
620 wcscat(szBuffer, &pInName[2]);
621 if(wcslen(szBuffer) > MAX_PATH)
622 szBuffer[MAX_PATH] = '\0';
623
624 DoGetFullPathNameW(szBuffer, (sizeof(szLocalBufferW)/sizeof(WCHAR)), szLocalBufferW);
625 }
626 }
627 else {
628 /* no drive letter */
629 if (IsPathSep(pInName[1]) && IsPathSep(pInName[0])) {
630 /* UNC name */
631 wcscpy(szLocalBufferW, pInName);
632 }
633 else {
634 wcscpy(szBuffer, GetDefaultDirW());
635 if (IsPathSep(pInName[0])) {
636 /* absolute path */
637 szLocalBufferW[0] = szBuffer[0];
638 szLocalBufferW[1] = szBuffer[1];
639 wcscpy(&szLocalBufferW[2], pInName);
640 }
641 else {
642 /* relative path */
e2694c95 643 if (IsSpecialFileName(pInName)) {
644 return (WCHAR*)pInName;
645 }
646 else {
647 wcscat(szBuffer, pInName);
648 if (wcslen(szBuffer) > MAX_PATH)
649 szBuffer[MAX_PATH] = '\0';
7766f137 650
e2694c95 651 DoGetFullPathNameW(szBuffer, (sizeof(szLocalBufferW)/sizeof(WCHAR)), szLocalBufferW);
652 }
7766f137 653 }
654 }
655 }
656 return szLocalBufferW;
657}
658
fd9459bc 659int VDir::SetCurrentDirectoryW(WCHAR *lpBuffer)
660{
93153446 661 WCHAR *pPtr;
fd9459bc 662 int length, nRet = -1;
663
93153446 664 pPtr = MapPathW(lpBuffer);
665 length = wcslen(pPtr);
666 if(length > 3 && IsPathSep(pPtr[length-1])) {
667 /* don't remove the trailing slash from 'x:\' */
668 pPtr[length-1] = '\0';
fd9459bc 669 }
670
93153446 671 DWORD r = GetFileAttributesW(pPtr);
672 if ((r != 0xffffffff) && (r & FILE_ATTRIBUTE_DIRECTORY))
673 {
674 SetDefaultDirW(pPtr, DriveIndex((char)pPtr[0]));
fd9459bc 675 nRet = 0;
676 }
93153446 677
fd9459bc 678 return nRet;
679}
7766f137 680
681#endif /* ___VDir_H___ */