1  /*	$NetBSD: t_vis.c,v 1.7 2014/09/08 19:01:03 christos Exp $	*/
  2  
  3  /*-
  4   * Copyright (c) 2002 The NetBSD Foundation, Inc.
  5   * All rights reserved.
  6   *
  7   * This code was contributed to The NetBSD Foundation by Christos Zoulas.
  8   *
  9   * Redistribution and use in source and binary forms, with or without
 10   * modification, are permitted provided that the following conditions
 11   * are met:
 12   * 1. Redistributions of source code must retain the above copyright
 13   *    notice, this list of conditions and the following disclaimer.
 14   * 2. Redistributions in binary form must reproduce the above copyright
 15   *    notice, this list of conditions and the following disclaimer in the
 16   *    documentation and/or other materials provided with the distribution.
 17   *
 18   * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
 19   * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
 20   * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
 21   * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
 22   * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
 23   * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
 24   * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
 25   * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
 26   * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
 27   * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
 28   * POSSIBILITY OF SUCH DAMAGE.
 29   */
 30  
 31  #include <string.h>
 32  #include <stdlib.h>
 33  #include <err.h>
 34  #include <vis.h>
 35  
 36  #include <darwintest.h>
 37  
 38  static int styles[] = {
 39  	VIS_OCTAL,
 40  	VIS_CSTYLE,
 41  	VIS_SP,
 42  	VIS_TAB,
 43  	VIS_NL,
 44  	VIS_WHITE,
 45  	VIS_SAFE,
 46  #if 0	/* Not reversible */
 47  	VIS_NOSLASH,
 48  #endif
 49  	VIS_HTTP1808,
 50  	VIS_MIMESTYLE,
 51  #if 0	/* Not supported by vis(3) */
 52  	VIS_HTTP1866,
 53  #endif
 54  };
 55  
 56  #define SIZE	256
 57  
 58  T_DECL(strvis_basic, "strvis(3)")
 59  {
 60  	char *srcbuf, *dstbuf, *visbuf;
 61  	unsigned int i, j;
 62  
 63  	T_ASSERT_NOTNULL((srcbuf = malloc(SIZE)), NULL);
 64  	T_ASSERT_NOTNULL((dstbuf = malloc(SIZE + 1)), NULL);
 65  	T_ASSERT_NOTNULL((visbuf = malloc(SIZE * 4 + 1)), NULL);
 66  
 67  	for (i = 0; i < SIZE; i++)
 68  		srcbuf[i] = (char)i;
 69  
 70  	for (i = 0; i < sizeof(styles)/sizeof(styles[0]); i++) {
 71  		T_ASSERT_GT(strsvisx(visbuf, srcbuf, SIZE, styles[i], ""), 0, NULL);
 72  		memset(dstbuf, 0, SIZE);
 73  		T_ASSERT_GT(strunvisx(dstbuf, visbuf, 
 74  		    styles[i] & (VIS_HTTP1808|VIS_MIMESTYLE)), 0, NULL);
 75  		for (j = 0; j < SIZE; j++)
 76  			if (dstbuf[j] != (char)j)
 77  				T_FAIL("Failed for style %x, char %d [%d]", styles[i], j, dstbuf[j]);
 78  		if (dstbuf[SIZE] != '\0')
 79  			T_FAIL("Failed for style %x, the result must be null-terminated [%d]", styles[i], dstbuf[SIZE]);
 80  	}
 81  	free(dstbuf);
 82  	free(srcbuf);
 83  	free(visbuf);
 84  }
 85  
 86  T_DECL(strvis_null, "strvis(3) NULL")
 87  {
 88  	char dst[] = "fail";
 89  	strvis(dst, NULL, VIS_SAFE);
 90  	T_ASSERT_EQ(dst[0], '\0', NULL);
 91  	T_ASSERT_EQ(dst[1], 'a', NULL);
 92  }
 93  
 94  T_DECL(strvis_empty, "strvis(3) empty")
 95  {
 96  	char dst[] = "fail";
 97  	strvis(dst, "", VIS_SAFE);
 98  	T_ASSERT_EQ(dst[0], '\0', NULL);
 99  	T_ASSERT_EQ(dst[1], 'a', NULL);
100  }
101  
102  T_DECL(strunvis_hex, "strunvis(3) \\Xxx")
103  {
104  	static const struct {
105  		const char *e;
106  		const char *d;
107  		int error;
108  	} ed[] = {
109  		{ "\\xff", "\xff", 1 },
110  		{ "\\x1", "\x1", 1 },
111  		{ "\\x1\\x02", "\x1\x2", 2 },
112  		{ "\\x1x", "\x1x", 2 },
113  		{ "\\xx", "", -1 },
114  	};
115  	char uv[10];
116  
117  	for (size_t i = 0; i < sizeof(ed)/sizeof(ed[0]); i++) {
118  		T_ASSERT_EQ(strunvis(uv, ed[i].e), ed[i].error, NULL);
119  		if (ed[i].error > 0)
120  			T_ASSERT_EQ(memcmp(ed[i].d, uv, (unsigned long)ed[i].error), 0, NULL);
121  	}
122  }