xprintf_quote.c revision 330897 
1/*-
2 * SPDX-License-Identifier: BSD-2-Clause-FreeBSD
3 *
4 * Copyright (c) 2005 Poul-Henning Kamp
5 * All rights reserved.
6 *
7 * Redistribution and use in source and binary forms, with or without
8 * modification, are permitted provided that the following conditions
9 * are met:
10 * 1. Redistributions of source code must retain the above copyright
11 *    notice, this list of conditions and the following disclaimer.
12 * 2. Redistributions in binary form must reproduce the above copyright
13 *    notice, this list of conditions and the following disclaimer in the
14 *    documentation and/or other materials provided with the distribution.
15 *
16 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
17 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
18 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
19 * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
20 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
21 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
22 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
23 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
24 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
25 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
26 * SUCH DAMAGE.
27 *
28 * $FreeBSD: stable/11/lib/libc/stdio/xprintf_quote.c 330897 2018-03-14 03:19:51Z eadler $
29 */
30
31#include <namespace.h>
32#include <stdio.h>
33#include <string.h>
34#include <stdlib.h>
35#include <ctype.h>
36#include <wchar.h>
37#include <vis.h>
38#include <assert.h>
39#include <sys/time.h>
40#include "printf.h"
41
42int
43__printf_arginfo_quote(const struct printf_info *pi __unused, size_t n, int *argt)
44{
45
46	assert(n >= 1);
47	argt[0] = PA_POINTER;
48	return (1);
49}
50
51int
52__printf_render_quote(struct __printf_io *io, const struct printf_info *pi __unused, const void *const *arg)
53{
54	const char *str, *p, *t, *o;
55	char r[5];
56	int i, ret;
57
58	str = *((const char *const *)arg[0]);
59	if (str == NULL)
60		return (__printf_out(io, pi, "\"(null)\"", 8));
61	if (*str == '\0')
62		return (__printf_out(io, pi, "\"\"", 2));
63
64	for (i = 0, p = str; *p; p++)
65		if (isspace(*p) || *p == '\\' || *p == '"')
66			i++;
67	if (!i)
68		return (__printf_out(io, pi, str, strlen(str)));
69
70	ret = __printf_out(io, pi, "\"", 1);
71	for (t = p = str; *p; p++) {
72		o = NULL;
73		if (*p == '\\')
74			o = "\\\\";
75		else if (*p == '\n')
76			o = "\\n";
77		else if (*p == '\r')
78			o = "\\r";
79		else if (*p == '\t')
80			o = "\\t";
81		else if (*p == ' ')
82			o = " ";
83		else if (*p == '"')
84			o = "\\\"";
85		else if (isspace(*p)) {
86			sprintf(r, "\\%03o", *p);
87			o = r;
88		} else
89			continue;
90		if (p != t)
91			ret += __printf_out(io, pi, t, p - t);
92		ret += __printf_out(io, pi, o, strlen(o));
93		t = p + 1;
94	}
95	if (p != t)
96		ret += __printf_out(io, pi, t, p - t);
97	ret += __printf_out(io, pi, "\"", 1);
98	__printf_flush(io);
99	return(ret);
100}
101