#include "incs.h"

/*
 * put/get/has attributes
 */

void test01(void)
{
	RADIUS_PACKET *packet;
	uint8_t buf[256];
	size_t len;
	const void *ptr;

	static const uint8_t data0[] = { 0xfe, 0xdc, 0xba, 0x98 };
	static const uint8_t data1[] = { 0x76, 0x54, 0x32 };
	static const uint8_t data2[] = { 0x0f, 0x1e, 0x2d, 0x3c };
	static const uint8_t data3[] = { 0x4b, 0x5a, 0x69 };
	static const uint8_t attrs[] = {
		10, 6, 0xfe, 0xdc, 0xba, 0x98,
		20, 5, 0x76, 0x54, 0x32,
		RADIUS_TYPE_VENDOR_SPECIFIC, 12, 0, 0, 300/256, 300%256, 40, 6, 0x0f, 0x1e, 0x2d, 0x3c,
		RADIUS_TYPE_VENDOR_SPECIFIC, 11, 0, 0, 500/256, 500%256, 60, 5, 0x4b, 0x5a, 0x69,
		20, 6, 0x0f, 0x1e, 0x2d, 0x3c,
		RADIUS_TYPE_VENDOR_SPECIFIC, 11, 0, 0, 300/256, 300%256, 40, 5, 0x76, 0x54, 0x32,
	};

	packet = radius_new_request_packet(RADIUS_CODE_ACCESS_REQUEST);

	radius_put_raw_attr(packet, 10, data0, sizeof(data0));
	radius_put_raw_attr(packet, 20, data1, sizeof(data1));
	radius_put_vs_raw_attr(packet, 300, 40, data2, sizeof(data2));
	radius_put_vs_raw_attr(packet, 500, 60, data3, sizeof(data3));
	radius_put_raw_attr(packet, 20, data2, sizeof(data2));
	radius_put_vs_raw_attr(packet, 300, 40, data1, sizeof(data1));

	CHECK(radius_get_length(packet) == sizeof(attrs) + 20);
	CHECK(memcmp(radius_get_data(packet) + 20, attrs, sizeof(attrs)) == 0);

	len = sizeof(buf);
	CHECK(radius_get_raw_attr(packet, 10, buf, &len) == 0);
	CHECK(len == sizeof(data0));
	CHECK(memcmp(buf, data0, len) == 0);

	len = sizeof(buf);
	CHECK(radius_get_raw_attr(packet, 20, buf, &len) == 0);
	CHECK(len == sizeof(data1));
	CHECK(memcmp(buf, data1, len) == 0);

	len = sizeof(buf);
	CHECK(radius_get_vs_raw_attr(packet, 300, 40, buf, &len) == 0);
	CHECK(len == sizeof(data2));
	CHECK(memcmp(buf, data2, len) == 0);

	len = sizeof(buf);
	CHECK(radius_get_vs_raw_attr(packet, 500, 60, buf, &len) == 0);
	CHECK(len == sizeof(data3));
	CHECK(memcmp(buf, data3, len) == 0);

	len = 2;
	CHECK(radius_get_raw_attr(packet, 10, buf, &len) == 0);
	CHECK(len == 2);
	CHECK(memcmp(buf, data0, len) == 0);

	len = 2;
	CHECK(radius_get_vs_raw_attr(packet, 300, 40, buf, &len) == 0);
	CHECK(len == 2);
	CHECK(memcmp(buf, data2, len) == 0);

	len = sizeof(buf);
	CHECK(radius_get_raw_attr(packet, 90, buf, &len) != 0);
	CHECK(radius_get_vs_raw_attr(packet, 300, 90, buf, &len) != 0);
	CHECK(radius_get_vs_raw_attr(packet, 900, 40, buf, &len) != 0);

	len = 0;
	CHECK(radius_get_raw_attr_ptr(packet, 10, &ptr, &len) == 0);
	CHECK(len == 4);
	CHECK(ptr == radius_get_data(packet) + 20 +  0 + 2);

	len = 0;
	CHECK(radius_get_vs_raw_attr_ptr(packet, 500, 60, &ptr, &len) == 0);
	CHECK(len == 3);
	CHECK(ptr == radius_get_data(packet) + 20 + 23 + 8);

	CHECK(radius_get_raw_attr_ptr(packet, 90, &ptr, &len) != 0);
	CHECK(radius_get_vs_raw_attr_ptr(packet, 300, 90, &ptr, &len) != 0);
	CHECK(radius_get_vs_raw_attr_ptr(packet, 900, 40, &ptr, &len) != 0);

	CHECK(radius_has_attr(packet, 10));
	CHECK(radius_has_attr(packet, 20));
	CHECK(radius_has_vs_attr(packet, 300, 40));
	CHECK(radius_has_vs_attr(packet, 500, 60));
	CHECK(!radius_has_attr(packet, 90));
	CHECK(!radius_has_vs_attr(packet, 300, 90));
	CHECK(!radius_has_vs_attr(packet, 900, 40));
}

ADD_TEST(test01)