1//程序清单6-1 回应服务器代码
  2// 编译命令Compile:cl -o Server Server.c ws2_32.lib
  3//
  4// 命令行选项:
  5// server [-p:x] [-i:IP] [-o]
  6// -p:x 监听的端口号
  7// -i:str 监听的网卡
  8// -o 只接收,不回显数据
  9//
 10
 11
 12
 13#include <winsock2.h>
 14#include <stdio.h>
 15#include <stdlib.h>
 16
 17#pragma  comment(lib, "Ws2_32.lib ")
 18
 19#define DEFAULT_PORT 5150
 20#define DEFAULT_BUFFER 4096
 21
 22int iPort = DEFAULT_PORT; // 监听客户端的端口
 23BOOL bInterface = FALSE, // 在指定网卡上监听
 24bRecvOnly = FALSE; // 只接收,不回显
 25char szAddress[128];
 26
 27
 28
 29void usage() ;
 30void ValidateArgs(int argc, char **argv) ;
 31DWORD WINAPI ClientThread(LPVOID lpParam) ;
 32
 33
 34// 函数:main
 35// 说明:执行主线程、初始化Winsock、解释命令行参数、创建监听套接字、捆绑到本地地址、
 36//等待
 37// 客户端连接
 38int main(int argc, char **argv)
 39{
 40    WSADATA wsd;
 41
 42    SOCKET sListen,sClient;
 43
 44    int iAddrSize;
 45
 46
 47    HANDLE hThread;
 48
 49    DWORD dwThreadId;
 50
 51    struct sockaddr_in local, client;
 52
 53    int i = 0 ;
 54
 55    printf("程序开始:\n");
 56
 57//    ValidateArgs(argc, argv);
 58
 59    if (WSAStartup(MAKEWORD(2,2), &wsd) != 0)
 60    {
 61        printf("Failed to load Winsock!\n");
 62        return 1;
 63    }

 64    // 创建监听套接字
 65    sListen = socket(AF_INET, SOCK_STREAM, IPPROTO_IP);
 66
 67    if (sListen == SOCKET_ERROR)
 68    {
 69        printf("socket() failed: %d\n", WSAGetLastError());
 70        return 1;
 71    }

 72    // 选择本地网络接口卡,并且捆绑到它上面
 73    if (bInterface)
 74    {
 75        local.sin_addr.s_addr = inet_addr(szAddress);
 76        if (local.sin_addr.s_addr == INADDR_NONE)
 77            usage();
 78    }

 79    else
 80        local.sin_addr.s_addr = htonl(INADDR_ANY);
 81
 82
 83    local.sin_family = AF_INET;
 84    local.sin_port = htons(iPort);
 85    if (bind(sListen, (struct sockaddr *)&local,sizeof(local)) == SOCKET_ERROR)
 86    {
 87        printf("bind() failed: %d\n", WSAGetLastError());
 88        return 1;
 89    }

 90    listen(sListen, 8);
 91
 92    // 循环等待进入的客户端连接,一旦检查到连接,创建线程并且传递句柄
 93    printf("主程序循环前:\n");
 94    while (1)
 95    {
 96        printf("主程序第%d循环\n",i);
 97
 98        iAddrSize = sizeof(client);
 99
100        sClient = accept(sListen, (struct sockaddr *)&client,&iAddrSize);   //会挂起,直到有连接
101
102        if (sClient == INVALID_SOCKET)
103        {
104            printf("accept() failed: %d\n", WSAGetLastError());
105            break;
106        }

107        printf("Accepted client: %s:%d\n",inet_ntoa(client.sin_addr), ntohs(client.sin_port));
108
109        hThread = CreateThread(NULL, 0, ClientThread,(LPVOID)sClient, 0&dwThreadId);  //创建处理纯种
110
111        if (hThread == NULL)
112        {
113            printf("CreateThread() failed: %d\n", GetLastError());
114            break;
115        }

116        CloseHandle(hThread);
117
118        i++ ;
119    }

120    closesocket(sListen);
121
122
123    WSACleanup();
124
125    printf("程序结束:\n");
126
127    return 0;
128}

