/*
 * torture_string.c - torture tests for ssh_string functions
 *
 * This file is part of the SSH Library
 *
 * Copyright (c) 2025 Praneeth Sarode <praneethsarode@gmail.com>
 *
 * The SSH Library 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.
 *
 * The SSH Library 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 the SSH Library; see the file COPYING.  If not, write to
 * the Free Software Foundation, Inc., 59 Temple Place - Suite 330, Boston,
 * MA 02111-1307, USA.
 */

#include "config.h"

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

#define LIBSSH_STATIC

#include "libssh/string.h"
#include "string.c"
#include "torture.h"

static void torture_ssh_string_new(void **state)
{
    struct ssh_string_struct *str = NULL;

    (void)state;

    /* Test normal allocation */
    str = ssh_string_new(100);
    assert_non_null(str);
    assert_int_equal(ssh_string_len(str), 100);
    ssh_string_free(str);

    /* Test zero size */
    str = ssh_string_new(0);
    assert_non_null(str);
    assert_int_equal(ssh_string_len(str), 0);
    ssh_string_free(str);

    /* Test maximum size */
    str = ssh_string_new(STRING_SIZE_MAX - 1);
    assert_non_null(str);
    assert_int_equal(ssh_string_len(str), STRING_SIZE_MAX - 1);
    ssh_string_free(str);

    /* Test size too large - should fail */
    str = ssh_string_new(STRING_SIZE_MAX + 1);
    assert_null(str);
    assert_int_equal(errno, EINVAL);
}

static void torture_ssh_string_from_char(void **state)
{
    struct ssh_string_struct *str = NULL;
    const char *test_string = "Hello, World!";
    const char *empty_string = "";

    (void)state;

    /* Test normal string */
    str = ssh_string_from_char(test_string);
    assert_non_null(str);
    assert_int_equal(ssh_string_len(str), strlen(test_string));
    assert_memory_equal(ssh_string_data(str), test_string, strlen(test_string));
    ssh_string_free(str);

    /* Test empty string */
    str = ssh_string_from_char(empty_string);
    assert_non_null(str);
    assert_int_equal(ssh_string_len(str), 0);
    ssh_string_free(str);

    /* Test NULL input */
    str = ssh_string_from_char(NULL);
    assert_null(str);
    assert_int_equal(errno, EINVAL);
}

static void torture_ssh_string_fill(void **state)
{
    struct ssh_string_struct *str = NULL;
    const char *test_data = "Test data";
    int rc;

    (void)state;

    /* Test normal fill */
    str = ssh_string_new(20);
    assert_non_null(str);

    rc = ssh_string_fill(str, test_data, strlen(test_data));
    assert_int_equal(rc, 0);
    assert_memory_equal(ssh_string_data(str), test_data, strlen(test_data));
    ssh_string_free(str);

    /* Test fill with exact size */
    str = ssh_string_new(strlen(test_data));
    assert_non_null(str);

    rc = ssh_string_fill(str, test_data, strlen(test_data));
    assert_int_equal(rc, 0);
    ssh_string_free(str);

    /* Test NULL data */
    str = ssh_string_new(10);
    assert_non_null(str);

    rc = ssh_string_fill(str, NULL, 5);
    assert_int_equal(rc, -1);
    ssh_string_free(str);

    /* Test zero length */
    str = ssh_string_new(10);
    assert_non_null(str);

    rc = ssh_string_fill(str, test_data, 0);
    assert_int_equal(rc, -1);
    ssh_string_free(str);
}

static void torture_ssh_string_to_char(void **state)
{
    struct ssh_string_struct *str = NULL;
    const char *test_string = "Convert to char";
    char *result = NULL;

    (void)state;

    /* Test normal string */
    str = ssh_string_from_char(test_string);
    assert_non_null(str);

    result = ssh_string_to_char(str);
    assert_non_null(result);
    assert_string_equal(result, test_string);

    ssh_string_free_char(result);
    ssh_string_free(str);

    /* Test empty string */
    str = ssh_string_from_char("");
    assert_non_null(str);

    result = ssh_string_to_char(str);
    assert_non_null(result);
    assert_string_equal(result, "");

    ssh_string_free_char(result);
    ssh_string_free(str);

    /* Test NULL string */
    result = ssh_string_to_char(NULL);
    assert_null(result);
}

