socket.h 3.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116
  1. // original : core.h 2003/03/14 11:55:25 Rev 1.4
  2. #ifndef _SOCKET_H_
  3. #define _SOCKET_H_
  4. #include <stdio.h>
  5. #ifdef __WIN32
  6. #include <winsock.h>
  7. #define close(fd) closesocket(fd)
  8. #else
  9. #include <sys/types.h>
  10. #include <sys/socket.h>
  11. #include <netinet/in.h>
  12. #endif
  13. #include <time.h>
  14. #include "malloc.h"
  15. extern time_t tick_;
  16. extern time_t stall_time_;
  17. // define declaration
  18. #define RFIFOP(fd,pos) (session[fd]->rdata+session[fd]->rdata_pos+(pos))
  19. #define RFIFOB(fd,pos) (*(unsigned char*)(session[fd]->rdata+session[fd]->rdata_pos+(pos)))
  20. #define RFIFOW(fd,pos) (*(unsigned short*)(session[fd]->rdata+session[fd]->rdata_pos+(pos)))
  21. #define RFIFOL(fd,pos) (*(unsigned int*)(session[fd]->rdata+session[fd]->rdata_pos+(pos)))
  22. //#define RFIFOSKIP(fd,len) ((session[fd]->rdata_size-session[fd]->rdata_pos-(len)<0) ? (fprintf(stderr,"too many skip\n"),exit(1)) : (session[fd]->rdata_pos+=(len)))
  23. #define RFIFOREST(fd) (session[fd]->rdata_size-session[fd]->rdata_pos)
  24. #define RFIFOFLUSH(fd) (memmove(session[fd]->rdata,RFIFOP(fd,0),RFIFOREST(fd)),session[fd]->rdata_size=RFIFOREST(fd),session[fd]->rdata_pos=0)
  25. #define RFIFOSPACE(fd) (session[fd]->max_rdata-session[fd]->rdata_size)
  26. #define RBUFP(p,pos) (((unsigned char*)(p))+(pos))
  27. #define RBUFB(p,pos) (*(unsigned char*)RBUFP((p),(pos)))
  28. #define RBUFW(p,pos) (*(unsigned short*)RBUFP((p),(pos)))
  29. #define RBUFL(p,pos) (*(unsigned int*)RBUFP((p),(pos)))
  30. #define WFIFOSPACE(fd) (session[fd]->max_wdata-session[fd]->wdata_size)
  31. #define WFIFOP(fd,pos) (session[fd]->wdata+session[fd]->wdata_size+(pos))
  32. #define WFIFOB(fd,pos) (*(unsigned char*)(session[fd]->wdata+session[fd]->wdata_size+(pos)))
  33. #define WFIFOW(fd,pos) (*(unsigned short*)(session[fd]->wdata+session[fd]->wdata_size+(pos)))
  34. #define WFIFOL(fd,pos) (*(unsigned int*)(session[fd]->wdata+session[fd]->wdata_size+(pos)))
  35. // use function instead of macro.
  36. //#define WFIFOSET(fd,len) (session[fd]->wdata_size = (session[fd]->wdata_size+(len)+2048 < session[fd]->max_wdata) ? session[fd]->wdata_size+len : session[fd]->wdata_size)
  37. #define WBUFP(p,pos) (((unsigned char*)(p))+(pos))
  38. #define WBUFB(p,pos) (*(unsigned char*)WBUFP((p),(pos)))
  39. #define WBUFW(p,pos) (*(unsigned short*)WBUFP((p),(pos)))
  40. #define WBUFL(p,pos) (*(unsigned int*)WBUFP((p),(pos)))
  41. #ifdef __INTERIX
  42. #define FD_SETSIZE 4096
  43. #endif // __INTERIX
  44. /* Removed Cygwin FD_SETSIZE declarations, now are directly passed on to the compiler through Makefile [Valaris] */
  45. // Struct declaration
  46. struct socket_data{
  47. int eof;
  48. unsigned char *rdata,*wdata;
  49. int max_rdata,max_wdata;
  50. int rdata_size,wdata_size;
  51. time_t rdata_tick;
  52. int rdata_pos;
  53. struct sockaddr_in client_addr;
  54. int (*func_recv)(int);
  55. int (*func_send)(int);
  56. int (*func_parse)(int);
  57. int (*func_console)(char*);
  58. void* session_data;
  59. };
  60. // Data prototype declaration
  61. #ifdef _WIN32
  62. #undef FD_SETSIZE
  63. #define FD_SETSIZE 4096
  64. #endif
  65. extern struct socket_data *session[FD_SETSIZE];
  66. extern int rfifo_size,wfifo_size;
  67. extern int fd_max;
  68. // Function prototype declaration
  69. int make_listen_port(int);
  70. int make_listen_bind(long,int);
  71. int make_connection(long,int);
  72. int delete_session(int);
  73. int realloc_fifo(int fd,int rfifo_size,int wfifo_size);
  74. int WFIFOSET(int fd,int len);
  75. int RFIFOSKIP(int fd,int len);
  76. int do_sendrecv(int next);
  77. int do_parsepacket(void);
  78. void do_socket(void);
  79. extern void flush_fifos();
  80. extern void set_nonblocking(int fd, int yes);
  81. int start_console(void);
  82. void set_defaultparse(int (*defaultparse)(int));
  83. void set_defaultconsoleparse(int (*defaultparse)(char*));
  84. int Net_Init(void);
  85. extern unsigned int addr_[16]; // ip addresses of local host (host byte order)
  86. extern unsigned int naddr_; // # of ip addresses
  87. #endif // _SOCKET_H_