diff options
author | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-11 08:21:29 +0000 |
---|---|---|
committer | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-11 08:21:29 +0000 |
commit | 29cd838eab01ed7110f3ccb2e8c6a35c8a31dbcc (patch) | |
tree | 63ef546b10a81d461e5cf5ed9e98a68cd7dee1aa /src/lib/nt_fullpath.c | |
parent | Initial commit. (diff) | |
download | kbuild-29cd838eab01ed7110f3ccb2e8c6a35c8a31dbcc.tar.xz kbuild-29cd838eab01ed7110f3ccb2e8c6a35c8a31dbcc.zip |
Adding upstream version 1:0.1.9998svn3589+dfsg.upstream/1%0.1.9998svn3589+dfsg
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to '')
-rw-r--r-- | src/lib/nt_fullpath.c | 580 |
1 files changed, 580 insertions, 0 deletions
diff --git a/src/lib/nt_fullpath.c b/src/lib/nt_fullpath.c new file mode 100644 index 0000000..fa9a7cc --- /dev/null +++ b/src/lib/nt_fullpath.c @@ -0,0 +1,580 @@ +/* $Id: nt_fullpath.c 3174 2018-03-21 21:37:52Z bird $ */ +/** @file + * fixcase - fixes the case of paths, windows specific. + */ + +/* + * Copyright (c) 2004-2010 knut st. osmundsen <bird-kBuild-spamx@anduin.net> + * + * This file is part of kBuild. + * + * kBuild is free software; you can redistribute it and/or modify + * it under the terms of the GNU General Public License as published by + * the Free Software Foundation; either version 3 of the License, or + * (at your option) any later version. + * + * kBuild is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with kBuild. If not, see <http://www.gnu.org/licenses/> + * + */ + +/******************************************************************************* +* Header Files * +*******************************************************************************/ +#include <Windows.h> +#include <stdio.h> +#include <stdlib.h> +#include <string.h> +#include <ctype.h> +#include <direct.h> + +#include "nt_fullpath.h" + + +/* + * Corrects the case of a path. + * Expects a fullpath! + * Added by bird for the $(abspath ) function and w32ify + */ +static void w32_fixcase(char *pszPath) +{ +#if 0 /* no mp safe */ + static char s_szLast[260]; + size_t cchLast; +#endif + +#ifndef NDEBUG +# define my_assert(expr) \ + do { \ + if (!(expr)) { \ + printf("my_assert: %s, file %s, line %d\npszPath=%s\npsz=%s\n", \ + #expr, __FILE__, __LINE__, pszPath, psz); \ + __debugbreak(); \ + exit(1); \ + } \ + } while (0) +#else +# define my_assert(expr) do {} while (0) +#endif + + char *psz = pszPath; + if (*psz == '/' || *psz == '\\') + { + if (psz[1] == '/' || psz[1] == '\\') + { + /* UNC */ + my_assert(psz[1] == '/' || psz[1] == '\\'); + my_assert(psz[2] != '/' && psz[2] != '\\'); + + /* skip server name */ + psz += 2; + while (*psz != '\\' && *psz != '/') + { + if (!*psz) + return; + *psz++ = toupper(*psz); + } + + /* skip the share name */ + psz++; + my_assert(*psz != '/' && *psz != '\\'); + while (*psz != '\\' && *psz != '/') + { + if (!*psz) + return; + *psz++ = toupper(*psz); + } + my_assert(*psz == '/' || *psz == '\\'); + psz++; + } + else + { + /* Unix spec */ + psz++; + } + } + else + { + /* Drive letter */ + my_assert(psz[1] == ':'); + *psz = toupper(*psz); + my_assert(psz[0] >= 'A' && psz[0] <= 'Z'); + my_assert(psz[2] == '/' || psz[2] == '\\'); + psz += 3; + } + +#if 0 /* not mp safe */ + /* + * Try make use of the result from the previous call. + * This is ignorant to slashes and similar, but may help even so. + */ + if ( s_szLast[0] == pszPath[0] + && (psz - pszPath == 1 || s_szLast[1] == pszPath[1]) + && (psz - pszPath <= 2 || s_szLast[2] == pszPath[2]) + ) + { + char *pszLast = &s_szLast[psz - pszPath]; + char *pszCur = psz; + char *pszSrc0 = pszLast; + char *pszDst0 = pszCur; + for (;;) + { + const char ch1 = *pszCur; + const char ch2 = *pszLast; + if ( ch1 != ch2 + && (ch1 != '\\' || ch2 != '/') + && (ch1 != '/' || ch2 != '\\') + && tolower(ch1) != tolower(ch2) + && toupper(ch1) != toupper(ch2)) + break; + if (ch1 == '/' || ch1 == '\\') + { + psz = pszCur + 1; + *pszLast = ch1; /* preserve the slashes */ + } + else if (ch1 == '\0') + { + psz = pszCur; + break; + } + pszCur++; + pszLast++; + } + if (psz != pszDst0) + memcpy(pszDst0, pszSrc0, psz - pszDst0); + } +#endif + + /* + * Pointing to the first char after the unc or drive specifier, + * or in case of a cache hit, the first non-matching char (following a slash of course). + */ + while (*psz) + { + WIN32_FIND_DATA FindFileData; + HANDLE hDir; + char chSaved0; + char chSaved1; + char *pszEnd; + int iLongNameDiff; + size_t cch; + + + /* find the end of the component. */ + pszEnd = psz; + while (*pszEnd && *pszEnd != '/' && *pszEnd != '\\') + pszEnd++; + cch = pszEnd - psz; + + /* replace the end with "?\0" */ + chSaved0 = pszEnd[0]; + chSaved1 = pszEnd[1]; + pszEnd[0] = '?'; + pszEnd[1] = '\0'; + + /* find the right filename. */ + hDir = FindFirstFile(pszPath, &FindFileData); + pszEnd[1] = chSaved1; + if (!hDir) + { +#if 0 /* not MP safe */ + cchLast = psz - pszPath; + memcpy(s_szLast, pszPath, cchLast + 1); + s_szLast[cchLast + 1] = '\0'; +#endif + pszEnd[0] = chSaved0; + return; + } + pszEnd[0] = '\0'; + while ( (iLongNameDiff = stricmp(FindFileData.cFileName, psz)) + && stricmp(FindFileData.cAlternateFileName, psz)) + { + if (!FindNextFile(hDir, &FindFileData)) + { +#if 0 /* not MP safe */ + cchLast = psz - pszPath; + memcpy(s_szLast, pszPath, cchLast + 1); + s_szLast[cchLast + 1] = '\0'; +#endif + pszEnd[0] = chSaved0; + return; + } + } + pszEnd[0] = chSaved0; + if ( iLongNameDiff /* matched the short name */ + || !FindFileData.cAlternateFileName[0] /* no short name */ + || !memchr(psz, ' ', cch)) /* no spaces in the matching name */ + memcpy(psz, !iLongNameDiff ? FindFileData.cFileName : FindFileData.cAlternateFileName, cch); + else + { + /* replace spacy name with the short name. */ + const size_t cchAlt = strlen(FindFileData.cAlternateFileName); + const size_t cchDelta = cch - cchAlt; + my_assert(cchAlt > 0); + if (!cchDelta) + memcpy(psz, FindFileData.cAlternateFileName, cch); + else + { + size_t cbLeft = strlen(pszEnd) + 1; + if ((psz - pszPath) + cbLeft + cchAlt <= _MAX_PATH) + { + memmove(psz + cchAlt, pszEnd, cbLeft); + pszEnd -= cchDelta; + memcpy(psz, FindFileData.cAlternateFileName, cchAlt); + } + else + fprintf(stderr, "kBuild: case & space fixed filename is growing too long (%d bytes)! '%s'\n", + (psz - pszPath) + cbLeft + cchAlt, pszPath); + } + } + my_assert(pszEnd[0] == chSaved0); + FindClose(hDir); + + /* advance to the next component */ + if (!chSaved0) + { + psz = pszEnd; + break; + } + psz = pszEnd + 1; + my_assert(*psz != '/' && *psz != '\\'); + } + +#if 0 /* not MP safe */ + /* *psz == '\0', the end. */ + cchLast = psz - pszPath; + memcpy(s_szLast, pszPath, cchLast + 1); +#endif +#undef my_assert +} + +#define MY_FileNameInformation 9 +typedef struct _MY_FILE_NAME_INFORMATION +{ + ULONG FileNameLength; + WCHAR FileName[1]; +} MY_FILE_NAME_INFORMATION, *PMY_FILE_NAME_INFORMATION; + +#define MY_FileInternalInformation 6 +typedef struct _MY_FILE_INTERNAL_INFORMATION { + LARGE_INTEGER IndexNumber; +} MY_FILE_INTERNAL_INFORMATION, *PMY_FILE_INTERNAL_INFORMATION; + +#define MY_FileFsVolumeInformation 1 +typedef struct _MY_FILE_FS_VOLUME_INFORMATION +{ + LARGE_INTEGER VolumeCreationTime; + ULONG VolumeSerialNumber; + ULONG VolumeLabelLength; + BOOLEAN SupportsObjects; + WCHAR VolumeLabel[/*1*/128]; +} MY_FILE_FS_VOLUME_INFORMATION, *PMY_FILE_FS_VOLUME_INFORMATION; + +#define MY_FileFsAttributeInformation 5 +typedef struct _MY_FILE_FS_ATTRIBUTE_INFORMATION +{ + ULONG FileSystemAttributes; + LONG MaximumComponentNameLength; + ULONG FileSystemNameLength; + WCHAR FileSystemName[/*1*/64]; +} MY_FILE_FS_ATTRIBUTE_INFORMATION, *PMY_FILE_FS_ATTRIBUTE_INFORMATION; + +#define MY_FileFsDeviceInformation 4 +typedef struct MY_FILE_FS_DEVICE_INFORMATION +{ + ULONG DeviceType; + ULONG Characteristics; +} MY_FILE_FS_DEVICE_INFORMATION, *PMY_FILE_FS_DEVICE_INFORMATION; +#define MY_FILE_DEVICE_DISK 7 +#define MY_FILE_DEVICE_DISK_FILE_SYSTEM 8 +#define MY_FILE_DEVICE_FILE_SYSTEM 9 +#define MY_FILE_DEVICE_VIRTUAL_DISK 36 + + +typedef struct +{ + union + { + LONG Status; + PVOID Pointer; + }; + ULONG_PTR Information; +} MY_IO_STATUS_BLOCK, *PMY_IO_STATUS_BLOCK; + +static BOOL g_fInitialized = FALSE; +static int g_afNtfsDrives['Z' - 'A' + 1]; +static MY_FILE_FS_VOLUME_INFORMATION g_aVolumeInfo['Z' - 'A' + 1]; + +static LONG (NTAPI *g_pfnNtQueryInformationFile)(HANDLE FileHandle, + PMY_IO_STATUS_BLOCK IoStatusBlock, PVOID FileInformation, + ULONG Length, ULONG FileInformationClass); +static LONG (NTAPI *g_pfnNtQueryVolumeInformationFile)(HANDLE FileHandle, + PMY_IO_STATUS_BLOCK IoStatusBlock, PVOID FsInformation, + ULONG Length, ULONG FsInformationClass); + + +int +nt_get_filename_info(const char *pszPath, char *pszFull, size_t cchFull) +{ + char abBuf[8192]; + PMY_FILE_NAME_INFORMATION pFileNameInfo = (PMY_FILE_NAME_INFORMATION)abBuf; + PMY_FILE_FS_VOLUME_INFORMATION pFsVolInfo = (PMY_FILE_FS_VOLUME_INFORMATION)abBuf; + MY_IO_STATUS_BLOCK Ios; + LONG rcNt; + HANDLE hFile; + int cchOut; + char *psz; + int iDrv; + int rc; + + /* + * Check for NtQueryInformationFile the first time around. + */ + if (!g_fInitialized) + { + g_fInitialized = TRUE; + if (!getenv("KMK_DONT_USE_NT_QUERY_INFORMATION_FILE")) + { + *(FARPROC *)&g_pfnNtQueryInformationFile = + GetProcAddress(LoadLibrary("ntdll.dll"), "NtQueryInformationFile"); + *(FARPROC *)&g_pfnNtQueryVolumeInformationFile = + GetProcAddress(LoadLibrary("ntdll.dll"), "NtQueryVolumeInformationFile"); + } + if ( g_pfnNtQueryInformationFile + && g_pfnNtQueryVolumeInformationFile) + { + unsigned i; + for (i = 0; i < sizeof(g_afNtfsDrives) / sizeof(g_afNtfsDrives[0]); i++ ) + g_afNtfsDrives[i] = -1; + } + else + { + g_pfnNtQueryVolumeInformationFile = NULL; + g_pfnNtQueryInformationFile = NULL; + } + } + if (!g_pfnNtQueryInformationFile) + return -1; + + /* + * The FileNameInformation we get is relative to where the volume is mounted, + * so we have to extract the driveletter prefix ourselves. + * + * FIXME: This will probably not work for volumes mounted in NTFS sub-directories. + */ + psz = pszFull; + if (pszPath[0] == '\\' || pszPath[0] == '/') + { + /* unc or root of volume */ + if ( (pszPath[1] == '\\' || pszPath[1] == '/') + && (pszPath[2] != '\\' || pszPath[2] == '/')) + { +#if 0 /* don't bother with unc yet. */ + /* unc - we get the server + name back */ + *psz++ = '\\'; +#endif + return -1; + } + /* root slash */ + *psz++ = _getdrive() + 'A' - 1; + *psz++ = ':'; + } + else if (pszPath[1] == ':' && isalpha(pszPath[0])) + { + /* drive letter */ + *psz++ = toupper(pszPath[0]); + *psz++ = ':'; + } + else + { + /* relative */ + *psz++ = _getdrive() + 'A' - 1; + *psz++ = ':'; + } + iDrv = *pszFull - 'A'; + + /* + * Fat32 doesn't return filenames with the correct case, so restrict it + * to NTFS volumes for now. + */ + if (g_afNtfsDrives[iDrv] == -1) + { + /* FSCTL_GET_REPARSE_POINT? Enumerate mount points? */ + g_afNtfsDrives[iDrv] = 0; + psz[0] = '\\'; + psz[1] = '\0'; +#if 1 + hFile = CreateFile(pszFull, + GENERIC_READ, + FILE_SHARE_READ | FILE_SHARE_WRITE | FILE_SHARE_DELETE, + NULL, + OPEN_EXISTING, + FILE_FLAG_BACKUP_SEMANTICS, + NULL); + if (hFile != INVALID_HANDLE_VALUE) + { + PMY_FILE_FS_ATTRIBUTE_INFORMATION pFsAttrInfo = (PMY_FILE_FS_ATTRIBUTE_INFORMATION)abBuf; + + memset(&Ios, 0, sizeof(Ios)); + rcNt = g_pfnNtQueryVolumeInformationFile(hFile, &Ios, abBuf, sizeof(abBuf), + MY_FileFsAttributeInformation); + if ( rcNt >= 0 + //&& pFsAttrInfo->FileSystemNameLength == 4 + && pFsAttrInfo->FileSystemName[0] == 'N' + && pFsAttrInfo->FileSystemName[1] == 'T' + && pFsAttrInfo->FileSystemName[2] == 'F' + && pFsAttrInfo->FileSystemName[3] == 'S' + && pFsAttrInfo->FileSystemName[4] == '\0') + { + memset(&Ios, 0, sizeof(Ios)); + rcNt = g_pfnNtQueryVolumeInformationFile(hFile, &Ios, &g_aVolumeInfo[iDrv], + sizeof(MY_FILE_FS_VOLUME_INFORMATION), + MY_FileFsVolumeInformation); + if (rcNt >= 0) + { + DWORD dwDriveType = GetDriveType(pszFull); + if ( dwDriveType == DRIVE_FIXED + || dwDriveType == DRIVE_RAMDISK) + g_afNtfsDrives[iDrv] = 1; + } + } + CloseHandle(hFile); + } +#else + { + char szFSName[32]; + if ( GetVolumeInformation(pszFull, + NULL, 0, /* volume name */ + NULL, /* serial number */ + NULL, /* max component */ + NULL, /* volume attribs */ + szFSName, + sizeof(szFSName)) + && !strcmp(szFSName, "NTFS")) + { + g_afNtfsDrives[iDrv] = 1; + } + } +#endif + } + if (!g_afNtfsDrives[iDrv]) + return -1; + + /* + * Try open the path and query its file name information. + */ + hFile = CreateFile(pszPath, + GENERIC_READ, + FILE_SHARE_READ | FILE_SHARE_WRITE | FILE_SHARE_DELETE, + NULL, + OPEN_EXISTING, + FILE_FLAG_BACKUP_SEMANTICS, + NULL); + if (hFile != INVALID_HANDLE_VALUE) + { + /* check that the driver letter is correct first (reparse / symlink issues). */ + memset(&Ios, 0, sizeof(Ios)); + rcNt = g_pfnNtQueryVolumeInformationFile(hFile, &Ios, pFsVolInfo, sizeof(*pFsVolInfo), MY_FileFsVolumeInformation); + if (rcNt >= 0) + { + /** @todo do a quick search and try correct the drive letter? */ + if ( pFsVolInfo->VolumeCreationTime.QuadPart == g_aVolumeInfo[iDrv].VolumeCreationTime.QuadPart + && pFsVolInfo->VolumeSerialNumber == g_aVolumeInfo[iDrv].VolumeSerialNumber) + { + memset(&Ios, 0, sizeof(Ios)); + rcNt = g_pfnNtQueryInformationFile(hFile, &Ios, abBuf, sizeof(abBuf), MY_FileNameInformation); + if (rcNt >= 0) + { + cchOut = WideCharToMultiByte(CP_ACP, 0, + pFileNameInfo->FileName, pFileNameInfo->FileNameLength / sizeof(WCHAR), + psz, (int)(cchFull - (psz - pszFull) - 2), NULL, NULL); + if (cchOut > 0) + { + const char *pszEnd; +#if 0 + /* upper case the server and share */ + if (fUnc) + { + for (psz++; *psz != '/' && *psz != '\\'; psz++) + *psz = toupper(*psz); + for (psz++; *psz != '/' && *psz != '\\'; psz++) + *psz = toupper(*psz); + } +#endif + /* add trailing slash on directories if input has it. */ + pszEnd = strchr(pszPath, '\0'); + if ( (pszEnd[-1] == '/' || pszEnd[-1] == '\\') + && psz[cchOut - 1] != '\\' + && psz[cchOut - 1] != '//') + psz[cchOut++] = '\\'; + + /* make sure it's terminated */ + psz[cchOut] = '\0'; + rc = 0; + } + else + rc = -3; + } + else + rc = -4; + } + else + rc = -5; + } + else + rc = -6; + CloseHandle(hFile); + } + else + rc = -7; + return rc; +} + +/** + * Somewhat similar to fullpath, except that it will fix + * the case of existing path components. + */ +void +nt_fullpath(const char *pszPath, char *pszFull, size_t cchFull) +{ +#if 0 + static int s_cHits = 0; + static int s_cFallbacks = 0; +#endif + + /* + * The simple case, the file / dir / whatever exists and can be + * queried without problems and spaces. + */ + if (nt_get_filename_info(pszPath, pszFull, cchFull) == 0) + { + /** @todo make nt_get_filename_info return spaceless path. */ + if (strchr(pszFull, ' ')) + w32_fixcase(pszFull); +#if 0 + fprintf(stdout, "nt #%d - %s\n", ++s_cHits, pszFull); + fprintf(stdout, " #%d - %s\n", s_cHits, pszPath); +#endif + return; + } + if (g_pfnNtQueryInformationFile) + { + /* do _fullpath and drop off path elements until we get a hit... - later */ + } + + /* + * For now, simply fall back on the old method. + */ + _fullpath(pszFull, pszPath, cchFull); + w32_fixcase(pszFull); +#if 0 + fprintf(stderr, "fb #%d - %s\n", ++s_cFallbacks, pszFull); + fprintf(stderr, " #%d - %s\n", s_cFallbacks, pszPath); +#endif +} + |