/* depend: */
/* cflags: */
-/* linker: color.o debug.o */
+/* linker: color.o debug.o http.o server.o signal.o */
+#include <assert.h>
#include <stdio.h>
#include <stdlib.h>
+#include <unistd.h>
+#include <signal.h>
#include "debug.h"
+#include "http.h"
+#include "server.h"
+#include "signal.h"
+
+/* types */
/* constants */
/* gobal variables */
char *progname = NULL;
+int port = 8080;
+socket_t sock = INVALID_SOCKET;
+socket_t conn = INVALID_SOCKET;
/* help function */
FILE *fid = ret ? stderr : stdout;
fprintf (fid, "usage: %s\n", progname);
fprintf (fid, " -h : help message\n");
+ fprintf (fid, " -p : port number (%d)\n", port);
fprintf (fid, " -v : verbose level (%d)\n", verbose);
return ret;
}
+void stop_server (__attribute__((unused)) int sig)
+{
+ if (conn != INVALID_SOCKET) {
+ close_socket (conn);
+ }
+ if (sock != INVALID_SOCKET) {
+ close_socket (sock);
+ }
+ terminate_network_context ();
+ exit (0);
+}
+
/* main function */
int main (int argc, char *argv[])
{
int i = 0;
- int ret = 0;
/* program name */
while (argc-- > 1) {
char *arg = *(++argv);
if (arg[0] != '-') {
- VERBOSE (ERROR, fprintf (stderr, "%s: invalid option -- '%s'\n", progname, arg); usage (1));
+ VERBOSE (ERROR, PERROR ("%s: invalid option -- '%s'\n", progname, arg); usage (1));
return 1;
}
char c = arg[1];
switch (c) {
-/*
- case 'i':
+ case 'p':
arg = (arg[2]) ? arg + 2 : (--argc > 0) ? *(++argv) : NULL;
if (arg == NULL) {
- VERBOSE (ERROR, fprintf (stderr, "%s: missing input prompt\n", progname); usage (1));
+ VERBOSE (ERROR, PERROR ("%s: missing port number\n", progname); usage (1));
+ return 1;
+ }
+ port = atoi (arg);
+ if (port <= 0) {
+ VERBOSE (ERROR, PERROR ("%s: incorrect port number (%s)\n", progname, arg); usage (1));
return 1;
}
- iprompt = arg;
break;
-*/
case 'v':
arg = (arg[2]) ? arg + 2 : (--argc > 0) ? *(++argv) : NULL;
if (arg == NULL) {
- VERBOSE (ERROR, fprintf (stderr, "%s: missing verbose level\n", progname); usage (1));
+ VERBOSE (ERROR, PERROR ("%s: missing verbose level\n", progname); usage (1));
return 1;
}
verbose = atoi (arg);
}
}
- return ret;
+ VERBOSE (DEBUG, PRINT ("Initializing socket\n"));
+ init_network_context ();
+ sock = open_listening_socket (port);
+ if (sock == INVALID_SOCKET) {
+ VERBOSE (ERROR, PERROR ("Can't open listening socket\n"));
+ return 1;
+ }
+ VERBOSE (INFO, PRINT ("Listening socket on port %d\n", port));
+
+ signal (SIGTERM, &stop_server);
+
+ /* main loop */
+ while (1) {
+ conn = accept_incoming_connection (sock);
+ if (conn == INVALID_SOCKET) {
+ usleep (1e5);
+ continue;
+ }
+
+ VERBOSE (DEBUG, PRINT ("Server connected, waiting for data\n"));
+
+ unsigned char *data = {0};
+
+ int len = receive_data (conn, &data);
+ if (len == 0) {
+ VERBOSE (WARNING, PRINT ("Connection closed by peer (rx)\n"));
+ } else if (len < 0) {
+ VERBOSE (WARNING, PRINT ("Connection in error (rx)\n"));
+ } else {
+ VERBOSE (DEBUG, PRINT ("Received %d bytes\n", len));
+
+ // processing
+ VERBOSE (DEBUG, PRINT ("Processing %s\n", data));
+ len = processing (data, len, &data);
+
+ int rc = send_data (conn, data, len);
+ if (rc == 0) {
+ VERBOSE (WARNING, PRINT ("Connection closed by peer (tx)\n"));
+ } else if (rc < 0) {
+ VERBOSE (WARNING, PRINT ("Connection in error (tx)\n"));
+ }
+ }
+
+ VERBOSE (DEBUG, PRINT ("Closing connection\n"));
+ if (data) {
+ free (data);
+ }
+ close_socket (conn);
+ conn = INVALID_SOCKET;
+ }
+
+ VERBOSE (DEBUG, PRINT ("Closing socket\n"));
+ close_socket (sock);
+ terminate_network_context ();
+
+ return 2;
}
// test: webserver.exe -h
// test: webserver.exe -h | awk '/usage:/ { rc=1 } END { exit (1-rc) }'
-// test: echo 1 | webserver.exe -v3 | grep -q value
-// test: webserver.exe -_ 2> /dev/null | awk 'END { if (NR == 0) { exit(0) } else exit (1) }'
+// test: webserver.exe -_ 2> /dev/null | wc -l | xargs test 0 =
// test: webserver.exe -_ 2>&1 | awk '/usage:/ { rc=1 } END { exit (1-rc) }'
// test: webserver.exe error 2>&1 | grep -q 'invalid option'
+// test: webserver.exe -v 2>&1 | grep -q 'missing verbose level'
+// test: webserver.exe -p 2>&1 | grep -q 'missing port number'
+// test: webserver.exe -p -1 2>&1 | grep -q 'incorrect port number'
+// test: webserver.exe > test.log & pid=$!; sleep 1; kill -9 $pid; grep -q 'Listening socket on port 8080' test.log
+// test: webserver.exe -p 8000 > test.log & pid=$!; sleep 1; kill -9 $pid; grep -q 'Listening socket on port 8000' test.log
+// test: webserver.exe & pid=$!; sleep 1; kill -TERM $pid; ps aux | grep -q [w]ebserver.exe && kill -9 $pid || rc=1; test x$rc = x1
/* vim: set ts=4 sw=4 et: */