static void torture_ssh_string_copy(void **state)
{
    struct ssh_string_struct *str = NULL, *copy = NULL;
    const char *test_string = "Copy me!";

    (void)state;

    /* Test normal copy */
    str = ssh_string_from_char(test_string);
    assert_non_null(str);

    copy = ssh_string_copy(str);
    assert_non_null(copy);
    assert_int_equal(ssh_string_len(copy), ssh_string_len(str));
    assert_memory_equal(ssh_string_data(copy),
                        ssh_string_data(str),
                        ssh_string_len(str));

    /* Ensure they are different objects */
    assert_ptr_not_equal(str, copy);
    assert_ptr_not_equal(ssh_string_data(str), ssh_string_data(copy));

    ssh_string_free(str);
    ssh_string_free(copy);

    /* Test copy of empty string */
    str = ssh_string_from_char("");
    assert_non_null(str);

    copy = ssh_string_copy(str);
    assert_non_null(copy);
    assert_int_equal(ssh_string_len(copy), 0);

    ssh_string_free(str);
    ssh_string_free(copy);

    /* Test NULL string */
    copy = ssh_string_copy(NULL);
    assert_null(copy);
}

static void torture_ssh_string_burn(void **state)
{
    struct ssh_string_struct *str = NULL;
    const char *test_string = "Secret data";
    void *data = NULL;
    size_t len;
    int i;

    (void)state;

    /* Test burning a string */
    str = ssh_string_from_char(test_string);
    assert_non_null(str);

    data = ssh_string_data(str);
    len = ssh_string_len(str);

    /* Verify data is there initially */
    assert_memory_equal(data, test_string, len);

    /* Burn the string */
    ssh_string_burn(str);

    /* Verify data is zeroed out */
    for (i = 0; i < (int)len; i++) {
        assert_int_equal(((unsigned char *)data)[i], 0);
    }

    ssh_string_free(str);

    /* Test burning NULL string (should not crash) */
    ssh_string_burn(NULL);

    /* Test burning zero-size string */
    str = ssh_string_new(0);
    assert_non_null(str);
    ssh_string_burn(str);
    ssh_string_free(str);
}

