TCP_IP_Stack/graph.c at main · treesturn/TCP_IP_Stack · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
#include "graph.h"
#include <stdlib.h>
#include <string.h>
#include <stdio.h>
#include <memory.h>

graph_t* create_new_graph(char* topology_name){
	graph_t* graph = calloc(1, sizeof(graph_t));
	strncpy(graph->topology_name, topology_name, 32);
	graph->topology_name[31] = '\0';

	init_glthread(&graph->node_list);
	return graph;
}

node_t* create_graph_node(graph_t* graph, char* node_name){
	node_t* node = calloc(1, sizeof(node_t));
	strncpy(node->node_name, node_name, NODE_NAME_SIZE);
	node->node_name[NODE_NAME_SIZE] = '\0';

	init_udp_socket(node);

	init_node_nw_prop(&node->node_nw_prop);
	init_glthread(&node->graph_glue);
	glthread_add_next(&graph->node_list, &node->graph_glue);
	return node;
}

void insert_link_between_two_nodes(
		node_t* node1,
		node_t* node2,
		char* from_if_name,
		char* to_if_name,
		unsigned int cost){

	link_t* link = calloc(1, sizeof(link_t));

	strncpy(link->intf1.if_name, from_if_name, IF_NAME_SIZE);
	link->intf1.if_name[IF_NAME_SIZE] = '\0';
	strncpy(link->intf2.if_name, to_if_name, IF_NAME_SIZE);
	link->intf2.if_name[IF_NAME_SIZE] = '\0';

	link->intf1.link = link;
	link->intf2.link = link;

	link->intf1.att_node = node1;
	link->intf2.att_node = node2;
	link->cost = cost;

	int empty_intf_slot;

	empty_intf_slot = get_node_intf_available_slot(node1);
	node1->intf[empty_intf_slot] = &link->intf1;

	empty_intf_slot = get_node_intf_available_slot(node2);
	node2->intf[empty_intf_slot] = &link->intf2;

	init_intf_nw_prop(&link->intf1.intf_nw_props);
	init_intf_nw_prop(&link->intf2.intf_nw_props);

	//Assign random generated MAC address to the interface
	interface_assign_mac_address(&link->intf1);
	interface_assign_mac_address(&link->intf2);
}

void dump_graph(graph_t* graph){
	node_t* node;
	glthread_t* curr;

	printf("Topology name: %s\n", graph->topology_name);

	ITERATE_GLTHREAD_BEGIN(&graph->node_list, curr){
		node = graph_glue_to_node(curr);
		dump_node(node);
	} ITERATE_GLTHREAD_END(&graph->node_list, curr);
}

void dump_node(node_t* node){
	unsigned int i = 0;
	interface_t* intf;

	printf("Node Name: %s\n", node->node_name);
	for(; i < MAX_INTF_PER_NODE; i++){
		intf = node->intf[i];
		if(!intf){
			break;
		}
		dump_interface(intf);
	}

}

void dump_interface(interface_t* interface){
	link_t* link = interface->link;
	node_t* nbr_node = get_nbr_node(interface);

	printf("Interface Name: %s\n\tNeighbour Node: %s, Local Node: %s, cost: %u\n",
			interface->if_name,
			nbr_node->node_name,
			interface->att_node->node_name,
			link->cost);
}