【网络】Windows 下 socket 编程范例

  1 #include "winsock2.h"
  2 #include <stdint.h>
  3 #include <stdio.h>
  4 
  5 #include <ctype.h>
  6 #include <time.h>
  7 
  8 #define MAX_BUF_LEN 1024
  9 
 10 #define ERR (-1)
 11 #define OK  0
 12 
 13 enum {
 14     TCP = 6,
 15     UDP = 17,
 16 };
 17 
 18 struct ClientPeer;
 19 typedef int (*ClientVoidHandler)(struct ClientPeer *);
 20 typedef int (*ClientRecvHandler)(struct ClientPeer *peer, char *buf, int len);
 21 typedef int (*ClientSendHandler)(struct ClientPeer *peer, const char *buf, int len);
 22 
 23 struct ClientPeerOp {
 24     ClientVoidHandler createsocket;
 25     ClientVoidHandler connect;
 26     ClientRecvHandler recv;
 27     ClientSendHandler send;
 28     ClientVoidHandler close;
 29 };
 30 
 31 struct ClientPeer {
 32     ClientPeerOp op;
 33 
 34     SOCKET fd;
 35 
 36     struct sockaddr_in remote;
 37 };
 38 
 39 int PeerEnvInit(void)
 40 {
 41     WORD sockVersion = MAKEWORD(2, 2);
 42     WSADATA wsaData;
 43     if (WSAStartup(sockVersion, &wsaData) != 0) {
 44         return ERR;
 45     }
 46     return OK;
 47 }
 48 
 49 int PeerEnvDestroy(void)
 50 {
 51     WSACleanup();
 52     return OK;
 53 }
 54 
 55 struct ClientPeer *ClientPeerNewWithRemote(const char *ip, unsigned short port)
 56 {
 57     struct ClientPeer *peer = (struct ClientPeer *)calloc(1, sizeof(struct ClientPeer));
 58 
 59     peer->remote.sin_family = AF_INET;
 60     peer->remote.sin_port = htons(port);
 61     peer->remote.sin_addr.S_un.S_addr = inet_addr(ip);
 62 
 63     return peer;
 64 };
 65 
 66 int TCPClientPeerCreateSocket(struct ClientPeer *peer)
 67 {
 68     SOCKET fd = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
 69     if (fd == INVALID_SOCKET) {
 70         printf("socket error !");
 71         return ERR;
 72     }
 73     peer->fd = fd;
 74     return OK;
 75 }
 76 
 77 int TCPClientPeerConnect(struct ClientPeer *peer)
 78 {
 79     if (connect(peer->fd, (struct sockaddr *)&(peer->remote), sizeof(struct sockaddr)) == SOCKET_ERROR) {
 80         printf("connect error !");
 81         return ERR;
 82     }
 83     return OK;
 84 }
 85 
 86 int TCPClientPeerSend(struct ClientPeer *peer, const char *buf, int len)
 87 {
 88     return send(peer->fd, buf, len, 0);
 89 }
 90 
 91 int TCPClientPeerRecv(struct ClientPeer *peer, char *buf, int len)
 92 {
 93     return recv(peer->fd, buf, len, 0);
 94 }
 95 
 96 int TCPClientPeerClose(struct ClientPeer *peer)
 97 {
 98     closesocket(peer->fd);
 99     return OK;
100 }
101 
102 
103 int UDPClientPeerCreateSocket(struct ClientPeer *peer)
104 {
105     SOCKET fd = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP);
106     if (fd == INVALID_SOCKET) {
107         printf("socket error !");
108         return ERR;
109     }
110     peer->fd = fd;
111     return OK;
112 }
113 
114 int UDPClientPeerConnect(struct ClientPeer *peer)
115 {
116     return OK;
117 }
118 
119 int UDPClientPeerSend(struct ClientPeer *peer, const char *buf, int len)
120 {
121     return sendto(peer->fd, buf, len, 0, (struct sockaddr *)&(peer->remote), sizeof(struct sockaddr));
122 }
123 
124 int UDPClientPeerRecv(struct ClientPeer *peer, char *buf, int len)
125 {
126     int addrlen = sizeof(struct sockaddr_in);
127     return recvfrom(peer->fd, buf, len, 0, (struct sockaddr *)&(peer->remote), &addrlen);
128 }
129 
130 int UDPClientPeerClose(struct ClientPeer *peer)
131 {
132     closesocket(peer->fd);
133     return OK;
134 }
135 
136 struct ClientPeer *ClientPeerFactory(int type, const char *ip, unsigned short port)
137 {
138     struct ClientPeer *peer = ClientPeerNewWithRemote(ip, port);
139 
140     switch (type) {
141         case TCP:
142             peer->op.createsocket = TCPClientPeerCreateSocket;
143             peer->op.connect = TCPClientPeerConnect;
144             peer->op.recv = TCPClientPeerRecv;
145             peer->op.send = TCPClientPeerSend;
146             peer->op.close = TCPClientPeerClose;
147             break;
148 
149         case UDP:
150             peer->op.createsocket = UDPClientPeerCreateSocket;
151             peer->op.connect = UDPClientPeerConnect;
152             peer->op.recv = UDPClientPeerRecv;
153             peer->op.send = UDPClientPeerSend;
154             peer->op.close = UDPClientPeerClose;
155             break;
156     }
157 
158     return peer;
159 }
160 
161 
162 
163 
164 struct ServerPeer;
165 typedef int (*ServerVoidHandler)(struct ServerPeer *);
166 typedef int (*ServerRecvHandler)(struct ServerPeer *peer, char *buf, int len);
167 typedef int (*ServerSendHandler)(struct ServerPeer *peer, const char *buf, int len);
168 
169 struct ServerPeerOp {
170     ServerVoidHandler createsocket;
171     ServerVoidHandler bindaddr;
172     ServerVoidHandler listen;
173     ServerVoidHandler accept;
174     ServerRecvHandler recv;
175     ServerSendHandler send;
176     ServerVoidHandler closeclient;
177     ServerVoidHandler close;
178 };
179 
180 struct ServerPeer {
181     SOCKET lfd;
182     SOCKET rfd;
183 
184     struct sockaddr_in laddr;
185     struct sockaddr_in raddr;
186 
187     struct ServerPeerOp op;
188 };
189 
190 
191 struct ServerPeer *ServerPeerNewWithAddr(const char *ip, unsigned short port)
192 {
193     struct ServerPeer *peer = (struct ServerPeer *)calloc(1, sizeof(struct ServerPeer));
194 
195     peer->laddr.sin_family = AF_INET;
196     peer->laddr.sin_port = htons(port);
197     peer->laddr.sin_addr.S_un.S_addr = inet_addr(ip);
198 
199     return peer;
200 }
201 
202 int TCPServerPeerCreateSocket(struct ServerPeer *peer)
203 {
204 
205     SOCKET fd = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
206     if (fd == INVALID_SOCKET) {
207         printf("socket error !");
208         return ERR;
209     }
210     peer->lfd = fd;
211     return OK;
212 }
213 
214 int TCPServerPeerBindaddr(struct ServerPeer *peer)
215 {
216     if (bind(peer->lfd, (struct sockaddr *)&(peer->laddr), sizeof(struct sockaddr)) == SOCKET_ERROR) {
217         printf("bind error !");
218         return ERR;
219     }
220 
221     return OK;
222 }
223 
224 int TCPServerPeerListen(struct ServerPeer *peer)
225 {
226     if (listen(peer->lfd, 5) == SOCKET_ERROR) {
227         printf("listen error !");
228         return ERR;
229     }
230 
231     return OK;
232 }
233 
234 int TCPServerPeerAccept(struct ServerPeer *peer)
235 {
236     int len = sizeof(struct sockaddr_in);
237     peer->rfd = accept(peer->lfd, (struct sockaddr *)&(peer->raddr), &len);
238     if (peer->rfd == INVALID_SOCKET) {
239         printf("accept error !");
240         return ERR;
241     }
242     return OK;
243 }
244 
245 int TCPServerPeerSend(struct ServerPeer *peer, const char *buf, int len)
246 {
247     return send(peer->rfd, buf, len, 0);
248 }
249 
250 int TCPServerPeerRecv(struct ServerPeer *peer, char *buf, int len)
251 {
252     return recv(peer->rfd, buf, len, 0);
253 }
254 
255 int TCPServerPeerCloseClient(struct ServerPeer *peer)
256 {
257     closesocket(peer->rfd);
258     return OK;
259 }
260 
261 int TCPServerPeerClose(struct ServerPeer *peer)
262 {
263     closesocket(peer->lfd);
264     return OK;
265 }
266 
267 
268 
269 
270 
271 
272 
273 
274 int UDPServerPeerCreateSocket(struct ServerPeer *peer)
275 {
276 
277     SOCKET fd = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP);
278     if (fd == INVALID_SOCKET) {
279         printf("socket error !");
280         return ERR;
281     }
282     peer->lfd = fd;
283     return OK;
284 }
285 
286 int UDPServerPeerBindaddr(struct ServerPeer *peer)
287 {
288     if (bind(peer->lfd, (struct sockaddr *)&(peer->laddr), sizeof(struct sockaddr)) == SOCKET_ERROR) {
289         printf("bind error !");
290         return ERR;
291     }
292 
293     return OK;
294 }
295 
296 int UDPServerPeerListen(struct ServerPeer *peer)
297 {
298     return OK;
299 }
300 
301 int UDPServerPeerAccept(struct ServerPeer *peer)
302 {
303     return OK;
304 }
305 
306 int UDPServerPeerSend(struct ServerPeer *peer, const char *buf, int len)
307 {
308     return sendto(peer->lfd, buf, len, 0, (struct sockaddr *)&(peer->raddr), sizeof(struct sockaddr_in));
309 }
310 
311 int UDPServerPeerRecv(struct ServerPeer *peer, char *buf, int len)
312 {
313     int addrlen = sizeof(struct sockaddr_in);
314     return recvfrom(peer->lfd, buf, len, 0, (struct sockaddr *)&(peer->raddr), &addrlen);
315 }
316 
317 int UDPServerPeerCloseClient(struct ServerPeer *peer)
318 {
319     return OK;
320 }
321 
322 int UDPServerPeerClose(struct ServerPeer *peer)
323 {
324     closesocket(peer->lfd);
325     return OK;
326 }
327 
328 
329 struct ServerPeer *ServerPeerFactory(int type, const char *ip, unsigned short port)
330 {
331     struct ServerPeer *peer = ServerPeerNewWithAddr(ip, port);
332 
333     switch (type) {
334         case TCP:
335             peer->op.createsocket = TCPServerPeerCreateSocket;
336             peer->op.bindaddr = TCPServerPeerBindaddr;
337             peer->op.listen = TCPServerPeerListen;
338             peer->op.accept = TCPServerPeerAccept;
339             peer->op.recv = TCPServerPeerRecv;
340             peer->op.send = TCPServerPeerSend;
341             peer->op.closeclient = TCPServerPeerCloseClient;
342             peer->op.close = TCPServerPeerClose;
343             break;
344 
345         case UDP:
346             peer->op.createsocket = UDPServerPeerCreateSocket;
347             peer->op.bindaddr = UDPServerPeerBindaddr;
348             peer->op.listen = UDPServerPeerListen;
349             peer->op.accept = UDPServerPeerAccept;
350             peer->op.recv = UDPServerPeerRecv;
351             peer->op.send = UDPServerPeerSend;
352             peer->op.closeclient = UDPServerPeerCloseClient;
353             peer->op.close = UDPServerPeerClose;
354             break;
355     }
356 
357     return peer;
358 }
359 
360 
361 struct Peer;
362 typedef void (*runhandler)(struct Peer *);
363 
364 
365 
366 struct Peer {
367     union {
368         ClientPeer *client;
369         ServerPeer *server;
370     };
371     runhandler op;
372 };
373 
374 void ServerRun(struct Peer *p)
375 {
376     struct ServerPeer *peer = p->server;
377     struct ServerPeerOp *op = &(peer->op);
378 
379     op->createsocket(peer);
380     op->bindaddr(peer);
381     op->listen(peer);
382     op->accept(peer);
383     while (1) {
384         int ret;
385         char buf[MAX_BUF_LEN];
386         time_t t = time(NULL);
387 
388         memset(buf, 0, sizeof(buf));
389         op->recv(peer, buf, sizeof(buf));
390 
391         printf("server recv: %s\n", buf);
392 
393         strncpy(buf, ctime(&t), sizeof(buf));
394         ret = op->send(peer, buf, strlen(buf));
395         if (ret <= 0) {
396             printf("send failed(%d - %d).\n", ret, WSAGetLastError());
397             break;
398         }
399     }
400 }
401 
402 void ClientRun(struct Peer *p)
403 {
404     struct ClientPeer *peer = p->client;
405     struct ClientPeerOp *op = &(peer->op);
406 
407     op->createsocket(peer);
408     op->connect(peer);
409 
410     char hello[] = "Hello, I am client.\n";
411 
412     op->send(peer, hello, strlen(hello));
413 
414     while (1) {
415         char buf[MAX_BUF_LEN] = {0};
416         int len = 0;
417 
418         len = op->recv(peer, buf, sizeof(buf));
419         if (len <= 0) {
420             printf("recv error (%d - %d) !\n", len, WSAGetLastError());
421             break;
422         }
423         printf("client recv: %s\n", buf);
424 
425         for (int i = 0; i < len; i++) {
426             buf[i] = toupper(buf[i]);
427         }
428         op->send(peer, buf, strlen(buf));
429 
430         Sleep(1000);
431     }
432 
433     op->close(peer);
434 }
435 
436 
437 enum {
438     TCP_SERVER = 1,
439     TCP_CLIENT,
440     UDP_SERVER,
441     UDP_CLIENT,
442 };
443 
444 struct Peer *PeerFactory(int type, const char *ip, unsigned short port)
445 {
446     struct Peer *peer = (struct Peer *)calloc(1, sizeof(struct Peer));
447 
448     switch (type) {
449         case TCP_SERVER:
450             peer->server = ServerPeerFactory(TCP, ip, port);
451             peer->op = ServerRun;
452             break;
453         case TCP_CLIENT:
454             peer->client = ClientPeerFactory(TCP, ip, port);
455             peer->op = ClientRun;
456             break;
457         case UDP_SERVER:
458             peer->server = ServerPeerFactory(UDP, ip, port);
459             peer->op = ServerRun;
460             break;
461         case UDP_CLIENT:
462             peer->client = ClientPeerFactory(UDP, ip, port);
463             peer->op = ClientRun;
464             break;
465     }
466 
467     return peer;
468 };
469 
470 
471 void PeerRun(struct Peer *peer)
472 {
473     peer->op(peer);
474 }
475 
476 
477 int main()
478 {
479     int type;
480     printf("input way:              \n\
481               1. tcp server         \n482               2. tcp client         \n483               3. udp server         \n484               4. udp client         \n485            ");
486     scanf("%d", &type);
487 
488     const char *ip = "127.0.0.1";
489     unsigned short port = 10240;
490 
491     PeerEnvInit();
492 
493     struct Peer *peer = NULL;
494 
495     switch(type) {
496         case 1:
497         case 2:
498         case 3:
499         case 4:
500             peer = PeerFactory(type, ip, port);
501             PeerRun(peer);
502             break;
503         default:
504             printf("unknown.");
505             break;
506     }
507 
508     PeerEnvDestroy();
509 }