/*

 * Soft:        Keepalived is a failover program for the LVS project

 *              <www.linuxvirtualserver.org>. It monitor & manipulate

 *              a loadbalanced server pool using multi-layer checks.

 *

 * Part:        utils.h include file.

 *

 * Author:      Alexandre Cassen, <acassen@linux-vs.org>

 *

 *              This program 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.

 *

 *              This program 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 of the License, or (at your option) any later version.

 *

 * Copyright (C) 2001-2017 Alexandre Cassen, <acassen@gmail.com>

 */

#ifndef _UTILS_H

#define _UTILS_H

#include "config.h"

/* system includes */

#include <sys/types.h>

#include <stdint.h>

#include <stdbool.h>

#include <sys/socket.h>

#include <netinet/in.h>

#include <stdio.h>

#include "vector.h"

#ifdef _DEBUG_

#include "logger.h"

#endif

/* Global debugging logging facilities */

#ifdef _DEBUG_

#define DBG(fmt, msg...) log_message(LOG_DEBUG, fmt, ## msg)

#else

#define DBG(fmt, msg...)

#endif

#define STR(x)  #x

#ifdef _WITH_PERF_

typedef enum {

	PERF_NONE,

	PERF_RUN,

	PERF_ALL,

	PERF_END,

} perf_t;

#endif

/* inline stuff */

static inline int __ip6_addr_equal(const struct in6_addr *a1,

				   const struct in6_addr *a2)

{

	return (((a1->s6_addr32[0] ^ a2->s6_addr32[0]) |

		 (a1->s6_addr32[1] ^ a2->s6_addr32[1]) |

		 (a1->s6_addr32[2] ^ a2->s6_addr32[2]) |

		 (a1->s6_addr32[3] ^ a2->s6_addr32[3])) == 0);

}

static inline bool sockstorage_equal(const struct sockaddr_storage *s1,

				    const struct sockaddr_storage *s2)

{

	if (s1->ss_family != s2->ss_family)

		return false;

	if (s1->ss_family == AF_INET6) {

		struct sockaddr_in6 *a1 = (struct sockaddr_in6 *) s1;

		struct sockaddr_in6 *a2 = (struct sockaddr_in6 *) s2;

//		if (IN6_ARE_ADDR_EQUAL(a1, a2) && (a1->sin6_port == a2->sin6_port))

		if (__ip6_addr_equal(&a1->sin6_addr, &a2->sin6_addr) &&

		    (a1->sin6_port == a2->sin6_port))

			return true;

	} else if (s1->ss_family == AF_INET) {

		struct sockaddr_in *a1 = (struct sockaddr_in *) s1;

		struct sockaddr_in *a2 = (struct sockaddr_in *) s2;

		if ((a1->sin_addr.s_addr == a2->sin_addr.s_addr) &&

		    (a1->sin_port == a2->sin_port))

			return true;

	} else if (s1->ss_family == AF_UNSPEC)

		return true;

	return false;

}

static inline bool inaddr_equal(sa_family_t family, void *addr1, void *addr2)

{

	if (family == AF_INET6) {

		struct in6_addr *a1 = (struct in6_addr *) addr1;

		struct in6_addr *a2 = (struct in6_addr *) addr2;

		if (__ip6_addr_equal(a1, a2))

			return true;

	} else if (family == AF_INET) {

		struct in_addr *a1 = (struct in_addr *) addr1;

		struct in_addr *a2 = (struct in_addr *) addr2;

		if (a1->s_addr == a2->s_addr)

			return true;

	}

	return false;

}

static inline uint16_t csum_incremental_update32(const uint16_t old_csum, const uint32_t old_val, const uint32_t new_val)

{

	/* This technique for incremental IP checksum update is described in RFC1624,

	 * along with accompanying errata */

	if (old_val == new_val)

		return old_csum;

	uint32_t acc = (~old_csum & 0xffff) + (~(old_val >> 16 ) & 0xffff) + (~old_val & 0xffff);

	acc += (new_val >> 16) + (new_val & 0xffff);

	/* finally compute vrrp checksum */

	acc = (acc & 0xffff) + (acc >> 16);

	acc += acc >> 16;

	return ~acc & 0xffff;

}

static inline uint16_t csum_incremental_update16(const uint16_t old_csum, const uint16_t old_val, const uint16_t new_val)

{

	/* This technique for incremental IP checksum update is described in RFC1624,

	 * along with accompanying errata */

	if (old_val == new_val)

		return old_csum;

	uint32_t acc = (~old_csum & 0xffff) + (~old_val & 0xffff);

	acc += new_val;

	/* finally compute vrrp checksum */

	acc = (acc & 0xffff) + (acc >> 16);

	acc += acc >> 16;

	return ~acc & 0xffff;

}

/* global vars exported */

extern unsigned long debug;

#ifdef _WITH_PERF_

extern perf_t perf_run;

#endif

/* Prototypes defs */

extern void dump_buffer(char *, size_t, FILE *, int);

#ifdef _WITH_STACKTRACE_

extern void write_stacktrace(const char *, const char *);

#endif

extern char *make_file_name(const char *, const char *, const char *, const char *);

#ifdef _WITH_PERF_

extern void run_perf(const char *, const char *, const char *);

#endif

extern uint16_t in_csum(const uint16_t *, size_t, uint32_t, uint32_t *);

extern char *inet_ntop2(uint32_t);

extern bool inet_stor(const char *, uint32_t *);

extern int domain_stosockaddr(const char *, const char *, struct sockaddr_storage *);

extern int inet_stosockaddr(char *, const char *, struct sockaddr_storage *);

extern void inet_ip4tosockaddr(struct in_addr *, struct sockaddr_storage *);

extern void inet_ip6tosockaddr(struct in6_addr *, struct sockaddr_storage *);

extern char *inet_sockaddrtos(struct sockaddr_storage *);

extern char *inet_sockaddrtopair(struct sockaddr_storage *);

extern char *inet_sockaddrtotrio(struct sockaddr_storage *, uint16_t);

extern uint16_t inet_sockaddrport(struct sockaddr_storage *);

extern uint32_t inet_sockaddrip4(struct sockaddr_storage *);

extern int inet_sockaddrip6(struct sockaddr_storage *, struct in6_addr *);

extern int inet_inaddrcmp(int, const void *, const void *);

extern int inet_sockaddrcmp(const struct sockaddr_storage *, const struct sockaddr_storage *);

extern char *get_local_name(void);

extern bool string_equal(const char *, const char *);

extern FILE *fopen_safe(const char *, const char *);

extern void set_std_fd(bool);

extern void close_std_fd(void);

#if !defined _HAVE_LIBIPTC_ || defined _LIBIPTC_DYNAMIC_

extern int fork_exec(char **);

#endif

#if defined _WITH_VRRP_ || defined _WITH_BFD_

extern int open_pipe(int [2]);

#endif

extern int memcmp_constant_time(const void *, const void *, size_t);

#endif