这个类的主要作用就是封装socket。
由于博主是在windows平台下,所以用的头文件时windows.h
和Ws2tcpip.h
。
linux平台下应该用sys/socket.h
、netinet/in.h
和arpa/inet.h
。
InetAddress.h
#pragma once/* linux下 * #include<arpa/inet.h> //for sockaddr * #include <netinet/in.h> */
#include <string>
#include <windows.h>//封装socket地址类型
class InetAddress
{
public://接收端口号和IP地址,填充addr_explicit InetAddress(uint16_t port ,std::string ip = "127.0.0.1");explicit InetAddress(const sockaddr_in& addr):addr_(addr){}std::string toIP() const;std::string toIPPort() const;uint16_t toPort() const;const sockaddr_in* getSockAddr() const { return &addr_; }
private:sockaddr_in addr_;};
InetAddress
#include "InetAddress.h"#include <Ws2tcpip.h>
#include <string.h>
/* * windows下: * #include <WS2tcpip.h> * linux下: * #include <sys/socket.h> * #include <netinet/in.h> * #include<arpa/inet.h> */
//无法包含strings.h头文件 自己写了一下bzero
#define bzero(a, b) memset(a, 0, b)//接收端口号和IP地址,填充addr_
InetAddress::InetAddress(uint16_t port,std::string ip)
{bzero(&addr_,sizeof addr_);addr_.sin_family = AF_INET;//主机字节序到网络字节序addr_.sin_port = htons(port);//字符串转成整型数据addr_.sin_addr.s_addr = inet_addr(ip.c_str());
}std::string InetAddress::toIP() const
{//addr_ 读取ip地址,转换成点分十进制char buf[64] = {0};inet_ntop(AF_INET, &addr_.sin_addr, buf, sizeof buf); return buf;
}std::string InetAddress::toIPPort() const
{//ip:portchar buf[64] = {0};inet_ntop(AF_INET, &addr_.sin_addr, buf, sizeof buf); uint16_t port = ntohs(addr_.sin_port);sprintf(buf+strlen(buf),":%u",port);return buf;
}uint16_t InetAddress::toPort() const
{return ntohs(addr_.sin_port);
}
测试代码
int main()
{InetAddress addr(8000);std::cout << addr.toIPPort()<< std::endl;
}
测试结果:
127.0.0.1:8000
测试正常!
参考文献
[1] 施磊.重写moduo库.图论科技.2020.7.