diff options
Diffstat (limited to 'listener.c')
-rw-r--r-- | listener.c | 101 |
1 files changed, 101 insertions, 0 deletions
diff --git a/listener.c b/listener.c new file mode 100644 index 0000000..04e3030 --- /dev/null +++ b/listener.c @@ -0,0 +1,101 @@ +/* +** listener.c -- a datagram sockets "server" demo +*/ + +#include <stdio.h> +#include <stdlib.h> +#include <unistd.h> +#include <errno.h> +#include <string.h> +#include <sys/types.h> +#include <sys/socket.h> +#include <netinet/in.h> +#include <arpa/inet.h> +#include <netdb.h> + +// #define MYPORT "4950" // the port users will be connecting to + +#define MAXBUFLEN 16384 + +// get sockaddr, IPv4 or IPv6: +void *get_in_addr(struct sockaddr *sa) +{ + if (sa->sa_family == AF_INET) { + return &(((struct sockaddr_in*)sa)->sin_addr); + } + + return &(((struct sockaddr_in6*)sa)->sin6_addr); +} + +int main(int argc, char *argv[]) +{ + int sockfd; + struct addrinfo hints, *servinfo, *p; + int rv; + int numbytes; + struct sockaddr_storage their_addr; + socklen_t addr_len; + char s[INET6_ADDRSTRLEN]; + + if (argc != 2) { + fprintf(stderr,"<udp port>\n"); + exit(1); + } + + const char* udp_port = argv[1]; + + + memset(&hints, 0, sizeof hints); + hints.ai_family = AF_INET; // set to AF_INET to use IPv4 + hints.ai_socktype = SOCK_DGRAM; + hints.ai_flags = AI_PASSIVE; // use my IP + + if ((rv = getaddrinfo(NULL, udp_port, &hints, &servinfo)) != 0) { + fprintf(stderr, "getaddrinfo: %s\n", gai_strerror(rv)); + return 1; + } + + // loop through all the results and bind to the first we can + for(p = servinfo; p != NULL; p = p->ai_next) { + if ((sockfd = socket(p->ai_family, p->ai_socktype, + p->ai_protocol)) == -1) { + perror("listener: socket"); + continue; + } + + if (bind(sockfd, p->ai_addr, p->ai_addrlen) == -1) { + close(sockfd); + perror("listener: bind"); + continue; + } + + break; + } + + if (p == NULL) { + fprintf(stderr, "listener: failed to bind socket\n"); + return 2; + } + + freeaddrinfo(servinfo); + + int count = 0; + + char buf[MAXBUFLEN]; + while(1) { + addr_len = sizeof their_addr; + if ((numbytes = recvfrom(sockfd, buf, MAXBUFLEN , 0, + (struct sockaddr *)&their_addr, &addr_len)) == -1) { + perror("recvfrom"); + exit(1); + } + // print the buffer + write(STDOUT_FILENO, buf, numbytes); + + memset(buf, 0, MAXBUFLEN); + } + + close(sockfd); + + return 0; +} |