source: benchmark/io/http/worker.cfa @ 97748ee

arm-ehjacob/cs343-translationnew-ast-unique-expr
Last change on this file since 97748ee was 97748ee, checked in by Thierry Delisle <tdelisle@…>, 11 months ago

Worker now supports /ping uri which just answers empty reponse.

  • Property mode set to 100644
File size: 3.0 KB
Line 
1#include "worker.hfa"
2
3#include <errno.h>
4#include <stdio.h>
5#include <string.h>
6#include <unistd.h>
7
8#include <iofwd.hfa>
9
10#include "options.hfa"
11#include "protocol.hfa"
12#include "filecache.hfa"
13
14//=============================================================================================
15// Worker Thread
16//=============================================================================================
17void ?{}( Worker & this ) {
18        ((thread&)this){ "Server Worker Thread", *options.clopts.instance };
19        this.pipe[0] = -1;
20        this.pipe[1] = -1;
21        this.done = false;
22}
23
24extern "C" {
25extern int accept4(int sockfd, struct sockaddr *addr, socklen_t *addrlen, int flags);
26}
27
28void main( Worker & this ) {
29        park();
30        /* paranoid */ assert( this.pipe[0] != -1 );
31        /* paranoid */ assert( this.pipe[1] != -1 );
32
33        CONNECTION:
34        for() {
35                if( options.log ) printf("=== Accepting connection ===\n");
36                int fd = cfa_accept4( this.[sockfd, addr, addrlen, flags], 0, -1`s, &this.cancel, 0p );
37                // int fd = accept4( this.[sockfd, addr, addrlen, flags] );
38                if(fd < 0) {
39                        if( errno == ECONNABORTED ) break;
40                        if( errno == EINVAL && this.done ) break;
41                        abort( "accept error: (%d) %s\n", (int)errno, strerror(errno) );
42                }
43
44                if( options.log ) printf("=== New connection %d, waiting for requests ===\n", fd);
45                REQUEST:
46                for() {
47                        bool closed;
48                        HttpCode code;
49                        const char * file;
50                        size_t name_size;
51
52                        // Read the http request
53                        size_t len = options.socket.buflen;
54                        char buffer[len];
55                        if( options.log ) printf("=== Reading request ===\n");
56                        [code, closed, file, name_size] = http_read(fd, buffer, len, &this.cancel);
57
58                        // if we are done, break out of the loop
59                        if( closed ) {
60                                if( options.log ) printf("=== Connection closed ===\n");
61                                close(fd);
62                                continue CONNECTION;
63                        }
64
65                        // If this wasn't a request retrun 400
66                        if( code != OK200 ) {
67                                printf("=== Invalid Request : %d ===\n", code_val(code));
68                                answer_error(fd, code);
69                                continue REQUEST;
70                        }
71
72                        if(0 == strncmp(file, "plaintext", min(name_size, sizeof("plaintext") ))) {
73                                if( options.log ) printf("=== Request for /plaintext ===\n");
74
75                                char text[] = "Hello, World!\n";
76
77                                // Send the header
78                                answer_plain(fd, text, sizeof(text));
79
80                                if( options.log ) printf("=== Answer sent ===\n");
81                                continue REQUEST;
82                        }
83
84                        if(0 == strncmp(file, "ping", min(name_size, sizeof("ping") ))) {
85                                if( options.log ) printf("=== Request for /ping ===\n");
86
87                                // Send the header
88                                answer_empty(fd);
89
90                                if( options.log ) printf("=== Answer sent ===\n");
91                                continue REQUEST;
92                        }
93
94                        if( options.log ) printf("=== Request for file %.*s ===\n", (int)name_size, file);
95
96                        // Get the fd from the file cache
97                        int ans_fd;
98                        size_t count;
99                        [ans_fd, count] = get_file( file, name_size );
100
101                        // If we can't find the file, return 404
102                        if( ans_fd < 0 ) {
103                                printf("=== File Not Found ===\n");
104                                answer_error(fd, E404);
105                                continue REQUEST;
106                        }
107
108                        // Send the header
109                        answer_header(fd, count);
110
111                        // Send the desired file
112                        sendfile( this.pipe, fd, ans_fd, count);
113
114                        if( options.log ) printf("=== Answer sent ===\n");
115                }
116        }
117}
Note: See TracBrowser for help on using the repository browser.