X-Git-Url: https://sigrok.org/gitweb/?a=blobdiff_plain;f=src%2Fbt%2Fbt_bluez.c;h=52567f9e8b7a04d4cd1d3ca68c55960ed4b8684c;hb=2d8a50897614e097cffa36dd7a3acd7d9032a9a1;hp=167216087133ba60c8aa6371444183aa30ee220b;hpb=0f92d5db036e266a04edfdfe5b7c8cefd7ff2193;p=libsigrok.git

diff --git a/src/bt/bt_bluez.c b/src/bt/bt_bluez.c
index 16721608..52567f9e 100644
--- a/src/bt/bt_bluez.c
+++ b/src/bt/bt_bluez.c
@@ -87,13 +87,15 @@
 #include <libsigrok/libsigrok.h>
 #include "libsigrok-internal.h"
 
-/** @cond PRIVATE */
 #define LOG_PREFIX "bt-bluez"
-/** @endcond */
 
+#define CONNECT_BLE_TIMEOUT	20	/* Connect timeout in seconds. */
 #define STORE_MAC_REVERSE	1
 #define ACCEPT_NONSEP_MAC	1
 
+#define CONNECT_RFCOMM_TRIES	3
+#define CONNECT_RFCOMM_RETRY_MS	100
+
 /* Silence warning about (currently) unused routine. */
 #define WITH_WRITE_TYPE_HANDLE	0
 
@@ -668,6 +670,7 @@ SR_PRIV int sr_bt_connect_ble(struct sr_bt_desc *desc)
 	struct sockaddr_l2 sl2;
 	bdaddr_t mac;
 	int s, ret;
+	gint64 deadline;
 
 	if (!desc)
 		return -1;
@@ -710,6 +713,8 @@ SR_PRIV int sr_bt_connect_ble(struct sr_bt_desc *desc)
 		}
 	}
 
+	deadline = g_get_monotonic_time();
+	deadline += CONNECT_BLE_TIMEOUT * 1000 * 1000;
 	str2ba(desc->remote_addr, &mac);
 	memcpy(&sl2.l2_bdaddr, &mac, sizeof(sl2.l2_bdaddr));
 	sl2.l2_bdaddr_type = BDADDR_LE_PUBLIC;
@@ -745,6 +750,10 @@ SR_PRIV int sr_bt_connect_ble(struct sr_bt_desc *desc)
 				continue;
 			if (!(fds[0].revents & POLLOUT))
 				continue;
+			if (g_get_monotonic_time() >= deadline) {
+				sr_warn("Connect attempt timed out");
+				return SR_ERR_IO;
+			}
 		} while (1);
 		memset(fds, 0, sizeof(fds));
 		fds[0].fd = s;
@@ -789,7 +798,7 @@ SR_PRIV int sr_bt_connect_ble(struct sr_bt_desc *desc)
 SR_PRIV int sr_bt_connect_rfcomm(struct sr_bt_desc *desc)
 {
 	struct sockaddr_rc addr;
-	int fd, rc;
+	int i, fd, rc;
 
 	if (!desc)
 		return -1;
@@ -801,25 +810,40 @@ SR_PRIV int sr_bt_connect_rfcomm(struct sr_bt_desc *desc)
 	if (!desc->rfcomm_channel)
 		desc->rfcomm_channel = 1;
 
-	fd = socket(AF_BLUETOOTH, SOCK_STREAM, BTPROTO_RFCOMM);
-	if (fd < 0) {
-		perror("socket");
-		return -1;
-	}
-	desc->fd = fd;
-
 	memset(&addr, 0, sizeof(addr));
 	addr.rc_family = AF_BLUETOOTH;
 	str2ba(desc->remote_addr, &addr.rc_bdaddr);
 	addr.rc_channel = desc->rfcomm_channel;
-	rc = connect(fd, (struct sockaddr *)&addr, sizeof(addr));
-	if (rc < 0) {
-		perror("connect");
-		return -2;
+
+	/*
+	 * There are cases where connect returns EBUSY if we are re-connecting
+	 * to a device. Try multiple times to work around this issue.
+	 */
+	for (i = 0; i < CONNECT_RFCOMM_TRIES; i++) {
+		fd = socket(AF_BLUETOOTH, SOCK_STREAM, BTPROTO_RFCOMM);
+		if (fd < 0) {
+			perror("socket");
+			return -1;
+		}
+
+		rc = connect(fd, (struct sockaddr *)&addr, sizeof(addr));
+		if (rc >= 0) {
+			sr_spew("connected");
+			desc->fd = fd;
+			return 0;
+		} else if (rc < 0 && errno == EBUSY) {
+			close(fd);
+			g_usleep(CONNECT_RFCOMM_RETRY_MS * 1000);
+		} else {
+			close(fd);
+			perror("connect");
+			return -2;
+		}
 	}
-	sr_spew("connected");
 
-	return 0;
+	sr_err("Connect failed, device busy.");
+
+	return -2;
 }
 
 SR_PRIV void sr_bt_disconnect(struct sr_bt_desc *desc)