static void torture_ssh_string_cmp(void **state)
{
    struct ssh_string_struct *str1 = NULL, *str2 = NULL;
    const char *test_string1 = "Hello, World!";
    const char *test_string2 = "Hello, libssh";
    const char *test_string3 = "Hello";
    const char *test_string4 = "Apple";

    const char data1[] = "Hello\x00World!";
    const char data2[] = "Hello\x00libssh";
    const char data3[] = "Hello";

    int rc;
    (void)state;

    /* Test comparing two NULL strings - should be equal */
    assert_int_equal(ssh_string_cmp(NULL, NULL), 0);

    /* Test comparing NULL with non-NULL string - NULL should be less */
    str1 = ssh_string_from_char(test_string1);
    assert_non_null(str1);
    assert_true(ssh_string_cmp(NULL, str1) < 0);
    assert_true(ssh_string_cmp(str1, NULL) > 0);
    ssh_string_free(str1);

    /* Test comparing empty strings */
    str1 = ssh_string_from_char("");
    str2 = ssh_string_from_char("");
    assert_non_null(str1);
    assert_non_null(str2);

    /* Both empty strings should be equal */
    assert_int_equal(ssh_string_cmp(str1, str2), 0);
    ssh_string_free(str1);
    ssh_string_free(str2);

    /* Test comparing empty string with non-empty string */
    str1 = ssh_string_from_char("");
    str2 = ssh_string_from_char("test");
    assert_non_null(str1);
    assert_non_null(str2);

    /* Empty string should be less than non-empty string */
    assert_true(ssh_string_cmp(str1, str2) < 0);
    assert_true(ssh_string_cmp(str2, str1) > 0);
    ssh_string_free(str1);
    ssh_string_free(str2);

    /* Test comparing strings where one is a prefix of another */
    str1 = ssh_string_from_char(test_string1); /* "Hello, World!" */
    str2 = ssh_string_from_char(test_string3); /* "Hello" - prefix */
    assert_non_null(str1);
    assert_non_null(str2);

    /* "Hello" is shorter and a prefix, so it should be < "Hello, World!" */
    assert_true(ssh_string_cmp(str2, str1) < 0);
    assert_true(ssh_string_cmp(str1, str2) > 0);
    ssh_string_free(str1);
    ssh_string_free(str2);

    /* Test comparing different strings with same length */
    str1 = ssh_string_from_char(test_string1); /* "Hello, World!" */
    str2 = ssh_string_from_char(test_string2); /* "Hello, libssh" */
    assert_non_null(str1);
    assert_non_null(str2);

    /* "Hello, World!" vs "Hello, libssh" - 'W' < 'l' */
    assert_true(ssh_string_cmp(str1, str2) < 0);
    assert_true(ssh_string_cmp(str2, str1) > 0);
    ssh_string_free(str1);
    ssh_string_free(str2);

    /* Test comparing strings with different lengths and different characters */
    str1 = ssh_string_from_char(test_string1); /* "Hello, World!" */
    str2 = ssh_string_from_char(test_string4); /* "Apple" */
    assert_non_null(str1);
    assert_non_null(str2);

    /* 'A' < 'H' so "Apple" < "Hello, World!" */
    assert_true(ssh_string_cmp(str2, str1) < 0);
    assert_true(ssh_string_cmp(str1, str2) > 0);
    ssh_string_free(str1);
    ssh_string_free(str2);

    /* Test comparing identical strings - should be equal */
    str1 = ssh_string_from_char(test_string1);
    str2 = ssh_string_from_char(test_string1);
    assert_non_null(str1);
    assert_non_null(str2);
    assert_int_equal(ssh_string_cmp(str1, str2), 0);
    assert_int_equal(ssh_string_cmp(str2, str1), 0);
    ssh_string_free(str1);
    ssh_string_free(str2);

    /* Test comparing strings with embedded null characters */
    str1 = ssh_string_new(sizeof(data1)); /* "Hello\x00World!" */
    str2 = ssh_string_new(sizeof(data3)); /* "Hello" */
    assert_non_null(str1);
    assert_non_null(str2);
    rc = ssh_string_fill(str1, data1, sizeof(data1));
    assert_int_equal(rc, 0);
    rc = ssh_string_fill(str2, data3, sizeof(data3));
    assert_int_equal(rc, 0);

    /* "Hello\x00World!" > "Hello" because its length is greater */
    assert_true(ssh_string_cmp(str1, str2) > 0); /* data1 > data3 */
    assert_true(ssh_string_cmp(str2, str1) < 0); /* data3 < data1 */
    ssh_string_free(str1);
    ssh_string_free(str2);

    /* Comparing binary strings with same length, but different characters */
    str1 = ssh_string_new(sizeof(data1)); /* "Hello\x00World!" */
    str2 = ssh_string_new(sizeof(data2)); /* "Hello\x00libssh" */
    assert_non_null(str1);
    assert_non_null(str2);
    rc = ssh_string_fill(str1, data1, sizeof(data1));
    assert_int_equal(rc, 0);
    rc = ssh_string_fill(str2, data2, sizeof(data2));
    assert_int_equal(rc, 0);

    /* 'W' < 'l' so str1 < str2 */
    assert_true(ssh_string_cmp(str1, str2) < 0); /* data1 < data2 */
    assert_true(ssh_string_cmp(str2, str1) > 0); /* data2 > data1 */
    ssh_string_free(str1);
    ssh_string_free(str2);
}

int torture_run_tests(void)
{
    int rc;
    struct CMUnitTest tests[] = {
        cmocka_unit_test(torture_ssh_string_new),
        cmocka_unit_test(torture_ssh_string_from_char),
        cmocka_unit_test(torture_ssh_string_fill),
        cmocka_unit_test(torture_ssh_string_to_char),
        cmocka_unit_test(torture_ssh_string_copy),
        cmocka_unit_test(torture_ssh_string_burn),
        cmocka_unit_test(torture_ssh_string_cmp),
    };

    ssh_init();
    torture_filter_tests(tests);
    rc = cmocka_run_group_tests(tests, NULL, NULL);
    ssh_finalize();

    return rc;
}