spanning_tree.h

Go to the documentation of this file.

#pragma once

#ifdef _WIN32
#define NOMINMAX
#include <WinSock2.h>
#include <Windows.h>
#include <WS2tcpip.h>
#include <io.h>

#define CLOSESOCK closesocket
#define inet_ntop InetNtopA

typedef unsigned int uint32_t;
typedef unsigned short uint16_t;
typedef int socklen_t;
typedef SOCKET socket_t;

namespace std
{
// forward declare promise as C++/CLI doesn't allow usage in header files
template <typename T>
class future;
}  // namespace std
#else
#include <unistd.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <netinet/tcp.h>
#include <netdb.h>
#include <strings.h>
#include <arpa/inet.h>

#define CLOSESOCK close

typedef int socket_t;

#include <future>
#endif

namespace VW
{
class SpanningTree
{
 private:
  bool m_stop;
  socket_t sock;
  uint16_t m_port;

  // future to signal end of thread running.
  // Need a pointer since C++/CLI doesn't like futures yet
  std::future<void>* m_future;

  bool m_quiet;

 public:
  SpanningTree(short unsigned int port = 26543, bool quiet = false);
  ~SpanningTree();

  short unsigned int BoundPort();

  void Start();
  void Run();
  void Stop();
};
}  // namespace VW