http服务器长连接的实现

1. #include <stdio.h>
2. #include <stdlib.h>
3. #include <process.h>
4. #include <winsock2.h>
5. #pragma comment(lib,"ws2_32")
7. #define I_GET               0x00000001
8. #define I_POST              0x00000002
10. typedef struct web_socket
11. {
12.     SOCKET socket;
13.     struct sockaddr_in client_in;
14.     WSAEVENT event;
16.     char *buffer;
17.     u_long length;
19.     char *data;             //post数据
20.     u_long ul_datalength;   //post长度
21.     
22.     char *httppath;         //请求路径
23.     char *pathparam;    //请求路径后面的参数
25.     int  i_method;          //提交方式
26.     bool b_isConnect;       //长连接
28. }web_socket;
30. unsigned __stdcall thread_client_web(void*);
32. int main(int argc, char* argv[])
33. {
34.     WSADATA WSAData;
35.     WSAStartup(MAKEWORD(2,2),&WSAData);
37.     struct sockaddr_in server_in;
38.     SOCKET sock;
40.     server_in.sin_family = AF_INET;
41.     server_in.sin_port = htons(99);
42.     server_in.sin_addr.s_addr = INADDR_ANY;
44.     sock = socket(AF_INET,SOCK_STREAM,IPPROTO_TCP);
45.     bind(sock,(struct sockaddr far*)&server_in,sizeof(server_in));
47.     listen(sock,10);
49.     while(true)
50.     {
51.         web_socket *web = (web_socket*)malloc(sizeof(web_socket));
52.         memset(web,0x0,sizeof(web_socket));
53.         int c_len = sizeof(web->client_in);
54.         web->socket = accept(sock,(struct sockaddr far*)&web->client_in,&c_len);
55.         
56.         _beginthreadex(NULL,0,thread_client_web,web,0,NULL);
58.     }
59.     printf("Hello World!/n");
61.     WSACleanup();
62.     return 0;
63. }
65. char * __stdcall _thread_client_recv_find_header_key_to_string(char *header,char *key,char *outval)
66. {
67.     char *find,*find_over;
68.     find = strstr(header,key);
69.     if(find == NULL || (find_over = strstr(find,"/r/n")) == NULL )
70.         return NULL;
71.     else
72.     {
73.         find+=strlen(key);
74.         find++;
75.         memcpy(outval,find,find_over-find);
76.         return outval;
77.     }
78. }
80. bool __stdcall _thread_client_recv_find_header_key_bool(char *header,char *key,char *val)
81. {
82.     char outval[32]={0};
83.     if(_thread_client_recv_find_header_key_to_string(header,key,outval)==NULL)
84.         return false;
85.     else
86.     {
87.         if(strstr(outval,val) != NULL)
88.             return true;
89.         else
90.             return false;
91.     }
92. }
94. int __stdcall _thread_client_recv_find_header_key_to_integer(char *header,char *key)
95. {
96.     char val[32]={0};
97.     if(_thread_client_recv_find_header_key_to_string(header,key,val)==NULL)
98.         return 0;
99.     else
100.         return atoi(val);
101. }
103. bool __stdcall _thread_client_recv_post_http_header(web_socket *web)
104. {
105.     char *t_find=NULL;
106.     if(memcmp(web->buffer,"GET",3)==0)
107.     {
108.         web->i_method = I_GET;
109.         web->httppath = strstr(web->buffer,"GET ");
110.         web->httppath+=strlen("GET ");
111.     }
112.     else if(memcmp(web->buffer,"POST",4)==0)
113.     {
114.         web->i_method = I_POST;
115.         web->httppath = strstr(web->buffer,"POST ");
116.         web->httppath+=strlen("POST ");
118.         web->data+=2;
119.         *web->data = '/0';
120.         web->data = web->data+2;
121.         web->ul_datalength = _thread_client_recv_find_header_key_to_integer(web->buffer,"Content-Length");
122.     }
123.     
124.     web->b_isConnect = _thread_client_recv_find_header_key_bool(web->buffer,"Connection","keep-alive");
126.     if(web->httppath == NULL)
127.         return false;
128.     t_find = strstr(web->httppath," ");
129.     if(t_find != NULL)
130.     {
131.         *t_find = '/0';
132.     }
133.     t_find = strstr(web->httppath,"?");
134.     if(t_find != NULL)
135.     {
136.         *t_find = '/0';
137.         web->pathparam = t_find+1;
138.     }
139.     return true;
140. }
142. bool __stdcall _thread_client_check_is_recv(web_socket *web)
143. {
144.     if(web->i_method == 0 )
145.     {
146.         web->data = strstr(web->buffer,"/r/n/r/n");
147.         if(NULL == web->data)
148.             return true;
149.         else
150.             _thread_client_recv_post_http_header(web);
151.     }
152.     if(web->i_method == I_GET)
153.     {
154.         return false;
155.     }
156.     else if(web->i_method == I_POST)
157.     {
158.         if(web->ul_datalength == strlen(web->data))
159.             return false;
160.     }
161.     return true;
162. }
164. unsigned __stdcall thread_client_web_send(web_socket *web)
165. {
166.     if(web->i_method == I_GET)
167.         printf("RECV:%d/n%s:%s/n",web->length,web->httppath,web->pathparam);
168.     else if(web->i_method == I_POST)
169.         printf("RECV:%d/n%s:%s/n%s/n",web->length,web->httppath,web->pathparam,web->data);
170.     char t[]="aaaaa";
171.     char buff[512]={0};
173.     sprintf(buff,"http/1.0 200 ok/r/nContent-Length:%d/r/nConnection:Keep-Alive/r/n/r/n%s",strlen(t),t);
175.     send(web->socket,buff,strlen(buff),0);
176.     return 1;
177. }
179. unsigned __stdcall thread_client_web(void* param)
180. {
181. #define PAGE_SIZE 0x2000
182.     
183.     bool isRecvOver;
184.     web_socket *web = (web_socket*)param;
185.     web->event = WSACreateEvent();
186.     web->buffer = (char*)malloc(PAGE_SIZE*10);
187.     
188.     WSAEventSelect(web->socket,web->event,FD_READ|FD_CLOSE);
190.     WSANETWORKEVENTS networkevent;
191.     u_long cRecv;
193. web_loop_next:
194.     isRecvOver=true;
195.     
196.     memset(web->buffer,0x0,PAGE_SIZE*10);
197.     web->length=0;
198.     web->b_isConnect = true;
199.     web->i_method=0;
200.     web->data=NULL;
201.     web->httppath=NULL;
202.     web->pathparam=NULL;
204.     while(isRecvOver)
205.     {
206.         WaitForSingleObject(web->event,INFINITE);
207.     
208.         WSAEnumNetworkEvents(web->socket,web->event,&networkevent);
209.         if(networkevent.lNetworkEvents & FD_READ)
210.         {
211.             ioctlsocket(web->socket,FIONREAD,&cRecv);
212.             web->length += recv(web->socket,web->buffer+web->length,cRecv,0);
213.             isRecvOver = _thread_client_check_is_recv(web);
214.         }
215.         else if(networkevent.lNetworkEvents & FD_CLOSE)
216.         {
217.             closesocket(web->socket);
218.             isRecvOver=false;
219.             web->b_isConnect=false;
220.             goto _exit;
221.         }
222.     }
224.     thread_client_web_send(web);
226. //  if(web->b_isConnect)
227.         goto web_loop_next;
228.     
229. _exit:
230.     printf("over");
231.     
232.     free(web->buffer);
233.     free(web);
234.     return 0;
235. }