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

链接ws2_32.lib库

头文件#include <winsock2.h>

  int WSAStartup(WORD wVersionRequested,LPWSADATA lpWSAData);

初始化Winsock库

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

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

 int WSACleanup();

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

 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)。

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

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

 int listen(SOCKET s, int backlog);

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

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

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

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

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

 int closesocket(SOCKET s);

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

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

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

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

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

len：要传输的字节数。

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

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

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

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

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

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

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

 #include <WinSock2.h>
 #include <iostream>

 namespace WinSock {

     void ErrorHandling(char* message) {
         std::cerr << message << std::endl;
         exit();
     }

     int Server(int port) {
         WSADATA wsaData;
         SOCKET hServSock, hClntSock;
         SOCKADDR_IN servAddr, clntAddr;
         int szClntAddr;
         char message[] = "Hello World!";
         if (WSAStartup(MAKEWORD(, ), &wsaData) != ) ErrorHandling("WSAStartup() error");
         hServSock = socket(PF_INET, SOCK_STREAM, );
         if (hServSock == INVALID_SOCKET) ErrorHandling("socket() error");
         memset(&servAddr, , sizeof(servAddr));
         servAddr.sin_family = AF_INET;
         servAddr.sin_addr.S_un.S_addr = htonl(INADDR_ANY);
         servAddr.sin_port = htons(port);
         if (bind(hServSock, (SOCKADDR*)&servAddr, sizeof(servAddr)) == SOCKET_ERROR) ErrorHandling("bind() error");
         if (listen(hServSock, ) == SOCKET_ERROR) ErrorHandling("listen() error");
         szClntAddr = sizeof(clntAddr);
         hClntSock = accept(hServSock, (SOCKADDR*)&clntAddr, &szClntAddr);
         if (hClntSock == INVALID_SOCKET) ErrorHandling("accept() error");
         send(hClntSock, message, sizeof(message), );
         closesocket(hClntSock);
         closesocket(hServSock);
         WSACleanup();
         return ;
     }

     int Client(char ip[], int port) {
         WSADATA wsaData;
         SOCKET hSocket;
         SOCKADDR_IN servAddr;
         char message[];
         int strLen;
         if (WSAStartup(MAKEWORD(, ), &wsaData) != ) ErrorHandling("WSAStartup() error");
         hSocket = socket(PF_INET, SOCK_STREAM, );
         if (hSocket == INVALID_SOCKET) ErrorHandling("socket() error");
         memset(&servAddr, , sizeof(servAddr));
         servAddr.sin_family = AF_INET;
         servAddr.sin_addr.S_un.S_addr = inet_addr(ip);
         servAddr.sin_port = htons(port);
         if (connect(hSocket, (SOCKADDR*)&servAddr, sizeof(servAddr)) == SOCKET_ERROR) ErrorHandling("connect() error");
         strLen = recv(hSocket, message, sizeof(message) - , );
         if (strLen == -) ErrorHandling("read() error");
         std::cout << "Message from server: " << message << std::endl;
         closesocket(hSocket);
         WSACleanup();
         return ;
     }

 };