jsocket.hpp 26 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667
  1. /*##############################################################################
  2. HPCC SYSTEMS software Copyright (C) 2012 HPCC Systems®.
  3. Licensed under the Apache License, Version 2.0 (the "License");
  4. you may not use this file except in compliance with the License.
  5. You may obtain a copy of the License at
  6. http://www.apache.org/licenses/LICENSE-2.0
  7. Unless required by applicable law or agreed to in writing, software
  8. distributed under the License is distributed on an "AS IS" BASIS,
  9. WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  10. See the License for the specific language governing permissions and
  11. limitations under the License.
  12. ############################################################################## */
  13. // Socket abstraction
  14. #ifndef __JSOCKIO_H__
  15. #define __JSOCKIO_H__
  16. #ifndef _VER_C5
  17. #include <time.h>
  18. #endif
  19. #include "jiface.hpp"
  20. #include "jexcept.hpp"
  21. #include "jthread.hpp"
  22. #if 0 // spurious mplink closed tracing info
  23. # define _TRACELINKCLOSED
  24. #endif
  25. #define DEFAULT_LISTEN_QUEUE_SIZE 200 // maximum for windows 2000 server
  26. #define DEFAULT_LINGER_TIME 1000 // seconds
  27. #ifndef WAIT_FOREVER
  28. #define WAIT_FOREVER ((unsigned)-1)
  29. #endif
  30. enum JSOCKET_ERROR_CODES {
  31. JSOCKERR_ok = 0,
  32. JSOCKERR_not_opened = -1, // accept,name,peer_name,read,write
  33. JSOCKERR_bad_address = -2, // connect
  34. JSOCKERR_connection_failed = -3, // connect
  35. JSOCKERR_broken_pipe = -4, // read,write
  36. JSOCKERR_invalid_access_mode = -5, // accept
  37. JSOCKERR_timeout_expired = -6, // read
  38. JSOCKERR_port_in_use = -7, // create
  39. JSOCKERR_cancel_accept = -8, // accept
  40. JSOCKERR_connectionless_socket = -9, // accept, cancel_accept
  41. JSOCKERR_graceful_close = -10, // read,send
  42. JSOCKERR_handle_too_large = -11, // select, connect etc (linux only)
  43. JSOCKERR_bad_netaddr = -12, // get/set net address
  44. JSOCKERR_ipv6_not_implemented = -13 // various
  45. };
  46. // Block operation flags
  47. #define BF_ASYNC_TRANSFER 0 // send_block sends immediately (default)
  48. #define BF_SYNC_TRANSFER_PULL 1 // send_block waits until receiver ready (i.e. receives first)
  49. #define BF_LZW_COMPRESS 2 // compress using LZW compression
  50. #define BF_REC_COMPRESS 4 // compress using record difference compression
  51. #define BF_RELIABLE_TRANSFER 8 // retries on socket failure
  52. #define BF_SYNC_TRANSFER_PUSH 16 // send_block pushes that has data (i.e. sends first)
  53. // shutdown options
  54. #define SHUTDOWN_READ 0
  55. #define SHUTDOWN_WRITE 1
  56. #define SHUTDOWN_READWRITE 2
  57. //
  58. // Abstract socket interface
  59. //
  60. class jlib_decl IpAddress
  61. {
  62. unsigned netaddr[4] = { 0, 0, 0, 0 };
  63. public:
  64. IpAddress() = default;
  65. IpAddress(const IpAddress& other) { ipset(other); }
  66. explicit IpAddress(const char *text) { ipset(text); }
  67. bool ipset(const char *text); // sets to NULL if fails or text=NULL
  68. void ipset(const IpAddress& other) { memcpy(&netaddr,&other.netaddr,sizeof(netaddr)); }
  69. bool ipequals(const IpAddress & other) const;
  70. int ipcompare(const IpAddress & other) const; // depreciated
  71. unsigned iphash(unsigned prev=0) const;
  72. unsigned fasthash() const;
  73. bool isNull() const; // is null
  74. bool isHost() const; // is primary host NIC ip
  75. bool isLoopBack() const; // is loopback (localhost: 127.0.0.1 or ::1)
  76. bool isLocal() const; // matches local interface
  77. bool isIp4() const;
  78. StringBuffer &getIpText(StringBuffer & out) const;
  79. void ipserialize(MemoryBuffer & out) const;
  80. void ipdeserialize(MemoryBuffer & in);
  81. unsigned ipdistance(const IpAddress &ip,unsigned offset=0) const; // network order distance (offset: 0-3 word (leat sig.), 0=Ipv4)
  82. bool ipincrement(unsigned count,byte minoctet=0,byte maxoctet=255,unsigned short minipv6piece=0,unsigned maxipv6piece=0xffff);
  83. unsigned ipsetrange( const char *text); // e.g. 10.173.72.1-65 ('-' may be omitted)
  84. // returns number in range (use ipincrement to iterate through)
  85. size32_t getNetAddress(size32_t maxsz,void *dst) const; // for internal use - returns 0 if address doesn't fit
  86. void setNetAddress(size32_t sz,const void *src); // for internal use
  87. inline bool operator == ( const IpAddress & other) const { return ipequals(other); }
  88. inline IpAddress & operator = ( const IpAddress &other )
  89. {
  90. ipset(other);
  91. return *this;
  92. }
  93. };
  94. struct IpComparator
  95. {
  96. bool operator()(const IpAddress &a, const IpAddress &b) const
  97. {
  98. // return true if the first argument goes before the second argument, and false otherwise
  99. return a.ipcompare(b) < 0;
  100. }
  101. };
  102. class jlib_decl IpAddressArray : public StructArrayOf<IpAddress>
  103. {
  104. public:
  105. StringBuffer &getText(StringBuffer &text);
  106. void fromText(const char *s,unsigned defport);
  107. };
  108. extern jlib_decl IpAddress & queryHostIP();
  109. extern jlib_decl IpAddress & queryLocalIP();
  110. extern jlib_decl const char * GetCachedHostName();
  111. inline StringBuffer & GetHostName(StringBuffer &str) { return str.append(GetCachedHostName()); }
  112. extern jlib_decl IpAddress &GetHostIp(IpAddress &ip);
  113. extern jlib_decl IpAddress &localHostToNIC(IpAddress &ip);
  114. class jlib_decl SocketEndpoint : extends IpAddress
  115. {
  116. public:
  117. SocketEndpoint() = default;
  118. SocketEndpoint(const char *name,unsigned short _port=0) { set(name,_port); };
  119. SocketEndpoint(unsigned short _port) { setLocalHost(_port); };
  120. SocketEndpoint(unsigned short _port, const IpAddress & _ip) { set(_port,_ip); };
  121. SocketEndpoint(const SocketEndpoint &other) = default;
  122. void deserialize(MemoryBuffer & in);
  123. void serialize(MemoryBuffer & out) const;
  124. bool set(const char *name,unsigned short _port=0);
  125. inline void set(const SocketEndpoint & value) { ipset(value); port = value.port; }
  126. inline void setLocalHost(unsigned short _port) { port = _port; GetHostIp(*this); } // NB *not* localhost(127.0.0.1)
  127. inline void set(unsigned short _port, const IpAddress & _ip) { ipset(_ip); port = _port; };
  128. inline bool equals(const SocketEndpoint &ep) const { return ((port==ep.port)&&ipequals(ep)); }
  129. void getUrlStr(char * str, size32_t len) const; // in form ip4:port or [ip6]:port
  130. StringBuffer &getUrlStr(StringBuffer &str) const; // in form ip4:port or [ip6]:port
  131. inline SocketEndpoint & operator = ( const SocketEndpoint &other )
  132. {
  133. ipset(other);
  134. port = other.port;
  135. return *this;
  136. }
  137. bool operator == (const SocketEndpoint &other) const { return equals(other); }
  138. bool operator != (const SocketEndpoint &other) const { return !equals(other); }
  139. unsigned hash(unsigned prev) const;
  140. unsigned short port = 0;
  141. // Ensure that all the bytes in the data structure are initialised to avoid complains from valgrind when it is written to a socket
  142. unsigned short portPadding = 0;
  143. };
  144. class jlib_decl SocketEndpointArray : public StructArrayOf<SocketEndpoint>
  145. {
  146. public:
  147. StringBuffer &getText(StringBuffer &text);
  148. void fromText(const char *s,unsigned defport);
  149. };
  150. interface ISocketEndpointHashTable: implements IInterface
  151. {
  152. virtual void add(const SocketEndpoint &ep, IInterface *i)=0; // takes ownership
  153. virtual void remove(const SocketEndpoint &ep)=0; // releases
  154. virtual IInterface *find(const SocketEndpoint &ep)=0; // does not link
  155. };
  156. extern jlib_decl ISocketEndpointHashTable *createSocketEndpointHashTable();
  157. class jlib_decl IpSubNet
  158. {
  159. unsigned net[4];
  160. unsigned mask[4];
  161. public:
  162. IpSubNet() {set(NULL,NULL); }
  163. IpSubNet(const char *_net,const char *_mask) { set(_net,_mask); }
  164. bool set(const char *_net,const char *_mask); // _net NULL means match everything
  165. // _mask NULL means match exact
  166. bool test(const IpAddress &ip) const;
  167. StringBuffer &getNetText(StringBuffer &text) const;
  168. StringBuffer &getMaskText(StringBuffer &text) const;
  169. bool isNull() const;
  170. bool operator==(IpSubNet const &other) const
  171. {
  172. if ((0 == memcmp(net, other.net, sizeof(net))) && (0 == memcmp(mask, other.mask, sizeof(mask))))
  173. return true;
  174. return false;
  175. }
  176. };
  177. class jlib_decl ISocket : extends IInterface
  178. {
  179. public:
  180. //
  181. // Create client socket connected to a TCP server socket
  182. static ISocket* connect( const SocketEndpoint &ep );
  183. // general connect
  184. static ISocket* connect_timeout( const SocketEndpoint &ep , unsigned timeout);
  185. // connect where should must take longer than timeout (in ms) to connect
  186. static ISocket* connect_wait( const SocketEndpoint &ep, unsigned timems);
  187. // connect where should try connecting for *at least* time specified
  188. // (e.g. if don't know that server listening yet)
  189. // if 0 specified for time then does single (blocking) connect try
  190. // Create client socket connected to a UDP server socket
  191. //
  192. static ISocket* udp_connect( unsigned short port, char const* host);
  193. static ISocket* udp_connect( const SocketEndpoint &ep);
  194. //
  195. // Create server TCP socket
  196. //
  197. static ISocket* create( unsigned short port,
  198. int listen_queue_size = DEFAULT_LISTEN_QUEUE_SIZE);
  199. //
  200. // Create server TCP socket listening a specific IP
  201. //
  202. static ISocket* create_ip( unsigned short port,
  203. const char *host,
  204. int listen_queue_size = DEFAULT_LISTEN_QUEUE_SIZE);
  205. //
  206. // Create server UDP socket
  207. //
  208. static ISocket* udp_create( unsigned short port);
  209. // Create client socket connected to a multicast server socket
  210. //
  211. static ISocket* multicast_connect( unsigned short port, const char *mcgroupip, unsigned _ttl);
  212. static ISocket* multicast_connect( const SocketEndpoint &ep, unsigned _ttl);
  213. //
  214. // Create server multicast socket
  215. //
  216. static ISocket* multicast_create( unsigned short port, const char *mcgroupip, unsigned _ttl);
  217. static ISocket* multicast_create( unsigned short port, const IpAddress &mcgroupip, unsigned _ttl);
  218. //
  219. // Creates an ISocket for an already created socket
  220. //
  221. static ISocket* attach(int s,bool tcpip=true);
  222. virtual void read(void* buf, size32_t min_size, size32_t max_size, size32_t &size_read,
  223. unsigned timeoutsecs = WAIT_FOREVER) = 0;
  224. virtual void readtms(void* buf, size32_t min_size, size32_t max_size, size32_t &size_read,
  225. unsigned timeout) = 0;
  226. virtual void read(void* buf, size32_t size) = 0;
  227. virtual size32_t write(void const* buf, size32_t size) = 0; // returns amount written normally same as in size (see set_nonblock)
  228. virtual size32_t writetms(void const* buf, size32_t size, unsigned timeoutms=WAIT_FOREVER) = 0;
  229. virtual size32_t get_max_send_size() = 0;
  230. //
  231. // This method is called by server to accept client connection
  232. //
  233. virtual ISocket* accept(bool allowcancel=false, SocketEndpoint *peerEp = nullptr) = 0; // not needed for UDP
  234. //
  235. // log poll() errors
  236. //
  237. virtual int logPollError(unsigned revents, const char *rwstr) = 0;
  238. //
  239. // This method is called to check whether a socket has data ready
  240. //
  241. virtual int wait_read(unsigned timeout) = 0;
  242. //
  243. // This method is called to check whether a socket is ready to write (i.e. some free buffer space)
  244. //
  245. virtual int wait_write(unsigned timeout) = 0;
  246. //
  247. // can be used with write to allow it to return if it would block
  248. // be sure and restore to old state before calling other functions on this socket
  249. //
  250. virtual bool set_nonblock(bool on) = 0; // returns old state
  251. // enable 'nagling' - small packet coalescing (implies delayed transmission)
  252. //
  253. virtual bool set_nagle(bool on) = 0; // returns old state
  254. // set 'linger' time - time close will linger so that outstanding unsent data will be transmitted
  255. //
  256. virtual void set_linger(int lingersecs) = 0;
  257. //
  258. // Cancel accept operation and close socket
  259. //
  260. virtual void cancel_accept() = 0; // not needed for UDP
  261. //
  262. // Shutdown socket: prohibit write and/or read operations on socket
  263. //
  264. virtual void shutdown(unsigned mode=SHUTDOWN_READWRITE) = 0; // not needed for UDP
  265. // Get local name of accepted (or connected) socket and returns port
  266. virtual int name(char *name,size32_t namemax)=0;
  267. // Get peer name of socket and returns port - in UDP returns return addr
  268. virtual int peer_name(char *name,size32_t namemax)=0;
  269. // Get peer endpoint of socket - in UDP returns return addr
  270. virtual SocketEndpoint &getPeerEndpoint(SocketEndpoint &ep)=0;
  271. // Get peer ip of socket - in UDP returns return addr
  272. virtual IpAddress &getPeerAddress(IpAddress &addr)=0;
  273. // Get local endpoint of socket
  274. virtual SocketEndpoint &getEndpoint(SocketEndpoint &ep) const = 0;
  275. //
  276. // Close socket
  277. //
  278. virtual bool connectionless()=0; // true if accept need not be called (i.e. UDP)
  279. virtual void set_return_addr(int port,const char *name) = 0; // used for UDP servers only
  280. // Block functions
  281. virtual void set_block_mode ( // must be called before block operations
  282. unsigned flags, // BF_* flags (must match receive_block)
  283. size32_t recsize=0, // record size (required for rec compression)
  284. unsigned timeoutms=0 // timeout in milisecs (0 for no timeout)
  285. )=0;
  286. virtual bool send_block(
  287. const void *blk, // data to send
  288. size32_t sz // size to send (0 for eof)
  289. )=0;
  290. virtual size32_t receive_block_size ()=0; // get size of next block (always must call receive_block after)
  291. virtual size32_t receive_block(
  292. void *blk, // receive pointer
  293. size32_t sz // max size to read (0 for sync eof)
  294. // if less than block size truncates block
  295. )=0;
  296. virtual void close() = 0;
  297. virtual unsigned OShandle() = 0; // for internal use
  298. virtual size32_t avail_read() = 0; // called after wait_read to see how much data available
  299. virtual size32_t write_multiple(unsigned num,void const**buf, size32_t *size) = 0; // same as write except writes multiple blocks
  300. virtual size32_t get_send_buffer_size() =0; // get OS send buffer
  301. virtual void set_send_buffer_size(size32_t sz) =0; // set OS send buffer size
  302. virtual bool join_multicast_group(SocketEndpoint &ep) = 0; // for udp multicast
  303. virtual bool leave_multicast_group(SocketEndpoint &ep) = 0; // for udp multicast
  304. virtual void set_ttl(unsigned _ttl) = 0; // set TTL
  305. virtual size32_t get_receive_buffer_size() = 0; // get OS receive buffer
  306. virtual void set_receive_buffer_size(size32_t sz) = 0; // set OS receive buffer size
  307. virtual void set_keep_alive(bool set) = 0; // set option SO_KEEPALIVE
  308. virtual size32_t udp_write_to(const SocketEndpoint &ep,void const* buf, size32_t size) = 0;
  309. virtual bool check_connection() = 0;
  310. virtual bool isSecure() const = 0;
  311. /*
  312. Exceptions raised: (when set_raise_exceptions(TRUE))
  313. create
  314. sys:(socket, bind, listen)
  315. udp_create
  316. sys:(socket, bind, listen)
  317. accept
  318. JSOCKERR_not_opened, sys:(accept,setsockopt), JSOCKERR_invalid_access_mode, JSOCKERR_cancel_accept, JSOCKERR_connectionless_socket
  319. name
  320. JSOCKERR_not_opened, sys:(getsockname)
  321. peer_name
  322. JSOCKERR_not_opened, sys:(getpeername)
  323. cancel_accept
  324. {connect}, sys:(gethostname), JSOCKERR_connectionless_socket
  325. connect
  326. JSOCKERR_bad_address, JSOCKERR_connection_failed, sys:(socket, connect, setsockopt)
  327. udp_connect
  328. JSOCKERR_bad_address, sys:(socket, connect, setsockopt)
  329. read (timeout)
  330. JSOCKERR_not_opened, JSOCKERR_broken_pipe, JSOCKERR_timeout_expired ,sys:(select, read), JSOCKERR_graceful_close
  331. read (no timeout)
  332. JSOCKERR_not_opened, JSOCKERR_broken_pipe, sys:(read), JSOCKERR_graceful_close
  333. write
  334. JSOCKERR_not_opened, JSOCKERR_broken_pipe, sys:(write), JSOCKERR_graceful_close
  335. close
  336. sys:(write)
  337. shutdown
  338. sys:(shutdown),JSOCKERR_broken_pipe
  339. */
  340. };
  341. interface jlib_thrown_decl IJSOCK_Exception: extends IException
  342. {
  343. };
  344. extern jlib_decl IJSOCK_Exception *IPv6NotImplementedException(const char *filename,unsigned lineno);
  345. #define IPV6_NOT_IMPLEMENTED() throw IPv6NotImplementedException(sanitizeSourceFile(__FILE__), __LINE__)
  346. //---------------------------------------------------------------------------
  347. // These classes are useful for compressing a list of ip:ports to pass around.
  348. class jlib_decl SocketListCreator
  349. {
  350. public:
  351. SocketListCreator();
  352. void addSocket(const SocketEndpoint &ep);
  353. void addSocket(const char * ip, unsigned port);
  354. const char * getText();
  355. void addSockets(SocketEndpointArray &array);
  356. protected:
  357. StringBuffer fullText;
  358. StringAttr lastIp;
  359. unsigned lastPort;
  360. };
  361. class jlib_decl SocketListParser
  362. // This class depreciated - new code should use SocketEndpointArray::fromText and getText
  363. {
  364. public:
  365. SocketListParser(const char * text);
  366. void first(unsigned defport=0);
  367. bool get(StringAttr & ip, unsigned & port, unsigned index, unsigned defport=0); // alternative to iterating..
  368. bool next(StringAttr & ip, unsigned & port);
  369. unsigned getSockets(SocketEndpointArray &array,unsigned defport=0);
  370. protected:
  371. StringAttr fullText;
  372. StringAttr lastIp;
  373. const char * cursor;
  374. unsigned lastPort;
  375. };
  376. struct JSocketStatistics
  377. {
  378. unsigned connects; // successful
  379. unsigned connecttime; // all times in microsecs
  380. unsigned failedconnects;
  381. unsigned failedconnecttime;
  382. unsigned reads;
  383. unsigned readtime;
  384. __int64 readsize; // all sizes in bytes
  385. unsigned writes;
  386. unsigned writetime;
  387. __int64 writesize;
  388. unsigned activesockets;
  389. unsigned numblockrecvs;
  390. unsigned numblocksends;
  391. __int64 blockrecvsize;
  392. __int64 blocksendsize;
  393. unsigned blockrecvtime; // not including initial handshake
  394. unsigned blocksendtime;
  395. unsigned longestblocksend;
  396. unsigned longestblocksize;
  397. };
  398. extern jlib_decl void getSocketStatistics(JSocketStatistics &stats);
  399. extern jlib_decl void resetSocketStatistics();
  400. extern jlib_decl StringBuffer &getSocketStatisticsString(JSocketStatistics &stats,StringBuffer &buf);
  401. // Select Thread
  402. #define SELECTMODE_READ 1
  403. #define SELECTMODE_WRITE 2
  404. #define SELECTMODE_EXCEPT 4
  405. interface ISocketSelectNotify: extends IInterface
  406. {
  407. virtual bool notifySelected(ISocket *sock,unsigned selected)=0; // return false to continue to next selected, true to re-select
  408. };
  409. interface ISocketSelectHandler: extends IInterface
  410. {
  411. public:
  412. virtual void start()=0;
  413. virtual void add(ISocket *sock,unsigned mode,ISocketSelectNotify *nfy)=0;
  414. virtual void remove(ISocket *sock)=0;
  415. virtual void stop(bool wait)=0;
  416. };
  417. extern jlib_decl ISocketSelectHandler *createSocketSelectHandler(const char *trc=NULL, unsigned hdlPerThrd=0);
  418. extern jlib_decl ISocketSelectHandler *createSocketEpollHandler(const char *trc=NULL, unsigned hdlPerThrd=0);
  419. class MemoryBuffer;
  420. // sends/receives length as well as contents.
  421. extern jlib_decl void readBuffer(ISocket * socket, MemoryBuffer & buffer);
  422. extern jlib_decl void readBuffer(ISocket * socket, MemoryBuffer & buffer, unsigned timeoutms);
  423. extern jlib_decl void writeBuffer(ISocket * socket, MemoryBuffer & buffer);
  424. // ditto but catches any exceptions
  425. extern jlib_decl bool catchReadBuffer(ISocket * socket, MemoryBuffer & buffer);
  426. extern jlib_decl bool catchReadBuffer(ISocket * socket, MemoryBuffer & buffer, unsigned timeoutms);
  427. extern jlib_decl bool catchWriteBuffer(ISocket * socket, MemoryBuffer & buffer);
  428. // utility interface for simple conversations
  429. // conversation is always between two ends,
  430. // at any given time one end must be receiving and other sending (though these may swap during the conversation)
  431. interface IConversation: extends IInterface
  432. {
  433. virtual bool accept(unsigned timeoutms)=0; // one side accepts
  434. virtual void set_keep_alive(bool keepalive)=0; // enable keepalive for socket
  435. virtual bool connect(unsigned timeoutms)=0; // other side connects
  436. virtual bool send(MemoryBuffer &mb)=0; // 0 length buffer can be sent
  437. virtual bool recv(MemoryBuffer &mb, unsigned timeoutms)=0; // up to protocol to terminate conversation (e.g. by zero length buffer)
  438. virtual void cancel()=0; // cancels above methods (from separate thread)
  439. virtual unsigned short setRandomPort(unsigned short base, unsigned num)=0; // sets a random unique port for accept use
  440. };
  441. extern jlib_decl IConversation *createSingletonSocketConnection(unsigned short port,SocketEndpoint *ep=NULL);
  442. // the end that listens may omit ep
  443. // this function does not connect so raises no socket exceptions
  444. // interface for reading from multiple sockets using the BF_SYNC_TRANSFER_PUSH protocol
  445. interface ISocketBufferReader: extends IInterface
  446. {
  447. public:
  448. virtual void init(unsigned num,ISocket **sockets,size32_t buffermax=(unsigned)-1)=0;
  449. virtual unsigned get(MemoryBuffer &mb)=0;
  450. virtual void done(bool wait)=0;
  451. };
  452. extern jlib_decl ISocketBufferReader *createSocketBufferReader(const char *trc=NULL);
  453. extern jlib_decl void markNodeCentral(SocketEndpoint &ep); // random delay for linux
  454. interface ISocketConnectNotify
  455. {
  456. public:
  457. virtual void connected(unsigned idx,const SocketEndpoint &ep,ISocket *socket)=0; // must link socket if kept
  458. virtual void failed(unsigned idx,const SocketEndpoint &ep,int err)=0;
  459. };
  460. extern jlib_decl void multiConnect(const SocketEndpointArray &eps,ISocketConnectNotify &inotify,unsigned timeout);
  461. extern jlib_decl void multiConnect(const SocketEndpointArray &eps,IPointerArrayOf<ISocket> &retsockets,unsigned timeout);
  462. interface ISocketConnectWait: extends IInterface
  463. {
  464. public:
  465. virtual ISocket *wait(unsigned waittimems)=0; // return NULL if time expired, throws exception if connect failed
  466. // releasing ISocketConnectWait cancels the connect iff wait has never returned socket
  467. };
  468. extern jlib_decl ISocketConnectWait *nonBlockingConnect(SocketEndpoint &ep,unsigned connectimeoutms=0);
  469. // buffered socket
  470. interface IBufferedSocket : implements IInterface
  471. {
  472. virtual int readline(char* buf, int maxlen, IMultiException *me) = 0;
  473. virtual int read(char* buf, int maxlen) = 0;
  474. virtual int readline(char* buf, int maxlen, bool keepcrlf, IMultiException *me) = 0;
  475. virtual void setReadTimeout(unsigned int timeout) = 0;
  476. };
  477. #define BSOCKET_READ_TIMEOUT 600
  478. #define BSOCKET_CLIENT_READ_TIMEOUT 7200
  479. extern jlib_decl IBufferedSocket* createBufferedSocket(ISocket* socket);
  480. #define MAX_NET_ADDRESS_SIZE (16)
  481. extern jlib_decl IpSubNet &queryPreferredSubnet(); // preferred subnet when resolving multiple NICs
  482. extern jlib_decl bool setPreferredSubnet(const char *ip,const char *mask); // also resets cached host IP
  483. extern jlib_decl StringBuffer &lookupHostName(const IpAddress &ip,StringBuffer &ret);
  484. extern jlib_decl bool isInterfaceIp(const IpAddress &ip, const char *ifname);
  485. extern jlib_decl bool getInterfaceIp(IpAddress &ip, const char *ifname);
  486. extern jlib_decl bool getInterfaceName(StringBuffer &ifname);
  487. //Given a list of server sockets, wait until any one or more are ready to be read/written (wont block)
  488. //return array of ready sockets
  489. extern jlib_decl int wait_read_multiple(UnsignedArray &socks, //IN sockets to be checked for read readiness
  490. unsigned timeoutMS, //IN timeout
  491. UnsignedArray &readySocks);//OUT sockets ready to be read
  492. extern jlib_decl int wait_write_multiple(UnsignedArray &socks, //IN sockets to be checked for write readiness
  493. unsigned timeoutMS, //IN timeout
  494. UnsignedArray &readySocks);//OUT sockets ready to be written
  495. extern jlib_decl void throwJSocketException(int jsockErr);
  496. extern jlib_decl IJSOCK_Exception* createJSocketException(int jsockErr, const char *_msg);
  497. extern jlib_decl bool isIPV4(const char *ip);
  498. extern jlib_decl bool isIPV6(const char *ip);
  499. extern jlib_decl bool isIPAddress(const char *ip);
  500. interface IWhiteListHandler : extends IInterface
  501. {
  502. virtual bool isWhiteListed(const char *ip, unsigned __int64 role, StringBuffer *responseText=nullptr) const = 0;
  503. virtual StringBuffer &getWhiteList(StringBuffer &out) const = 0;
  504. virtual void refresh() = 0;
  505. };
  506. interface IWhiteListWriter : extends IInterface
  507. {
  508. virtual void add(const char *ip, unsigned __int64 role) = 0;
  509. virtual void setAllowAnonRoles(bool tf) = 0;
  510. };
  511. typedef std::function<bool(IWhiteListWriter &)> WhiteListPopulateFunction;
  512. typedef std::function<StringBuffer &(StringBuffer &, unsigned __int64)> WhiteListFormatFunction;
  513. extern jlib_decl IWhiteListHandler *createWhiteListHandler(WhiteListPopulateFunction populateFunc, WhiteListFormatFunction roleFormatFunc = {}); // format function optional
  514. #endif