1const std = @import("std");
 2
 3/// Returns 1 on success, 0 on failure
 4export fn verify(argc: c_int, argv: [*]const [*:0]const u16) c_int {
 5    const argv_slice = argv[0..@intCast(argc)];
 6    testArgv(argv_slice) catch |err| switch (err) {
 7        error.OutOfMemory => @panic("oom"),
 8        error.Overflow => @panic("bytes needed to contain args would overflow usize"),
 9        error.ArgvMismatch => return 0,
10    };
11    return 1;
12}
13
14fn testArgv(expected_args: []const [*:0]const u16) !void {
15    var arena_state = std.heap.ArenaAllocator.init(std.heap.page_allocator);
16    defer arena_state.deinit();
17    const allocator = arena_state.allocator();
18
19    const args = try std.process.argsAlloc(allocator);
20    var wtf8_buf = std.array_list.Managed(u8).init(allocator);
21
22    var eql = true;
23    if (args.len != expected_args.len) eql = false;
24
25    const min_len = @min(expected_args.len, args.len);
26    for (expected_args[0..min_len], args[0..min_len], 0..) |expected_arg, arg_wtf8, i| {
27        wtf8_buf.clearRetainingCapacity();
28        try std.unicode.wtf16LeToWtf8ArrayList(&wtf8_buf, std.mem.span(expected_arg));
29        if (!std.mem.eql(u8, wtf8_buf.items, arg_wtf8)) {
30            std.debug.print("{}: expected: \"{f}\"\n", .{ i, std.zig.fmtString(wtf8_buf.items) });
31            std.debug.print("{}:   actual: \"{f}\"\n", .{ i, std.zig.fmtString(arg_wtf8) });
32            eql = false;
33        }
34    }
35    if (!eql) {
36        for (expected_args[min_len..], min_len..) |arg, i| {
37            wtf8_buf.clearRetainingCapacity();
38            try std.unicode.wtf16LeToWtf8ArrayList(&wtf8_buf, std.mem.span(arg));
39            std.debug.print("{}: expected: \"{f}\"\n", .{ i, std.zig.fmtString(wtf8_buf.items) });
40        }
41        for (args[min_len..], min_len..) |arg, i| {
42            std.debug.print("{}:   actual: \"{f}\"\n", .{ i, std.zig.fmtString(arg) });
43        }
44        const peb = std.os.windows.peb();
45        const lpCmdLine: [*:0]u16 = @ptrCast(peb.ProcessParameters.CommandLine.Buffer);
46        wtf8_buf.clearRetainingCapacity();
47        try std.unicode.wtf16LeToWtf8ArrayList(&wtf8_buf, std.mem.span(lpCmdLine));
48        std.debug.print("command line: \"{f}\"\n", .{std.zig.fmtString(wtf8_buf.items)});
49        std.debug.print("expected argv:\n", .{});
50        std.debug.print("&.{{\n", .{});
51        for (expected_args) |arg| {
52            wtf8_buf.clearRetainingCapacity();
53            try std.unicode.wtf16LeToWtf8ArrayList(&wtf8_buf, std.mem.span(arg));
54            std.debug.print("    \"{f}\",\n", .{std.zig.fmtString(wtf8_buf.items)});
55        }
56        std.debug.print("}}\n", .{});
57        return error.ArgvMismatch;
58    }
59}