Added basic template for server - accepting connnections, command line
arguments, etc
This commit is contained in:
parent
76a7548409
commit
8189722dce
21
Makefile
Normal file
21
Makefile
Normal file
@ -0,0 +1,21 @@
|
||||
RM=rm -f
|
||||
CFLAGS=
|
||||
LDFLAGS=
|
||||
LIBS=
|
||||
TARGET=wstationd
|
||||
SRC= \
|
||||
src/wstationd.c
|
||||
OBJ=$(SRC:.c=.o)
|
||||
|
||||
.PHONY: clean all
|
||||
|
||||
.c.o:
|
||||
$(CC) -Wall -std=c99 $(CFLAGS) -c $< -o $@
|
||||
|
||||
all: $(TARGET)
|
||||
|
||||
$(TARGET): $(OBJ)
|
||||
$(CC) $(LDFLAGS) $(OBJ) $(LIBS) -o $@
|
||||
|
||||
clean:
|
||||
$(RM) $(OBJ) $(TARGET)
|
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"
|
Loading…
Reference in New Issue
Block a user