forked from OSchip/llvm-project
86 lines
2.1 KiB
C++
86 lines
2.1 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 "RNBDefs.h"
|
|
#include <sys/socket.h>
|
|
#include <sys/types.h>
|
|
#include <string>
|
|
#include "DNBTimer.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__
|