ipaddr.h

Go to the documentation of this file.

/*
 * Copyright (C) 2013,2014 CESNET
 *
 * LICENSE TERMS
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. 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.
 * 3. Neither the name of the Company nor the names of its contributors
 *    may be used to endorse or promote products derived from this
 *    software without specific prior written permission.
 *
 * ALTERNATIVELY, provided that this notice is retained in full, this
 * product may be distributed under the terms of the GNU General Public
 * License (GPL) version 2 or later, in which case the provisions
 * of the GPL apply INSTEAD OF those given above.
 *
 * This software is provided ``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 company or contributors 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.
 *
 */

#ifndef _IPADDR_H_
#define _IPADDR_H_

#ifdef __cplusplus
extern "C" {
#endif

#include "inline.h"

#include <stdint.h>
#include <string.h>
#ifndef __WIN32
#include <arpa/inet.h>
#include <sys/socket.h>
#include <sys/types.h>
#else
#define ntohl(x) ((x & 0x0000000ff) << 24) | ((x & 0x0000ff00) << 8) | \
                 ((x & 0x00ff0000) >> 8) | ((x & 0xff000000) >> 24)
#define htonl(x) ntohl(x)
#endif

typedef union ip_addr_u {
   uint8_t  bytes[16];
   uint8_t  ui8[16];
   uint16_t ui16[8];
   uint32_t ui32[4];
   uint64_t ui64[2];
} ip_addr_t;


INLINE int ip_is4(const ip_addr_t *addr)
{
   return (addr->ui64[0] == 0 && addr->ui32[3] == 0xffffffff);
}

INLINE int ip_is6(const ip_addr_t *addr)
{
   return !ip_is4(addr);
}

INLINE uint32_t ip_get_v4_as_int(ip_addr_t *addr)
{
   return ntohl(addr->ui32[2]);
}

INLINE char *ip_get_v4_as_bytes(const ip_addr_t *addr)
{
   return (char *) &addr->bytes[8];
}


INLINE ip_addr_t ip_from_int(uint32_t i)
{
   ip_addr_t a;
   a.ui64[0] = 0;
   a.ui32[2] = htonl(i);
   a.ui32[3] = 0xffffffff;
   return a;
}

INLINE ip_addr_t ip_from_4_bytes_be(char b[4])
{
   ip_addr_t a;
   a.ui64[0] = 0;
   a.bytes[8] = b[0];
   a.bytes[9] = b[1];
   a.bytes[10] = b[2];
   a.bytes[11] = b[3];
   a.ui32[3] = 0xffffffff;
   return a;
}

INLINE ip_addr_t ip_from_4_bytes_le(char b[4])
{
   ip_addr_t a;
   a.ui64[0] = 0;
   a.bytes[8]  = b[3];
   a.bytes[9]  = b[2];
   a.bytes[10] = b[1];
   a.bytes[11] = b[0];
   a.ui32[3] = 0xffffffff;
   return a;
}

INLINE ip_addr_t ip_from_16_bytes_be(char b[16])
{
   ip_addr_t a;
   memcpy(&a, b, 16);
   return a;
}

INLINE ip_addr_t ip_from_16_bytes_le(char b[16])
{
   ip_addr_t a;
   int i;
   for (i = 0; i < 16; i++) {
      a.bytes[i] = b[15-i];
   }
   return a;
}

INLINE int ip_cmp(const ip_addr_t *addr1, const ip_addr_t *addr2)
{
   return memcmp((const char *)addr1, (const char *)addr2, 16);
}

INLINE int ip_is_null(const ip_addr_t *addr)
{
   if (addr->ui64[0] == 0) {
      if (addr->ui64[1] == 0 || (addr->ui32[2] == 0 && addr->ui32[3] == 0xffffffff)) {
         return 1;
      }
   }
   return 0;
}

#ifndef __WIN32

INLINE int ip_from_str(const char *str, ip_addr_t *addr)
{
   char tmp[16];
   if (strchr(str, ':') == NULL) { // IPv4
      if (inet_pton(AF_INET, str, (void *) tmp) != 1) {
         return 0; // err
      }
      *addr = ip_from_4_bytes_be(tmp);
      return 1;
   } else { // IPv6
      if (inet_pton(AF_INET6, str, (void *) tmp) != 1) {
         return 0; // err
      }
      *addr = ip_from_16_bytes_be(tmp);
      return 1;
   }
}

INLINE void ip_to_str(const ip_addr_t *addr, char *str)
{
   if (ip_is4(addr)) { // IPv4
      inet_ntop(AF_INET, ip_get_v4_as_bytes(addr), str, INET6_ADDRSTRLEN);
   } else { // IPv6
      inet_ntop(AF_INET6, addr, str, INET6_ADDRSTRLEN);
   }
}
#endif

#ifdef __cplusplus
}
#endif

#endif