| ... | @@ -924,23 +924,12 @@ pub fn GetFinalPathNameByHandle( | ... | @@ -924,23 +924,12 @@ pub fn GetFinalPathNameByHandle( |
| 924 | out_buffer: []u16, | 924 | out_buffer: []u16, |
| 925 | ) GetFinalPathNameByHandleError![]u16 { | 925 | ) GetFinalPathNameByHandleError![]u16 { |
| 926 | // Get normalized path; doesn't include volume name though. | 926 | // Get normalized path; doesn't include volume name though. |
| 927 | var path_buffer: [PATH_MAX_WIDE * 2]u8 align(@alignOf(FILE_NAME_INFORMATION)) = undefined; | 927 | var path_buffer: [@sizeOf(FILE_NAME_INFORMATION) + PATH_MAX_WIDE * 2]u8 align(@alignOf(FILE_NAME_INFORMATION)) = undefined; |
| 928 | var io: IO_STATUS_BLOCK = undefined; | 928 | try QueryInformationFile(hFile, FILE_INFORMATION_CLASS.FileNormalizedNameInformation, path_buffer[0..]); |
| 929 | var rc = ntdll.NtQueryInformationFile(hFile, &io, &path_buffer, path_buffer.len, FILE_INFORMATION_CLASS.FileNormalizedNameInformation); | | |
| 930 | switch (rc) { | | |
| 931 | .SUCCESS => {}, | | |
| 932 | .INVALID_PARAMETER => unreachable, | | |
| 933 | else => return unexpectedStatus(rc), | | |
| 934 | } | | |
| 935 | | 929 | |
| 936 | // Get NT volume name. | 930 | // Get NT volume name. |
| 937 | var volume_buffer: [MAX_PATH]u8 align(@alignOf(FILE_NAME_INFORMATION)) = undefined; // MAX_PATH bytes should be enough since it's Windows-defined name | 931 | var volume_buffer: [@sizeOf(FILE_NAME_INFORMATION) + MAX_PATH]u8 align(@alignOf(FILE_NAME_INFORMATION)) = undefined; // MAX_PATH bytes should be enough since it's Windows-defined name |
| 938 | rc = ntdll.NtQueryInformationFile(hFile, &io, &volume_buffer, volume_buffer.len, FILE_INFORMATION_CLASS.FileVolumeNameInformation); | 932 | try QueryInformationFile(hFile, FILE_INFORMATION_CLASS.FileVolumeNameInformation, volume_buffer[0..]); |
| 939 | switch (rc) { | | |
| 940 | .SUCCESS => {}, | | |
| 941 | .INVALID_PARAMETER => unreachable, | | |
| 942 | else => return unexpectedStatus(rc), | | |
| 943 | } | | |
| 944 | | 933 | |
| 945 | const file_name = @ptrCast(*const FILE_NAME_INFORMATION, &path_buffer[0]); | 934 | const file_name = @ptrCast(*const FILE_NAME_INFORMATION, &path_buffer[0]); |
| 946 | const file_name_u16 = @ptrCast([*]const u16, &file_name.FileName[0])[0 .. file_name.FileNameLength / 2]; | 935 | const file_name_u16 = @ptrCast([*]const u16, &file_name.FileName[0])[0 .. file_name.FileNameLength / 2]; |
| ... | @@ -1014,9 +1003,7 @@ pub fn GetFinalPathNameByHandle( | ... | @@ -1014,9 +1003,7 @@ pub fn GetFinalPathNameByHandle( |
| 1014 | // with traditional DOS drive letters, so pick the first one available. | 1003 | // with traditional DOS drive letters, so pick the first one available. |
| 1015 | const prefix = &[_]u16{ '\\', 'D', 'o', 's', 'D', 'e', 'v', 'i', 'c', 'e', 's', '\\' }; | 1004 | const prefix = &[_]u16{ '\\', 'D', 'o', 's', 'D', 'e', 'v', 'i', 'c', 'e', 's', '\\' }; |
| 1016 | | 1005 | |
| 1017 | if (std.mem.indexOf(u16, symlink, prefix)) |idx| { | 1006 | if (std.mem.startsWith(u16, symlink, prefix)) { |
| 1018 | if (idx != 0) continue; | | |
| 1019 | | | |
| 1020 | const drive_letter = symlink[prefix.len..]; | 1007 | const drive_letter = symlink[prefix.len..]; |
| 1021 | | 1008 | |
| 1022 | if (out_buffer.len < drive_letter.len + file_name_u16.len) return error.NameTooLong; | 1009 | if (out_buffer.len < drive_letter.len + file_name_u16.len) return error.NameTooLong; |
| ... | @@ -1035,6 +1022,25 @@ pub fn GetFinalPathNameByHandle( | ... | @@ -1035,6 +1022,25 @@ pub fn GetFinalPathNameByHandle( |
| 1035 | } | 1022 | } |
| 1036 | } | 1023 | } |
| 1037 | | 1024 | |
| | 1025 | pub const QueryInformationFileError = error{Unexpected}; |
| | 1026 | |
| | 1027 | pub fn QueryInformationFile( |
| | 1028 | handle: HANDLE, |
| | 1029 | info_class: FILE_INFORMATION_CLASS, |
| | 1030 | out_buffer: []u8, |
| | 1031 | ) QueryInformationFileError!void { |
| | 1032 | var io: IO_STATUS_BLOCK = undefined; |
| | 1033 | const len_bytes = std.math.cast(u32, out_buffer.len) catch |err| switch (err) { |
| | 1034 | error.Overflow => std.math.maxInt(u32), // If the provided buffer is larger than what we can handle, set size to max what we can handle |
| | 1035 | }; |
| | 1036 | const rc = ntdll.NtQueryInformationFile(handle, &io, out_buffer.ptr, len_bytes, info_class); |
| | 1037 | switch (rc) { |
| | 1038 | .SUCCESS => {}, |
| | 1039 | .INVALID_PARAMETER => unreachable, |
| | 1040 | else => return unexpectedStatus(rc), |
| | 1041 | } |
| | 1042 | } |
| | 1043 | |
| 1038 | pub const GetFileSizeError = error{Unexpected}; | 1044 | pub const GetFileSizeError = error{Unexpected}; |
| 1039 | | 1045 | |
| 1040 | pub fn GetFileSizeEx(hFile: HANDLE) GetFileSizeError!u64 { | 1046 | pub fn GetFileSizeEx(hFile: HANDLE) GetFileSizeError!u64 { |
Jakub Konka <kubkon@jakubkonka.com> 2020-08-09 10:47:52+02:00