nortega/neocomm
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Implemented node removal function.

Browse Source
This commit is contained in:
Nicolás Ortega Froysa 2017-08-05 17:28:31 -05:00
parent 5a53d3681d
commit e4de99086a
No known key found for this signature in database
GPG Key ID: FEC70E3BAE2E69BF
1 changed files with 43 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

62
src/connectivity.c
View File

@ -27,9 +27,9 @@
#include <sys/socket.h>
#include <netdb.h>
#include <unistd.h>
#include <string.h>
#include <strings.h>
#include <stdio.h>
#include <stdlib.h>
static struct NeoComm_directory_node *node_list = NULL;
@ -240,43 +240,67 @@ unsigned int NeoComm_get_num_nodes() {
}
int NeoComm_add_node(const struct NeoComm_address addr) {
struct hostent *host = NULL;
host = gethostbyname(addr.address);
if(num_nodes == max_nodes)
{
NeoComm_error("Directory full");
return 0;
}
struct hostent *host = gethostbyname(addr.address);
if(!host)
{
NeoComm_error("Error getting host");
return 0;
}
struct NeoComm_directory_node *new_node;
for(unsigned int i = 0; i < max_nodes; ++i)
unsigned int i;
for(i = 0; i < max_nodes; ++i)
{
if(!node_list[i].in_use)
new_node = &node_list[i];
}
if(!new_node)
{
NeoComm_error("No more unused slots in directory");
return 0;
break;
}
bzero((char*) &new_node->addr, sizeof(new_node->addr));
new_node->addr.sin_family = AF_INET;
bzero((char*) &node_list[i].addr, sizeof(node_list[i].addr));
node_list[i].addr.sin_family = AF_INET;
bcopy((char*) host->h_addr_list[0],
(char*) &new_node->addr.sin_addr.s_addr,
(char*) &node_list[i].addr.sin_addr.s_addr,
host->h_length);
new_node->addr.sin_port = htons(addr.port);
node_list[i].addr.sin_port = htons(addr.port);
if(connect(new_node->sockfd,
(struct sockaddr*) &new_node->addr,
sizeof(new_node->addr)) < 0)
if(connect(node_list[i].sockfd,
(struct sockaddr*) &node_list[i].addr,
sizeof(node_list[i].addr)) < 0)
{
NeoComm_error("Failed to connect to new node");
return 0;
}
new_node->in_use = 1;
node_list[i].in_use = 1;
++num_nodes;
// TODO: listen to node on new thread
return 1;
}
void NeoComm_remove_node(const struct NeoComm_address addr) {
struct hostent *host = gethostbyname(addr.address);
if(!host)
return;
unsigned int i;
for(i = 0; i < max_nodes; ++i)
{
if(strncmp((char*) host->h_addr_list[0],
(char*) &node_list[i].addr.sin_addr.s_addr,
host->h_length) == 0)
break;
if(i == max_nodes - 1)
++i;
}
if(i == max_nodes)
return;
pthread_cancel(node_list[i].thread);
close(node_list[i].sockfd);
node_list[i].in_use = 0;
--num_nodes;
}