Retro68/LaunchAPPL/Common/Test.cc

#include "ReliableStream.h"
#include <iostream>
#include <vector>
#include <string>
#include <stdint.h>

class Forwarder : public Stream
{
public:
    Forwarder *other;
    std::vector<std::vector<uint8_t>> packets;

    virtual void write(const void* p, size_t n)
    {
        other->enqueueReceive(p,n);
    }

    void enqueueReceive(const void* p, size_t n)
    {
        packets.emplace_back((const uint8_t*)p, ((const uint8_t*)p)+n);
    }

    void idle()
    {
        std::vector<std::vector<uint8_t>> packets2;
        packets2.swap(packets);
        for(auto& p : packets2)
            notifyReceive(p.data(), p.size());
    }
};

class DumpToConsole : public StreamListener
{
public:
    std::string prefix;

    DumpToConsole(std::string prefix) : prefix(prefix) {}

    size_t onReceive(const uint8_t* p, size_t n)
    {
        std::cout << prefix;
        for(int i = 0; i < n; i++)
        {
            if(p[i] >= 128 || p[i] < 32)
                std::cout << "\\x" << std::hex << (unsigned)p[i];
            else
                std::cout << p[i];
        }
        std::cout << std::endl;
        return n;
    }
};

int main()
{
    Forwarder fwd1, fwd2;
    fwd1.other = &fwd2;
    fwd2.other = &fwd1;

    ReliableStream rel1(fwd1), rel2(fwd2);
    fwd1.setListener(&rel1);
    fwd2.setListener(&rel2);

    DumpToConsole dump1("one:");
    DumpToConsole dump2("two:");
    
    rel1.setListener(&dump1);
    rel2.setListener(&dump2);

    rel1.write("Hello, world.", 13);
    rel1.flushWrite();

    for(int i = 0; i < 100; i++)
    {
        fwd1.idle();
        fwd2.idle();
    }
    return 0;
}