casbin/ip_parser/parser/IPNet.cpp

#include "casbin/pch.h"

#ifndef IPNET_CPP
#define IPNET_CPP


#include "casbin/ip_parser/parser/IPNet.h"

namespace casbin {

std::string IPNet :: NETIP_toString() {
    std::string ip1, ip2, ip3, ip4;
    std::stringstream ss1, ss2, ss3, ss4;
    ss1 << net_ip.ip[0];
    ss1 >> ip1;
    ss2 << net_ip.ip[1];
    ss2 >> ip2;
    ss3 << net_ip.ip[2];
    ss3 >> ip3;
    ss4 << net_ip.ip[3];
    ss4 >> ip4;
    return ip1 + "." + ip2 + "." + ip3 + "." + ip4;
}

std::string IPNet :: IPMask_toString() {
    std::string mask1, mask2, mask3, mask4;
    std::stringstream ss1, ss2, ss3, ss4;
    ss1 << mask[0];
    ss1 >> mask1;
    ss2 << mask[1];
    ss2 >>mask2;
    ss3 << mask[2];
    ss3 >>mask3;
    ss4 << mask[3];
    ss4 >>mask4;
    return mask1 + "." + mask2 + "." + mask3 + "." + mask4;
}

// Contains reports whether the network includes ip.
bool IPNet :: contains(IP ipNew) {
    std::pair<IP, IPMask> p = networkNumberAndMask(*this);
    IP x;
    x = ipNew.To4();
    if(x.isLegal == true) {
        ipNew = x;
    }
    int l = int(ipNew.ip.size());
    if(l != p.first.ip.size()) {
        return false;
    }
    for(int i = 0 ; i < l ; i++) {
        if((p.first.ip[i]&p.second[i]) != (ipNew.ip[i]&p.second[i])) {
            return false;
        }
    }
    return true;
}

std::pair<IP, IPMask> IPNet :: networkNumberAndMask(IPNet n) {
    IP newIp;
    newIp = n.net_ip.To4();
    std::pair<IP, IPMask> p;
    p.first = newIp;
    if(newIp.isLegal == false) {
        newIp = n.net_ip;
        if(newIp.ip.size() != IP :: IPv6len) {
            p.first.isLegal = false;
            return p;
        }
    }
    IPMask m = n.mask;
    p.second = m;
    const byte ipv4len = IP :: IPv4len;
    const byte ipv6len = IP :: IPv6len;
    if(m.size() == ipv4len){
        if(newIp.ip.size() != IP :: IPv4len) {
            p.first.isLegal = false;
        }
    } else if(m.size() == ipv6len){
        if(newIp.ip.size() == IP :: IPv4len) {
            IPMask newM(m.begin() + 12, m.end());
            m = newM;
            p.first = newIp;
            p.second = m;
        }
    } else {
        p.first.isLegal = false;
    }
    return p;
}

} // namespace casbin

#endif // IPNET_CPP