summaryrefslogtreecommitdiffstats
path: root/src
diff options
context:
space:
mode:
authorGravatar Nikias Bassen2021-06-07 02:01:35 +0200
committerGravatar Nikias Bassen2021-06-07 03:41:07 +0200
commit046b26150e004a8ac740e699c6c3e11be29e8f11 (patch)
tree8f59597463dccd81ca180a87668e53864c4b09ae /src
downloadlibimobiledevice-glue-046b26150e004a8ac740e699c6c3e11be29e8f11.tar.gz
libimobiledevice-glue-046b26150e004a8ac740e699c6c3e11be29e8f11.tar.bz2
Initial check-in of sources
Diffstat (limited to 'src')
-rw-r--r--src/Makefile.am23
-rw-r--r--src/collection.c101
-rw-r--r--src/common.h38
-rw-r--r--src/libimobiledevice-glue-1.0.pc.in11
-rw-r--r--src/socket.c1169
-rw-r--r--src/thread.c203
-rw-r--r--src/utils.c551
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);
+ }
+}