socket.h 4.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140
  1. // Copyright (c) Athena Dev Teams - Licensed under GNU GPL
  2. // For more information, see LICENCE in the main folder
  3. #ifndef _SOCKET_H_
  4. #define _SOCKET_H_
  5. #ifdef WIN32
  6. #include <windows.h>
  7. typedef long in_addr_t;
  8. #else
  9. #include <sys/types.h>
  10. #include <sys/socket.h>
  11. #include <netinet/in.h>
  12. #endif
  13. #include "../common/cbasetypes.h"
  14. #include <time.h>
  15. // socket I/O macros
  16. #ifdef TURBO
  17. #define RFIFOVAR(fd) rbPtr ## fd
  18. #define WFIFOVAR(fd) wbPtr ## fd
  19. #define RFIFOHEAD(fd) uint8 *RFIFOVAR(fd) = session[fd]->rdata+session[fd]->rdata_pos
  20. #define WFIFOHEAD(fd, x) uint8 *WFIFOVAR(fd) = ( (fd) > 0 && session[fd] ? session[fd]->wdata+session[fd]->wdata_size : NULL )
  21. #define RFIFOP(fd,pos) ( &RFIFOVAR(fd) + (pos) )
  22. #define WFIFOP(fd,pos) ( &WFIFOVAR(fd) + (pos) )
  23. #else
  24. #define RFIFOHEAD(fd)
  25. #define WFIFOHEAD(fd, size) do{ if((fd) && session[fd]->wdata_size + (size) > session[fd]->max_wdata ) realloc_writefifo(fd, size); }while(0)
  26. #define RFIFOP(fd,pos) (session[fd]->rdata + session[fd]->rdata_pos + (pos))
  27. #define WFIFOP(fd,pos) (session[fd]->wdata + session[fd]->wdata_size + (pos))
  28. #endif
  29. #define RFIFOB(fd,pos) (*(uint8*)RFIFOP(fd,pos))
  30. #define WFIFOB(fd,pos) (*(uint8*)WFIFOP(fd,pos))
  31. #define RFIFOW(fd,pos) (*(uint16*)RFIFOP(fd,pos))
  32. #define WFIFOW(fd,pos) (*(uint16*)WFIFOP(fd,pos))
  33. #define RFIFOL(fd,pos) (*(uint32*)RFIFOP(fd,pos))
  34. #define WFIFOL(fd,pos) (*(uint32*)WFIFOP(fd,pos))
  35. #define RFIFOSPACE(fd) (session[fd]->max_rdata - session[fd]->rdata_size)
  36. #define WFIFOSPACE(fd) (session[fd]->max_wdata - session[fd]->wdata_size)
  37. #define RFIFOREST(fd) (session[fd]->rdata_size - session[fd]->rdata_pos)
  38. #define RFIFOFLUSH(fd) \
  39. do { \
  40. if(session[fd]->rdata_size == session[fd]->rdata_pos){ \
  41. session[fd]->rdata_size = session[fd]->rdata_pos = 0; \
  42. } else { \
  43. session[fd]->rdata_size -= session[fd]->rdata_pos; \
  44. memmove(session[fd]->rdata, session[fd]->rdata+session[fd]->rdata_pos, session[fd]->rdata_size); \
  45. session[fd]->rdata_pos = 0; \
  46. } \
  47. } while(0)
  48. // buffer I/O macros
  49. #define RBUFP(p,pos) (((uint8*)(p)) + (pos))
  50. #define RBUFB(p,pos) (*(uint8*)RBUFP((p),(pos)))
  51. #define RBUFW(p,pos) (*(uint16*)RBUFP((p),(pos)))
  52. #define RBUFL(p,pos) (*(uint32*)RBUFP((p),(pos)))
  53. #define WBUFP(p,pos) (((uint8*)(p)) + (pos))
  54. #define WBUFB(p,pos) (*(uint8*)WBUFP((p),(pos)))
  55. #define WBUFW(p,pos) (*(uint16*)WBUFP((p),(pos)))
  56. #define WBUFL(p,pos) (*(uint32*)WBUFP((p),(pos)))
  57. #define TOB(n) ((uint8)((n)&UINT8_MAX))
  58. #define TOW(n) ((uint16)((n)&UINT16_MAX))
  59. #define TOL(n) ((uint32)((n)&UINT32_MAX))
  60. // Struct declaration
  61. typedef int (*RecvFunc)(int fd);
  62. typedef int (*SendFunc)(int fd);
  63. typedef int (*ParseFunc)(int fd);
  64. struct socket_data {
  65. unsigned char eof;
  66. unsigned char *rdata, *wdata;
  67. size_t max_rdata, max_wdata;
  68. size_t rdata_size, wdata_size;
  69. size_t rdata_pos;
  70. time_t rdata_tick; // time of last receive (for detecting timeouts)
  71. uint32 client_addr; // remote client address (zero for s2s connections)
  72. void* session_data;
  73. RecvFunc func_recv;
  74. SendFunc func_send;
  75. ParseFunc func_parse;
  76. };
  77. // Data prototype declaration
  78. extern struct socket_data* session[FD_SETSIZE];
  79. extern int fd_max;
  80. extern time_t last_tick;
  81. extern time_t stall_time;
  82. //////////////////////////////////
  83. // some checking on sockets
  84. extern int session_isValid(int fd);
  85. extern int session_isActive(int fd);
  86. //////////////////////////////////
  87. // Function prototype declaration
  88. int make_listen_bind(uint32 ip, uint16 port);
  89. int make_connection(uint32 ip, uint16 port);
  90. int realloc_fifo(int fd, unsigned int rfifo_size, unsigned int wfifo_size);
  91. int realloc_writefifo(int fd, size_t addition);
  92. int WFIFOSET(int fd, int len);
  93. int RFIFOSKIP(int fd, int len);
  94. int do_sendrecv(int next);
  95. int do_parsepacket(void);
  96. void do_close(int fd);
  97. void socket_init(void);
  98. void socket_final(void);
  99. extern void flush_fifo(int fd);
  100. extern void flush_fifos(void);
  101. extern void set_nonblocking(int fd, unsigned long yes);
  102. void set_defaultparse(ParseFunc defaultparse);
  103. // hostname/ip conversion functions
  104. uint32 host2ip(const char* hostname);
  105. const char* ip2str(uint32 ip, char ip_str[16]);
  106. uint32 str2ip(const char* ip_str);
  107. #define CONVIP(ip) (ip>>24)&0xFF,(ip>>16)&0xFF,(ip>>8)&0xFF,(ip>>0)&0xFF
  108. uint16 ntows(uint16 netshort);
  109. int socket_getips(uint32* ips, int max);
  110. extern uint32 addr_[16]; // ip addresses of local host (host byte order)
  111. extern int naddr_; // # of ip addresses
  112. #endif /* _SOCKET_H_ */