aboutsummaryrefslogtreecommitdiff
path: root/tests
diff options
context:
space:
mode:
authorAris Adamantiadis <aris@0xbadc0de.be>2014-04-09 20:57:08 +0200
committerAndreas Schneider <asn@cryptomilk.org>2014-08-06 09:20:45 +0200
commit1d97f75b0aa3f47a0dc9589af30570103b35c490 (patch)
tree37e0ccfd0b36620e6b5b96f06252d3eec550894c /tests
parent835e34d1eb407465eac8744ff392d029a117fae3 (diff)
downloadlibssh-1d97f75b0aa3f47a0dc9589af30570103b35c490.tar.gz
libssh-1d97f75b0aa3f47a0dc9589af30570103b35c490.tar.xz
libssh-1d97f75b0aa3f47a0dc9589af30570103b35c490.zip
tests: test for ssh_buffer_(un)pack
Reviewed-by: Andreas Schneider <asn@samba.org>
Diffstat (limited to 'tests')
-rw-r--r--tests/unittests/torture_buffer.c98
1 files changed, 98 insertions, 0 deletions
diff --git a/tests/unittests/torture_buffer.c b/tests/unittests/torture_buffer.c
index 0ec2cf73..f4fa465e 100644
--- a/tests/unittests/torture_buffer.c
+++ b/tests/unittests/torture_buffer.c
@@ -122,7 +122,102 @@ static void torture_buffer_get_ssh_string(void **state) {
}
}
+static void torture_buffer_add_format(void **state) {
+ ssh_buffer buffer=*state;
+ uint8_t b;
+ uint16_t w;
+ uint32_t d;
+ uint64_t q;
+ ssh_string s;
+ int rc;
+ size_t len;
+ uint8_t verif[]="\x42\x13\x37\x0b\xad\xc0\xde\x13\x24\x35\x46"
+ "\xac\xbd\xce\xdf"
+ "\x00\x00\x00\x06" "libssh"
+ "\x00\x00\x00\x05" "rocks"
+ "So much";
+
+ b=0x42;
+ w=0x1337;
+ d=0xbadc0de;
+ q=0x13243546acbdcedf;
+ s=ssh_string_from_char("libssh");
+ rc=ssh_buffer_pack(buffer, "bwdqSsP",b,w,d,q,s,"rocks",7,"So much");
+ assert_int_equal(rc, SSH_OK);
+
+ len = buffer_get_rest_len(buffer);
+ assert_int_equal(len, sizeof(verif) - 1);
+ assert_memory_equal(buffer_get_rest(buffer), verif, sizeof(verif) -1);
+}
+static void torture_buffer_get_format(void **state) {
+ ssh_buffer buffer=*state;
+ uint8_t b=0;
+ uint16_t w=0;
+ uint32_t d=0;
+ uint64_t q=0;
+ ssh_string s=NULL;
+ char *s1=NULL, *s2=NULL;
+ int rc;
+ size_t len;
+ uint8_t verif[]="\x42\x13\x37\x0b\xad\xc0\xde\x13\x24\x35\x46"
+ "\xac\xbd\xce\xdf"
+ "\x00\x00\x00\x06" "libssh"
+ "\x00\x00\x00\x05" "rocks"
+ "So much";
+
+ rc = ssh_buffer_add_data(buffer, verif, sizeof(verif) - 1);
+ assert_int_equal(rc, SSH_OK);
+ rc = ssh_buffer_unpack(buffer, "bwdqSsP",&b,&w,&d,&q,&s,&s1,(size_t)7,&s2);
+ assert_int_equal(rc, SSH_OK);
+
+ assert_int_equal(b, 0x42);
+ assert_int_equal(w, 0x1337);
+
+ assert_true(d == 0xbadc0de);
+ assert_true(q == 0x13243546acbdcedf);
+
+ assert_true(s != NULL);
+ assert_int_equal(ssh_string_len(s), 6);
+ assert_memory_equal(ssh_string_data(s), "libssh", 6);
+
+ assert_true(s1 != NULL);
+ assert_string_equal(s1, "rocks");
+
+ assert_true(s2 != NULL);
+ assert_memory_equal(s2, "So much", 7);
+
+ len = buffer_get_rest_len(buffer);
+ assert_int_equal(len, 0);
+ SAFE_FREE(s);
+ SAFE_FREE(s1);
+ SAFE_FREE(s2);
+}
+
+static void torture_buffer_get_format_error(void **state) {
+ ssh_buffer buffer=*state;
+ uint8_t b=0;
+ uint16_t w=0;
+ uint32_t d=0;
+ uint64_t q=0;
+ ssh_string s=NULL;
+ char *s1=NULL, *s2=NULL;
+ int rc;
+ uint8_t verif[]="\x42\x13\x37\x0b\xad\xc0\xde\x13\x24\x35\x46"
+ "\xac\xbd\xce\xdf"
+ "\x00\x00\x00\x06" "libssh"
+ "\x00\x00\x00\x05" "rocks"
+ "So much";
+
+ rc = ssh_buffer_add_data(buffer, verif, sizeof(verif) - 1);
+ assert_int_equal(rc, SSH_OK);
+ rc = ssh_buffer_unpack(buffer, "bwdqSsPb",&b,&w,&d,&q,&s,&s1,(size_t)7,&s2,&b);
+ assert_int_equal(rc, SSH_ERROR);
+
+ assert_true(s==NULL);
+ assert_true(s1 == NULL);
+ assert_true(s2 == NULL);
+}
int torture_run_tests(void) {
int rc;
@@ -131,6 +226,9 @@ int torture_run_tests(void) {
unit_test_setup_teardown(torture_growing_buffer_shifting, setup, teardown),
unit_test_setup_teardown(torture_buffer_prepend, setup, teardown),
unit_test(torture_buffer_get_ssh_string),
+ unit_test_setup_teardown(torture_buffer_add_format, setup, teardown),
+ unit_test_setup_teardown(torture_buffer_get_format, setup, teardown),
+ unit_test_setup_teardown(torture_buffer_get_format_error, setup, teardown)
};
ssh_init();