libhv/examples/curl.cpp

/*
 * @build: make examples
 * @server bin/httpd -s restart -d
 * @usage: bin/curl -v www.baidu.com
 *         bin/curl -v 127.0.0.1:8080
 *         bin/curl -v 127.0.0.1:8080/ping
 *         bin/curl -v 127.0.0.1:8080/echo -d 'hello,world!'
 */

#include "http_client.h"

#ifdef _MSC_VER
#include "misc/win32_getopt.h"
#else
#include <getopt.h>
#endif

static int  http_version    = 1;
static int  grpc            = 0;
static bool verbose         = false;
static const char* url      = NULL;
static const char* method   = NULL;
static const char* headers  = NULL;
static const char* range    = NULL;
static const char* data     = NULL;
static const char* form     = NULL;
static int send_count       = 1;

static const char* options = "hVvX:H:r:d:F:n:";
static const struct option long_options[] = {
    {"help",    no_argument,        NULL,   'h'},
    {"verion",  no_argument,        NULL,   'V'},
    {"verbose", no_argument,        NULL,   'v'},
    {"method",  required_argument,  NULL,   'X'},
    {"header",  required_argument,  NULL,   'H'},
    {"range",   required_argument,  NULL,   'r'},
    {"data",    required_argument,  NULL,   'd'},
    {"form",    required_argument,  NULL,   'F'},
    {"http2",   no_argument,        &http_version, 2},
    {"grpc",    no_argument,        &grpc,  1},
    {"count",   required_argument,  NULL,   'n'},
    {NULL,      0,                  NULL,   0}
};
static const char* help = R"(Options:
    -h|--help           Print this message.
    -V|--version        Print version.
    -v|--verbose        Show verbose infomation.
    -X|--method         Set http method.
    -H|--header         Add http headers, -H "Content-Type:application/json Accept:*/*"
    -r|--range          Add http header Range:bytes=0-1023
    -d|--data           Set http body.
    -F|--form           Set http form, -F "name1=content name2=@filename"
    -n|--count          Send request count, used for test keep-alive
       --http2          Use http2
       --grpc           Use grpc over http2
Examples:
    curl -v GET  httpbin.org/get
    curl -v POST httpbin.org/post   user=admin pswd=123456
    curl -v PUT  httpbin.org/put    user=admin pswd=123456
    curl -v localhost:8080
    curl -v localhost:8080 -r 0-9
    curl -v localhost:8080/ping
    curl -v localhost:8080/query?page_no=1\&page_size=10
    curl -v localhost:8080/echo     hello,world!
    curl -v localhost:8080/kv       user=admin\&pswd=123456
    curl -v localhost:8080/json     user=admin pswd=123456
    curl -v localhost:8080/form     -F file=@filename
)";

static void print_usage() {
    printf("Usage: curl [%s] [METHOD] url [header_field:header_value] [body_key=body_value]\n", options);
}
static void print_version() {
    printf("curl version 1.0.0\n");
}
static void print_help() {
    print_usage();
    puts(help);
    print_version();
}

static bool is_upper_string(const char* str) {
    const char* p = str;
    while (*p >= 'A' && *p <= 'Z') ++p;
    return *p == '\0';
}

int parse_cmdline(int argc, char* argv[]) {
    int opt;
    int opt_idx;
    while ((opt = getopt_long(argc, argv, options, long_options, &opt_idx)) != EOF) {
        switch(opt) {
        case 'h': print_help();     exit(0);
        case 'V': print_version();  exit(0);
        case 'v': verbose = true;   break;
        case 'X': method = optarg;  break;
        case 'H': headers = optarg; break;
        case 'r': range = optarg;   break;
        case 'd': data = optarg;    break;
        case 'F': form = optarg;    break;
        case 'n': send_count = atoi(optarg); break;
        default: break;
        }
    }

    if (optind == argc) {
        printf("Missing url\n");
        print_usage();
        exit(-1);
    }

    if (is_upper_string(argv[optind])) {
        method = argv[optind++];
    }
    url = argv[optind++];

    return 0;
}

