windows API(TCP/IP)通讯编程

server端程序：

1. #include <winsock2.h>
2. #include <stdio.h>
3. #include <stdlib.h>
4. #include <windows.h>
5. #include <process.h>
7. #pragma comment(lib,"Ws2_32")
9. #define DEFAULT_PORT 5150
10. #define DEFAULT_BUFFER 4096
12. int iPort = DEFAULT_PORT;
13. BOOL bInterface = FALSE,
14. bRecvOnly = FALSE;
15. char szAddress[128];
17. void usage()
18. {
19.  printf("usage:server[-p:x] [-i:IP] [-o]/n/n");
20.  printf("       -p:x   Port number to listen on/n");
21.  printf("       -i:str Interface to listen on/n");
22.  printf("       -o     Don't echo the data back/n/n");
23.  ExitProcess(1);
24. }
26. void ValidateArgs(int argc,char **argv)
27. {
28.  int i;
29.  for(i=1;i<argc;i++)
30.  {
31.   if((argv[i][0] == '-')||(argv[i][0] == '/'))
32.   {
33.    switch(tolower(argv[i][1]))
34.    {
35.    case 'p':
36.     iPort = atoi(&argv[i][3]);
37.     break;
38.    case 'i':
39.     bInterface = TRUE;
40.     if(strlen(argv[i])>3)
41.      strcpy(szAddress,&argv[i][3]);
42.     break;
43.    case 'o':
44.     bRecvOnly = TRUE;
45.     break;
46.    default:
47.     usage();
48.     break;
49.    }
50.   }
51.  }
52. }
53. DWORD WINAPI ClientThread(LPVOID lpParam)
54. {
55.  SOCKET sock = (SOCKET) lpParam;
56.  char   szBuff[DEFAULT_BUFFER];
57.  int ret,nLeft,idx;
58.  while(1)
59.  {
60.   ret = recv(sock,szBuff,DEFAULT_BUFFER,0);
61.   if(ret == 0)
62.    break;
63.   else if(ret == SOCKET_ERROR)
64.   {
65.    printf("recv() failed:%d/n",WSAGetLastError());
66.    break;
67.   }
68.   szBuff[ret]='/0';
69.   printf("recv:'%s'/n",szBuff);
70.   if(!bRecvOnly)
71.   {
72.    nLeft = ret;
73.    idx = 0;
74.    while(nLeft>0)
75.    {
76.     ret = send(sock,&szBuff[idx],nLeft,0);
77.     if(ret == 0)
78.      break;
79.     else if (ret == SOCKET_ERROR)
80.     {
81.      printf("send() failed:%d/n",WSAGetLastError());
82.      break;
83.     }
84.     nLeft -=ret;
85.     idx+=ret;
86.    }
87.   }
88.  }
89.  return 0;
90. }
92. int main(int argc,char **argv)
93. {
94.  WSADATA wsd;
95.  SOCKET sListen,sClient;
96.  int iAddrSize;
97.  HANDLE hThread;
98.  DWORD dwThreadId;
99.  struct sockaddr_in local,client;
100.  ValidateArgs(argc,argv);
101.  if(WSAStartup(MAKEWORD(2,2),&wsd) != 0)
102.  {
103.   printf("failed to load winsock!/n");
104.   return 1;
105.  }
106.  sListen = socket(AF_INET, SOCK_STREAM,IPPROTO_IP);
107.  if(sListen == SOCKET_ERROR)
108.  {
109.   printf("socket() failed:%d/n",WSAGetLastError());
110.   return 1;
111.  }
112.  else
113.   printf("socket() successful!/n");
114.  if(bInterface)
115.  {
116.   local.sin_addr.s_addr = inet_addr(szAddress);
117.   if(local.sin_addr.s_addr==INADDR_NONE)
118.    usage();
119.  }
120.  else
121.   local.sin_addr.s_addr = htonl(INADDR_ANY);
122.  local.sin_family= AF_INET;
123.  local.sin_port = htons(iPort);
124.  if(bind(sListen,(struct sockaddr *) &local,sizeof(local)) == SOCKET_ERROR)
125.  {
126.   printf("bind() failed:%d/n",WSAGetLastError());
127.   return 1;
128.  }
129.  else
130.   printf("bind() successful!/n");
131.  int listenState = listen(sListen, 8);
132.  printf("%d/n",listenState);
133.  while(1)
134.  {
135.   iAddrSize = sizeof(client);
136.   sClient = accept(sListen,(struct sockaddr *) &client,&iAddrSize);
137.   if(sClient ==INVALID_SOCKET)
138.   {
139.    printf("accept() failed:%d/n",WSAGetLastError());
140.    break;
141.   }
142.   printf("Accept client:%s:%d/n",inet_ntoa(client.sin_addr),ntohs(client.sin_port));
143.   hThread = ::CreateThread(NULL,0,ClientThread,(LPVOID)sClient,0,&dwThreadId);
144.   if(hThread == NULL)
145.   {
146.    printf("creatThread() failed:%d/n",GetLastError());
147.    break;
148.   }
149.   CloseHandle(hThread);
150.  }
151.  closesocket(sListen);
152.  WSACleanup();
153.  return 0;
154. }

