diff options
Diffstat (limited to 'src/fl2k_tcp.c')
| -rw-r--r-- | src/fl2k_tcp.c | 14 |
1 files changed, 13 insertions, 1 deletions
diff --git a/src/fl2k_tcp.c b/src/fl2k_tcp.c index bd01758..871ac26 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; } } @@ -233,8 +245,8 @@ int main(int argc, char **argv) sleep_ms(500); out: - free(txbuf); fl2k_close(dev); + free(txbuf); closesocket(s); #ifdef _WIN32 WSACleanup(); |