1*a9643ea8Slogwang /*-
2*a9643ea8Slogwang * Copyright (c) 2013-2015 Sandvine Inc. All rights reserved.
3*a9643ea8Slogwang * All rights reserved.
4*a9643ea8Slogwang *
5*a9643ea8Slogwang * Redistribution and use in source and binary forms, with or without
6*a9643ea8Slogwang * modification, are permitted provided that the following conditions
7*a9643ea8Slogwang * are met:
8*a9643ea8Slogwang * 1. Redistributions of source code must retain the above copyright
9*a9643ea8Slogwang * notice, this list of conditions and the following disclaimer.
10*a9643ea8Slogwang * 2. Redistributions in binary form must reproduce the above copyright
11*a9643ea8Slogwang * notice, this list of conditions and the following disclaimer in the
12*a9643ea8Slogwang * documentation and/or other materials provided with the distribution.
13*a9643ea8Slogwang *
14*a9643ea8Slogwang * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
15*a9643ea8Slogwang * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
16*a9643ea8Slogwang * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
17*a9643ea8Slogwang * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
18*a9643ea8Slogwang * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
19*a9643ea8Slogwang * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
20*a9643ea8Slogwang * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
21*a9643ea8Slogwang * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
22*a9643ea8Slogwang * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
23*a9643ea8Slogwang * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
24*a9643ea8Slogwang * SUCH DAMAGE.
25*a9643ea8Slogwang */
26*a9643ea8Slogwang
27*a9643ea8Slogwang #include <sys/cdefs.h>
28*a9643ea8Slogwang __FBSDID("$FreeBSD$");
29*a9643ea8Slogwang
30*a9643ea8Slogwang #include <sys/param.h>
31*a9643ea8Slogwang #include <sys/systm.h>
32*a9643ea8Slogwang #include <sys/malloc.h>
33*a9643ea8Slogwang
34*a9643ea8Slogwang #include <machine/stdarg.h>
35*a9643ea8Slogwang
36*a9643ea8Slogwang int
vasprintf(char ** buf,struct malloc_type * mtp,const char * format,va_list va)37*a9643ea8Slogwang vasprintf(char **buf, struct malloc_type *mtp, const char *format, va_list va)
38*a9643ea8Slogwang {
39*a9643ea8Slogwang int len, ret;
40*a9643ea8Slogwang va_list tmp_va;
41*a9643ea8Slogwang char dummy;
42*a9643ea8Slogwang
43*a9643ea8Slogwang va_copy(tmp_va, va);
44*a9643ea8Slogwang len = vsnprintf(&dummy, 0, format, tmp_va);
45*a9643ea8Slogwang va_end(tmp_va);
46*a9643ea8Slogwang if (len < 0) {
47*a9643ea8Slogwang *buf = NULL;
48*a9643ea8Slogwang return (len);
49*a9643ea8Slogwang }
50*a9643ea8Slogwang
51*a9643ea8Slogwang /* Account for null terminator. */
52*a9643ea8Slogwang len += 1;
53*a9643ea8Slogwang *buf = malloc(len, mtp, M_NOWAIT);
54*a9643ea8Slogwang if (*buf == NULL)
55*a9643ea8Slogwang return (-1);
56*a9643ea8Slogwang
57*a9643ea8Slogwang ret = vsnprintf(*buf, len, format, va);
58*a9643ea8Slogwang if (ret < 0) {
59*a9643ea8Slogwang free(*buf, mtp);
60*a9643ea8Slogwang *buf = NULL;
61*a9643ea8Slogwang }
62*a9643ea8Slogwang
63*a9643ea8Slogwang return (ret);
64*a9643ea8Slogwang }
65*a9643ea8Slogwang
66*a9643ea8Slogwang int
asprintf(char ** buf,struct malloc_type * mtp,const char * format,...)67*a9643ea8Slogwang asprintf(char **buf, struct malloc_type *mtp, const char *format, ...)
68*a9643ea8Slogwang {
69*a9643ea8Slogwang int ret;
70*a9643ea8Slogwang va_list va;
71*a9643ea8Slogwang
72*a9643ea8Slogwang va_start(va, format);
73*a9643ea8Slogwang ret = vasprintf(buf, mtp, format, va);
74*a9643ea8Slogwang va_end(va);
75*a9643ea8Slogwang
76*a9643ea8Slogwang return (ret);
77*a9643ea8Slogwang }
78