initial commit

iris-legacy/iris/irisnet/jdns/qjdns_sock.cpp

@@ -0,0 +1,184 @@
+/*
+ * Copyright (C) 2005,2006  Justin Karneges
+ *
+ * Permission is hereby granted, free of charge, to any person obtaining a
+ * copy of this software and associated documentation files (the
+ * "Software"), to deal in the Software without restriction, including
+ * without limitation the rights to use, copy, modify, merge, publish,
+ * distribute, sublicense, and/or sell copies of the Software, and to
+ * permit persons to whom the Software is furnished to do so, subject to
+ * the following conditions:
+ *
+ * The above copyright notice and this permission notice shall be included
+ * in all copies or substantial portions of the Software.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
+ * OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+ * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.
+ * IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY
+ * CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
+ * TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
+ * SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
+ */
+
+#include "qjdns_sock.h"
+
+#include <QtGlobal>
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+
+#ifdef Q_OS_WIN
+# include <winsock2.h>
+# include <ws2tcpip.h>
+#endif
+
+#ifdef Q_OS_UNIX
+# include <sys/time.h>
+# include <sys/types.h>
+# include <sys/socket.h>
+# include <netinet/in.h>
+# include <fcntl.h>
+# include <errno.h>
+# include <signal.h>
+# include <arpa/inet.h>
+#endif
+
+#ifndef QT_NO_IPV6
+# define HAVE_IPV6
+# ifndef IPPROTO_IPV6
+#  define IPPROTO_IPV6 41
+   struct in6_addr
+   {
+     union
+     {
+       unsigned char  _S6_u8[16];
+       unsigned short _S6_u16[8];
+       unsigned long  _S6_u32[4];
+     } _S6_un;
+   };
+#  define s6_addr _S6_un._S6_u8
+# endif
+# ifndef IPV6_JOIN_GROUP
+#  define IPV6_JOIN_GROUP 12
+#  define IPV6_MULTICAST_HOPS 10
+   struct ipv6_mreq
+   {
+     struct in6_addr ipv6mr_multiaddr;
+     unsigned int ipv6mr_interface;
+   };
+# endif
+#endif
+
+static int get_last_error()
+{
+	int x;
+#ifdef Q_OS_WIN
+	x = WSAGetLastError();
+#else
+	x = errno;
+#endif
+	return x;
+}
+
+bool qjdns_sock_setMulticast4(int s, unsigned long int addr, int *errorCode)
+{
+	int ret;
+	struct ip_mreq mc;
+
+	memset(&mc, 0, sizeof(mc));
+	mc.imr_multiaddr.s_addr = htonl(addr);
+	mc.imr_interface.s_addr = INADDR_ANY;
+
+	ret = setsockopt(s, IPPROTO_IP, IP_ADD_MEMBERSHIP, (const char *)&mc, sizeof(mc));
+	if(ret != 0)
+	{
+		if(errorCode)
+			*errorCode = get_last_error();
+		return false;
+	}
+	return true;
+}
+
+bool qjdns_sock_setMulticast6(int s, unsigned char *addr, int *errorCode)
+{
+#ifdef HAVE_IPV6
+	int ret;
+	struct ipv6_mreq mc;
+
+	memset(&mc, 0, sizeof(mc));
+	memcpy(mc.ipv6mr_multiaddr.s6_addr, addr, 16);
+	mc.ipv6mr_interface = 0;
+
+	ret = setsockopt(s, IPPROTO_IPV6, IPV6_JOIN_GROUP, (const char *)&mc, sizeof(mc));
+	if(ret != 0)
+	{
+		if(errorCode)
+			*errorCode = get_last_error();
+		return false;
+	}
+	return true;
+#else
+	Q_UNUSED(s);
+	Q_UNUSED(addr);
+	Q_UNUSED(errorCode);
+	return false;
+#endif
+}
+
+bool qjdns_sock_setTTL4(int s, int ttl)
+{
+	unsigned char cttl;
+	int ret, ittl;
+
+	cttl = ttl;
+	ittl = ttl;
+
+	// IP_MULTICAST_TTL might take 1 byte or 4, try both
+	ret = setsockopt(s, IPPROTO_IP, IP_MULTICAST_TTL, (const char *)&cttl, sizeof(cttl));
+	if(ret != 0)
+	{
+		ret = setsockopt(s, IPPROTO_IP, IP_MULTICAST_TTL, (const char *)&ittl, sizeof(ittl));
+		if(ret != 0)
+			return false;
+	}
+	return true;
+}
+
+bool qjdns_sock_setTTL6(int s, int ttl)
+{
+#ifdef HAVE_IPV6
+	unsigned char cttl;
+	int ret, ittl;
+
+	cttl = ttl;
+	ittl = ttl;
+
+	// IPV6_MULTICAST_HOPS might take 1 byte or 4, try both
+	ret = setsockopt(s, IPPROTO_IPV6, IPV6_MULTICAST_HOPS, (const char *)&cttl, sizeof(cttl));
+	if(ret != 0)
+	{
+		ret = setsockopt(s, IPPROTO_IPV6, IPV6_MULTICAST_HOPS, (const char *)&ittl, sizeof(ittl));
+		if(ret != 0)
+			return false;
+	}
+	return true;
+#else
+	Q_UNUSED(s);
+	Q_UNUSED(ttl);
+	return false;
+#endif
+}
+
+bool qjdns_sock_setIPv6Only(int s)
+{
+#if defined(HAVE_IPV6) && defined(IPV6_V6ONLY)
+	int x = 1;
+	if(setsockopt(s, IPPROTO_IPV6, IPV6_V6ONLY, (char *)&x, sizeof(x)) != 0)
+		return false;
+	return true;
+#else
+	Q_UNUSED(s);
+	return false;
+#endif
+}