80 lines
2.0 KiB
C++
80 lines
2.0 KiB
C++
//===-- RNBSocket.h ---------------------------------------------*- C++ -*-===//
|
|
//
|
|
// The LLVM Compiler Infrastructure
|
|
//
|
|
// This file is distributed under the University of Illinois Open Source
|
|
// License. See LICENSE.TXT for details.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// Created by Greg Clayton on 12/12/07.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#ifndef __RNBSocket_h__
|
|
#define __RNBSocket_h__
|
|
|
|
#include "DNBTimer.h"
|
|
#include "RNBDefs.h"
|
|
#include <string>
|
|
#include <sys/socket.h>
|
|
#include <sys/types.h>
|
|
|
|
#ifdef WITH_LOCKDOWN
|
|
#include "lockdown.h"
|
|
#endif
|
|
|
|
class RNBSocket {
|
|
public:
|
|
typedef void (*PortBoundCallback)(const void *baton, uint16_t port);
|
|
|
|
RNBSocket()
|
|
: m_fd(-1),
|
|
#ifdef WITH_LOCKDOWN
|
|
m_fd_from_lockdown(false), m_ld_conn(),
|
|
#endif
|
|
m_timer(true) // Make a thread safe timer
|
|
{
|
|
}
|
|
~RNBSocket(void) { Disconnect(false); }
|
|
|
|
rnb_err_t Listen(const char *listen_host, uint16_t port,
|
|
PortBoundCallback callback, const void *callback_baton);
|
|
rnb_err_t Connect(const char *host, uint16_t port);
|
|
|
|
rnb_err_t useFD(int fd);
|
|
|
|
#ifdef WITH_LOCKDOWN
|
|
rnb_err_t ConnectToService();
|
|
#endif
|
|
rnb_err_t OpenFile(const char *path);
|
|
rnb_err_t Disconnect(bool save_errno);
|
|
rnb_err_t Read(std::string &p);
|
|
rnb_err_t Write(const void *buffer, size_t length);
|
|
|
|
bool IsConnected() const { return m_fd != -1; }
|
|
void SaveErrno(int curr_errno);
|
|
DNBTimer &Timer() { return m_timer; }
|
|
|
|
static int SetSocketOption(int fd, int level, int option_name,
|
|
int option_value);
|
|
|
|
private:
|
|
// Outlaw some constructors
|
|
RNBSocket(const RNBSocket &);
|
|
|
|
protected:
|
|
rnb_err_t ClosePort(int &fd, bool save_errno);
|
|
|
|
int m_fd; // Socket we use to communicate once conn established
|
|
|
|
#ifdef WITH_LOCKDOWN
|
|
bool m_fd_from_lockdown;
|
|
lockdown_connection m_ld_conn;
|
|
#endif
|
|
|
|
DNBTimer m_timer;
|
|
};
|
|
|
|
#endif // #ifndef __RNBSocket_h__
|