#include <stdio.h>

#include <stdbool.h>

#include <errno.h>

#include <string.h>

#include <assert.h>

#include "util.h"

#include "macro.h"

#include "dhcp-protocol.h"

#include "dhcp-internal.h"

struct option_desc {

uint8_t sname[64];

int snamelen;

uint8_t file[128];

int filelen;

uint8_t options[128];

int len;

bool success;

int filepos;

int snamepos;

int pos;

};

static bool verbose = false;

static struct option_desc option_tests[] = {

{ {}, 0, {}, 0, { 42, 5, 65, 66, 67, 68, 69 }, 7, false, },

{ {}, 0, {}, 0, { 42, 5, 65, 66, 67, 68, 69, 0, 0,

DHCP_OPTION_MESSAGE_TYPE, 1, DHCP_ACK }, 12, true, },

{ {}, 0, {}, 0, { 8, 255, 70, 71, 72 }, 5, false, },

{ {}, 0, {}, 0, { 0x35, 0x01, 0x05, 0x36, 0x04, 0x01, 0x00, 0xa8,

0xc0, 0x33, 0x04, 0x00, 0x01, 0x51, 0x80, 0x01,

0x04, 0xff, 0xff, 0xff, 0x00, 0x03, 0x04, 0xc0,

0xa8, 0x00, 0x01, 0x06, 0x04, 0xc0, 0xa8, 0x00,

0x01, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, },

40, true, },

{ {}, 0, {}, 0, { DHCP_OPTION_MESSAGE_TYPE, 1, DHCP_OFFER,

42, 3, 0, 0, 0 }, 8, true, },

{ {}, 0, {}, 0, { 42, 2, 1, 2, 44 }, 5, false, },

{ {}, 0,

{ 222, 3, 1, 2, 3, DHCP_OPTION_MESSAGE_TYPE, 1, DHCP_NAK }, 8,

{ DHCP_OPTION_OVERLOAD, 1, DHCP_OVERLOAD_FILE }, 3, true, },

{ { 1, 4, 1, 2, 3, 4, DHCP_OPTION_MESSAGE_TYPE, 1, DHCP_ACK }, 9,

{ 222, 3, 1, 2, 3 }, 5,

{ DHCP_OPTION_OVERLOAD, 1,

DHCP_OVERLOAD_FILE|DHCP_OVERLOAD_SNAME }, 3, true, },

};

static const char *dhcp_type(int type)

{

switch(type) {

case DHCP_DISCOVER:

return "DHCPDISCOVER";

case DHCP_OFFER:

return "DHCPOFFER";

case DHCP_REQUEST:

return "DHCPREQUEST";

case DHCP_DECLINE:

return "DHCPDECLINE";

case DHCP_ACK:

return "DHCPACK";

case DHCP_NAK:

return "DHCPNAK";

case DHCP_RELEASE:

return "DHCPRELEASE";

default:

return "unknown";

}

}

static void test_invalid_buffer_length(void)

{

DHCPMessage message;

assert_se(dhcp_option_parse(&message, 0, NULL, NULL) == -EINVAL);

assert_se(dhcp_option_parse(&message, sizeof(DHCPMessage) - 1, NULL, NULL)

== -EINVAL);

}

static void test_message_init(void)

{

_cleanup_free_ DHCPMessage *message = NULL;

size_t optlen = 4, optoffset;

size_t len = sizeof(DHCPMessage) + optlen;

uint8_t *magic;

message = malloc0(len);

assert_se(dhcp_message_init(message, BOOTREQUEST, 0x12345678,

DHCP_DISCOVER, ARPHRD_ETHER, optlen, &optoffset) >= 0);

assert_se(message->xid == htobe32(0x12345678));

assert_se(message->op == BOOTREQUEST);

magic = (uint8_t*)&message->magic;

assert_se(magic[0] == 99);

assert_se(magic[1] == 130);

assert_se(magic[2] == 83);

assert_se(magic[3] == 99);

assert_se(dhcp_option_parse(message, len, NULL, NULL) >= 0);

}

