C302/c302.c

#include <stdint.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/select.h>
#include <sys/uio.h>
#include <time.h>
#include "c302.h"

static void http_error(errcode_enum_t code, char* msg) {
    uint32_t len = strlen(msg) + typel[code];
    char* str = malloc(len);
    sprintf(str, types[code], msg);
    const struct iovec iov[2] = {{&len, sizeof(uint32_t)}, {str, len}};
    writev(1, (const struct iovec *)&iov, 2);
    free(str);
}

// Usage: c302 method query_string
int main(int argc, char **argv) {
    if (argc != 3) {
        http_error(HTTP500, "Argument Count Error.");
        return 1;
    }
    int i = time(NULL) % (sizeof(urls)/sizeof(char*));
    uint32_t len = sizeof(HTTP302HEAD)-1 + urlsl[i] + 4 + sizeof(CONTENT)-1;
    const struct iovec iov[4] = {
        {(void *)&len, sizeof(uint32_t)},
        {HTTP302HEAD, sizeof(HTTP302HEAD)-1},
        {(void *)urls[i], urlsl[i]},
        {"\r\n\r\n" CONTENT, 4+sizeof(CONTENT)-1}
    };
    return writev(1, (const struct iovec *)&iov, 4) != len+sizeof(uint32_t);
}

static void __attribute__((destructor)) defer_close_fp() {
    fflush(stdout);
    char buf[256];
    fd_set rfds;
    struct timeval tv;
    int ret;
    FD_ZERO(&rfds);
    FD_SET(0, &rfds);
    tv.tv_sec = 1;
    tv.tv_usec = 0;
    if (select(1, &rfds, NULL, NULL, &tv) > 0 && FD_ISSET(0, &rfds)) {
        read(0, buf, sizeof(buf));
    }
}