Added basic template for server - accepting connnections, command line
arguments, etc
This commit is contained in:
		
							
								
								
									
										123
									
								
								src/wstationd.c
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										123
									
								
								src/wstationd.c
									
									
									
									
									
										Normal file
									
								
							@ -0,0 +1,123 @@
 | 
			
		||||
#include <stdio.h>
 | 
			
		||||
#include <unistd.h>
 | 
			
		||||
#include <errno.h>
 | 
			
		||||
#include <string.h>
 | 
			
		||||
#include <stdlib.h>
 | 
			
		||||
#include <limits.h>
 | 
			
		||||
#include <sys/types.h>
 | 
			
		||||
#include <sys/socket.h>
 | 
			
		||||
#include <netinet/in.h>
 | 
			
		||||
#include <arpa/inet.h>
 | 
			
		||||
#include <libgen.h>
 | 
			
		||||
#include <getopt.h>
 | 
			
		||||
#include "wstationd.h"
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
static char *exec_name;
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
static void print_help()
 | 
			
		||||
{
 | 
			
		||||
	fprintf(stdout, "usage: %s\n", exec_name);
 | 
			
		||||
	fprintf(stdout, "  -p, --port <port>     Port to listen on (default: 10800)\n");
 | 
			
		||||
	fprintf(stdout, "  -b, --bind <address>  Address to bind to (default: 0.0.0.0)\n");
 | 
			
		||||
	fprintf(stdout, "  -h, --help            Print this message and exit\n");
 | 
			
		||||
	fprintf(stdout, "  -v, --version         Print version and exit\n");
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
int main(int argc, char *argv[])
 | 
			
		||||
{
 | 
			
		||||
	const struct option longopts[] = {
 | 
			
		||||
		{ "port", required_argument, NULL, 'p' },
 | 
			
		||||
		{ "bind", required_argument, NULL, 'b' },
 | 
			
		||||
		{ "version", no_argument, NULL, 'v' },
 | 
			
		||||
		{ "help", no_argument, NULL, 'h' },
 | 
			
		||||
		{ NULL, 0, NULL, 0 }
 | 
			
		||||
	};
 | 
			
		||||
 | 
			
		||||
	exec_name = basename(argv[0]);
 | 
			
		||||
 | 
			
		||||
	int flag = 1;
 | 
			
		||||
	int sockfd = 0, connfd = 0;
 | 
			
		||||
	int port = 10800;
 | 
			
		||||
	uint32_t addr = INADDR_ANY;
 | 
			
		||||
 | 
			
		||||
	for(int ch; (ch = getopt_long(argc, argv, "b:p:vh", longopts, NULL)) != -1;){
 | 
			
		||||
		switch(ch){
 | 
			
		||||
			case 'p':
 | 
			
		||||
				port = (int)strtol(optarg, NULL, 10);
 | 
			
		||||
				if(port < 1 || port > 65535){
 | 
			
		||||
					fprintf(stderr, "%s: invalid port number\n", exec_name);
 | 
			
		||||
					print_help();
 | 
			
		||||
					goto shutdown_error;
 | 
			
		||||
				}
 | 
			
		||||
			break;
 | 
			
		||||
 | 
			
		||||
			case 'b':
 | 
			
		||||
				if(inet_pton(AF_INET, optarg, &addr) == -1){
 | 
			
		||||
					fprintf(stderr, "%s: invalid bind address\n", exec_name);
 | 
			
		||||
					print_help();
 | 
			
		||||
					goto shutdown_error;
 | 
			
		||||
				}
 | 
			
		||||
			break;
 | 
			
		||||
 | 
			
		||||
			case 'v':
 | 
			
		||||
				fprintf(stdout, "wstationd v%s (%s %s)\n", VERSION,__DATE__, __TIME__);
 | 
			
		||||
			goto shutdown_clean;
 | 
			
		||||
 | 
			
		||||
			case '?':
 | 
			
		||||
			case 'h':
 | 
			
		||||
				print_help();
 | 
			
		||||
			goto shutdown_clean;
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	sockfd = socket(AF_INET, SOCK_STREAM, 0);
 | 
			
		||||
	if(sockfd == -1){
 | 
			
		||||
		fprintf(stderr, "%s: cannot open socket - %s\n",
 | 
			
		||||
			exec_name, strerror(errno)
 | 
			
		||||
		);
 | 
			
		||||
		goto shutdown_error;
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	setsockopt(sockfd, SOL_SOCKET, SO_REUSEADDR, &flag, sizeof(flag));
 | 
			
		||||
 | 
			
		||||
	struct sockaddr_in sockin;
 | 
			
		||||
	memset(&sockin, 0, sizeof(sockin));
 | 
			
		||||
	sockin.sin_family = AF_INET;
 | 
			
		||||
	sockin.sin_addr.s_addr = addr;
 | 
			
		||||
	sockin.sin_port = htons(port);
 | 
			
		||||
 | 
			
		||||
	if(bind(sockfd, (struct sockaddr*)&sockin, sizeof(sockin)) < 0){
 | 
			
		||||
		fprintf(stderr, "%s: cannot bind to port - %s\n",
 | 
			
		||||
			exec_name, strerror(errno)
 | 
			
		||||
		);
 | 
			
		||||
		goto shutdown_error;
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	if(listen(sockfd, 10) != 0){
 | 
			
		||||
		fprintf(stderr, "%s: cannot listen - %s\n",
 | 
			
		||||
			exec_name, strerror(errno)
 | 
			
		||||
		);
 | 
			
		||||
		goto shutdown_error;
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	connfd = accept(sockfd, NULL, NULL);
 | 
			
		||||
	if(connfd == -1){
 | 
			
		||||
		fprintf(stderr, "%s: connection failed - %s\n",
 | 
			
		||||
			exec_name, strerror(errno)
 | 
			
		||||
		);
 | 
			
		||||
		goto shutdown_error;
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	shutdown_clean:
 | 
			
		||||
	if(connfd > 0) close(connfd);
 | 
			
		||||
	if(sockfd > 0) close(sockfd);
 | 
			
		||||
	return 0;
 | 
			
		||||
 | 
			
		||||
	shutdown_error:
 | 
			
		||||
	if(connfd > 0) close(connfd);
 | 
			
		||||
	if(sockfd > 0) close(sockfd);
 | 
			
		||||
	return 1;
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										1
									
								
								src/wstationd.h
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										1
									
								
								src/wstationd.h
									
									
									
									
									
										Normal file
									
								
							@ -0,0 +1 @@
 | 
			
		||||
#define VERSION "0.1.0"
 | 
			
		||||
		Reference in New Issue
	
	Block a user