1 |
|
---|
2 | /***********************************************************************
|
---|
3 |
|
---|
4 | $Id: literal.c 1416 2009-04-25 15:46:39Z gyoung $
|
---|
5 |
|
---|
6 | string quoting utilities
|
---|
7 | wildcarding utilities
|
---|
8 |
|
---|
9 | Copyright (c) 1993-98 M. Kimes
|
---|
10 | Copyright (c) 2004, 2007 Steven H.Levine
|
---|
11 |
|
---|
12 | Archive containers
|
---|
13 |
|
---|
14 | 01 Aug 04 SHL Rework fixup to avoid overflows
|
---|
15 | 16 Jun 06 SHL liternal: comments
|
---|
16 | 22 Jul 06 SHL Check more run time errors
|
---|
17 | 20 Aug 07 GKY Move #pragma alloc_text to end for OpenWatcom compat
|
---|
18 | 16 Nov 07 SHL Report fixup buffer overflow
|
---|
19 | 15 Mar 08 KOMH Fix wildcard for multiple dots
|
---|
20 | 24 Apr 09 SHL Rework wildcard for clarity
|
---|
21 |
|
---|
22 | ***********************************************************************/
|
---|
23 |
|
---|
24 | #include <stdlib.h>
|
---|
25 | #include <string.h>
|
---|
26 | #include <ctype.h>
|
---|
27 |
|
---|
28 | #define INCL_OS2
|
---|
29 | #define INCL_WIN
|
---|
30 | #define INCL_LONGLONG // dircnrs.h
|
---|
31 |
|
---|
32 | #include "fm3dll.h"
|
---|
33 | #include "errutil.h" // Dos_Error...
|
---|
34 | #include "literal.h"
|
---|
35 | #include "wrappers.h" // xmalloc
|
---|
36 | #include "fortify.h"
|
---|
37 |
|
---|
38 | static PSZ pszSrcFile = __FILE__;
|
---|
39 |
|
---|
40 | static INT index(const CHAR * s, const CHAR c);
|
---|
41 |
|
---|
42 | /* Get index of char in string
|
---|
43 | * @parm s string to search
|
---|
44 | * @parm c char to search for
|
---|
45 | * @return 0 relative index of c in s or -1
|
---|
46 | */
|
---|
47 |
|
---|
48 | static INT index(const CHAR * s, const CHAR c)
|
---|
49 | {
|
---|
50 | CHAR *p;
|
---|
51 |
|
---|
52 | p = strchr(s, c);
|
---|
53 | if (p == NULL || !*p)
|
---|
54 | return -1;
|
---|
55 | return (INT) (p - s);
|
---|
56 | }
|
---|
57 |
|
---|
58 | /* literal()
|
---|
59 | * Translate a string with \ escape tokens to binary equivalent
|
---|
60 | * Translates in place
|
---|
61 | *
|
---|
62 | * 1. \x1b translates to CHAR(0x1b)
|
---|
63 | * 2. \27 translates to CHAR(27)
|
---|
64 | * 3. \" translates to "
|
---|
65 | * 4. \' translates to '
|
---|
66 | * 5. \\ translates to \
|
---|
67 | * 6. \r translates to carriage return
|
---|
68 | * 7. \n translates to linefeed
|
---|
69 | * 8. \b translates to backspace
|
---|
70 | * 9. \t translates to tab
|
---|
71 | * 10. \a translates to bell
|
---|
72 | * 11. \f translates to formfeed
|
---|
73 | *
|
---|
74 | * Synopsis
|
---|
75 | * *s = "this\x20is\32a test of \\MSC\\CSM\7"
|
---|
76 | * literal(s);
|
---|
77 | *
|
---|
78 | * ( s now equals "this is a test of \MSC\CSM")
|
---|
79 | *
|
---|
80 | * Return converted character count like strlen()
|
---|
81 | * Count does not include terminating nul
|
---|
82 | */
|
---|
83 |
|
---|
84 | #define HEX "0123456789ABCDEF"
|
---|
85 | #define DEC "0123456789"
|
---|
86 |
|
---|
87 | UINT literal(PSZ pszBuf)
|
---|
88 | {
|
---|
89 | INT wpos;
|
---|
90 | INT iBuf;
|
---|
91 | UINT cBufBytes;
|
---|
92 | INT iBufSave;
|
---|
93 | PSZ pszOut;
|
---|
94 | PSZ pszWork;
|
---|
95 | CHAR wchar;
|
---|
96 |
|
---|
97 | if (!pszBuf || !*pszBuf)
|
---|
98 | return 0;
|
---|
99 | cBufBytes = strlen(pszBuf) + 1;
|
---|
100 | pszWork = pszOut = xmalloc(cBufBytes + 1, pszSrcFile, __LINE__);
|
---|
101 | if (!pszWork)
|
---|
102 | return 0;
|
---|
103 |
|
---|
104 | iBuf = 0; /* set index to first character */
|
---|
105 | while (pszBuf[iBuf]) {
|
---|
106 | switch (pszBuf[iBuf]) {
|
---|
107 | case '\\':
|
---|
108 | switch (pszBuf[iBuf + 1]) {
|
---|
109 | case 'x': /* hexadecimal */
|
---|
110 | wchar = 0;
|
---|
111 | iBuf += 2; /* get past "\x" */
|
---|
112 | if (index(HEX, (CHAR) toupper(pszBuf[iBuf])) != -1) {
|
---|
113 | iBufSave = iBuf;
|
---|
114 | while (((wpos = index(HEX, (CHAR) toupper(pszBuf[iBuf]))) != -1) &&
|
---|
115 | iBuf < iBufSave + 2) {
|
---|
116 | wchar = (CHAR) (wchar << 4) + (CHAR) wpos;
|
---|
117 | iBuf++;
|
---|
118 | }
|
---|
119 | }
|
---|
120 | else
|
---|
121 | wchar = 'x'; /* just an x */
|
---|
122 | iBuf--;
|
---|
123 | *pszOut++ = wchar;
|
---|
124 | break;
|
---|
125 |
|
---|
126 | case '\\': /* we want a "\" */
|
---|
127 | iBuf++;
|
---|
128 | *pszOut++ = '\\';
|
---|
129 | break;
|
---|
130 |
|
---|
131 | case 't': /* tab CHAR */
|
---|
132 | iBuf++;
|
---|
133 | *pszOut++ = '\t';
|
---|
134 | break;
|
---|
135 |
|
---|
136 | case 'n': /* new line */
|
---|
137 | iBuf++;
|
---|
138 | *pszOut++ = '\n';
|
---|
139 | break;
|
---|
140 |
|
---|
141 | case 'r': /* carr return */
|
---|
142 | iBuf++;
|
---|
143 | *pszOut++ = '\r';
|
---|
144 | break;
|
---|
145 |
|
---|
146 | case 'b': /* back space */
|
---|
147 | iBuf++;
|
---|
148 | *pszOut++ = '\b';
|
---|
149 | break;
|
---|
150 |
|
---|
151 | case 'f': /* formfeed */
|
---|
152 | iBuf++;
|
---|
153 | *pszOut++ = '\x0c';
|
---|
154 | break;
|
---|
155 |
|
---|
156 | case 'a': /* bell */
|
---|
157 | iBuf++;
|
---|
158 | *pszOut++ = '\07';
|
---|
159 | break;
|
---|
160 |
|
---|
161 | case '\'': /* single quote */
|
---|
162 | iBuf++;
|
---|
163 | *pszOut++ = '\'';
|
---|
164 | break;
|
---|
165 |
|
---|
166 | case '\"': /* double quote */
|
---|
167 |
|
---|
168 | iBuf++;
|
---|
169 | *pszOut++ = '\"';
|
---|
170 | break;
|
---|
171 |
|
---|
172 | default: /* decimal */
|
---|
173 | iBuf++; /* get past "\" */
|
---|
174 | wchar = 0;
|
---|
175 | if (index(DEC, pszBuf[iBuf]) != -1) {
|
---|
176 | iBufSave = iBuf;
|
---|
177 | do { /* cvt to binary */
|
---|
178 | wchar = (CHAR) (wchar * 10 + (pszBuf[iBuf++] - 48));
|
---|
179 | } while (index(DEC, pszBuf[iBuf]) != -1 && iBuf < iBufSave + 3);
|
---|
180 | iBuf--;
|
---|
181 | }
|
---|
182 | else
|
---|
183 | wchar = pszBuf[iBuf];
|
---|
184 | *pszOut++ = wchar;
|
---|
185 | break;
|
---|
186 | } // switch
|
---|
187 | break;
|
---|
188 |
|
---|
189 | default:
|
---|
190 | *pszOut++ = pszBuf[iBuf];
|
---|
191 | break;
|
---|
192 | } // switch
|
---|
193 | iBuf++;
|
---|
194 | } // while
|
---|
195 | *pszOut = 0; /* Always terminate, even if not string */
|
---|
196 |
|
---|
197 | cBufBytes = pszOut - pszWork; /* Calc string length excluding terminator */
|
---|
198 | memcpy(pszBuf, pszWork, cBufBytes + 1); /* Overwrite including terminator */
|
---|
199 | free(pszWork);
|
---|
200 |
|
---|
201 | return cBufBytes; /* Return string length */
|
---|
202 | }
|
---|
203 |
|
---|
204 | /** Check wildcard match
|
---|
205 | * @parm pszBuf Buffer to check
|
---|
206 | * @parm pszWildCard wildcard to match
|
---|
207 | * @parm fIgnorePathSep TRUE if match ignores path separators
|
---|
208 | * @return TRUE if matched else FALSE
|
---|
209 | * @fixme need to rework to scan from right to left if not ignoring path separators
|
---|
210 | */
|
---|
211 |
|
---|
212 | BOOL wildcard(const PSZ pszBufIn, const PSZ pszWildCardIn,
|
---|
213 | const BOOL fIgnorePathSep)
|
---|
214 | {
|
---|
215 | PSZ pszBuf = pszBufIn;
|
---|
216 | PSZ pszWild = pszWildCardIn;
|
---|
217 |
|
---|
218 | while (*pszBuf && *pszWild) {
|
---|
219 | switch (*pszWild) {
|
---|
220 | case '*' :
|
---|
221 | {
|
---|
222 | PSZ pszLook;
|
---|
223 |
|
---|
224 | // find next non-wild character in wildcard
|
---|
225 | while (*pszWild && ( *pszWild == '*' || *pszWild == '?'))
|
---|
226 | pszWild++;
|
---|
227 |
|
---|
228 | // if last char of wildcard is *, got match
|
---|
229 | if (!*pszWild)
|
---|
230 | return TRUE;
|
---|
231 |
|
---|
232 | pszLook = pszBuf;
|
---|
233 | while (*pszLook) {
|
---|
234 | // scan until match, eos or path separator (maybe)
|
---|
235 | while (*pszLook && toupper(*pszLook) != toupper(*pszWild) &&
|
---|
236 | (fIgnorePathSep || ( *pszLook != '/' && *pszLook != '\\')))
|
---|
237 | pszLook++;
|
---|
238 |
|
---|
239 | // If eos or path separator (maybe), stop scan
|
---|
240 | if (!*pszLook || (!fIgnorePathSep && (*pszLook == '/' || *pszLook == '\\')))
|
---|
241 | break;
|
---|
242 |
|
---|
243 | // Not ignoring path separators, match next path component
|
---|
244 | if (wildcard(pszLook, pszWild, fIgnorePathSep) == TRUE)
|
---|
245 | return TRUE;
|
---|
246 |
|
---|
247 | pszLook++;
|
---|
248 | } // while
|
---|
249 |
|
---|
250 | pszBuf = pszLook;
|
---|
251 | break;
|
---|
252 | }
|
---|
253 |
|
---|
254 | case '?' : // character substitution
|
---|
255 | pszWild++;
|
---|
256 |
|
---|
257 | if (fIgnorePathSep || (*pszBuf != '.' && *pszBuf != '/' && *pszBuf != '\\'))
|
---|
258 | pszBuf++; // skip (match) next character
|
---|
259 | break;
|
---|
260 |
|
---|
261 | default :
|
---|
262 | if (fIgnorePathSep || (*pszBuf != '/' && *pszBuf != '\\') ||
|
---|
263 | (*pszWild != '/' && *pszWild != '\\')) {
|
---|
264 | if (toupper( *pszBuf ) != toupper( *pszWild))
|
---|
265 | return FALSE;
|
---|
266 | }
|
---|
267 |
|
---|
268 | pszWild++;
|
---|
269 | pszBuf++;
|
---|
270 | break;
|
---|
271 | } // switch
|
---|
272 | } // while
|
---|
273 |
|
---|
274 | if (!*pszBuf) {
|
---|
275 | // Skip trailing * and ?
|
---|
276 | while (*pszWild && (*pszWild == '?' || *pszWild == '*'))
|
---|
277 | pszWild++;
|
---|
278 |
|
---|
279 | if (!fIgnorePathSep) {
|
---|
280 | // remove trailing .
|
---|
281 | while (*pszWild && *pszWild == '.')
|
---|
282 | pszWild++;
|
---|
283 | }
|
---|
284 | }
|
---|
285 |
|
---|
286 | return (*pszBuf == *pszWild);
|
---|
287 | }
|
---|
288 |
|
---|
289 | // fixup - quote literal character array
|
---|
290 |
|
---|
291 | PSZ fixup(const PCH pachIn, PSZ pszOutBuf, const UINT cBufBytes,
|
---|
292 | const UINT cInBytes)
|
---|
293 | {
|
---|
294 | PCH pchIn = pachIn;
|
---|
295 | PCH pchOut = pszOutBuf;
|
---|
296 |
|
---|
297 | // input is a character array, not a string - may not be null terminated
|
---|
298 | // cBufBytes is buffer size
|
---|
299 | if (pachIn) {
|
---|
300 | // 16 Nov 07 SHL fixme to optimize counting and speed
|
---|
301 | // Ensure room for null and possible \ escape
|
---|
302 | while (pchIn - pachIn < cInBytes) {
|
---|
303 | if (pchOut - pszOutBuf + 4 >= cBufBytes) {
|
---|
304 | *pchOut = 0;
|
---|
305 | Runtime_Error(pszSrcFile, __LINE__, "buffer too small for %s", pszOutBuf);
|
---|
306 | break;
|
---|
307 | }
|
---|
308 |
|
---|
309 | if (!isprint(*pchIn)) {
|
---|
310 | if (*pchIn == '\r') {
|
---|
311 | *pchOut++ = '\\';
|
---|
312 | *pchOut++ = 'r';
|
---|
313 | }
|
---|
314 | else if (*pchIn == '\n') {
|
---|
315 | *pchOut++ = '\\';
|
---|
316 | *pchOut++ = 'n';
|
---|
317 | }
|
---|
318 | else if (*pchIn == '\b') {
|
---|
319 | *pchOut++ = '\\';
|
---|
320 | *pchOut++ = 'b';
|
---|
321 | }
|
---|
322 | else {
|
---|
323 | sprintf(pchOut, "\\x%02x", (UCHAR)*pchIn);
|
---|
324 | pchOut += 4;
|
---|
325 | }
|
---|
326 | pchIn++;
|
---|
327 | }
|
---|
328 | else if (*pchIn == '\\') {
|
---|
329 | *pchOut++ = '\\';
|
---|
330 | *pchOut++ = '\\';
|
---|
331 | pchIn++;
|
---|
332 | }
|
---|
333 | else
|
---|
334 | *pchOut++ = *pchIn++;
|
---|
335 | } // while
|
---|
336 |
|
---|
337 | } // if pachIn
|
---|
338 | *pchOut = 0;
|
---|
339 | return pszOutBuf;
|
---|
340 | }
|
---|
341 |
|
---|
342 | #pragma alloc_text(LITERAL,literal,index,fixup,wildcard)
|
---|
343 |
|
---|