客户端:
新版本中,加入了#include< Ws2tcpip.h> #pragma comment(lib, "Ws2_32.lib")进行预编译
#include <iostream>
#include <string>
#include <WinSock2.h>
#include< Ws2tcpip.h>
#pragma comment(lib, "Ws2_32.lib")
class tcpClient
{
public:
tcpClient() {}
void buildClient()
{
std::cout << "### TCP/IP Client Demo ###" << std::endl
<< "# " << std::endl
<< "# run the program as: TcpClient " << std::endl
<< "# or: TcpClient <IPADDR>" << std::endl
<< "# or: TcpClient <IPADDR> <IPPORT>" << std::endl
<< "# the default IPADDR is \"127.0.0.1\"" << std::endl
<< "# the default IPPORT is \"6789\"" << std::endl
<< "# " << std::endl
<< "### TCP/IP Server Demo ###" << std::endl
<< std::endl;
//网络初始化
WSAData wsaData;
/*
socket编程中:
声明调用不同的Winsock版本。例如MAKEWORD(2,2)就是调用2.2版,
MAKEWORD(1,1)就是调用1.1版。
不同版本是有区别的,例如1.1版只支持TCP/IP协议,而2.0版可以支持多协议。
2.0版有良好的向后兼容性,任何使用1.1版的源代码、二进制文件、
应用程序都可以不加修改地在2.0规范下使用。此外winsock 2.0支持异步 1.1不支持异步.
*/
if (0 != WSAStartup(MAKEWORD(2, 2), &wsaData)) {
std::cout << "[E] Init network failed!" << std::endl
<< "# " << std::endl;
system("pause");
}
//客户端
/*
原型为:
#include <sys/types.h>
#include <sys/socket.h>
int socket(int domain, int type, int protocol);
1.domain表示协议簇,常用协议簇有:
AF_UNIX(本机通信)
AF_INET(TCP/IP – IPv4)
AF_INET6(TCP/IP – IPv6)
2.type表示套接字类型,常用套接字有:
SOCK_STREAM(TCP流)
SOCK_DGRAM(UDP数据报)
SOCK_RAW(原始套接字)
3.最后一个 “protocol”一般设置为“0”,
也就是当确定套接字使用的协议簇和类型时,这个参数的值就为0,
但是有时候创建原始套接字时,并不知道要使用的协议簇和类型,
也就是domain参数未知情况下,这时protocol这个参数就起作用了,
它可以确定协议的种类。
4.socket是一个函数,那么它也有返回值,当套接字创建成功时,返回套接字,失败返回“-1”,错误代码则写入“errno”中。
*/
SOCKET sockClient = socket(AF_INET, SOCK_STREAM, 0);
if (INVALID_SOCKET == sockClient) {
std::cout << "[E] Create socket failed! Error code is: " << WSAGetLastError() << std::endl
<< "# " << std::endl;
WSACleanup();
system("pause");
}
/*
1.sockaddr:
sockaddr在头文件#include <sys/socket.h>中定义,
sockaddr的缺陷是:sa_data把目标地址和端口信息混在一起了,如下:
struct sockaddr {
sa_family_t sin_family;//地址族
char sa_data[14]; //14字节,包含套接字中的目标地址和端口信息
};
2.sockaddr_in:
sockaddr_in在头文件#include<netinet/in.h>或#include <arpa/inet.h>中定义,
该结构体解决了sockaddr的缺陷,把port和addr 分开储存在两个变量中
sin_family:地址族
sin_port:端口
sin_addr:32位IP地址
sin_zero:不需要
*/
SOCKADDR_IN addrServer;
inet_pton(AF_INET, "127.0.0.1", &addrServer.sin_addr.s_addr);
//addrServer.sin_addr.s_addr = inet_addr("127.0.0.1") /*inet_addr(m_server_addr)*/;
addrServer.sin_family = AF_INET;
addrServer.sin_port = htons(8000)/*htons(m_server_port)*/;
//std::cout << "[I] Current addr is: " << m_server_addr << std::endl;
//std::cout << "[I] port is: " << m_server_port << std::endl;
int errcode;
errcode = connect(sockClient, (SOCKADDR*)&addrServer, sizeof(SOCKADDR));
if (SOCKET_ERROR == errcode) {
std::cout << "[E] Connect to socket failed! Error code is: " << WSAGetLastError() << std::endl
<< "# " << std::endl;
WSACleanup();
system("pause");
}
while (true) {
std::string strBuf;
char sendBuf[] = { 0 };
std::cout << "[I] Please input message: ";
std::cin >> strBuf;
send(sockClient, strBuf.data(), strBuf.length(), 0);
char recvBuf[100] = { 0 };
recv(sockClient, recvBuf, 100, 0);
std::cout << "[I] Receive data from server: " << recvBuf << std::endl
<< "# " << std::endl;
std::cout << std::endl;
}
closesocket(sockClient);
WSACleanup();
system("pause");
}
};
服务端:
#include <iostream>
#include <WinSock2.h>
#include< Ws2tcpip.h>
#pragma comment(lib, "Ws2_32.lib")
class tcpServer
{
public:
tcpServer() {}
void buildServer()
{
std::cout << "### TCP/IP Server Demo ###" << std::endl
<< "# " << std::endl
<< "# run the program as: TcpServer" << std::endl
<< "# or: TcpServer <IPPORT>" << std::endl
<< "# the default IPPORT is \"6789\"" << std::endl
<< "# " << std::endl
<< "### TCP/IP Server Demo ###" << std::endl
<< std::endl;
//初始化
WSADATA wsaData;
if (0 != WSAStartup(MAKEWORD(2, 2), &wsaData)) {
std::cout << "[E] Init network failed!" << std::endl;
system("pause");
}
SOCKET sockServer = socket(AF_INET, SOCK_STREAM, 0);
if (INVALID_SOCKET == sockServer) {
std::cout << "[E] Create socket failed! Error code is: " << WSAGetLastError() << std::endl;
WSACleanup();
system("pause");
}
SOCKADDR_IN addrServer;
addrServer.sin_addr.S_un.S_addr = ADDR_ANY;
addrServer.sin_family = AF_INET;
addrServer.sin_port = htons(8000);
//std::cout << "[I] Current port is " << m_server_port << std::endl
// << "# " << std::endl;
int errcode;
//和client不同,bind
errcode = bind(sockServer, (SOCKADDR*)&addrServer, sizeof(SOCKADDR));
if (SOCKET_ERROR == errcode) {
std::cout << "[E] Bind to socket failed! Error code is: " << WSAGetLastError() << std::endl
<< "# " << std::endl;
WSACleanup();
system("pause");
}
//开启监听
errcode = listen(sockServer, 5);
if (SOCKET_ERROR == errcode) {
std::cout << "[E] Listen the socket failed! Error code is: " << WSAGetLastError() << std::endl
<< "# " << std::endl;
WSACleanup();
system("pause");
}
SOCKADDR_IN addrClient;
int len = sizeof(SOCKADDR);
while (true) {
SOCKET sockConn = accept(sockServer, (SOCKADDR*)&addrClient, &len);
if (INVALID_SOCKET == sockConn) {
std::cout << "[E] Accept from socket failed! Error code is: " << WSAGetLastError() << std::endl
<< "# " << std::endl;
continue;
}
char recvBuf[100];
recv(sockConn, recvBuf, 100, 0);
//std::cout << "[I] Receive data from " << inet_ntoa(addrClient.sin_addr) << " : " << recvBuf << std::endl
// << "# " << std::endl;
char sendBuf[100] = { 0 };
//sprintf(sendBuf, "Welcome, %s", inet_ntoa(addrClient.sin_addr));
send(sockConn, sendBuf, strlen(sendBuf) + 1, 0);
closesocket(sockConn);
}
WSACleanup();
system("pause");
}
};