static DHCPMessage *create_message(uint8_t *options, uint16_t optlen,

uint8_t *file, uint8_t filelen,

uint8_t *sname, uint8_t snamelen)

{

DHCPMessage *message;

size_t len = sizeof(DHCPMessage) + optlen;

message = malloc0(len);

assert_se(message);

if (options && optlen)

memcpy(&message->options, options, optlen);

if (file && filelen <= 128)

memcpy(&message->file, file, filelen);

if (sname && snamelen <= 64)

memcpy(&message->sname, sname, snamelen);

return message;

}

static void test_ignore_opts(uint8_t *descoption, int *descpos, int *desclen)

{

while (*descpos < *desclen) {

switch(descoption[*descpos]) {

case DHCP_OPTION_PAD:

*descpos += 1;

break;

case DHCP_OPTION_MESSAGE_TYPE:

case DHCP_OPTION_OVERLOAD:

*descpos += 3;

break;

default:

return;

}

}

}

static int test_options_cb(uint8_t code, uint8_t len, const uint8_t *option,

void *user_data)

{

struct option_desc *desc = user_data;

uint8_t *descoption = NULL;

int *desclen = NULL, *descpos = NULL;

uint8_t optcode = 0;

uint8_t optlen = 0;

uint8_t i;

assert_se((!desc && !code && !len) || desc);

if (!desc)

return -EINVAL;

assert_se(code != DHCP_OPTION_PAD);

assert_se(code != DHCP_OPTION_END);

assert_se(code != DHCP_OPTION_MESSAGE_TYPE);

assert_se(code != DHCP_OPTION_OVERLOAD);

while (desc->pos >= 0 || desc->filepos >= 0 || desc->snamepos >= 0) {

if (desc->pos >= 0) {

descoption = &desc->options[0];

desclen = &desc->len;

descpos = &desc->pos;

} else if (desc->filepos >= 0) {

descoption = &desc->file[0];

desclen = &desc->filelen;

descpos = &desc->filepos;

} else if (desc->snamepos >= 0) {

descoption = &desc->sname[0];

desclen = &desc->snamelen;

descpos = &desc->snamepos;

}

assert_se(descoption && desclen && descpos);

if (*desclen)

test_ignore_opts(descoption, descpos, desclen);

if (*descpos < *desclen)

break;

if (*descpos == *desclen)

*descpos = -1;

}

assert_se(descpos);

assert_se(*descpos != -1);

optcode = descoption[*descpos];

optlen = descoption[*descpos + 1];

if (verbose)

printf("DHCP code %2d(%2d) len %2d(%2d) ", code, optcode,

len, optlen);

assert_se(code == optcode);

assert_se(len == optlen);

for (i = 0; i < len; i++) {

if (verbose)

printf("0x%02x(0x%02x) ", option[i],

descoption[*descpos + 2 + i]);

assert_se(option[i] == descoption[*descpos + 2 + i]);

}

if (verbose)

printf("\n");

*descpos += optlen + 2;

test_ignore_opts(descoption, descpos, desclen);

if (desc->pos != -1 && desc->pos == desc->len)

desc->pos = -1;

if (desc->filepos != -1 && desc->filepos == desc->filelen)

desc->filepos = -1;

if (desc->snamepos != -1 && desc->snamepos == desc->snamelen)

desc->snamepos = -1;

return 0;

}

static void test_options(struct option_desc *desc)

