#include "incs.h"

/*
 * put/get cat attributes
 */

void test04(void)
{
	RADIUS_PACKET *packet;
	uint8_t buf[1024];
	size_t len;
	uint8_t *p;

#define ATTRLEN (256 + 256 + 103)
	uint8_t data0[ATTRLEN];
	uint8_t data1[ATTRLEN];
	uint8_t data2[] = { 0x10, 0x20, 0x30, 0x40 };
	uint8_t data3[] = { 0x10, 0x20, 0x30, 0x40, 0x10, 0x20, 0x30, 0x40 };
	uint8_t attrs[2048];

	packet = radius_new_request_packet(RADIUS_CODE_ACCESS_REQUEST);

	for (int i = 0; i < ATTRLEN; i++)
		data0[i] = random();
	for (int i = 0; i < ATTRLEN; i++)
		data1[i] = random();

	p = attrs;
	*p++ = 20; *p++ = 6;
	memcpy(p, data2, 4); p += 4;

	*p++ = 10; *p++ = 255;
	memcpy(p, data0, 253); p += 253;
	*p++ = 10; *p++ = 255;
	memcpy(p, data0 + 253, 253); p += 253;
	*p++ = 10; *p++ = ATTRLEN-253*2+2;
	memcpy(p, data0 + 253*2, ATTRLEN-253*2); p += ATTRLEN-253*2;

	*p++ = RADIUS_TYPE_VENDOR_SPECIFIC; *p++ = 255;
	*p++ = 0; *p++ = 0; *p++ = 0; *p++ = 20; *p++ = 30; *p++ = 249;
	memcpy(p, data1, 247); p += 247;
	*p++ = RADIUS_TYPE_VENDOR_SPECIFIC; *p++ = 255;
	*p++ = 0; *p++ = 0; *p++ = 0; *p++ = 20; *p++ = 30; *p++ = 249;
	memcpy(p, data1 + 247, 247); p += 247;
	*p++ = RADIUS_TYPE_VENDOR_SPECIFIC; *p++ = ATTRLEN-247*2+8;
	*p++ = 0; *p++ = 0; *p++ = 0; *p++ = 20; *p++ = 30; *p++ = ATTRLEN-247*2+2;
	memcpy(p, data1 + 247*2, ATTRLEN-247*2); p += ATTRLEN-247*2;

	*p++ = 20; *p++ = 6;
	memcpy(p, data2, 4); p += 4;


	radius_put_raw_attr(packet, 20, data2, sizeof(data2));
	radius_put_raw_attr_cat(packet, 10, data0, ATTRLEN);
	radius_put_vs_raw_attr_cat(packet, 20, 30, data1, ATTRLEN);
	radius_put_raw_attr(packet, 20, data2, sizeof(data2));

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

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

	len = sizeof(buf);
	CHECK(radius_get_vs_raw_attr_cat(packet, 20, 30, buf, &len) == 0);
	CHECK(len == ATTRLEN);
	CHECK(memcmp(buf, data1, len) == 0);

	len = sizeof(buf);
	CHECK(radius_get_raw_attr_cat(packet, 20,buf, &len) == 0);
	CHECK(len == sizeof(data3));
	CHECK(memcmp(buf, data3, len) == 0);
}

ADD_TEST(test04)