t_client.c

/* -*- mode: c; c-basic-offset: 4; indent-tabs-mode: nil -*- */
/* lib/krad/t_client.c - Client request test program */
/*
 * Copyright 2013 Red Hat, Inc.  All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions are met:
 *
 *    1. Redistributions of source code must retain the above copyright
 *       notice, this list of conditions and the following disclaimer.
 *
 *    2. Redistributions in binary form must reproduce the above copyright
 *       notice, this list of conditions and the following disclaimer in
 *       the documentation and/or other materials provided with the
 *       distribution.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS
 * IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
 * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A
 * PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER
 * OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
 * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
 * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
 * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
 * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
 * NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
 * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

#include "t_daemon.h"

#define EVENT_COUNT 4

static struct
{
    int count;
    struct event events[EVENT_COUNT];
} record;

static verto_ctx *vctx;

static void
callback(krb5_error_code retval, const krad_packet *request,
         const krad_packet *response, void *data)
{
    struct event *evt;

    evt = &record.events[record.count++];
    evt->error = retval != 0;
    if (evt->error)
        evt->result.retval = retval;
    else
        evt->result.code = krad_packet_get_code(response);
    verto_break(vctx);
}

int
main(int argc, const char **argv)
{
    krad_attrset *attrs;
    krad_client *rc;
    krb5_context kctx;
    krb5_data tmp;

    if (!daemon_start(argc, argv)) {
        fprintf(stderr, "Unable to start pyrad daemon, skipping test...\n");
        return 0;
    }

    noerror(krb5_init_context(&kctx));
    vctx = verto_new(NULL, VERTO_EV_TYPE_IO | VERTO_EV_TYPE_TIMEOUT);
    insist(vctx != NULL);
    noerror(krad_client_new(kctx, vctx, &rc));

    tmp = string2data("testUser");
    noerror(krad_attrset_new(kctx, &attrs));
    noerror(krad_attrset_add(attrs, KRAD_ATTR_USER_NAME, &tmp));

    /* Test accept. */
    tmp = string2data("accept");
    noerror(krad_attrset_add(attrs, KRAD_ATTR_USER_PASSWORD, &tmp));
    noerror(krad_client_send(rc, KRAD_CODE_ACCESS_REQUEST, attrs, "localhost",
                             "foo", 1000, 3, callback, NULL));
    verto_run(vctx);

    /* Test reject. */
    tmp = string2data("reject");
    krad_attrset_del(attrs, KRAD_ATTR_USER_PASSWORD, 0);
    noerror(krad_attrset_add(attrs, KRAD_ATTR_USER_PASSWORD, &tmp));
    noerror(krad_client_send(rc, KRAD_CODE_ACCESS_REQUEST, attrs, "localhost",
                             "foo", 1000, 3, callback, NULL));
    verto_run(vctx);

    /* Test timeout. */
    daemon_stop();
    noerror(krad_client_send(rc, KRAD_CODE_ACCESS_REQUEST, attrs, "localhost",
                             "foo", 1000, 3, callback, NULL));
    verto_run(vctx);

    /* Test outstanding packet freeing. */
    noerror(krad_client_send(rc, KRAD_CODE_ACCESS_REQUEST, attrs, "localhost",
                             "foo", 1000, 3, callback, NULL));
    krad_client_free(rc);
    rc = NULL;

    /* Verify the results. */
    insist(record.count == EVENT_COUNT);
    insist(record.events[0].error == FALSE);
    insist(record.events[0].result.code == KRAD_CODE_ACCESS_ACCEPT);
    insist(record.events[1].error == FALSE);
    insist(record.events[1].result.code == KRAD_CODE_ACCESS_REJECT);
    insist(record.events[2].error == TRUE);
    insist(record.events[2].result.retval == ETIMEDOUT);
    insist(record.events[3].error == TRUE);
    insist(record.events[3].result.retval == ECANCELED);

    krad_attrset_free(attrs);
    krad_client_free(rc);
    verto_free(vctx);
    krb5_free_context(kctx);
    return 0;
}