58 lines
1.9 KiB
Zig
58 lines
1.9 KiB
Zig
const std = @import("std");
|
|
const net = std.net;
|
|
const fmt = std.fmt;
|
|
const os = std.os;
|
|
|
|
const print = std.debug.print;
|
|
const testing = std.testing;
|
|
const print_eq = @import("./util.zig").print_eq;
|
|
const URI = @import("./util.zig").URI;
|
|
|
|
fn init_tcp_server(allocator: std.mem.Allocator) !net.StreamServer {
|
|
var buffer: [1000]u8 = undefined;
|
|
var fbs = std.io.fixedBufferStream(&buffer);
|
|
var writer = fbs.writer();
|
|
|
|
var address = std.process.getEnvVarOwned(allocator, "ADDRESS") catch |err| switch(err) {
|
|
error.EnvironmentVariableNotFound => blk: {
|
|
print("no ADDRESS envvar: TCPd will listen on 127.0.0.1:9000\n", .{});
|
|
break :blk try allocator.dupe(u8, "127.0.0.1:9000");
|
|
},
|
|
else => { return err; },
|
|
};
|
|
defer allocator.free(address);
|
|
|
|
try writer.print("{s}", .{address});
|
|
var tcp_address = fbs.getWritten();
|
|
|
|
var iterator = std.mem.split(u8, tcp_address, ":");
|
|
var real_tcp_address = iterator.first();
|
|
var real_tcp_port = try std.fmt.parseUnsigned(u16, iterator.rest(), 10);
|
|
|
|
print ("TCP address [{s}] port [{}]\n", .{real_tcp_address, real_tcp_port});
|
|
|
|
var server = net.StreamServer.init(.{.reuse_address = true});
|
|
var socket_addr = try net.Address.parseIp(real_tcp_address, real_tcp_port);
|
|
try server.listen(socket_addr);
|
|
|
|
return server;
|
|
}
|
|
|
|
fn accept_read(server: *net.StreamServer) !void {
|
|
var client = try server.accept(); // net.StreamServer.Connection
|
|
var buffer: [1000]u8 = undefined;
|
|
var size = try client.stream.read(&buffer);
|
|
print ("received: {s}\n", .{buffer[0..size]});
|
|
client.stream.close();
|
|
}
|
|
|
|
pub fn main() !u8 {
|
|
const config = .{.safety = true};
|
|
var gpa = std.heap.GeneralPurposeAllocator(config){};
|
|
defer _ = gpa.deinit();
|
|
const allocator = gpa.allocator();
|
|
var streamserver = try init_tcp_server(allocator);
|
|
try accept_read(&streamserver);
|
|
return 0;
|
|
}
|