windows socket编程实例-UDP(2007-07-22 14:05)

 

写了个mudp类, 省去了一些比较讨厌的实现细节...

刚开始把closesocket放在了析构函数里, 导致调用copyconstructor时把socket给关了=.=

发送和接收的example都在一个main里面了.

项目属性那还得设置为多线程

1. #include "stdio.h"
2. #include "winsock2.h"
3. #pragma     comment(lib,"WS2_32.LIB")
5. class mudp{
6. public:
7.       SOCKET socket1;
8.       sockaddr_in local;//本地地址
9.       sockaddr_in from;//udp包来源地址
10.       sockaddr_in dir;//目标地址
11.       int len;//上面这些东西的大小
12.       char buffer[1024];//内部用的缓冲
13. public:
14.       mudp(int port);//需要receive的话,提供一个port
16.       mudp();//不需要receive则不必提供
18.       void close();//关闭socket
20.       bool receive(char* mes,char* address,int& port);//udp包的信息,地址,端口号
22.       bool receive(char* mes);//只接收信息
24.       void setaddress(char* ip,int port);//设置目的地址
26.       bool send(char* mes,int size,char* address,int port);//发到信息到参数指定的地址,size为信息长度
28.       bool send(char* mes,int size);//发到信息到dir指定的地址,size为信息长度
30.       bool sendlast(char* mes,int size);//发送到上次接收udp包的来源地址
32. };
34. mudp::mudp(int port){
35.       len =sizeof(from);
36.       from.sin_family=AF_INET;
37.       local.sin_family=AF_INET;
38.       local.sin_port=htons(port); 
39.       local.sin_addr.s_addr=INADDR_ANY; 
40.       from=dir=local;
41.       socket1=socket(AF_INET,SOCK_DGRAM,0);
42.       bind(socket1,(struct sockaddr*)&local,len);
43. }
45. mudp::mudp(){
46.       len =sizeof(from);
47.       from.sin_family=AF_INET;
48.       socket1=socket(AF_INET,SOCK_DGRAM,0);
49. }
51. void mudp::close(){
52.       closesocket(socket1);
53. }
55. bool mudp::receive(char* mes,char* address,int& port){
56.       if (recvfrom(socket1,buffer,sizeof(buffer),0,(struct sockaddr*)&from,&len)!=SOCKET_ERROR){
57.           strcpy(mes,buffer);
58.           strcpy(address,inet_ntoa(from.sin_addr));
59.           port=from.sin_port;
60.           return true;
61.       }
62.       else{
63.           return false;
64.       }
65. }
67. bool mudp::receive(char* mes){
68.       if (recvfrom(socket1,buffer,sizeof(buffer),0,(struct sockaddr*)&from,&len)!=SOCKET_ERROR){
69.           strcpy(mes,buffer);
70.           return true;
71.       }
72.       else{
73.           return false;
74.       }
75. }
77. void mudp::setaddress(char* ip,int port){
78.       dir.sin_port=htons(port); 
79.       dir.sin_addr.s_addr=inet_addr(ip);
80. }
82. bool mudp::send(char* mes,int size,char* address,int port){    
83.       dir.sin_family=AF_INET;
84.       dir.sin_addr.s_addr=inet_addr(address); 
85.       dir.sin_port=htons(port);
86.       if(sendto(socket1,mes,size,0,(struct sockaddr*)&dir,len)!=SOCKET_ERROR){
87.           return true;
88.       }
89.       return false;
90. }
92. bool mudp::send(char* mes,int size){
93.       if(sendto(socket1,mes,size,0,(struct sockaddr*)&dir,len)!=SOCKET_ERROR){
94.           return true;
95.       }
96.       return false;
97. }
99. bool mudp::sendlast(char* mes,int size){
100.       if(sendto(socket1,mes,size,0,(struct sockaddr*)&from,len)!=SOCKET_ERROR){
101.           return true;
102.       }
103.       return false;
104. }
107. HANDLE hlisten;
109. mudp sock;
111. void mlisten(void* arg){
112.       char str[1024];
113.       char ip[30];
114.       int port;
115.       printf("thread started/n");
116.       while(1){
117.           if(sock.receive(str,ip,port)){
118.               printf("message received from %s:%d/n%s/n",ip,port,str);
119.               if(strcmp(str,"quit")==0){
120.                   break;
121.               }
122.           }
123.           else{
124.               printf("can not reveive message/n");
125.           }
126.       }
127.       printf("listen thread quit/n");
128. }
131. void main(){
132.       WSADATA wsaData;
133.       if (WSAStartup(MAKEWORD(2,1),&wsaData)){
134.            printf("Winsock initializing fail/n");
135.            WSACleanup();
136.            return;
137.       }
138.       sock=mudp(12345);
139.       hlisten=CreateThread(NULL,0,(LPTHREAD_START_ROUTINE) mlisten,NULL,CREATE_SUSPENDED,NULL);
140.       ResumeThread(hlisten);
141.       if(hlisten==NULL){
142.           printf("thread create fail/n");
143.       }
144.       char str[1024];
145.       mudp a(54321);
146.       a.setaddress("172.18.48.7",12345);
147.       while(1){
148.           scanf("%s",str);
149.           a.send(str,1024);
150.           if(strcmp(str,"quit")==0){
151.               break;
152.           }
153.       }
155.       Sleep(100);
156.       sock.close();
157.       WSACleanup();
158. }