• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 //===-- RNBSocket.h ---------------------------------------------*- C++ -*-===//
2 //
3 //                     The LLVM Compiler Infrastructure
4 //
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
7 //
8 //===----------------------------------------------------------------------===//
9 //
10 //  Created by Greg Clayton on 12/12/07.
11 //
12 //===----------------------------------------------------------------------===//
13 
14 #ifndef __RNBSocket_h__
15 #define __RNBSocket_h__
16 
17 #include "RNBDefs.h"
18 #include <sys/socket.h>
19 #include <sys/types.h>
20 #include <string>
21 #include "DNBTimer.h"
22 
23 #ifdef WITH_LOCKDOWN
24 #include "lockdown.h"
25 #endif
26 
27 class RNBSocket
28 {
29 public:
30     typedef void (*PortBoundCallback) (const void *baton, in_port_t port);
31 
RNBSocket()32     RNBSocket () :
33         m_fd (-1),
34 #ifdef WITH_LOCKDOWN
35         m_fd_from_lockdown (false),
36         m_ld_conn (),
37 #endif
38         m_timer (true)      // Make a thread safe timer
39     {
40     }
~RNBSocket(void)41     ~RNBSocket (void)
42     {
43         Disconnect (false);
44     }
45 
46     rnb_err_t Listen (const char *listen_host,
47                       in_port_t port,
48                       PortBoundCallback callback,
49                       const void *callback_baton);
50     rnb_err_t Connect (const char *host, uint16_t port);
51 
52     rnb_err_t useFD(int fd);
53 
54 #ifdef WITH_LOCKDOWN
55     rnb_err_t ConnectToService();
56 #endif
57     rnb_err_t OpenFile (const char *path);
58     rnb_err_t Disconnect (bool save_errno);
59     rnb_err_t Read (std::string &p);
60     rnb_err_t Write (const void *buffer, size_t length);
61 
IsConnected()62     bool IsConnected () const { return m_fd != -1; }
63     void SaveErrno (int curr_errno);
Timer()64     DNBTimer& Timer() { return m_timer; }
65 
66     static int SetSocketOption(int fd, int level, int option_name, int option_value);
67 private:
68     // Outlaw some constructors
69     RNBSocket (const RNBSocket &);
70 
71 protected:
72     rnb_err_t ClosePort (int& fd, bool save_errno);
73 
74     int m_fd;    // Socket we use to communicate once conn established
75 
76 #ifdef WITH_LOCKDOWN
77     bool m_fd_from_lockdown;
78     lockdown_connection m_ld_conn;
79 #endif
80 
81     DNBTimer m_timer;
82 };
83 
84 
85 #endif // #ifndef __RNBSocket_h__
86