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