source: mainline/uspace/lib/c/generic/inet/addr.c@ b323a3a

/*
 * Copyright (c) 2013 Jiri Svoboda
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 *
 * - Redistributions of source code must retain the above copyright
 *   notice, this list of conditions and the following disclaimer.
 * - Redistributions in binary form must reproduce the above copyright
 *   notice, this list of conditions and the following disclaimer in the
 *   documentation and/or other materials provided with the distribution.
 * - The name of the author may not be used to endorse or promote products
 *   derived from this software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

/** @addtogroup libc
 * @{
 */
/** @file Internet address parsing and formatting.
 */

#include <errno.h>
#include <unistd.h>
#include <net/socket_codes.h>
#include <inet/addr.h>
#include <net/inet.h>
#include <stdio.h>
#include <malloc.h>
#include <bitops.h>

#define INET_PREFIXSTRSIZE  5

#if !(defined(__BE__) ^ defined(__LE__))
        #error The architecture must be either big-endian or little-endian.
#endif

const addr48_t addr48_broadcast = {
        0xff, 0xff, 0xff, 0xff, 0xff, 0xff
};

static const inet_addr_t inet_addr_any_addr = {
        .family = AF_INET,
        .addr = 0
};

static const inet_addr_t inet_addr_any_addr6 = {
        .family = AF_INET6,
        .addr6 = {0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0}
};

void addr48(const addr48_t src, addr48_t dst)
{
        memcpy(dst, src, 6);
}

void addr128(const addr128_t src, addr128_t dst)
{
        memcpy(dst, src, 16);
}

int addr128_compare(const addr128_t a, const addr128_t b)
{
        return memcmp(a, b, 16);
}

void host2addr128_t_be(const addr128_t host, addr128_t be)
{
#ifdef __BE__
        memcpy(be, host, 16);
#else
        be[0] = host[15];
        be[1] = host[14];
        be[2] = host[13];
        be[3] = host[12];
        be[4] = host[11];
        be[5] = host[10];
        be[6] = host[9];
        be[7] = host[8];
        be[8] = host[7];
        be[9] = host[6];
        be[10] = host[5];
        be[11] = host[4];
        be[12] = host[3];
        be[13] = host[2];
        be[14] = host[1];
        be[15] = host[0];
#endif
}

void addr128_t_be2host(const addr128_t be, addr128_t host)
{
#ifdef __BE__
        memcpy(host, be, 16);
#else
        host[0] = be[15];
        host[1] = be[14];
        host[2] = be[13];
        host[3] = be[12];
        host[4] = be[11];
        host[5] = be[10];
        host[6] = be[9];
        host[7] = be[8];
        host[8] = be[7];
        host[9] = be[6];
        host[10] = be[5];
        host[11] = be[4];
        host[12] = be[3];
        host[13] = be[2];
        host[14] = be[1];
        host[15] = be[0];
#endif
}

void inet_addr(inet_addr_t *addr, uint8_t a, uint8_t b, uint8_t c, uint8_t d)
{
        addr->family = AF_INET;
        addr->addr = ((addr32_t) a << 24) | ((addr32_t) b << 16) |
            ((addr32_t) c << 8) | ((addr32_t) d);
}

void inet_naddr(inet_naddr_t *naddr, uint8_t a, uint8_t b, uint8_t c, uint8_t d,
    uint8_t prefix)
{
        naddr->family = AF_INET;
        naddr->addr = ((addr32_t) a << 24) | ((addr32_t) b << 16) |
            ((addr32_t) c << 8) | ((addr32_t) d);
        naddr->prefix = prefix;
}

void inet_addr6(inet_addr_t *addr, uint16_t a, uint16_t b, uint16_t c,
    uint16_t d, uint16_t e, uint16_t f, uint16_t g, uint16_t h)
{
        addr->family = AF_INET6;
        addr->addr6[0] = (a >> 8) & 0xff;
        addr->addr6[1] = a & 0xff;
        addr->addr6[2] = (b >> 8) & 0xff;
        addr->addr6[3] = b & 0xff;
        addr->addr6[4] = (c >> 8) & 0xff;
        addr->addr6[5] = c & 0xff;
        addr->addr6[6] = (d >> 8) & 0xff;
        addr->addr6[7] = d & 0xff;
        addr->addr6[8] = (e >> 8) & 0xff;
        addr->addr6[9] = e & 0xff;
        addr->addr6[10] = (f >> 8) & 0xff;
        addr->addr6[11] = f & 0xff;
        addr->addr6[12] = (g >> 8) & 0xff;
        addr->addr6[13] = g & 0xff;
        addr->addr6[14] = (h >> 8) & 0xff;
        addr->addr6[15] = h & 0xff;
}

