ks_test.c

Go to the documentation of this file.

/*
 * libkarmaclient - A C Library to the Karmasphere Reputation Server
 * Copyright (C) 2006 Karmasphere <http://www.karmasphere.com/>
 *  - Shevek <shevek@karmasphere.com>
 *  - Dave Stewart <dave.stewart@karmasphere.com>
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) any later version.
 *
 * This library is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this library; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307  USA
 */

#include "ks_config.h"

#ifdef HAVE_STDIO_H
#include <stdio.h>
#endif

#ifdef HAVE_STRING_H
#include <string.h>
#endif

#ifdef HAVE_NETINET_IN_H
#include <netinet/in.h>
#endif

#include "ks_bquery.h"
#include "ks_response.h"
#include "ks_bparse.h"

static const char* tests[] = {
        /* these next few should pass */
        "32:Start of tests which should pass",
        "15:aaabbbcccdddeeefff",
        "i35e",
        "le", 
        "de",
        "li1ei2ei3ei4ee",
        "d3:foo4:arsee",
        "35: f305=-1*1.0 f306=-1*1.0 f531=1*1.0",
        "d1:d35: f305=-1*1.0 f306=-1*1.0 f531=1*1.01:vi-1ee",
        "d9:combinersd7:defaultd1:d35: f305=-1*1.0 f306=-1*1.0 f531=1*1.01:vi-1eee5:factsld1:d42:http://www.mail-abuse.org/cgi-bin/lookup?$1:fi305e1:vi-1eed1:d42:http://www.mail-abuse.org/cgi-bin/lookup?$1:fi306e1:vi-1eed1:d31:Listed in trusted-forwarder.org1:fi531e1:vi1eee4:timei18ee",
        "30:End of tests which should pass",

        /* these next few should fail */
        "",
        "ddee",
        "ddd",
        "lll",
        "ie", 
        "aa",
};

static void
test_parser()
{
        ks_string_t             *p;
        ks_base_t               *b;
        unsigned int     i;

        for (i = 0; i < (sizeof(tests) / sizeof(tests[0])); i++) {
                printf("Parse %s\n", tests[i]);
                b = ks_bparse(tests[i], strlen(tests[i]));
                printf("Parsed\n");
                if (b == NULL)
                        continue;
                p = ks_string_new("", 0);
                printf("Appending...\n");
                ks_bappend(p, b);
                printf("Appended\n");
                printf("Got %s\n\n", ks_string_get(p));
                ks_destroy((ks_base_t *)b);
                ks_destroy((ks_base_t *)p);
        }
}

static void
test_query(const char *id)
{
        ks_bquery_t             *q;
        ks_string_t             *p;

        q = ks_bquery_new();

        ks_bquery_feed_add(q, "cymru.bogons");
        ks_bquery_feed_add(q, "spamhaus.sbl");

        ks_bquery_combiner_add(q, "default");

        ks_bquery_identity_add(q, "fred@foo.com", KS_IDT_EMAIL_ADDRESS);
        ks_bquery_identity_add(q, "123.45.6.7", KS_IDT_IP4_ADDRESS);

        if (id)
                ks_bquery_id_set(q, id, strlen(id));
        ks_bquery_flags_set(q, 123);

        p = ks_bquery_packet(q);

        /* Send this packet over the network. */
        printf("Packet is %s\n", ks_string_get(p));

        ks_destroy((ks_base_t *)q);
        ks_destroy((ks_base_t *)p);
}

static void
test_response()
{
        ks_response_t   *e;
        ks_combination_t*c;
        ks_fact_t               *f;
        const char              *s;
        int                              i;

        /* This packet was received over the network. */
        s = "d9:combinersd7:defaultd1:d35: f305=-1*1.0 f306=-1*1.0 f531=1*1.01:vi-1eee5:factsld1:d42:http://www.mail-abuse.org/cgi-bin/lookup?$1:fi305e1:vi-1eed1:d42:http://www.mail-abuse.org/cgi-bin/lookup?$1:fi306e1:vi-1eed1:d31:Listed in trusted-forwarder.org1:fi531e1:vi1eee4:timei18ee";

        printf("Response data %s\n", s);
        e = ks_response_bparse(s, strlen(s));

        c = ks_response_combination(e, "default");
        printf("Score: %ld, data %s\n",
                                        ks_combination_score(c),
                                        ks_combination_data(c));

        for (i = 0; i < ks_response_facts_length(e); i++) {
                f = ks_response_fact(e, i);
                printf("fact %d: %ld, data %s\n",
                                                i,
                                                ks_fact_score(f),
                                                ks_fact_data(f));
        }

        ks_destroy((ks_base_t *)e);
}

int
main(void)
{
        test_parser();
        test_query("fooIDID");
        test_query(NULL);
        test_query(NULL);
        test_query(NULL);
        test_response();

        return 0;
}

---