3 files changed, 61 insertions, 6 deletions

diff --git a/src/ipcpd/normal/pff.c b/src/ipcpd/normal/pff.c
index 8cab7936..77c2c551 100644
--- a/src/ipcpd/normal/pff.c
+++ b/src/ipcpd/normal/pff.c
@@ -68,6 +68,16 @@ void pff_destroy(struct pff * instance)
         free(instance);
 }
 
+void pff_lock(struct pff * instance)
+{
+        pthread_mutex_lock(&instance->lock);
+}
+
+void pff_unlock(struct pff * instance)
+{
+        pthread_mutex_unlock(&instance->lock);
+}
+
 int pff_add(struct pff * instance, uint64_t addr, int fd)
 {
         int * val;
@@ -132,6 +142,15 @@ int pff_remove(struct pff * instance, uint64_t addr)
         return 0;
 }
 
+void pff_flush(struct pff * instance)
+{
+        assert(instance);
+
+        pthread_mutex_lock(&instance->lock);
+        htable_flush(instance->table);
+        pthread_mutex_unlock(&instance->lock);
+}
+
 int pff_nhop(struct pff * instance, uint64_t addr)
 {
         int * j;
diff --git a/src/ipcpd/normal/pff.h b/src/ipcpd/normal/pff.h
index 667c341e..7d8ce469 100644
--- a/src/ipcpd/normal/pff.h
+++ b/src/ipcpd/normal/pff.h
@@ -34,6 +34,10 @@ struct pff * pff_create(void);
 
 void         pff_destroy(struct pff * instance);
 
+void         pff_lock(struct pff * instance);
+
+void         pff_unlock(struct pff * instance);
+
 int          pff_add(struct pff * instance,
                      uint64_t     addr,
                      int          fd);
@@ -45,6 +49,8 @@ int          pff_update(struct pff * instance,
 int          pff_remove(struct pff * instance,
                         uint64_t     addr);
 
+void         pff_flush(struct pff * instance);
+
 /* Returns fd towards next hop */
 int          pff_nhop(struct pff * instance,
                       uint64_t     addr);
diff --git a/src/ipcpd/normal/routing.c b/src/ipcpd/normal/routing.c
index f8705991..b750ca84 100644
--- a/src/ipcpd/normal/routing.c
+++ b/src/ipcpd/normal/routing.c
@@ -35,6 +35,7 @@
 #include "ribconfig.h"
 #include "ipcp.h"
 #include "graph.h"
+#include "neighbors.h"
 
 #include <assert.h>
 #include <stdlib.h>
@@ -63,14 +64,32 @@ struct {
         pthread_t          rib_listener;
 } routing;
 
+/* Take under neighbors lock */
+static int addr_to_fd(uint64_t addr)
+{
+        struct list_head * p = NULL;
+
+        list_for_each(p, &routing.nbs->list) {
+                struct nb * e = list_entry(p, struct nb, next);
+                if (e->conn.conn_info.addr == addr)
+                        return e->conn.flow_info.fd;
+        }
+
+        return -1;
+}
+
 static void * calculate_pff(void * o)
 {
+        struct routing_i *      instance;
         struct routing_table ** table;
         ssize_t                 n_table;
+        int                     i;
+        int                     fd;
 
-        (void) o;
+        instance = (struct routing_i *) o;
 
         while (true) {
+                table = NULL;
                 n_table = graph_routing_table(routing.graph,
                                               ipcpi.dt_addr, &table);
                 if (table == NULL) {
@@ -78,10 +97,21 @@ static void * calculate_pff(void * o)
                         continue;
                 }
 
-                /*
-                 * FIXME: Calculate address to fd here
-                 * and fill in PFF
-                 */
+                pthread_mutex_lock(&routing.nbs->list_lock);
+                pff_lock(instance->pff);
+
+                pff_flush(instance->pff);
+
+                for (i = 0; i < n_table; i++) {
+                        fd = addr_to_fd(table[i]->nhop);
+                        if (fd == -1)
+                                continue;
+
+                        pff_add(instance->pff, table[i]->dst, fd);
+                }
+
+                pff_unlock(instance->pff);
+                pthread_mutex_unlock(&routing.nbs->list_lock);
 
                 freepp(struct routing_table, table, n_table);
                 sleep(RECALC_TIME);
@@ -102,7 +132,7 @@ struct routing_i * routing_i_create(struct pff * pff)
 
         tmp->pff = pff;
 
-        pthread_create(&tmp->calculator, NULL, calculate_pff, NULL);
+        pthread_create(&tmp->calculator, NULL, calculate_pff, (void *) tmp);
 
         return tmp;
 }