void inet_naddr6(inet_naddr_t *naddr, uint16_t a, uint16_t b, uint16_t c,
    uint16_t d, uint16_t e, uint16_t f, uint16_t g, uint16_t h, uint8_t prefix)
{
        naddr->family = AF_INET6;
        naddr->addr6[0] = (a >> 8) & 0xff;
        naddr->addr6[1] = a & 0xff;
        naddr->addr6[2] = (b >> 8) & 0xff;
        naddr->addr6[3] = b & 0xff;
        naddr->addr6[4] = (c >> 8) & 0xff;
        naddr->addr6[5] = c & 0xff;
        naddr->addr6[6] = (d >> 8) & 0xff;
        naddr->addr6[7] = d & 0xff;
        naddr->addr6[8] = (e >> 8) & 0xff;
        naddr->addr6[9] = e & 0xff;
        naddr->addr6[10] = (f >> 8) & 0xff;
        naddr->addr6[11] = f & 0xff;
        naddr->addr6[12] = (g >> 8) & 0xff;
        naddr->addr6[13] = g & 0xff;
        naddr->addr6[14] = (h >> 8) & 0xff;
        naddr->addr6[15] = h & 0xff;
        naddr->prefix = prefix;
}

/** Parse network address family.
 *
 * @param text Network address in common notation.
 * @param af   Place to store network address family.
 *
 * @return EOK on success, EINVAL if input is not in valid format.
 *
 */
int inet_addr_family(const char *text, uint16_t *af)
{
        char *dot = str_chr(text, '.');
        if (dot != NULL) {
                *af = AF_INET;
                return EOK;
        }
        
        char *collon = str_chr(text, ':');
        if (collon != NULL) {
                *af = AF_INET6;
                return EOK;
        }
        
        return EINVAL;
}

void inet_naddr_addr(const inet_naddr_t *naddr, inet_addr_t *addr)
{
        addr->family = naddr->family;
        memcpy(addr->addr6, naddr->addr6, 16);
}

void inet_addr_naddr(const inet_addr_t *addr, uint8_t prefix,
    inet_naddr_t *naddr)
{
        naddr->family = addr->family;
        memcpy(naddr->addr6, addr->addr6, 16);
        naddr->prefix = prefix;
}

void inet_addr_any(inet_addr_t *addr)
{
        addr->family = AF_NONE;
        memset(addr->addr6, 0, 16);
}

void inet_naddr_any(inet_naddr_t *naddr)
{
        naddr->family = AF_NONE;
        memset(naddr->addr6, 0, 16);
        naddr->prefix = 0;
}

int inet_addr_compare(const inet_addr_t *a, const inet_addr_t *b)
{
        if (a->family != b->family)
                return 0;
        
        switch (a->family) {
        case AF_INET:
                return (a->addr == b->addr);
        case AF_INET6:
                return addr128_compare(a->addr6, b->addr6);
        default:
                return 0;
        }
}

int inet_addr_is_any(const inet_addr_t *addr)
{
        return ((addr->family == 0) ||
            (inet_addr_compare(addr, &inet_addr_any_addr)) ||
            (inet_addr_compare(addr, &inet_addr_any_addr6)));
}

int inet_naddr_compare(const inet_naddr_t *naddr, const inet_addr_t *addr)
{
        if (naddr->family != addr->family)
                return 0;
        
        switch (naddr->family) {
        case AF_INET:
                return (naddr->addr == addr->addr);
        case AF_INET6:
                return addr128_compare(naddr->addr6, addr->addr6);
        default:
                return 0;
        }
}

int inet_naddr_compare_mask(const inet_naddr_t *naddr, const inet_addr_t *addr)
{
        if (naddr->family != addr->family)
                return 0;
        
        switch (naddr->family) {
        case AF_INET:
                if (naddr->prefix > 32)
                        return 0;
                
                addr32_t mask =
                    BIT_RANGE(addr32_t, 31, 31 - (naddr->prefix - 1));
                return ((naddr->addr & mask) == (addr->addr & mask));
        case AF_INET6:
                if (naddr->prefix > 128)
                        return 0;
                
                size_t pos = 0;
                for (size_t i = 0; i < 16; i++) {
                        /* Further bits do not matter */
                        if (naddr->prefix < pos)
                                break;
                        
                        if (naddr->prefix - pos > 8) {
                                /* Comparison without masking */
                                if (naddr->addr6[i] != addr->addr6[i])
                                        return 0;
                        } else {
                                /* Comparison with masking */
                                uint8_t mask =
                                    BIT_RANGE(uint8_t, 8, 8 - (naddr->prefix - pos - 1));
                                if ((naddr->addr6[i] & mask) != (addr->addr6[i] & mask))
                                        return 0;
                        }
                        
                        pos += 8;
                }
                
                return 1;
        default:
                return 0;
        }
}

