diff options
author | Nikias Bassen | 2021-06-07 02:01:35 +0200 |
---|---|---|
committer | Nikias Bassen | 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); + } +} |