client端程序：

1. #include <winsock2.h>
2. #include <stdio.h>
3. #include <stdlib.h>
5. #pragma comment(lib,"Ws2_32")
7. #define DEFAULT_COUNT 20
8. #define DEFAULT_PORT 5150
9. #define DEFAULT_BUFFER 2048
10. #define DEFAULT_MESSAGE "This is a test of the emergency broadcasting system"
12.  
14. char szServer[128],szMessage[1024];
15. int iPort = DEFAULT_PORT;
16. DWORD  dwCount = DEFAULT_COUNT;
17. BOOL bSendOnly = FALSE;
19. void usage()
20. {
21.  printf("usage:client  [-p:x] [-i:IP] [-o]/n/n");
22.  printf("       -p:x   Remote port to send to/n");
23.  printf("       -s:IP  server's IP address or host name/n");
24.  printf("       -n:x   number of times to send message/n");
25.  printf("       -o     send message only; don't receive/n");
26.  ExitProcess(1);
27. }
29. void ValidateArgs(int argc,char **argv)
30. {
31.  int i;
32.  for(i=1;i<argc;i++)
33.  {
34.   if((argv[i][0] == '-')||(argv[i][0] == '/'))
35.   {
36.    switch(tolower(argv[i][1]))
37.    {
38.    case 'p':
39.     if (strlen(argv[i])>3)
40.      iPort = atoi(&argv[i][3]);
41.     break;
42.    case 's':
43.     if(strlen(argv[i])>3)
44.      strcpy(szServer,&argv[i][3]);
45.     break;
46.    case 'n':
47.     if(strlen(argv[i])>3)
48.      dwCount = atol(&argv[1][3]);
49.     break;
50.    case 'o':
51.     bSendOnly = TRUE;
52.     break;
53.    default:
54.     usage();
55.     break;
56.    }
57.   }
58.  }
59. }
61. int main(int argc, char **argv)
62. {
63.  WSADATA wsd;
64.  SOCKET sClient;
65.  char szBuffer[DEFAULT_BUFFER];
66.  int ret,i;
67.  struct sockaddr_in server;
68.  struct hostent *host = NULL;
70.  ValidateArgs(argc,argv);
72.  if(WSAStartup(MAKEWORD(2,2),&wsd)!=0)
73.  {
74.   printf("failed to load winsock library!/n");
75.   return 1;
76.  }
77.  strcpy(szMessage,DEFAULT_MESSAGE);
78.  sClient= socket(AF_INET,SOCK_STREAM,IPPROTO_TCP);
79.  if(sClient == INVALID_SOCKET)
80.  {
81.   printf("socket() fialed: %d/n", WSAGetLastError());
82.   return 1;
83.  }
84.  server.sin_family = AF_INET;
85.  server.sin_port = htons(iPort);
86.  server.sin_addr.s_addr = inet_addr("192.168.0.105");
87.  if(server.sin_addr.s_addr == INADDR_NONE)
88.  {
89.   host = gethostbyname(szServer);
90.   if( host == NULL)
91.   {
92.    printf("Unable to resolve server:%s/n",szServer);
93.    return 1;
94.   }
95.   CopyMemory(&server.sin_addr,host->h_addr_list[0],host->h_length );
96.  }
98.  int m = connect(sClient,(struct sockaddr *)&server,sizeof(server));
99.  if(m==SOCKET_ERROR)
100.  {
101.   int errorCode = WSAGetLastError();
102.   printf("connect() fialed:%d/n",errorCode);
103.   return 1;
104.  }
105.  for(i =0;i < dwCount; i++)
106.  {
107.   ret = send(sClient,szMessage,strlen(szMessage),0);
108.   if(ret ==0)
109.    break;
110.   else if (ret==SOCKET_ERROR)
111.   {
112.    printf("send() fialed:%d/n",WSAGetLastError());
113.    break;
114.   }
115.    printf("send %d bytes/n",ret);
116.    if(!bSendOnly)
117.    {
118.     ret = recv(sClient,szBuffer,DEFAULT_BUFFER,0);
119.     if(ret ==0)
120.      break;
121.     else if (ret == SOCKET_ERROR)
122.     {
123.      printf("recv() fialed:%d/n",WSAGetLastError());
124.      break;
125.     }
126.     szBuffer[ret] = '/0';
127.     printf("recv[%d bytes]:'%S'=n",ret,szBuffer);
128.    }
129.   }
130.   closesocket(sClient);
131.   WSACleanup();
132.   return 0 ; 
133.  }
134.