diff options
-rw-r--r-- | src/ipcpd/normal/pol/graph.c | 64 |
1 files changed, 26 insertions, 38 deletions
diff --git a/src/ipcpd/normal/pol/graph.c b/src/ipcpd/normal/pol/graph.c index 80a7de6f..422977cd 100644 --- a/src/ipcpd/normal/pol/graph.c +++ b/src/ipcpd/normal/pol/graph.c @@ -119,6 +119,7 @@ static struct vertex * add_vertex(struct graph * graph, list_head_init(&vertex->edges); vertex->addr = addr; + /* Keep them ordered on address. */ list_for_each(p, &graph->vertices) { struct vertex * v = list_entry(p, struct vertex, next); if (v->addr > addr) @@ -319,59 +320,48 @@ int graph_del_edge(struct graph * graph, return 0; } -static int get_min_vertex(struct vertex ** vertices, - int nr_vertices, +static int get_min_vertex(struct graph * graph, int * dist, + bool * used, struct vertex ** v) { - int min = INT_MAX; - int index = -1; - int i; + int min = INT_MAX; + int index = -1; + int i = 0; + struct list_head * p = NULL; *v = NULL; - for (i = 0; i < nr_vertices; i++) { - if (vertices[i] == NULL) + list_for_each(p, &graph->vertices) { + if (used[i] == true) { + i++; continue; + } if (dist[i] < min) { - *v = vertices[i]; min = dist[i]; index = i; + *v = list_entry(p, struct vertex, next); } + + i++; } if (index != -1) - vertices[index] = NULL; + used[index] = true; return index; } -static int get_vertex_number(struct vertex ** vertices, - int nr_vertices, - struct vertex * v) +static int get_vertex_number(struct graph * graph, + struct vertex * v) { - int i; - - for (i = 0; i < nr_vertices; i++) { - if (vertices[i] == v) - return i; - } - - return -1; -} - -static int get_vertex_index(struct graph * graph, - struct vertex * v) - -{ - struct list_head * p = NULL; - struct vertex * vertex; int i = 0; + struct list_head * p = NULL; list_for_each(p, &graph->vertices) { - vertex = list_entry(p, struct vertex, next); + struct vertex * vertex = list_entry(p, struct vertex, next); if (vertex == v) return i; i++; @@ -384,7 +374,7 @@ static struct vertex ** dijkstra(struct graph * graph, uint64_t src) { int dist[graph->nr_vertices]; - struct vertex * vertices[graph->nr_vertices]; + bool used[graph->nr_vertices]; struct list_head * p = NULL; int i = 0; int j = 0; @@ -400,24 +390,22 @@ static struct vertex ** dijkstra(struct graph * graph, /* Init the data structures */ list_for_each(p, &graph->vertices) { v = list_entry(p, struct vertex, next); - vertices[i] = v; if (v->addr == src) dist[i] = 0; else dist[i] = INT_MAX; prev[i] = NULL; + used[i] = false; i++; } /* Perform actual Dijkstra */ - i = get_min_vertex(vertices, graph->nr_vertices, dist, &v); + i = get_min_vertex(graph, dist, used, &v); while (v != NULL) { list_for_each(p, &v->edges) { e = list_entry(p, struct edge, next); - j = get_vertex_number(vertices, - graph->nr_vertices, - e->nb); + j = get_vertex_number(graph, e->nb); if (j == -1) continue; @@ -432,7 +420,7 @@ static struct vertex ** dijkstra(struct graph * graph, prev[j] = v; } } - i = get_min_vertex(vertices, graph->nr_vertices, dist, &v); + i = get_min_vertex(graph, dist, used, &v); } return prev; @@ -512,11 +500,11 @@ int graph_routing_table(struct graph * graph, continue; } - index = get_vertex_index(graph, prev); + index = get_vertex_number(graph, prev); while (prevs[index] != NULL) { nhop = prev; prev = prevs[index]; - index = get_vertex_index(graph, prev); + index = get_vertex_number(graph, prev); } t = malloc(sizeof(*t)); |