封装了一个客户端接口:
写的过程中遇到两个主要的问题:
-
inet_addr(ip)已过期
我们熟悉的写法是:add.sin_addr.S_un.S_addr = inet_addr(ip);
可以用:
inet_pton(AF_INET,ip,&add.sin_addr);
来等效替代,其中ip是字符串,add是
SOCKADDR_IN
类,AF_INET
表示使用IPv4地址。注意还要添加包含该函数声明的头文件:WS2tcpip.h
-
调用
sock.recv(bBuffer,iBufferLen,0);
时接收字符串打印出来出现大量乱码屯屯屯屯
观察了一下发现是因为recv函数不会在写入数据后在末尾添加\0
,于是我们来手动添加,考虑到recv会返回接收数据的大小,我们自己封装的函数这样处理:void Recv(char* data) { int end=recv(sock_client, data, 1024, 0); data[end] = '\0'; }
即可解决乱码问题。
完整代码:
#include <Winsock2.h>
#include <WS2tcpip.h> //InetPton声明文件
#pragma comment(lib,"ws2_32.lib")
using namespace std;
#define MALLOC(n, type) ( (type *) malloc( (n) * sizeof(type) ) )
class TCP_Client {
private:
SOCKET sock_client;
SOCKADDR_IN serv_add;
public:
int err = 0; //0: no error 1:winsock load error 2:failed to create socket 3:failed to connect
int state = 0 ;
TCP_Client(const char* serv_ip, int serv_port)
{
WORD w = MAKEWORD(2, 2);
WSADATA wsa;
if (WSAStartup(w, &wsa) != 0) {
err = 1; return;
}
serv_add.sin_family = AF_INET;
serv_add.sin_port = htons(serv_port);
//serv_add.sin_addr.S_un.S_addr = inet_addr(serv_ip); 旧函数,已过时
inet_pton(AF_INET,serv_ip,&serv_add.sin_addr);
sock_client = socket(AF_INET, SOCK_STREAM, 0);
if (SOCKET_ERROR == sock_client) {
err = 2;
return;
}
}
int Connect()
{
if (connect(sock_client, (struct sockaddr*) & serv_add, sizeof(serv_add) ) == INVALID_SOCKET) {
err = 3;
return 0;
}
state = 1;
return 1;
}
void Recv(char* data)
{
int end=recv(sock_client, data, 1024, 0);
data[end] = '\0';
}
void Send(const char data[])
{
send(sock_client, data, strlen(data), 0);
}
};