From 9b056f5091ad0bc35cd0188cfcb887eb30269fef Mon Sep 17 00:00:00 2001 From: Romain Vimont Date: Thu, 15 Feb 2018 22:59:21 +0100 Subject: [PATCH] Replace SDL_net by custom implementation SDL_net is not very suitable for scrcpy. For example, SDLNet_TCP_Accept() is non-blocking, so we have to wrap it by calling many SDL_Net-specific functions to make it blocking. But above all, SDLNet_TCP_Open() is a server socket only when no IP is provided; otherwise, it's a client socket. Therefore, it is not possible to create a server socket bound to localhost, so it accepts connections from anywhere. This is a problem for scrcpy, because on start, the application listens for nearly 1 second until it accepts the first connection, supposedly from the device. If someone on the local network manages to connect to the server socket first, then they can stream arbitrary H.264 video. This may be troublesome, for example during a public presentation ;-) Provide our own simplified API (net.h) instead, implemented for the different platforms. --- app/meson.build | 20 +++++++++------ app/src/command.h | 1 + app/src/controller.c | 4 +-- app/src/controller.h | 8 +++--- app/src/decoder.c | 7 +++--- app/src/decoder.h | 8 +++--- app/src/device.c | 4 +-- app/src/device.h | 4 +-- app/src/main.c | 2 -- app/src/net.c | 56 ++++++++++++++++++++++++++++++++++++++++++ app/src/net.h | 26 ++++++++++++++++++++ app/src/netutil.c | 31 ----------------------- app/src/netutil.h | 9 ------- app/src/scrcpy.c | 7 +++--- app/src/server.c | 37 ++++++++++++++-------------- app/src/server.h | 12 ++++----- app/src/sys/unix/net.c | 16 ++++++++++++ app/src/sys/win/net.c | 21 ++++++++++++++++ 18 files changed, 178 insertions(+), 95 deletions(-) create mode 100644 app/src/net.c create mode 100644 app/src/net.h delete mode 100644 app/src/netutil.c delete mode 100644 app/src/netutil.h create mode 100644 app/src/sys/unix/net.c create mode 100644 app/src/sys/win/net.c diff --git a/app/meson.build b/app/meson.build index 83375ee2..e574a835 100644 --- a/app/meson.build +++ b/app/meson.build @@ -10,7 +10,7 @@ src = [ 'src/frames.c', 'src/inputmanager.c', 'src/lockutil.c', - 'src/netutil.c', + 'src/net.c', 'src/scrcpy.c', 'src/screen.c', 'src/server.c', @@ -18,20 +18,24 @@ src = [ 'src/tinyxpm.c', ] -if host_machine.system() == 'windows' - src += [ 'src/sys/win/command.c' ] -else - src += [ 'src/sys/unix/command.c' ] -endif - dependencies = [ dependency('libavformat'), dependency('libavcodec'), dependency('libavutil'), dependency('sdl2'), - dependency('SDL2_net'), ] +cc = meson.get_compiler('c') + +if host_machine.system() == 'windows' + src += [ 'src/sys/win/command.c' ] + src += [ 'src/sys/win/net.c' ] + dependencies += cc.find_library('ws2_32') +else + src += [ 'src/sys/unix/command.c' ] + src += [ 'src/sys/unix/net.c' ] +endif + conf = configuration_data() # expose the build type diff --git a/app/src/command.h b/app/src/command.h index adaad836..8232f46c 100644 --- a/app/src/command.h +++ b/app/src/command.h @@ -20,6 +20,7 @@ #endif #ifdef __WINDOWS__ +# include // not needed here, but must never be included AFTER windows.h # include # define PROCESS_NONE NULL typedef HANDLE process_t; diff --git a/app/src/controller.c b/app/src/controller.c index 8f5bddf8..24485bdd 100644 --- a/app/src/controller.c +++ b/app/src/controller.c @@ -3,7 +3,7 @@ #include "lockutil.h" #include "log.h" -SDL_bool controller_init(struct controller *controller, TCPsocket video_socket) { +SDL_bool controller_init(struct controller *controller, socket_t video_socket) { if (!control_event_queue_init(&controller->queue)) { return SDL_FALSE; } @@ -47,7 +47,7 @@ static SDL_bool process_event(struct controller *controller, const struct contro if (!length) { return SDL_FALSE; } - int w = SDLNet_TCP_Send(controller->video_socket, serialized_event, length); + int w = net_send(controller->video_socket, serialized_event, length); return w == length; } diff --git a/app/src/controller.h b/app/src/controller.h index 14b93a2a..7c22fe07 100644 --- a/app/src/controller.h +++ b/app/src/controller.h @@ -4,11 +4,13 @@ #include "controlevent.h" #include -#include #include +#include + +#include "net.h" struct controller { - TCPsocket video_socket; + socket_t video_socket; SDL_Thread *thread; SDL_mutex *mutex; SDL_cond *event_cond; @@ -16,7 +18,7 @@ struct controller { struct control_event_queue queue; }; -SDL_bool controller_init(struct controller *controller, TCPsocket video_socket); +SDL_bool controller_init(struct controller *controller, socket_t video_socket); void controller_destroy(struct controller *controller); SDL_bool controller_start(struct controller *controller); diff --git a/app/src/decoder.c b/app/src/decoder.c index c3700006..8f93b9c9 100644 --- a/app/src/decoder.c +++ b/app/src/decoder.c @@ -1,8 +1,8 @@ #include "decoder.h" #include +#include #include -#include #include #include @@ -11,13 +11,12 @@ #include "frames.h" #include "lockutil.h" #include "log.h" -#include "netutil.h" #define BUFSIZE 0x10000 static int read_packet(void *opaque, uint8_t *buf, int buf_size) { struct decoder *decoder = opaque; - return SDLNet_TCP_Recv(decoder->video_socket, buf, buf_size); + return net_recv(decoder->video_socket, buf, buf_size); } // set the decoded frame as ready for rendering, and notify @@ -147,7 +146,7 @@ run_finally_free_codec_ctx: return ret; } -void decoder_init(struct decoder *decoder, struct frames *frames, TCPsocket video_socket) { +void decoder_init(struct decoder *decoder, struct frames *frames, socket_t video_socket) { decoder->frames = frames; decoder->video_socket = video_socket; } diff --git a/app/src/decoder.h b/app/src/decoder.h index 04ff14a6..87346114 100644 --- a/app/src/decoder.h +++ b/app/src/decoder.h @@ -2,18 +2,20 @@ #define DECODER_H #include -#include +#include + +#include "net.h" struct frames; struct decoder { struct frames *frames; - TCPsocket video_socket; + socket_t video_socket; SDL_Thread *thread; SDL_mutex *mutex; }; -void decoder_init(struct decoder *decoder, struct frames *frames, TCPsocket video_socket); +void decoder_init(struct decoder *decoder, struct frames *frames, socket_t video_socket); SDL_bool decoder_start(struct decoder *decoder); void decoder_stop(struct decoder *decoder); void decoder_join(struct decoder *decoder); diff --git a/app/src/device.c b/app/src/device.c index c1dd7a5d..761e2446 100644 --- a/app/src/device.c +++ b/app/src/device.c @@ -1,9 +1,9 @@ #include "device.h" #include "log.h" -SDL_bool device_read_info(TCPsocket device_socket, char *device_name, struct size *size) { +SDL_bool device_read_info(socket_t device_socket, char *device_name, struct size *size) { unsigned char buf[DEVICE_NAME_FIELD_LENGTH + 4]; - if (SDLNet_TCP_Recv(device_socket, buf, sizeof(buf)) <= 0) { + if (net_recv(device_socket, buf, sizeof(buf)) <= 0) { LOGE("Could not retrieve device information"); return SDL_FALSE; } diff --git a/app/src/device.h b/app/src/device.h index ffaeaa28..d01d6ed2 100644 --- a/app/src/device.h +++ b/app/src/device.h @@ -1,14 +1,14 @@ #ifndef DEVICE_H #define DEVICE_H -#include #include #include "common.h" +#include "net.h" #define DEVICE_NAME_FIELD_LENGTH 64 // name must be at least DEVICE_NAME_FIELD_LENGTH bytes -SDL_bool device_read_info(TCPsocket device_socket, char *name, struct size *frame_size); +SDL_bool device_read_info(socket_t device_socket, char *name, struct size *frame_size); #endif diff --git a/app/src/main.c b/app/src/main.c index a0e36b74..46abd845 100644 --- a/app/src/main.c +++ b/app/src/main.c @@ -4,7 +4,6 @@ #include #include #include -#include #include "config.h" #include "log.h" @@ -97,7 +96,6 @@ static void print_version(void) { fprintf(stderr, "dependencies:\n"); fprintf(stderr, " - SDL %d.%d.%d\n", SDL_MAJOR_VERSION, SDL_MINOR_VERSION, SDL_PATCHLEVEL); - fprintf(stderr, " - SDL_net %d.%d.%d\n", SDL_NET_MAJOR_VERSION, SDL_NET_MINOR_VERSION, SDL_NET_PATCHLEVEL); fprintf(stderr, " - libavcodec %d.%d.%d\n", LIBAVCODEC_VERSION_MAJOR, LIBAVCODEC_VERSION_MINOR, LIBAVCODEC_VERSION_MICRO); fprintf(stderr, " - libavformat %d.%d.%d\n", LIBAVFORMAT_VERSION_MAJOR, LIBAVFORMAT_VERSION_MINOR, LIBAVFORMAT_VERSION_MICRO); fprintf(stderr, " - libavutil %d.%d.%d\n", LIBAVUTIL_VERSION_MAJOR, LIBAVUTIL_VERSION_MINOR, LIBAVUTIL_VERSION_MICRO); diff --git a/app/src/net.c b/app/src/net.c new file mode 100644 index 00000000..11a66e4c --- /dev/null +++ b/app/src/net.c @@ -0,0 +1,56 @@ +#include "net.h" + +#include "log.h" + +#ifdef __WINDOWS__ + typedef int socklen_t; +#else +# include +# include +# include +# include +# include +# define SOCKET_ERROR -1 + typedef struct sockaddr_in SOCKADDR_IN; + typedef struct sockaddr SOCKADDR; + typedef struct in_addr IN_ADDR; +#endif + +socket_t net_listen(Uint32 addr, Uint16 port, int backlog) { + socket_t sock = socket(AF_INET, SOCK_STREAM, 0); + if (sock == INVALID_SOCKET) { + LOGE("Cannot create socket"); + return INVALID_SOCKET; + } + + SOCKADDR_IN sin; + sin.sin_family = AF_INET; + sin.sin_addr.s_addr = htonl(addr); // htonl() harmless on INADDR_ANY + sin.sin_port = htons(port); + + if (bind(sock, (SOCKADDR *) &sin, sizeof(sin)) == SOCKET_ERROR) { + LOGE("Cannot bind"); + return INVALID_SOCKET; + } + + if (listen(sock, backlog) == SOCKET_ERROR) { + LOGE("Cannot listen on port %" PRIu16, port); + return INVALID_SOCKET; + } + + return sock; +} + +socket_t net_accept(socket_t server_socket) { + SOCKADDR_IN csin; + socklen_t sinsize = sizeof(csin); + return accept(server_socket, (SOCKADDR *) &csin, &sinsize); +} + +ssize_t net_recv(socket_t socket, void *buf, size_t len) { + return recv(socket, buf, len, 0); +} + +ssize_t net_send(socket_t socket, void *buf, size_t len) { + return send(socket, buf, len, 0); +} diff --git a/app/src/net.h b/app/src/net.h new file mode 100644 index 00000000..bf22605c --- /dev/null +++ b/app/src/net.h @@ -0,0 +1,26 @@ +#ifndef NET_H +#define NET_H + +#include +#include + +#ifdef __WINDOWS__ +# include + typedef SIZE_T size_t; + typedef SSIZE_T ssize_t; + typedef SOCKET socket_t; +#else +# define INVALID_SOCKET -1 + typedef int socket_t; +#endif + +SDL_bool net_init(void); +void net_cleanup(void); + +socket_t net_listen(Uint32 addr, Uint16 port, int backlog); +socket_t net_accept(socket_t server_socket); +ssize_t net_recv(socket_t socket, void *buf, size_t len); +ssize_t net_send(socket_t socket, void *buf, size_t len); +void net_close(socket_t socket); + +#endif diff --git a/app/src/netutil.c b/app/src/netutil.c deleted file mode 100644 index b3054587..00000000 --- a/app/src/netutil.c +++ /dev/null @@ -1,31 +0,0 @@ -#include "netutil.h" - -#include - -#include "log.h" - -// contrary to SDLNet_TCP_Send and SDLNet_TCP_Recv, SDLNet_TCP_Accept is non-blocking -// so we need to block before calling it -TCPsocket server_socket_accept(TCPsocket server_socket, Uint32 timeout_ms) { - SDLNet_SocketSet set = SDLNet_AllocSocketSet(1); - if (!set) { - LOGC("Could not allocate socket set"); - return NULL; - } - - if (SDLNet_TCP_AddSocket(set, server_socket) == -1) { - LOGC("Could not add socket to set"); - SDLNet_FreeSocketSet(set); - return NULL; - } - - if (SDLNet_CheckSockets(set, timeout_ms) != 1) { - LOGE("No connection to accept"); - SDLNet_FreeSocketSet(set); - return NULL; - } - - SDLNet_FreeSocketSet(set); - - return SDLNet_TCP_Accept(server_socket); -} diff --git a/app/src/netutil.h b/app/src/netutil.h deleted file mode 100644 index d0a3d2a8..00000000 --- a/app/src/netutil.h +++ /dev/null @@ -1,9 +0,0 @@ -#ifndef NETUTIL_H -#define NETUTIL_H - -#include - -// blocking accept on the server socket -TCPsocket server_socket_accept(TCPsocket server_socket, Uint32 timeout_ms); - -#endif diff --git a/app/src/scrcpy.c b/app/src/scrcpy.c index 90eb1e52..5891658d 100644 --- a/app/src/scrcpy.c +++ b/app/src/scrcpy.c @@ -6,7 +6,6 @@ #include #include #include -#include #include "command.h" #include "common.h" @@ -19,7 +18,7 @@ #include "inputmanager.h" #include "log.h" #include "lockutil.h" -#include "netutil.h" +#include "net.h" #include "screen.h" #include "server.h" #include "tinyxpm.h" @@ -104,8 +103,8 @@ SDL_bool scrcpy(const char *serial, Uint16 local_port, Uint16 max_size, Uint32 b // managed by the event loop. This blocking call blocks the event loop, so // timeout the connection not to block indefinitely in case of SIGTERM. #define SERVER_CONNECT_TIMEOUT_MS 2000 - TCPsocket device_socket = server_connect_to(&server, serial, SERVER_CONNECT_TIMEOUT_MS); - if (!device_socket) { + socket_t device_socket = server_connect_to(&server, serial, SERVER_CONNECT_TIMEOUT_MS); + if (device_socket == INVALID_SOCKET) { server_stop(&server, serial); ret = SDL_FALSE; goto finally_destroy_server; diff --git a/app/src/server.c b/app/src/server.c index 614f78d7..0cb4f0e6 100644 --- a/app/src/server.c +++ b/app/src/server.c @@ -1,12 +1,12 @@ #include "server.h" -#include #include +#include #include +#include #include "config.h" #include "log.h" -#include "netutil.h" #define SOCKET_NAME "scrcpy" @@ -62,12 +62,9 @@ static void terminate_server(process_t server) { } } -static TCPsocket listen_on_port(Uint16 port) { - IPaddress addr = { - .host = INADDR_ANY, - .port = SDL_SwapBE16(port), - }; - return SDLNet_TCP_Open(&addr); +static socket_t listen_on_port(Uint16 port) { +#define IPV4_LOCALHOST 0x7F000001 + return net_listen(IPV4_LOCALHOST, port, 1); } void server_init(struct server *server) { @@ -91,7 +88,7 @@ SDL_bool server_start(struct server *server, const char *serial, Uint16 local_po // connect until the server socket is listening on the device. server->server_socket = listen_on_port(local_port); - if (!server->server_socket) { + if (server->server_socket == INVALID_SOCKET) { LOGE("Could not listen on port %" PRIu16, local_port); disable_tunnel(serial); return SDL_FALSE; @@ -100,7 +97,7 @@ SDL_bool server_start(struct server *server, const char *serial, Uint16 local_po // server will connect to our server socket server->process = execute_server(serial, max_size, bit_rate); if (server->process == PROCESS_NONE) { - SDLNet_TCP_Close(server->server_socket); + net_close(server->server_socket); disable_tunnel(serial); return SDL_FALSE; } @@ -110,13 +107,15 @@ SDL_bool server_start(struct server *server, const char *serial, Uint16 local_po return SDL_TRUE; } -TCPsocket server_connect_to(struct server *server, const char *serial, Uint32 timeout_ms) { - SDL_assert(server->server_socket); - server->device_socket = server_socket_accept(server->server_socket, timeout_ms); +socket_t server_connect_to(struct server *server, const char *serial, Uint32 timeout_ms) { + server->device_socket = net_accept(server->server_socket); + if (server->device_socket == INVALID_SOCKET) { + return INVALID_SOCKET; + } // we don't need the server socket anymore - SDLNet_TCP_Close(server->server_socket); - server->server_socket = NULL; + net_close(server->server_socket); + server->server_socket = INVALID_SOCKET; // we don't need the adb tunnel anymore disable_tunnel(serial); // ignore failure @@ -136,10 +135,10 @@ void server_stop(struct server *server, const char *serial) { } void server_destroy(struct server *server) { - if (server->server_socket) { - SDLNet_TCP_Close(server->server_socket); + if (server->server_socket != INVALID_SOCKET) { + net_close(server->server_socket); } - if (server->device_socket) { - SDLNet_TCP_Close(server->device_socket); + if (server->device_socket != INVALID_SOCKET) { + net_close(server->device_socket); } } diff --git a/app/src/server.h b/app/src/server.h index 3d877f0c..95eb7d16 100644 --- a/app/src/server.h +++ b/app/src/server.h @@ -1,20 +1,20 @@ #ifndef SERVER_H #define SERVER_H -#include #include "command.h" +#include "net.h" struct server { process_t process; - TCPsocket server_socket; - TCPsocket device_socket; + socket_t server_socket; + socket_t device_socket; SDL_bool adb_reverse_enabled; }; #define SERVER_INITIALIZER { \ .process = PROCESS_NONE, \ - .server_socket = NULL, \ - .device_socket = NULL, \ + .server_socket = INVALID_SOCKET, \ + .device_socket = INVALID_SOCKET, \ .adb_reverse_enabled = SDL_FALSE, \ } @@ -26,7 +26,7 @@ SDL_bool server_start(struct server *server, const char *serial, Uint16 local_po Uint16 max_size, Uint32 bit_rate); // block until the communication with the server is established -TCPsocket server_connect_to(struct server *server, const char *serial, Uint32 timeout_ms); +socket_t server_connect_to(struct server *server, const char *serial, Uint32 timeout_ms); // disconnect and kill the server process void server_stop(struct server *server, const char *serial); diff --git a/app/src/sys/unix/net.c b/app/src/sys/unix/net.c new file mode 100644 index 00000000..4219ddb8 --- /dev/null +++ b/app/src/sys/unix/net.c @@ -0,0 +1,16 @@ +#include "../../net.h" + +# include + +SDL_bool net_init(void) { + // do nothing + return SDL_TRUE; +} + +void net_cleanup(void) { + // do nothing +} + +void net_close(socket_t socket) { + close(socket); +} diff --git a/app/src/sys/win/net.c b/app/src/sys/win/net.c new file mode 100644 index 00000000..b328361e --- /dev/null +++ b/app/src/sys/win/net.c @@ -0,0 +1,21 @@ +#include "../../net.h" + +#include "../../log.h" + +SDL_bool net_init(void) { + WSADATA wsa; + int res = WSAStartup(MAKEWORD(2, 2), &wsa) < 0; + if (res < 0) { + LOGC("WSAStartup failed with error %d", res); + return SDL_FALSE; + } + return SDL_TRUE; +} + +void net_cleanup(void) { + WSACleanup(); +} + +void net_close(socket_t socket) { + closesocket(socket); +} -- GitLab