|
| 1 | +#define USE_THE_REPOSITORY_VARIABLE |
| 2 | +#include "../../git-compat-util.h" |
| 3 | +#include "../win32.h" |
| 4 | +#include "../../repository.h" |
| 5 | +#include "config.h" |
| 6 | +#include "ntifs.h" |
| 7 | +#include "wsl.h" |
| 8 | + |
| 9 | +int are_wsl_compatible_mode_bits_enabled(void) |
| 10 | +{ |
| 11 | + /* default to `false` during initialization */ |
| 12 | + static const int fallback = 0; |
| 13 | + static int enabled = -1; |
| 14 | + |
| 15 | + if (enabled < 0) { |
| 16 | + /* avoid infinite recursion */ |
| 17 | + if (!the_repository) |
| 18 | + return fallback; |
| 19 | + |
| 20 | + if (the_repository->config && |
| 21 | + the_repository->config->hash_initialized && |
| 22 | + repo_config_get_bool(the_repository, "core.wslcompat", &enabled) < 0) |
| 23 | + enabled = 0; |
| 24 | + } |
| 25 | + |
| 26 | + return enabled < 0 ? fallback : enabled; |
| 27 | +} |
| 28 | + |
| 29 | +int copy_wsl_mode_bits_from_disk(const wchar_t *wpath, ssize_t wpathlen, |
| 30 | + _mode_t *mode) |
| 31 | +{ |
| 32 | + int ret = -1; |
| 33 | + HANDLE h; |
| 34 | + if (wpathlen >= 0) { |
| 35 | + /* |
| 36 | + * It's caller's duty to make sure wpathlen is reasonable so |
| 37 | + * it does not overflow. |
| 38 | + */ |
| 39 | + wchar_t *fn2 = (wchar_t*)alloca((wpathlen + 1) * sizeof(wchar_t)); |
| 40 | + memcpy(fn2, wpath, wpathlen * sizeof(wchar_t)); |
| 41 | + fn2[wpathlen] = 0; |
| 42 | + wpath = fn2; |
| 43 | + } |
| 44 | + h = CreateFileW(wpath, FILE_READ_EA | SYNCHRONIZE, |
| 45 | + FILE_SHARE_READ | FILE_SHARE_WRITE | FILE_SHARE_DELETE, |
| 46 | + NULL, OPEN_EXISTING, |
| 47 | + FILE_FLAG_BACKUP_SEMANTICS | |
| 48 | + FILE_FLAG_OPEN_REPARSE_POINT, |
| 49 | + NULL); |
| 50 | + if (h != INVALID_HANDLE_VALUE) { |
| 51 | + ret = get_wsl_mode_bits_by_handle(h, mode); |
| 52 | + CloseHandle(h); |
| 53 | + } |
| 54 | + return ret; |
| 55 | +} |
| 56 | + |
| 57 | +#ifndef LX_FILE_METADATA_HAS_UID |
| 58 | +#define LX_FILE_METADATA_HAS_UID 0x1 |
| 59 | +#define LX_FILE_METADATA_HAS_GID 0x2 |
| 60 | +#define LX_FILE_METADATA_HAS_MODE 0x4 |
| 61 | +#define LX_FILE_METADATA_HAS_DEVICE_ID 0x8 |
| 62 | +#define LX_FILE_CASE_SENSITIVE_DIR 0x10 |
| 63 | +typedef struct _FILE_STAT_LX_INFORMATION { |
| 64 | + LARGE_INTEGER FileId; |
| 65 | + LARGE_INTEGER CreationTime; |
| 66 | + LARGE_INTEGER LastAccessTime; |
| 67 | + LARGE_INTEGER LastWriteTime; |
| 68 | + LARGE_INTEGER ChangeTime; |
| 69 | + LARGE_INTEGER AllocationSize; |
| 70 | + LARGE_INTEGER EndOfFile; |
| 71 | + uint32_t FileAttributes; |
| 72 | + uint32_t ReparseTag; |
| 73 | + uint32_t NumberOfLinks; |
| 74 | + ACCESS_MASK EffectiveAccess; |
| 75 | + uint32_t LxFlags; |
| 76 | + uint32_t LxUid; |
| 77 | + uint32_t LxGid; |
| 78 | + uint32_t LxMode; |
| 79 | + uint32_t LxDeviceIdMajor; |
| 80 | + uint32_t LxDeviceIdMinor; |
| 81 | +} FILE_STAT_LX_INFORMATION, *PFILE_STAT_LX_INFORMATION; |
| 82 | +#endif |
| 83 | + |
| 84 | +/* |
| 85 | + * This struct is extended from the original FILE_FULL_EA_INFORMATION of |
| 86 | + * Microsoft Windows. |
| 87 | + */ |
| 88 | +struct wsl_full_ea_info_t { |
| 89 | + uint32_t NextEntryOffset; |
| 90 | + uint8_t Flags; |
| 91 | + uint8_t EaNameLength; |
| 92 | + uint16_t EaValueLength; |
| 93 | + char EaName[7]; |
| 94 | + char EaValue[4]; |
| 95 | + char Padding[1]; |
| 96 | +}; |
| 97 | + |
| 98 | +enum { |
| 99 | + FileStatLxInformation = 70, |
| 100 | +}; |
| 101 | +__declspec(dllimport) NTSTATUS WINAPI |
| 102 | + NtQueryInformationFile(HANDLE FileHandle, |
| 103 | + PIO_STATUS_BLOCK IoStatusBlock, |
| 104 | + PVOID FileInformation, ULONG Length, |
| 105 | + uint32_t FileInformationClass); |
| 106 | +__declspec(dllimport) NTSTATUS WINAPI |
| 107 | + NtSetInformationFile(HANDLE FileHandle, PIO_STATUS_BLOCK IoStatusBlock, |
| 108 | + PVOID FileInformation, ULONG Length, |
| 109 | + uint32_t FileInformationClass); |
| 110 | +__declspec(dllimport) NTSTATUS WINAPI |
| 111 | + NtSetEaFile(HANDLE FileHandle, PIO_STATUS_BLOCK IoStatusBlock, |
| 112 | + PVOID EaBuffer, ULONG EaBufferSize); |
| 113 | + |
| 114 | +int set_wsl_mode_bits_by_handle(HANDLE h, _mode_t mode) |
| 115 | +{ |
| 116 | + uint32_t value = mode; |
| 117 | + struct wsl_full_ea_info_t ea_info; |
| 118 | + IO_STATUS_BLOCK iob; |
| 119 | + /* mode should be valid to make WSL happy */ |
| 120 | + assert(S_ISREG(mode) || S_ISDIR(mode)); |
| 121 | + ea_info.NextEntryOffset = 0; |
| 122 | + ea_info.Flags = 0; |
| 123 | + ea_info.EaNameLength = 6; |
| 124 | + ea_info.EaValueLength = sizeof(value); /* 4 */ |
| 125 | + strlcpy(ea_info.EaName, "$LXMOD", sizeof(ea_info.EaName)); |
| 126 | + memcpy(ea_info.EaValue, &value, sizeof(value)); |
| 127 | + ea_info.Padding[0] = 0; |
| 128 | + return NtSetEaFile(h, &iob, &ea_info, sizeof(ea_info)); |
| 129 | +} |
| 130 | + |
| 131 | +int get_wsl_mode_bits_by_handle(HANDLE h, _mode_t *mode) |
| 132 | +{ |
| 133 | + FILE_STAT_LX_INFORMATION fxi; |
| 134 | + IO_STATUS_BLOCK iob; |
| 135 | + if (NtQueryInformationFile(h, &iob, &fxi, sizeof(fxi), |
| 136 | + FileStatLxInformation) == 0) { |
| 137 | + if (fxi.LxFlags & LX_FILE_METADATA_HAS_MODE) |
| 138 | + *mode = (_mode_t)fxi.LxMode; |
| 139 | + return 0; |
| 140 | + } |
| 141 | + return -1; |
| 142 | +} |
0 commit comments