jsocket.hpp 26 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668
  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. bool fromName(const char *name, unsigned defport);
  149. void fromText(const char *s,unsigned defport);
  150. };
  151. interface ISocketEndpointHashTable: implements IInterface
  152. {
  153. virtual void add(const SocketEndpoint &ep, IInterface *i)=0; // takes ownership
  154. virtual void remove(const SocketEndpoint &ep)=0; // releases
  155. virtual IInterface *find(const SocketEndpoint &ep)=0; // does not link
  156. };
  157. extern jlib_decl ISocketEndpointHashTable *createSocketEndpointHashTable();
  158. class jlib_decl IpSubNet
  159. {
  160. unsigned net[4];
  161. unsigned mask[4];
  162. public:
  163. IpSubNet() {set(NULL,NULL); }
  164. IpSubNet(const char *_net,const char *_mask) { set(_net,_mask); }
  165. bool set(const char *_net,const char *_mask); // _net NULL means match everything
  166. // _mask NULL means match exact
  167. bool test(const IpAddress &ip) const;
  168. StringBuffer &getNetText(StringBuffer &text) const;
  169. StringBuffer &getMaskText(StringBuffer &text) const;
  170. bool isNull() const;
  171. bool operator==(IpSubNet const &other) const
  172. {
  173. if ((0 == memcmp(net, other.net, sizeof(net))) && (0 == memcmp(mask, other.mask, sizeof(mask))))
  174. return true;
  175. return false;
  176. }
  177. };
  178. class jlib_decl ISocket : extends IInterface
  179. {
  180. public:
  181. //
  182. // Create client socket connected to a TCP server socket
  183. static ISocket* connect( const SocketEndpoint &ep );
  184. // general connect
  185. static ISocket* connect_timeout( const SocketEndpoint &ep , unsigned timeout);
  186. // connect where should must take longer than timeout (in ms) to connect
  187. static ISocket* connect_wait( const SocketEndpoint &ep, unsigned timems);
  188. // connect where should try connecting for *at least* time specified
  189. // (e.g. if don't know that server listening yet)
  190. // if 0 specified for time then does single (blocking) connect try
  191. // Create client socket connected to a UDP server socket
  192. //
  193. static ISocket* udp_connect( unsigned short port, char const* host);
  194. static ISocket* udp_connect( const SocketEndpoint &ep);
  195. //
  196. // Create server TCP socket
  197. //
  198. static ISocket* create( unsigned short port,
  199. int listen_queue_size = DEFAULT_LISTEN_QUEUE_SIZE);
  200. //
  201. // Create server TCP socket listening a specific IP
  202. //
  203. static ISocket* create_ip( unsigned short port,
  204. const char *host,
  205. int listen_queue_size = DEFAULT_LISTEN_QUEUE_SIZE);
  206. //
  207. // Create server UDP socket
  208. //
  209. static ISocket* udp_create( unsigned short port);
  210. // Create client socket connected to a multicast server socket
  211. //
  212. static ISocket* multicast_connect( unsigned short port, const char *mcgroupip, unsigned _ttl);
  213. static ISocket* multicast_connect( const SocketEndpoint &ep, unsigned _ttl);
  214. //
  215. // Create server multicast socket
  216. //
  217. static ISocket* multicast_create( unsigned short port, const char *mcgroupip, unsigned _ttl);
  218. static ISocket* multicast_create( unsigned short port, const IpAddress &mcgroupip, unsigned _ttl);
  219. //
  220. // Creates an ISocket for an already created socket
  221. //
  222. static ISocket* attach(int s,bool tcpip=true);
  223. virtual void read(void* buf, size32_t min_size, size32_t max_size, size32_t &size_read,
  224. unsigned timeoutsecs = WAIT_FOREVER) = 0;
  225. virtual void readtms(void* buf, size32_t min_size, size32_t max_size, size32_t &size_read,
  226. unsigned timeout) = 0;
  227. virtual void read(void* buf, size32_t size) = 0;
  228. virtual size32_t write(void const* buf, size32_t size) = 0; // returns amount written normally same as in size (see set_nonblock)
  229. virtual size32_t writetms(void const* buf, size32_t size, unsigned timeoutms=WAIT_FOREVER) = 0;
  230. virtual size32_t get_max_send_size() = 0;
  231. //
  232. // This method is called by server to accept client connection
  233. //
  234. virtual ISocket* accept(bool allowcancel=false, SocketEndpoint *peerEp = nullptr) = 0; // not needed for UDP
  235. //
  236. // log poll() errors
  237. //
  238. virtual int logPollError(unsigned revents, const char *rwstr) = 0;
  239. //
  240. // This method is called to check whether a socket has data ready
  241. //
  242. virtual int wait_read(unsigned timeout) = 0;
  243. //
  244. // This method is called to check whether a socket is ready to write (i.e. some free buffer space)
  245. //
  246. virtual int wait_write(unsigned timeout) = 0;
  247. //
  248. // can be used with write to allow it to return if it would block
  249. // be sure and restore to old state before calling other functions on this socket
  250. //
  251. virtual bool set_nonblock(bool on) = 0; // returns old state
  252. // enable 'nagling' - small packet coalescing (implies delayed transmission)
  253. //
  254. virtual bool set_nagle(bool on) = 0; // returns old state
  255. // set 'linger' time - time close will linger so that outstanding unsent data will be transmitted
  256. //
  257. virtual void set_linger(int lingersecs) = 0;
  258. //
  259. // Cancel accept operation and close socket
  260. //
  261. virtual void cancel_accept() = 0; // not needed for UDP
  262. //
  263. // Shutdown socket: prohibit write and/or read operations on socket
  264. //
  265. virtual void shutdown(unsigned mode=SHUTDOWN_READWRITE) = 0; // not needed for UDP
  266. // Get local name of accepted (or connected) socket and returns port
  267. virtual int name(char *name,size32_t namemax)=0;
  268. // Get peer name of socket and returns port - in UDP returns return addr
  269. virtual int peer_name(char *name,size32_t namemax)=0;
  270. // Get peer endpoint of socket - in UDP returns return addr
  271. virtual SocketEndpoint &getPeerEndpoint(SocketEndpoint &ep)=0;
  272. // Get peer ip of socket - in UDP returns return addr
  273. virtual IpAddress &getPeerAddress(IpAddress &addr)=0;
  274. // Get local endpoint of socket
  275. virtual SocketEndpoint &getEndpoint(SocketEndpoint &ep) const = 0;
  276. //
  277. // Close socket
  278. //
  279. virtual bool connectionless()=0; // true if accept need not be called (i.e. UDP)
  280. virtual void set_return_addr(int port,const char *name) = 0; // used for UDP servers only
  281. // Block functions
  282. virtual void set_block_mode ( // must be called before block operations
  283. unsigned flags, // BF_* flags (must match receive_block)
  284. size32_t recsize=0, // record size (required for rec compression)
  285. unsigned timeoutms=0 // timeout in milisecs (0 for no timeout)
  286. )=0;
  287. virtual bool send_block(
  288. const void *blk, // data to send
  289. size32_t sz // size to send (0 for eof)
  290. )=0;
  291. virtual size32_t receive_block_size ()=0; // get size of next block (always must call receive_block after)
  292. virtual size32_t receive_block(
  293. void *blk, // receive pointer
  294. size32_t sz // max size to read (0 for sync eof)
  295. // if less than block size truncates block
  296. )=0;
  297. virtual void close() = 0;
  298. virtual unsigned OShandle() = 0; // for internal use
  299. virtual size32_t avail_read() = 0; // called after wait_read to see how much data available
  300. virtual size32_t write_multiple(unsigned num,void const**buf, size32_t *size) = 0; // same as write except writes multiple blocks
  301. virtual size32_t get_send_buffer_size() =0; // get OS send buffer
  302. virtual void set_send_buffer_size(size32_t sz) =0; // set OS send buffer size
  303. virtual bool join_multicast_group(SocketEndpoint &ep) = 0; // for udp multicast
  304. virtual bool leave_multicast_group(SocketEndpoint &ep) = 0; // for udp multicast
  305. virtual void set_ttl(unsigned _ttl) = 0; // set TTL
  306. virtual size32_t get_receive_buffer_size() = 0; // get OS receive buffer
  307. virtual void set_receive_buffer_size(size32_t sz) = 0; // set OS receive buffer size
  308. virtual void set_keep_alive(bool set) = 0; // set option SO_KEEPALIVE
  309. virtual size32_t udp_write_to(const SocketEndpoint &ep,void const* buf, size32_t size) = 0;
  310. virtual bool check_connection() = 0;
  311. virtual bool isSecure() const = 0;
  312. /*
  313. Exceptions raised: (when set_raise_exceptions(TRUE))
  314. create
  315. sys:(socket, bind, listen)
  316. udp_create
  317. sys:(socket, bind, listen)
  318. accept
  319. JSOCKERR_not_opened, sys:(accept,setsockopt), JSOCKERR_invalid_access_mode, JSOCKERR_cancel_accept, JSOCKERR_connectionless_socket
  320. name
  321. JSOCKERR_not_opened, sys:(getsockname)
  322. peer_name
  323. JSOCKERR_not_opened, sys:(getpeername)
  324. cancel_accept
  325. {connect}, sys:(gethostname), JSOCKERR_connectionless_socket
  326. connect
  327. JSOCKERR_bad_address, JSOCKERR_connection_failed, sys:(socket, connect, setsockopt)
  328. udp_connect
  329. JSOCKERR_bad_address, sys:(socket, connect, setsockopt)
  330. read (timeout)
  331. JSOCKERR_not_opened, JSOCKERR_broken_pipe, JSOCKERR_timeout_expired ,sys:(select, read), JSOCKERR_graceful_close
  332. read (no timeout)
  333. JSOCKERR_not_opened, JSOCKERR_broken_pipe, sys:(read), JSOCKERR_graceful_close
  334. write
  335. JSOCKERR_not_opened, JSOCKERR_broken_pipe, sys:(write), JSOCKERR_graceful_close
  336. close
  337. sys:(write)
  338. shutdown
  339. sys:(shutdown),JSOCKERR_broken_pipe
  340. */
  341. };
  342. interface jlib_thrown_decl IJSOCK_Exception: extends IException
  343. {
  344. };
  345. extern jlib_decl IJSOCK_Exception *IPv6NotImplementedException(const char *filename,unsigned lineno);
  346. #define IPV6_NOT_IMPLEMENTED() throw IPv6NotImplementedException(sanitizeSourceFile(__FILE__), __LINE__)
  347. //---------------------------------------------------------------------------
  348. // These classes are useful for compressing a list of ip:ports to pass around.
  349. class jlib_decl SocketListCreator
  350. {
  351. public:
  352. SocketListCreator();
  353. void addSocket(const SocketEndpoint &ep);
  354. void addSocket(const char * ip, unsigned port);
  355. const char * getText();
  356. void addSockets(SocketEndpointArray &array);
  357. protected:
  358. StringBuffer fullText;
  359. StringAttr lastIp;
  360. unsigned lastPort;
  361. };
  362. class jlib_decl SocketListParser
  363. // This class depreciated - new code should use SocketEndpointArray::fromText and getText
  364. {
  365. public:
  366. SocketListParser(const char * text);
  367. void first(unsigned defport=0);
  368. bool get(StringAttr & ip, unsigned & port, unsigned index, unsigned defport=0); // alternative to iterating..
  369. bool next(StringAttr & ip, unsigned & port);
  370. unsigned getSockets(SocketEndpointArray &array,unsigned defport=0);
  371. protected:
  372. StringAttr fullText;
  373. StringAttr lastIp;
  374. const char * cursor;
  375. unsigned lastPort;
  376. };
  377. struct JSocketStatistics
  378. {
  379. unsigned connects; // successful
  380. unsigned connecttime; // all times in microsecs
  381. unsigned failedconnects;
  382. unsigned failedconnecttime;
  383. unsigned reads;
  384. unsigned readtime;
  385. __int64 readsize; // all sizes in bytes
  386. unsigned writes;
  387. unsigned writetime;
  388. __int64 writesize;
  389. unsigned activesockets;
  390. unsigned numblockrecvs;
  391. unsigned numblocksends;
  392. __int64 blockrecvsize;
  393. __int64 blocksendsize;
  394. unsigned blockrecvtime; // not including initial handshake
  395. unsigned blocksendtime;
  396. unsigned longestblocksend;
  397. unsigned longestblocksize;
  398. };
  399. extern jlib_decl void getSocketStatistics(JSocketStatistics &stats);
  400. extern jlib_decl void resetSocketStatistics();
  401. extern jlib_decl StringBuffer &getSocketStatisticsString(JSocketStatistics &stats,StringBuffer &buf);
  402. // Select Thread
  403. #define SELECTMODE_READ 1
  404. #define SELECTMODE_WRITE 2
  405. #define SELECTMODE_EXCEPT 4
  406. interface ISocketSelectNotify: extends IInterface
  407. {
  408. virtual bool notifySelected(ISocket *sock,unsigned selected)=0; // return false to continue to next selected, true to re-select
  409. };
  410. interface ISocketSelectHandler: extends IInterface
  411. {
  412. public:
  413. virtual void start()=0;
  414. virtual void add(ISocket *sock,unsigned mode,ISocketSelectNotify *nfy)=0;
  415. virtual void remove(ISocket *sock)=0;
  416. virtual void stop(bool wait)=0;
  417. };
  418. extern jlib_decl ISocketSelectHandler *createSocketSelectHandler(const char *trc=NULL, unsigned hdlPerThrd=0);
  419. extern jlib_decl ISocketSelectHandler *createSocketEpollHandler(const char *trc=NULL, unsigned hdlPerThrd=0);
  420. class MemoryBuffer;
  421. // sends/receives length as well as contents.
  422. extern jlib_decl void readBuffer(ISocket * socket, MemoryBuffer & buffer);
  423. extern jlib_decl void readBuffer(ISocket * socket, MemoryBuffer & buffer, unsigned timeoutms);
  424. extern jlib_decl void writeBuffer(ISocket * socket, MemoryBuffer & buffer);
  425. // ditto but catches any exceptions
  426. extern jlib_decl bool catchReadBuffer(ISocket * socket, MemoryBuffer & buffer);
  427. extern jlib_decl bool catchReadBuffer(ISocket * socket, MemoryBuffer & buffer, unsigned timeoutms);
  428. extern jlib_decl bool catchWriteBuffer(ISocket * socket, MemoryBuffer & buffer);
  429. // utility interface for simple conversations
  430. // conversation is always between two ends,
  431. // at any given time one end must be receiving and other sending (though these may swap during the conversation)
  432. interface IConversation: extends IInterface
  433. {
  434. virtual bool accept(unsigned timeoutms)=0; // one side accepts
  435. virtual void set_keep_alive(bool keepalive)=0; // enable keepalive for socket
  436. virtual bool connect(unsigned timeoutms)=0; // other side connects
  437. virtual bool send(MemoryBuffer &mb)=0; // 0 length buffer can be sent
  438. virtual bool recv(MemoryBuffer &mb, unsigned timeoutms)=0; // up to protocol to terminate conversation (e.g. by zero length buffer)
  439. virtual void cancel()=0; // cancels above methods (from separate thread)
  440. virtual unsigned short setRandomPort(unsigned short base, unsigned num)=0; // sets a random unique port for accept use
  441. };
  442. extern jlib_decl IConversation *createSingletonSocketConnection(unsigned short port,SocketEndpoint *ep=NULL);
  443. // the end that listens may omit ep
  444. // this function does not connect so raises no socket exceptions
  445. // interface for reading from multiple sockets using the BF_SYNC_TRANSFER_PUSH protocol
  446. interface ISocketBufferReader: extends IInterface
  447. {
  448. public:
  449. virtual void init(unsigned num,ISocket **sockets,size32_t buffermax=(unsigned)-1)=0;
  450. virtual unsigned get(MemoryBuffer &mb)=0;
  451. virtual void done(bool wait)=0;
  452. };
  453. extern jlib_decl ISocketBufferReader *createSocketBufferReader(const char *trc=NULL);
  454. extern jlib_decl void markNodeCentral(SocketEndpoint &ep); // random delay for linux
  455. interface ISocketConnectNotify
  456. {
  457. public:
  458. virtual void connected(unsigned idx,const SocketEndpoint &ep,ISocket *socket)=0; // must link socket if kept
  459. virtual void failed(unsigned idx,const SocketEndpoint &ep,int err)=0;
  460. };
  461. extern jlib_decl void multiConnect(const SocketEndpointArray &eps,ISocketConnectNotify &inotify,unsigned timeout);
  462. extern jlib_decl void multiConnect(const SocketEndpointArray &eps,IPointerArrayOf<ISocket> &retsockets,unsigned timeout);
  463. interface ISocketConnectWait: extends IInterface
  464. {
  465. public:
  466. virtual ISocket *wait(unsigned waittimems)=0; // return NULL if time expired, throws exception if connect failed
  467. // releasing ISocketConnectWait cancels the connect iff wait has never returned socket
  468. };
  469. extern jlib_decl ISocketConnectWait *nonBlockingConnect(SocketEndpoint &ep,unsigned connectimeoutms=0);
  470. // buffered socket
  471. interface IBufferedSocket : implements IInterface
  472. {
  473. virtual int readline(char* buf, int maxlen, IMultiException *me) = 0;
  474. virtual int read(char* buf, int maxlen) = 0;
  475. virtual int readline(char* buf, int maxlen, bool keepcrlf, IMultiException *me) = 0;
  476. virtual void setReadTimeout(unsigned int timeout) = 0;
  477. };
  478. #define BSOCKET_READ_TIMEOUT 600
  479. #define BSOCKET_CLIENT_READ_TIMEOUT 7200
  480. extern jlib_decl IBufferedSocket* createBufferedSocket(ISocket* socket);
  481. #define MAX_NET_ADDRESS_SIZE (16)
  482. extern jlib_decl IpSubNet &queryPreferredSubnet(); // preferred subnet when resolving multiple NICs
  483. extern jlib_decl bool setPreferredSubnet(const char *ip,const char *mask); // also resets cached host IP
  484. extern jlib_decl StringBuffer &lookupHostName(const IpAddress &ip,StringBuffer &ret);
  485. extern jlib_decl bool isInterfaceIp(const IpAddress &ip, const char *ifname);
  486. extern jlib_decl bool getInterfaceIp(IpAddress &ip, const char *ifname);
  487. extern jlib_decl bool getInterfaceName(StringBuffer &ifname);
  488. //Given a list of server sockets, wait until any one or more are ready to be read/written (wont block)
  489. //return array of ready sockets
  490. extern jlib_decl int wait_read_multiple(UnsignedArray &socks, //IN sockets to be checked for read readiness
  491. unsigned timeoutMS, //IN timeout
  492. UnsignedArray &readySocks);//OUT sockets ready to be read
  493. extern jlib_decl int wait_write_multiple(UnsignedArray &socks, //IN sockets to be checked for write readiness
  494. unsigned timeoutMS, //IN timeout
  495. UnsignedArray &readySocks);//OUT sockets ready to be written
  496. extern jlib_decl void throwJSocketException(int jsockErr);
  497. extern jlib_decl IJSOCK_Exception* createJSocketException(int jsockErr, const char *_msg);
  498. extern jlib_decl bool isIPV4(const char *ip);
  499. extern jlib_decl bool isIPV6(const char *ip);
  500. extern jlib_decl bool isIPAddress(const char *ip);
  501. interface IAllowListHandler : extends IInterface
  502. {
  503. virtual bool isAllowListed(const char *ip, unsigned __int64 role, StringBuffer *responseText=nullptr) const = 0;
  504. virtual StringBuffer &getAllowList(StringBuffer &out) const = 0;
  505. virtual void refresh() = 0;
  506. };
  507. interface IAllowListWriter : extends IInterface
  508. {
  509. virtual void add(const char *ip, unsigned __int64 role) = 0;
  510. virtual void setAllowAnonRoles(bool tf) = 0;
  511. };
  512. typedef std::function<bool(IAllowListWriter &)> AllowListPopulateFunction;
  513. typedef std::function<StringBuffer &(StringBuffer &, unsigned __int64)> AllowListFormatFunction;
  514. extern jlib_decl IAllowListHandler *createAllowListHandler(AllowListPopulateFunction populateFunc, AllowListFormatFunction roleFormatFunc = {}); // format function optional
  515. #endif