[mirror_frr.git] / lib / ipaddr.h

/*
 * IP address structure (for generic IPv4 or IPv6 address)
 * Copyright (C) 2016, 2017 Cumulus Networks, Inc.
 *
 * This file is part of FRR.
 *
 * FRR is free software; you can redistribute it and/or modify it
 * under the terms of the GNU General Public License as published by the
 * Free Software Foundation; either version 2, or (at your option) any
 * later version.
 *
 * FRR is distributed in the hope that it will be useful, but
 * WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with FRR; see the file COPYING.  If not, write to the Free
 * Software Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA
 * 02111-1307, USA.
 */

#ifndef __IPADDR_H__
#define __IPADDR_H__

#include <zebra.h>

/*
 * Generic IP address - union of IPv4 and IPv6 address.
 */
enum ipaddr_type_t {
	IPADDR_NONE = 0,
	IPADDR_V4 = 1, /* IPv4 */
	IPADDR_V6 = 2, /* IPv6 */
};

struct ipaddr {
	enum ipaddr_type_t ipa_type;
	union {
		u_char addr;
		struct in_addr _v4_addr;
		struct in6_addr _v6_addr;
	} ip;
#define ipaddr_v4 ip._v4_addr
#define ipaddr_v6 ip._v6_addr
};

#define IS_IPADDR_NONE(p) ((p)->ipa_type == IPADDR_NONE)
#define IS_IPADDR_V4(p)   ((p)->ipa_type == IPADDR_V4)
#define IS_IPADDR_V6(p)   ((p)->ipa_type == IPADDR_V6)

#define SET_IPADDR_V4(p)  (p)->ipa_type = IPADDR_V4
#define SET_IPADDR_V6(p)  (p)->ipa_type = IPADDR_V6

static inline int str2ipaddr(const char *str, struct ipaddr *ip)
{
	int ret;

	memset(ip, 0, sizeof(struct ipaddr));

	ret = inet_pton(AF_INET, str, &ip->ipaddr_v4);
	if (ret > 0) /* Valid IPv4 address. */
	{
		ip->ipa_type = IPADDR_V4;
		return 0;
	}
	ret = inet_pton(AF_INET6, str, &ip->ipaddr_v6);
	if (ret > 0) /* Valid IPv6 address. */
	{
		ip->ipa_type = IPADDR_V6;
		return 0;
	}

	return -1;
}

static inline char *ipaddr2str(struct ipaddr *ip, char *buf, int size)
{
	buf[0] = '\0';
	if (ip) {
		if (IS_IPADDR_V4(ip))
			inet_ntop(AF_INET, &ip->ip.addr, buf, size);
		else if (IS_IPADDR_V6(ip))
			inet_ntop(AF_INET6, &ip->ip.addr, buf, size);
	}
	return buf;
}
#endif /* __IPADDR_H__ */
Commit	Line	Data
482c0fb0	1	/*
	2	* IP address structure (for generic IPv4 or IPv6 address)
	3	* Copyright (C) 2016, 2017 Cumulus Networks, Inc.
	4	*
	5	* This file is part of FRR.
	6	*
	7	* FRR is free software; you can redistribute it and/or modify it
	8	* under the terms of the GNU General Public License as published by the
	9	* Free Software Foundation; either version 2, or (at your option) any
	10	* later version.
	11	*
	12	* FRR is distributed in the hope that it will be useful, but
	13	* WITHOUT ANY WARRANTY; without even the implied warranty of
	14	* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
	15	* General Public License for more details.
	16	*
	17	* You should have received a copy of the GNU General Public License
	18	* along with FRR; see the file COPYING. If not, write to the Free
	19	* Software Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA
	20	* 02111-1307, USA.
	21	*/
	22
	23	#ifndef __IPADDR_H__
	24	#define __IPADDR_H__
	25
	26	#include <zebra.h>
	27
	28	/*
	29	* Generic IP address - union of IPv4 and IPv6 address.
	30	*/
d62a17ae	31	enum ipaddr_type_t {
	32	IPADDR_NONE = 0,
	33	IPADDR_V4 = 1, /* IPv4 */
	34	IPADDR_V6 = 2, /* IPv6 */
482c0fb0	35	};
482c0fb0	36
d62a17ae	37	struct ipaddr {
	38	enum ipaddr_type_t ipa_type;
	39	union {
	40	u_char addr;
	41	struct in_addr _v4_addr;
	42	struct in6_addr _v6_addr;
	43	} ip;
482c0fb0	44	#define ipaddr_v4 ip._v4_addr
482c0fb0	45	#define ipaddr_v6 ip._v6_addr
7fb9d20f	46	};
482c0fb0	47
	48	#define IS_IPADDR_NONE(p) ((p)->ipa_type == IPADDR_NONE)
	49	#define IS_IPADDR_V4(p) ((p)->ipa_type == IPADDR_V4)
	50	#define IS_IPADDR_V6(p) ((p)->ipa_type == IPADDR_V6)
	51
	52	#define SET_IPADDR_V4(p) (p)->ipa_type = IPADDR_V4
	53	#define SET_IPADDR_V6(p) (p)->ipa_type = IPADDR_V6
	54
d62a17ae	55	static inline int str2ipaddr(const char str, struct ipaddr ip)
482c0fb0	56	{
d62a17ae	57	int ret;
482c0fb0	58
d62a17ae	59	memset(ip, 0, sizeof(struct ipaddr));
482c0fb0	60
d62a17ae	61	ret = inet_pton(AF_INET, str, &ip->ipaddr_v4);
	62	if (ret > 0) /* Valid IPv4 address. */
	63	{
	64	ip->ipa_type = IPADDR_V4;
	65	return 0;
	66	}
	67	ret = inet_pton(AF_INET6, str, &ip->ipaddr_v6);
	68	if (ret > 0) /* Valid IPv6 address. */
	69	{
	70	ip->ipa_type = IPADDR_V6;
	71	return 0;
	72	}
482c0fb0	73
d62a17ae	74	return -1;
482c0fb0	75	}
482c0fb0	76
d62a17ae	77	static inline char ipaddr2str(struct ipaddr ip, char *buf, int size)
482c0fb0	78	{
d62a17ae	79	buf[0] = '\0';
	80	if (ip) {
	81	if (IS_IPADDR_V4(ip))
	82	inet_ntop(AF_INET, &ip->ip.addr, buf, size);
	83	else if (IS_IPADDR_V6(ip))
	84	inet_ntop(AF_INET6, &ip->ip.addr, buf, size);
	85	}
	86	return buf;
482c0fb0	87	}
482c0fb0	88	#endif /* __IPADDR_H__ */