MP-SPDZ/Networking/Receiver.h

/*
 * Receiver.h
 *
 */

#ifndef NETWORKING_RECEIVER_H_
#define NETWORKING_RECEIVER_H_

#include <pthread.h>

#include "Tools/octetStream.h"
#include "Tools/WaitQueue.h"
#include "Tools/time-func.h"

class CommunicationThread
{
    int other;

protected:
    CommunicationThread(int other);
    virtual ~CommunicationThread() {}

    void run();
    virtual void run_with_error() = 0;
};

template<class T>
class Receiver : CommunicationThread
{
    T socket;
    WaitQueue<octetStream*> in;
    WaitQueue<octetStream*> out;
    pthread_t thread;

    static void* run_thread(void* receiver);

    // prevent copying
    Receiver(const Receiver& other);

    void start();
    void stop();
    void run_with_error();

public:
    Timer timer;

    Receiver(T socket, int other);
    ~Receiver();

    T get_socket()
    {
        return socket;
    }

    void request(octetStream& os);
    void wait(octetStream& os);
};

#endif /* NETWORKING_RECEIVER_H_ */