| 1 | #include "options.hfa"
 | 
|---|
| 2 | 
 | 
|---|
| 3 | #define _GNU_SOURCE
 | 
|---|
| 4 | #define __USE_GNU
 | 
|---|
| 5 | extern "C" {
 | 
|---|
| 6 |         #include <sys/types.h>
 | 
|---|
| 7 |         #include <sys/stat.h>
 | 
|---|
| 8 |         #include <fcntl.h>
 | 
|---|
| 9 | }
 | 
|---|
| 10 | 
 | 
|---|
| 11 | #include <bitmanip.hfa>
 | 
|---|
| 12 | #include <fstream.hfa>
 | 
|---|
| 13 | #include <kernel.hfa>
 | 
|---|
| 14 | #include <parseargs.hfa>
 | 
|---|
| 15 | #include <stdlib.hfa>
 | 
|---|
| 16 | 
 | 
|---|
| 17 | #include <stdlib.h>
 | 
|---|
| 18 | #include <string.h>
 | 
|---|
| 19 | 
 | 
|---|
| 20 | Options options @= {
 | 
|---|
| 21 |         false, // log
 | 
|---|
| 22 |         false, // stats
 | 
|---|
| 23 | 
 | 
|---|
| 24 |         { // file_cache
 | 
|---|
| 25 |                 0,     // path
 | 
|---|
| 26 |                 0,     // open_flags;
 | 
|---|
| 27 |                 42u,   // hash_seed;
 | 
|---|
| 28 |                 0,     // size;
 | 
|---|
| 29 |                 false, // list;
 | 
|---|
| 30 |                 false  // fixed_fds
 | 
|---|
| 31 |         },
 | 
|---|
| 32 | 
 | 
|---|
| 33 |         { // socket
 | 
|---|
| 34 |                 8080, // port
 | 
|---|
| 35 |                 10,   // backlog
 | 
|---|
| 36 |                 1024  // buflen
 | 
|---|
| 37 |         },
 | 
|---|
| 38 | 
 | 
|---|
| 39 |         { // cluster
 | 
|---|
| 40 |                 1,     // nclusters;
 | 
|---|
| 41 |                 1,     // nprocs;
 | 
|---|
| 42 |                 1,     // nworkers;
 | 
|---|
| 43 |                 {},     // params;
 | 
|---|
| 44 |                 false, // procstats
 | 
|---|
| 45 |                 false, // viewhalts
 | 
|---|
| 46 |                 0,     // instance;
 | 
|---|
| 47 |         }
 | 
|---|
| 48 | };
 | 
|---|
| 49 | 
 | 
