uu_alloc.c revision 7887:b6618727fabf
1/*
2 * CDDL HEADER START
3 *
4 * The contents of this file are subject to the terms of the
5 * Common Development and Distribution License (the "License").
6 * You may not use this file except in compliance with the License.
7 *
8 * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
9 * or http://www.opensolaris.org/os/licensing.
10 * See the License for the specific language governing permissions
11 * and limitations under the License.
12 *
13 * When distributing Covered Code, include this CDDL HEADER in each
14 * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
15 * If applicable, add the following below this CDDL HEADER, with the
16 * fields enclosed by brackets "[]" replaced with your own identifying
17 * information: Portions Copyright [yyyy] [name of copyright owner]
18 *
19 * CDDL HEADER END
20 */
21/*
22 * Copyright 2008 Sun Microsystems, Inc.  All rights reserved.
23 * Use is subject to license terms.
24 */
25
26#include "libuutil_common.h"
27
28#include <stdarg.h>
29#include <stdio.h>
30#include <stdlib.h>
31#include <string.h>
32
33void *
34uu_zalloc(size_t n)
35{
36	void *p = malloc(n);
37
38	if (p == NULL) {
39		uu_set_error(UU_ERROR_SYSTEM);
40		return (NULL);
41	}
42
43	(void) memset(p, 0, n);
44
45	return (p);
46}
47
48void
49uu_free(void *p)
50{
51	free(p);
52}
53
54char *
55uu_strdup(const char *str)
56{
57	char *buf = NULL;
58
59	if (str != NULL) {
60		size_t sz;
61
62		sz = strlen(str) + 1;
63		buf = uu_zalloc(sz);
64		if (buf != NULL)
65			(void) memcpy(buf, str, sz);
66	}
67	return (buf);
68}
69
70char *
71uu_msprintf(const char *format, ...)
72{
73	va_list args;
74	char attic[1];
75	uint_t M, m;
76	char *b;
77
78	va_start(args, format);
79	M = vsnprintf(attic, 1, format, args);
80	va_end(args);
81
82	for (;;) {
83		m = M;
84		if ((b = uu_zalloc(m + 1)) == NULL)
85			return (NULL);
86
87		va_start(args, format);
88		M = vsnprintf(b, m + 1, format, args);
89		va_end(args);
90
91		if (M == m)
92			break;		/* sizes match */
93
94		uu_free(b);
95	}
96
97	return (b);
98}
99