src/tests/tcurl_test_tool.c

91b0592cdab22915dff27ceae6d8e49c608aea4aJakub Hrozek/*
91b0592cdab22915dff27ceae6d8e49c608aea4aJakub Hrozek   SSSD
91b0592cdab22915dff27ceae6d8e49c608aea4aJakub Hrozek
91b0592cdab22915dff27ceae6d8e49c608aea4aJakub Hrozek   libcurl tevent integration test tool
91b0592cdab22915dff27ceae6d8e49c608aea4aJakub Hrozek
91b0592cdab22915dff27ceae6d8e49c608aea4aJakub Hrozek   Copyright (C) Red Hat, 2016
91b0592cdab22915dff27ceae6d8e49c608aea4aJakub Hrozek
91b0592cdab22915dff27ceae6d8e49c608aea4aJakub Hrozek   This program is free software; you can redistribute it and/or modify
91b0592cdab22915dff27ceae6d8e49c608aea4aJakub Hrozek   it under the terms of the GNU General Public License as published by
91b0592cdab22915dff27ceae6d8e49c608aea4aJakub Hrozek   the Free Software Foundation; either version 3 of the License, or
91b0592cdab22915dff27ceae6d8e49c608aea4aJakub Hrozek   (at your option) any later version.
91b0592cdab22915dff27ceae6d8e49c608aea4aJakub Hrozek
91b0592cdab22915dff27ceae6d8e49c608aea4aJakub Hrozek   This program is distributed in the hope that it will be useful,
91b0592cdab22915dff27ceae6d8e49c608aea4aJakub Hrozek   but WITHOUT ANY WARRANTY; without even the implied warranty of
91b0592cdab22915dff27ceae6d8e49c608aea4aJakub Hrozek   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
91b0592cdab22915dff27ceae6d8e49c608aea4aJakub Hrozek   GNU General Public License for more details.
91b0592cdab22915dff27ceae6d8e49c608aea4aJakub Hrozek
91b0592cdab22915dff27ceae6d8e49c608aea4aJakub Hrozek   You should have received a copy of the GNU General Public License
91b0592cdab22915dff27ceae6d8e49c608aea4aJakub Hrozek   along with this program.  If not, see <http://www.gnu.org/licenses/>.
91b0592cdab22915dff27ceae6d8e49c608aea4aJakub Hrozek*/
91b0592cdab22915dff27ceae6d8e49c608aea4aJakub Hrozek
91b0592cdab22915dff27ceae6d8e49c608aea4aJakub Hrozek#include <popt.h>
91b0592cdab22915dff27ceae6d8e49c608aea4aJakub Hrozek
91b0592cdab22915dff27ceae6d8e49c608aea4aJakub Hrozek#include "util/util.h"
91b0592cdab22915dff27ceae6d8e49c608aea4aJakub Hrozek#include "util/tev_curl.h"
91b0592cdab22915dff27ceae6d8e49c608aea4aJakub Hrozek
91b0592cdab22915dff27ceae6d8e49c608aea4aJakub Hrozek#define MAXREQ 64
91b0592cdab22915dff27ceae6d8e49c608aea4aJakub Hrozek
91b0592cdab22915dff27ceae6d8e49c608aea4aJakub Hrozekstruct tool_ctx {
91b0592cdab22915dff27ceae6d8e49c608aea4aJakub Hrozek    bool verbose;
91b0592cdab22915dff27ceae6d8e49c608aea4aJakub Hrozek    bool done;
91b0592cdab22915dff27ceae6d8e49c608aea4aJakub Hrozek
91b0592cdab22915dff27ceae6d8e49c608aea4aJakub Hrozek    size_t nreqs;
91b0592cdab22915dff27ceae6d8e49c608aea4aJakub Hrozek};
91b0592cdab22915dff27ceae6d8e49c608aea4aJakub Hrozek
b800a6d09244359959404aca81c6796a58cafbcbPavel Březinastruct tool_options {
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina    int debug;
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina    int verbose;
36e49a842e257ac9bde71728ee3bef4299b6e6e2Pavel Březina    int raw;
886e0f75e6f4c7877a23a3625f8a20c09109b09dPavel Březina    int tls;
886e0f75e6f4c7877a23a3625f8a20c09109b09dPavel Březina    int verify_peer;
886e0f75e6f4c7877a23a3625f8a20c09109b09dPavel Březina    int verify_host;
d1ed11fc50922aab2332758a9300f3fbf814f112Pavel Březina    const char **headers;
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina    enum tcurl_http_method method;
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina    const char *socket_path;
886e0f75e6f4c7877a23a3625f8a20c09109b09dPavel Březina    const char *capath;
886e0f75e6f4c7877a23a3625f8a20c09109b09dPavel Březina    const char *cacert;
c2ea75da72b426d98ba489039e220d417bfb4c2aPavel Březina
ae6b11229d9961e26922918183c7c1de7780b8d6Pavel Březina    const char *clientcert;
ae6b11229d9961e26922918183c7c1de7780b8d6Pavel Březina    const char *clientkey;
ae6b11229d9961e26922918183c7c1de7780b8d6Pavel Březina
c2ea75da72b426d98ba489039e220d417bfb4c2aPavel Březina    const char *username;
c2ea75da72b426d98ba489039e220d417bfb4c2aPavel Březina    const char *password;
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina};
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina
91b0592cdab22915dff27ceae6d8e49c608aea4aJakub Hrozekstatic void request_done(struct tevent_req *req)
91b0592cdab22915dff27ceae6d8e49c608aea4aJakub Hrozek{
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina    struct tool_ctx *tool_ctx;
91b0592cdab22915dff27ceae6d8e49c608aea4aJakub Hrozek    struct sss_iobuf *outbuf;
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina    int http_code;
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina    errno_t ret;
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina    tool_ctx = tevent_req_callback_data(req, struct tool_ctx);
91b0592cdab22915dff27ceae6d8e49c608aea4aJakub Hrozek
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina    ret = tcurl_request_recv(tool_ctx, req, &outbuf, &http_code);
91b0592cdab22915dff27ceae6d8e49c608aea4aJakub Hrozek    talloc_zfree(req);
91b0592cdab22915dff27ceae6d8e49c608aea4aJakub Hrozek
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina    tool_ctx->nreqs--;
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina    if (tool_ctx->nreqs == 0) {
91b0592cdab22915dff27ceae6d8e49c608aea4aJakub Hrozek        tool_ctx->done = true;
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina    }
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina    if (ret != EOK) {
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina        DEBUG(SSSDBG_FATAL_FAILURE, "HTTP request failed [%d]: %s\n",
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina              ret, sss_strerror(ret));
91b0592cdab22915dff27ceae6d8e49c608aea4aJakub Hrozek        return;
91b0592cdab22915dff27ceae6d8e49c608aea4aJakub Hrozek    } else if (tool_ctx->verbose) {
91b0592cdab22915dff27ceae6d8e49c608aea4aJakub Hrozek        printf("Request HTTP code: %d\n", http_code);
91b0592cdab22915dff27ceae6d8e49c608aea4aJakub Hrozek        printf("Request HTTP body: \n%s\n",
91b0592cdab22915dff27ceae6d8e49c608aea4aJakub Hrozek               (const char *) sss_iobuf_get_data(outbuf));
91b0592cdab22915dff27ceae6d8e49c608aea4aJakub Hrozek        talloc_zfree(outbuf);
91b0592cdab22915dff27ceae6d8e49c608aea4aJakub Hrozek    }
91b0592cdab22915dff27ceae6d8e49c608aea4aJakub Hrozek}
91b0592cdab22915dff27ceae6d8e49c608aea4aJakub Hrozek
b800a6d09244359959404aca81c6796a58cafbcbPavel Březinastatic errno_t
b800a6d09244359959404aca81c6796a58cafbcbPavel Březinaparse_options(poptContext pc, struct tool_options *opts)
91b0592cdab22915dff27ceae6d8e49c608aea4aJakub Hrozek{
91b0592cdab22915dff27ceae6d8e49c608aea4aJakub Hrozek    int opt;
91b0592cdab22915dff27ceae6d8e49c608aea4aJakub Hrozek
91b0592cdab22915dff27ceae6d8e49c608aea4aJakub Hrozek    while ((opt = poptGetNextOpt(pc)) > 0) {
91b0592cdab22915dff27ceae6d8e49c608aea4aJakub Hrozek        switch (opt) {
91b0592cdab22915dff27ceae6d8e49c608aea4aJakub Hrozek        case 'g':
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina            opts->method = TCURL_HTTP_GET;
91b0592cdab22915dff27ceae6d8e49c608aea4aJakub Hrozek            break;
91b0592cdab22915dff27ceae6d8e49c608aea4aJakub Hrozek        case 'p':
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina            opts->method = TCURL_HTTP_PUT;
91b0592cdab22915dff27ceae6d8e49c608aea4aJakub Hrozek            break;
c9db8b8b19827c3d492b8d2769aa77a37dbc12d3Jakub Hrozek        case 'o':
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina            opts->method = TCURL_HTTP_POST;
300b9e9217ee1ed8d845ed2370c5ccf5c87afb36Pavel Březina            break;
300b9e9217ee1ed8d845ed2370c5ccf5c87afb36Pavel Březina        case 'd':
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina            opts->method = TCURL_HTTP_DELETE;
91b0592cdab22915dff27ceae6d8e49c608aea4aJakub Hrozek            break;
91b0592cdab22915dff27ceae6d8e49c608aea4aJakub Hrozek        default:
91b0592cdab22915dff27ceae6d8e49c608aea4aJakub Hrozek            DEBUG(SSSDBG_FATAL_FAILURE, "Unexpected option\n");
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina            return EINVAL;
91b0592cdab22915dff27ceae6d8e49c608aea4aJakub Hrozek        }
91b0592cdab22915dff27ceae6d8e49c608aea4aJakub Hrozek    }
91b0592cdab22915dff27ceae6d8e49c608aea4aJakub Hrozek
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina    if (opt != -1) {
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina        poptPrintUsage(pc, stderr, 0);
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina        fprintf(stderr, "%s", poptStrerror(opt));
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina        return EINVAL;
91b0592cdab22915dff27ceae6d8e49c608aea4aJakub Hrozek    }
91b0592cdab22915dff27ceae6d8e49c608aea4aJakub Hrozek
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina    return EOK;
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina}
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina
b800a6d09244359959404aca81c6796a58cafbcbPavel Březinastatic errno_t
b800a6d09244359959404aca81c6796a58cafbcbPavel Březinaprepare_requests(TALLOC_CTX *mem_ctx,
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina                 poptContext pc,
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina                 struct tool_options *opts,
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina                 struct tcurl_request ***_requests,
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina                 size_t *_num_requests)
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina{
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina    struct tcurl_request **requests;
d1ed11fc50922aab2332758a9300f3fbf814f112Pavel Březina    struct sss_iobuf *body;
d1ed11fc50922aab2332758a9300f3fbf814f112Pavel Březina    const char **headers;
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina    const char *arg;
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina    const char *url;
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina    errno_t ret;
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina    size_t i;
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina
d1ed11fc50922aab2332758a9300f3fbf814f112Pavel Březina    static const char *default_headers[] = {
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina        "Content-type: application/octet-stream",
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina        NULL,
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina    };
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina    requests = talloc_zero_array(mem_ctx, struct tcurl_request *, MAXREQ + 1);
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina    if (requests == NULL) {
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina        return ENOMEM;
91b0592cdab22915dff27ceae6d8e49c608aea4aJakub Hrozek    }
91b0592cdab22915dff27ceae6d8e49c608aea4aJakub Hrozek
d1ed11fc50922aab2332758a9300f3fbf814f112Pavel Březina    headers = opts->headers == NULL ? default_headers : opts->headers;
d1ed11fc50922aab2332758a9300f3fbf814f112Pavel Březina
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina    i = 0;
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina    while ((arg = poptGetArg(pc)) != NULL) {
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina        if (i >= MAXREQ) {
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina            fprintf(stderr, _("Too many requests!\n"));
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina            ret = EINVAL;
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina            goto done;
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina        }
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina        switch (opts->method) {
91b0592cdab22915dff27ceae6d8e49c608aea4aJakub Hrozek        case TCURL_HTTP_GET:
91b0592cdab22915dff27ceae6d8e49c608aea4aJakub Hrozek        case TCURL_HTTP_DELETE:
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina            url = arg;
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina            body = NULL;
91b0592cdab22915dff27ceae6d8e49c608aea4aJakub Hrozek            break;
91b0592cdab22915dff27ceae6d8e49c608aea4aJakub Hrozek        case TCURL_HTTP_PUT:
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina        case TCURL_HTTP_POST:
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina            url = arg;
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina            arg = poptGetArg(pc);
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina            if (arg == NULL) {
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina                body = NULL;
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina                break;
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina            }
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina            body = sss_iobuf_init_readonly(requests,
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina                                           discard_const_p(uint8_t, arg),
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina                                           strlen(arg));
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina            if (body == NULL) {
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina                ret = ENOMEM;
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina                goto done;
91b0592cdab22915dff27ceae6d8e49c608aea4aJakub Hrozek            }
91b0592cdab22915dff27ceae6d8e49c608aea4aJakub Hrozek            break;
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina        default:
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina            DEBUG(SSSDBG_CRIT_FAILURE, "Invalid method!\n");
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina            ret = EINVAL;
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina            goto done;
91b0592cdab22915dff27ceae6d8e49c608aea4aJakub Hrozek        }
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina        requests[i] = tcurl_http(requests, opts->method, opts->socket_path,
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina                                 url, headers, body);
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina        if (requests[i] == NULL) {
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina            ret = ENOMEM;
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina            goto done;
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina        }
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina
36e49a842e257ac9bde71728ee3bef4299b6e6e2Pavel Březina        if (opts->raw) {
36e49a842e257ac9bde71728ee3bef4299b6e6e2Pavel Březina            ret = tcurl_req_enable_rawoutput(requests[i]);
36e49a842e257ac9bde71728ee3bef4299b6e6e2Pavel Březina            if (ret != EOK) {
36e49a842e257ac9bde71728ee3bef4299b6e6e2Pavel Březina                goto done;
36e49a842e257ac9bde71728ee3bef4299b6e6e2Pavel Březina            }
36e49a842e257ac9bde71728ee3bef4299b6e6e2Pavel Březina        }
36e49a842e257ac9bde71728ee3bef4299b6e6e2Pavel Březina
886e0f75e6f4c7877a23a3625f8a20c09109b09dPavel Březina        if (opts->tls) {
886e0f75e6f4c7877a23a3625f8a20c09109b09dPavel Březina            ret = tcurl_req_verify_peer(requests[i], opts->capath, opts->cacert,
886e0f75e6f4c7877a23a3625f8a20c09109b09dPavel Březina                                        opts->verify_peer, opts->verify_host);
886e0f75e6f4c7877a23a3625f8a20c09109b09dPavel Březina            if (ret != EOK) {
886e0f75e6f4c7877a23a3625f8a20c09109b09dPavel Březina                goto done;
886e0f75e6f4c7877a23a3625f8a20c09109b09dPavel Březina            }
886e0f75e6f4c7877a23a3625f8a20c09109b09dPavel Březina        }
886e0f75e6f4c7877a23a3625f8a20c09109b09dPavel Březina
ae6b11229d9961e26922918183c7c1de7780b8d6Pavel Březina        if (opts->clientcert != NULL) {
ae6b11229d9961e26922918183c7c1de7780b8d6Pavel Březina            ret = tcurl_req_set_client_cert(requests[i], opts->clientcert,
ae6b11229d9961e26922918183c7c1de7780b8d6Pavel Březina                                            opts->clientkey);
ae6b11229d9961e26922918183c7c1de7780b8d6Pavel Březina            if (ret != EOK) {
ae6b11229d9961e26922918183c7c1de7780b8d6Pavel Březina                goto done;
ae6b11229d9961e26922918183c7c1de7780b8d6Pavel Březina            }
ae6b11229d9961e26922918183c7c1de7780b8d6Pavel Březina        }
ae6b11229d9961e26922918183c7c1de7780b8d6Pavel Březina
c2ea75da72b426d98ba489039e220d417bfb4c2aPavel Březina        if (opts->username != NULL && opts->password != NULL) {
c2ea75da72b426d98ba489039e220d417bfb4c2aPavel Březina            ret = tcurl_req_http_basic_auth(requests[i], opts->username,
c2ea75da72b426d98ba489039e220d417bfb4c2aPavel Březina                                            opts->password);
c2ea75da72b426d98ba489039e220d417bfb4c2aPavel Březina            if (ret != EOK) {
c2ea75da72b426d98ba489039e220d417bfb4c2aPavel Březina                goto done;
c2ea75da72b426d98ba489039e220d417bfb4c2aPavel Březina            }
c2ea75da72b426d98ba489039e220d417bfb4c2aPavel Březina        }
c2ea75da72b426d98ba489039e220d417bfb4c2aPavel Březina
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina        i++;
91b0592cdab22915dff27ceae6d8e49c608aea4aJakub Hrozek    }
91b0592cdab22915dff27ceae6d8e49c608aea4aJakub Hrozek
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina    *_requests = requests;
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina    *_num_requests = i;
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina    ret = EOK;
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina
b800a6d09244359959404aca81c6796a58cafbcbPavel Březinadone:
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina    if (ret != EOK) {
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina        talloc_free(requests);
91b0592cdab22915dff27ceae6d8e49c608aea4aJakub Hrozek    }
91b0592cdab22915dff27ceae6d8e49c608aea4aJakub Hrozek
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina    return ret;
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina}
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina
b800a6d09244359959404aca81c6796a58cafbcbPavel Březinastatic errno_t
b800a6d09244359959404aca81c6796a58cafbcbPavel Březinarun_requests(struct tool_ctx *tool_ctx,
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina             struct tcurl_request **requests)
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina{
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina    TALLOC_CTX *tmp_ctx;
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina    struct tcurl_ctx *tcurl_ctx;
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina    struct tevent_context *ev;
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina    struct tevent_req *req;
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina    errno_t ret;
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina    int i;
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina    tmp_ctx = talloc_new(NULL);
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina    if (tmp_ctx == NULL) {
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina        DEBUG(SSSDBG_FATAL_FAILURE, "Out of memory!\n");
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina        return ENOMEM;
91b0592cdab22915dff27ceae6d8e49c608aea4aJakub Hrozek    }
91b0592cdab22915dff27ceae6d8e49c608aea4aJakub Hrozek
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina    if (requests == NULL || requests[0] == NULL) {
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina        ret = EOK;
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina        goto done;
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina    }
91b0592cdab22915dff27ceae6d8e49c608aea4aJakub Hrozek
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina    ev = tevent_context_init(tmp_ctx);
91b0592cdab22915dff27ceae6d8e49c608aea4aJakub Hrozek    if (ev == NULL) {
91b0592cdab22915dff27ceae6d8e49c608aea4aJakub Hrozek        DEBUG(SSSDBG_CRIT_FAILURE, "Could not init tevent context\n");
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina        ret = ENOMEM;
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina        goto done;
91b0592cdab22915dff27ceae6d8e49c608aea4aJakub Hrozek    }
91b0592cdab22915dff27ceae6d8e49c608aea4aJakub Hrozek
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina    tcurl_ctx = tcurl_init(tmp_ctx, ev);
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina    if (tcurl_ctx == NULL) {
91b0592cdab22915dff27ceae6d8e49c608aea4aJakub Hrozek        DEBUG(SSSDBG_FATAL_FAILURE, "Could not init tcurl context\n");
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina        ret = ENOMEM;
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina        goto done;
91b0592cdab22915dff27ceae6d8e49c608aea4aJakub Hrozek    }
91b0592cdab22915dff27ceae6d8e49c608aea4aJakub Hrozek
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina    for (i = 0; requests[i] != NULL; i++) {
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina        req = tcurl_request_send(tmp_ctx, ev, tcurl_ctx, requests[i], 5);
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina        if (req == NULL) {
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina            DEBUG(SSSDBG_FATAL_FAILURE, "Could not create tevent request\n");
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina            ret = ENOMEM;
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina            goto done;
300b9e9217ee1ed8d845ed2370c5ccf5c87afb36Pavel Březina        }
300b9e9217ee1ed8d845ed2370c5ccf5c87afb36Pavel Březina
91b0592cdab22915dff27ceae6d8e49c608aea4aJakub Hrozek        tevent_req_set_callback(req, request_done, tool_ctx);
91b0592cdab22915dff27ceae6d8e49c608aea4aJakub Hrozek    }
91b0592cdab22915dff27ceae6d8e49c608aea4aJakub Hrozek
91b0592cdab22915dff27ceae6d8e49c608aea4aJakub Hrozek    while (tool_ctx->done == false) {
91b0592cdab22915dff27ceae6d8e49c608aea4aJakub Hrozek        tevent_loop_once(ev);
91b0592cdab22915dff27ceae6d8e49c608aea4aJakub Hrozek    }
91b0592cdab22915dff27ceae6d8e49c608aea4aJakub Hrozek
91b0592cdab22915dff27ceae6d8e49c608aea4aJakub Hrozek    if (tool_ctx->nreqs > 0) {
91b0592cdab22915dff27ceae6d8e49c608aea4aJakub Hrozek        DEBUG(SSSDBG_FATAL_FAILURE,
91b0592cdab22915dff27ceae6d8e49c608aea4aJakub Hrozek              "The tool finished with some pending requests, fail!\n");
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina        ret = EEXIST;
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina        goto done;
91b0592cdab22915dff27ceae6d8e49c608aea4aJakub Hrozek    }
91b0592cdab22915dff27ceae6d8e49c608aea4aJakub Hrozek
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina    ret = EOK;
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina
b800a6d09244359959404aca81c6796a58cafbcbPavel Březinadone:
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina    talloc_free(tmp_ctx);
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina    return ret;
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina}
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina
b800a6d09244359959404aca81c6796a58cafbcbPavel Březinaint main(int argc, const char *argv[])
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina{
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina    struct tool_options opts = { 0 };
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina    struct tool_ctx *tool_ctx;
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina    struct tcurl_request **requests;
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina    poptContext pc;
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina    errno_t ret;
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina    struct poptOption long_options[] = {
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina        POPT_AUTOHELP
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina        { "debug", '\0', POPT_ARG_INT, &opts.debug, 0, "The debug level to run with", NULL },
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina        { "socket-path", 's', POPT_ARG_STRING, &opts.socket_path, 0, "The path to the HTTP server socket", NULL },
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina        { "get", 'g', POPT_ARG_NONE, NULL, 'g', "Perform a HTTP GET (default)", NULL },
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina        { "put", 'p', POPT_ARG_NONE, NULL, 'p', "Perform a HTTP PUT", NULL },
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina        { "post", 'o', POPT_ARG_NONE, NULL, 'o', "Perform a HTTP POST", NULL },
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina        { "del", 'd', POPT_ARG_NONE, NULL, 'd', "Perform a HTTP DELETE", NULL },
d1ed11fc50922aab2332758a9300f3fbf814f112Pavel Březina#ifdef POPT_ARG_ARGV
d1ed11fc50922aab2332758a9300f3fbf814f112Pavel Březina        { "header", 'h', POPT_ARG_ARGV, &opts.headers, '\0', "Add HTTP header", NULL },
d1ed11fc50922aab2332758a9300f3fbf814f112Pavel Březina#endif
36e49a842e257ac9bde71728ee3bef4299b6e6e2Pavel Březina        { "raw", 'r', POPT_ARG_NONE, &opts.raw, '\0', "Print raw protocol output", NULL },
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina        { "verbose", 'v', POPT_ARG_NONE, &opts.verbose, '\0', "Print response code and body", NULL },
886e0f75e6f4c7877a23a3625f8a20c09109b09dPavel Březina        /* TLS */
886e0f75e6f4c7877a23a3625f8a20c09109b09dPavel Březina        { "tls", '\0', POPT_ARG_NONE, &opts.tls, '\0', "Enable TLS", NULL },
886e0f75e6f4c7877a23a3625f8a20c09109b09dPavel Březina        { "verify-peer", '\0', POPT_ARG_NONE, &opts.verify_peer, '\0', "Verify peer when TLS is enabled", NULL },
886e0f75e6f4c7877a23a3625f8a20c09109b09dPavel Březina        { "verify-host", '\0', POPT_ARG_NONE, &opts.verify_host, '\0', "Verify host when TLS is enabled", NULL },
886e0f75e6f4c7877a23a3625f8a20c09109b09dPavel Březina        { "capath", '\0', POPT_ARG_STRING, &opts.capath, '\0', "Path to CA directory where peer certificate is stored", NULL },
886e0f75e6f4c7877a23a3625f8a20c09109b09dPavel Březina        { "cacert", '\0', POPT_ARG_STRING, &opts.cacert, '\0', "Path to CA certificate", NULL },
ae6b11229d9961e26922918183c7c1de7780b8d6Pavel Březina        { "clientcert", '\0', POPT_ARG_STRING, &opts.clientcert, '\0', "Path to client's certificate", NULL },
ae6b11229d9961e26922918183c7c1de7780b8d6Pavel Březina        { "clientkey", '\0', POPT_ARG_STRING, &opts.clientkey, '\0', "Path to client's private key", NULL },
c2ea75da72b426d98ba489039e220d417bfb4c2aPavel Březina        /* BASIC AUTH */
c2ea75da72b426d98ba489039e220d417bfb4c2aPavel Březina        { "username", '\0', POPT_ARG_STRING, &opts.username, '\0', "Username for basic authentication", NULL },
c2ea75da72b426d98ba489039e220d417bfb4c2aPavel Březina        { "password", '\0', POPT_ARG_STRING, &opts.password, '\0', "Password for basic authentication", NULL },
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina        POPT_TABLEEND
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina    };
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina    pc = poptGetContext(NULL, argc, argv, long_options, 0);
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina    poptSetOtherOptionHelp(pc, "[URL HTTPDATA]*");
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina    tool_ctx = talloc_zero(NULL, struct tool_ctx);
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina    if (tool_ctx == NULL) {
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina        DEBUG(SSSDBG_CRIT_FAILURE, "Could not init tool context\n");
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina        ret = ENOMEM;
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina        goto done;
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina    }
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina    ret = parse_options(pc, &opts);
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina    if (ret != EOK) {
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina        DEBUG(SSSDBG_FATAL_FAILURE, "Unable to parse options [%d]: %s\n",
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina              ret, sss_strerror(ret));
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina        goto done;
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina    }
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina    DEBUG_CLI_INIT(opts.debug);
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina    tool_ctx->verbose = opts.verbose;
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina    ret = prepare_requests(tool_ctx, pc, &opts, &requests, &tool_ctx->nreqs);
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina    if (ret != EOK) {
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina        DEBUG(SSSDBG_FATAL_FAILURE, "Unable to prepare requests [%d]: %s\n",
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina              ret, sss_strerror(ret));
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina        goto done;
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina    }
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina    ret = run_requests(tool_ctx, requests);
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina    if (ret != EOK) {
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina        DEBUG(SSSDBG_FATAL_FAILURE, "Unable to issue requests [%d]: %s\n",
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina              ret, sss_strerror(ret));
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina        goto done;
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina    }
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina
b800a6d09244359959404aca81c6796a58cafbcbPavel Březinadone:
91b0592cdab22915dff27ceae6d8e49c608aea4aJakub Hrozek    talloc_free(tool_ctx);
91b0592cdab22915dff27ceae6d8e49c608aea4aJakub Hrozek    poptFreeContext(pc);
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina    if (ret != EOK) {
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina        return EXIT_FAILURE;
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina    }
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina
b800a6d09244359959404aca81c6796a58cafbcbPavel Březina    return EXIT_SUCCESS;
91b0592cdab22915dff27ceae6d8e49c608aea4aJakub Hrozek}