/*
 * Copyright 2009, Axel Dörfler, axeld@pinc-software.de.
 * This file may be used under the terms of the MIT License.
 */


#include <errno.h>
#include <stdio.h>
#include <string.h>

#include <fs_attr.h>
#include <TypeConstants.h>


const char* kTestFileName = "/tmp/fs_attr_test";


void
test_read(int fd, const char* attribute, type_code type, const char* data,
        size_t length)
{
        attr_info info;
        if (fs_stat_attr(fd, attribute, &info) != 0) {
                fprintf(stderr, "Could not stat attribute \"%s\": %s\n", attribute,
                        strerror(errno));
                exit(1);
        }

        if (info.size != length) {
                fprintf(stderr, "Length does not match for \"%s\": should be %ld, is "
                        "%lld\n", data, length, info.size);
                exit(1);
        }

        if (info.type != type) {
                fprintf(stderr, "Type does not match B_RAW_TYPE!\n");
                exit(1);
        }

        char buffer[4096];
        ssize_t bytesRead = fs_read_attr(fd, attribute, B_RAW_TYPE, 0, buffer,
                length);
        if (bytesRead != (ssize_t)length) {
                fprintf(stderr, "Bytes read does not match: should be %ld, is %ld\n",
                        length, bytesRead);
                exit(1);
        }

        if (memcmp(data, buffer, length)) {
                fprintf(stderr, "Bytes do not match: should be \"%s\", is \"%s\"\n",
                        data, buffer);
                exit(1);
        }
}


int
main(int argc, char** argv)
{
        int fd = open(kTestFileName, O_CREAT | O_TRUNC | O_WRONLY);
        if (fd < 0) {
                fprintf(stderr, "Creating test file \"%s\" failed: %s\n", kTestFileName,
                        strerror(errno));
                return 1;
        }

        // Test the old BeOS API

        fs_write_attr(fd, "TEST", B_STRING_TYPE, 0, "Hello BeOS", 11);
        test_read(fd, "TEST", B_STRING_TYPE, "Hello BeOS", 11);

        // TODO: this shows a bug in BFS; the index is not updated correctly
        fs_write_attr(fd, "TEST", B_STRING_TYPE, 6, "Haiku", 6);
        test_read(fd, "TEST", B_STRING_TYPE, "Hello Haiku", 12);

        fs_write_attr(fd, "TESTraw", B_RAW_TYPE, 16, "Haiku", 6);
        test_read(fd, "TESTraw", B_RAW_TYPE, "\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0Haiku",
                22);

        fs_write_attr(fd, "TESTraw", B_RAW_TYPE, 0, "Haiku", 6);
        test_read(fd, "TESTraw", B_RAW_TYPE, "Haiku", 6);

        char buffer[4096];
        memset(buffer, 0, sizeof(buffer));
        strcpy(buffer, "Hello");
        fs_write_attr(fd, "TESTswitch", B_RAW_TYPE, 0, buffer,
                strlen(buffer) + 1);
        test_read(fd, "TESTswitch", B_RAW_TYPE, buffer, strlen(buffer) + 1);

        strcpy(buffer + 4000, "Haiku");
        fs_write_attr(fd, "TESTswitch", B_RAW_TYPE, 0, buffer, 4006);
        test_read(fd, "TESTswitch", B_RAW_TYPE, buffer, 4006);

        return 0;
}