lib, irmd, ipcpd: Flow deallocation over shim UDP

The shim UDP now supports deallocating a flow end-to-end. Contains
some stability fixes for flow allocation and some missing close()
calls in lib/sockets.

1 files changed, 26 insertions, 0 deletions

diff --git a/src/lib/dev.c b/src/lib/dev.c
index 63cf92f7..95fca1ec 100644
--- a/src/lib/dev.c
+++ b/src/lib/dev.c
@@ -154,6 +154,26 @@ static int port_id_to_fd(int port_id)
 }
 #endif
 
+static void clean_fds()
+{
+        int i;
+        for (i = 0; i < AP_MAX_FLOWS; ++i) {
+                if (flow_alloc_res(i) < 0) {
+                        rw_lock_wrlock(&_ap_instance->flows_lock);
+                        if (_ap_instance->flows[i].port_id < 0) {
+                                rw_lock_unlock(&_ap_instance->flows_lock);
+                                continue;
+                        }
+
+                        bmp_release(_ap_instance->fds, i);
+                        _ap_instance->flows[i].port_id = -1;
+                        shm_ap_rbuff_close(_ap_instance->flows[i].rb);
+                        _ap_instance->flows[i].rb = NULL;
+                        rw_lock_unlock(&_ap_instance->flows_lock);
+                }
+        }
+}
+
 int ap_reg(char ** difs,
            size_t  len)
 {
@@ -202,6 +222,7 @@ int ap_reg(char ** difs,
         rw_lock_wrlock(&_ap_instance->flows_lock);
 
         fd = bmp_allocate(_ap_instance->fds);
+        _ap_instance->flows[fd].port_id = -1;
 
         rw_lock_unlock(&_ap_instance->flows_lock);
         rw_lock_unlock(&_ap_instance->data_lock);
@@ -268,6 +289,8 @@ int flow_accept(int     fd,
 
         msg.pid     = _ap_instance->api->id;
 
+        clean_fds();
+
         rw_lock_unlock(&_ap_instance->data_lock);
 
         recv_msg = send_recv_irm_msg(&msg);
@@ -389,6 +412,8 @@ int flow_alloc(char * dst_name,
         msg.pid         = _ap_instance->api->id;
         msg.ap_name     = _ap_instance->api->name;
 
+        clean_fds();
+
         rw_lock_unlock(&_ap_instance->data_lock);
 
         recv_msg = send_recv_irm_msg(&msg);
@@ -402,6 +427,7 @@ int flow_alloc(char * dst_name,
         }
 
         rw_lock_rdlock(&_ap_instance->data_lock);
+
         rw_lock_wrlock(&_ap_instance->flows_lock);
 
         fd = bmp_allocate(_ap_instance->fds);