/* vim: set tabstop=8 shiftwidth=4 softtabstop=4 expandtab smarttab colorcolumn=80: */
/*
 * Copyright (c) 2016 Red Hat, Inc.
 * Author: Nathaniel McCallum <npmccallum@redhat.com>
 *
 * This program is free software: you can redistribute it and/or modify it
 * under the terms of the GNU Lesser General Public License as published by
 * the Free Software Foundation, either version 2.1 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

#include "test.h"
#include <error.h>
#include <stdlib.h>
#include <string.h>

static const luksmeta_uuid_t UUID0 = {
    0x35, 0x08, 0x50, 0xc3, 0x25, 0xc9, 0x85, 0xea,
    0x1b, 0x55, 0x93, 0x56, 0x36, 0x2a, 0xd9, 0x85
};

static const luksmeta_uuid_t UUID1 = {
    0xb4, 0xcb, 0x8c, 0x1c, 0x34, 0xea, 0xcc, 0x21,
    0x0b, 0x9c, 0xc3, 0x9c, 0x9a, 0x09, 0xc0, 0x0f
};

int
main(int argc, char *argv[])
{
    uint8_t data[sizeof(UUID0)] = {};
    struct crypt_device *cd = NULL;
    luksmeta_uuid_t uuid = {};
    uint32_t offset = 0;
    uint32_t length = 0;
    int r;

    crypt_free(test_format());
    cd = test_init();
    test_hole(cd, &offset, &length);

    /* Add one metadata. */
    r = luksmeta_save(cd, 0, UUID0, UUID0, sizeof(UUID0));
    if (r < 0)
        error(EXIT_FAILURE, -r, "luksmeta_save()");

    assert(test_layout((range_t[]) {
        { 0, 1024 },                   /* LUKS header */
        { 1024, 3072, true },          /* Keyslot Area */
        { offset, 4096 },              /* luksmeta header */
        { offset + 4096, 4096 },       /* luksmeta slot 0 */
        END(offset + 8192),            /* Rest of the file */
    }));

    assert(luksmeta_load(cd, 0, uuid, data, sizeof(data)) == sizeof(data));
    assert(memcmp(uuid, UUID0, sizeof(UUID0)) == 0);
    assert(memcmp(data, UUID0, sizeof(UUID0)) == 0);

    /* Add a second metadata. */
    r = luksmeta_save(cd, 1, UUID1, UUID1, sizeof(UUID1));
    if (r < 0)
        error(EXIT_FAILURE, -r, "luksmeta_save()");

    assert(test_layout((range_t[]) {
        { 0, 1024 },                   /* LUKS header */
        { 1024, 3072, true },          /* Keyslot Area */
        { offset, 4096 },              /* luksmeta header */
        { offset + 4096, 4096 },       /* luksmeta slot 0 */
        { offset + 8192, 4096 },       /* luksmeta slot 1 */
        END(offset + 12288),           /* Rest of the file */
    }));

    assert(luksmeta_load(cd, 0, uuid, data, sizeof(data)) == sizeof(data));
    assert(memcmp(uuid, UUID0, sizeof(UUID0)) == 0);
    assert(memcmp(data, UUID0, sizeof(UUID0)) == 0);
    assert(luksmeta_load(cd, 1, uuid, data, sizeof(data)) == sizeof(data));
    assert(memcmp(uuid, UUID1, sizeof(UUID1)) == 0);
    assert(memcmp(data, UUID1, sizeof(UUID1)) == 0);

    /* Delete the first metadata. */
    assert(luksmeta_wipe(cd, 0, UUID0) == 0);
    assert(test_layout((range_t[]) {
        { 0, 1024 },                   /* LUKS header */
        { 1024, 3072, true },          /* Keyslot Area */
        { offset, 4096 },              /* luksmeta header */
        { offset + 4096, 4096, true }, /* luksmeta slot 0 */
        { offset + 8192, 4096 },       /* luksmeta slot 1 */
        END(offset + 12288),           /* Rest of the file */
    }));

    /* Delete the second metadata. */
    assert(luksmeta_wipe(cd, 1, UUID1) == 0);
    assert(test_layout((range_t[]) {
        { 0, 1024 },                   /* LUKS header */
        { 1024, 3072, true },          /* Keyslot Area */
        { offset, 4096 },              /* luksmeta header */
        END(offset + 4096),            /* Rest of the file */
    }));

    crypt_free(cd);
    unlink(filename);
    return 0;
}