mirror of https://github.com/pvnis/srsRAN_4G.git
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
137 lines
3.5 KiB
C
137 lines
3.5 KiB
C
/**
|
|
*
|
|
* \section COPYRIGHT
|
|
*
|
|
* Copyright 2013-2015 Software Radio Systems Limited
|
|
*
|
|
* \section LICENSE
|
|
*
|
|
* This file is part of the srsLTE library.
|
|
*
|
|
* srsLTE is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU Affero General Public License as
|
|
* published by the Free Software Foundation, either version 3 of
|
|
* the License, or (at your option) any later version.
|
|
*
|
|
* srsLTE is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU Affero General Public License for more details.
|
|
*
|
|
* A copy of the GNU Affero General Public License can be found in
|
|
* the LICENSE file in the top-level directory of this distribution
|
|
* and at http://www.gnu.org/licenses/.
|
|
*
|
|
*/
|
|
|
|
#include <sys/socket.h>
|
|
#include <netinet/in.h>
|
|
#include <arpa/inet.h>
|
|
#include <unistd.h>
|
|
#include <fcntl.h>
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <strings.h>
|
|
#include <errno.h>
|
|
|
|
#include "srslte/io/netsource.h"
|
|
|
|
int srslte_netsource_init(srslte_netsource_t *q, const char *address, uint16_t port, srslte_netsource_type_t type) {
|
|
bzero(q, sizeof(srslte_netsource_t));
|
|
|
|
q->sockfd=socket(AF_INET,type==SRSLTE_NETSOURCE_TCP?SOCK_STREAM:SOCK_DGRAM,0);
|
|
|
|
if (q->sockfd < 0) {
|
|
perror("socket");
|
|
return -1;
|
|
}
|
|
|
|
// Make sockets reusable
|
|
int enable = 1;
|
|
#if defined (SO_REUSEADDR)
|
|
if (setsockopt(q->sockfd, SOL_SOCKET, SO_REUSEADDR, &enable, sizeof(int)) < 0)
|
|
perror("setsockopt(SO_REUSEADDR) failed");
|
|
#endif
|
|
#if defined (SO_REUSEPORT)
|
|
if (setsockopt(q->sockfd, SOL_SOCKET, SO_REUSEPORT, &enable, sizeof(int)) < 0)
|
|
perror("setsockopt(SO_REUSEPORT) failed");
|
|
#endif
|
|
q->type = type;
|
|
|
|
q->servaddr.sin_family = AF_INET;
|
|
q->servaddr.sin_addr.s_addr=inet_addr(address);
|
|
q->servaddr.sin_port=htons(port);
|
|
|
|
if (bind(q->sockfd,(struct sockaddr *)&q->servaddr,sizeof(struct sockaddr_in))) {
|
|
perror("bind");
|
|
return -1;
|
|
}
|
|
q->connfd = 0;
|
|
|
|
return 0;
|
|
}
|
|
|
|
void srslte_netsource_free(srslte_netsource_t *q) {
|
|
if (q->sockfd) {
|
|
close(q->sockfd);
|
|
}
|
|
bzero(q, sizeof(srslte_netsource_t));
|
|
}
|
|
|
|
int srslte_netsource_read(srslte_netsource_t *q, void *buffer, int nbytes) {
|
|
if (q->type == SRSLTE_NETSOURCE_UDP) {
|
|
int n = recv(q->sockfd, buffer, nbytes, 0);
|
|
|
|
if (n == -1) {
|
|
if (errno == EAGAIN) {
|
|
return 0;
|
|
} else {
|
|
return -1;
|
|
}
|
|
} else {
|
|
return n;
|
|
}
|
|
} else {
|
|
if (q->connfd == 0) {
|
|
printf("Waiting for TCP connection\n");
|
|
listen(q->sockfd, 1);
|
|
socklen_t clilen = sizeof(q->cliaddr);
|
|
q->connfd = accept(q->sockfd, (struct sockaddr *)&q->cliaddr, &clilen);
|
|
if (q->connfd < 0) {
|
|
perror("accept");
|
|
return -1;
|
|
}
|
|
}
|
|
int n = read(q->connfd, buffer, nbytes);
|
|
if (n == 0) {
|
|
printf("Connection closed\n");
|
|
close(q->connfd);
|
|
q->connfd = 0;
|
|
return 0;
|
|
}
|
|
if (n == -1) {
|
|
perror("read");
|
|
}
|
|
return n;
|
|
}
|
|
}
|
|
|
|
int srslte_netsource_set_nonblocking(srslte_netsource_t *q) {
|
|
if (fcntl(q->sockfd, F_SETFL, O_NONBLOCK)) {
|
|
perror("fcntl");
|
|
return -1;
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
int srslte_netsource_set_timeout(srslte_netsource_t *q, uint32_t microseconds) {
|
|
struct timeval t;
|
|
t.tv_sec = 0;
|
|
t.tv_usec = microseconds;
|
|
if (setsockopt(q->sockfd, SOL_SOCKET, SO_RCVTIMEO, &t, sizeof(struct timeval))) {
|
|
perror("setsockopt");
|
|
return -1;
|
|
}
|
|
return 0;
|
|
}
|