diff options
Diffstat (limited to 'src/irmd')
| -rw-r--r-- | src/irmd/main.c | 25 | ||||
| -rw-r--r-- | src/irmd/registry.c | 3 |
2 files changed, 17 insertions, 11 deletions
diff --git a/src/irmd/main.c b/src/irmd/main.c index c7c1566d..8503fcfa 100644 --- a/src/irmd/main.c +++ b/src/irmd/main.c @@ -787,10 +787,10 @@ static int flow_alloc_resp(pid_t n_api, return -1; } - registry_del_api(&irmd->registry, n_api); - pthread_mutex_unlock(&rne->state_lock); + registry_del_api(&irmd->registry, n_api); + pthread_rwlock_unlock(&irmd->reg_lock); if (!response) { @@ -1401,10 +1401,11 @@ void * irm_flow_cleaner() } pthread_rwlock_unlock(&irmd->flows_lock); - pthread_rwlock_wrlock(&irmd->reg_lock); registry_sanitize_apis(&irmd->registry); + pthread_rwlock_wrlock(&irmd->reg_lock); + list_for_each_safe(pos, n, &irmd->spawned_apis) { struct spawned_api * api = list_entry(pos, struct spawned_api, next); @@ -1431,6 +1432,11 @@ static void clean_msg(void * msg) irm_msg__free_unpacked(msg, NULL); } +static void close_ptr(void * o) +{ + close(*((int *) o)); +} + void * mainloop() { uint8_t buf[IRM_MSG_BUF_SIZE]; @@ -1465,6 +1471,7 @@ void * mainloop() continue; } + pthread_cleanup_push(close_ptr, &cli_sockfd); pthread_cleanup_push(clean_msg, (void *) msg); switch (msg->code) { @@ -1595,6 +1602,7 @@ void * mainloop() } pthread_cleanup_pop(true); + pthread_cleanup_pop(false); buffer.len = irm_msg__get_packed_size(&ret_msg); if (buffer.len == 0) { @@ -1615,17 +1623,12 @@ void * mainloop() irm_msg__pack(&ret_msg, buffer.data); - if (write(cli_sockfd, buffer.data, buffer.len) == -1) { - free(buffer.data); - if (apis != NULL) - free(apis); - close(cli_sockfd); - continue; - } - if (apis != NULL) free(apis); + if (write(cli_sockfd, buffer.data, buffer.len) == -1) + LOG_ERR("Failed to send reply message."); + free(buffer.data); close(cli_sockfd); } diff --git a/src/irmd/registry.c b/src/irmd/registry.c index ae8c3b0a..04f60009 100644 --- a/src/irmd/registry.c +++ b/src/irmd/registry.c @@ -544,6 +544,8 @@ void registry_del_api(struct list_head * registry, reg_api_destroy(i); + pthread_mutex_lock(&e->state_lock); + if (list_empty(&e->reg_apis)) { if (reg_entry_has_auto_binding(e)) e->state = REG_NAME_AUTO_ACCEPT; @@ -554,6 +556,7 @@ void registry_del_api(struct list_head * registry, } pthread_cond_signal(&e->state_cond); + pthread_mutex_unlock(&e->state_lock); return; } |