Zero entries were skipped, fix from Adrian Goalby
[p5sagit/p5-mst-13.2.git] / win32 / vdir.h
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
13 /*
14  * Allow one slot for each possible drive letter
15  * and one additional slot for a UNC name
16  */
17 const int driveCount = ('Z'-'A')+1+1;
18
19 class VDir
20 {
21 public:
22     VDir(int bManageDir = 1);
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);
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
59 protected:
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     };
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     };
109
110     inline int DriveIndex(char chr)
111     {
112         if (chr == '\\' || chr == '/')
113             return ('Z'-'A')+1;
114         return (chr | 0x20)-'a';
115     };
116
117     VMem *pMem;
118     int nDefault, bManageDirectory;
119     char *dirTableA[driveCount];
120     char szLocalBufferA[MAX_PATH+1];
121     WCHAR *dirTableW[driveCount];
122     WCHAR szLocalBufferW[MAX_PATH+1];
123 };
124
125
126 VDir::VDir(int bManageDir /* = 1 */)
127 {
128     nDefault = 0;
129     bManageDirectory = bManageDir;
130     memset(dirTableA, 0, sizeof(dirTableA));
131     memset(dirTableW, 0, sizeof(dirTableW));
132 }
133
134 void VDir::Init(VDir* pDir, VMem *p)
135 {
136     int index;
137     DWORD driveBits;
138     int nSave;
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 {
149         nSave = bManageDirectory;
150         bManageDirectory = 0;
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(".");
164         bManageDirectory = nSave;
165     }
166 }
167
168 int 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     }
195
196     if(bManageDirectory)
197         ::SetCurrentDirectoryA(pPath);
198
199     return length;
200 }
201
202 void 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
214 void 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
227 int 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     }
253
254     if(bManageDirectory)
255         ::SetCurrentDirectoryW(pPath);
256
257     return length;
258 }
259
260 void 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
273 inline BOOL IsPathSep(char ch)
274 {
275     return (ch == '\\' || ch == '/');
276 }
277
278 inline 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
292 inline 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
368 char *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];
375     int length = strlen(pInName);
376
377     if (!length)
378         return (char*)pInName;
379
380     if (length > MAX_PATH) {
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 */
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';
431
432                     DoGetFullPathNameA(szBuffer, sizeof(szLocalBufferA), szLocalBufferA);
433                 }
434             }
435         }
436     }
437
438     return szLocalBufferA;
439 }
440
441 int VDir::SetCurrentDirectoryA(char *lpBuffer)
442 {
443     char *pPtr;
444     int length, nRet = -1;
445
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';
451     }
452
453     DWORD r = GetFileAttributesA(pPtr);
454     if ((r != 0xffffffff) && (r & FILE_ATTRIBUTE_DIRECTORY))
455     {
456         SetDefaultDirA(pPtr, DriveIndex(pPtr[0]));
457         nRet = 0;
458     }
459
460     return nRet;
461 }
462
463 DWORD 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
475 LPSTR 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
492 inline BOOL IsPathSep(WCHAR ch)
493 {
494     return (ch == '\\' || ch == '/');
495 }
496
497 inline 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
511 inline 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
587 WCHAR* 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];
594     int length = wcslen(pInName);
595
596     if (!length)
597         return (WCHAR*)pInName;
598
599     if (length > MAX_PATH) {
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 */
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';
650
651                     DoGetFullPathNameW(szBuffer, (sizeof(szLocalBufferW)/sizeof(WCHAR)), szLocalBufferW);
652                 }
653             }
654         }
655     }
656     return szLocalBufferW;
657 }
658
659 int VDir::SetCurrentDirectoryW(WCHAR *lpBuffer)
660 {
661     WCHAR *pPtr;
662     int length, nRet = -1;
663
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';
669     }
670
671     DWORD r = GetFileAttributesW(pPtr);
672     if ((r != 0xffffffff) && (r & FILE_ATTRIBUTE_DIRECTORY))
673     {
674         SetDefaultDirW(pPtr, DriveIndex((char)pPtr[0]));
675         nRet = 0;
676     }
677
678     return nRet;
679 }
680
681 #endif  /* ___VDir_H___ */