mirror of
https://git.libssh.org/projects/libssh.git
synced 2026-02-09 18:04:25 +09:00
Compare commits
6 Commits
72c282434b
...
ccb8cf88c8
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
ccb8cf88c8 | ||
|
|
b43392c31d | ||
|
|
5fc65e7270 | ||
|
|
8310b8cc2b | ||
|
|
b0063b52d8 | ||
|
|
33a947dcb0 |
@@ -66,7 +66,6 @@ int authenticate_kbdint(ssh_session session, const char *password)
|
||||
return SSH_AUTH_ERROR;
|
||||
}
|
||||
|
||||
buffer[sizeof(buffer) - 1] = '\0';
|
||||
if ((p = strchr(buffer, '\n'))) {
|
||||
*p = '\0';
|
||||
}
|
||||
@@ -75,7 +74,7 @@ int authenticate_kbdint(ssh_session session, const char *password)
|
||||
return SSH_AUTH_ERROR;
|
||||
}
|
||||
|
||||
memset(buffer, 0, strlen(buffer));
|
||||
memset(buffer, 0, sizeof(buffer));
|
||||
} else {
|
||||
if (password && strstr(prompt, "Password:")) {
|
||||
answer = password;
|
||||
|
||||
@@ -16,7 +16,7 @@ clients must be made or how a client should react.
|
||||
#include <libssh/libssh.h>
|
||||
|
||||
/** Zero a structure */
|
||||
#define ZERO_STRUCT(x) memset((char *)&(x), 0, sizeof(x))
|
||||
#define ZERO_STRUCT(x) memset(&(x), 0, sizeof(x))
|
||||
|
||||
int authenticate_console(ssh_session session);
|
||||
int authenticate_kbdint(ssh_session session, const char *password);
|
||||
|
||||
@@ -357,10 +357,10 @@ int ssh_connector_remove_event(ssh_connector connector);
|
||||
#define SAFE_FREE(x) do { if ((x) != NULL) {free(x); x=NULL;} } while(0)
|
||||
|
||||
/** Zero a structure */
|
||||
#define ZERO_STRUCT(x) memset((char *)&(x), 0, sizeof(x))
|
||||
#define ZERO_STRUCT(x) memset(&(x), 0, sizeof(x))
|
||||
|
||||
/** Zero a structure given a pointer to the structure */
|
||||
#define ZERO_STRUCTP(x) do { if ((x) != NULL) memset((char *)(x), 0, sizeof(*(x))); } while(0)
|
||||
#define ZERO_STRUCTP(x) do { if ((x) != NULL) memset((x), 0, sizeof(*(x))); } while(0)
|
||||
|
||||
/** Get the size of an array */
|
||||
#define ARRAY_SIZE(a) (sizeof(a)/sizeof(a[0]))
|
||||
|
||||
@@ -533,6 +533,7 @@ ssh_string ssh_agent_sign_data(ssh_session session,
|
||||
* STRING_SIZE_MAX is (UINT32_MAX >> 8) + 1 */
|
||||
rc = ssh_buffer_allocate_size(request, (uint32_t)request_len);
|
||||
if (rc < 0) {
|
||||
SSH_STRING_FREE(key_blob);
|
||||
SSH_BUFFER_FREE(request);
|
||||
return NULL;
|
||||
}
|
||||
|
||||
@@ -19,6 +19,7 @@ set(LIBSSH_UNIT_TESTS
|
||||
torture_temp_file
|
||||
torture_push_pop_dir
|
||||
torture_session_keys
|
||||
torture_string
|
||||
torture_tokens
|
||||
)
|
||||
|
||||
|
||||
@@ -2468,9 +2468,9 @@ static void torture_config_make_absolute_int(void **state, bool no_sshdir_fails)
|
||||
char *home = NULL;
|
||||
struct passwd *pw = getpwuid(getuid());
|
||||
assert_non_null(pw);
|
||||
user = pw->pw_name;
|
||||
user = strdup(pw->pw_name);
|
||||
assert_non_null(user);
|
||||
home = pw->pw_dir;
|
||||
home = strdup(pw->pw_dir);
|
||||
assert_non_null(home);
|
||||
#endif
|
||||
|
||||
@@ -2528,6 +2528,8 @@ static void torture_config_make_absolute_int(void **state, bool no_sshdir_fails)
|
||||
snprintf(h, 256 - 1, "/etc/ssh/~%s/.ssh/config.d/*.conf", user);
|
||||
assert_string_equal(result, h);
|
||||
free(result);
|
||||
free(home);
|
||||
free(user);
|
||||
#endif
|
||||
}
|
||||
|
||||
|
||||
268
tests/unittests/torture_string.c
Normal file
268
tests/unittests/torture_string.c
Normal file
@@ -0,0 +1,268 @@
|
||||
/*
|
||||
* 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);
|
||||
}
|
||||
|
||||
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),
|
||||
};
|
||||
|
||||
ssh_init();
|
||||
torture_filter_tests(tests);
|
||||
rc = cmocka_run_group_tests(tests, NULL, NULL);
|
||||
ssh_finalize();
|
||||
|
||||
return rc;
|
||||
}
|
||||
Reference in New Issue
Block a user