int main(int argc, char* argv[]) {
    if (argc < 2) {
        print_usage();
        return 0;
    }

    parse_cmdline(argc, argv);

    int ret = 0;
    HttpRequest req;
    if (grpc) {
        http_version = 2;
        req.content_type = APPLICATION_GRPC;
    }
    if (http_version == 2) {
        req.http_major = 2;
        req.http_minor = 0;
    }
    req.url = url;
    if (method) {
        req.method = http_method_enum(method);
    }
    enum {
        s_key,
        s_value,
    } state = s_key;
    if (headers) {
        const char* p = headers;
        const char* key = p;
        const char* value = NULL;
        int key_len = 0;
        int value_len = 0;
        state = s_key;
        while (*p != '\0') {
            if (*p == ' ') {
                if (key_len && value_len) {
                    req.headers[std::string(key,key_len)] = std::string(value,value_len);
                    key_len = value_len = 0;
                    state = s_key;
                }
            }
            else if (*p == ':') {
                state = s_value;
            }
            else {
                if (state == s_key) {
                    if (++key_len == 1) key = p;
                }
                else {
                    if (++value_len == 1) value = p;
                }
            }
            ++p;
        }
        if (key_len && value_len) {
            req.headers[std::string(key,key_len)] = std::string(value,value_len);
            key_len = value_len = 0;
        }
    }
    if (range) {
        req.headers["Range"] = std::string("bytes=").append(range);
    }
    if (data || form) {
        if (method == NULL) {
            req.method = HTTP_POST;
        }
        if (data) {
            req.body = data;
        }
        else if (form) {
            const char* p = form;
            const char* key = p;
            const char* value = NULL;
            int key_len = 0;
            int value_len = 0;
            state = s_key;
            while (*p != '\0') {
                if (*p == ' ') {
                    if (key_len && value_len) {
                        FormData data;
                        if (*value == '@') {
                            data.filename = std::string(value+1, value_len-1);
                        }
                        else {
                            data.content = std::string(value, value_len);
                        }
                        req.form[std::string(key,key_len)] = data;
                        key_len = value_len = 0;
                        state = s_key;
                    }
                }
                else if (*p == '=') {
                    state = s_value;
                }
                else {
                    if (state == s_key) {
                        if (++key_len == 1) key = p;
                    }
                    else {
                        if (++value_len == 1) value = p;
                    }
                }
                ++p;
            }
            if (key_len && value_len) {
                // printf("key=%.*s value=%.*s\n", key_len, key, value_len, value);
                FormData data;
                if (*value == '@') {
                    data.filename = std::string(value+1, value_len-1);
                }
                else {
                    data.content = std::string(value, value_len);
                }
                req.form[std::string(key,key_len)] = data;
            }
        }
    }
    for (int d = optind; d < argc; ++d) {
        const char* arg = argv[d];
        const char* pos = NULL;
        if ((pos = strchr(arg, ':')) != NULL) {
            // header_field:header_value
            *(char*)pos = '\0';
            req.headers[arg] = pos + 1;
        } else {
            if (method == NULL) {
                req.method = HTTP_POST;
            }
            if ((pos = strchr(arg, '&')) != NULL) {
                if (req.ContentType() == CONTENT_TYPE_NONE) {
                    req.content_type = X_WWW_FORM_URLENCODED;
                }
                req.body = arg;
            }
            else if ((pos = strchr(arg, '=')) != NULL) {
                // body_key=body_value
                if (req.ContentType() == CONTENT_TYPE_NONE) {
                    req.content_type = APPLICATION_JSON;
                }
                *(char*)pos = '\0';
                req.Set(arg, pos + 1);
            }
            else {
                if (req.ContentType() == CONTENT_TYPE_NONE) {
                    req.content_type = TEXT_PLAIN;
                }
                req.body = arg;
            }
        }
    }

    HttpResponse res;
    /*
    res.head_cb = [](const http_headers& headers){
        for (auto& header : headers) {
            printf("%s: %s\r\n", header.first.c_str(), header.second.c_str());
        }
        printf("\r\n");
    };
    res.body_cb = [](const char* data, size_t size){
        printf("%.*s", (int)size, data);
    };
    */
    res.chunked_cb = [](const char* data, size_t size){
        printf("%.*s", (int)size, data);
    };
    http_client_t* cli = http_client_new();
send:
    if (verbose) {
        printf("%s\n", req.Dump(true,true).c_str());
    }
    ret = http_client_send(cli, &req, &res);
    if (ret != 0) {
        printf("* Failed:%s:%d\n", http_client_strerror(ret), ret);
    } else {
        if (verbose) {
            printf("%s\n", res.Dump(true,true).c_str());
        }
        else {
            printf("%s\n", res.body.c_str());
        }
    }
    if (--send_count > 0) {
        printf("send again later...%d\n", send_count);
#ifdef _WIN32
        Sleep(3*1000);
#else
        sleep(3);
#endif
        goto send;
    }
    http_client_del(cli);
    return ret;
}