| /* $OpenBSD: test_sshbuf_misc.c,v 1.1 2014/04/30 05:32:00 djm Exp $ */ |
| /* |
| * Regress test for sshbuf.h buffer API |
| * |
| * Placed in the public domain |
| */ |
| |
| #include <sys/types.h> |
| #include <sys/param.h> |
| #include <stdio.h> |
| #include <stdint.h> |
| #include <stdlib.h> |
| #include <string.h> |
| |
| #include "test_helper.h" |
| |
| #include "sshbuf.h" |
| |
| void sshbuf_misc_tests(void); |
| |
| void |
| sshbuf_misc_tests(void) |
| { |
| struct sshbuf *p1; |
| char tmp[512], *p; |
| FILE *out; |
| size_t sz; |
| |
| TEST_START("sshbuf_dump"); |
| out = tmpfile(); |
| ASSERT_PTR_NE(out, NULL); |
| p1 = sshbuf_new(); |
| ASSERT_PTR_NE(p1, NULL); |
| ASSERT_INT_EQ(sshbuf_put_u32(p1, 0x12345678), 0); |
| sshbuf_dump(p1, out); |
| fflush(out); |
| rewind(out); |
| sz = fread(tmp, 1, sizeof(tmp), out); |
| ASSERT_INT_EQ(ferror(out), 0); |
| ASSERT_INT_NE(feof(out), 0); |
| ASSERT_SIZE_T_GT(sz, 0); |
| tmp[sz] = '\0'; |
| ASSERT_PTR_NE(strstr(tmp, "12 34 56 78"), NULL); |
| fclose(out); |
| sshbuf_free(p1); |
| TEST_DONE(); |
| |
| TEST_START("sshbuf_dtob16"); |
| p1 = sshbuf_new(); |
| ASSERT_PTR_NE(p1, NULL); |
| ASSERT_INT_EQ(sshbuf_put_u32(p1, 0x12345678), 0); |
| p = sshbuf_dtob16(p1); |
| ASSERT_PTR_NE(p, NULL); |
| ASSERT_STRING_EQ(p, "12345678"); |
| free(p); |
| sshbuf_free(p1); |
| TEST_DONE(); |
| |
| TEST_START("sshbuf_dtob64 len 1"); |
| p1 = sshbuf_new(); |
| ASSERT_PTR_NE(p1, NULL); |
| ASSERT_INT_EQ(sshbuf_put_u8(p1, 0x11), 0); |
| p = sshbuf_dtob64(p1); |
| ASSERT_PTR_NE(p, NULL); |
| ASSERT_STRING_EQ(p, "EQ=="); |
| free(p); |
| sshbuf_free(p1); |
| TEST_DONE(); |
| |
| TEST_START("sshbuf_dtob64 len 2"); |
| p1 = sshbuf_new(); |
| ASSERT_PTR_NE(p1, NULL); |
| ASSERT_INT_EQ(sshbuf_put_u8(p1, 0x11), 0); |
| ASSERT_INT_EQ(sshbuf_put_u8(p1, 0x22), 0); |
| p = sshbuf_dtob64(p1); |
| ASSERT_PTR_NE(p, NULL); |
| ASSERT_STRING_EQ(p, "ESI="); |
| free(p); |
| sshbuf_free(p1); |
| TEST_DONE(); |
| |
| TEST_START("sshbuf_dtob64 len 3"); |
| p1 = sshbuf_new(); |
| ASSERT_PTR_NE(p1, NULL); |
| ASSERT_INT_EQ(sshbuf_put_u8(p1, 0x11), 0); |
| ASSERT_INT_EQ(sshbuf_put_u8(p1, 0x22), 0); |
| ASSERT_INT_EQ(sshbuf_put_u8(p1, 0x33), 0); |
| p = sshbuf_dtob64(p1); |
| ASSERT_PTR_NE(p, NULL); |
| ASSERT_STRING_EQ(p, "ESIz"); |
| free(p); |
| sshbuf_free(p1); |
| TEST_DONE(); |
| |
| TEST_START("sshbuf_dtob64 len 8191"); |
| p1 = sshbuf_new(); |
| ASSERT_PTR_NE(p1, NULL); |
| ASSERT_INT_EQ(sshbuf_reserve(p1, 8192, NULL), 0); |
| bzero(sshbuf_mutable_ptr(p1), 8192); |
| p = sshbuf_dtob64(p1); |
| ASSERT_PTR_NE(p, NULL); |
| ASSERT_SIZE_T_EQ(strlen(p), ((8191 + 2) / 3) * 4); |
| free(p); |
| sshbuf_free(p1); |
| TEST_DONE(); |
| |
| TEST_START("sshbuf_b64tod len 1"); |
| p1 = sshbuf_new(); |
| ASSERT_PTR_NE(p1, NULL); |
| ASSERT_INT_EQ(sshbuf_b64tod(p1, "0A=="), 0); |
| ASSERT_SIZE_T_EQ(sshbuf_len(p1), 1); |
| ASSERT_U8_EQ(*sshbuf_ptr(p1), 0xd0); |
| sshbuf_free(p1); |
| TEST_DONE(); |
| |
| TEST_START("sshbuf_b64tod len 2"); |
| p1 = sshbuf_new(); |
| ASSERT_PTR_NE(p1, NULL); |
| ASSERT_INT_EQ(sshbuf_b64tod(p1, "0A8="), 0); |
| ASSERT_SIZE_T_EQ(sshbuf_len(p1), 2); |
| ASSERT_U16_EQ(PEEK_U16(sshbuf_ptr(p1)), 0xd00f); |
| sshbuf_free(p1); |
| TEST_DONE(); |
| |
| TEST_START("sshbuf_b64tod len 4"); |
| p1 = sshbuf_new(); |
| ASSERT_PTR_NE(p1, NULL); |
| ASSERT_INT_EQ(sshbuf_b64tod(p1, "0A/QDw=="), 0); |
| ASSERT_SIZE_T_EQ(sshbuf_len(p1), 4); |
| ASSERT_U32_EQ(PEEK_U32(sshbuf_ptr(p1)), 0xd00fd00f); |
| sshbuf_free(p1); |
| TEST_DONE(); |
| } |
| |