Ticket #4112: test.cpp

#include <stdio.h>
#include <net/NetBuffer.h>
#include <net/NetEndpoint.h>

int32 PingHandler(void* data) {
    BNetEndpoint* client = (BNetEndpoint*)data;
    while (true) {
        int32 size = client->Send("PING\n", 5);
        snooze(5000000L);
    }
}

int32 ConnectionHandler(void* data) {
    BNetEndpoint* client = (BNetEndpoint*)data;
    BNetBuffer buffer;
    bool connection_closed = false;
    while (!connection_closed) {
        bool result = client->IsDataPending(10000000L);
        if (result) {
            int32 size = client->Receive(buffer, 1024);
            printf("Got data. Size = %d\n", size);
            if (size == 0) {
                printf("Connection closed by remote host.\n");
                client->Close();
                delete client;
                connection_closed = true;
            } 
        } else {
            printf("Timeout.\n");
        }
    }
}

void StartConnectionHandler(BNetEndpoint* client) {
    printf("Starting connection thread.\n");
    thread_id handler_thread = spawn_thread(ConnectionHandler,
        "Connection Handler", B_NORMAL_PRIORITY, client);
    if (handler_thread < B_OK) {
        printf("Error when spawning connection handler thread.\n");
        delete client;
    } else {
        resume_thread(handler_thread);
        printf("Connection Handler thread started.\n");
    }
    
    printf("Starting ping thread.\n");
    thread_id ping_thread = spawn_thread(PingHandler,
        "Ping Handler", B_NORMAL_PRIORITY, client);
    if (ping_thread < B_OK) {
        printf("Error when spawning ping thread.\n");
        delete client;
    } else {
        resume_thread(ping_thread);
        printf("Ping thread started.\n");
    }
    
}

int main(void) {
    BNetEndpoint server_port;
    
    if (server_port.InitCheck() != B_OK) {
        printf("Error creating end point.\n");
        exit (0); 
    }
    
    if (server_port.Bind(8080) != B_OK) {
        printf("Error binding to port 8080 : %s\n", server_port.ErrorStr());
        exit (0);
    }
    
    if (server_port.Listen() != B_OK) {
        printf("Error listenning or port 8080 : %s\n", server_port.ErrorStr());
        exit (0);
    }
    
    while (true) {
        BNetEndpoint* client = NULL;
        client = server_port.Accept();
        printf("Got new connection request.\n");
        if (client) {
            StartConnectionHandler(client);
        }
    }   
}