UDP socket (UDP套接口通信)

经常用TCP套接口通信，UDP套接口也是很重要。今天使用了一下，小节如下：

无论是 Server 还是 Client，首先包含头文件和库

#include <winsock.h>

#pragma comment(lib, "wsock32")

对于创建Server：

1. WSAStartup() 加载套接字库

2. socket() 创建套接字 

3. bind() 绑定

4. recvfrom() 接收 / sendto() 发送

5. closesocket() 关闭套接字

6. WSACleanup() 卸载套接字库

对于创建Client:  (不需要bind)

1. WSAStartup() 加载套接字库

2. socket() 创建套接字 

3. recvfrom() 接收 / sendto() 发送

4. closesocket() 关闭套接字

5. WSACleanup() 卸载套接字库

Server端代码如下：(功能 - 接收从Client端发送的计算机使用信息)

2. #include <stdio.h>
3. #include <winsock.h>
4. #pragma comment(lib, "wsock32")
6. typedef struct
7. {
8.     char ComputerName[30];
9.     char UserName[30];
10.     UINT CPURate;
11. }COMPUTER_INFORM;
13. int main()
14. {
15.     /*  start WSA   */
16.     WSADATA wsaData;
17.     int iResult;
18.     
19.     iResult = WSAStartup(MAKEWORD(2,2), &wsaData);
20.     if (iResult != NO_ERROR)
21.     {
22.         printf("Error at WSAStartup()/n");
23.         return 0;
24.     }
26.     /*  Create socket   */
27.     SOCKET sktSrv;
28.     sktSrv = socket(AF_INET, SOCK_DGRAM, 0 );
29.     if( sktSrv == INVALID_SOCKET )
30.     {
31.         printf("Error at socket()/n");
32.         return 0;
33.     }
35.     /*  Create address  */
36.     SOCKADDR_IN addrSrv;
37.     addrSrv.sin_family = AF_INET;
38.     addrSrv.sin_addr.S_un.S_addr = htonl(INADDR_ANY);
39.     addrSrv.sin_port = htons(4345);
41.     /*  Bind socket */
42.     int result = bind(sktSrv,(SOCKADDR*)&addrSrv,sizeof(SOCKADDR));
43.     if(result == SOCKET_ERROR)
44.     {
45.         printf("Error at bind() : %d/n", WSAGetLastError());
46.         return 0;
47.     }
49.     /*  Receive data    */
50.     COMPUTER_INFORM info;
51.     SOCKADDR addrClt;
52.     int addrLen, recvLen;
53.     UINT index;
54.     recvLen = -1;
55.     index = 0;
56.     addrLen = sizeof(SOCKADDR);
57.     printf("Begin to receive/n");
58.     while(1)
59.     {
60.         recvLen = recvfrom(sktSrv, (char *)&info, sizeof(info), 0, (SOCKADDR *)&addrClt, &addrLen);
61.         // Add your process for "info"
    
62.         printf("%d, %s: %s, %3.1d%%/n", index, info.ComputerName, info.UserName, info.CPURate);
63.         index ++;
64.     }
66.     /*  Close socket    */
67.     closesocket(sktSrv);
68.     WSACleanup();
69.     return 0;
70. }

Client端代码如下： （功能 - 获取本地机的计算机名、用户名、CPU使用率等信息，发送到Server）

