tests/dns.c

   1 /*
   2  * Copyright (c) 2006-2009 Bjorn Andersson <flex@kryo.se>, Erik Ekman <yarrick@kryo.se>
   3  *
   4  * Permission to use, copy, modify, and distribute this software for any
   5  * purpose with or without fee is hereby granted, provided that the above
   6  * copyright notice and this permission notice appear in all copies.
   7  *
   8  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
   9  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
  10  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
  11  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
  12  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
  13  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
  14  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
  15  */
  16
  17 #include <check.h>
  18 #include <stdio.h>
  19 #include <stdlib.h>
  20 #include <string.h>
  21 #include <ctype.h>
  22 #include <sys/types.h>
  23 #include <sys/socket.h>
  24 #include <netinet/in.h>
  25 #include <sys/stat.h>
  26 #include <arpa/nameser.h>
  27
  28 #include "common.h"
  29 #include "dns.h"
  30 #include "encoding.h"
  31 #include "base32.h"
  32 #include "test.h"
  33
  34 static void dump_packet(char *, size_t);
  35
  36 static char query_packet[] =
  37         "\x05\x39\x01\x00\x00\x01\x00\x00\x00\x00\x00\x01\x2D\x41\x6A\x62\x63"
  38         "\x75\x79\x74\x63\x70\x65\x62\x30\x67\x71\x30\x6C\x74\x65\x62\x75\x78"
  39         "\x67\x69\x64\x75\x6E\x62\x73\x73\x61\x33\x64\x66\x6F\x6E\x30\x63\x61"
  40         "\x7A\x64\x62\x6F\x72\x71\x71\x04\x6B\x72\x79\x6F\x02\x73\x65\x00\x00"
  41         "\x0A\x00\x01\x00\x00\x29\x10\x00\x00\x00\x80\x00\x00\x00";
  42
  43 static char answer_packet[] =
  44         "\x05\x39\x84\x00\x00\x01\x00\x01\x00\x00\x00\x00\x05\x73\x69\x6C\x6C"
  45         "\x79\x04\x68\x6F\x73\x74\x02\x6F\x66\x06\x69\x6F\x64\x69\x6E\x65\x04"
  46         "\x63\x6F\x64\x65\x04\x6B\x72\x79\x6F\x02\x73\x65\x00\x00\x0A\x00\x01"
  47         "\xC0\x0C\x00\x0A\x00\x01\x00\x00\x00\x00\x00\x23\x74\x68\x69\x73\x20"
  48         "\x69\x73\x20\x74\x68\x65\x20\x6D\x65\x73\x73\x61\x67\x65\x20\x74\x6F"
  49         "\x20\x62\x65\x20\x64\x65\x6C\x69\x76\x65\x72\x65\x64";
  50
  51 static char answer_packet_high_trans_id[] =
  52         "\x85\x39\x84\x00\x00\x01\x00\x01\x00\x00\x00\x00\x05\x73\x69\x6C\x6C"
  53         "\x79\x04\x68\x6F\x73\x74\x02\x6F\x66\x06\x69\x6F\x64\x69\x6E\x65\x04"
  54         "\x63\x6F\x64\x65\x04\x6B\x72\x79\x6F\x02\x73\x65\x00\x00\x0A\x00\x01"
  55         "\xC0\x0C\x00\x0A\x00\x01\x00\x00\x00\x00\x00\x23\x74\x68\x69\x73\x20"
  56         "\x69\x73\x20\x74\x68\x65\x20\x6D\x65\x73\x73\x61\x67\x65\x20\x74\x6F"
  57         "\x20\x62\x65\x20\x64\x65\x6C\x69\x76\x65\x72\x65\x64";
  58 static char *msgData = "this is the message to be delivered";
  59 static char *topdomain = "kryo.se";
  60
  61 static char *innerData = "HELLO this is the test data";
  62
  63 START_TEST(test_encode_query)
  64 {
  65         char buf[512];
  66         char resolv[512];
  67         struct query q;
  68         struct encoder *enc;
  69         char *d;
  70         size_t len;
  71         int ret;
  72
  73         len = sizeof(buf);
  74         memset(&buf, 0, sizeof(buf));
  75         memset(&resolv, 0, sizeof(resolv));
  76         memset(&q, 0, sizeof(struct query));
  77         q.type = T_NULL;
  78         q.id = 1337;
  79         d = resolv;
  80         enc = get_base32_encoder();
  81
  82         *d++ = 'A';
  83         enc->encode(d, &len, innerData, strlen(innerData));
  84         d = resolv + strlen(resolv);
  85         if (*d != '.') {
  86                 *d++ = '.';
  87         }
  88         strcpy(d, topdomain);
  89         ret = dns_encode(buf, len, &q, QR_QUERY, resolv, strlen(resolv));
  90         len = sizeof(query_packet) - 1; /* Skip extra null character */
  91
  92         if (strncmp(query_packet, buf, sizeof(query_packet)) || ret != len) {
  93                 printf("\n");
  94                 dump_packet(query_packet, len);
  95                 dump_packet(buf, ret);
  96         }
  97         fail_unless(strncmp(query_packet, buf, sizeof(query_packet)) == 0, "Did not compile expected packet");
  98         fail_unless(ret == len, "Bad packet length: %d, expected %d", ret, len);
  99 }
 100 END_TEST
 101
 102 START_TEST(test_decode_query)
 103 {
 104         char buf[512];
 105         char *domain;
 106         struct query q;
 107         struct encoder *enc;
 108         size_t len;
 109
 110         memset(&q, 0, sizeof(struct query));
 111         memset(&buf, 0, sizeof(buf));
 112         q.id = 0;
 113         len = sizeof(query_packet) - 1;
 114         enc = get_base32_encoder();
 115
 116         dns_decode(buf, sizeof(buf), &q, QR_QUERY, query_packet, len);
 117         domain = strstr(q.name, topdomain);
 118         len = sizeof(buf);
 119         unpack_data(buf, len, &(q.name[1]), (int) (domain - q.name) - 1, enc);
 120
 121         fail_unless(strncmp(buf, innerData, strlen(innerData)) == 0, "Did not extract expected host: '%s'", buf);
 122         fail_unless(strlen(buf) == strlen(innerData), "Bad host length: %d, expected %d: '%s'", strlen(buf), strlen(innerData), buf);
 123 }
 124 END_TEST
 125
 126 START_TEST(test_encode_response)
 127 {
 128         char buf[512];
 129         char *host = "silly.host.of.iodine.code.kryo.se";
 130         struct query q;
 131         int len;
 132         int ret;
 133
 134         len = sizeof(buf);
 135         memset(&buf, 0, sizeof(buf));
 136         memset(&q, 0, sizeof(struct query));
 137         strncpy(q.name, host, strlen(host));
 138         q.type = T_NULL;
 139         q.id = 1337;
 140
 141         ret = dns_encode(buf, len, &q, QR_ANSWER, msgData, strlen(msgData));
 142         len = sizeof(answer_packet) - 1; /* Skip extra null character */
 143
 144         fail_unless(strncmp(answer_packet, buf, sizeof(answer_packet)) == 0, "Did not compile expected packet");
 145         fail_unless(ret == len, "Bad packet length: %d, expected %d", ret, len);
 146 }
 147 END_TEST
 148
 149 START_TEST(test_decode_response)
 150 {
 151         char buf[512];
 152         struct query q;
 153         int len;
 154         int ret;
 155
 156         len = sizeof(buf);
 157         memset(&buf, 0, sizeof(buf));
 158
 159         ret = dns_decode(buf, len, &q, QR_ANSWER, answer_packet, sizeof(answer_packet)-1);
 160         fail_unless(strncmp(msgData, buf, sizeof(msgData)) == 0, "Did not extract expected data");
 161         fail_unless(ret == strlen(msgData), "Bad data length: %d, expected %d", ret, strlen(msgData));
 162         fail_unless(q.id == 0x0539);
 163 }
 164 END_TEST
 165
 166 START_TEST(test_decode_response_with_high_trans_id)
 167 {
 168         char buf[512];
 169         struct query q;
 170         int len;
 171         int ret;
 172
 173         len = sizeof(buf);
 174         memset(&buf, 0, sizeof(buf));
 175
 176         ret = dns_decode(buf, len, &q, QR_ANSWER, answer_packet_high_trans_id, sizeof(answer_packet_high_trans_id)-1);
 177         fail_unless(strncmp(msgData, buf, sizeof(msgData)) == 0, "Did not extract expected data");
 178         fail_unless(ret == strlen(msgData), "Bad data length: %d, expected %d", ret, strlen(msgData));
 179         fail_unless(q.id == 0x8539, "q.id was %08X instead of %08X!", q.id, 0x8539);
 180 }
 181 END_TEST
 182
 183 START_TEST(test_get_id_short_packet)
 184 {
 185         char buf[5];
 186         int len;
 187         unsigned short id;
 188
 189         len = sizeof(buf);
 190         memset(&buf, 5, sizeof(buf));
 191
 192         id = dns_get_id(buf, len);
 193         fail_unless(id == 0);
 194 }
 195 END_TEST
 196
 197 START_TEST(test_get_id_low)
 198 {
 199         unsigned short id;
 200
 201         id = dns_get_id(answer_packet, sizeof(answer_packet));
 202         fail_unless(id == 1337);
 203 }
 204 END_TEST
 205
 206 START_TEST(test_get_id_high)
 207 {
 208         unsigned short id;
 209
 210         id = dns_get_id(answer_packet_high_trans_id, sizeof(answer_packet_high_trans_id));
 211         fail_unless(id == 0x8539);
 212 }
 213 END_TEST
 214
 215 static void
 216 dump_packet(char *buf, size_t len)
 217 {
 218         int pos;
 219
 220         for (pos = 0; pos < len; pos++) {
 221                 printf("\\x%02X", (unsigned char) buf[pos]);
 222         }
 223         printf("\n");
 224         for (pos = 0; pos < len; pos++) {
 225                 if (isalnum((unsigned char) buf[pos])) {
 226                         printf(" %c  ", (unsigned char) buf[pos]);
 227                 } else {
 228                         printf("    ");
 229                 }
 230         }
 231         printf("\n");
 232 }
 233
 234 TCase *
 235 test_dns_create_tests()
 236 {
 237         TCase *tc;
 238
 239         tc = tcase_create("Dns");
 240         tcase_add_test(tc, test_encode_query);
 241         tcase_add_test(tc, test_decode_query);
 242         tcase_add_test(tc, test_encode_response);
 243         tcase_add_test(tc, test_decode_response);
 244         tcase_add_test(tc, test_decode_response_with_high_trans_id);
 245         tcase_add_test(tc, test_get_id_short_packet);
 246         tcase_add_test(tc, test_get_id_low);
 247         tcase_add_test(tc, test_get_id_high);
 248
 249         return tc;
 250 }