socket.h 3.5 KB

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