RepoMirrors/openssh
1
0
Fork 0
mirror of git://anongit.mindrot.org/openssh.git synced 2024-12-25 19:32:09 +00:00
Code Issues Packages Projects Releases Wiki Activity
26bf693661
openssh/regress/unittests/sshbuf/test_sshbuf_misc.c
djm@openbsd.org f8829fe57f upstream: adapt to sshbuf_dtob64() change 
OpenBSD-Regress-ID: 82374a83edf0955fd1477169eee3f5d6467405a6
2019-07-17 11:04:25 +10:00

219 lines
6.1 KiB
C
Raw Blame History

/* $OpenBSD: test_sshbuf_misc.c,v 1.4 2019/07/16 22:16:49 djm Exp $ */
/*
* Regress test for sshbuf.h buffer API
*
* Placed in the public domain
*/
#include "includes.h"
#include <sys/types.h>
#include <sys/param.h>
#include <stdio.h>
#ifdef HAVE_STDINT_H
# include <stdint.h>
#endif
#include <stdlib.h>
#include <string.h>
#include "../test_helper/test_helper.h"
#include "sshbuf.h"
#include "ssherr.h"
void sshbuf_misc_tests(void);
void
sshbuf_misc_tests(void)
{
struct sshbuf *p1;
char tmp[512], msg[] = "imploring ping silence ping over", *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_string len 1");
p1 = sshbuf_new();
ASSERT_PTR_NE(p1, NULL);
ASSERT_INT_EQ(sshbuf_put_u8(p1, 0x11), 0);
p = sshbuf_dtob64_string(p1, 0);
ASSERT_PTR_NE(p, NULL);
ASSERT_STRING_EQ(p, "EQ==");
free(p);
sshbuf_free(p1);
TEST_DONE();
TEST_START("sshbuf_dtob64_string 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_string(p1, 0);
ASSERT_PTR_NE(p, NULL);
ASSERT_STRING_EQ(p, "ESI=");
free(p);
sshbuf_free(p1);
TEST_DONE();
TEST_START("sshbuf_dtob64_string 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_string(p1, 0);
ASSERT_PTR_NE(p, NULL);
ASSERT_STRING_EQ(p, "ESIz");
free(p);
sshbuf_free(p1);
TEST_DONE();
TEST_START("sshbuf_dtob64_string 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_string(p1, 0);
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();
TEST_START("sshbuf_dup_string");
p1 = sshbuf_new();
ASSERT_PTR_NE(p1, NULL);
/* Check empty buffer */
p = sshbuf_dup_string(p1);
ASSERT_PTR_NE(p, NULL);
ASSERT_SIZE_T_EQ(strlen(p), 0);
free(p);
/* Check buffer with string */
ASSERT_INT_EQ(sshbuf_put(p1, "quad1", strlen("quad1")), 0);
p = sshbuf_dup_string(p1);
ASSERT_PTR_NE(p, NULL);
ASSERT_SIZE_T_EQ(strlen(p), strlen("quad1"));
ASSERT_STRING_EQ(p, "quad1");
free(p);
/* Check buffer with terminating nul */
ASSERT_INT_EQ(sshbuf_put(p1, "\0", 1), 0);
p = sshbuf_dup_string(p1);
ASSERT_PTR_NE(p, NULL);
ASSERT_SIZE_T_EQ(strlen(p), strlen("quad1"));
ASSERT_STRING_EQ(p, "quad1");
free(p);
/* Check buffer with data after nul (expect failure) */
ASSERT_INT_EQ(sshbuf_put(p1, "quad2", strlen("quad2")), 0);
p = sshbuf_dup_string(p1);
ASSERT_PTR_EQ(p, NULL);
sshbuf_free(p1);
TEST_DONE();
TEST_START("sshbuf_cmp");
p1 = sshbuf_from(msg, sizeof(msg) - 1);
ASSERT_PTR_NE(p1, NULL);
ASSERT_INT_EQ(sshbuf_cmp(p1, 0, "i", 1), 0);
ASSERT_INT_EQ(sshbuf_cmp(p1, 0, "j", 1), SSH_ERR_INVALID_FORMAT);
ASSERT_INT_EQ(sshbuf_cmp(p1, 0, "imploring", 9), 0);
ASSERT_INT_EQ(sshbuf_cmp(p1, 0, "implored", 9), SSH_ERR_INVALID_FORMAT);
ASSERT_INT_EQ(sshbuf_cmp(p1, 10, "ping", 4), 0);
ASSERT_INT_EQ(sshbuf_cmp(p1, 10, "ring", 4), SSH_ERR_INVALID_FORMAT);
ASSERT_INT_EQ(sshbuf_cmp(p1, 28, "over", 4), 0);
ASSERT_INT_EQ(sshbuf_cmp(p1, 28, "rove", 4), SSH_ERR_INVALID_FORMAT);
ASSERT_INT_EQ(sshbuf_cmp(p1, 28, "overt", 5),
SSH_ERR_MESSAGE_INCOMPLETE);
ASSERT_INT_EQ(sshbuf_cmp(p1, 32, "ping", 4),
SSH_ERR_MESSAGE_INCOMPLETE);
ASSERT_INT_EQ(sshbuf_cmp(p1, 1000, "silence", 7),
SSH_ERR_MESSAGE_INCOMPLETE);
ASSERT_INT_EQ(sshbuf_cmp(p1, 0, msg, sizeof(msg) - 1), 0);
TEST_DONE();
TEST_START("sshbuf_find");
p1 = sshbuf_from(msg, sizeof(msg) - 1);
ASSERT_PTR_NE(p1, NULL);
ASSERT_INT_EQ(sshbuf_find(p1, 0, "i", 1, &sz), 0);
ASSERT_SIZE_T_EQ(sz, 0);
ASSERT_INT_EQ(sshbuf_find(p1, 0, "j", 1, &sz), SSH_ERR_INVALID_FORMAT);
ASSERT_INT_EQ(sshbuf_find(p1, 0, "imploring", 9, &sz), 0);
ASSERT_SIZE_T_EQ(sz, 0);
ASSERT_INT_EQ(sshbuf_find(p1, 0, "implored", 9, &sz),
SSH_ERR_INVALID_FORMAT);
ASSERT_INT_EQ(sshbuf_find(p1, 3, "ping", 4, &sz), 0);
ASSERT_SIZE_T_EQ(sz, 10);
ASSERT_INT_EQ(sshbuf_find(p1, 11, "ping", 4, &sz), 0);
ASSERT_SIZE_T_EQ(sz, 23);
ASSERT_INT_EQ(sshbuf_find(p1, 20, "over", 4, &sz), 0);
ASSERT_SIZE_T_EQ(sz, 28);
ASSERT_INT_EQ(sshbuf_find(p1, 28, "over", 4, &sz), 0);
ASSERT_SIZE_T_EQ(sz, 28);
ASSERT_INT_EQ(sshbuf_find(p1, 28, "rove", 4, &sz),
SSH_ERR_INVALID_FORMAT);
ASSERT_INT_EQ(sshbuf_find(p1, 28, "overt", 5, &sz),
SSH_ERR_MESSAGE_INCOMPLETE);
ASSERT_INT_EQ(sshbuf_find(p1, 32, "ping", 4, &sz),
SSH_ERR_MESSAGE_INCOMPLETE);
ASSERT_INT_EQ(sshbuf_find(p1, 1000, "silence", 7, &sz),
SSH_ERR_MESSAGE_INCOMPLETE);
ASSERT_INT_EQ(sshbuf_find(p1, 0, msg + 1, sizeof(msg) - 2, &sz), 0);
ASSERT_SIZE_T_EQ(sz, 1);
TEST_DONE();
}
Reference in New Issue View Git Blame Copy Permalink