summaryrefslogtreecommitdiff
path: root/src/irmd/main.c
diff options
context:
space:
mode:
Diffstat (limited to 'src/irmd/main.c')
-rw-r--r--src/irmd/main.c243
1 files changed, 36 insertions, 207 deletions
diff --git a/src/irmd/main.c b/src/irmd/main.c
index f9acb238..5e5039b1 100644
--- a/src/irmd/main.c
+++ b/src/irmd/main.c
@@ -108,6 +108,28 @@ struct irm {
pthread_t shm_sanitize; /* keep track of rdrbuff use */
} irmd;
+static enum irm_state irmd_get_state(void)
+{
+ enum irm_state state;
+
+ pthread_rwlock_rdlock(&irmd.state_lock);
+
+ state = irmd.state;
+
+ pthread_rwlock_unlock(&irmd.state_lock);
+
+ return state;
+}
+
+static void irmd_set_state(enum irm_state state)
+{
+ pthread_rwlock_wrlock(&irmd.state_lock);
+
+ irmd.state = state;
+
+ pthread_rwlock_unlock(&irmd.state_lock);
+}
+
static void clear_irm_flow(struct irm_flow * f) {
ssize_t idx;
@@ -257,19 +279,11 @@ static pid_t create_ipcp(char * name,
if (api == NULL)
return -ENOMEM;
- pthread_rwlock_rdlock(&irmd.state_lock);
-
- if (irmd.state != IRMD_RUNNING) {
- pthread_rwlock_unlock(&irmd.state_lock);
- return -1;
- }
-
pthread_rwlock_wrlock(&irmd.reg_lock);
entry = get_ipcp_entry_by_name(name);
if (entry != NULL) {
pthread_rwlock_unlock(&irmd.reg_lock);
- pthread_rwlock_unlock(&irmd.state_lock);
log_err("IPCP by that name already exists.");
return -1;
}
@@ -277,7 +291,6 @@ static pid_t create_ipcp(char * name,
api->pid = ipcp_create(name, ipcp_type);
if (api->pid == -1) {
pthread_rwlock_unlock(&irmd.reg_lock);
- pthread_rwlock_unlock(&irmd.state_lock);
log_err("Failed to create IPCP.");
return -1;
}
@@ -285,7 +298,6 @@ static pid_t create_ipcp(char * name,
tmp = ipcp_entry_create();
if (tmp == NULL) {
pthread_rwlock_unlock(&irmd.reg_lock);
- pthread_rwlock_unlock(&irmd.state_lock);
return -1;
}
@@ -296,7 +308,6 @@ static pid_t create_ipcp(char * name,
if (tmp->name == NULL) {
ipcp_entry_destroy(tmp);
pthread_rwlock_unlock(&irmd.reg_lock);
- pthread_rwlock_unlock(&irmd.state_lock);
return -1;
}
@@ -320,7 +331,6 @@ static pid_t create_ipcp(char * name,
pthread_mutex_lock(&tmp->init_lock);
pthread_rwlock_unlock(&irmd.reg_lock);
- pthread_rwlock_unlock(&irmd.state_lock);
while (tmp->init == false)
pthread_cond_wait(&tmp->init_cond, &tmp->init_lock);
@@ -340,7 +350,6 @@ static int create_ipcp_r(pid_t api,
if (result != 0)
return result;
- pthread_rwlock_rdlock(&irmd.state_lock);
pthread_rwlock_rdlock(&irmd.reg_lock);
list_for_each(pos, &irmd.ipcps) {
@@ -356,7 +365,6 @@ static int create_ipcp_r(pid_t api,
}
pthread_rwlock_unlock(&irmd.reg_lock);
- pthread_rwlock_unlock(&irmd.state_lock);
return 0;
}
@@ -380,7 +388,6 @@ static int destroy_ipcp(pid_t api)
struct list_head * pos = NULL;
struct list_head * n = NULL;
- pthread_rwlock_rdlock(&irmd.state_lock);
pthread_rwlock_wrlock(&irmd.reg_lock);
list_for_each_safe(pos, n, &(irmd.ipcps)) {
@@ -399,7 +406,6 @@ static int destroy_ipcp(pid_t api)
}
pthread_rwlock_unlock(&irmd.reg_lock);
- pthread_rwlock_unlock(&irmd.state_lock);
return 0;
}
@@ -409,33 +415,23 @@ static int bootstrap_ipcp(pid_t api,
{
struct ipcp_entry * entry = NULL;
- pthread_rwlock_rdlock(&irmd.state_lock);
-
- if (irmd.state != IRMD_RUNNING) {
- pthread_rwlock_unlock(&irmd.state_lock);
- return -1;
- }
-
pthread_rwlock_wrlock(&irmd.reg_lock);
entry = get_ipcp_entry_by_api(api);
if (entry == NULL) {
pthread_rwlock_unlock(&irmd.reg_lock);
- pthread_rwlock_unlock(&irmd.state_lock);
log_err("No such IPCP.");
return -1;
}
if (entry->type != (enum ipcp_type) conf->ipcp_type) {
pthread_rwlock_unlock(&irmd.reg_lock);
- pthread_rwlock_unlock(&irmd.state_lock);
log_err("Configuration does not match IPCP type.");
return -1;
}
if (ipcp_bootstrap(entry->api, conf)) {
pthread_rwlock_unlock(&irmd.reg_lock);
- pthread_rwlock_unlock(&irmd.state_lock);
log_err("Could not bootstrap IPCP.");
return -1;
}
@@ -443,13 +439,11 @@ static int bootstrap_ipcp(pid_t api,
entry->dif_name = strdup(conf->dif_name);
if (entry->dif_name == NULL) {
pthread_rwlock_unlock(&irmd.reg_lock);
- pthread_rwlock_unlock(&irmd.state_lock);
log_warn("Failed to set name of DIF.");
return -ENOMEM;
}
pthread_rwlock_unlock(&irmd.reg_lock);
- pthread_rwlock_unlock(&irmd.state_lock);
log_info("Bootstrapped IPCP %d in DIF %s.",
entry->api, conf->dif_name);
@@ -462,26 +456,17 @@ static int enroll_ipcp(pid_t api,
{
struct ipcp_entry * entry = NULL;
- pthread_rwlock_rdlock(&irmd.state_lock);
-
- if (irmd.state != IRMD_RUNNING) {
- pthread_rwlock_unlock(&irmd.state_lock);
- return -1;
- }
-
pthread_rwlock_wrlock(&irmd.reg_lock);
entry = get_ipcp_entry_by_api(api);
if (entry == NULL) {
pthread_rwlock_unlock(&irmd.reg_lock);
- pthread_rwlock_unlock(&irmd.state_lock);
log_err("No such IPCP.");
return -1;
}
if (entry->dif_name != NULL) {
pthread_rwlock_unlock(&irmd.reg_lock);
- pthread_rwlock_unlock(&irmd.state_lock);
log_err("IPCP in wrong state");
return -1;
}
@@ -489,21 +474,17 @@ static int enroll_ipcp(pid_t api,
entry->dif_name = strdup(dif_name);
if (entry->dif_name == NULL) {
pthread_rwlock_unlock(&irmd.reg_lock);
- pthread_rwlock_unlock(&irmd.state_lock);
log_err("Failed to strdup.");
return -1;
}
pthread_rwlock_unlock(&irmd.reg_lock);
- pthread_rwlock_unlock(&irmd.state_lock);
if (ipcp_enroll(api, dif_name)) {
- pthread_rwlock_rdlock(&irmd.state_lock);
pthread_rwlock_wrlock(&irmd.reg_lock);
free(entry->dif_name);
entry->dif_name = NULL;
pthread_rwlock_unlock(&irmd.reg_lock);
- pthread_rwlock_unlock(&irmd.state_lock);
log_err("Could not enroll IPCP.");
return -1;
}
@@ -531,13 +512,6 @@ static int bind_ap(char * ap,
if (ap == NULL || name == NULL)
return -EINVAL;
- pthread_rwlock_rdlock(&irmd.state_lock);
-
- if (irmd.state != IRMD_RUNNING) {
- pthread_rwlock_unlock(&irmd.state_lock);
- return -1;
- }
-
pthread_rwlock_wrlock(&irmd.reg_lock);
e = apn_table_get(&irmd.apn_table, path_strip(ap));
@@ -546,14 +520,12 @@ static int bind_ap(char * ap,
aps = strdup(path_strip(ap));
if (aps == NULL) {
pthread_rwlock_unlock(&irmd.reg_lock);
- pthread_rwlock_unlock(&irmd.state_lock);
return -ENOMEM;
}
apn = strdup(name);
if (apn == NULL) {
pthread_rwlock_unlock(&irmd.reg_lock);
- pthread_rwlock_unlock(&irmd.state_lock);
free(aps);
return -ENOMEM;
}
@@ -566,8 +538,6 @@ static int bind_ap(char * ap,
argv_dup[i] = strdup(argv[i - 1]);
if (argv_dup[i] == NULL) {
pthread_rwlock_unlock(&irmd.reg_lock);
- pthread_rwlock_unlock(
- &irmd.state_lock);
argvfree(argv_dup);
log_err("Failed to bind ap %s to %s.",
ap, name);
@@ -581,7 +551,6 @@ static int bind_ap(char * ap,
e = apn_entry_create(apn, aps, flags, argv_dup);
if (e == NULL) {
pthread_rwlock_unlock(&irmd.reg_lock);
- pthread_rwlock_unlock(&irmd.state_lock);
free(aps);
free(apn);
argvfree(argv_dup);
@@ -595,14 +564,12 @@ static int bind_ap(char * ap,
name_dup = strdup(name);
if (name_dup == NULL) {
pthread_rwlock_unlock(&irmd.reg_lock);
- pthread_rwlock_unlock(&irmd.state_lock);
return -ENOMEM;
}
if (apn_entry_add_name(e, name_dup)) {
log_err("Failed adding name.");
pthread_rwlock_unlock(&irmd.reg_lock);
- pthread_rwlock_unlock(&irmd.state_lock);
free(name_dup);
return -ENOMEM;
}
@@ -612,7 +579,6 @@ static int bind_ap(char * ap,
log_err("Failed adding AP %s for name %s.", ap, name);
pthread_rwlock_unlock(&irmd.reg_lock);
- pthread_rwlock_unlock(&irmd.state_lock);
log_info("Bound AP %s to name %s.", ap, name);
@@ -629,33 +595,23 @@ static int bind_api(pid_t api,
if (name == NULL)
return -EINVAL;
- pthread_rwlock_rdlock(&irmd.state_lock);
-
- if (irmd.state != IRMD_RUNNING) {
- pthread_rwlock_unlock(&irmd.state_lock);
- return -1;
- }
-
pthread_rwlock_wrlock(&irmd.reg_lock);
e = api_table_get(&irmd.api_table, api);
if (e == NULL) {
log_err("AP-I %d does not exist.", api);
pthread_rwlock_unlock(&irmd.reg_lock);
- pthread_rwlock_unlock(&irmd.state_lock);
return -1;
}
name_dup = strdup(name);
if (name_dup == NULL) {
pthread_rwlock_unlock(&irmd.reg_lock);
- pthread_rwlock_unlock(&irmd.state_lock);
return -ENOMEM;
}
if (api_entry_add_name(e, name_dup)) {
pthread_rwlock_unlock(&irmd.reg_lock);
- pthread_rwlock_unlock(&irmd.state_lock);
log_err("Failed to add name %s to api %d.", name, api);
free(name_dup);
return -1;
@@ -666,7 +622,6 @@ static int bind_api(pid_t api,
log_err("Failed adding AP-I %d for name %s.", api, name);
pthread_rwlock_unlock(&irmd.reg_lock);
- pthread_rwlock_unlock(&irmd.state_lock);
log_info("Bound AP-I %d to name %s.", api, name);
@@ -679,13 +634,6 @@ static int unbind_ap(char * ap,
if (ap == NULL)
return -EINVAL;
- pthread_rwlock_rdlock(&irmd.state_lock);
-
- if (irmd.state != IRMD_RUNNING) {
- pthread_rwlock_unlock(&irmd.state_lock);
- return -1;
- }
-
pthread_rwlock_wrlock(&irmd.reg_lock);
if (name == NULL)
@@ -696,7 +644,6 @@ static int unbind_ap(char * ap,
}
pthread_rwlock_unlock(&irmd.reg_lock);
- pthread_rwlock_unlock(&irmd.state_lock);
if (name == NULL)
log_info("AP %s removed.", ap);
@@ -709,13 +656,6 @@ static int unbind_ap(char * ap,
static int unbind_api(pid_t api,
char * name)
{
- pthread_rwlock_rdlock(&irmd.state_lock);
-
- if (irmd.state != IRMD_RUNNING) {
- pthread_rwlock_unlock(&irmd.state_lock);
- return -1;
- }
-
pthread_rwlock_wrlock(&irmd.reg_lock);
if (name == NULL)
@@ -726,7 +666,6 @@ static int unbind_api(pid_t api,
}
pthread_rwlock_unlock(&irmd.reg_lock);
- pthread_rwlock_unlock(&irmd.state_lock);
if (name == NULL)
log_info("AP-I %d removed.", api);
@@ -743,7 +682,6 @@ static ssize_t list_ipcps(char * name,
size_t count = 0;
int i = 0;
- pthread_rwlock_rdlock(&irmd.state_lock);
pthread_rwlock_rdlock(&irmd.reg_lock);
list_for_each(pos, &irmd.ipcps) {
@@ -756,7 +694,6 @@ static ssize_t list_ipcps(char * name,
*apis = malloc(count * sizeof(pid_t));
if (*apis == NULL) {
pthread_rwlock_unlock(&irmd.reg_lock);
- pthread_rwlock_unlock(&irmd.state_lock);
return -1;
}
@@ -768,7 +705,6 @@ static ssize_t list_ipcps(char * name,
}
pthread_rwlock_unlock(&irmd.reg_lock);
- pthread_rwlock_unlock(&irmd.state_lock);
return count;
}
@@ -784,18 +720,10 @@ static int name_reg(char * name,
if (name == NULL || difs == NULL || len == 0 || difs[0] == NULL)
return -EINVAL;
- pthread_rwlock_rdlock(&irmd.state_lock);
-
- if (irmd.state != IRMD_RUNNING) {
- pthread_rwlock_unlock(&irmd.state_lock);
- return -1;
- }
-
pthread_rwlock_wrlock(&irmd.reg_lock);
if (list_is_empty(&irmd.ipcps)) {
pthread_rwlock_unlock(&irmd.reg_lock);
- pthread_rwlock_unlock(&irmd.state_lock);
return -1;
}
@@ -805,7 +733,6 @@ static int name_reg(char * name,
if (re == NULL) {
log_err("Failed creating registry entry for %s.", name);
pthread_rwlock_unlock(&irmd.reg_lock);
- pthread_rwlock_unlock(&irmd.state_lock);
return -1;
}
@@ -863,7 +790,6 @@ static int name_reg(char * name,
}
pthread_rwlock_unlock(&irmd.reg_lock);
- pthread_rwlock_unlock(&irmd.state_lock);
return (ret > 0 ? 0 : -1);
}
@@ -879,13 +805,6 @@ static int name_unreg(char * name,
if (name == NULL || len == 0 || difs == NULL || difs[0] == NULL)
return -1;
- pthread_rwlock_rdlock(&irmd.state_lock);
-
- if (irmd.state != IRMD_RUNNING) {
- pthread_rwlock_unlock(&irmd.state_lock);
- return -1;
- }
-
pthread_rwlock_wrlock(&irmd.reg_lock);
list_for_each(pos, &irmd.ipcps) {
@@ -914,7 +833,6 @@ static int name_unreg(char * name,
}
pthread_rwlock_unlock(&irmd.reg_lock);
- pthread_rwlock_unlock(&irmd.state_lock);
return (ret > 0 ? 0 : -1);
}
@@ -928,22 +846,13 @@ static int api_announce(pid_t api,
if (apn == NULL)
return -EINVAL;
- pthread_rwlock_rdlock(&irmd.state_lock);
-
- if (irmd.state != IRMD_RUNNING) {
- pthread_rwlock_unlock(&irmd.state_lock);
- return -EPERM;
- }
-
apn_dup = strdup(apn);
if (apn_dup == NULL) {
- pthread_rwlock_unlock(&irmd.state_lock);
return -ENOMEM;
}
e = api_entry_create(api, apn_dup);
if (e == NULL) {
- pthread_rwlock_unlock(&irmd.state_lock);
return -ENOMEM;
}
@@ -961,13 +870,11 @@ static int api_announce(pid_t api,
struct str_el * n = malloc(sizeof(*n));
if (n == NULL) {
pthread_rwlock_unlock(&irmd.reg_lock);
- pthread_rwlock_unlock(&irmd.state_lock);
return -ENOMEM;
}
n->str = strdup(s->str);
if (n->str == NULL) {
pthread_rwlock_unlock(&irmd.reg_lock);
- pthread_rwlock_unlock(&irmd.state_lock);
free(n);
}
@@ -978,7 +885,6 @@ static int api_announce(pid_t api,
}
pthread_rwlock_unlock(&irmd.reg_lock);
- pthread_rwlock_unlock(&irmd.state_lock);
return 0;
}
@@ -1005,20 +911,12 @@ static int flow_accept(pid_t api,
ts_add(&now, timeo, &dl);
}
- pthread_rwlock_rdlock(&irmd.state_lock);
-
- if (irmd.state != IRMD_RUNNING) {
- pthread_rwlock_unlock(&irmd.state_lock);
- return -EIRMD;
- }
-
pthread_rwlock_wrlock(&irmd.reg_lock);
e = api_table_get(&irmd.api_table, api);
if (e == NULL) {
/* Can only happen if server called ap_init(NULL); */
pthread_rwlock_unlock(&irmd.reg_lock);
- pthread_rwlock_unlock(&irmd.state_lock);
log_err("Unknown instance %d calling accept.", api);
return -EINVAL;
}
@@ -1035,7 +933,6 @@ static int flow_accept(pid_t api,
}
pthread_rwlock_unlock(&irmd.reg_lock);
- pthread_rwlock_unlock(&irmd.state_lock);
while (true) {
if (timeo != NULL && ts_diff_ns(&now, &dl) < 0) {
@@ -1043,14 +940,8 @@ static int flow_accept(pid_t api,
return -ETIMEDOUT;
}
- pthread_rwlock_rdlock(&irmd.state_lock);
-
- if (irmd.state != IRMD_RUNNING) {
- pthread_rwlock_unlock(&irmd.state_lock);
+ if (irmd_get_state() != IRMD_RUNNING)
return -EIRMD;
- }
-
- pthread_rwlock_unlock(&irmd.state_lock);
ret = api_entry_sleep(e);
if (ret == -ETIMEDOUT) {
@@ -1066,11 +957,8 @@ static int flow_accept(pid_t api,
break;
}
- pthread_rwlock_rdlock(&irmd.state_lock);
-
- if (irmd.state != IRMD_RUNNING) {
+ if (irmd_get_state() != IRMD_RUNNING) {
reg_entry_set_state(re, REG_NAME_NULL);
- pthread_rwlock_unlock(&irmd.state_lock);
return -EIRMD;
}
@@ -1079,7 +967,6 @@ static int flow_accept(pid_t api,
f = get_irm_flow_n(api);
if (f == NULL) {
pthread_rwlock_unlock(&irmd.flows_lock);
- pthread_rwlock_unlock(&irmd.state_lock);
log_warn("Port_id was not created yet.");
return -EPERM;
}
@@ -1098,7 +985,6 @@ static int flow_accept(pid_t api,
list_del(&f->next);
bmp_release(irmd.port_ids, f->port_id);
pthread_rwlock_unlock(&irmd.flows_lock);
- pthread_rwlock_unlock(&irmd.state_lock);
ipcp_flow_alloc_resp(api_n1, port_id, api_n, -1);
clear_irm_flow(f);
irm_flow_set_state(f, FLOW_NULL);
@@ -1119,7 +1005,6 @@ static int flow_accept(pid_t api,
list_del(&f->next);
bmp_release(irmd.port_ids, f->port_id);
pthread_rwlock_unlock(&irmd.flows_lock);
- pthread_rwlock_unlock(&irmd.state_lock);
ipcp_flow_alloc_resp(api_n1, port_id, api_n, -1);
clear_irm_flow(f);
irm_flow_set_state(f, FLOW_NULL);
@@ -1131,14 +1016,11 @@ static int flow_accept(pid_t api,
registry_del_api(&irmd.registry, api);
pthread_rwlock_unlock(&irmd.reg_lock);
- pthread_rwlock_unlock(&irmd.state_lock);
if (ipcp_flow_alloc_resp(api_n1, port_id, api_n, 0)) {
- pthread_rwlock_rdlock(&irmd.state_lock);
pthread_rwlock_wrlock(&irmd.flows_lock);
list_del(&f->next);
pthread_rwlock_unlock(&irmd.flows_lock);
- pthread_rwlock_unlock(&irmd.state_lock);
log_dbg("Failed to respond to alloc. Port_id invalidated.");
clear_irm_flow(f);
irm_flow_set_state(f, FLOW_NULL);
@@ -1166,19 +1048,11 @@ static int flow_alloc(pid_t api,
int port_id;
int state;
- pthread_rwlock_rdlock(&irmd.state_lock);
-
- if (irmd.state != IRMD_RUNNING) {
- pthread_rwlock_unlock(&irmd.state_lock);
- return -1;
- }
-
pthread_rwlock_rdlock(&irmd.reg_lock);
ipcp = get_ipcp_by_dst_name(dst_name);
if (ipcp == -1) {
pthread_rwlock_unlock(&irmd.reg_lock);
- pthread_rwlock_unlock(&irmd.state_lock);
log_info("Destination unreachable.");
return -1;
}
@@ -1188,7 +1062,6 @@ static int flow_alloc(pid_t api,
port_id = bmp_allocate(irmd.port_ids);
if (!bmp_is_id_valid(irmd.port_ids, port_id)) {
pthread_rwlock_unlock(&irmd.flows_lock);
- pthread_rwlock_unlock(&irmd.state_lock);
log_err("Could not allocate port_id.");
return -EBADF;
}
@@ -1197,7 +1070,6 @@ static int flow_alloc(pid_t api,
if (f == NULL) {
bmp_release(irmd.port_ids, port_id);
pthread_rwlock_unlock(&irmd.flows_lock);
- pthread_rwlock_unlock(&irmd.state_lock);
log_err("Could not allocate port_id.");
return -ENOMEM;
}
@@ -1205,7 +1077,6 @@ static int flow_alloc(pid_t api,
list_add(&f->next, &irmd.irm_flows);
pthread_rwlock_unlock(&irmd.flows_lock);
- pthread_rwlock_unlock(&irmd.state_lock);
assert(irm_flow_get_state(f) == FLOW_ALLOC_PENDING);
@@ -1243,13 +1114,11 @@ static int flow_dealloc(pid_t api,
struct irm_flow * f = NULL;
- pthread_rwlock_rdlock(&irmd.state_lock);
pthread_rwlock_wrlock(&irmd.flows_lock);
f = get_irm_flow(port_id);
if (f == NULL) {
pthread_rwlock_unlock(&irmd.flows_lock);
- pthread_rwlock_unlock(&irmd.state_lock);
log_dbg("Deallocate unknown port %d by %d.", port_id, api);
return 0;
}
@@ -1261,7 +1130,6 @@ static int flow_dealloc(pid_t api,
f->n_1_api = -1;
} else {
pthread_rwlock_unlock(&irmd.flows_lock);
- pthread_rwlock_unlock(&irmd.state_lock);
log_dbg("Dealloc called by wrong AP-I.");
return -EPERM;
}
@@ -1283,7 +1151,6 @@ static int flow_dealloc(pid_t api,
}
pthread_rwlock_unlock(&irmd.flows_lock);
- pthread_rwlock_unlock(&irmd.state_lock);
if (n_1_api != -1)
ret = ipcp_flow_dealloc(n_1_api, port_id);
@@ -1342,19 +1209,16 @@ static struct irm_flow * flow_req_arr(pid_t api,
log_dbg("Flow req arrived from IPCP %d for %s.", api, dst_name);
- pthread_rwlock_rdlock(&irmd.state_lock);
pthread_rwlock_rdlock(&irmd.reg_lock);
re = registry_get_entry(&irmd.registry, dst_name);
if (re == NULL) {
pthread_rwlock_unlock(&irmd.reg_lock);
- pthread_rwlock_unlock(&irmd.state_lock);
log_err("Unknown name: %s.", dst_name);
return NULL;
}
pthread_rwlock_unlock(&irmd.reg_lock);
- pthread_rwlock_unlock(&irmd.state_lock);
/* Give the AP a bit of slop time to call accept */
if (reg_entry_leave_state(re, REG_NAME_IDLE, &wt) == -1) {
@@ -1362,20 +1226,17 @@ static struct irm_flow * flow_req_arr(pid_t api,
return NULL;
}
- pthread_rwlock_rdlock(&irmd.state_lock);
pthread_rwlock_wrlock(&irmd.reg_lock);
switch (reg_entry_get_state(re)) {
case REG_NAME_IDLE:
pthread_rwlock_unlock(&irmd.reg_lock);
- pthread_rwlock_unlock(&irmd.state_lock);
log_err("No APs for %s.", dst_name);
return NULL;
case REG_NAME_AUTO_ACCEPT:
c_api = malloc(sizeof(*c_api));
if (c_api == NULL) {
pthread_rwlock_unlock(&irmd.reg_lock);
- pthread_rwlock_unlock(&irmd.state_lock);
return NULL;
}
@@ -1386,7 +1247,6 @@ static struct irm_flow * flow_req_arr(pid_t api,
if (a == NULL || (c_api->pid = auto_execute(a->argv)) < 0) {
reg_entry_set_state(re, REG_NAME_AUTO_ACCEPT);
pthread_rwlock_unlock(&irmd.reg_lock);
- pthread_rwlock_unlock(&irmd.state_lock);
log_err("Could not get start apn for reg_entry %s.",
re->name);
free(c_api);
@@ -1396,18 +1256,15 @@ static struct irm_flow * flow_req_arr(pid_t api,
list_add(&c_api->next, &irmd.spawned_apis);
pthread_rwlock_unlock(&irmd.reg_lock);
- pthread_rwlock_unlock(&irmd.state_lock);
if (reg_entry_leave_state(re, REG_NAME_AUTO_EXEC, NULL))
return NULL;
- pthread_rwlock_rdlock(&irmd.state_lock);
pthread_rwlock_wrlock(&irmd.reg_lock);
case REG_NAME_FLOW_ACCEPT:
h_api = reg_entry_get_api(re);
if (h_api == -1) {
pthread_rwlock_unlock(&irmd.reg_lock);
- pthread_rwlock_unlock(&irmd.state_lock);
log_err("Invalid api returned.");
return NULL;
}
@@ -1415,7 +1272,6 @@ static struct irm_flow * flow_req_arr(pid_t api,
break;
default:
pthread_rwlock_unlock(&irmd.reg_lock);
- pthread_rwlock_unlock(&irmd.state_lock);
log_err("IRMd in wrong state.");
return NULL;
}
@@ -1425,7 +1281,6 @@ static struct irm_flow * flow_req_arr(pid_t api,
port_id = bmp_allocate(irmd.port_ids);
if (!bmp_is_id_valid(irmd.port_ids, port_id)) {
pthread_rwlock_unlock(&irmd.flows_lock);
- pthread_rwlock_unlock(&irmd.state_lock);
return NULL;
}
@@ -1433,7 +1288,6 @@ static struct irm_flow * flow_req_arr(pid_t api,
if (f == NULL) {
bmp_release(irmd.port_ids, port_id);
pthread_rwlock_unlock(&irmd.flows_lock);
- pthread_rwlock_unlock(&irmd.state_lock);
log_err("Could not allocate port_id.");
return NULL;
}
@@ -1453,7 +1307,6 @@ static struct irm_flow * flow_req_arr(pid_t api,
bmp_release(irmd.port_ids, f->port_id);
list_del(&f->next);
pthread_rwlock_unlock(&irmd.flows_lock);
- pthread_rwlock_unlock(&irmd.state_lock);
log_err("Could not get api table entry for %d.", h_api);
irm_flow_destroy(f);
return NULL;
@@ -1462,7 +1315,6 @@ static struct irm_flow * flow_req_arr(pid_t api,
api_entry_wake(e, re);
pthread_rwlock_unlock(&irmd.reg_lock);
- pthread_rwlock_unlock(&irmd.state_lock);
reg_entry_leave_state(re, REG_NAME_FLOW_ARRIVED, NULL);
@@ -1474,13 +1326,11 @@ static int flow_alloc_reply(int port_id,
{
struct irm_flow * f;
- pthread_rwlock_rdlock(&irmd.state_lock);
pthread_rwlock_rdlock(&irmd.flows_lock);
f = get_irm_flow(port_id);
if (f == NULL) {
pthread_rwlock_unlock(&irmd.flows_lock);
- pthread_rwlock_unlock(&irmd.state_lock);
return -1;
}
@@ -1490,7 +1340,6 @@ static int flow_alloc_reply(int port_id,
irm_flow_set_state(f, FLOW_NULL);
pthread_rwlock_unlock(&irmd.flows_lock);
- pthread_rwlock_unlock(&irmd.state_lock);
return 0;
}
@@ -1500,9 +1349,7 @@ static void irm_fini(void)
struct list_head * p;
struct list_head * h;
- pthread_rwlock_rdlock(&irmd.state_lock);
-
- if (irmd.state != IRMD_NULL)
+ if (irmd_get_state() != IRMD_NULL)
log_warn("Unsafe destroy.");
pthread_mutex_lock(&irmd.threads_lock);
@@ -1567,8 +1414,6 @@ static void irm_fini(void)
if (irmd.lf != NULL)
lockfile_destroy(irmd.lf);
- pthread_rwlock_unlock(&irmd.state_lock);
-
pthread_rwlock_destroy(&irmd.reg_lock);
pthread_rwlock_destroy(&irmd.state_lock);
}
@@ -1584,21 +1429,14 @@ void irmd_sig_handler(int sig,
case SIGINT:
case SIGTERM:
case SIGHUP:
- pthread_rwlock_rdlock(&irmd.state_lock);
- if (irmd.state == IRMD_NULL) {
+ if (irmd_get_state() == IRMD_NULL) {
log_info("Patience is bitter, but its fruit is sweet.");
- pthread_rwlock_unlock(&irmd.state_lock);
return;
}
- pthread_rwlock_unlock(&irmd.state_lock);
log_info("IRMd shutting down...");
- pthread_rwlock_wrlock(&irmd.state_lock);
-
- irmd.state = IRMD_NULL;
-
- pthread_rwlock_unlock(&irmd.state_lock);
+ irmd_set_state(IRMD_NULL);
break;
case SIGPIPE:
log_dbg("Ignored SIGPIPE.");
@@ -1619,7 +1457,6 @@ void * shm_sanitize(void * o)
while (true) {
shm_rdrbuff_wait_full(irmd.rdrb);
- pthread_rwlock_rdlock(&irmd.state_lock);
pthread_rwlock_wrlock(&irmd.flows_lock);
list_for_each(p, &irmd.irm_flows) {
@@ -1639,7 +1476,6 @@ void * shm_sanitize(void * o)
}
pthread_rwlock_unlock(&irmd.flows_lock);
- pthread_rwlock_unlock(&irmd.state_lock);
nanosleep(&ts, NULL);
}
@@ -1663,9 +1499,7 @@ void * irm_sanitize(void * o)
if (clock_gettime(CLOCK_MONOTONIC, &now) < 0)
log_warn("Failed to get time.");
- pthread_rwlock_rdlock(&irmd.state_lock);
-
- if (irmd.state != IRMD_RUNNING) {
+ if (irmd_get_state() != IRMD_RUNNING) {
/* Clean up all flows first to kill mainloops */
pthread_rwlock_wrlock(&irmd.flows_lock);
list_for_each_safe(p, h, &irmd.irm_flows) {
@@ -1686,7 +1520,6 @@ void * irm_sanitize(void * o)
api_entry_destroy(e);
}
pthread_rwlock_unlock(&irmd.reg_lock);
- pthread_rwlock_unlock(&irmd.state_lock);
return (void *) 0;
}
@@ -1781,7 +1614,6 @@ void * irm_sanitize(void * o)
}
pthread_rwlock_unlock(&irmd.flows_lock);
- pthread_rwlock_unlock(&irmd.state_lock);
nanosleep(&timeout, NULL);
}
@@ -1855,16 +1687,11 @@ void * mainloop(void * o)
struct timeval tv = {(SOCKET_TIMEOUT / 1000),
(SOCKET_TIMEOUT % 1000) * 1000};
- pthread_rwlock_rdlock(&irmd.state_lock);
-
- if (irmd.state != IRMD_RUNNING || thread_check()) {
+ if (irmd_get_state() != IRMD_RUNNING || thread_check()) {
thread_exit(id);
- pthread_rwlock_unlock(&irmd.state_lock);
break;
}
- pthread_rwlock_unlock(&irmd.state_lock);
-
ret_msg.code = IRM_MSG_CODE__IRM_REPLY;
#ifdef __FreeBSD__
FD_ZERO(&fds);
@@ -1888,6 +1715,12 @@ void * mainloop(void * o)
continue;
}
+ if (irmd_get_state() != IRMD_RUNNING) {
+ close(cli_sockfd);
+ thread_exit(id);
+ break;
+ }
+
msg = irm_msg__unpack(NULL, count, buf);
if (msg == NULL) {
close(cli_sockfd);
@@ -2080,9 +1913,7 @@ void * threadpoolmgr(void * o)
clock_gettime(PTHREAD_COND_CLOCK, &dl);
ts_add(&dl, &to, &dl);
- pthread_rwlock_rdlock(&irmd.state_lock);
- if (irmd.state != IRMD_RUNNING) {
- pthread_rwlock_unlock(&irmd.state_lock);
+ if (irmd_get_state() != IRMD_RUNNING) {
pthread_attr_destroy(&pattr);
log_dbg("Waiting for threads to exit.");
pthread_mutex_lock(&irmd.threads_lock);
@@ -2094,8 +1925,6 @@ void * threadpoolmgr(void * o)
break;
}
- pthread_rwlock_unlock(&irmd.state_lock);
-
pthread_mutex_lock(&irmd.threads_lock);
if (irmd.threads < IRMD_MIN_AV_THREADS) {
generated by cgit v1.2.3 (git 2.39.1) at 2024-11-01 15:17:02 +0000