41 #include <sys/types.h> 43 #include <sys/socket.h> 46 #include <sys/ioctl.h> 47 #include <sys/param.h> 48 #include <netinet/in.h> 49 #include <arpa/inet.h> 65 #include <qb/qbdefs.h> 66 #include <qb/qbloop.h> 67 #define LOGSYS_UTILS_ONLY 1 80 #define MSG_NOSIGNAL 0 83 #define MCAST_SOCKET_BUFFER_SIZE (TRANSMITS_ALLOWED * FRAME_SIZE_MAX) 84 #define NETIF_STATE_REPORT_UP 1 85 #define NETIF_STATE_REPORT_DOWN 2 87 #define BIND_STATE_UNBOUND 0 88 #define BIND_STATE_REGULAR 1 89 #define BIND_STATE_LOOPBACK 2 117 void (*totemudp_deliver_fn) (
120 unsigned int msg_len);
122 void (*totemudp_iface_change_fn) (
126 void (*totemudp_target_set_completed) (
void *context);
143 void (*totemudp_log_printf) (
146 const char *
function,
158 struct iovec totemudp_iov_recv;
160 struct iovec totemudp_iov_recv_flush;
176 struct timeval stats_tv_start;
201 static int totemudp_build_sockets (
229 #define log_printf(level, format, args...) \ 231 instance->totemudp_log_printf ( \ 232 level, instance->totemudp_subsys_id, \ 233 __FUNCTION__, __FILE__, __LINE__, \ 234 (const char *)format, ##args); \ 237 #define LOGSYS_PERROR(err_num, level, fmt, args...) \ 239 char _error_str[LOGSYS_MAX_PERROR_MSG_LEN]; \ 240 const char *_error_ptr = qb_strerror_r(err_num, _error_str, sizeof(_error_str)); \ 241 instance->totemudp_log_printf ( \ 242 level, instance->totemudp_subsys_id, \ 243 __FUNCTION__, __FILE__, __LINE__, \ 244 fmt ": %s (%d)\n", ##args, _error_ptr, err_num); \ 249 const char *cipher_type,
250 const char *hash_type)
257 static inline void ucast_sendmsg (
261 unsigned int msg_len)
263 struct msghdr msg_ucast;
267 struct sockaddr_storage sockaddr;
276 (
const unsigned char *)msg,
279 &buf_out_len) != 0) {
284 iovec.iov_base = (
void *)buf_out;
285 iovec.iov_len = buf_out_len;
290 memset(&msg_ucast, 0,
sizeof(msg_ucast));
293 msg_ucast.msg_name = &sockaddr;
294 msg_ucast.msg_namelen = addrlen;
295 msg_ucast.msg_iov = (
void *)&iovec;
296 msg_ucast.msg_iovlen = 1;
297 #ifdef HAVE_MSGHDR_CONTROL 298 msg_ucast.msg_control = 0;
300 #ifdef HAVE_MSGHDR_CONTROLLEN 301 msg_ucast.msg_controllen = 0;
303 #ifdef HAVE_MSGHDR_FLAGS 304 msg_ucast.msg_flags = 0;
306 #ifdef HAVE_MSGHDR_ACCRIGHTS 307 msg_ucast.msg_accrights = NULL;
309 #ifdef HAVE_MSGHDR_ACCRIGHTSLEN 310 msg_ucast.msg_accrightslen = 0;
322 "sendmsg(ucast) failed (non-critical)");
326 static inline void mcast_sendmsg (
329 unsigned int msg_len)
331 struct msghdr msg_mcast;
336 struct sockaddr_storage sockaddr;
344 (
const unsigned char *)msg,
347 &buf_out_len) != 0) {
352 iovec.iov_base = (
void *)&buf_out;
353 iovec.iov_len = buf_out_len;
360 memset(&msg_mcast, 0,
sizeof(msg_mcast));
361 msg_mcast.msg_name = &sockaddr;
362 msg_mcast.msg_namelen = addrlen;
363 msg_mcast.msg_iov = (
void *)&iovec;
364 msg_mcast.msg_iovlen = 1;
365 #ifdef HAVE_MSGHDR_CONTROL 366 msg_mcast.msg_control = 0;
368 #ifdef HAVE_MSGHDR_CONTROLLEN 369 msg_mcast.msg_controllen = 0;
371 #ifdef HAVE_MSGHDR_FLAGS 372 msg_mcast.msg_flags = 0;
374 #ifdef HAVE_MSGHDR_ACCRIGHTS 375 msg_mcast.msg_accrights = NULL;
377 #ifdef HAVE_MSGHDR_ACCRIGHTSLEN 378 msg_mcast.msg_accrightslen = 0;
389 "sendmsg(mcast) failed (non-critical)");
399 msg_mcast.msg_name = NULL;
400 msg_mcast.msg_namelen = 0;
406 "sendmsg(local mcast loop) failed (non-critical)");
444 static int net_deliver_fn (
450 struct msghdr msg_recv;
452 struct sockaddr_storage system_from;
466 msg_recv.msg_namelen =
sizeof (
struct sockaddr_storage);
467 msg_recv.msg_iov = iovec;
468 msg_recv.msg_iovlen = 1;
469 #ifdef HAVE_MSGHDR_CONTROL 470 msg_recv.msg_control = 0;
472 #ifdef HAVE_MSGHDR_CONTROLLEN 473 msg_recv.msg_controllen = 0;
475 #ifdef HAVE_MSGHDR_FLAGS 476 msg_recv.msg_flags = 0;
478 #ifdef HAVE_MSGHDR_ACCRIGHTS 479 msg_recv.msg_accrights = NULL;
481 #ifdef HAVE_MSGHDR_ACCRIGHTSLEN 482 msg_recv.msg_accrightslen = 0;
485 bytes_received = recvmsg (fd, &msg_recv,
MSG_NOSIGNAL | MSG_DONTWAIT);
486 if (bytes_received == -1) {
499 "Invalid packet data");
503 iovec->iov_len = bytes_received;
517 static int netif_determine (
527 interface_up, interface_num,
539 static void timer_function_netif_check_timeout (
550 netif_determine (instance,
553 &interface_up, &interface_num);
559 interface_up == 0) ||
563 interface_up == 1)) {
569 timer_function_netif_check_timeout,
570 &instance->timer_netif_check_timeout);
598 if (interface_up == 0) {
603 bind_address = &localhost;
612 timer_function_netif_check_timeout,
613 &instance->timer_netif_check_timeout);
624 (void)totemudp_build_sockets (instance,
634 POLLIN, instance, net_deliver_fn);
640 POLLIN, instance, net_deliver_fn);
646 POLLIN, instance, net_deliver_fn);
656 "The network interface [%s] is now up.",
669 timer_function_netif_check_timeout,
670 &instance->timer_netif_check_timeout);
676 "The network interface is down.");
686 static void totemudp_traffic_control_set(
struct totemudp_instance *instance,
int sock)
691 if (setsockopt(sock, SOL_SOCKET, SO_PRIORITY, &prio,
sizeof(
int))) {
697 static int totemudp_build_sockets_ip (
705 struct sockaddr_storage sockaddr;
706 struct ipv6_mreq mreq6;
708 struct sockaddr_storage mcast_ss, boundto_ss;
709 struct sockaddr_in6 *mcast_sin6 = (
struct sockaddr_in6 *)&mcast_ss;
710 struct sockaddr_in *mcast_sin = (
struct sockaddr_in *)&mcast_ss;
711 struct sockaddr_in *boundto_sin = (
struct sockaddr_in *)&boundto_ss;
712 unsigned int sendbuf_size;
713 unsigned int recvbuf_size;
714 unsigned int optlen =
sizeof (sendbuf_size);
715 unsigned int retries;
733 res = fcntl (sockets->
mcast_recv, F_SETFL, O_NONBLOCK);
736 "Could not set non-blocking operation on multicast socket");
744 if ( setsockopt(sockets->
mcast_recv, SOL_SOCKET, SO_REUSEADDR, (
char *)&flag, sizeof (flag)) < 0) {
746 "setsockopt(SO_REUSEADDR) failed");
759 for (i = 0; i < 2; i++) {
764 "Could not set non-blocking operation on multicast socket");
782 res = fcntl (sockets->
mcast_send, F_SETFL, O_NONBLOCK);
785 "Could not set non-blocking operation on multicast socket");
793 if ( setsockopt(sockets->
mcast_send, SOL_SOCKET, SO_REUSEADDR, (
char *)&flag, sizeof (flag)) < 0) {
795 "setsockopt(SO_REUSEADDR) failed");
800 &sockaddr, &addrlen);
804 res = bind (sockets->
mcast_send, (
struct sockaddr *)&sockaddr, addrlen);
809 "Unable to bind the socket to send multicast packets");
826 sockets->
token = socket (bindnet_address->
family, SOCK_DGRAM, 0);
827 if (sockets->
token == -1) {
834 res = fcntl (sockets->
token, F_SETFL, O_NONBLOCK);
837 "Could not set non-blocking operation on token socket");
845 if ( setsockopt(sockets->
token, SOL_SOCKET, SO_REUSEADDR, (
char *)&flag, sizeof (flag)) < 0) {
847 "setsockopt(SO_REUSEADDR) failed");
859 res = bind (sockets->
token, (
struct sockaddr *)&sockaddr, addrlen);
864 "Unable to bind UDP unicast socket");
883 res = setsockopt (sockets->
mcast_recv, SOL_SOCKET, SO_RCVBUF, &recvbuf_size, optlen);
886 "Unable to set SO_RCVBUF size on UDP mcast socket");
889 res = setsockopt (sockets->
mcast_send, SOL_SOCKET, SO_SNDBUF, &sendbuf_size, optlen);
892 "Unable to set SO_SNDBUF size on UDP mcast socket");
895 res = setsockopt (sockets->
local_mcast_loop[0], SOL_SOCKET, SO_RCVBUF, &recvbuf_size, optlen);
898 "Unable to set SO_RCVBUF size on UDP local mcast loop socket");
901 res = setsockopt (sockets->
local_mcast_loop[1], SOL_SOCKET, SO_SNDBUF, &sendbuf_size, optlen);
904 "Unable to set SO_SNDBUF size on UDP local mcast loop socket");
908 res = getsockopt (sockets->
mcast_recv, SOL_SOCKET, SO_RCVBUF, &recvbuf_size, &optlen);
911 "Receive multicast socket recv buffer size (%d bytes).", recvbuf_size);
914 res = getsockopt (sockets->
mcast_send, SOL_SOCKET, SO_SNDBUF, &sendbuf_size, &optlen);
917 "Transmit multicast socket send buffer size (%d bytes).", sendbuf_size);
920 res = getsockopt (sockets->
local_mcast_loop[0], SOL_SOCKET, SO_RCVBUF, &recvbuf_size, &optlen);
923 "Local receive multicast loop socket recv buffer size (%d bytes).", recvbuf_size);
926 res = getsockopt (sockets->
local_mcast_loop[1], SOL_SOCKET, SO_SNDBUF, &sendbuf_size, &optlen);
929 "Local transmit multicast loop socket send buffer size (%d bytes).", sendbuf_size);
940 unsigned int broadcast = 1;
942 if ((setsockopt(sockets->
mcast_recv, SOL_SOCKET,
943 SO_BROADCAST, &broadcast, sizeof (broadcast))) == -1) {
945 "setting broadcast option failed");
948 if ((setsockopt(sockets->
mcast_send, SOL_SOCKET,
949 SO_BROADCAST, &broadcast, sizeof (broadcast))) == -1) {
951 "setting broadcast option failed");
955 switch (bindnet_address->
family) {
957 memset(&mreq, 0,
sizeof(mreq));
958 mreq.imr_multiaddr.s_addr = mcast_sin->sin_addr.s_addr;
959 mreq.imr_interface.s_addr = boundto_sin->sin_addr.s_addr;
960 res = setsockopt (sockets->
mcast_recv, IPPROTO_IP, IP_ADD_MEMBERSHIP,
961 &mreq, sizeof (mreq));
964 "join ipv4 multicast group failed");
969 memset(&mreq6, 0,
sizeof(mreq6));
970 memcpy(&mreq6.ipv6mr_multiaddr, &mcast_sin6->sin6_addr,
sizeof(
struct in6_addr));
971 mreq6.ipv6mr_interface = interface_num;
973 res = setsockopt (sockets->
mcast_recv, IPPROTO_IPV6, IPV6_JOIN_GROUP,
974 &mreq6, sizeof (mreq6));
977 "join ipv6 multicast group failed");
989 switch ( bindnet_address->
family ) {
992 res = setsockopt (sockets->
mcast_send, IPPROTO_IP, IP_MULTICAST_LOOP,
993 &sflag, sizeof (sflag));
996 res = setsockopt (sockets->
mcast_send, IPPROTO_IPV6, IPV6_MULTICAST_LOOP,
997 &flag, sizeof (flag));
1001 "Unable to turn off multicast loopback");
1009 if (bindnet_address->
family == AF_INET6) {
1010 res = setsockopt (sockets->
mcast_send, IPPROTO_IPV6, IPV6_MULTICAST_HOPS,
1011 &flag, sizeof (flag));
1014 "set mcast v6 TTL failed");
1019 res = setsockopt(sockets->
mcast_send, IPPROTO_IP, IP_MULTICAST_TTL,
1020 &sflag,
sizeof(sflag));
1023 "set mcast v4 TTL failed");
1031 switch ( bindnet_address->
family ) {
1033 if (setsockopt (sockets->
mcast_send, IPPROTO_IP, IP_MULTICAST_IF,
1034 &boundto_sin->sin_addr, sizeof (boundto_sin->sin_addr)) < 0) {
1036 "cannot select interface for multicast packets (send)");
1039 if (setsockopt (sockets->
mcast_recv, IPPROTO_IP, IP_MULTICAST_IF,
1040 &boundto_sin->sin_addr, sizeof (boundto_sin->sin_addr)) < 0) {
1042 "cannot select interface for multicast packets (recv)");
1047 if (setsockopt (sockets->
mcast_send, IPPROTO_IPV6, IPV6_MULTICAST_IF,
1048 &interface_num, sizeof (interface_num)) < 0) {
1050 "cannot select interface for multicast packets (send v6)");
1053 if (setsockopt (sockets->
mcast_recv, IPPROTO_IPV6, IPV6_MULTICAST_IF,
1054 &interface_num, sizeof (interface_num)) < 0) {
1056 "cannot select interface for multicast packets (recv v6)");
1073 res = bind (sockets->
mcast_recv, (
struct sockaddr *)&sockaddr, addrlen);
1078 "Unable to bind the socket to receive multicast packets");
1095 static int totemudp_build_sockets (
1109 res = netif_determine (instance,
1121 res = totemudp_build_sockets_ip (instance, mcast_address,
1122 bindnet_address, sockets, bound_to, interface_num);
1127 "Unable to create sockets, exiting");
1132 totemudp_traffic_control_set(instance, sockets->
token);
1145 qb_loop_t *poll_handle,
1152 void (*deliver_fn) (
1155 unsigned int msg_len),
1157 void (*iface_change_fn) (
1161 void (*target_set_completed) (
1167 if (instance == NULL) {
1171 totemudp_instance_initialize (instance);
1230 100*QB_TIME_NS_IN_MSEC,
1232 timer_function_netif_check_timeout,
1233 &instance->timer_netif_check_timeout);
1235 *udp_context = instance;
1251 int processor_count)
1259 if (processor_count == 1) {
1264 timer_function_netif_check_timeout,
1265 &instance->timer_netif_check_timeout);
1282 for (i = 0; i < 2; i++) {
1294 ufd.events = POLLIN;
1295 nfds = poll (&ufd, 1, 0);
1296 if (nfds == 1 && ufd.revents & POLLIN) {
1297 net_deliver_fn (sock, ufd.revents, instance);
1299 }
while (nfds == 1);
1315 unsigned int msg_len)
1320 ucast_sendmsg (instance, &instance->
token_target, msg, msg_len);
1327 unsigned int msg_len)
1332 mcast_sendmsg (instance, msg, msg_len);
1340 unsigned int msg_len)
1345 mcast_sendmsg (instance, msg, msg_len);
1355 timer_function_netif_check_timeout (instance);
1372 const char *ret_char;
1411 struct sockaddr_storage system_from;
1412 struct msghdr msg_recv;
1415 int msg_processed = 0;
1423 msg_recv.msg_namelen =
sizeof (
struct sockaddr_storage);
1425 msg_recv.msg_iovlen = 1;
1426 #ifdef HAVE_MSGHDR_CONTROL 1427 msg_recv.msg_control = 0;
1429 #ifdef HAVE_MSGHDR_CONTROLLEN 1430 msg_recv.msg_controllen = 0;
1432 #ifdef HAVE_MSGHDR_FLAGS 1433 msg_recv.msg_flags = 0;
1435 #ifdef HAVE_MSGHDR_ACCRIGHTS 1436 msg_recv.msg_accrights = NULL;
1438 #ifdef HAVE_MSGHDR_ACCRIGHTSLEN 1439 msg_recv.msg_accrightslen = 0;
1442 for (i = 0; i < 2; i++) {
1454 ufd.events = POLLIN;
1455 nfds = poll (&ufd, 1, 0);
1456 if (nfds == 1 && ufd.revents & POLLIN) {
1457 res = recvmsg (sock, &msg_recv,
MSG_NOSIGNAL | MSG_DONTWAIT);
1464 }
while (nfds == 1);
1467 return (msg_processed);
unsigned int clear_node_high_bit
int totemudp_crypto_set(void *udp_context, const char *cipher_type, const char *hash_type)
int totemudp_processor_count_set(void *udp_context, int processor_count)
int totemip_localhost(int family, struct totem_ip_address *localhost)
struct iovec totemudp_iov_recv_flush
struct totem_interface * interfaces
unsigned int interface_count
struct totemudp_instance * instance
struct crypto_instance * crypto_inst
size_t crypto_sec_header_size(const char *crypto_cipher_type, const char *crypto_hash_type)
The totem_ip_address struct.
const char * totemip_print(const struct totem_ip_address *addr)
void(* totemudp_iface_change_fn)(void *context, const struct totem_ip_address *iface_address)
void(* totemudp_deliver_fn)(void *context, const void *msg, unsigned int msg_len)
void totemudp_net_mtu_adjust(void *udp_context, struct totem_config *totem_config)
void * totemudp_buffer_alloc(void)
int totemudp_mcast_flush_send(void *udp_context, const void *msg, unsigned int msg_len)
int totemudp_token_target_set(void *udp_context, const struct totem_ip_address *token_target)
unsigned char private_key[TOTEM_PRIVATE_KEY_LEN]
char iov_buffer[FRAME_SIZE_MAX]
struct totemudp_socket totemudp_sockets
unsigned char addr[TOTEMIP_ADDRLEN]
struct crypto_instance * crypto_init(const unsigned char *private_key, unsigned int private_key_len, const char *crypto_cipher_type, const char *crypto_hash_type, void(*log_printf_func)(int level, int subsys, const char *function, const char *file, int line, const char *format,...) __attribute__((format(printf, 6, 7))), int log_level_security, int log_level_notice, int log_level_error, int log_subsys_id)
void totemip_copy(struct totem_ip_address *addr1, const struct totem_ip_address *addr2)
struct totem_ip_address mcast_address
unsigned int downcheck_timeout
int totemudp_mcast_noflush_send(void *udp_context, const void *msg, unsigned int msg_len)
unsigned int private_key_len
int totemudp_log_level_security
#define BIND_STATE_REGULAR
struct totem_config * totem_config
#define NETIF_STATE_REPORT_DOWN
#define totemip_nosigpipe(s)
int totemudp_log_level_error
int totemudp_send_flush(void *udp_context)
int totemudp_recv_flush(void *udp_context)
struct iovec totemudp_iov_recv
int totemip_iface_check(struct totem_ip_address *bindnet, struct totem_ip_address *boundto, int *interface_up, int *interface_num, int mask_high_bit)
int crypto_encrypt_and_sign(struct crypto_instance *instance, const unsigned char *buf_in, const size_t buf_in_len, unsigned char *buf_out, size_t *buf_out_len)
int totemudp_token_send(void *udp_context, const void *msg, unsigned int msg_len)
struct totem_interface * totem_interface
int totemudp_initialize(qb_loop_t *poll_handle, void **udp_context, struct totem_config *totem_config, totemsrp_stats_t *stats, int interface_no, void *context, void(*deliver_fn)(void *context, const void *msg, unsigned int msg_len), void(*iface_change_fn)(void *context, const struct totem_ip_address *iface_address), void(*target_set_completed)(void *context))
Create an instance.
qb_loop_t * totemudp_poll_handle
unsigned int my_memb_entries
struct totem_ip_address mcast_addr
#define BIND_RETRIES_INTERVAL
void(* totemudp_log_printf)(int level, int subsys, const char *function, const char *file, int line, const char *format,...) __attribute__((format(printf
int totemudp_finalize(void *udp_context)
struct totem_ip_address boundto
#define BIND_STATE_LOOPBACK
size_t totemip_udpip_header_size(int family)
const char * totemudp_iface_print(void *udp_context)
#define NETIF_STATE_REPORT_UP
void(* log_printf)(int level, int subsys, const char *function_name, const char *file_name, int file_line, const char *format,...) __attribute__((format(printf
#define MCAST_SOCKET_BUFFER_SIZE
void(*) void udp_context)
char iov_buffer_flush[FRAME_SIZE_MAX]
int totemudp_log_level_debug
int crypto_authenticate_and_decrypt(struct crypto_instance *instance, unsigned char *buf, int *buf_len)
int totemudp_log_level_notice
unsigned int broadcast_use
int totemudp_iface_get(void *udp_context, struct totem_ip_address *addr)
#define LOGSYS_LEVEL_CRIT
void(* totemudp_target_set_completed)(void *context)
int totemip_totemip_to_sockaddr_convert(struct totem_ip_address *ip_addr, uint16_t port, struct sockaddr_storage *saddr, int *addrlen)
struct totem_logging_configuration totem_logging_configuration
#define LOGSYS_PERROR(err_num, level, fmt, args...)
struct srp_addr system_from
#define log_printf(level, format, args...)
char * crypto_cipher_type
struct totem_ip_address my_id
int totemudp_recv_mcast_empty(void *udp_context)
int totemudp_iface_check(void *udp_context)
struct totem_ip_address bindnet
uint32_t continuous_sendmsg_failures
struct totem_ip_address token_target
qb_loop_timer_handle timer_netif_check_timeout
void totemudp_buffer_release(void *ptr)
int totemudp_log_level_warning