root/crypto/openssh/regress/unittests/misc/test_hpdelim.c
/*      $OpenBSD: test_hpdelim.c,v 1.2 2022/02/06 22:58:33 dtucker Exp $ */
/*
 * Regress test for misc hpdelim() and co
 *
 * Placed in the public domain.
 */

#include "includes.h"

#include <sys/types.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 "log.h"
#include "misc.h"
#include "xmalloc.h"

void test_hpdelim(void);

void
test_hpdelim(void)
{
        char *orig, *str, *cp, *port;

#define START_STRING(x) orig = str = xstrdup(x)
#define DONE_STRING()   free(orig)

        TEST_START("hpdelim host only");
        START_STRING("host");
        cp = hpdelim(&str);
        ASSERT_STRING_EQ(cp, "host");
        ASSERT_PTR_EQ(str, NULL);
        DONE_STRING();
        TEST_DONE();

        TEST_START("hpdelim :port");
        START_STRING(":1234");
        cp = hpdelim(&str);
        ASSERT_STRING_EQ(cp, "");
        ASSERT_PTR_NE(str, NULL);
        port = hpdelim(&str);
        ASSERT_STRING_EQ(port, "1234");
        ASSERT_PTR_EQ(str, NULL);
        DONE_STRING();
        TEST_DONE();

        TEST_START("hpdelim host:port");
        START_STRING("host:1234");
        cp = hpdelim(&str);
        ASSERT_STRING_EQ(cp, "host");
        ASSERT_PTR_NE(str, NULL);
        port = hpdelim(&str);
        ASSERT_STRING_EQ(port, "1234");
        ASSERT_PTR_EQ(str, NULL);
        DONE_STRING();
        TEST_DONE();

        TEST_START("hpdelim [host]:port");
        START_STRING("[::1]:1234");
        cp = hpdelim(&str);
        ASSERT_STRING_EQ(cp, "[::1]");
        ASSERT_PTR_NE(str, NULL);
        port = hpdelim(&str);
        ASSERT_STRING_EQ(port, "1234");
        ASSERT_PTR_EQ(str, NULL);
        DONE_STRING();
        TEST_DONE();

        TEST_START("hpdelim missing ] error");
        START_STRING("[::1:1234");
        cp = hpdelim(&str);
        ASSERT_PTR_EQ(cp, NULL);
        DONE_STRING();
        TEST_DONE();

}