summaryrefslogtreecommitdiffstats
path: root/src/lib/nt_fullpath.c
diff options
context:
space:
mode:
Diffstat (limited to 'src/lib/nt_fullpath.c')
-rw-r--r--src/lib/nt_fullpath.c580
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
+}
+
generated by cgit v1.2.3 (git 2.39.1) at 2024-05-22 06:48:05 +0000