1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
|
//===-- 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__
|
