diff options
| author | 2021-06-07 02:01:35 +0200 | |
|---|---|---|
| committer | 2021-06-07 03:41:07 +0200 | |
| commit | 046b26150e004a8ac740e699c6c3e11be29e8f11 (patch) | |
| tree | 8f59597463dccd81ca180a87668e53864c4b09ae /src | |
| download | libimobiledevice-glue-046b26150e004a8ac740e699c6c3e11be29e8f11.tar.gz libimobiledevice-glue-046b26150e004a8ac740e699c6c3e11be29e8f11.tar.bz2  | |
Initial check-in of sources
Diffstat (limited to 'src')
| -rw-r--r-- | src/Makefile.am | 23 | ||||
| -rw-r--r-- | src/collection.c | 101 | ||||
| -rw-r--r-- | src/common.h | 38 | ||||
| -rw-r--r-- | src/libimobiledevice-glue-1.0.pc.in | 11 | ||||
| -rw-r--r-- | src/socket.c | 1169 | ||||
| -rw-r--r-- | src/thread.c | 203 | ||||
| -rw-r--r-- | src/utils.c | 551 | 
7 files changed, 2096 insertions, 0 deletions
diff --git a/src/Makefile.am b/src/Makefile.am new file mode 100644 index 0000000..2856eab --- /dev/null +++ b/src/Makefile.am @@ -0,0 +1,23 @@ +AM_CPPFLAGS = -I$(top_srcdir)/include -I$(top_srcdir) + +AM_CFLAGS = $(GLOBAL_CFLAGS) $(PTHREAD_CFLAGS) $(libplist_CFLAGS) + +AM_LDFLAGS = $(PTHREAD_LIBS) $(libplist_LIBS) + +lib_LTLIBRARIES = libimobiledevice-glue-1.0.la +libimobiledevice_glue_1_0_la_LDFLAGS = $(AM_LDFLAGS) -version-info $(LIBIMOBILEDEVICE_GLUE_SO_VERSION) -no-undefined +libimobiledevice_glue_1_0_la_LIBADD = +libimobiledevice_glue_1_0_la_SOURCES =	\ +	socket.c				\ +	thread.c				\ +	utils.c					\ +	collection.c				\ +	common.h + +if WIN32 +libimobiledevice_glue_1_0_la_LDFLAGS += -avoid-version -static-libgcc +libimobiledevice_glue_1_0_la_LIBADD += -lws2_32 -lIphlpapi +endif + +pkgconfigdir = $(libdir)/pkgconfig +pkgconfig_DATA = libimobiledevice-glue-1.0.pc diff --git a/src/collection.c b/src/collection.c new file mode 100644 index 0000000..ef47217 --- /dev/null +++ b/src/collection.c @@ -0,0 +1,101 @@ +/* + * collection.c + * + * Copyright (C) 2009 Hector Martin <hector@marcansoft.com> + * Copyright (C) 2009 Nikias Bassen <nikias@gmx.li> + * + * This library is free software; you can redistribute it and/or + * modify it under the terms of the GNU Lesser General Public + * License as published by the Free Software Foundation; either + * version 2.1 of the License, or (at your option) any later version. + * + * This library 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 + * Lesser General Public License for more details. + * + * You should have received a copy of the GNU Lesser General Public + * License along with this library; if not, write to the Free Software + * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA + */ + +#ifdef HAVE_CONFIG_H +#include <config.h> +#endif + +#include <stdlib.h> +#include <string.h> +#include <stdio.h> + +#include "common.h" +#include "libimobiledevice-glue/collection.h" + +#undef NDEBUG // we need to make sure we still get assertions because we can't handle memory allocation errors +#include <assert.h> + +#define INIT_NULL(addr, count) { unsigned int i_ = 0; for (i_ = 0; i_ < (count); i_++) ((void**)(addr))[i_] = NULL; } + +#define CAPACITY_STEP 8 + +LIBIMOBILEDEVICE_GLUE_API void collection_init(struct collection *col) +{ +	col->list = malloc(sizeof(void *) * CAPACITY_STEP); +	assert(col->list); +	INIT_NULL(col->list, CAPACITY_STEP); +	col->capacity = CAPACITY_STEP; +} + +LIBIMOBILEDEVICE_GLUE_API void collection_free(struct collection *col) +{ +	free(col->list); +	col->list = NULL; +	col->capacity = 0; +} + +LIBIMOBILEDEVICE_GLUE_API void collection_add(struct collection *col, void *element) +{ +	int i; +	for(i=0; i<col->capacity; i++) { +		if(!col->list[i]) { +			col->list[i] = element; +			return; +		} +	} +	void **newlist = realloc(col->list, sizeof(void*) * (col->capacity + CAPACITY_STEP)); +	assert(newlist); +	col->list = newlist; +	INIT_NULL(&col->list[col->capacity], CAPACITY_STEP); +	col->list[col->capacity] = element; +	col->capacity += CAPACITY_STEP; +} + +LIBIMOBILEDEVICE_GLUE_API int collection_remove(struct collection *col, void *element) +{ +	int i; +	for(i=0; i<col->capacity; i++) { +		if(col->list[i] == element) { +			col->list[i] = NULL; +			return 0; +		} +	} +	fprintf(stderr, "%s: WARNING: element %p not present in collection %p (cap %d)", __func__, element, col, col->capacity); +	return -1; +} + +LIBIMOBILEDEVICE_GLUE_API int collection_count(struct collection *col) +{ +	int i, cnt = 0; +	for(i=0; i<col->capacity; i++) { +		if(col->list[i]) +			cnt++; +	} +	return cnt; +} + +LIBIMOBILEDEVICE_GLUE_API void collection_copy(struct collection *dest, struct collection *src) +{ +	if (!dest || !src) return; +	dest->capacity = src->capacity; +	dest->list = malloc(sizeof(void*) * src->capacity); +	memcpy(dest->list, src->list, sizeof(void*) * src->capacity); +} diff --git a/src/common.h b/src/common.h new file mode 100644 index 0000000..bd22e3d --- /dev/null +++ b/src/common.h @@ -0,0 +1,38 @@ +/* + * common.h + * + * Copyright (c) 2020 Nikias Bassen, All Rights Reserved. + * + * This library is free software; you can redistribute it and/or + * modify it under the terms of the GNU Lesser General Public + * License as published by the Free Software Foundation; either + * version 2.1 of the License, or (at your option) any later version. + * + * This library 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 + * Lesser General Public License for more details. + * + * You should have received a copy of the GNU Lesser General Public + * License along with this library; if not, write to the Free Software + * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA + */ + +#ifndef __COMMON_H +#define __COMMON_H + +#ifdef HAVE_CONFIG_H +#include <config.h> +#endif + +#ifdef WIN32 +#define LIBIMOBILEDEVICE_GLUE_API __declspec( dllexport ) +#else +#ifdef HAVE_FVISIBILITY +#define LIBIMOBILEDEVICE_GLUE_API __attribute__((visibility("default"))) +#else +#define LIBIMOBILEDEVICE_GLUE_API +#endif +#endif + +#endif diff --git a/src/libimobiledevice-glue-1.0.pc.in b/src/libimobiledevice-glue-1.0.pc.in new file mode 100644 index 0000000..814a33c --- /dev/null +++ b/src/libimobiledevice-glue-1.0.pc.in @@ -0,0 +1,11 @@ +prefix=@prefix@ +exec_prefix=@exec_prefix@ +libdir=@libdir@ +includedir=@includedir@ + +Name: @PACKAGE_NAME@ +Description: Common library for libimobiledevice and co. +Version: @PACKAGE_VERSION@ +Libs: -L${libdir} -limobiledevice-glue-1.0 +Cflags: -I${includedir} +Requires: libplist-2.0 >= @LIBPLIST_VERSION@ diff --git a/src/socket.c b/src/socket.c new file mode 100644 index 0000000..067ef41 --- /dev/null +++ b/src/socket.c @@ -0,0 +1,1169 @@ +/* + * socket.c + * + * Copyright (C) 2012-2020 Nikias Bassen <nikias@gmx.li> + * Copyright (C) 2012 Martin Szulecki <m.szulecki@libimobiledevice.org> + * + * This library is free software; you can redistribute it and/or + * modify it under the terms of the GNU Lesser General Public + * License as published by the Free Software Foundation; either + * version 2.1 of the License, or (at your option) any later version. + * + * This library 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 + * Lesser General Public License for more details. + * + * You should have received a copy of the GNU Lesser General Public + * License along with this library; if not, write to the Free Software + * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA + */ + +#ifdef HAVE_CONFIG_H +#include <config.h> +#endif +#include <stdio.h> +#include <stddef.h> +#include <stdlib.h> +#include <string.h> +#include <unistd.h> +#include <errno.h> +#include <sys/time.h> +#include <sys/stat.h> +#ifdef WIN32 +#include <winsock2.h> +#include <ws2tcpip.h> +#include <windows.h> +#ifndef HAVE_GETIFADDRS +#include <iphlpapi.h> +#endif +static int wsa_init = 0; +#ifndef IFF_RUNNING +#define IFF_RUNNING IFF_UP +#endif +#ifndef AI_NUMERICSERV +#define AI_NUMERICSERV 0 +#endif +#else +#include <sys/socket.h> +#include <sys/un.h> +#include <netinet/in.h> +#include <netinet/tcp.h> +#include <netdb.h> +#include <arpa/inet.h> +#include <fcntl.h> +#ifdef AF_INET6 +#include <net/if.h> +#include <ifaddrs.h> +#endif +#endif +#include "common.h" +#include "libimobiledevice-glue/socket.h" + +#define RECV_TIMEOUT 20000 +#define SEND_TIMEOUT 10000 +#define CONNECT_TIMEOUT 5000 + +#ifndef EAFNOSUPPORT +#define EAFNOSUPPORT 102 +#endif +#ifndef ECONNRESET +#define ECONNRESET 108 +#endif +#ifndef ETIMEDOUT +#define ETIMEDOUT 138 +#endif + +static int verbose = 0; + +LIBIMOBILEDEVICE_GLUE_API void socket_set_verbose(int level) +{ +	verbose = level; +} + +LIBIMOBILEDEVICE_GLUE_API const char *socket_addr_to_string(struct sockaddr *addr, char *addr_out, size_t addr_out_size) +{ +#ifdef WIN32 +	WSADATA wsa_data; +	if (!wsa_init) { +		if (WSAStartup(MAKEWORD(2,2), &wsa_data) != ERROR_SUCCESS) { +			fprintf(stderr, "WSAStartup failed!\n"); +			ExitProcess(-1); +		} +		wsa_init = 1; +	} +	DWORD addr_out_len = addr_out_size; +	DWORD addrlen = 0; + +	if (addr->sa_family == AF_INET) { +		addrlen = sizeof(struct sockaddr_in); +	} +#ifdef AF_INET6 +	else if (addr->sa_family == AF_INET6) { +		addrlen = sizeof(struct sockaddr_in6); +	} +#endif +	else { +		errno = EAFNOSUPPORT; +		return NULL; +	} + +	if (WSAAddressToString(addr, addrlen, NULL, addr_out, &addr_out_len) == 0) { +		return addr_out; +	} +#else +	const void *addrdata = NULL; + +	if (addr->sa_family == AF_INET) { +		addrdata = &((struct sockaddr_in*)addr)->sin_addr; +	} +#ifdef AF_INET6 +	else if (addr->sa_family == AF_INET6) { +		addrdata = &((struct sockaddr_in6*)addr)->sin6_addr; +	} +#endif +	else { +		errno = EAFNOSUPPORT; +		return NULL; +	} + +	if (inet_ntop(addr->sa_family, addrdata, addr_out, addr_out_size)) { +		return addr_out; +	} +#endif +	return NULL; +} + +#ifndef WIN32 +LIBIMOBILEDEVICE_GLUE_API int socket_create_unix(const char *filename) +{ +	struct sockaddr_un name; +	int sock; +#ifdef SO_NOSIGPIPE +	int yes = 1; +#endif + +	// remove if still present +	unlink(filename); + +	/* Create the socket. */ +	sock = socket(PF_UNIX, SOCK_STREAM, 0); +	if (sock < 0) { +		perror("socket"); +		return -1; +	} + +#ifdef SO_NOSIGPIPE +	if (setsockopt(sock, SOL_SOCKET, SO_NOSIGPIPE, (void*)&yes, sizeof(int)) == -1) { +		perror("setsockopt()"); +		socket_close(sock); +		return -1; +	} +#endif + +	/* Bind a name to the socket. */ +	name.sun_family = AF_UNIX; +	strncpy(name.sun_path, filename, sizeof(name.sun_path)); +	name.sun_path[sizeof(name.sun_path) - 1] = '\0'; + +	if (bind(sock, (struct sockaddr*)&name, sizeof(name)) < 0) { +		perror("bind"); +		socket_close(sock); +		return -1; +	} + +	if (listen(sock, 100) < 0) { +		perror("listen"); +		socket_close(sock); +		return -1; +	} + +	return sock; +} + +LIBIMOBILEDEVICE_GLUE_API int socket_connect_unix(const char *filename) +{ +	struct sockaddr_un name; +	int sfd = -1; +	struct stat fst; +#ifdef SO_NOSIGPIPE +	int yes = 1; +#endif +	int bufsize = 0x20000; + +	// check if socket file exists... +	if (stat(filename, &fst) != 0) { +		if (verbose >= 2) +			fprintf(stderr, "%s: stat '%s': %s\n", __func__, filename, +					strerror(errno)); +		return -1; +	} +	// ... and if it is a unix domain socket +	if (!S_ISSOCK(fst.st_mode)) { +		if (verbose >= 2) +			fprintf(stderr, "%s: File '%s' is not a socket!\n", __func__, +					filename); +		return -1; +	} +	// make a new socket +	if ((sfd = socket(PF_UNIX, SOCK_STREAM, 0)) < 0) { +		if (verbose >= 2) +			fprintf(stderr, "%s: socket: %s\n", __func__, strerror(errno)); +		return -1; +	} + +	if (setsockopt(sfd, SOL_SOCKET, SO_SNDBUF, (void*)&bufsize, sizeof(int)) == -1) { +		perror("Could not set send buffer for socket"); +	} + +	if (setsockopt(sfd, SOL_SOCKET, SO_RCVBUF, (void*)&bufsize, sizeof(int)) == -1) { +		perror("Could not set receive buffer for socket"); +	} + +#ifdef SO_NOSIGPIPE +	if (setsockopt(sfd, SOL_SOCKET, SO_NOSIGPIPE, (void*)&yes, sizeof(int)) == -1) { +		perror("setsockopt()"); +		socket_close(sfd); +		return -1; +	} +#endif +	// and connect to 'filename' +	name.sun_family = AF_UNIX; +	strncpy(name.sun_path, filename, sizeof(name.sun_path)); +	name.sun_path[sizeof(name.sun_path) - 1] = 0; + +	int flags = fcntl(sfd, F_GETFL, 0); +	fcntl(sfd, F_SETFL, flags | O_NONBLOCK); + +	do { +		if (connect(sfd, (struct sockaddr*)&name, sizeof(name)) != -1) { +			break; +		} +		if (errno == EINPROGRESS) { +			fd_set fds; +			FD_ZERO(&fds); +			FD_SET(sfd, &fds); + +			struct timeval timeout; +			timeout.tv_sec = CONNECT_TIMEOUT / 1000; +			timeout.tv_usec = (CONNECT_TIMEOUT - (timeout.tv_sec * 1000)) * 1000; +			if (select(sfd + 1, NULL, &fds, NULL, &timeout) == 1) { +				int so_error; +				socklen_t len = sizeof(so_error); +				getsockopt(sfd, SOL_SOCKET, SO_ERROR, (void*)&so_error, &len); +				if (so_error == 0) { +					break; +				} +			} +		} +		socket_close(sfd); +		sfd = -1; +	} while (0); + +	if (sfd < 0) { +		if (verbose >= 2) +			fprintf(stderr, "%s: connect: %s\n", __func__, strerror(errno)); +		return -1; +	} + +	return sfd; +} +#endif + +LIBIMOBILEDEVICE_GLUE_API int socket_create(const char* addr, uint16_t port) +{ +	int sfd = -1; +	int yes = 1; +	struct addrinfo hints; +	struct addrinfo *result, *rp; +	char portstr[8]; +	int res; +#ifdef WIN32 +	WSADATA wsa_data; +	if (!wsa_init) { +		if (WSAStartup(MAKEWORD(2,2), &wsa_data) != ERROR_SUCCESS) { +			fprintf(stderr, "WSAStartup failed!\n"); +			ExitProcess(-1); +		} +		wsa_init = 1; +	} +#endif + +	memset(&hints, '\0', sizeof(struct addrinfo)); +	hints.ai_family = AF_UNSPEC; +	hints.ai_socktype = SOCK_STREAM; +	hints.ai_flags = AI_PASSIVE | AI_NUMERICSERV; +	hints.ai_protocol = IPPROTO_TCP; + +	sprintf(portstr, "%d", port); + +	if (!addr) { +		addr = "localhost"; +	} +	res = getaddrinfo(addr, portstr, &hints, &result); +	if (res != 0) { +		fprintf(stderr, "%s: getaddrinfo: %s\n", __func__, gai_strerror(res)); +		return -1; +	} + +	for (rp = result; rp != NULL; rp = rp->ai_next) { +		sfd = socket(rp->ai_family, rp->ai_socktype, rp->ai_protocol); +		if (sfd == -1) { +			continue; +		} + +		if (setsockopt(sfd, SOL_SOCKET, SO_REUSEADDR, (void*)&yes, sizeof(int)) == -1) { +			perror("setsockopt()"); +			socket_close(sfd); +			continue; +		} + +#ifdef SO_NOSIGPIPE +		if (setsockopt(sfd, SOL_SOCKET, SO_NOSIGPIPE, (void*)&yes, sizeof(int)) == -1) { +			perror("setsockopt()"); +			socket_close(sfd); +			continue; +		} +#endif + +#if defined(AF_INET6) && defined(IPV6_V6ONLY) +		if (rp->ai_family == AF_INET6) { +			if (setsockopt(sfd, IPPROTO_IPV6, IPV6_V6ONLY, (void*)&yes, sizeof(int)) == -1) { +				perror("setsockopt() IPV6_V6ONLY"); +			} +		} +#endif + +		if (bind(sfd, rp->ai_addr, rp->ai_addrlen) < 0) { +			perror("bind()"); +			socket_close(sfd); +			continue; +		} + +		if (listen(sfd, 100) < 0) { +			perror("listen()"); +			socket_close(sfd); +			continue; +		} +		break; +	} + +	freeaddrinfo(result); + +	if (rp == NULL) { +		return -1; +	} + +	return sfd; +} + +#ifdef AF_INET6 +static uint32_t _in6_addr_scope(struct in6_addr* addr) +{ +	uint32_t scope = 0; + +	if (IN6_IS_ADDR_MULTICAST(addr)) { +		if (IN6_IS_ADDR_MC_NODELOCAL(addr)) { +			scope = 1; +		} else if (IN6_IS_ADDR_MC_LINKLOCAL(addr)) { +			scope = 2; +		} else if (IN6_IS_ADDR_MC_SITELOCAL(addr)) { +			scope = 5; +		} + +		return scope; +	} + +	if (IN6_IS_ADDR_LINKLOCAL(addr)) { +		scope = 2; +	} else if (IN6_IS_ADDR_LOOPBACK(addr)) { +		scope = 2; +	} else if (IN6_IS_ADDR_SITELOCAL(addr)) { +		scope = 5; +	} else if (IN6_IS_ADDR_UNSPECIFIED(addr)) { +		scope = 0; +	} + +	return scope; +} + +#ifndef HAVE_GETIFADDRS +#ifdef WIN32 + +struct ifaddrs { +	struct ifaddrs  *ifa_next;    /* Next item in list */ +	char            *ifa_name;    /* Name of interface */ +	unsigned int     ifa_flags;   /* Flags from SIOCGIFFLAGS */ +	struct sockaddr *ifa_addr;    /* Address of interface */ +	struct sockaddr *ifa_netmask; /* Netmask of interface */ +	union { +		struct sockaddr *ifu_broadaddr; /* Broadcast address of interface */ +		struct sockaddr *ifu_dstaddr;   /* Point-to-point destination address */ +	} ifa_ifu; +#define                  ifa_broadaddr ifa_ifu.ifu_broadaddr +#define                  ifa_dstaddr   ifa_ifu.ifu_dstaddr +	void            *ifa_data;    /* Address-specific data */ +}; + +#define WORKING_BUFFER_SIZE 15000 +#define MAX_TRIES 3 + +static void freeifaddrs(struct ifaddrs *ifa) +{ +	if (!ifa) { +		return; +	} +	free(ifa->ifa_name); +	free(ifa->ifa_addr); +	free(ifa->ifa_netmask); +	free(ifa->ifa_dstaddr); +	freeifaddrs(ifa->ifa_next); +	free(ifa); +} + +/* + * getifaddrs() reference implementation for win32. + * Heavily based on openpgm's implementation found here: + * https://github.com/steve-o/openpgm/blob/master/openpgm/pgm/getifaddrs.c + */ +static int getifaddrs(struct ifaddrs** ifap) +{ +	struct ifaddrs* ifa = NULL; + +	DWORD dwRetVal = 0; + +	PIP_ADAPTER_ADDRESSES pAddresses = NULL; +	ULONG outBufLen = 0; +	ULONG Iterations = 0; + +	ULONG flags = GAA_FLAG_INCLUDE_PREFIX | +		GAA_FLAG_SKIP_ANYCAST | +		GAA_FLAG_SKIP_DNS_SERVER | +		GAA_FLAG_SKIP_FRIENDLY_NAME | +		GAA_FLAG_SKIP_MULTICAST; + +	PIP_ADAPTER_ADDRESSES adapter = NULL; + +	if (!ifap) { +		errno = EINVAL; +		return -1; +	} +	*ifap = NULL; + +	outBufLen = WORKING_BUFFER_SIZE; +	do { +		pAddresses = (IP_ADAPTER_ADDRESSES*)malloc(outBufLen); +		if (pAddresses == NULL) { +			printf("Memory allocation failed for IP_ADAPTER_ADDRESSES struct\n"); +			return -1; +		} +		dwRetVal = GetAdaptersAddresses(AF_UNSPEC, flags, NULL, pAddresses, &outBufLen); +		if (dwRetVal == ERROR_BUFFER_OVERFLOW) { +			free(pAddresses); +			pAddresses = NULL; +		} else { +			break; +		} +		Iterations++; +	} while ((dwRetVal == ERROR_BUFFER_OVERFLOW) && (Iterations < MAX_TRIES)); + +	if (dwRetVal != NO_ERROR) { +		free(pAddresses); +		return -1; +	} + +	for (adapter = pAddresses; adapter; adapter = adapter->Next) { +		int unicastIndex = 0; +		for (IP_ADAPTER_UNICAST_ADDRESS *unicast = adapter->FirstUnicastAddress; unicast; unicast = unicast->Next, ++unicastIndex) { +			/* ensure IP adapter */ +			if (AF_INET != unicast->Address.lpSockaddr->sa_family && AF_INET6 != unicast->Address.lpSockaddr->sa_family) { +				continue; +			} + +			if (!ifa) { +				ifa = malloc(sizeof(struct ifaddrs)); +				if (!ifa) { +					errno = ENOMEM; +					free(pAddresses); +					return -1; +				} +				*ifap = ifa; +				ifa->ifa_next = NULL; +			} else { +				struct ifaddrs* ifanew = malloc(sizeof(struct ifaddrs)); +				if (!ifanew) { +					freeifaddrs(*ifap); +					free(pAddresses); +					errno = ENOMEM; +					return -1; +				} +				ifa->ifa_next = ifanew; +				ifa = ifanew; +				ifa->ifa_next = NULL; +			} + +			/* name */ +			ifa->ifa_name = strdup(adapter->AdapterName); + +			/* flags */ +			ifa->ifa_flags = 0; +			if (IfOperStatusUp == adapter->OperStatus) +				ifa->ifa_flags |= IFF_UP; +			if (IF_TYPE_SOFTWARE_LOOPBACK == adapter->IfType) +				ifa->ifa_flags |= IFF_LOOPBACK; +			if (!(adapter->Flags & IP_ADAPTER_NO_MULTICAST)) +				ifa->ifa_flags |= IFF_MULTICAST; + +			/* address */ +			ifa->ifa_addr = (struct sockaddr*)malloc(sizeof(struct sockaddr_storage)); +			memcpy(ifa->ifa_addr, unicast->Address.lpSockaddr, unicast->Address.iSockaddrLength); + +			/* netmask */ +			ifa->ifa_netmask = (struct sockaddr*)malloc(sizeof(struct sockaddr_storage)); +			memset(ifa->ifa_netmask, 0, sizeof(struct sockaddr_storage)); + +/* pre-Vista must hunt for matching prefix in linked list, otherwise use + * OnLinkPrefixLength from IP_ADAPTER_UNICAST_ADDRESS structure. + * FirstPrefix requires Windows XP SP1, from SP1 to pre-Vista provides a + * single adapter prefix for each IP address.  Vista and later provides + * host IP address prefix, subnet IP address, and subnet broadcast IP + * address.  In addition there is a multicast and broadcast address prefix. + */ +			ULONG prefixLength = 0; + +#if defined( _WIN32 ) && ( _WIN32_WINNT >= 0x0600 ) +/* For a unicast IPv4 address, any value greater than 32 is an illegal + * value. For a unicast IPv6 address, any value greater than 128 is an + * illegal value. A value of 255 is commonly used to represent an illegal + * value. + * + * Windows 7 SP1 returns 64 for Teredo links which is incorrect. + */ + +#define IN6_IS_ADDR_TEREDO(addr) \ +	(((const uint32_t *)(addr))[0] == ntohl (0x20010000)) + +			if (AF_INET6 == unicast->Address.lpSockaddr->sa_family && +/* TunnelType only applies to one interface on the adapter and no + * convenient method is provided to determine which. + */ +				TUNNEL_TYPE_TEREDO == adapter->TunnelType && +/* Test the interface with the known Teredo network prefix. + */ +				IN6_IS_ADDR_TEREDO( &((struct sockaddr_in6*)(unicast->Address.lpSockaddr))->sin6_addr) && +/* Test that this version is actually wrong, subsequent releases from Microsoft + * may resolve the issue. + */ +				32 != unicast->OnLinkPrefixLength) +			{ +				prefixLength = 32; +			} +			else +				prefixLength = unicast->OnLinkPrefixLength; +#else +/* The order of linked IP_ADAPTER_UNICAST_ADDRESS structures pointed to by + * the FirstUnicastAddress member does not have any relationship with the + * order of linked IP_ADAPTER_PREFIX structures pointed to by the FirstPrefix + * member. + * + * Example enumeration: + *    [ no subnet ] + *   ::1/128            - address + *   ff00::%1/8         - multicast (no IPv6 broadcast) + *   127.0.0.0/8        - subnet + *   127.0.0.1/32       - address + *   127.255.255.255/32 - subnet broadcast + *   224.0.0.0/4        - multicast + *   255.255.255.255/32 - broadcast + * + * Which differs from most adapters listing three IPv6: + *   fe80::%10/64       - subnet + *   fe80::51e9:5fe5:4202:325a%10/128 - address + *   ff00::%10/8        - multicast + * + * !IfOperStatusUp IPv4 addresses are skipped: + *   fe80::%13/64       - subnet + *   fe80::d530:946d:e8df:8c91%13/128 - address + *   ff00::%13/8        - multicast + *    [ no subnet  ] + *    [ no address ] + *   224.0.0.0/4        - multicast + *   255.255.255.255/32 - broadcast + * + * On PTP links no multicast or broadcast addresses are returned: + *    [ no subnet ] + *   fe80::5efe:10.203.9.30/128 - address + *    [ no multicast ] + *    [ no multicast ] + *    [ no broadcast ] + * + * Active primary IPv6 interfaces are a bit overloaded: + *   ::/0               - default route + *   2001::/32          - global subnet + *   2001:0:4137:9e76:2443:d6:ba87:1a2a/128 - global address + *   fe80::/64          - link-local subnet + *   fe80::2443:d6:ba87:1a2a/128 - link-local address + *   ff00::/8           - multicast + */ + +#define IN_LINKLOCAL(a)	((((uint32_t) (a)) & 0xaffff0000) == 0xa9fe0000) + +			for (IP_ADAPTER_PREFIX *prefix = adapter->FirstPrefix; prefix; prefix = prefix->Next) { +				LPSOCKADDR lpSockaddr = prefix->Address.lpSockaddr; +				if (lpSockaddr->sa_family != unicast->Address.lpSockaddr->sa_family) +					continue; +/* special cases */ +/* RFC2863: IPv4 interface not up */ +				if (AF_INET == lpSockaddr->sa_family && adapter->OperStatus != IfOperStatusUp) { +/* RFC3927: link-local IPv4 always has 16-bit CIDR */ +					if (IN_LINKLOCAL( ntohl (((struct sockaddr_in*)(unicast->Address.lpSockaddr))->sin_addr.s_addr))) { +						prefixLength = 16; +					} +					break; +				} +/* default IPv6 route */ +				if (AF_INET6 == lpSockaddr->sa_family && 0 == prefix->PrefixLength && IN6_IS_ADDR_UNSPECIFIED( &((struct sockaddr_in6*)(lpSockaddr))->sin6_addr)) { +					continue; +				} +/* Assume unicast address for first prefix of operational adapter */ +				if (AF_INET == lpSockaddr->sa_family) +					if (IN_MULTICAST( ntohl (((struct sockaddr_in*)(lpSockaddr))->sin_addr.s_addr))) { +						fprintf(stderr, "FATAL: first prefix is non a unicast address\n"); +						break; +					} +				if (AF_INET6 == lpSockaddr->sa_family) +					if (IN6_IS_ADDR_MULTICAST( &((struct sockaddr_in6*)(lpSockaddr))->sin6_addr)) { +						fprintf(stderr, "FATAL: first prefix is not a unicast address\n"); +						break; +					} +/* Assume subnet or host IP address for XP backward compatibility */ + +				prefixLength = prefix->PrefixLength; +				break; +			} +#endif /* defined( _WIN32 ) && ( _WIN32_WINNT >= 0x0600 ) */ + +/* map prefix to netmask */ +			ifa->ifa_netmask->sa_family = unicast->Address.lpSockaddr->sa_family; +			switch (unicast->Address.lpSockaddr->sa_family) { +			case AF_INET: +				if (0 == prefixLength || prefixLength > 32) { +					prefixLength = 32; +				} +#if defined( _WIN32) && ( _WIN32_WINNT >= 0x0600 ) +/* Added in Vista, but no IPv6 equivalent. */ +				{ +				ULONG Mask; +				ConvertLengthToIpv4Mask (prefixLength, &Mask); +				((struct sockaddr_in*)ifa->ifa_netmask)->sin_addr.s_addr = Mask;	/* network order */ +				} +#else +/* NB: left-shift of full bit-width is undefined in C standard. */ +				((struct sockaddr_in*)ifa->ifa_netmask)->sin_addr.s_addr = htonl( 0xffffffffU << ( 32 - prefixLength ) ); +#endif +				break; + +			case AF_INET6: +				if (0 == prefixLength || prefixLength > 128) { +					prefixLength = 128; +				} +				for (LONG i = prefixLength, j = 0; i > 0; i -= 8, ++j) { +					((struct sockaddr_in6*)ifa->ifa_netmask)->sin6_addr.s6_addr[ j ] = i >= 8 ? 0xff : (ULONG)(( 0xffU << ( 8 - i ) ) & 0xffU ); +				} +				break; +			default: +				break; +			} +		} +	} +	free(pAddresses); + +	return 0; +} +#else +#error No reference implementation for getifaddrs available for this platform. +#endif +#endif + +static int32_t _sockaddr_in6_scope_id(struct sockaddr_in6* addr) +{ +	int32_t res = -1; +	struct ifaddrs *ifaddr = NULL, *ifa = NULL; +	uint32_t addr_scope; + +	/* get scope for requested address */ +	addr_scope = _in6_addr_scope(&addr->sin6_addr); +	if (addr_scope == 0) { +		/* global scope doesn't need a specific scope id */ +		return addr_scope; +	} + +	/* get interfaces */ +	if (getifaddrs(&ifaddr) == -1) { +		perror("getifaddrs"); +		return res; +	} + +	/* loop over interfaces */ +	for (ifa = ifaddr; ifa != NULL; ifa = ifa->ifa_next) { +		/* skip if no address is available */ +		if (ifa->ifa_addr == NULL) { +			continue; +		} + +		/* skip if wrong family */ +		if (ifa->ifa_addr->sa_family != AF_INET6) { +			continue; +		} + +		/* skip if not up */ +		if ((ifa->ifa_flags & IFF_UP) == 0) { +			continue; +		} + +		/* skip if not running */ +		if ((ifa->ifa_flags & IFF_RUNNING) == 0) { +			continue; +		} + +		struct sockaddr_in6* addr_in = (struct sockaddr_in6*)ifa->ifa_addr; + +		/* skip if scopes do not match */ +		if (_in6_addr_scope(&addr_in->sin6_addr) != addr_scope) { +			continue; +		} + +		/* use if address is equal */ +		if (memcmp(&addr->sin6_addr.s6_addr, &addr_in->sin6_addr.s6_addr, sizeof(addr_in->sin6_addr.s6_addr)) == 0) { +			/* if scope id equals the requested one then assume it was valid */ +			if (addr->sin6_scope_id == addr_in->sin6_scope_id) { +				res = addr_in->sin6_scope_id; +				break; +			} + +			if ((addr_in->sin6_scope_id > addr->sin6_scope_id) && (res >= 0)) { +				// use last valid scope id as we're past the requested scope id +				break; +			} +			res = addr_in->sin6_scope_id; +			continue; +		} + +		/* skip loopback interface if not already matched exactly above */ +		if ((ifa->ifa_flags & IFF_LOOPBACK) != 0) { +			continue; +		} + +		if ((addr_in->sin6_scope_id > addr->sin6_scope_id) && (res >= 0)) { +			// use last valid scope id as we're past the requested scope id +			break; +		} + +		res = addr_in->sin6_scope_id; + +		/* if scope id equals the requested one then assume it was valid */ +		if (addr->sin6_scope_id == addr_in->sin6_scope_id) { +			/* set the scope id of this interface as most likely candidate */ +			break; +		} +	} + +	freeifaddrs(ifaddr); + +	return res; +} +#endif + +LIBIMOBILEDEVICE_GLUE_API int socket_connect_addr(struct sockaddr* addr, uint16_t port) +{ +	int sfd = -1; +	int yes = 1; +	int bufsize = 0x20000; +	int addrlen = 0; +#ifdef WIN32 +	u_long l_yes = 1; +	WSADATA wsa_data; +	if (!wsa_init) { +		if (WSAStartup(MAKEWORD(2,2), &wsa_data) != ERROR_SUCCESS) { +			fprintf(stderr, "WSAStartup failed!\n"); +			ExitProcess(-1); +		} +		wsa_init = 1; +	} +#endif + +	if (addr->sa_family == AF_INET) { +		struct sockaddr_in* addr_in = (struct sockaddr_in*)addr; +		addr_in->sin_port = htons(port); +		addrlen = sizeof(struct sockaddr_in); +	} +#ifdef AF_INET6 +	else if (addr->sa_family == AF_INET6) { +		struct sockaddr_in6* addr_in = (struct sockaddr_in6*)addr; +		addr_in->sin6_port = htons(port); + +		/* +		 * IPv6 Routing Magic: +		 * +		 * If the scope of the address is a link-local one, IPv6 requires the +		 * scope id set to an interface number to allow proper routing. However, +		 * as the provided sockaddr might contain a wrong scope id, we must find +		 * a scope id from a suitable interface on this system or routing might +		 * fail. An IPv6 guru should have another look though... +		 */ +		addr_in->sin6_scope_id = _sockaddr_in6_scope_id(addr_in); + +		addrlen = sizeof(struct sockaddr_in6); +	} +#endif +	else { +		fprintf(stderr, "ERROR: Unsupported address family"); +		return -1; +	} + +	sfd = socket(addr->sa_family, SOCK_STREAM, IPPROTO_TCP); +	if (sfd == -1) { +		perror("socket()"); +		return -1; +	} + +#ifdef SO_NOSIGPIPE +	if (setsockopt(sfd, SOL_SOCKET, SO_NOSIGPIPE, (void*)&yes, sizeof(int)) == -1) { +		perror("setsockopt()"); +		socket_close(sfd); +		return -1; +	} +#endif + +	if (setsockopt(sfd, SOL_SOCKET, SO_REUSEADDR, (void*)&yes, sizeof(int)) == -1) { +		perror("setsockopt()"); +		socket_close(sfd); +		return -1; +	} + +#ifdef WIN32 +	ioctlsocket(sfd, FIONBIO, &l_yes); +#else +	int flags = fcntl(sfd, F_GETFL, 0); +	fcntl(sfd, F_SETFL, flags | O_NONBLOCK); +#endif + +	do { +		if (connect(sfd, addr, addrlen) != -1) { +			break; +		} +#ifdef WIN32 +		if (WSAGetLastError() == WSAEWOULDBLOCK) +#else +		if (errno == EINPROGRESS) +#endif +		{ +			fd_set fds; +			FD_ZERO(&fds); +			FD_SET(sfd, &fds); + +			struct timeval timeout; +			timeout.tv_sec = CONNECT_TIMEOUT / 1000; +			timeout.tv_usec = (CONNECT_TIMEOUT - (timeout.tv_sec * 1000)) * 1000; +			if (select(sfd + 1, NULL, &fds, NULL, &timeout) == 1) { +				int so_error; +				socklen_t len = sizeof(so_error); +				getsockopt(sfd, SOL_SOCKET, SO_ERROR, (void*)&so_error, &len); +				if (so_error == 0) { +					errno = 0; +					break; +				} +				errno = so_error; +			} +		} +		socket_close(sfd); +		sfd = -1; +	} while (0); + +	if (sfd < 0) { +		if (verbose >= 2) { +			char addrtxt[48]; +			socket_addr_to_string(addr, addrtxt, sizeof(addrtxt)); +			fprintf(stderr, "%s: Could not connect to %s port %d\n", __func__, addrtxt, port); +		} +		return -1; +	} + +	if (setsockopt(sfd, IPPROTO_TCP, TCP_NODELAY, (void*)&yes, sizeof(int)) == -1) { +		perror("Could not set TCP_NODELAY on socket"); +	} + +	if (setsockopt(sfd, SOL_SOCKET, SO_SNDBUF, (void*)&bufsize, sizeof(int)) == -1) { +		perror("Could not set send buffer for socket"); +	} + +	if (setsockopt(sfd, SOL_SOCKET, SO_RCVBUF, (void*)&bufsize, sizeof(int)) == -1) { +		perror("Could not set receive buffer for socket"); +	} + +	return sfd; +} + +LIBIMOBILEDEVICE_GLUE_API int socket_connect(const char *addr, uint16_t port) +{ +	int sfd = -1; +	int yes = 1; +	int bufsize = 0x20000; +	struct addrinfo hints; +	struct addrinfo *result, *rp; +	char portstr[8]; +	int res; +#ifdef WIN32 +	u_long l_yes = 1; +	WSADATA wsa_data; +	if (!wsa_init) { +		if (WSAStartup(MAKEWORD(2,2), &wsa_data) != ERROR_SUCCESS) { +			fprintf(stderr, "WSAStartup failed!\n"); +			ExitProcess(-1); +		} +		wsa_init = 1; +	} +#else +	int flags = 0; +#endif + +	if (!addr) { +		errno = EINVAL; +		return -1; +	} + +	memset(&hints, '\0', sizeof(struct addrinfo)); +	hints.ai_family = AF_UNSPEC; +	hints.ai_socktype = SOCK_STREAM; +	hints.ai_flags = AI_NUMERICSERV; +	hints.ai_protocol = IPPROTO_TCP; + +	sprintf(portstr, "%d", port); + +	res = getaddrinfo(addr, portstr, &hints, &result); +	if (res != 0) { +		fprintf(stderr, "%s: getaddrinfo: %s\n", __func__, gai_strerror(res)); +		return -1; +	} + +	for (rp = result; rp != NULL; rp = rp->ai_next) { +		sfd = socket(rp->ai_family, rp->ai_socktype, rp->ai_protocol); +		if (sfd == -1) { +			continue; +		} + +#ifdef SO_NOSIGPIPE +		if (setsockopt(sfd, SOL_SOCKET, SO_NOSIGPIPE, (void*)&yes, sizeof(int)) == -1) { +			perror("setsockopt()"); +			socket_close(sfd); +			return -1; +		} +#endif + +		if (setsockopt(sfd, SOL_SOCKET, SO_REUSEADDR, (void*)&yes, sizeof(int)) == -1) { +			perror("setsockopt()"); +			socket_close(sfd); +			continue; +		} + +#ifdef WIN32 +		ioctlsocket(sfd, FIONBIO, &l_yes); +#else +		flags = fcntl(sfd, F_GETFL, 0); +		fcntl(sfd, F_SETFL, flags | O_NONBLOCK); +#endif + +		if (connect(sfd, rp->ai_addr, rp->ai_addrlen) != -1) { +			break; +		} +#ifdef WIN32 +		if (WSAGetLastError() == WSAEWOULDBLOCK) +#else +		if (errno == EINPROGRESS) +#endif +		{ +			fd_set fds; +			FD_ZERO(&fds); +			FD_SET(sfd, &fds); + +			struct timeval timeout; +			timeout.tv_sec = CONNECT_TIMEOUT / 1000; +			timeout.tv_usec = (CONNECT_TIMEOUT - (timeout.tv_sec * 1000)) * 1000; +			if (select(sfd + 1, NULL, &fds, NULL, &timeout) == 1) { +				int so_error; +				socklen_t len = sizeof(so_error); +				getsockopt(sfd, SOL_SOCKET, SO_ERROR, (void*)&so_error, &len); +				if (so_error == 0) { +					break; +				} +			} +		} +		socket_close(sfd); +	} + +	freeaddrinfo(result); + +	if (rp == NULL) { +		if (verbose >= 2) +			fprintf(stderr, "%s: Could not connect to %s:%d\n", __func__, addr, port); +		return -1; +	} + +	if (setsockopt(sfd, IPPROTO_TCP, TCP_NODELAY, (void*)&yes, sizeof(int)) == -1) { +		perror("Could not set TCP_NODELAY on socket"); +	} + +	if (setsockopt(sfd, SOL_SOCKET, SO_SNDBUF, (void*)&bufsize, sizeof(int)) == -1) { +		perror("Could not set send buffer for socket"); +	} + +	if (setsockopt(sfd, SOL_SOCKET, SO_RCVBUF, (void*)&bufsize, sizeof(int)) == -1) { +		perror("Could not set receive buffer for socket"); +	} + +	return sfd; +} + +LIBIMOBILEDEVICE_GLUE_API int socket_check_fd(int fd, fd_mode fdm, unsigned int timeout) +{ +	fd_set fds; +	int sret; +	int eagain; +	struct timeval to; +	struct timeval *pto; + +	if (fd < 0) { +		if (verbose >= 2) +			fprintf(stderr, "ERROR: invalid fd in check_fd %d\n", fd); +		return -1; +	} + +	FD_ZERO(&fds); +	FD_SET(fd, &fds); + +	sret = -1; + +	do { +		if (timeout > 0) { +			to.tv_sec = (time_t) (timeout / 1000); +			to.tv_usec = (time_t) ((timeout - (to.tv_sec * 1000)) * 1000); +			pto = &to; +		} else { +			pto = NULL; +		} +		eagain = 0; +		switch (fdm) { +		case FDM_READ: +			sret = select(fd + 1, &fds, NULL, NULL, pto); +			break; +		case FDM_WRITE: +			sret = select(fd + 1, NULL, &fds, NULL, pto); +			break; +		case FDM_EXCEPT: +			sret = select(fd + 1, NULL, NULL, &fds, pto); +			break; +		default: +			return -1; +		} + +		if (sret < 0) { +			switch (errno) { +			case EINTR: +				// interrupt signal in select +				if (verbose >= 2) +					fprintf(stderr, "%s: EINTR\n", __func__); +				eagain = 1; +				break; +			case EAGAIN: +				if (verbose >= 2) +					fprintf(stderr, "%s: EAGAIN\n", __func__); +				break; +			default: +				if (verbose >= 2) +					fprintf(stderr, "%s: select failed: %s\n", __func__, +							strerror(errno)); +				return -1; +			} +		} else if (sret == 0) { +			return -ETIMEDOUT; +		} +	} while (eagain); + +	return sret; +} + +LIBIMOBILEDEVICE_GLUE_API int socket_accept(int fd, uint16_t port) +{ +#ifdef WIN32 +	int addr_len; +#else +	socklen_t addr_len; +#endif +	int result; +	struct sockaddr_storage addr; +	addr_len = sizeof(addr); + +	result = accept(fd, (struct sockaddr*)&addr, &addr_len); + +	return result; +} + +LIBIMOBILEDEVICE_GLUE_API int socket_shutdown(int fd, int how) +{ +	return shutdown(fd, how); +} + +LIBIMOBILEDEVICE_GLUE_API int socket_close(int fd) { +#ifdef WIN32 +	return closesocket(fd); +#else +	return close(fd); +#endif +} + +LIBIMOBILEDEVICE_GLUE_API int socket_receive(int fd, void *data, size_t length) +{ +	return socket_receive_timeout(fd, data, length, 0, RECV_TIMEOUT); +} + +LIBIMOBILEDEVICE_GLUE_API int socket_peek(int fd, void *data, size_t length) +{ +	return socket_receive_timeout(fd, data, length, MSG_PEEK, RECV_TIMEOUT); +} + +LIBIMOBILEDEVICE_GLUE_API int socket_receive_timeout(int fd, void *data, size_t length, int flags, unsigned int timeout) +{ +	int res; +	int result; + +	// check if data is available +	res = socket_check_fd(fd, FDM_READ, timeout); +	if (res <= 0) { +		return res; +	} +	// if we get here, there _is_ data available +	result = recv(fd, data, length, flags); +	if (res > 0 && result == 0) { +		// but this is an error condition +		if (verbose >= 3) +			fprintf(stderr, "%s: fd=%d recv returned 0\n", __func__, fd); +		return -ECONNRESET; +	} +	if (result < 0) { +		return -errno; +	} +	return result; +} + +LIBIMOBILEDEVICE_GLUE_API int socket_send(int fd, void *data, size_t length) +{ +	int flags = 0; +	int res = socket_check_fd(fd, FDM_WRITE, SEND_TIMEOUT); +	if (res <= 0) { +		return res; +	} +#ifdef MSG_NOSIGNAL +	flags |= MSG_NOSIGNAL; +#endif +	return send(fd, data, length, flags); +} diff --git a/src/thread.c b/src/thread.c new file mode 100644 index 0000000..dbe93c8 --- /dev/null +++ b/src/thread.c @@ -0,0 +1,203 @@ +/* + * thread.c + * + * Copyright (c) 2012-2019 Nikias Bassen, All Rights Reserved. + * Copyright (c) 2012 Martin Szulecki, All Rights Reserved. + * + * This library is free software; you can redistribute it and/or + * modify it under the terms of the GNU Lesser General Public + * License as published by the Free Software Foundation; either + * version 2.1 of the License, or (at your option) any later version. + * + * This library 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 + * Lesser General Public License for more details. + * + * You should have received a copy of the GNU Lesser General Public + * License along with this library; if not, write to the Free Software + * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA + */ + +#ifdef HAVE_CONFIG_H +#include <config.h> +#endif +#include "common.h" +#include "libimobiledevice-glue/thread.h" + +LIBIMOBILEDEVICE_GLUE_API int thread_new(THREAD_T *thread, thread_func_t thread_func, void* data) +{ +#ifdef WIN32 +	HANDLE th = CreateThread(NULL, 0, (LPTHREAD_START_ROUTINE)thread_func, data, 0, NULL); +	if (th == NULL) { +		return -1; +	} +	*thread = th; +	return 0; +#else +	int res = pthread_create(thread, NULL, thread_func, data); +	return res; +#endif +} + +LIBIMOBILEDEVICE_GLUE_API void thread_detach(THREAD_T thread) +{ +#ifdef WIN32 +	CloseHandle(thread); +#else +	pthread_detach(thread); +#endif +} + +LIBIMOBILEDEVICE_GLUE_API void thread_free(THREAD_T thread) +{ +#ifdef WIN32 +	CloseHandle(thread); +#endif +} + +LIBIMOBILEDEVICE_GLUE_API int thread_join(THREAD_T thread) +{ +	/* wait for thread to complete */ +#ifdef WIN32 +	return (int)WaitForSingleObject(thread, INFINITE); +#else +	return pthread_join(thread, NULL); +#endif +} + +LIBIMOBILEDEVICE_GLUE_API int thread_alive(THREAD_T thread) +{ +	if (!thread) +		return 0; +#ifdef WIN32 +	return WaitForSingleObject(thread, 0) == WAIT_TIMEOUT; +#else +	return pthread_kill(thread, 0) == 0; +#endif +} + +LIBIMOBILEDEVICE_GLUE_API int thread_cancel(THREAD_T thread) +{ +#ifdef WIN32 +	return -1; +#else +#ifdef HAVE_PTHREAD_CANCEL +	return pthread_cancel(thread); +#else +	return -1; +#endif +#endif +} + +LIBIMOBILEDEVICE_GLUE_API void mutex_init(mutex_t* mutex) +{ +#ifdef WIN32 +	InitializeCriticalSection(mutex); +#else +	pthread_mutex_init(mutex, NULL); +#endif +} + +LIBIMOBILEDEVICE_GLUE_API void mutex_destroy(mutex_t* mutex) +{ +#ifdef WIN32 +	DeleteCriticalSection(mutex); +#else +	pthread_mutex_destroy(mutex); +#endif +} + +LIBIMOBILEDEVICE_GLUE_API void mutex_lock(mutex_t* mutex) +{ +#ifdef WIN32 +	EnterCriticalSection(mutex); +#else +	pthread_mutex_lock(mutex); +#endif +} + +LIBIMOBILEDEVICE_GLUE_API void mutex_unlock(mutex_t* mutex) +{ +#ifdef WIN32 +	LeaveCriticalSection(mutex); +#else +	pthread_mutex_unlock(mutex); +#endif +} + +LIBIMOBILEDEVICE_GLUE_API void thread_once(thread_once_t *once_control, void (*init_routine)(void)) +{ +#ifdef WIN32 +	while (InterlockedExchange(&(once_control->lock), 1) != 0) { +		Sleep(1); +	} +	if (!once_control->state) { +		once_control->state = 1; +		init_routine(); +	} +	InterlockedExchange(&(once_control->lock), 0); +#else +	pthread_once(once_control, init_routine); +#endif +} + +void cond_init(cond_t* cond) +{ +#ifdef WIN32 +	cond->sem = CreateSemaphore(NULL, 0, 32767, NULL); +#else +	pthread_cond_init(cond, NULL); +#endif +} + +void cond_destroy(cond_t* cond) +{ +#ifdef WIN32 +	CloseHandle(cond->sem); +#else +	pthread_cond_destroy(cond); +#endif +} + +int cond_signal(cond_t* cond) +{ +#ifdef WIN32 +	int result = 0; +	if (!ReleaseSemaphore(cond->sem, 1, NULL)) { +		result = -1; +	} +	return result; +#else +	return pthread_cond_signal(cond); +#endif +} + +int cond_wait(cond_t* cond, mutex_t* mutex) +{ +#ifdef WIN32 +	mutex_unlock(mutex); +	WaitForSingleObject(cond->sem, INFINITE); +#else +	return pthread_cond_wait(cond, mutex); +#endif +} + +int cond_wait_timeout(cond_t* cond, mutex_t* mutex, unsigned int timeout_ms) +{ +#ifdef WIN32 +	mutex_unlock(mutex); +	WaitForSingleObject(cond->sem, timeout_ms); +#else +	struct timespec ts; +	struct timeval now; +	gettimeofday(&now, NULL); + +	ts.tv_sec = now.tv_sec + timeout_ms / 1000; +	ts.tv_nsec = now.tv_usec * 1000 + 1000 * 1000 * (timeout_ms % 1000); +	ts.tv_sec += ts.tv_nsec / (1000 * 1000 * 1000); +	ts.tv_nsec %= (1000 * 1000 * 1000); + +	return pthread_cond_timedwait(cond, mutex, &ts); +#endif +} diff --git a/src/utils.c b/src/utils.c new file mode 100644 index 0000000..2c70b80 --- /dev/null +++ b/src/utils.c @@ -0,0 +1,551 @@ +/* + * utils.c + * Miscellaneous utilities for string manipulation, + * file I/O and plist helper. + * + * Copyright (c) 2014-2019 Nikias Bassen, All Rights Reserved. + * Copyright (c) 2013-2014 Martin Szulecki, All Rights Reserved. + * Copyright (c) 2013 Federico Mena Quintero + * + * This library is free software; you can redistribute it and/or + * modify it under the terms of the GNU Lesser General Public + * License as published by the Free Software Foundation; either + * version 2.1 of the License, or (at your option) any later version. + * + * This library 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 + * Lesser General Public License for more details. + * + * You should have received a copy of the GNU Lesser General Public + * License along with this library; if not, write to the Free Software + * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA + */ + +#ifdef HAVE_CONFIG_H +#include <config.h> +#endif + +#include <stdarg.h> +#include <stdlib.h> +#include <string.h> +#include <time.h> +#include <sys/time.h> +#include <inttypes.h> +#include <ctype.h> +#include <errno.h> + +#include "common.h" +#include "libimobiledevice-glue/utils.h" + +#ifndef HAVE_STPCPY +/** + * Copy characters from one string into another + * + * @note: The strings should not overlap, as the behavior is undefined. + * + * @s1: The source string. + * @s2: The destination string. + * + * @return a pointer to the terminating `\0' character of @s1, + * or NULL if @s1 or @s2 is NULL. + */ +char *stpcpy(char *s1, const char *s2) +{ +	if (s1 == NULL || s2 == NULL) +		return NULL; + +	strcpy(s1, s2); + +	return s1 + strlen(s2); +} +#endif + +/** + * Concatenate strings into a newly allocated string + * + * @note: Specify NULL for the last string in the varargs list + * + * @str: The first string in the list + * @...: Subsequent strings.  Use NULL for the last item. + * + * @return a newly allocated string, or NULL if @str is NULL.  This will also + * return NULL and set errno to ENOMEM if memory is exhausted. + */ +LIBIMOBILEDEVICE_GLUE_API char *string_concat(const char *str, ...) +{ +	size_t len; +	va_list args; +	char *s; +	char *result; +	char *dest; + +	if (!str) +		return NULL; + +	/* Compute final length */ + +	len = strlen(str) + 1; /* plus 1 for the null terminator */ + +	va_start(args, str); +	s = va_arg(args, char *); +	while (s) { +		len += strlen(s); +		s = va_arg(args, char*); +	} +	va_end(args); + +	/* Concat each string */ + +	result = malloc(len); +	if (!result) +		return NULL; /* errno remains set */ + +	dest = result; + +	dest = stpcpy(dest, str); + +	va_start(args, str); +	s = va_arg(args, char *); +	while (s) { +		dest = stpcpy(dest, s); +		s = va_arg(args, char *); +	} +	va_end(args); + +	return result; +} + +LIBIMOBILEDEVICE_GLUE_API char *string_append(char* str, ...) +{ +	size_t len = 0; +	size_t slen; +	va_list args; +	char *s; +	char *result; +	char *dest; + +	/* Compute final length */ + +	if (str) { +		len = strlen(str); +	} +	slen = len; +	len++; /* plus 1 for the null terminator */ + +	va_start(args, str); +	s = va_arg(args, char *); +	while (s) { +		len += strlen(s); +		s = va_arg(args, char*); +	} +	va_end(args); + +	result = realloc(str, len); +	if (!result) +		return NULL; /* errno remains set */ + +	dest = result + slen; + +	/* Concat additional strings */ + +	va_start(args, str); +	s = va_arg(args, char *); +	while (s) { +		dest = stpcpy(dest, s); +		s = va_arg(args, char *); +	} +	va_end(args); + +	return result; +} + +LIBIMOBILEDEVICE_GLUE_API char *string_build_path(const char *elem, ...) +{ +	if (!elem) +		return NULL; +	va_list args; +	int len = strlen(elem)+1; +	va_start(args, elem); +	char *arg = va_arg(args, char*); +	while (arg) { +		len += strlen(arg)+1; +		arg = va_arg(args, char*); +	} +	va_end(args); + +	char* out = (char*)malloc(len); +	strcpy(out, elem); + +	va_start(args, elem); +	arg = va_arg(args, char*); +	while (arg) { +		strcat(out, "/"); +		strcat(out, arg); +		arg = va_arg(args, char*); +	} +	va_end(args); +	return out; +} + +LIBIMOBILEDEVICE_GLUE_API char *string_format_size(uint64_t size) +{ +	char buf[80]; +	double sz; +	if (size >= 1000000000000LL) { +		sz = ((double)size / 1000000000000.0f); +		sprintf(buf, "%0.1f TB", sz); +	} else if (size >= 1000000000LL) { +		sz = ((double)size / 1000000000.0f); +		sprintf(buf, "%0.1f GB", sz); +	} else if (size >= 1000000LL) { +		sz = ((double)size / 1000000.0f); +		sprintf(buf, "%0.1f MB", sz); +	} else if (size >= 1000LL) { +		sz = ((double)size / 1000.0f); +		sprintf(buf, "%0.1f KB", sz); +	} else { +		sprintf(buf, "%d Bytes", (int)size); +	} +	return strdup(buf); +} + +LIBIMOBILEDEVICE_GLUE_API char *string_toupper(char* str) +{ +	char *res = strdup(str); +	unsigned int i; +	for (i = 0; i < strlen(res); i++) { +		res[i] = toupper(res[i]); +	} +	return res; +} + +static int get_rand(int min, int max) +{ +	int retval = (rand() % (max - min)) + min; +	return retval; +} + +LIBIMOBILEDEVICE_GLUE_API char *generate_uuid() +{ +	const char *chars = "ABCDEF0123456789"; +	int i = 0; +	char *uuid = (char *) malloc(sizeof(char) * 37); + +	srand(time(NULL)); + +	for (i = 0; i < 36; i++) { +		if (i == 8 || i == 13 || i == 18 || i == 23) { +			uuid[i] = '-'; +			continue; +		} else { +			uuid[i] = chars[get_rand(0, 16)]; +		} +	} + +	/* make it a real string */ +	uuid[36] = '\0'; + +	return uuid; +} + +LIBIMOBILEDEVICE_GLUE_API int buffer_read_from_filename(const char *filename, char **buffer, uint64_t *length) +{ +	FILE *f; +	uint64_t size; + +	*length = 0; + +	f = fopen(filename, "rb"); +	if (!f) { +		return 0; +	} + +	fseek(f, 0, SEEK_END); +	size = ftell(f); +	rewind(f); + +	if (size == 0) { +		fclose(f); +		return 0; +	} + +	*buffer = (char*)malloc(sizeof(char)*(size+1)); + +	if (!buffer) { +		return 0; +	} + +	int ret = 1; +	if (fread(*buffer, sizeof(char), size, f) != size) { +		free(*buffer); +		ret = 0; +		errno = EIO; +	} +	fclose(f); + +	*length = size; +	return ret; +} + +LIBIMOBILEDEVICE_GLUE_API int buffer_write_to_filename(const char *filename, const char *buffer, uint64_t length) +{ +	FILE *f; + +	f = fopen(filename, "wb"); +	if (f) { +		size_t written = fwrite(buffer, sizeof(char), length, f); +		fclose(f); + +		if (written == length) { +			return 1; +		} +		else { +			// Not all data could be written. +			errno = EIO; +			return 0; +		} +	} +	else { +		// Failed to open the file, let the caller know. +		return 0; +	} +} + +LIBIMOBILEDEVICE_GLUE_API int plist_read_from_filename(plist_t *plist, const char *filename) +{ +	char *buffer = NULL; +	uint64_t length; + +	if (!filename) +		return 0; + +	if (!buffer_read_from_filename(filename, &buffer, &length)) { +		return 0; +	} + +	if ((length > 8) && (memcmp(buffer, "bplist00", 8) == 0)) { +		plist_from_bin(buffer, length, plist); +	} else { +		plist_from_xml(buffer, length, plist); +	} + +	free(buffer); + +	return 1; +} + +LIBIMOBILEDEVICE_GLUE_API int plist_write_to_filename(plist_t plist, const char *filename, enum plist_format_t format) +{ +	char *buffer = NULL; +	uint32_t length; + +	if (!plist || !filename) +		return 0; + +	if (format == PLIST_FORMAT_XML) +		plist_to_xml(plist, &buffer, &length); +	else if (format == PLIST_FORMAT_BINARY) +		plist_to_bin(plist, &buffer, &length); +	else +		return 0; + +	int res = buffer_write_to_filename(filename, buffer, length); + +	free(buffer); + +	return res; +} + +static const char base64_str[] = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/"; +static const char base64_pad = '='; + +static char *base64encode(const unsigned char *buf, size_t size) +{ +	if (!buf || !(size > 0)) return NULL; +	int outlen = (size / 3) * 4; +	char *outbuf = (char*)malloc(outlen+5); // 4 spare bytes + 1 for '\0' +	size_t n = 0; +	size_t m = 0; +	unsigned char input[3]; +	unsigned int output[4]; +	while (n < size) { +		input[0] = buf[n]; +		input[1] = (n+1 < size) ? buf[n+1] : 0; +		input[2] = (n+2 < size) ? buf[n+2] : 0; +		output[0] = input[0] >> 2; +		output[1] = ((input[0] & 3) << 4) + (input[1] >> 4); +		output[2] = ((input[1] & 15) << 2) + (input[2] >> 6); +		output[3] = input[2] & 63; +		outbuf[m++] = base64_str[(int)output[0]]; +		outbuf[m++] = base64_str[(int)output[1]]; +		outbuf[m++] = (n+1 < size) ? base64_str[(int)output[2]] : base64_pad; +		outbuf[m++] = (n+2 < size) ? base64_str[(int)output[3]] : base64_pad; +		n+=3; +	} +	outbuf[m] = 0; // 0-termination! +	return outbuf; +} + +static void plist_node_print_to_stream(plist_t node, int* indent_level, FILE* stream); + +static void plist_array_print_to_stream(plist_t node, int* indent_level, FILE* stream) +{ +	/* iterate over items */ +	int i, count; +	plist_t subnode = NULL; + +	count = plist_array_get_size(node); + +	for (i = 0; i < count; i++) { +		subnode = plist_array_get_item(node, i); +		fprintf(stream, "%*s", *indent_level, ""); +		fprintf(stream, "%d: ", i); +		plist_node_print_to_stream(subnode, indent_level, stream); +	} +} + +static void plist_dict_print_to_stream(plist_t node, int* indent_level, FILE* stream) +{ +	/* iterate over key/value pairs */ +	plist_dict_iter it = NULL; + +	char* key = NULL; +	plist_t subnode = NULL; +	plist_dict_new_iter(node, &it); +	plist_dict_next_item(node, it, &key, &subnode); +	while (subnode) +	{ +		fprintf(stream, "%*s", *indent_level, ""); +		fprintf(stream, "%s", key); +		if (plist_get_node_type(subnode) == PLIST_ARRAY) +			fprintf(stream, "[%d]: ", plist_array_get_size(subnode)); +		else +			fprintf(stream, ": "); +		free(key); +		key = NULL; +		plist_node_print_to_stream(subnode, indent_level, stream); +		plist_dict_next_item(node, it, &key, &subnode); +	} +	free(it); +} + +static void plist_node_print_to_stream(plist_t node, int* indent_level, FILE* stream) +{ +	char *s = NULL; +	char *data = NULL; +	double d; +	uint8_t b; +	uint64_t u = 0; +	struct timeval tv = { 0, 0 }; + +	plist_type t; + +	if (!node) +		return; + +	t = plist_get_node_type(node); + +	switch (t) { +	case PLIST_BOOLEAN: +		plist_get_bool_val(node, &b); +		fprintf(stream, "%s\n", (b ? "true" : "false")); +		break; + +	case PLIST_UINT: +		plist_get_uint_val(node, &u); +		fprintf(stream, "%"PRIu64"\n", u); +		break; + +	case PLIST_REAL: +		plist_get_real_val(node, &d); +		fprintf(stream, "%f\n", d); +		break; + +	case PLIST_STRING: +		plist_get_string_val(node, &s); +		fprintf(stream, "%s\n", s); +		free(s); +		break; + +	case PLIST_KEY: +		plist_get_key_val(node, &s); +		fprintf(stream, "%s: ", s); +		free(s); +		break; + +	case PLIST_DATA: +		plist_get_data_val(node, &data, &u); +		if (u > 0) { +			s = base64encode((unsigned char*)data, u); +			free(data); +			if (s) { +				fprintf(stream, "%s\n", s); +				free(s); +			} else { +				fprintf(stream, "\n"); +			} +		} else { +			fprintf(stream, "\n"); +		} +		break; + +	case PLIST_DATE: +		plist_get_date_val(node, (int32_t*)&tv.tv_sec, (int32_t*)&tv.tv_usec); +		{ +			time_t ti = (time_t)tv.tv_sec; +			struct tm *btime = localtime(&ti); +			if (btime) { +				s = (char*)malloc(24); + 				memset(s, 0, 24); +				if (strftime(s, 24, "%Y-%m-%dT%H:%M:%SZ", btime) <= 0) { +					free (s); +					s = NULL; +				} +			} +		} +		if (s) { +			fprintf(stream, "%s\n", s); +			free(s); +		} else { +			fprintf(stream, "\n"); +		} +		break; + +	case PLIST_ARRAY: +		fprintf(stream, "\n"); +		(*indent_level)++; +		plist_array_print_to_stream(node, indent_level, stream); +		(*indent_level)--; +		break; + +	case PLIST_DICT: +		fprintf(stream, "\n"); +		(*indent_level)++; +		plist_dict_print_to_stream(node, indent_level, stream); +		(*indent_level)--; +		break; + +	default: +		break; +	} +} + +LIBIMOBILEDEVICE_GLUE_API void plist_print_to_stream(plist_t plist, FILE* stream) +{ +	int indent = 0; + +	if (!plist || !stream) +		return; + +	switch (plist_get_node_type(plist)) { +	case PLIST_DICT: +		plist_dict_print_to_stream(plist, &indent, stream); +		break; +	case PLIST_ARRAY: +		plist_array_print_to_stream(plist, &indent, stream); +		break; +	default: +		plist_node_print_to_stream(plist, &indent, stream); +	} +}  | 
