1 #ifndef _BABELTRACE_COMPAT_STRING_H
2 #define _BABELTRACE_COMPAT_STRING_H
3 
4 /*
5  * Copyright (C) 2013 Mathieu Desnoyers <mathieu.desnoyers@efficios.com>
6  *
7  * Permission is hereby granted, free of charge, to any person obtaining a copy
8  * of this software and associated documentation files (the "Software"), to deal
9  * in the Software without restriction, including without limitation the rights
10  * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
11  * copies of the Software, and to permit persons to whom the Software is
12  * furnished to do so, subject to the following conditions:
13  *
14  * The above copyright notice and this permission notice shall be included in
15  * all copies or substantial portions of the Software.
16  *
17  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
18  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
19  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
20  * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
21  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
22  * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
23  * SOFTWARE.
24  */
25 
26 #include <string.h>
27 #include <stdlib.h>
28 
29 #ifdef HAVE_STRNLEN
30 static inline
bt_strnlen(const char * str,size_t max)31 size_t bt_strnlen(const char *str, size_t max)
32 {
33 	return strnlen(str, max);
34 }
35 #else
36 static inline
bt_strnlen(const char * str,size_t max)37 size_t bt_strnlen(const char *str, size_t max)
38 {
39 	size_t ret;
40 	const char *end;
41 
42 	end = memchr(str, 0, max);
43 
44 	if (end) {
45 		ret = (size_t) (end - str);
46 	} else {
47 		ret = max;
48 	}
49 
50 	return ret;
51 }
52 #endif /* HAVE_STRNLEN */
53 
54 #ifdef HAVE_STRNDUP
55 static inline
bt_strndup(const char * s,size_t n)56 char *bt_strndup(const char *s, size_t n)
57 {
58 	return strndup(s, n);
59 }
60 #else
61 static inline
bt_strndup(const char * s,size_t n)62 char *bt_strndup(const char *s, size_t n)
63 {
64 	char *ret;
65 	size_t navail;
66 
67 	if (!s) {
68 		ret = NULL;
69 		goto end;
70 	}
71 
72 	/* min() */
73 	navail = strlen(s) + 1;
74 	if ((n + 1) < navail) {
75 		navail = n + 1;
76 	}
77 
78 	ret = malloc(navail);
79 	if (!ret) {
80 		goto end;
81 	}
82 
83 	memcpy(ret, s, navail);
84 	ret[navail - 1] = '\0';
85 end:
86 	return ret;
87 }
88 #endif /* HAVE_STRNDUP */
89 
90 #endif /* _BABELTRACE_COMPAT_STRING_H */
91