| ... | ... | @@ -1715,15 +1715,13 @@ pub fn getenvZ(key: [*:0]const u8) ?[]const u8 { |
| 1715 | 1715 | |
| 1716 | 1716 | /// Windows-only. Get an environment variable with a null-terminated, WTF-16 encoded name. |
| 1717 | 1717 | /// See also `getenv`. |
| 1718 | | /// This function first attempts a case-sensitive lookup. If no match is found, and `key` |
| 1719 | | /// is ASCII, then it attempts a second case-insensitive lookup. |
| 1718 | /// This function performs a Unicode-aware case-insensitive lookup using RtlEqualUnicodeString. |
| 1720 | 1719 | pub fn getenvW(key: [*:0]const u16) ?[:0]const u16 { |
| 1721 | 1720 | if (builtin.os.tag != .windows) { |
| 1722 | 1721 | @compileError("std.os.getenvW is a Windows-only API"); |
| 1723 | 1722 | } |
| 1724 | 1723 | const key_slice = mem.sliceTo(key, 0); |
| 1725 | 1724 | const ptr = windows.peb().ProcessParameters.Environment; |
| 1726 | | var ascii_match: ?[:0]const u16 = null; |
| 1727 | 1725 | var i: usize = 0; |
| 1728 | 1726 | while (ptr[i] != 0) { |
| 1729 | 1727 | const key_start = i; |
| ... | ... | @@ -1737,22 +1735,25 @@ pub fn getenvW(key: [*:0]const u16) ?[:0]const u16 { |
| 1737 | 1735 | while (ptr[i] != 0) : (i += 1) {} |
| 1738 | 1736 | const this_value = ptr[value_start..i :0]; |
| 1739 | 1737 | |
| 1740 | | if (mem.eql(u16, key_slice, this_key)) return this_value; |
| 1741 | | |
| 1742 | | ascii_check: { |
| 1743 | | if (ascii_match != null) break :ascii_check; |
| 1744 | | if (key_slice.len != this_key.len) break :ascii_check; |
| 1745 | | for (key_slice) |a_c, key_index| { |
| 1746 | | const a = math.cast(u8, a_c) catch break :ascii_check; |
| 1747 | | const b = math.cast(u8, this_key[key_index]) catch break :ascii_check; |
| 1748 | | if (std.ascii.toLower(a) != std.ascii.toLower(b)) break :ascii_check; |
| 1749 | | } |
| 1750 | | ascii_match = this_value; |
| 1738 | const key_string_bytes = @intCast(u16, key_slice.len * 2); |
| 1739 | const key_string = windows.UNICODE_STRING{ |
| 1740 | .Length = key_string_bytes, |
| 1741 | .MaximumLength = key_string_bytes, |
| 1742 | .Buffer = @intToPtr([*]u16, @ptrToInt(key)), |
| 1743 | }; |
| 1744 | const this_key_string_bytes = @intCast(u16, this_key.len * 2); |
| 1745 | const this_key_string = windows.UNICODE_STRING{ |
| 1746 | .Length = this_key_string_bytes, |
| 1747 | .MaximumLength = this_key_string_bytes, |
| 1748 | .Buffer = this_key.ptr, |
| 1749 | }; |
| 1750 | if (windows.ntdll.RtlEqualUnicodeString(&key_string, &this_key_string, windows.TRUE) == windows.TRUE) { |
| 1751 | return this_value; |
| 1751 | 1752 | } |
| 1752 | 1753 | |
| 1753 | 1754 | i += 1; // skip over null byte |
| 1754 | 1755 | } |
| 1755 | | return ascii_match; |
| 1756 | return null; |
| 1756 | 1757 | } |
| 1757 | 1758 | |
| 1758 | 1759 | pub const GetCwdError = error{ |
Ryan Liptak <squeek502@hotmail.com> 2022-01-15 04:56:32-08:00