Skip to content

Commit

Permalink
Introduce gattlib_connection_is_valid() to not access 'connection->de…
Browse files Browse the repository at this point in the history
…vice'
  • Loading branch information
oliviermartin committed Apr 8, 2024
1 parent 2d771d9 commit db04a0e
Show file tree
Hide file tree
Showing 7 changed files with 57 additions and 15 deletions.
2 changes: 1 addition & 1 deletion common/gattlib_callback_connected_device.c
Original file line number Diff line number Diff line change
Expand Up @@ -87,7 +87,7 @@ static void* _connected_device_thread_args_allocator(va_list args) {
}

void gattlib_on_connected_device(gattlib_connection_t* connection) {
if (!gattlib_device_is_valid(connection->device)) {
if (!gattlib_connection_is_valid(connection)) {
GATTLIB_LOG(GATTLIB_ERROR, "gattlib_on_connected_device: Device is not valid");
return;
}
Expand Down
2 changes: 1 addition & 1 deletion common/gattlib_callback_disconnected_device.c
Original file line number Diff line number Diff line change
Expand Up @@ -33,7 +33,7 @@ void gattlib_disconnected_device_python_callback(gattlib_connection_t* connectio
void gattlib_on_disconnected_device(gattlib_connection_t* connection) {
g_rec_mutex_lock(&m_gattlib_mutex);

if (!gattlib_device_is_valid(connection->device)) {
if (!gattlib_connection_is_valid(connection)) {
GATTLIB_LOG(GATTLIB_ERROR, "gattlib_on_disconnected_device: Device not valid");
g_rec_mutex_unlock(&m_gattlib_mutex);
return;
Expand Down
6 changes: 3 additions & 3 deletions common/gattlib_common.c
Original file line number Diff line number Diff line change
Expand Up @@ -20,7 +20,7 @@ int gattlib_register_notification(gattlib_connection_t* connection, gattlib_even
goto EXIT;
}

if (!gattlib_device_is_valid(connection->device)) {
if (!gattlib_connection_is_valid(connection)) {
GATTLIB_LOG(GATTLIB_ERROR, "gattlib_register_notification: Device not valid");
ret = GATTLIB_DEVICE_DISCONNECTED;
goto EXIT;
Expand Down Expand Up @@ -58,7 +58,7 @@ int gattlib_register_indication(gattlib_connection_t* connection, gattlib_event_
goto EXIT;
}

if (!gattlib_device_is_valid(connection->device)) {
if (!gattlib_connection_is_valid(connection)) {
GATTLIB_LOG(GATTLIB_ERROR, "gattlib_register_indication: Device not valid");
ret = GATTLIB_DEVICE_DISCONNECTED;
goto EXIT;
Expand Down Expand Up @@ -93,7 +93,7 @@ int gattlib_register_on_disconnect(gattlib_connection_t *connection, gattlib_dis
goto EXIT;
}

if (!gattlib_device_is_valid(connection->device)) {
if (!gattlib_connection_is_valid(connection)) {
GATTLIB_LOG(GATTLIB_ERROR, "gattlib_register_on_disconnect: Device not valid");
ret = GATTLIB_DEVICE_DISCONNECTED;
goto EXIT;
Expand Down
39 changes: 37 additions & 2 deletions common/gattlib_common_adapter.c
Original file line number Diff line number Diff line change
Expand Up @@ -103,16 +103,51 @@ bool gattlib_device_is_valid(gattlib_device_t* device) {
return device_is_valid.found;
}

struct _connection_is_connected {
struct _connection_is_valid {
gattlib_connection_t* connection;
bool is_connected;
bool is_valid;
};

static gint _is_device_connection(gconstpointer a, gconstpointer b) {
const gattlib_device_t* device = a;
return (&device->connection == b) ? 0 : -1; // We need to return 0 when it matches
}

static void _gattlib_connection_is_valid(gpointer data, gpointer user_data) {
gattlib_adapter_t* adapter = data;
struct _connection_is_valid* connection_is_valid = user_data;

//printf("_gattlib_connection_is_connected: Check device in adapter:%s\n", adapter->id);

GSList *device_entry = g_slist_find_custom(adapter->devices, connection_is_valid->connection, _is_device_connection);
if (device_entry == NULL) {
//printf("_gattlib_connection_is_connected: Did not find device %s\n", connection_is_connected->connection->device->device_id);
return;
}

connection_is_valid->is_valid = true;
}

bool gattlib_connection_is_valid(gattlib_connection_t* connection) {
struct _connection_is_valid connection_is_valid = {
.connection = connection,
.is_valid = false
};

g_rec_mutex_lock(&m_gattlib_mutex);
//printf("gattlib_connection_is_connected A");
g_slist_foreach(m_adapter_list, _gattlib_connection_is_valid, &connection_is_valid);
//printf("gattlib_connection_is_connected B");
g_rec_mutex_unlock(&m_gattlib_mutex);

return connection_is_valid.is_valid;
}

struct _connection_is_connected {
gattlib_connection_t* connection;
bool is_connected;
};

static void _gattlib_connection_is_connected(gpointer data, gpointer user_data) {
gattlib_adapter_t* adapter = data;
struct _connection_is_connected* connection_is_connected = user_data;
Expand Down
7 changes: 7 additions & 0 deletions common/gattlib_internal.h
Original file line number Diff line number Diff line change
Expand Up @@ -133,6 +133,13 @@ int gattlib_adapter_unref(gattlib_adapter_t* adapter);
bool gattlib_device_is_valid(gattlib_device_t* device);
int gattlib_device_ref(gattlib_device_t* device);
int gattlib_device_unref(gattlib_device_t* device);
/**
* This function is similar to 'gattlib_device_is_valid()' except we check if
* the connection (connected or not) still belongs to a valid device.
*
* It is to avoid to use 'connection->device' when the device has been freed
*/
bool gattlib_connection_is_valid(gattlib_connection_t* connection);
bool gattlib_connection_is_connected(gattlib_connection_t* connection);

void gattlib_handler_dispatch_to_thread(struct gattlib_handler* handler, void (*python_callback)(),
Expand Down
14 changes: 7 additions & 7 deletions dbus/gattlib.c
Original file line number Diff line number Diff line change
Expand Up @@ -21,7 +21,7 @@ static void _on_device_connect(gattlib_connection_t* connection) {

g_rec_mutex_lock(&m_gattlib_mutex);

if (!gattlib_device_is_valid(connection->device)) {
if (!gattlib_connection_is_valid(connection)) {
GATTLIB_LOG(GATTLIB_ERROR, "_on_device_connect: Device not valid");
goto EXIT;
}
Expand Down Expand Up @@ -142,7 +142,7 @@ static gboolean _stop_connect_func(gpointer data) {

g_rec_mutex_lock(&m_gattlib_mutex);

if (!gattlib_device_is_valid(connection->device)) {
if (!gattlib_connection_is_valid(connection)) {
GATTLIB_LOG(GATTLIB_ERROR, "_stop_connect_func: Device not valid");
goto EXIT;
}
Expand Down Expand Up @@ -424,7 +424,7 @@ int gattlib_discover_primary(gattlib_connection_t* connection, gattlib_primary_s
return GATTLIB_INVALID_PARAMETER;
}

if (!gattlib_device_is_valid(connection->device)) {
if (!gattlib_connection_is_valid(connection)) {
GATTLIB_LOG(GATTLIB_ERROR, "gattlib_discover_primary: Device not valid");
g_rec_mutex_unlock(&m_gattlib_mutex);
return GATTLIB_DEVICE_DISCONNECTED;
Expand Down Expand Up @@ -517,7 +517,7 @@ int gattlib_discover_primary(gattlib_connection_t* connection, gattlib_primary_s
goto EXIT;
}

if (!gattlib_device_is_valid(connection->device)) {
if (!gattlib_connection_is_valid(connection)) {
GATTLIB_LOG(GATTLIB_ERROR, "gattlib_discover_primary: Device not valid");
ret = GATTLIB_DEVICE_DISCONNECTED;
goto EXIT;
Expand Down Expand Up @@ -679,7 +679,7 @@ int gattlib_discover_char_range(gattlib_connection_t* connection, uint16_t start
// Increase bluez_device object reference counter to avoid to keep locking the mutex
g_rec_mutex_lock(&m_gattlib_mutex);

if (!gattlib_device_is_valid(connection->device)) {
if (!gattlib_connection_is_valid(connection)) {
GATTLIB_LOG(GATTLIB_ERROR, "gattlib_discover_char_range: Device not valid");
g_rec_mutex_unlock(&m_gattlib_mutex);
return GATTLIB_DEVICE_DISCONNECTED;
Expand Down Expand Up @@ -939,7 +939,7 @@ int gattlib_discover_char_range(gattlib_connection_t* connection, uint16_t start

g_rec_mutex_lock(&m_gattlib_mutex);

if (!gattlib_device_is_valid(connection->device)) {
if (!gattlib_connection_is_valid(connection)) {
GATTLIB_LOG(GATTLIB_ERROR, "gattlib_discover_char_range: Device not valid");
ret = GATTLIB_DEVICE_DISCONNECTED;
goto EXIT;
Expand Down Expand Up @@ -1119,7 +1119,7 @@ int gattlib_get_rssi(gattlib_connection_t *connection, int16_t *rssi)
return GATTLIB_INVALID_PARAMETER;
}

if (!gattlib_device_is_valid(connection->device)) {
if (!gattlib_connection_is_valid(connection)) {
GATTLIB_LOG(GATTLIB_ERROR, "gattlib_get_rssi: Device not valid");
g_rec_mutex_unlock(&m_gattlib_mutex);
return GATTLIB_DEVICE_DISCONNECTED;
Expand Down
2 changes: 1 addition & 1 deletion dbus/gattlib_advertisement.c
Original file line number Diff line number Diff line change
Expand Up @@ -125,7 +125,7 @@ int gattlib_get_advertisement_data(gattlib_connection_t *connection,
return GATTLIB_INVALID_PARAMETER;
}

if (!gattlib_device_is_valid(connection->device)) {
if (!gattlib_connection_is_valid(connection)) {
g_rec_mutex_unlock(&m_gattlib_mutex);
return GATTLIB_DEVICE_DISCONNECTED;
}
Expand Down

0 comments on commit db04a0e

Please sign in to comment.