129
130
131// 函数:ClientThread
132// 说明:函数以线程方式调用,并且处理给定客户端连接,传递参数为从accept()调用中返回
133//的套接
134// 字句柄,该函数从客户端读取数据,并且将读到的数据写回去
135DWORD WINAPI ClientThread(LPVOID lpParam)
136{
137    SOCKET sock=(SOCKET)lpParam;
138    char szBuff[DEFAULT_BUFFER];
139    int ret,
140        nLeft,
141        idx;
142
143    printf("Client开始:\n");
144
145    while(1)
146    {
147        // 阻塞recv()调用
148
149        printf("recv开始:\n");
150
151        ret = recv(sock, szBuff, DEFAULT_BUFFER, 0);
152        if (ret == 0// 常规关闭
153            break;
154        else if (ret == SOCKET_ERROR)
155        {
156            printf("recv() failed: %d\n", WSAGetLastError());
157            break;
158        }

159        szBuff[ret] = '\0';
160        printf("RECV: '%s'\n", szBuff);
161        //
162        // 选择回显数据
163        if (!bRecvOnly)
164        {
165            nLeft = ret;
166            idx = 0;
167            // 确保写所有数据
168            while(nLeft > 0)
169            {
170                ret = send(sock, &szBuff[idx], nLeft, 0);
171                if (ret == 0)
172                    break;
173                else if (ret == SOCKET_ERROR)
174                    
175                {
176                    printf("send() failed: %d\n",
177                        WSAGetLastError());
178                    break;
179                }

180                nLeft -= ret;
181                idx += ret;
182            }

183        }

184
185    }

186
187    printf("Client结束:\n");
188
189    return 0;
190}

191
192
193
194// 监听客户端的网卡
195// 函数:usage
196// 说明:打印使用信息,并且退出
197
198
199void usage()
200{
201    printf("usage: server [-p:x] [-i:IP] [-o]\n\n");
202    printf(" -p:x 监听的端口号\n");
203    printf(" -i:str 监听的网卡\n");
204    printf(" -o 只接收,不回显数据\n\n");
205    ExitProcess(1);
206}

207// 函数:ValidateArgs
208// 说明:解释命令行参数,设置指示操作如何进行的全局变量
209void ValidateArgs(int argc, char **argv)
210{
211    int i;
212
213
214    printf("解释命令行参数程序开始:\n");
215    for(i = 1; i < argc; i++)
216    {
217        if ((argv[i][0== '-'|| (argv[i][0== '/'))
218        {
219            switch (tolower(argv[i][1]))
220            {
221            case 'p':
222                iPort = atoi(&argv[i][3]);
223                break;
224            case 'i':
225                
226                    bInterface = TRUE;
227                if (strlen(argv[i]) > 3)
228                    strcpy(szAddress, &argv[i][3]);
229                break;
230            case 'o':
231                bRecvOnly = TRUE;
232                break;
233            default:
234                usage();
235                break;
236            }

237        }

238    }

239    printf("解释命令行参数程序结束:\n");
240}

241
242
243
244
245



  1
  2/*
  3
  4程序清单6-2 是客户端代码,客户端建立一个套接字,并对投入应用的服务器名进行解析,然后
  5与服务器建立连接。连接一旦建成,就可发送大量的消息了。每次发送数据之后,客户端都会等待服
  6务器发回的回应。客户端把得自套接字的数据打印出来。
  7回应客户端和服务器不能完全说明TCP 协议的流式传输。这是因为读取操作是在写操作之后进行
  8的,至少客户端这一端是这样的。当然,对服务器来说,还有另一种方式。因此,服务器每次调用读
  9取函数,一般都会返回客户端发出的整条消息。但不要误会,如果客户端的消息大到超过了TCP 的最
 10大传输单元,在线上,它会被分成几个小的数据包,这种情况下,接收端需要多次执行接收调用,才
 11能收完整条消息。为了更好地说明流式传输,运行客户端和服务器时带上-O 选项即可。这样,客户端
 12便只管发送数据,接收端只管读取数据。
 13
 14          
 15            
 16服务器如下执行:
 17server -p:5150 -o
 18而客户端如下执行:
 19client -p:5150 -s:IP -n:10 -o
 20大家最可能见到的是客户端进行了10 次send 调用,而服务器在一次或两次recv 调用中,就读
 21取了10 条消息。
 22
 23
 24*/

 25
 26
 27
 28//程序清单6-2 回应客户端代码
 29
 30
 31// 说明:回显客户端,连接TCP 服务器,发送数据,并且读服务器返回的数据
 32// 编译命令:cl -o Client Client.c ws2_32.lib
 33//
 34// 命令行参数:
 35// client [-p:x] [-s:IP] [-n:x] [-o]
 36// -p:x 发送的远程端口
 37// -s:IP 服务器IP 地址或主机名
 38// -n:x 发送消息次数
 39// -o 只发送消息,不接收
 40//
 41
 42#include <winsock2.h>
 43#include <stdio.h>
 44#include <stdlib.h>
 45
 46#pragma  comment(lib, "Ws2_32.lib ")
 47
 48#define DEFAULT_COUNT 1000
 49#define DEFAULT_PORT 5150
 50#define DEFAULT_BUFFER 2048
 51#define DEFAULT_MESSAGE "This is a test of the emergency \
 52broadcasting system"
 53
 54
 55char szServer[128], // 连接的服务器
 56
 57szMessage[1024]; // 发送给服务器的消息
 58
 59int iPort = DEFAULT_PORT; // 连接到服务器的端口
 60
 61DWORD dwCount = DEFAULT_COUNT; // 发送消息次数
 62
 63BOOL bSendOnly = FALSE; // 只发送数据,不接收
 64
 65
 66 void usage() ;
 67 void ValidateArgs(int argc, char **argv) ;
 68
 69// 函数:main
 70// 说明:执行主线程,初始化Winsock,解释命令行参数,创建套接字,连接服务器,然后发送
 71//和接
 72// 收数据
 73int main(int argc, char **argv)
 74{
 75    WSADATA wsd;
 76    SOCKET sClient;
 77    char szBuffer[DEFAULT_BUFFER];
 78    int ret,i;
 79
 80    struct sockaddr_in server;
 81
 82    struct hostent *host = NULL;
 83
 84    // 解释命令行并且载入Winsock
 85    ValidateArgs(argc, argv);
 86
 87    if (WSAStartup(MAKEWORD(2,2), &wsd) != 0)
 88    {
 89        printf("Failed to load Winsock library!\n");
 90        return 1;
 91    }

 92    strcpy(szMessage, DEFAULT_MESSAGE);
 93    // 创建套接字,并且尝试连接服务器
 94    sClient = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
 95    if (sClient == INVALID_SOCKET)
 96    {
 97        printf("socket() failed: %d\n", WSAGetLastError());
 98        return 1;
 99    }

100    server.sin_family = AF_INET;
101    server.sin_port = htons(iPort);
102    server.sin_addr.s_addr = inet_addr(szServer);
103    
104        // 如果提供的服务器地址不是形如"aaa.bbb.ccc.ddd",则为主机名,尝试解析它
105        if (server.sin_addr.s_addr == INADDR_NONE)
106        {
107            host = gethostbyname(szServer);
108            if (host == NULL)
109            {
110                printf("Unable to resolve server: %s\n", szServer);
111                return 1;
112            }

113
114            CopyMemory(&server.sin_addr, host->h_addr_list[0],
115                host->h_length);
116        }

117        if (connect(sClient, (struct sockaddr *)&server,sizeof(server)) == SOCKET_ERROR)
118        {
119            printf("connect() failed: %d\n", WSAGetLastError());
120            return 1;
121        }

122
123        // 发送和接收数据
124        for(i = 0; i < dwCount; i++)
125        {
126            ret = send(sClient, szMessage, strlen(szMessage), 0);
127            if (ret == 0)
128                break;
129            else if (ret == SOCKET_ERROR)
130            {
131                printf("send() failed: %d\n", WSAGetLastError());
132                break;
133            }

134            printf("Send %d bytes\n", ret);
135            if (!bSendOnly)
136            {
137                ret = recv(sClient, szBuffer, DEFAULT_BUFFER, 0);
138                if (ret == 0// Graceful close
139                    break;
140                else if (ret == SOCKET_ERROR)
141                {
142                    printf("recv() failed: %d\n", WSAGetLastError());
143                    break;
144                }

145                szBuffer[ret] = '\0';
146                printf("RECV [%d bytes]: '%s'\n", ret, szBuffer);
147            }

148        }

149        closesocket(sClient);
150        WSACleanup();
151        return 0;
152}

153
154
155
156
157 void usage()
158{
159    printf("usage: client [-p:x] [-s:IP] [-n:x] [-o]\n\n");
160    printf(" -p:x Remote port to send to\n");
161    printf(" -s:IP Server's IP address or hostname\n");
162    printf(" -n:x Number of times to send message\n");
163    printf(" -o Send messages only; don't receive\n");
164    ExitProcess(1);
165}

166
167
168// 函数:ValidateArgs
169// 说明:解释命令行参数,设置全局变量
170void ValidateArgs(int argc, char **argv)
171{
172    int i;
173    for(i = 1; i < argc; i++)
174    {
175        if ((argv[i][0== '-'|| (argv[i][0== '/'))
176        {
177            switch (tolower(argv[i][1]))
178            {
179            case 'p'// Remote port
180                if (strlen(argv[i]) > 3)
181                    iPort = atoi(&argv[i][3]);
182                break;
183                
184            case 's'// Server
185                if (strlen(argv[i]) > 3)
186                    strcpy(szServer, &argv[i][3]);
187                break;
188            case 'n'// Number of times to send message
189                if (strlen(argv[i]) > 3)
190                    dwCount = atol(&argv[i][3]);
191                break;
192            case 'o'// Only send message; don't receive
193                bSendOnly = TRUE;
194                break;
195            default:
196                usage();
197                break;
198            }

199        }

200    }

201}

202

只有注册用户登录后才能发表评论。
网站导航: 博客园   IT新闻   BlogJava   知识库   博问   管理


posts - 5, comments - 1, trackbacks - 0, articles - 3

Copyright © Seed-L