From 077613efc501195f4ad0018e9d94f848654f6c3e Mon Sep 17 00:00:00 2001 From: Steve Markgraf Date: Mon, 26 Aug 2019 19:26:30 +0200 Subject: improve exit handling on device removal --- src/fl2k_tcp.c | 12 ++++++++++++ 1 file changed, 12 insertions(+) (limited to 'src/fl2k_tcp.c') diff --git a/src/fl2k_tcp.c b/src/fl2k_tcp.c index bd01758..1fa9e5d 100644 --- a/src/fl2k_tcp.c +++ b/src/fl2k_tcp.c @@ -107,6 +107,12 @@ void fl2k_callback(fl2k_data_info_t *data_info) int r; struct timeval tv = { 1, 0 }; + if (data_info->device_error) { + fprintf(stderr, "Device error, exiting.\n"); + do_exit = 1; + return; + } + if (!connected) return; @@ -122,6 +128,12 @@ void fl2k_callback(fl2k_data_info_t *data_info) if (r) { received = recv(sock, txbuf + (FL2K_BUF_LEN - left), left, 0); + if (!received) { + fprintf(stderr, "Connection was closed!\n"); + fl2k_stop_tx(dev); + do_exit = 1; + } + left -= received; } } -- cgit v1.2.3