|---|
| 50 | void parse_options( int argc, char * argv[] ) {
 | 
|---|
| 51 |         // bool fixedfd = false;
 | 
|---|
| 52 |         // bool sqkpoll = false;
 | 
|---|
| 53 |         // bool iokpoll = false;
 | 
|---|
| 54 |         unsigned nentries = 16;
 | 
|---|
| 55 |         bool isolate = false;
 | 
|---|
| 56 | 
 | 
|---|
| 57 | 
 | 
|---|
| 58 |         static cfa_option opt[] = {
 | 
|---|
| 59 |                 { 'p', "port",           "Port the server will listen on", options.socket.port},
 | 
|---|
| 60 |                 { 'c', "cpus",           "Number of processors to use", options.clopts.nprocs},
 | 
|---|
| 61 |                 { 't', "threads",        "Number of worker threads to use", options.clopts.nworkers},
 | 
|---|
| 62 |                 {'\0', "isolate",        "Create one cluster per processor", isolate, parse_settrue},
 | 
|---|
| 63 |                 {'\0', "log",            "Enable logs", options.log, parse_settrue},
 | 
|---|
| 64 |                 {'\0', "stats",          "Enable statistics", options.stats, parse_settrue},
 | 
|---|
| 65 |                 {'\0', "accept-backlog", "Maximum number of pending accepts", options.socket.backlog},
 | 
|---|
| 66 |                 {'\0', "request_len",    "Maximum number of bytes in the http request, requests with more data will be answered with Http Code 414", options.socket.buflen},
 | 
|---|
| 67 |                 {'\0', "seed",           "seed to use for hashing", options.file_cache.hash_seed },
 | 
|---|
| 68 |                 {'\0', "cache-size",     "Size of the cache to use, if set to small, will uses closes power of 2", options.file_cache.size },
 | 
|---|
| 69 |                 {'\0', "list-files",     "List the files in the specified path and exit", options.file_cache.list, parse_settrue },
 | 
|---|
| 70 |                 // { 'f', "fixed-fds",      "If set, files are open eagerly and pre-registered with the cluster", fixedfd, parse_settrue},
 | 
|---|
| 71 |                 // { 'k', "kpollsubmit",    "If set, cluster uses IORING_SETUP_SQPOLL, implies -f", sqkpoll, parse_settrue },
 | 
|---|
| 72 |                 // { 'i', "kpollcomplete",  "If set, cluster uses IORING_SETUP_IOPOLL", iokpoll, parse_settrue },
 | 
|---|
| 73 |                 {'e', "numentries",     "Number of I/O entries", nentries },
 | 
|---|
| 74 | 
 | 
|---|
| 75 |         };
 | 
|---|
| 76 |         int opt_cnt = sizeof(opt) / sizeof(cfa_option);
 | 
|---|
| 77 | 
 | 
|---|
| 78 |         char **left;
 | 
|---|
| 79 |         parse_args( argc, argv, opt, opt_cnt, "[OPTIONS]... [PATH]\ncforall http server", left );
 | 
|---|
| 80 | 
 | 
|---|
| 81 |         if( !is_pow2(nentries) ) {
 | 
|---|
| 82 |                 unsigned v = nentries;
 | 
|---|
| 83 |                 v--;
 | 
|---|
| 84 |                 v |= v >> 1;
 | 
|---|
| 85 |                 v |= v >> 2;
 | 
|---|
| 86 |                 v |= v >> 4;
 | 
|---|
| 87 |                 v |= v >> 8;
 | 
|---|
| 88 |                 v |= v >> 16;
 | 
|---|
| 89 |                 v++;
 | 
|---|
| 90 |                 serr | "Warning: num_entries not a power of 2" | '(' | nentries | ')' | "raising to " | v;
 | 
|---|
| 91 |                 nentries = v;
 | 
|---|
| 92 |         }
 | 
|---|
| 93 |         if(isolate) {
 | 
|---|
| 94 |                 options.clopts.nclusters = options.clopts.nprocs;
 | 
|---|
| 95 |                 options.clopts.nprocs = 1;
 | 
|---|
| 96 |         }
 | 
|---|
| 97 |         options.clopts.params.num_entries = nentries;
 | 
|---|
| 98 |         options.clopts.instance = alloc(options.clopts.nclusters);
 | 
|---|
| 99 |         options.clopts.thrd_cnt = alloc(options.clopts.nclusters);
 | 
|---|
| 100 |         options.clopts.cltr_cnt = 0;
 | 
|---|
| 101 |         for(i; options.clopts.nclusters) {
 | 
|---|
| 102 |                 options.clopts.thrd_cnt[i] = 0;
 | 
|---|
| 103 |         }
 | 
|---|
| 104 | 
 | 
|---|
| 105 | 
 | 
|---|
| 106 |         // if( fixedfd ) {
 | 
|---|
| 107 |         //      options.file_cache.fixed_fds = true;
 | 
|---|
| 108 |         // }
 | 
|---|
| 109 | 
 | 
|---|
| 110 |         // if( sqkpoll ) {
 | 
|---|
| 111 |         //      options.file_cache.fixed_fds = true;
 | 
|---|
| 112 |         // }
 | 
|---|
| 113 | 
 | 
|---|
| 114 |         // if( iokpoll ) {
 | 
|---|
| 115 |         //      options.file_cache.open_flags |= O_DIRECT;
 | 
|---|
| 116 |         // }
 | 
|---|
| 117 | 
 | 
|---|
| 118 |         if( left[0] == 0p ) { return; }
 | 
|---|
| 119 | 
 | 
|---|
| 120 |         const char * path = left[0];
 | 
|---|
| 121 |         left++;
 | 
|---|
| 122 | 
 | 
|---|
| 123 |         if( left[0] != 0p ) {
 | 
|---|
| 124 |                 serr | "Too many trailing arguments!" | '\'' | path | '\'';
 | 
|---|
| 125 |                 while(left[0] != 0p) {
 | 
|---|
| 126 |                         serr | " - " | left[0];
 | 
|---|
| 127 |                         left++;
 | 
|---|
| 128 |                 }
 | 
|---|
| 129 |                 exit(EXIT_FAILURE);
 | 
|---|
| 130 |         }
 | 
|---|
| 131 | 
 | 
|---|
| 132 |         options.file_cache.path = path;
 | 
|---|
| 133 | }
 | 
|---|