/** Parse node address.
 *
 * @param text Network address in common notation.
 * @param addr Place to store node address.
 *
 * @return EOK on success, EINVAL if input is not in valid format.
 *
 */
int inet_addr_parse(const char *text, inet_addr_t *addr)
{
        int rc = inet_addr_family(text, &addr->family);
        if (rc != EOK)
                return rc;
        
        uint8_t buf[16];
        rc = inet_pton(addr->family, text, buf);
        if (rc != EOK)
                return rc;
        
        switch (addr->family) {
        case AF_INET:
                addr->addr = (buf[0] << 24) | (buf[1] << 16) | (buf[2] << 8) |
                    buf[3];
                break;
        case AF_INET6:
                memcpy(addr->addr6, buf, 16);
                break;
        default:
                return EINVAL;
        }
        
        return EOK;
}

/** Parse network address.
 *
 * @param text  Network address in common notation.
 * @param naddr Place to store network address.
 *
 * @return EOK on success, EINVAL if input is not in valid format.
 *
 */
int inet_naddr_parse(const char *text, inet_naddr_t *naddr)
{
        char *slash = str_chr(text, '/');
        if (slash == NULL)
                return EINVAL;
        
        *slash = 0;
        
        int rc = inet_addr_family(text, &naddr->family);
        if (rc != EOK)
                return rc;
        
        uint8_t buf[16];
        rc = inet_pton(naddr->family, text, buf);
        *slash = '/';
        
        if (rc != EOK)
                return rc;
        
        slash++;
        uint8_t prefix;
        
        switch (naddr->family) {
        case AF_INET:
                prefix = strtoul(slash, &slash, 10);
                if (prefix > 32)
                        return EINVAL;
                
                naddr->addr = (buf[0] << 24) | (buf[1] << 16) | (buf[2] << 8) |
                    buf[3];
                naddr->prefix = prefix;
                
                break;
        case AF_INET6:
                prefix = strtoul(slash, &slash, 10);
                if (prefix > 128)
                        return EINVAL;
                
                memcpy(naddr->addr6, buf, 16);
                naddr->prefix = prefix;
                
                break;
        default:
                return ENOTSUP;
        }
        
        return EOK;
}

/** Format node address.
 *
 * @param addr Node address.
 * @param bufp Place to store pointer to formatted string.
 *
 * @return EOK on success.
 * @return ENOMEM if out of memory.
 * @return ENOTSUP on unsupported address family.
 *
 */
int inet_addr_format(const inet_addr_t *addr, char **bufp)
{
        int rc = 0;
        
        switch (addr->family) {
        case AF_NONE:
                rc = asprintf(bufp, "none");
                break;
        case AF_INET:
                rc = asprintf(bufp, "%u.%u.%u.%u", (addr->addr >> 24) & 0xff,
                    (addr->addr >> 16) & 0xff, (addr->addr >> 8) & 0xff,
                    addr->addr & 0xff);
                break;
        case AF_INET6:
                *bufp = (char *) malloc(INET6_ADDRSTRLEN);
                if (*bufp == NULL)
                        return ENOMEM;
                
                return inet_ntop(AF_INET6, addr->addr6, *bufp, INET6_ADDRSTRLEN);
        default:
                return ENOTSUP;
        }
        
        if (rc < 0)
                return ENOMEM;
        
        return EOK;
}

/** Format network address.
 *
 * @param naddr Network address.
 * @param bufp  Place to store pointer to formatted string.
 *
 * @return EOK on success.
 * @return ENOMEM if out of memory.
 * @return ENOTSUP on unsupported address family.
 *
 */
