#ifndef __IP_H__
#define __IP_H__

#include <sys/types.h>
#include <ifaddrs.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <cstring>
#include <string>

#include <iostream>

//返回值：是否成功
//参数：prefix，通过运营商提供的ipv6地址前四个字符判断是否是自己需要的地址，如联通给我提供的地址前四个字符是2408
bool get_ipv6_address_linux(const std::string& prefix, std::string& ipv6Address)
{
    struct ifaddrs* ifAddrStruct = nullptr;
    void* tmpAddrPtr = nullptr;

    if(0 != getifaddrs(&ifAddrStruct))
    {
        return false;
    }

    struct ifaddrs* ptr = ifAddrStruct;
    char addressBuffer[INET6_ADDRSTRLEN] = {0};
    while(nullptr != ifAddrStruct)
    {
        if(AF_INET6 == ifAddrStruct->ifa_addr->sa_family)
        {
            tmpAddrPtr = &((struct sockaddr_in6*)ifAddrStruct->ifa_addr)->sin6_addr;
            inet_ntop(AF_INET6, tmpAddrPtr, addressBuffer, INET6_ADDRSTRLEN);
            std::string rst = std::string(addressBuffer);
            //过滤掉::1再获取前缀
            if(rst.size() > 4)
            {
                std::string tmp = rst.substr(0, 4);
                if(tmp == prefix)
                {
                    ipv6Address = rst;
                    memset(addressBuffer, 0, INET6_ADDRSTRLEN);
                    freeifaddrs(ptr);
                    ifAddrStruct = nullptr;
                    ptr = nullptr;
                    return true;
                }
            }

            memset(addressBuffer, 0, INET6_ADDRSTRLEN);
        }

        ifAddrStruct = ifAddrStruct->ifa_next;
    }

    freeifaddrs(ptr);
    ifAddrStruct = nullptr;
    ptr = nullptr;

    return false;
}

#endif