lib/libssh
1
0
Fork 0
mirror of https://git.libssh.org/projects/libssh.git synced 2025-07-29 13:01:13 +03:00
Code Activity

torture_misc.c : Add test for ssh_writen()

Browse Source 
Signed-off-by: Eshan Kelkar <eshankelkar@galorithm.com>
Reviewed-by: Jakub Jelen <jjelen@redhat.com>
Reviewed-by: Norbert Pocs <npocs@redhat.com>
This commit is contained in:
Eshan Kelkar
2023-07-30 12:46:03 +05:30
committed by Jakub Jelen
parent 85d7cc5cf2
commit 06fbf5c159
1 changed files with 96 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

97
tests/unittests/torture_misc.c
View File

@ -885,6 +885,100 @@ static void torture_ssh_readn(void **state)
free(write_buf);
}
static void torture_ssh_writen(void **state)
{
char *write_buf = NULL, *read_buf = NULL, *file_path = NULL;
const char *file_template = "libssh_torture_ssh_writen_test_XXXXXX";
size_t data_len = 10 * 1024 * 1024;
size_t i, total_bytes_read = 0;
ssize_t bytes_written, bytes_read;
off_t off;
int rc, fd, flags;
(void)state;
write_buf = malloc(data_len);
assert_non_null(write_buf);
/* Fill the write buffer with random data */
for (i = 0; i < data_len; ++i) {
rc = rand();
write_buf[i] = (char)(rc & 0xff);
}
read_buf = malloc(data_len);
assert_non_null(read_buf);
file_path = torture_create_temp_file(file_template);
assert_non_null(file_path);
/* Open a file for reading and writing */
flags = O_RDWR;
#ifdef _WIN32
flags |= O_BINARY;
#endif
fd = open(file_path, flags, 0);
assert_int_not_equal(fd, -1);
/* Write the data present in the write buffer to the file */
bytes_written = ssh_writen(fd, write_buf, data_len);
assert_int_equal(bytes_written, data_len);
/*
* Ensure that the file offset is incremented by the number of bytes
* written.
*/
off = lseek(fd, 0, SEEK_CUR);
assert_int_equal(off, data_len);
/*
* Ensure that the data present in the write buffer has been written to the
* file and not some garbage.
*/
off = lseek(fd, 0, SEEK_SET);
assert_int_not_equal(off, -1);
do {
bytes_read = read(fd,
read_buf + total_bytes_read,
data_len - total_bytes_read);
if (bytes_read == -1 && errno == EINTR) {
continue;
}
assert_int_not_equal(bytes_read, -1);
assert_int_not_equal(bytes_read, 0);
total_bytes_read += bytes_read;
} while (total_bytes_read < data_len);
assert_memory_equal(write_buf, read_buf, data_len);
/* Negative tests start */
bytes_written = ssh_writen(-3, write_buf, data_len);
assert_int_equal(bytes_written, -1);
bytes_written = ssh_writen(fd, NULL, data_len);
assert_int_equal(bytes_written, -1);
bytes_written = ssh_writen(fd, write_buf, 0);
assert_int_equal(bytes_written, -1);
/* Clean up */
rc = close(fd);
assert_int_equal(rc, 0);
rc = unlink(file_path);
assert_int_equal(rc, 0);
free(file_path);
free(read_buf);
free(write_buf);
}
int torture_run_tests(void) {
int rc;
struct CMUnitTest tests[] = {
@ -909,7 +1003,8 @@ int torture_run_tests(void) {
cmocka_unit_test(torture_ssh_quote_file_name),
cmocka_unit_test(torture_ssh_strreplace),
cmocka_unit_test(torture_ssh_strerror),
cmocka_unit_test(torture_ssh_readn)
cmocka_unit_test(torture_ssh_readn),
cmocka_unit_test(torture_ssh_writen)
};
ssh_init();