#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <errno.h>
#include <semanage/semanage.h>

void check_result_int(const char *expected, int real) {
    int exp = strtol(expected, NULL, 10);

    if (exp != real) {
        fprintf(stderr, "Expected %d but got %d\n", exp, real);
        exit(1);
    }
}

semanage_handle_t *test_handle_create() {
    semanage_handle_t *sh = NULL;

    sh = semanage_handle_create();
    printf("semanage_handle_create(): %p\n", (void *) sh);

    if (sh == NULL) {
        perror("semanage_handle_create");
        exit(2);
    }

    return sh;
}

int test_connect(semanage_handle_t *sh) {
    int result = semanage_connect(sh);
    printf("semanage_connect(%p): %d\n", (void *) sh, result);

    if (result != 0) {
        perror("semanage_connect");
        exit(2);
    }

    return result;
}

int test_disconnect(semanage_handle_t *sh) {
    int result = semanage_disconnect(sh);
    printf("semanage_disconnect(%p): %d\n", (void *) sh, result);

    if (result != 0) {
        perror("semanage_disconnect");
        exit(2);
    }

    return result;
}

int test_begin_transaction(semanage_handle_t *sh) {
    int result = semanage_begin_transaction(sh);
    printf("semanage_begin_transaction(%p): %d\n", (void *) sh, result);

    if (result != 0) {
        perror("semanage_begin_transaction");
        exit(2);
    }

    return result;
}

int test_commit(semanage_handle_t *sh) {
    int result = semanage_commit(sh);
    printf("semanage_commit(%p): %d\n", (void *) sh, result);

    if (result != 0) {
        perror("semanage_commit");
        exit(2);
    }

    return result;
}

semanage_seuser_key_t *test_get_key(semanage_handle_t *sh, const char *name) {
    semanage_seuser_key_t *key;
    int result = semanage_seuser_key_create(sh, name, &key);
    printf("semanage_seuser_key_create(%p, %s, %p): %d\n",
           (void *) sh, name, (void *) &key, result);

    if (key == NULL || result < 0) {
        perror("semanage_seuser_key_create");
        exit(2);
    }

    return key;
}

semanage_seuser_t *test_get_seuser_nth(semanage_handle_t *sh, unsigned int index) {
    int result;
    semanage_seuser_t **records;
    unsigned int count;

    result = semanage_seuser_list(sh, &records, &count);
    printf("semanage_seuser_list(%p, %p, %p): %d\n",
           (void *) sh, (void *) &records, (void *) &count, result);
    
    if (result < 0) {
        perror("semanage_seuser_list");
        exit(2);
    }

    if (count < index + 1)
        exit(2);

    return records[index];
}

semanage_seuser_t *test_get_seuser_new(semanage_handle_t *sh) {
    int result;
    semanage_seuser_t *seuser;

    result = semanage_seuser_create(sh, &seuser);
    printf("semanage_seuser_create(%p, %p): %d\n",
           (void *) sh, (void *) seuser, result);
    
    if (result < 0) {
        perror("semanage_seuser_create");
        exit(2);
    }

    return seuser;
}

semanage_seuser_t *test_get_seuser(semanage_handle_t *sh, const char *param) {
    if (strcmp(param, "new") == 0)
        return test_get_seuser_new(sh);
    
    if (strcmp(param, "first") == 0)
        return test_get_seuser_nth(sh, 0);

    if (strcmp(param, "second") == 0)
        return test_get_seuser_nth(sh, 1);

    fprintf(stderr, "Unknown seuser \"%s\" specified\n", param);
    exit(2);
}

void test_add_local_seuser(semanage_handle_t *sh, semanage_seuser_t *seuser) {
    int result;
    semanage_seuser_key_t *key;

    result = semanage_seuser_key_extract(sh, seuser, &key);
    printf("semanage_seuser_key_extract(%p, %p, %p): %d\n",
           (void *) sh, (void *) seuser, (void *) &key, result);

    if (result < 0) {
        perror("semanage_seuser_key_extract");
        exit(2);
    }

    result = semanage_seuser_modify_local(sh, key, seuser);
    printf("semanage_seuser_modify_local(%p, %p, %p): %d\n",
           (void *) seuser, (void *) key, (void *) seuser, result);

    if (result < 0) {
        perror("semanage_seuser_modify_local");
        exit(2);
    }
}

void test_del_local_seuser(semanage_handle_t *sh, semanage_seuser_t *seuser) {
    int result;
    semanage_seuser_key_t *key;

    result = semanage_seuser_key_extract(sh, seuser, &key);
    printf("semanage_seuser_key_extract(%p, %p, %p): %d\n",
           (void *) sh, (void *) seuser, (void *) &key, result);

    if (result < 0) {
        perror("semanage_seuser_key_extract");
        exit(2);
    }

    result = semanage_seuser_del_local(sh, key);
    printf("semanage_seuser_del_local(%p, %p): %d\n",
           (void *) seuser, (void *) key, result);

    if (result < 0) {
        perror("semanage_seuser_del_local");
        exit(2);
    }
}

#define STATE_INIT      1
#define STATE_HANDLE    2
#define STATE_CONN      3
#define STATE_TRANS     4

int get_state(const char *state_str) {
    if (strcmp(state_str, "init") == 0)
        return STATE_INIT;
    if (strcmp(state_str, "handle") == 0)
        return STATE_HANDLE;
    if (strcmp(state_str, "conn") == 0)
        return STATE_CONN;
    if (strcmp(state_str, "trans") == 0)
        return STATE_TRANS;

    return 0;
}

semanage_handle_t * get_handle(const char *state_str) {
    int state;
    semanage_handle_t *sh = NULL;

    state = get_state(state_str);

    if (state >= STATE_INIT)
        sh = NULL;

    if (state >= STATE_HANDLE)
        sh = test_handle_create();

    if (state >= STATE_CONN)
        test_connect(sh);

    if (state >= STATE_TRANS)
        test_begin_transaction(sh);

    return sh;
}

void destroy_handle(semanage_handle_t *sh, const char *state_str) {
    int state;

    state = get_state(state_str);

    if (state >= STATE_TRANS)
        test_commit(sh);

    if (state >= STATE_CONN)
        test_disconnect(sh);

    if (state >= STATE_HANDLE) {
        semanage_handle_destroy(sh);
        printf("semanage_handle_destroy(%p)\n", (void *) sh);
    }
}

int strcmp_null(const char *str1, const char *str2) {
    if (str1 == NULL && str2 == NULL)
        return 0;

    if (str1 == NULL) {
        if (strcmp(str2, "NULL") == 0)
            return 0;
        else
            return -1;
    }

    if (str2 == NULL) {
        if (strcmp(str1, "NULL") == 0)
            return 0;
        else
            return 1;
    }

    return strcmp(str1, str2);
}