net_tcp_timeout.c

TCP server demonstrating accept timeout handling.

TCP server demonstrating accept timeout handling.Demonstrates: bj_accept_tcp with timeout, BJ_ERROR_NETWORK_TIMEOUT, doing periodic work while waiting for connections.

 
#include <banjo/log.h>
#include <banjo/main.h>
#include <banjo/net.h>
#include <banjo/system.h>
#include <banjo/error.h>
 
#define PORT 8085
#define ACCEPT_TIMEOUT_MS 5000
 
int main(int argc, char* argv[]) {
    (void)argc;
    (void)argv;
 
    if (!bj_begin(BJ_NETWORK_SYSTEM, 0)) {
        return 1;
    }
 
    struct bj_error* error = 0;
 
    struct bj_tcp_listener* listener = bj_listen_tcp("0.0.0.0", PORT, BJ_RESOLVE_IPV4, 5, &error);
    if (!listener) {
        bj_err("server: failed to listen (%s)", bj_error_message(error));
        bj_clear_error(&error);
        bj_end();
        return 1;
    }
 
    bj_info("server: listening on port %d", bj_tcp_listener_port(listener));
    bj_info("server: accept timeout is %d ms", ACCEPT_TIMEOUT_MS);
 
    int timeout_count = 0;
    int connection_count = 0;
 
    while (1) {
        struct bj_tcp_stream* client = bj_accept_tcp(listener, ACCEPT_TIMEOUT_MS, &error);
 
        if (!client) {
            if (error && bj_error_code(error) == BJ_ERROR_NETWORK_TIMEOUT) {
                timeout_count++;
                bj_info("server: accept timed out (%d timeouts, %d connections so far)",
                        timeout_count, connection_count);
                bj_clear_error(&error);
                // Could do periodic maintenance here
                continue;
            }
 
            bj_err("server: accept failed (%s)", bj_error_message(error));
            bj_clear_error(&error);
            continue;
        }
 
        connection_count++;
 
        char addr[BJ_ADDR_STRLEN];
        bj_tcp_peer_address(client, addr, sizeof(addr));
        bj_info("server: client %d connected from %s", connection_count, addr);
 
        char buf[256];
        int n;
        while ((n = bj_tcp_recv(client, buf, sizeof(buf) - 1)) > 0) {
            buf[n] = '\0';
            bj_info("server: client %d: \"%s\"", connection_count, buf);
            bj_tcp_send(client, buf, n);
        }
 
        bj_info("server: client %d disconnected", connection_count);
        bj_close_tcp_stream(client);
    }
 
    bj_close_tcp_listener(listener);
    bj_end();
    return 0;
}