{

uint8_t *options = NULL;

uint8_t *file = NULL;

uint8_t *sname = NULL;

int optlen = 0;

int filelen = 0;

int snamelen = 0;

int buflen = 0;

_cleanup_free_ DHCPMessage *message = NULL;

int res;

if (desc) {

file = &desc->file[0];

filelen = desc->filelen;

if (!filelen)

desc->filepos = -1;

sname = &desc->sname[0];

snamelen = desc->snamelen;

if (!snamelen)

desc->snamepos = -1;

options = &desc->options[0];

optlen = desc->len;

desc->pos = 0;

}

message = create_message(options, optlen, file, filelen,

sname, snamelen);

buflen = sizeof(DHCPMessage) + optlen;

if (!desc) {

assert_se((res = dhcp_option_parse(message, buflen,

test_options_cb,

NULL)) == -ENOMSG);

} else if (desc->success) {

assert_se((res = dhcp_option_parse(message, buflen,

test_options_cb,

desc)) >= 0);

assert_se(desc->pos == -1 && desc->filepos == -1 &&

desc->snamepos == -1);

} else

assert_se((res = dhcp_option_parse(message, buflen,

test_options_cb,

desc)) < 0);

if (verbose)

printf("DHCP type %s\n", dhcp_type(res));

}

static uint8_t options[64] = {

'A', 'B', 'C', 'D',

160, 2, 0x11, 0x12,

0,

31, 8, 0x31, 0x32, 0x33, 0x34, 0x35, 0x36, 0x37, 0x38,

0,

55, 3, 0x51, 0x52, 0x53,

17, 7, 0x71, 0x72, 0x73, 0x74, 0x75, 0x76, 0x77,

255

};

static void test_option_set(void)

{

_cleanup_free_ DHCPMessage *result = NULL;

size_t offset = 0, len, pos;

unsigned i;

result = malloc0(sizeof(DHCPMessage) + 11);

assert_se(result);

result->options[0] = 'A';

result->options[1] = 'B';

result->options[2] = 'C';

result->options[3] = 'D';

assert_se(dhcp_option_append(result, 0, &offset, 0, DHCP_OPTION_PAD,

0, NULL) == -ENOBUFS);

assert_se(offset == 0);

offset = 4;

assert_se(dhcp_option_append(result, 5, &offset, 0, DHCP_OPTION_PAD,

0, NULL) == -ENOBUFS);

assert_se(offset == 4);

assert_se(dhcp_option_append(result, 6, &offset, 0, DHCP_OPTION_PAD,

0, NULL) >= 0);

assert_se(offset == 5);

offset = pos = 4;

len = 11;

while (pos < len && options[pos] != DHCP_OPTION_END) {

assert_se(dhcp_option_append(result, len, &offset, DHCP_OVERLOAD_SNAME,

options[pos],

options[pos + 1],

&options[pos + 2]) >= 0);

if (options[pos] == DHCP_OPTION_PAD)

pos++;

else

pos += 2 + options[pos + 1];

if (pos < len)

assert_se(offset == pos);

}

for (i = 0; i < 9; i++) {

if (verbose)

printf("%2u: 0x%02x(0x%02x) (options)\n", i, result->options[i],

options[i]);

assert_se(result->options[i] == options[i]);

}

if (verbose)

printf("%2d: 0x%02x(0x%02x) (options)\n", 9, result->options[9],

DHCP_OPTION_END);

assert_se(result->options[9] == DHCP_OPTION_END);

if (verbose)

printf("%2d: 0x%02x(0x%02x) (options)\n", 10, result->options[10],

DHCP_OPTION_PAD);

assert_se(result->options[10] == DHCP_OPTION_PAD);

for (i = 0; i < pos - 8; i++) {

if (verbose)

printf("%2u: 0x%02x(0x%02x) (sname)\n", i, result->sname[i],

options[i + 9]);

assert_se(result->sname[i] == options[i + 9]);

}

if (verbose)

printf ("\n");

}

int main(int argc, char *argv[])

{

unsigned int i;

test_invalid_buffer_length();

test_message_init();

test_options(NULL);

for (i = 0; i < ELEMENTSOF(option_tests); i++)

test_options(&option_tests[i]);

test_option_set();

return 0;

}