mirror/zig
1
0
Fork 0
mirror of https://github.com/ziglang/zig.git synced 2025-12-06 06:13:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

windows: use array of tmp bufs as backing store for input memory to ntdll

Browse Source
This commit is contained in:
Jakub Konka 2022-12-02 10:20:05 +01:00 committed by Andrew Kelley
parent 16dc86a49e
commit 5eaacf1ce9
1 changed files with 11 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
lib/std/zig/system/windows.zig
View File

@ -1,5 +1,6 @@
const std = @import("std");
const builtin = @import("builtin");
const assert = std.debug.assert;
const mem = std.mem;
const Target = std.Target;
@ -89,6 +90,8 @@ fn getCpuInfoFromRegistry(core: usize, args: anytype) !void {
.DefaultLength = 0,
};
var tmp_bufs: [fields_info.len][max_value_len]u8 align(@alignOf(std.os.windows.UNICODE_STRING)) = undefined;
inline for (fields_info) |field, i| {
const ctx: *anyopaque = blk: {
switch (@field(args, field.name).value_type) {
@ -96,26 +99,20 @@ fn getCpuInfoFromRegistry(core: usize, args: anytype) !void {
REG.EXPAND_SZ,
REG.MULTI_SZ,
=> {
var buf: [max_value_len / 2]u16 = undefined;
var unicode = std.os.windows.UNICODE_STRING{
comptime assert(@sizeOf(std.os.windows.UNICODE_STRING) % 2 == 0);
const unicode = @ptrCast(*std.os.windows.UNICODE_STRING, &tmp_bufs[i]);
unicode.* = .{
.Length = 0,
.MaximumLength = max_value_len,
.Buffer = &buf,
.MaximumLength = max_value_len - @sizeOf(std.os.windows.UNICODE_STRING),
.Buffer = @ptrCast([*]u16, tmp_bufs[i][@sizeOf(std.os.windows.UNICODE_STRING)..]),
};
break :blk &unicode;
break :blk unicode;
},
REG.DWORD,
REG.DWORD_BIG_ENDIAN,
=> {
var buf: [4]u8 = undefined;
break :blk &buf;
},
REG.QWORD => {
var buf: [8]u8 = undefined;
break :blk &buf;
},
REG.QWORD,
=> break :blk &tmp_bufs[i],
else => unreachable,
}