summaryrefslogtreecommitdiff
path: root/src/main.zig
blob: ebb7d333960fd091ada9c5f41ff25e984777b7d9 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
//! By convention, main.zig is where your main function lives in the case that
//! you are building an executable. If you are making a library, the convention
//! is to delete this file and start with root.zig instead.

const SaprusPacketType = enum(u16) {
    relay = 0x003C,
    file_transfer = 0x8888,
};

const SaprusHeaderFrame = struct {
    msg_type: SaprusPacketType,
    payload: []u8,

    const Self = @This();

    fn toBytes(s: Self, allocator: Allocator) ![]u8 {
        const buf = allocator.alloc(u8, 32 + s.payload.len);
        std.mem.writeInt(u16, buf[0..2], s.msg_type, .big);
        std.mem.writeInt(u16, buf[2..4], s.payload.len);
        std.mem.copyForwards(u8, buf[4..], s.payload);
        return buf;
    }
};

const SaprusRelayMessage = struct {
    dest: [4]u8,
    payload: []u8,

    const Self = @This();

    fn toBytes(s: Self, allocator: Allocator) ![]u8 {
        const buf = allocator.alloc(u8, 4 + s.payload.len);
        std.mem.copyForwards(u8, buf[0..4], s.dest);
        std.mem.copyForwards(u8, buf[4..], s.payload);
        return buf;
    }
};

pub fn main() !void {
    // Prints to stderr (it's a shortcut based on `std.io.getStdErr()`)
    std.debug.print("All your {s} are belong to us.\n", .{"codebase"});

    // stdout is for the actual output of your application, for example if you
    // are implementing gzip, then only the compressed bytes should be sent to
    // stdout, not any debugging messages.
    const stdout_file = std.io.getStdOut().writer();
    var bw = std.io.bufferedWriter(stdout_file);
    const stdout = bw.writer();

    try stdout.print("Run `zig build test` to run the tests.\n", .{});

    try bw.flush(); // Don't forget to flush!
}

test "simple test" {
    var list = std.ArrayList(i32).init(std.testing.allocator);
    defer list.deinit(); // Try commenting this out and see if zig detects the memory leak!
    try list.append(42);
    try std.testing.expectEqual(@as(i32, 42), list.pop());
}

test "fuzz example" {
    const Context = struct {
        fn testOne(context: @This(), input: []const u8) anyerror!void {
            _ = context;
            // Try passing `--fuzz` to `zig build test` and see if it manages to fail this test case!
            try std.testing.expect(!std.mem.eql(u8, "canyoufindme", input));
        }
    };
    try std.testing.fuzz(Context{}, Context.testOne, .{});
}

const std = @import("std");
const Allocator = std.mem.Allocator;