#include <cutter.h>
#include <freefare.h>

#include "fixture.h"

static nfc_context *context;
static nfc_device *device = NULL;
static FreefareTag *tags = NULL;
FreefareTag tag = NULL;

void
cut_setup(void)
{
    int res;
    nfc_connstring devices[8];
    size_t device_count;

    nfc_init(&context);
    cut_assert_not_null(context, cut_message("Unable to init libnfc (malloc)"));

    device_count = nfc_list_devices(context, devices, 8);
    if (device_count <= 0)
	cut_omit("No device found");

    for (size_t i = 0; i < device_count; i++) {
	device = nfc_open(context, devices[i]);
	if (!device)
	    cut_omit("nfc_open() failed.");

	tags = freefare_get_tags(device);
	cut_assert_not_null(tags, cut_message("freefare_get_tags() failed"));

	tag = NULL;
	for (int i = 0; tags[i]; i++) {
	    if ((freefare_get_tag_type(tags[i]) == MIFARE_CLASSIC_1K) ||
		(freefare_get_tag_type(tags[i]) == MIFARE_CLASSIC_4K)) {
		tag = tags[i];
		res = mifare_classic_connect(tag);
		cut_assert_equal_int(0, res, cut_message("mifare_classic_connect() failed"));
		return;
	    }
	}
	nfc_close(device);
	device = NULL;
	freefare_free_tags(tags);
	tags = NULL;
    }
    cut_omit("No MIFARE Classic tag on NFC device");
}

void
cut_teardown(void)
{
    if (tag)
	mifare_classic_disconnect(tag);

    if (tags) {
	freefare_free_tags(tags);
	tags = NULL;
    }

    if (device)
	nfc_close(device);

    nfc_exit(context);
}