/* Opens a file descriptor whcich could be a FIFO or just a file. On error returns -1*/
int fd_open (int type, void * data, int perms) {
int fd = 0;
if(type == _FD_FIFO) {
int key = fifo_exists(data);
if(key) {
return fd_find(type, key);
}
FIFO_allocator_size++;
if(FIFO_allocator_size == FIFO_allocator_end - FIFO_allocator - 1) {
return -1;
}
fd = FIFO_allocator_index++;
if(FIFO_allocator_index == FIFO_allocator_end - 1) {
FIFO_allocator_index = FIFO_allocator;
}
} else if(type == _FD_FILE) {
fd_allocator_size++;
if(fd_allocator_size == fd_allocator_end - fd_allocator - 1) {
return -1;
}
fd = fd_allocator_index++;
if(fd_allocator_index == fd_allocator_end - 1) {
fd_allocator_index = fd_allocator;
}
} else {
return -1;
}
return fd_open_with_index(fd, type, data, perms);
}
scamper_fd_t *scamper_fd_dl(int ifindex)
{
scamper_fd_t *fdn = NULL, findme;
int fd = -1;
findme.type = SCAMPER_FD_TYPE_DL;
findme.fd_dl_ifindex = ifindex;
if((fdn = fd_find(&findme)) != NULL)
{
scamper_fd_read_unpause(fdn);
return fdn;
}
/*
* open the file descriptor for the ifindex, and then allocate a scamper_fd
* for the file descriptor
*/
if((fd = scamper_dl_open(ifindex)) == -1 ||
(fdn = fd_alloc(SCAMPER_FD_TYPE_DL, fd)) == NULL)
{
goto err;
}
/*
* record the ifindex for the file descriptor, and then allocate the state
* that is maintained with it
*/
fdn->fd_dl_ifindex = ifindex;
/*
* 1. add the file descriptor to the splay tree
* 2. allocate state for the datalink file descriptor
*/
if((fdn->fd_tree_node = splaytree_insert(fd_tree, fdn)) == NULL ||
(fdn->fd_list_node = dlist_tail_push(fd_list, fdn)) == NULL ||
(fdn->fd_dl_dl = scamper_dl_state_alloc(fdn)) == NULL)
{
goto err;
}
/* set the file descriptor up for reading */
fdn->read.cb = scamper_dl_read_cb;
fdn->read.param = fdn->fd_dl_dl;
fdn->write.cb = NULL;
fdn->write.param = NULL;
scamper_fd_read_unpause(fdn);
return fdn;
err:
if(fdn != NULL) free(fdn);
if(fd != -1) scamper_dl_close(fd);
return NULL;
}
static scamper_fd_t *fd_tcp(int type, void *addr, uint16_t sport)
{
scamper_fd_t *fdn, findme;
size_t len = 0;
int fd = -1;
findme.type = type;
findme.fd_tcp_addr = addr;
findme.fd_tcp_sport = sport;
if((fdn = fd_find(&findme)) != NULL)
{
return fdn;
}
if(type == SCAMPER_FD_TYPE_TCP4)
{
fd = scamper_tcp4_open(addr, sport);
len = sizeof(struct in_addr);
}
else if(type == SCAMPER_FD_TYPE_TCP6)
{
fd = scamper_tcp6_open(addr, sport);
len = sizeof(struct in6_addr);
}
if(fd == -1 || (fdn = fd_alloc(type, fd)) == NULL ||
(addr != NULL && (fdn->fd_tcp_addr = memdup(addr, len)) == NULL))
{
goto err;
}
fdn->fd_tcp_sport = sport;
if((fdn->fd_tree_node = splaytree_insert(fd_tree, fdn)) == NULL ||
(fdn->fd_list_node = dlist_tail_push(fd_list, fdn)) == NULL)
{
goto err;
}
return fdn;
err:
if(fd != -1)
{
if(type == SCAMPER_FD_TYPE_TCP4)
scamper_tcp4_close(fd);
else if(type == SCAMPER_FD_TYPE_TCP6)
scamper_tcp6_close(fd);
}
if(fdn != NULL) fd_free(fdn);
return NULL;
}
static scamper_fd_t *fd_null(int type)
{
scamper_fd_t *fdn = NULL, findme;
int fd = -1;
/* first check if a sharable fd exists for this type */
findme.type = type;
if((fdn = fd_find(&findme)) != NULL)
{
return fdn;
}
if(type == SCAMPER_FD_TYPE_RTSOCK) fd = scamper_rtsock_open();
else if(type == SCAMPER_FD_TYPE_IFSOCK) fd = socket(AF_INET, SOCK_DGRAM, 0);
else if(type == SCAMPER_FD_TYPE_IP4) fd = scamper_ip4_openraw();
if(fd == -1 || (fdn = fd_alloc(type, fd)) == NULL ||
(fdn->fd_tree_node = splaytree_insert(fd_tree, fdn)) == NULL ||
(fdn->fd_list_node = dlist_tail_push(fd_list, fdn)) == NULL)
{
goto err;
}
return fdn;
err:
if(fd != -1)
{
if(type == SCAMPER_FD_TYPE_RTSOCK)
scamper_rtsock_close(fd);
else if(type == SCAMPER_FD_TYPE_IFSOCK)
close(fd);
else if(type == SCAMPER_FD_TYPE_IP4)
scamper_ip4_close(fd);
}
if(fdn != NULL) fd_free(fdn);
return NULL;
}
static scamper_fd_t *fd_udp(int type, void *addr, uint16_t sport)
{
scamper_fd_t *fdn, findme;
size_t len = 0;
int fd = -1;
findme.type = type;
findme.fd_udp_addr = addr;
findme.fd_udp_sport = sport;
if((fdn = fd_find(&findme)) != NULL)
return fdn;
if(type == SCAMPER_FD_TYPE_UDP4)
{
fd = scamper_udp4_openraw(addr);
len = sizeof(struct in_addr);
}
else if(type == SCAMPER_FD_TYPE_UDP6)
{
fd = scamper_udp6_open(addr, sport);
len = sizeof(struct in6_addr);
}
else if(type == SCAMPER_FD_TYPE_UDP4DG)
{
fd = scamper_udp4_opendgram(addr, sport);
len = sizeof(struct in_addr);
}
if(fd == -1 || (fdn = fd_alloc(type, fd)) == NULL ||
(addr != NULL && (fdn->fd_udp_addr = memdup(addr, len)) == NULL))
{
printerror(errno, strerror, __func__, "could not open socket");
goto err;
}
fdn->fd_udp_sport = sport;
if((fdn->fd_tree_node = splaytree_insert(fd_tree, fdn)) == NULL)
{
printerror(errno, strerror, __func__, "could not add socket to tree");
goto err;
}
if((fdn->fd_list_node = dlist_tail_push(fd_list, fdn)) == NULL)
{
printerror(errno, strerror, __func__, "could not add socket to list");
goto err;
}
return fdn;
err:
if(fd != -1)
{
if(type == SCAMPER_FD_TYPE_UDP4 || type == SCAMPER_FD_TYPE_UDP4DG)
scamper_udp4_close(fd);
else if(type == SCAMPER_FD_TYPE_UDP6)
scamper_udp6_close(fd);
}
if(fdn != NULL) fd_free(fdn);
return NULL;
}
