1 | /* |
2 | Copyright (c) 2012, Broadcom Europe Ltd |
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 are met: |
7 | * Redistributions of source code must retain the above copyright |
8 | notice, this list of conditions and the following disclaimer. |
9 | * Redistributions in binary form must reproduce the above copyright |
10 | notice, this list of conditions and the following disclaimer in the |
11 | documentation and/or other materials provided with the distribution. |
12 | * Neither the name of the copyright holder nor the |
13 | names of its contributors may be used to endorse or promote products |
14 | derived from this software without specific prior written permission. |
15 | |
16 | THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND |
17 | ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED |
18 | WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE |
19 | DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY |
20 | DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES |
21 | (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; |
22 | LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND |
23 | ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
24 | (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS |
25 | SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
26 | */ |
27 | |
28 | #include "interface/vcos/vcos.h" |
29 | |
30 | #if defined( __KERNEL__ ) |
31 | #include <linux/string.h> |
32 | #include <linux/module.h> |
33 | #else |
34 | #include <string.h> |
35 | #endif |
36 | |
37 | #include <stdarg.h> |
38 | |
39 | /** Like vsnprintf, except it places the output at the specified offset. |
40 | * Output is truncated to fit in buflen bytes, and is guaranteed to be NUL-terminated. |
41 | * Returns the string length before/without truncation. |
42 | */ |
43 | size_t vcos_safe_vsprintf(char *buf, size_t buflen, size_t offset, const char *fmt, va_list ap) |
44 | { |
45 | size_t space = (offset < buflen) ? (buflen - offset) : 0; |
46 | |
47 | offset += vcos_vsnprintf(buf ? (buf + offset) : NULL, space, fmt, ap); |
48 | |
49 | return offset; |
50 | } |
51 | |
52 | /** Like snprintf, except it places the output at the specified offset. |
53 | * Output is truncated to fit in buflen bytes, and is guaranteed to be NUL-terminated. |
54 | * Returns the string length before/without truncation. |
55 | */ |
56 | size_t vcos_safe_sprintf(char *buf, size_t buflen, size_t offset, const char *fmt, ...) |
57 | { |
58 | size_t space = (offset < buflen) ? (buflen - offset) : 0; |
59 | va_list ap; |
60 | |
61 | va_start(ap, fmt); |
62 | |
63 | offset += vcos_vsnprintf(buf ? (buf + offset) : NULL, space, fmt, ap); |
64 | |
65 | va_end(ap); |
66 | |
67 | return offset; |
68 | } |
69 | |
70 | /** Copies string src to dst at the specified offset. |
71 | * Output is truncated to fit in dstlen bytes, i.e. the string is at most |
72 | * (buflen - 1) characters long. Unlike strncpy, exactly one NUL is written |
73 | * to dst, which is always NUL-terminated. |
74 | * Returns the string length before/without truncation. |
75 | */ |
76 | size_t vcos_safe_strcpy(char *dst, const char *src, size_t dstlen, size_t offset) |
77 | { |
78 | if (offset < dstlen) |
79 | { |
80 | const char *p = src; |
81 | char *endp = dst + dstlen -1; |
82 | |
83 | dst += offset; |
84 | |
85 | for (; *p!='\0' && dst != endp; dst++, p++) |
86 | *dst = *p; |
87 | *dst = '\0'; |
88 | } |
89 | offset += strlen(src); |
90 | |
91 | return offset; |
92 | } |
93 | |