2. #include <stdio.h>
3. #include <winsock.h>
4. #pragma comment(lib, "wsock32")
6. #define ADDR_SOCKET_SRV     "192.168.0.117"
7. #define APP_SOCKET_PORT     4545
8. #define SERVER_UPDATE_INTERVAL      60000   //  ms
9. #define CLIENT_UPDATE_INTERVAL      6000    //  ms
11. #define SystemBasicInformation       0
12. #define SystemPerformanceInformation 2
13. #define SystemTimeInformation        3
15. #define Li2Double(x) ((double)((x).HighPart) * 4.294967296E9 + (double)((x).LowPart))
17. typedef struct
18. {
19.     char ComputerName[30];
20.     char UserName[30];
21.     UINT CPURate;
22. }COMPUTER_INFORM;
24. typedef struct
25. {
26.     LARGE_INTEGER   liIdleTime;
27.     DWORD           dwSpare[76];
28. } SYSTEM_PERFORMANCE_INFORMATION;
30. typedef struct
31. {
32.     DWORD   dwUnknown1;
33.     ULONG   uKeMaximumIncrement;
34.     ULONG   uPageSize;
35.     ULONG   uMmNumberOfPhysicalPages;
36.     ULONG   uMmLowestPhysicalPage;
37.     ULONG   uMmHighestPhysicalPage;
38.     ULONG   uAllocationGranularity;
39.     PVOID   pLowestUserAddress;
40.     PVOID   pMmHighestUserAddress;
41.     ULONG   uKeActiveProcessors;
42.     BYTE    bKeNumberProcessors;
43.     BYTE    bUnknown2;
44.     WORD    wUnknown3;
45. } SYSTEM_BASIC_INFORMATION;
47. typedef struct
48. {
49.     LARGE_INTEGER liKeBootTime;
50.     LARGE_INTEGER liKeSystemTime;
51.     LARGE_INTEGER liExpTimeZoneBias;
52.     ULONG         uCurrentTimeZoneId;
53.     DWORD         dwReserved;
54. } SYSTEM_TIME_INFORMATION;
56. typedef LONG (WINAPI *PROCNTQSI)(UINT,PVOID,ULONG,PULONG);
59. void main(void)
60. {
61.     SYSTEM_PERFORMANCE_INFORMATION SysPerfInfo;
62.     SYSTEM_TIME_INFORMATION        SysTimeInfo;
63.     SYSTEM_BASIC_INFORMATION       SysBaseInfo;
64.     double                         dbIdleTime;
65.     double                         dbSystemTime;
66.     LONG                           status;
67.     LARGE_INTEGER                  liOldIdleTime = {0,0};
68.     LARGE_INTEGER                  liOldSystemTime = {0,0};
69.     COMPUTER_INFORM                ComputerInfo;
70.     SOCKET                         m_sockClt=0;
71.     char                           ComputerName[30]; 
72.     char                           UserName[30];
73.     DWORD                          size=30;  
74.     WSADATA                        wsaData;
75.     sockaddr_in                    clientService; 
76.     int                            ret;
77.     int                            iResult;
78.     PROCNTQSI NtQuerySystemInformation;
80.     iResult = WSAStartup(MAKEWORD(2,2), &wsaData);
81.     if (iResult != NO_ERROR)
82.     {
83.         printf("Error at WSAStartup()/n");
84.         return;
85.     }
87.     memset(&ComputerInfo,0,sizeof(COMPUTER_INFORM));
89.     /* Init the client Socket */
90.     m_sockClt = socket(AF_INET, SOCK_DGRAM, 0);
92.     if(m_sockClt == INVALID_SOCKET)
93.     {           
94.        printf("Create Client Socket Error %d!/n",WSAGetLastError());
95.        return;
96.     }
98.     /* Connect to server */
99.     clientService.sin_family = AF_INET;
100.     clientService.sin_addr.s_addr = inet_addr(ADDR_SOCKET_SRV); //inet_addr("192.168.0.117" );
101.     clientService.sin_port = htons(APP_SOCKET_PORT);    //htons( 10500 );
103.     while(1)
104.     {
105.         NtQuerySystemInformation = (PROCNTQSI)GetProcAddress(
106.                                           GetModuleHandle("ntdll"),
107.                                          "NtQuerySystemInformation"
108.                                          );
109.         if (!NtQuerySystemInformation)
110.         {
111.             continue;
112.         }
113.         else
114.         {
115.             break;
116.         }
117.     }
119.     while(1)
120.     {
121.         /* Get Number of Processors in the system */
122.         status = NtQuerySystemInformation(SystemBasicInformation,&SysBaseInfo,sizeof(SysBaseInfo),NULL);
123.         if (status != NO_ERROR)
124.         {
125.             continue;
126.         }
127.         else
128.         {
129.             break;
130.         }
131.     }
132.     
133.     /* Get Computer Name */
134.     if(!GetComputerName(ComputerName,&size))   
135.     {   
136.         printf("Failed 1/n");
137.         strcpy(ComputerInfo.UserName,"EMPTY");
139.     }
140.     else
141.     {
142.         printf("The Computer Name is :%s/n",ComputerName);
143.         strcpy(ComputerInfo.ComputerName,ComputerName);
144.     }
146.     printf("/nCPU Usage (press any key to exit):    /n");
147.     UINT index = 0;
148.     while(1)
149.     {
150.         
151.         /* Get New System Time */
152.         status = NtQuerySystemInformation(SystemTimeInformation,&SysTimeInfo,sizeof(SysTimeInfo),0);
153.         if (status!=NO_ERROR)
154.             return;
156.         /* Get new CPU's idle time */
157.         status = NtQuerySystemInformation(SystemPerformanceInformation,&SysPerfInfo,sizeof(SysPerfInfo),NULL);
158.         if (status != NO_ERROR)
159.             return;
161.         /* if it's a first call - skip it */
162.        if (liOldIdleTime.QuadPart != 0)
163.        {
164.             /* CurrentValue = NewValue - OldValue */
165.             dbIdleTime = Li2Double(SysPerfInfo.liIdleTime) - Li2Double(liOldIdleTime);
166.             dbSystemTime = Li2Double(SysTimeInfo.liKeSystemTime) - Li2Double(liOldSystemTime);
168.             /* CurrentCpuIdle = IdleTime / SystemTime */
169.             dbIdleTime = dbIdleTime / dbSystemTime;
171.             /* CurrentCpuUsage% = 100 - (CurrentCpuIdle * 100) / NumberOfProcessors */
172.             dbIdleTime = 100.0 - dbIdleTime * 100.0 / (double)SysBaseInfo.bKeNumberProcessors + 0.5;
173.             
174.             if(dbIdleTime < 0)
175.             {
176.                 dbIdleTime = -dbIdleTime;
177.             }
179.             //printf("/b/b/b/b%3d%%/n",(UINT)dbIdleTime);
180.             //printf("index = %d/t", index);
182.             /* Get User Name */
183.             if(!GetUserName(UserName,&size))
184.             {   
185.                 printf("Failed 2/n");
186.                 //strcpy(ComputerInfo.UserName,"EMPTY");
187.                 continue;
188.             }
189.             else
190.             {
191.                 //printf("User Name : %s/t",UserName);
192.                 strcpy(ComputerInfo.UserName,UserName);
193.             }
195.             /* Write the CPU usage rate to the struct */
196.             ComputerInfo.CPURate = (UINT)dbIdleTime;
197.             //printf("%4.1d%%/n",(UINT)dbIdleTime);
198.             //ret = send(m_CSocket,(char *)&ComputerInfo,sizeof(ComputerInfo),0);
199.             printf("%d, %s: %s, %3.1d%%/n", 
200.                 index, ComputerInfo.ComputerName, ComputerInfo.UserName, ComputerInfo.CPURate);
201.             ret = sendto(m_sockClt, (char *)&ComputerInfo, sizeof(ComputerInfo), 0, 
202.                 (SOCKADDR *)&clientService, sizeof(SOCKADDR));
204.             if(ret == SOCKET_ERROR)
205.             {
206.                 printf("Send Computer Information Error!/n");
207.                 //goto Repeat;
208.             }
209.        }
210.        index ++;
212.         /* store new CPU's idle and system time */
213.         liOldIdleTime = SysPerfInfo.liIdleTime;
214.         liOldSystemTime = SysTimeInfo.liKeSystemTime;
216.         /* wait one minute */
217.         //Sleep(6000);
218.         Sleep(CLIENT_UPDATE_INTERVAL);
219.     }
220.     printf("/n");
221. }