int inet_naddr_format(const inet_naddr_t *naddr, char **bufp)
{
        int rc = 0;
        char prefix[INET_PREFIXSTRSIZE];
        
        switch (naddr->family) {
        case AF_NONE:
                rc = asprintf(bufp, "none");
                break;
        case AF_INET:
                rc = asprintf(bufp, "%" PRIu8 ".%" PRIu8 ".%" PRIu8 ".%" PRIu8
                    "/%" PRIu8, (naddr->addr >> 24) & 0xff,
                    (naddr->addr >> 16) & 0xff, (naddr->addr >> 8) & 0xff,
                    naddr->addr & 0xff, naddr->prefix);
                break;
        case AF_INET6:
                *bufp = (char *) malloc(INET6_ADDRSTRLEN + INET_PREFIXSTRSIZE);
                if (*bufp == NULL)
                        return ENOMEM;
                
                rc = inet_ntop(AF_INET6, naddr->addr6, *bufp,
                    INET6_ADDRSTRLEN + INET_PREFIXSTRSIZE);
                if (rc != EOK) {
                        free(*bufp);
                        return rc;
                }
                
                rc = snprintf(prefix, INET_PREFIXSTRSIZE, "/%" PRIu8,
                    naddr->prefix);
                if (rc < 0) {
                        free(*bufp);
                        return ENOMEM;
                }
                
                str_append(*bufp, INET6_ADDRSTRLEN + INET_PREFIXSTRSIZE, prefix);
                
                break;
        default:
                return ENOTSUP;
        }
        
        if (rc < 0)
                return ENOMEM;
        
        return EOK;
}

uint16_t inet_addr_get(const inet_addr_t *addr, addr32_t *v4, addr128_t *v6)
{
        switch (addr->family) {
        case AF_INET:
                if (v4 != NULL)
                        *v4 = addr->addr;
                
                break;
        case AF_INET6:
                if (v6 != NULL)
                        memcpy(*v6, addr->addr6, 16);
                
                break;
        }
        
        return addr->family;
}

uint16_t inet_naddr_get(const inet_naddr_t *naddr, addr32_t *v4, addr128_t *v6,
    uint8_t *prefix)
{
        switch (naddr->family) {
        case AF_INET:
                if (v4 != NULL)
                        *v4 = naddr->addr;
                
                if (prefix != NULL)
                        *prefix = naddr->prefix;
                
                break;
        case AF_INET6:
                if (v6 != NULL)
                        memcpy(*v6, naddr->addr6, 16);
                
                if (prefix != NULL)
                        *prefix = naddr->prefix;
                
                break;
        }
        
        return naddr->family;
}

void inet_addr_set(addr32_t v4, inet_addr_t *addr)
{
        addr->family = AF_INET;
        addr->addr = v4;
}

void inet_naddr_set(addr32_t v4, uint8_t prefix, inet_naddr_t *naddr)
{
        naddr->family = AF_INET;
        naddr->addr = v4;
        naddr->prefix = prefix;
}

void inet_sockaddr_in_addr(const sockaddr_in_t *sockaddr_in, inet_addr_t *addr)
{
        addr->family = AF_INET;
        addr->addr = uint32_t_be2host(sockaddr_in->sin_addr.s_addr);
}

void inet_addr_set6(addr128_t v6, inet_addr_t *addr)
{
        addr->family = AF_INET6;
        memcpy(addr->addr6, v6, 16);
}

void inet_naddr_set6(addr128_t v6, uint8_t prefix, inet_naddr_t *naddr)
{
        naddr->family = AF_INET6;
        memcpy(naddr->addr6, v6, 16);
        naddr->prefix = prefix;
}

void inet_sockaddr_in6_addr(const sockaddr_in6_t *sockaddr_in6,
    inet_addr_t *addr)
{
        addr->family = AF_INET6;
        addr128_t_be2host(sockaddr_in6->sin6_addr.s6_addr, addr->addr6);
}

uint16_t inet_addr_sockaddr_in(const inet_addr_t *addr,
    sockaddr_in_t *sockaddr_in, sockaddr_in6_t *sockaddr_in6)
{
        switch (addr->family) {
        case AF_INET:
                if (sockaddr_in != NULL) {
                        sockaddr_in->sin_family = AF_INET;
                        sockaddr_in->sin_addr.s_addr = host2uint32_t_be(addr->addr);
                }
                
                break;
        case AF_INET6:
                if (sockaddr_in6 != NULL) {
                        sockaddr_in6->sin6_family = AF_INET6;
                        host2addr128_t_be(addr->addr6, sockaddr_in6->sin6_addr.s6_addr);
                }
                
                break;
        }
        
        return addr->family;
}

/** @}
 */