blob: e81b2906f722ef1267c8b32e7686159a17961b16 [file] [log] [blame]
/*
* iproxy.c -- proxy that enables tcp service access to iOS devices
*
* Copyright (C) 2009-2020 Nikias Bassen <nikias@gmx.li>
* Copyright (C) 2014 Martin Szulecki <m.szulecki@libimobiledevice.org>
* Copyright (C) 2009 Paul Sladen <libiphone@paul.sladen.org>
*
* Based upon iTunnel source code, Copyright (c) 2008 Jing Su.
* http://www.cs.toronto.edu/~jingsu/itunnel/
*
* This program is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation; either version 2 of the License, or
* (at your option) any later version.
*
* This program 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 General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program; 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 <stdlib.h>
#include <string.h>
#include <fcntl.h>
#include <stddef.h>
#include <unistd.h>
#include <errno.h>
#include <getopt.h>
#ifdef WIN32
#include <windows.h>
#include <winsock2.h>
typedef unsigned int socklen_t;
#else
#include <sys/socket.h>
#include <sys/un.h>
#include <arpa/inet.h>
#include <pthread.h>
#include <netinet/in.h>
#include <signal.h>
#endif
#include "socket.h"
#include "usbmuxd.h"
#ifndef ETIMEDOUT
#define ETIMEDOUT 138
#endif
static int debug_level = 0;
struct client_data {
int fd;
int sfd;
char* udid;
enum usbmux_lookup_options lookup_opts;
uint16_t device_port;
};
#define CDATA_FREE(x) if (x) { \
if (x->fd > 0) socket_close(x->fd); \
if (x->sfd > 0) socket_close(x->sfd); \
free(x->udid); \
free(x); \
}
static void *acceptor_thread(void *arg)
{
char buffer[32768];
struct client_data *cdata = (struct client_data*)arg;
usbmuxd_device_info_t *dev_list = NULL;
usbmuxd_device_info_t *dev = NULL;
usbmuxd_device_info_t muxdev;
int count;
if (!cdata) {
fprintf(stderr, "invalid client_data provided!\n");
return NULL;
}
if (cdata->udid) {
if (usbmuxd_get_device(cdata->udid, &muxdev, cdata->lookup_opts) > 0) {
dev = &muxdev;
}
} else {
if ((count = usbmuxd_get_device_list(&dev_list)) < 0) {
printf("Connecting to usbmuxd failed, terminating.\n");
free(dev_list);
CDATA_FREE(cdata);
return NULL;
}
if (dev_list == NULL || dev_list[0].handle == 0) {
printf("No connected device found, terminating.\n");
free(dev_list);
CDATA_FREE(cdata);
return NULL;
}
int i;
for (i = 0; i < count; i++) {
if (dev_list[i].conn_type == CONNECTION_TYPE_USB && (cdata->lookup_opts & DEVICE_LOOKUP_USBMUX)) {
dev = &(dev_list[i]);
break;
} else if (dev_list[i].conn_type == CONNECTION_TYPE_NETWORK && (cdata->lookup_opts & DEVICE_LOOKUP_NETWORK)) {
dev = &(dev_list[i]);
break;
}
}
}
if (dev == NULL || dev->handle == 0) {
printf("No connected/matching device found, disconnecting client.\n");
free(dev_list);
CDATA_FREE(cdata);
return NULL;
}
cdata->sfd = -1;
if (dev->conn_type == CONNECTION_TYPE_NETWORK) {
unsigned char saddr_[32];
memset(saddr_, '\0', sizeof(saddr_));
struct sockaddr* saddr = (struct sockaddr*)&saddr_[0];
if (((char*)dev->conn_data)[1] == 0x02) { // AF_INET
saddr->sa_family = AF_INET;
memcpy(&saddr->sa_data[0], (char*)dev->conn_data+2, 14);
}
else if (((char*)dev->conn_data)[1] == 0x1E) { //AF_INET6 (bsd)
#ifdef AF_INET6
saddr->sa_family = AF_INET6;
memcpy(&saddr->sa_data[0], (char*)dev->conn_data+2, 26);
#else
fprintf(stderr, "ERROR: Got an IPv6 address but this system doesn't support IPv6\n");
CDATA_FREE(cdata);
return NULL;
#endif
}
else {
fprintf(stderr, "Unsupported address family 0x%02x\n", ((char*)dev->conn_data)[1]);
CDATA_FREE(cdata);
return NULL;
}
char addrtxt[48];
addrtxt[0] = '\0';
if (!socket_addr_to_string(saddr, addrtxt, sizeof(addrtxt))) {
fprintf(stderr, "Failed to convert network address: %d (%s)\n", errno, strerror(errno));
}
fprintf(stdout, "Requesting connecion to NETWORK device %s (serial: %s), port %d\n", addrtxt, dev->udid, cdata->device_port);
cdata->sfd = socket_connect_addr(saddr, cdata->device_port);
} else if (dev->conn_type == CONNECTION_TYPE_USB) {
fprintf(stdout, "Requesting connecion to USB device handle %d (serial: %s), port %d\n", dev->handle, dev->udid, cdata->device_port);
cdata->sfd = usbmuxd_connect(dev->handle, cdata->device_port);
}
free(dev_list);
if (cdata->sfd < 0) {
fprintf(stderr, "Error connecting to device: %s\n", strerror(errno));
} else {
fd_set fds;
FD_ZERO(&fds);
FD_SET(cdata->fd, &fds);
FD_SET(cdata->sfd, &fds);
while (1) {
fd_set read_fds = fds;
int ret_sel = select(cdata->sfd+1, &read_fds, NULL, NULL, NULL);
if (ret_sel < 0) {
perror("select");
break;
}
if (FD_ISSET(cdata->fd, &read_fds)) {
int r = socket_receive_timeout(cdata->fd, buffer, sizeof(buffer), 0, 100);
if (r <= 0) {
break;
}
int sent = 0;
while (sent < r) {
int s = socket_send(cdata->sfd, buffer+sent, r-sent);
if (s <= 0) {
break;
}
sent += s;
}
}
if (FD_ISSET(cdata->sfd, &read_fds)) {
int r = socket_receive_timeout(cdata->sfd, buffer, sizeof(buffer), 0, 100);
if (r <= 0) {
break;
}
int sent = 0;
while (sent < r) {
int s = socket_send(cdata->fd, buffer+sent, r-sent);
if (s <= 0) {
break;
}
sent += s;
}
}
}
}
CDATA_FREE(cdata);
return NULL;
}
static void print_usage(int argc, char **argv, int is_error)
{
char *name = NULL;
name = strrchr(argv[0], '/');
fprintf(is_error ? stderr : stdout, "Usage: %s [OPTIONS] LOCAL_PORT:DEVICE_PORT [LOCAL_PORT2:DEVICE_PORT2 ...]\n\n", (name ? name+1 : argv[0]));
fprintf(is_error ? stderr : stdout,
"Proxy that binds local TCP ports to be forwarded to the specified ports on a usbmux device.\n\n" \
"OPTIONS:\n" \
" -u, --udid UDID target specific device by UDID\n" \
" -n, --network connect to network device\n" \
" -l, --local connect to USB device (default)\n" \
" -s, --source ADDR source address for listening socket (default 127.0.0.1)\n" \
" -h, --help prints usage information\n" \
" -d, --debug increase debug level\n" \
"\n" \
"Homepage: <" PACKAGE_URL ">\n"
"Bug reports: <" PACKAGE_BUGREPORT ">\n"
"\n"
);
}
int main(int argc, char **argv)
{
char* device_udid = NULL;
char* source_addr = NULL;
uint16_t listen_port[16];
uint16_t device_port[16];
int listen_sock[16] = { -1, };
int num_pairs = 0;
int i = 0;
enum usbmux_lookup_options lookup_opts = 0;
const struct option longopts[] = {
{ "debug", no_argument, NULL, 'd' },
{ "help", no_argument, NULL, 'h' },
{ "udid", required_argument, NULL, 'u' },
{ "local", no_argument, NULL, 'l' },
{ "network", no_argument, NULL, 'n' },
{ "source", required_argument, NULL, 's' },
{ NULL, 0, NULL, 0}
};
int c = 0;
while ((c = getopt_long(argc, argv, "dhu:lns:", longopts, NULL)) != -1) {
switch (c) {
case 'd':
libusbmuxd_set_debug_level(++debug_level);
break;
case 'u':
if (!*optarg) {
fprintf(stderr, "ERROR: UDID must not be empty!\n");
print_usage(argc, argv, 1);
return 2;
}
free(device_udid);
device_udid = strdup(optarg);
break;
case 'l':
lookup_opts |= DEVICE_LOOKUP_USBMUX;
break;
case 'n':
lookup_opts |= DEVICE_LOOKUP_NETWORK;
break;
case 's':
if (!*optarg) {
fprintf(stderr, "ERROR: source address must not be empty!\n");
print_usage(argc, argv, 1);
return 2;
}
free(source_addr);
source_addr = strdup(optarg);
break;
case 'h':
print_usage(argc, argv, 0);
return 0;
default:
print_usage(argc, argv, 1);
return 2;
}
}
if (lookup_opts == 0) {
lookup_opts = DEVICE_LOOKUP_USBMUX;
}
argc -= optind;
argv += optind;
if (argc == 0) {
fprintf(stderr, "ERROR: Not enough parameters. Need at least one pair of ports.\n");
print_usage(argc + optind, argv - optind, 1);
free(device_udid);
free(source_addr);
return 2;
}
if (argc == 2 && (strchr(argv[0], ':') == NULL) && (strchr(argv[1], ':') == NULL)) {
/* support old-style port pair specification */
char* endp = NULL;
listen_port[0] = (uint16_t)strtol(argv[0], &endp, 10);
if (!listen_port[0] || *endp != '\0') {
fprintf(stderr, "Invalid listen port specified in argument '%s'!\n", argv[0]);
free(device_udid);
free(source_addr);
return -EINVAL;
}
endp = NULL;
device_port[0] = (uint16_t)strtol(argv[1], &endp, 10);
if (!device_port[0] || *endp != '\0') {
fprintf(stderr, "Invalid device port specified in argument '%s'!\n", argv[1]);
free(device_udid);
free(source_addr);
return -EINVAL;
}
num_pairs = 1;
} else {
/* new style, colon-separated local:device port pairs */
for (i = 0; i < argc; i++) {
char* endp = NULL;
listen_port[i] = (uint16_t)strtol(argv[i], &endp, 10);
if (!listen_port[i] || (*endp != ':')) {
fprintf(stderr, "Invalid listen port specified in argument '%s'!\n", argv[i]);
free(device_udid);
free(source_addr);
return -EINVAL;
}
device_port[i] = (uint16_t)strtol(endp+1, &endp, 10);
if (!device_port[i] || (*endp != '\0')) {
fprintf(stderr, "Invalid device port specified in argument '%s'!\n", argv[i+1]);
free(device_udid);
free(source_addr);
return -EINVAL;
}
}
num_pairs = argc;
}
#ifndef WIN32
signal(SIGPIPE, SIG_IGN);
#endif
// first create the listening sockets
for (i = 0; i < num_pairs; i++) {
printf("Creating listening port %d for device port %d\n", listen_port[i], device_port[i]);
listen_sock[i] = socket_create(source_addr, listen_port[i]);
if (listen_sock[i] < 0) {
int j;
fprintf(stderr, "Error creating socket for listen port %u: %s\n", listen_port[i], strerror(errno));
free(source_addr);
free(device_udid);
for (j = i; j >= 0; j--) {
socket_close(listen_sock[j]);
}
return -errno;
}
}
// make them non-blocking and add them to fd set
fd_set fds;
FD_ZERO(&fds);
for (i = 0; i < num_pairs; i++) {
#ifdef WIN32
u_long l_yes = 1;
ioctlsocket(listen_sock[i], FIONBIO, &l_yes);
#else
int flags = fcntl(listen_sock[i], F_GETFL, 0);
fcntl(listen_sock[i], F_SETFL, flags | O_NONBLOCK);
#endif
FD_SET(listen_sock[i], &fds);
}
// main loop
while (1) {
printf("waiting for connection\n");
fd_set read_fds = fds;
int ret_sel = select(listen_sock[num_pairs-1]+1, &read_fds, NULL, NULL, NULL);
if (ret_sel < 0) {
perror("select");
break;
}
for (i = 0; i < num_pairs; i++) {
if (FD_ISSET(listen_sock[i], &read_fds)) {
#ifdef WIN32
HANDLE acceptor = NULL;
#else
pthread_t acceptor;
#endif
struct client_data *cdata;
int c_sock = socket_accept(listen_sock[i], listen_port[i]);
if (c_sock < 0) {
fprintf(stderr, "accept: %s\n", strerror(errno));
break;
} else {
printf("New connection for %d->%d, fd = %d\n", listen_port[i], device_port[i], c_sock);
cdata = (struct client_data*)malloc(sizeof(struct client_data));
if (!cdata) {
socket_close(c_sock);
fprintf(stderr, "ERROR: Out of memory\n");
free(device_udid);
return -1;
}
cdata->fd = c_sock;
cdata->sfd = -1;
cdata->udid = (device_udid) ? strdup(device_udid) : NULL;
cdata->lookup_opts = lookup_opts;
cdata->device_port = device_port[i];
#ifdef WIN32
acceptor = CreateThread(NULL, 0, (LPTHREAD_START_ROUTINE)acceptor_thread, cdata, 0, NULL);
CloseHandle(acceptor);
#else
pthread_create(&acceptor, NULL, acceptor_thread, cdata);
pthread_detach(acceptor);
#endif
}
}
}
}
for (i = 0; i < num_pairs; i++) {
socket_close(listen_sock[i]);
}
free(device_udid);
free(source_addr);
return 0;
}