summaryrefslogtreecommitdiff
path: root/src/irmd/main.c
diff options
context:
space:
mode:
authorSander Vrijders <sander.vrijders@intec.ugent.be>2016-08-10 13:07:59 +0200
committerSander Vrijders <sander.vrijders@intec.ugent.be>2016-08-10 13:07:59 +0200
commit26b20744a441705accbe550aa0b996f8601a9404 (patch)
tree9b33b5d1888cd4fb338b57b7b1aa90349668eb8b /src/irmd/main.c
parente5cb8042c9f0c6374c4f14bf7ff3f4fe3cdcd155 (diff)
parent9740c4bf2966d57d6f7e0d034a862211cb3814c4 (diff)
downloadouroboros-26b20744a441705accbe550aa0b996f8601a9404.tar.gz
ouroboros-26b20744a441705accbe550aa0b996f8601a9404.zip
Merged in dstaesse/ouroboros/be-irmd-cleanup (pull request #201)
irmd: Graceful shutdown
Diffstat (limited to 'src/irmd/main.c')
-rw-r--r--src/irmd/main.c94
1 files changed, 43 insertions, 51 deletions
diff --git a/src/irmd/main.c b/src/irmd/main.c
index d0ba969f..e36fb98e 100644
--- a/src/irmd/main.c
+++ b/src/irmd/main.c
@@ -70,8 +70,7 @@ struct ipcp_entry {
enum irm_state {
IRMD_NULL = 0,
- IRMD_RUNNING,
- IRMD_SHUTDOWN
+ IRMD_RUNNING
};
struct spawned_api {
@@ -760,7 +759,14 @@ static struct irm_flow * flow_accept(pid_t api,
pthread_rwlock_unlock(&irmd->reg_lock);
pthread_rwlock_unlock(&irmd->state_lock);
- reg_api_sleep(rgi);
+ while (reg_api_sleep(rgi) == -ETIMEDOUT) {
+ pthread_rwlock_rdlock(&irmd->state_lock);
+ if (irmd->state != IRMD_RUNNING) {
+ pthread_rwlock_unlock(&irmd->state_lock);
+ break;
+ }
+ pthread_rwlock_unlock(&irmd->state_lock);
+ }
pthread_rwlock_rdlock(&irmd->state_lock);
pthread_rwlock_rdlock(&irmd->reg_lock);
@@ -929,15 +935,6 @@ static struct irm_flow * flow_alloc(pid_t api,
return f;
}
-static void cleanup_alloc_res(void * o)
-{
- struct irm_flow * f = (struct irm_flow *) o;
- if (f->state == FLOW_PENDING)
- f->state = FLOW_NULL;
- pthread_cond_broadcast(&f->state_cond);
- pthread_mutex_unlock(&f->state_lock);
-}
-
static int flow_alloc_res(int port_id)
{
struct irm_flow * f;
@@ -975,12 +972,11 @@ static int flow_alloc_res(int port_id)
pthread_rwlock_unlock(&irmd->state_lock);
pthread_mutex_lock(&f->state_lock);
- pthread_cleanup_push(cleanup_alloc_res, (void *) f);
while (f->state == FLOW_PENDING)
pthread_cond_wait(&f->state_cond, &f->state_lock);
- pthread_cleanup_pop(true);
+ pthread_mutex_unlock(&f->state_lock);
pthread_rwlock_rdlock(&irmd->state_lock);
pthread_rwlock_wrlock(&irmd->flows_lock);
@@ -1145,13 +1141,11 @@ static struct irm_flow * flow_req_arr(pid_t api,
pthread_rwlock_unlock(&irmd->state_lock);
pthread_mutex_lock(&rne->state_lock);
- pthread_cleanup_push((void (*)(void *)) pthread_mutex_unlock,
- (void *) &rne->state_lock);
while (rne->state == REG_NAME_AUTO_EXEC)
pthread_cond_wait(&rne->state_cond, &rne->state_lock);
- pthread_cleanup_pop(true);
+ pthread_mutex_unlock(&rne->state_lock);
pthread_rwlock_rdlock(&irmd->state_lock);
pthread_rwlock_rdlock(&irmd->reg_lock);
@@ -1207,13 +1201,11 @@ static struct irm_flow * flow_req_arr(pid_t api,
reg_api_wake(rgi);
pthread_mutex_lock(&rne->state_lock);
- pthread_cleanup_push((void (*)(void *)) pthread_mutex_unlock,
- (void *) &rne->state_lock);
while (rne->state == REG_NAME_FLOW_ARRIVED)
pthread_cond_wait(&rne->state_cond, &rne->state_lock);
- pthread_cleanup_pop(true);
+ pthread_mutex_unlock(&rne->state_lock);
return f;
}
@@ -1346,26 +1338,17 @@ static void irm_destroy()
void irmd_sig_handler(int sig, siginfo_t * info, void * c)
{
- int i;
-
switch(sig) {
case SIGINT:
case SIGTERM:
case SIGHUP:
+ LOG_INFO("IRMd shutting down...");
+
pthread_rwlock_wrlock(&irmd->state_lock);
irmd->state = IRMD_NULL;
pthread_rwlock_unlock(&irmd->state_lock);
-
- if (irmd->threadpool != NULL) {
- for (i = 0; i < IRMD_THREADPOOL_SIZE; i++)
- pthread_cancel(irmd->threadpool[i]);
-
- }
-
- pthread_cancel(irmd->shm_sanitize);
- pthread_cancel(irmd->cleanup_flows);
break;
case SIGPIPE:
LOG_DBG("Ignored SIGPIPE.");
@@ -1393,7 +1376,7 @@ void * irm_flow_cleaner()
pthread_rwlock_rdlock(&irmd->state_lock);
- if (irmd->state == IRMD_NULL) {
+ if (irmd->state != IRMD_RUNNING) {
pthread_rwlock_unlock(&irmd->state_lock);
return (void *) 0;
}
@@ -1482,16 +1465,6 @@ void * irm_flow_cleaner()
}
}
-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];
@@ -1505,13 +1478,18 @@ void * mainloop()
struct irm_flow * e = NULL;
pid_t * apis = NULL;
+ pthread_rwlock_rdlock(&irmd->state_lock);
+ if (irmd->state != IRMD_RUNNING) {
+ pthread_rwlock_unlock(&irmd->state_lock);
+ break;
+ }
+ pthread_rwlock_unlock(&irmd->state_lock);
+
ret_msg.code = IRM_MSG_CODE__IRM_REPLY;
cli_sockfd = accept(irmd->sockfd, 0, 0);
- if (cli_sockfd < 0) {
- LOG_ERR("Cannot accept new connection.");
+ if (cli_sockfd < 0)
continue;
- }
count = read(cli_sockfd, buf, IRM_MSG_BUF_SIZE);
if (count <= 0) {
@@ -1526,9 +1504,6 @@ void * mainloop()
continue;
}
- pthread_cleanup_push(close_ptr, &cli_sockfd);
- pthread_cleanup_push(clean_msg, (void *) msg);
-
switch (msg->code) {
case IRM_MSG_CODE__IRM_CREATE_IPCP:
ret_msg.has_result = true;
@@ -1660,8 +1635,7 @@ void * mainloop()
break;
}
- pthread_cleanup_pop(true);
- pthread_cleanup_pop(false);
+ irm_msg__free_unpacked(msg, NULL);
buffer.len = irm_msg__get_packed_size(&ret_msg);
if (buffer.len == 0) {
@@ -1691,11 +1665,15 @@ void * mainloop()
free(buffer.data);
close(cli_sockfd);
}
+
+ return (void *) 0;
}
static struct irm * irm_create()
{
struct stat st = {0};
+ struct timeval timeout = {(IRMD_ACCEPT_TIMEOUT / 1000),
+ (IRMD_ACCEPT_TIMEOUT % 1000) * 1000};
irmd = malloc(sizeof(*irmd));
if (irmd == NULL)
@@ -1752,6 +1730,16 @@ static struct irm * irm_create()
return NULL;
}
+ if (setsockopt(irmd->sockfd,
+ SOL_SOCKET,
+ SO_RCVTIMEO,
+ (char *) &timeout,
+ sizeof(timeout)) < 0) {
+ LOG_ERR("Failed setting socket option.");
+ irm_destroy();
+ return NULL;
+ }
+
if (chmod(IRM_SOCK_PATH, 0666)) {
LOG_ERR("Failed to chmod socket.");
irm_destroy();
@@ -1905,12 +1893,16 @@ int main(int argc, char ** argv)
for (t = 0; t < IRMD_THREADPOOL_SIZE; ++t)
pthread_join(irmd->threadpool[t], NULL);
- pthread_join(irmd->shm_sanitize, NULL);
pthread_join(irmd->cleanup_flows, NULL);
+ pthread_cancel(irmd->shm_sanitize);
+ pthread_join(irmd->shm_sanitize, NULL);
+
irm_destroy();
close_logfile();
+ LOG_INFO("Bye.");
+
exit(EXIT_SUCCESS);
}