difoss_cppbase/ipv4_address.cpp at master · difosschan/difoss_cppbase · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
#include "ipv4_address.h"

#include "str_util.h"
#include "ctime.h"

#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>

using namespace std;
//////////////////////////////////////////////////////////////////////////
Ipv4Address::Ipv4Address()
: ip(0), port(0) {}

Ipv4Address::Ipv4Address(uint32_t ip_, uint16_t port_)
: ip(ip_), port(port_) {}


Ipv4Address::Ipv4Address( const char* cstr )
: ip(0), port(0) {
    set(cstr);
}

bool Ipv4Address::set( const char* cstr ) {
    string ip_port(cstr);
    vector<string> v;
    StrUtil::Split(cstr, ":", v);
    switch (v.size())
    {
    case 1:
        StrUtil::ReplaceAll(v[0], "*", "255");
        ip = ::inet_network(v[0].c_str());
        port = 0;
        break;
    case 2:
        StrUtil::ReplaceAll(v[0], "*", "255");
        ip = ::inet_network(v[0].c_str());
        port = StrUtil::ToInt(v[1].c_str());
        break;
    default:
        break;
    }
    return true;
}
Ipv4Address& Ipv4Address::operator= ( const Ipv4Address& other ) {
    if (&other != this) {
        this->ip = other.ip;
        this->port = other.port;
    }
    return *this;
}

ostream& Ipv4Address::operator<<( ostream& os ) const
{
    in_addr ip_net;
    ip_net.s_addr = htonl(ip);
    os << "ip: " << ip << "/" << inet_ntoa(ip_net) << ", port: " << port;
    return os;
}

bool Ipv4Address::operator==( const Ipv4Address& other ) const
{
    if (&other == this)
        return true;
    return (this->ip == other.ip) && (this->port == other.port);
}

bool Ipv4Address::operator!=( const Ipv4Address& other ) const {
    return !(*this == other);
}

bool Ipv4Address::IsGroupAddress( uint32_t ip )
{
    uint32_t net_ip = htonl(ip);
    return (net_ip &= 0xE0) == 0xE0;
}

std::string Ipv4Address::ToString( uint32_t ip )
{
    in_addr ip_net;
    ip_net.s_addr = htonl(ip);
    return std::string(inet_ntoa(ip_net));
}


ostream& operator<<(ostream & os, const Ipv4Address& one) {
    one.operator<<(os);
    return os;
}