diff options
author | Damien Miller <djm@mindrot.org> | 1999-11-26 12:21:24 +1100 |
---|---|---|
committer | Damien Miller <djm@mindrot.org> | 1999-11-26 12:21:24 +1100 |
commit | 42b81ffe7d1e98b362696076edbc56a66368053e (patch) | |
tree | 0c11756e8b9def63760aa59e6d2a195fcc380196 /bsd-snprintf.c | |
parent | bf3f6ef19b8d43b09598807cc30de7e66be37563 (diff) |
- Add definition for __P()
- Added [v]snprintf() replacement for systems that lack it
Diffstat (limited to 'bsd-snprintf.c')
-rw-r--r-- | bsd-snprintf.c | 162 |
1 files changed, 162 insertions, 0 deletions
diff --git a/bsd-snprintf.c b/bsd-snprintf.c new file mode 100644 index 000000000..11c4ff39c --- /dev/null +++ b/bsd-snprintf.c | |||
@@ -0,0 +1,162 @@ | |||
1 | /* | ||
2 | * Revision 12: http://theos.com/~deraadt/snprintf.c | ||
3 | * | ||
4 | * Copyright (c) 1997 Theo de Raadt | ||
5 | * | ||
6 | * Redistribution and use in source and binary forms, with or without | ||
7 | * modification, are permitted provided that the following conditions | ||
8 | * are met: | ||
9 | * 1. Redistributions of source code must retain the above copyright | ||
10 | * notice, this list of conditions and the following disclaimer. | ||
11 | * 2. Redistributions in binary form must reproduce the above copyright | ||
12 | * notice, this list of conditions and the following disclaimer in the | ||
13 | * documentation and/or other materials provided with the distribution. | ||
14 | * | ||
15 | * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR | ||
16 | * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES | ||
17 | * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. | ||
18 | * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, | ||
19 | * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT | ||
20 | * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, | ||
21 | * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY | ||
22 | * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT | ||
23 | * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF | ||
24 | * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. | ||
25 | */ | ||
26 | |||
27 | #include "config.h" | ||
28 | |||
29 | #if !defined(HAVE_SNPRINTF) || !defined(HAVE_VSNPRINTF) | ||
30 | |||
31 | #include <sys/param.h> | ||
32 | #include <sys/types.h> | ||
33 | #include <sys/mman.h> | ||
34 | #include <signal.h> | ||
35 | #include <stdio.h> | ||
36 | #include <unistd.h> | ||
37 | #include <string.h> | ||
38 | #if __STDC__ | ||
39 | #include <stdarg.h> | ||
40 | #include <stdlib.h> | ||
41 | #else | ||
42 | #include <varargs.h> | ||
43 | #endif | ||
44 | #include <setjmp.h> | ||
45 | |||
46 | #ifndef roundup | ||
47 | #define roundup (x, y) ((((x)+((y)-1))/(y))*(y)) | ||
48 | #endif | ||
49 | |||
50 | static int pgsize; | ||
51 | static char *curobj; | ||
52 | static int caught; | ||
53 | static sigjmp_buf bail; | ||
54 | |||
55 | #define EXTRABYTES 2 /* XXX: why 2? you don't want to know */ | ||
56 | |||
57 | static char * | ||
58 | msetup(str, n) | ||
59 | char *str; | ||
60 | size_t n; | ||
61 | { | ||
62 | char *e; | ||
63 | |||
64 | if (n == 0) | ||
65 | return NULL; | ||
66 | if (pgsize == 0) | ||
67 | pgsize = getpagesize(); | ||
68 | curobj = (char *)malloc(n + EXTRABYTES + pgsize * 2); | ||
69 | if (curobj == NULL) | ||
70 | return NULL; | ||
71 | e = curobj + n + EXTRABYTES; | ||
72 | e = (char *)roundup((unsigned long)e, pgsize); | ||
73 | if (mprotect(e, pgsize, PROT_NONE) == -1) { | ||
74 | free(curobj); | ||
75 | curobj = NULL; | ||
76 | return NULL; | ||
77 | } | ||
78 | e = e - n - EXTRABYTES; | ||
79 | *e = '\0'; | ||
80 | return (e); | ||
81 | } | ||
82 | |||
83 | static void | ||
84 | mcatch() | ||
85 | { | ||
86 | siglongjmp(bail, 1); | ||
87 | } | ||
88 | |||
89 | static void | ||
90 | mcleanup(str, n, p) | ||
91 | char *str; | ||
92 | size_t n; | ||
93 | char *p; | ||
94 | { | ||
95 | strncpy(str, p, n-1); | ||
96 | str[n-1] = '\0'; | ||
97 | if (mprotect((caddr_t)(p + n + EXTRABYTES), pgsize, | ||
98 | PROT_READ|PROT_WRITE|PROT_EXEC) == -1) | ||
99 | mprotect((caddr_t)(p + n + EXTRABYTES), pgsize, | ||
100 | PROT_READ|PROT_WRITE); | ||
101 | free(curobj); | ||
102 | } | ||
103 | |||
104 | #if !defined(HAVE_SNPRINTF) | ||
105 | int | ||
106 | #if __STDC__ | ||
107 | snprintf(char *str, size_t n, char const *fmt, ...) | ||
108 | #else | ||
109 | snprintf(str, n, fmt, va_alist) | ||
110 | char *str; | ||
111 | size_t n; | ||
112 | char *fmt; | ||
113 | va_dcl | ||
114 | #endif | ||
115 | { | ||
116 | va_list ap; | ||
117 | #if __STDC__ | ||
118 | va_start(ap, fmt); | ||
119 | #else | ||
120 | va_start(ap); | ||
121 | #endif | ||
122 | |||
123 | return (vsnprintf(str, n, fmt, ap)); | ||
124 | va_end(ap); | ||
125 | } | ||
126 | #endif /* !defined(HAVE_SNPRINTF) */ | ||
127 | |||
128 | #if !defined(HAVE_VSNPRINTF) | ||
129 | int | ||
130 | vsnprintf(str, n, fmt, ap) | ||
131 | char *str; | ||
132 | size_t n; | ||
133 | char *fmt; | ||
134 | char *ap; | ||
135 | { | ||
136 | struct sigaction osa, nsa; | ||
137 | char *p; | ||
138 | int ret = n + 1; /* if we bail, indicated we overflowed */ | ||
139 | |||
140 | memset(&nsa, 0, sizeof nsa); | ||
141 | nsa.sa_handler = mcatch; | ||
142 | sigemptyset(&nsa.sa_mask); | ||
143 | |||
144 | p = msetup(str, n); | ||
145 | if (p == NULL) { | ||
146 | *str = '\0'; | ||
147 | return 0; | ||
148 | } | ||
149 | if (sigsetjmp(bail, 1) == 0) { | ||
150 | if (sigaction(SIGSEGV, &nsa, &osa) == -1) { | ||
151 | mcleanup(str, n, p); | ||
152 | return (0); | ||
153 | } | ||
154 | ret = vsprintf(p, fmt, ap); | ||
155 | } | ||
156 | mcleanup(str, n, p); | ||
157 | (void) sigaction(SIGSEGV, &osa, NULL); | ||
158 | return (ret); | ||
159 | } | ||
160 | #endif /* !defined(HAVE_VSNPRINTF) */ | ||
161 | |||
162 | #endif /* !defined(HAVE_SNPRINTF) || !defined(HAVE_VSNPRINTF) */ | ||