基于Windows的套接字相关函数及示例

链接ws2_32.lib库

头文件#include <winsock2.h>

1  int WSAStartup(WORD wVersionRequested,LPWSADATA lpWSAData);

初始化Winsock库

wVersionRequested：Winsock版本信息，用MAKEWORD宏构造，MAKEWORD(2,2)。

lpWSAData：WSADATA结构体变量的地址。无特殊含义，但是要传递结构体变量地址。

1 int WSACleanup();

注销Winsock相关库，成功返回0，失败返回SOCKET_ERROR。

1 SOCKET socket(int af, int type, int protocol);

成功返回套接字句柄，失败返回INVALID_SOCKET。

af：套接字中使用的协议族信息，常用的是PF_INET，IPv4互联网协议族。

type：套接字数据传输类型信息。有面向连接的套接字(SOCK_STREAM)，面向消息的套接字(SOCK_DGRAM)。

protocol：计算机间通信中使用的协议信息。选择最终协议，IPv4网络协议族中面向连接的协议(IPPROTO_TCP)，面向消息的协议(IPPROTO_UDP)。

1 int bind(SOCKET s, struct sockaddr * name, int namelen);

成功返回0，失败返回SOCKET_ERROR。

1 int listen(SOCKET s, int backlog);

成功返回0，失败返回SOCKET_ERROR。

1 SOCKET accept(SOCKET s, struct sockaddr * addr, int * addrlen);

成功返回套接字句柄，失败返回INVALID_SOCKET。

1 int connect(SOCKET s, struct sockaddr * name, int namelen);

成功返回0，失败返回SOCKET_ERROR。

1 int closesocket(SOCKET s);

关闭套接字，成功返回0，失败返回SOCKET_ERROR。

1 int send(SOCKET s,const char * buf, int len, int flags);

数据传输函数，成功返回传输字节数，失败返回SOCKET_ERROR。

s：数据传输对象连接的套接字句柄值。

buf：保存待传输数据的缓冲地址值。

len：要传输的字节数。

flags：传输数据时要用到的多种选项信息。

1 int recv(SOCKET s, const char * buf, int len, int flags);

成功时返回接受的字节数，收到EOF时为0，失败时返回SOCKET_ERROR。

s：数据接收对象连接的套接字句值。

buf：保存接收数据的缓冲地址值。

len：能够接受的最大字节数。

flags：接受数据时用到的多种选项信息。

 1 #include <WinSock2.h>
 2 #include <iostream>
 3 
 4 namespace WinSock {
 5 
 6     void ErrorHandling(char* message) {
 7         std::cerr << message << std::endl;
 8         exit(1);
 9     }
10 
11     int Server(int port) {
12         WSADATA wsaData;
13         SOCKET hServSock, hClntSock;
14         SOCKADDR_IN servAddr, clntAddr;
15         int szClntAddr;
16         char message[] = "Hello World!";
17         if (WSAStartup(MAKEWORD(2, 2), &wsaData) != 0) ErrorHandling("WSAStartup() error");
18         hServSock = socket(PF_INET, SOCK_STREAM, 0);
19         if (hServSock == INVALID_SOCKET) ErrorHandling("socket() error");
20         memset(&servAddr, 0, sizeof(servAddr));
21         servAddr.sin_family = AF_INET;
22         servAddr.sin_addr.S_un.S_addr = htonl(INADDR_ANY);
23         servAddr.sin_port = htons(port);
24         if (bind(hServSock, (SOCKADDR*)&servAddr, sizeof(servAddr)) == SOCKET_ERROR) ErrorHandling("bind() error");
25         if (listen(hServSock, 5) == SOCKET_ERROR) ErrorHandling("listen() error");
26         szClntAddr = sizeof(clntAddr);
27         hClntSock = accept(hServSock, (SOCKADDR*)&clntAddr, &szClntAddr);
28         if (hClntSock == INVALID_SOCKET) ErrorHandling("accept() error");
29         send(hClntSock, message, sizeof(message), 0);
30         closesocket(hClntSock);
31         closesocket(hServSock);
32         WSACleanup();
33         return 0;
34     }
35 
36     int Client(char ip[], int port) {
37         WSADATA wsaData;
38         SOCKET hSocket;
39         SOCKADDR_IN servAddr;
40         char message[30];
41         int strLen;
42         if (WSAStartup(MAKEWORD(2, 2), &wsaData) != 0) ErrorHandling("WSAStartup() error");
43         hSocket = socket(PF_INET, SOCK_STREAM, 0);
44         if (hSocket == INVALID_SOCKET) ErrorHandling("socket() error");
45         memset(&servAddr, 0, sizeof(servAddr));
46         servAddr.sin_family = AF_INET;
47         servAddr.sin_addr.S_un.S_addr = inet_addr(ip);
48         servAddr.sin_port = htons(port);
49         if (connect(hSocket, (SOCKADDR*)&servAddr, sizeof(servAddr)) == SOCKET_ERROR) ErrorHandling("connect() error");
50         strLen = recv(hSocket, message, sizeof(message) - 1, 0);
51         if (strLen == -1) ErrorHandling("read() error");
52         std::cout << "Message from server: " << message << std::endl;
53         closesocket(hSocket);
54         WSACleanup();
55         return 0;
56     }
57 
58 };

基于Windows的套接字相关函数及示例

推荐阅读