Changes in uspace/app/websrv/websrv.c [d076f16:8d2dd7f2] in mainline

File:

• uspace/app/websrv/websrv.c (modified) (10 diffs)

uspace/app/websrv/websrv.c

@@ -75 +75 @@
 static uint16_t port = DEFAULT_PORT;
 
-typedef struct {
-        tcp_conn_t *conn;
-
-        char rbuf[BUFFER_SIZE];
-        size_t rbuf_out;
-        size_t rbuf_in;
-
-        char lbuf[BUFFER_SIZE + 1];
-        size_t lbuf_used;
-} recv_t;
+static char rbuf[BUFFER_SIZE];
+static size_t rbuf_out;
+static size_t rbuf_in;
+
+static char lbuf[BUFFER_SIZE + 1];
+static size_t lbuf_used;
+
+static char fbuf[BUFFER_SIZE];
 
 static bool verbose = false;
@@ -133 +131 @@
     "</html>\r\n";
 
-
-static int recv_create(tcp_conn_t *conn, recv_t **rrecv)
-{
-        recv_t *recv;
-        
-        recv = calloc(1, sizeof(recv_t));
-        if (recv == NULL)
-                return ENOMEM;
-        
-        recv->conn = conn;
-        recv->rbuf_out = 0;
-        recv->rbuf_in = 0;
-        recv->lbuf_used = 0;
-        
-        *rrecv = recv;
-        return EOK;
-}
-
-static void recv_destroy(recv_t *recv)
-{
-        free(recv);
-}
-
 /** Receive one character (with buffering) */
-static int recv_char(recv_t *recv, char *c)
+static int recv_char(tcp_conn_t *conn, char *c)
 {
         size_t nrecv;
         int rc;
         
-        if (recv->rbuf_out == recv->rbuf_in) {
-                recv->rbuf_out = 0;
-                recv->rbuf_in = 0;
-                
-                rc = tcp_conn_recv_wait(recv->conn, recv->rbuf, BUFFER_SIZE, &nrecv);
+        if (rbuf_out == rbuf_in) {
+                rbuf_out = 0;
+                rbuf_in = 0;
+                
+                rc = tcp_conn_recv_wait(conn, rbuf, BUFFER_SIZE, &nrecv);
                 if (rc != EOK) {
                         fprintf(stderr, "tcp_conn_recv() failed (%d)\n", rc);
@@ -172 +147 @@
                 }
                 
-                recv->rbuf_in = nrecv;
-        }
-        
-        *c = recv->rbuf[recv->rbuf_out++];
+                rbuf_in = nrecv;
+        }
+        
+        *c = rbuf[rbuf_out++];
         return EOK;
 }
 
 /** Receive one line with length limit */
-static int recv_line(recv_t *recv, char **rbuf)
-{
-        char *bp = recv->lbuf;
+static int recv_line(tcp_conn_t *conn)
+{
+        char *bp = lbuf;
         char c = '\0';
         
-        while (bp < recv->lbuf + BUFFER_SIZE) {
+        while (bp < lbuf + BUFFER_SIZE) {
                 char prev = c;
-                int rc = recv_char(recv, &c);
+                int rc = recv_char(conn, &c);
                 
                 if (rc != EOK)
@@ -197 +172 @@
         }
         
-        recv->lbuf_used = bp - recv->lbuf;
+        lbuf_used = bp - lbuf;
         *bp = '\0';
         
-        if (bp == recv->lbuf + BUFFER_SIZE)
+        if (bp == lbuf + BUFFER_SIZE)
                 return ELIMIT;
         
-        *rbuf = recv->lbuf;
         return EOK;
 }
@@ -244 +218 @@
 static int uri_get(const char *uri, tcp_conn_t *conn)
 {
-        char *fbuf = NULL;
-        char *fname = NULL;
-        int rc;
-        int fd = -1;
-        
-        fbuf = calloc(BUFFER_SIZE, 1);
-        if (fbuf == NULL) {
-                rc = ENOMEM;
-                goto out;
-        }
-        
         if (str_cmp(uri, "/") == 0)
                 uri = "/index.html";
         
-        rc = asprintf(&fname, "%s%s", WEB_ROOT, uri);
-        if (rc < 0) {
-                rc = ENOMEM;
-                goto out;
-        }
-        
-        fd = vfs_lookup_open(fname, WALK_REGULAR, MODE_READ);
+        char *fname;
+        int rc = asprintf(&fname, "%s%s", WEB_ROOT, uri);
+        if (rc < 0)
+                return ENOMEM;
+        
+        int fd = vfs_lookup_open(fname, WALK_REGULAR, MODE_READ);
         if (fd < 0) {
                 rc = send_response(conn, msg_not_found);
-                goto out;
+                free(fname);
+                return rc;
         }
         
         free(fname);
-        fname = NULL;
         
         rc = send_response(conn, msg_ok);
         if (rc != EOK)
-                goto out;
+                return rc;
         
         aoff64_t pos = 0;
@@ -284 +246 @@
                 
                 if (nr < 0) {
-                        rc = EIO;
-                        goto out;
+                        vfs_put(fd);
+                        return EIO;
                 }
                 
@@ -291 +253 @@
                 if (rc != EOK) {
                         fprintf(stderr, "tcp_conn_send() failed\n");
-                        goto out;
+                        vfs_put(fd);
+                        return rc;
                 }
         }
         
-        rc = EOK;
-out:
-        if (fd >= 0)
-                vfs_put(fd);
-        free(fname);
-        free(fbuf);
-        return rc;
-}
-
-static int req_process(tcp_conn_t *conn, recv_t *recv)
-{
-        char *reqline = NULL;
-
-        int rc = recv_line(recv, &reqline);
+        vfs_put(fd);
+        
+        return EOK;
+}
+
+static int req_process(tcp_conn_t *conn)
+{
+        int rc = recv_line(conn);
         if (rc != EOK) {
                 fprintf(stderr, "recv_line() failed\n");
@@ -315 +272 @@
         
         if (verbose)
-                fprintf(stderr, "Request: %s", reqline);
-        
-        if (str_lcmp(reqline, "GET ", 4) != 0) {
+                fprintf(stderr, "Request: %s", lbuf);
+        
+        if (str_lcmp(lbuf, "GET ", 4) != 0) {
                 rc = send_response(conn, msg_not_implemented);
                 return rc;
         }
         
-        char *uri = reqline + 4;
+        char *uri = lbuf + 4;
         char *end_uri = str_chr(uri, ' ');
         if (end_uri == NULL) {
-                end_uri = reqline + str_size(reqline) - 2;
+                end_uri = lbuf + lbuf_used - 2;
                 assert(*end_uri == '\r');
         }
@@ -343 +300 @@
 static void usage(void)
 {
-        printf("Simple web server\n"
+        printf("Skeletal server\n"
             "\n"
             "Usage: " NAME " [options]\n"
@@ -406 +363 @@
 {
         int rc;
-        recv_t *recv = NULL;
         
         if (verbose)
                 fprintf(stderr, "New connection, waiting for request\n");
         
-        rc = recv_create(conn, &recv);
-        if (rc != EOK) {
-                fprintf(stderr, "Out of memory.\n");
-                goto error;
-        }
-        
-        rc = req_process(conn, recv);
+        rbuf_out = 0;
+        rbuf_in = 0;
+        
+        rc = req_process(conn);
         if (rc != EOK) {
                 fprintf(stderr, "Error processing request (%s)\n",
                     str_error(rc));
-                goto error;
-        }
-        
+                return;
+        }
+
         rc = tcp_conn_send_fin(conn);
         if (rc != EOK) {
                 fprintf(stderr, "Error sending FIN.\n");
-                goto error;
-        }
-
-        recv_destroy(recv);
-        return;
-error:
-        rc = tcp_conn_reset(conn);
-        if (rc != EOK)
-                fprintf(stderr, "Error resetting connection.\n");
-        
-        recv_destroy(recv);
+                return;
+        }
 }