diff options
author | Ingo Molnar <mingo@elte.hu> | 2009-06-06 14:33:43 -0400 |
---|---|---|
committer | Ingo Molnar <mingo@elte.hu> | 2009-06-06 14:33:43 -0400 |
commit | 864709302a80f26fa9da3be5b47304f0b8bae192 (patch) | |
tree | 8c2bab78f141fe43a38914bd3e3aae0a88f958e5 /tools/perf/util/quote.c | |
parent | 75b5032212641f6d38ac041416945e70da833b68 (diff) |
perf_counter tools: Move from Documentation/perf_counter/ to tools/perf/
Several people have suggested that 'perf' has become a full-fledged
tool that should be moved out of Documentation/. Move it to the
(new) tools/ directory.
Cc: Peter Zijlstra <a.p.zijlstra@chello.nl>
Cc: Mike Galbraith <efault@gmx.de>
Cc: Paul Mackerras <paulus@samba.org>
Cc: Arnaldo Carvalho de Melo <acme@redhat.com>
LKML-Reference: <new-submission>
Signed-off-by: Ingo Molnar <mingo@elte.hu>
Diffstat (limited to 'tools/perf/util/quote.c')
-rw-r--r-- | tools/perf/util/quote.c | 481 |
1 files changed, 481 insertions, 0 deletions
diff --git a/tools/perf/util/quote.c b/tools/perf/util/quote.c new file mode 100644 index 000000000000..f18c5212bc92 --- /dev/null +++ b/tools/perf/util/quote.c | |||
@@ -0,0 +1,481 @@ | |||
1 | #include "cache.h" | ||
2 | #include "quote.h" | ||
3 | |||
4 | int quote_path_fully = 1; | ||
5 | |||
6 | /* Help to copy the thing properly quoted for the shell safety. | ||
7 | * any single quote is replaced with '\'', any exclamation point | ||
8 | * is replaced with '\!', and the whole thing is enclosed in a | ||
9 | * | ||
10 | * E.g. | ||
11 | * original sq_quote result | ||
12 | * name ==> name ==> 'name' | ||
13 | * a b ==> a b ==> 'a b' | ||
14 | * a'b ==> a'\''b ==> 'a'\''b' | ||
15 | * a!b ==> a'\!'b ==> 'a'\!'b' | ||
16 | */ | ||
17 | static inline int need_bs_quote(char c) | ||
18 | { | ||
19 | return (c == '\'' || c == '!'); | ||
20 | } | ||
21 | |||
22 | void sq_quote_buf(struct strbuf *dst, const char *src) | ||
23 | { | ||
24 | char *to_free = NULL; | ||
25 | |||
26 | if (dst->buf == src) | ||
27 | to_free = strbuf_detach(dst, NULL); | ||
28 | |||
29 | strbuf_addch(dst, '\''); | ||
30 | while (*src) { | ||
31 | size_t len = strcspn(src, "'!"); | ||
32 | strbuf_add(dst, src, len); | ||
33 | src += len; | ||
34 | while (need_bs_quote(*src)) { | ||
35 | strbuf_addstr(dst, "'\\"); | ||
36 | strbuf_addch(dst, *src++); | ||
37 | strbuf_addch(dst, '\''); | ||
38 | } | ||
39 | } | ||
40 | strbuf_addch(dst, '\''); | ||
41 | free(to_free); | ||
42 | } | ||
43 | |||
44 | void sq_quote_print(FILE *stream, const char *src) | ||
45 | { | ||
46 | char c; | ||
47 | |||
48 | fputc('\'', stream); | ||
49 | while ((c = *src++)) { | ||
50 | if (need_bs_quote(c)) { | ||
51 | fputs("'\\", stream); | ||
52 | fputc(c, stream); | ||
53 | fputc('\'', stream); | ||
54 | } else { | ||
55 | fputc(c, stream); | ||
56 | } | ||
57 | } | ||
58 | fputc('\'', stream); | ||
59 | } | ||
60 | |||
61 | void sq_quote_argv(struct strbuf *dst, const char** argv, size_t maxlen) | ||
62 | { | ||
63 | int i; | ||
64 | |||
65 | /* Copy into destination buffer. */ | ||
66 | strbuf_grow(dst, 255); | ||
67 | for (i = 0; argv[i]; ++i) { | ||
68 | strbuf_addch(dst, ' '); | ||
69 | sq_quote_buf(dst, argv[i]); | ||
70 | if (maxlen && dst->len > maxlen) | ||
71 | die("Too many or long arguments"); | ||
72 | } | ||
73 | } | ||
74 | |||
75 | char *sq_dequote_step(char *arg, char **next) | ||
76 | { | ||
77 | char *dst = arg; | ||
78 | char *src = arg; | ||
79 | char c; | ||
80 | |||
81 | if (*src != '\'') | ||
82 | return NULL; | ||
83 | for (;;) { | ||
84 | c = *++src; | ||
85 | if (!c) | ||
86 | return NULL; | ||
87 | if (c != '\'') { | ||
88 | *dst++ = c; | ||
89 | continue; | ||
90 | } | ||
91 | /* We stepped out of sq */ | ||
92 | switch (*++src) { | ||
93 | case '\0': | ||
94 | *dst = 0; | ||
95 | if (next) | ||
96 | *next = NULL; | ||
97 | return arg; | ||
98 | case '\\': | ||
99 | c = *++src; | ||
100 | if (need_bs_quote(c) && *++src == '\'') { | ||
101 | *dst++ = c; | ||
102 | continue; | ||
103 | } | ||
104 | /* Fallthrough */ | ||
105 | default: | ||
106 | if (!next || !isspace(*src)) | ||
107 | return NULL; | ||
108 | do { | ||
109 | c = *++src; | ||
110 | } while (isspace(c)); | ||
111 | *dst = 0; | ||
112 | *next = src; | ||
113 | return arg; | ||
114 | } | ||
115 | } | ||
116 | } | ||
117 | |||
118 | char *sq_dequote(char *arg) | ||
119 | { | ||
120 | return sq_dequote_step(arg, NULL); | ||
121 | } | ||
122 | |||
123 | int sq_dequote_to_argv(char *arg, const char ***argv, int *nr, int *alloc) | ||
124 | { | ||
125 | char *next = arg; | ||
126 | |||
127 | if (!*arg) | ||
128 | return 0; | ||
129 | do { | ||
130 | char *dequoted = sq_dequote_step(next, &next); | ||
131 | if (!dequoted) | ||
132 | return -1; | ||
133 | ALLOC_GROW(*argv, *nr + 1, *alloc); | ||
134 | (*argv)[(*nr)++] = dequoted; | ||
135 | } while (next); | ||
136 | |||
137 | return 0; | ||
138 | } | ||
139 | |||
140 | /* 1 means: quote as octal | ||
141 | * 0 means: quote as octal if (quote_path_fully) | ||
142 | * -1 means: never quote | ||
143 | * c: quote as "\\c" | ||
144 | */ | ||
145 | #define X8(x) x, x, x, x, x, x, x, x | ||
146 | #define X16(x) X8(x), X8(x) | ||
147 | static signed char const sq_lookup[256] = { | ||
148 | /* 0 1 2 3 4 5 6 7 */ | ||
149 | /* 0x00 */ 1, 1, 1, 1, 1, 1, 1, 'a', | ||
150 | /* 0x08 */ 'b', 't', 'n', 'v', 'f', 'r', 1, 1, | ||
151 | /* 0x10 */ X16(1), | ||
152 | /* 0x20 */ -1, -1, '"', -1, -1, -1, -1, -1, | ||
153 | /* 0x28 */ X16(-1), X16(-1), X16(-1), | ||
154 | /* 0x58 */ -1, -1, -1, -1,'\\', -1, -1, -1, | ||
155 | /* 0x60 */ X16(-1), X8(-1), | ||
156 | /* 0x78 */ -1, -1, -1, -1, -1, -1, -1, 1, | ||
157 | /* 0x80 */ /* set to 0 */ | ||
158 | }; | ||
159 | |||
160 | static inline int sq_must_quote(char c) | ||
161 | { | ||
162 | return sq_lookup[(unsigned char)c] + quote_path_fully > 0; | ||
163 | } | ||
164 | |||
165 | /* returns the longest prefix not needing a quote up to maxlen if positive. | ||
166 | This stops at the first \0 because it's marked as a character needing an | ||
167 | escape */ | ||
168 | static size_t next_quote_pos(const char *s, ssize_t maxlen) | ||
169 | { | ||
170 | size_t len; | ||
171 | if (maxlen < 0) { | ||
172 | for (len = 0; !sq_must_quote(s[len]); len++); | ||
173 | } else { | ||
174 | for (len = 0; len < maxlen && !sq_must_quote(s[len]); len++); | ||
175 | } | ||
176 | return len; | ||
177 | } | ||
178 | |||
179 | /* | ||
180 | * C-style name quoting. | ||
181 | * | ||
182 | * (1) if sb and fp are both NULL, inspect the input name and counts the | ||
183 | * number of bytes that are needed to hold c_style quoted version of name, | ||
184 | * counting the double quotes around it but not terminating NUL, and | ||
185 | * returns it. | ||
186 | * However, if name does not need c_style quoting, it returns 0. | ||
187 | * | ||
188 | * (2) if sb or fp are not NULL, it emits the c_style quoted version | ||
189 | * of name, enclosed with double quotes if asked and needed only. | ||
190 | * Return value is the same as in (1). | ||
191 | */ | ||
192 | static size_t quote_c_style_counted(const char *name, ssize_t maxlen, | ||
193 | struct strbuf *sb, FILE *fp, int no_dq) | ||
194 | { | ||
195 | #undef EMIT | ||
196 | #define EMIT(c) \ | ||
197 | do { \ | ||
198 | if (sb) strbuf_addch(sb, (c)); \ | ||
199 | if (fp) fputc((c), fp); \ | ||
200 | count++; \ | ||
201 | } while (0) | ||
202 | #define EMITBUF(s, l) \ | ||
203 | do { \ | ||
204 | int __ret; \ | ||
205 | if (sb) strbuf_add(sb, (s), (l)); \ | ||
206 | if (fp) __ret = fwrite((s), (l), 1, fp); \ | ||
207 | count += (l); \ | ||
208 | } while (0) | ||
209 | |||
210 | size_t len, count = 0; | ||
211 | const char *p = name; | ||
212 | |||
213 | for (;;) { | ||
214 | int ch; | ||
215 | |||
216 | len = next_quote_pos(p, maxlen); | ||
217 | if (len == maxlen || !p[len]) | ||
218 | break; | ||
219 | |||
220 | if (!no_dq && p == name) | ||
221 | EMIT('"'); | ||
222 | |||
223 | EMITBUF(p, len); | ||
224 | EMIT('\\'); | ||
225 | p += len; | ||
226 | ch = (unsigned char)*p++; | ||
227 | if (sq_lookup[ch] >= ' ') { | ||
228 | EMIT(sq_lookup[ch]); | ||
229 | } else { | ||
230 | EMIT(((ch >> 6) & 03) + '0'); | ||
231 | EMIT(((ch >> 3) & 07) + '0'); | ||
232 | EMIT(((ch >> 0) & 07) + '0'); | ||
233 | } | ||
234 | } | ||
235 | |||
236 | EMITBUF(p, len); | ||
237 | if (p == name) /* no ending quote needed */ | ||
238 | return 0; | ||
239 | |||
240 | if (!no_dq) | ||
241 | EMIT('"'); | ||
242 | return count; | ||
243 | } | ||
244 | |||
245 | size_t quote_c_style(const char *name, struct strbuf *sb, FILE *fp, int nodq) | ||
246 | { | ||
247 | return quote_c_style_counted(name, -1, sb, fp, nodq); | ||
248 | } | ||
249 | |||
250 | void quote_two_c_style(struct strbuf *sb, const char *prefix, const char *path, int nodq) | ||
251 | { | ||
252 | if (quote_c_style(prefix, NULL, NULL, 0) || | ||
253 | quote_c_style(path, NULL, NULL, 0)) { | ||
254 | if (!nodq) | ||
255 | strbuf_addch(sb, '"'); | ||
256 | quote_c_style(prefix, sb, NULL, 1); | ||
257 | quote_c_style(path, sb, NULL, 1); | ||
258 | if (!nodq) | ||
259 | strbuf_addch(sb, '"'); | ||
260 | } else { | ||
261 | strbuf_addstr(sb, prefix); | ||
262 | strbuf_addstr(sb, path); | ||
263 | } | ||
264 | } | ||
265 | |||
266 | void write_name_quoted(const char *name, FILE *fp, int terminator) | ||
267 | { | ||
268 | if (terminator) { | ||
269 | quote_c_style(name, NULL, fp, 0); | ||
270 | } else { | ||
271 | fputs(name, fp); | ||
272 | } | ||
273 | fputc(terminator, fp); | ||
274 | } | ||
275 | |||
276 | extern void write_name_quotedpfx(const char *pfx, size_t pfxlen, | ||
277 | const char *name, FILE *fp, int terminator) | ||
278 | { | ||
279 | int needquote = 0; | ||
280 | |||
281 | if (terminator) { | ||
282 | needquote = next_quote_pos(pfx, pfxlen) < pfxlen | ||
283 | || name[next_quote_pos(name, -1)]; | ||
284 | } | ||
285 | if (needquote) { | ||
286 | fputc('"', fp); | ||
287 | quote_c_style_counted(pfx, pfxlen, NULL, fp, 1); | ||
288 | quote_c_style(name, NULL, fp, 1); | ||
289 | fputc('"', fp); | ||
290 | } else { | ||
291 | int ret; | ||
292 | |||
293 | ret = fwrite(pfx, pfxlen, 1, fp); | ||
294 | fputs(name, fp); | ||
295 | } | ||
296 | fputc(terminator, fp); | ||
297 | } | ||
298 | |||
299 | /* quote path as relative to the given prefix */ | ||
300 | char *quote_path_relative(const char *in, int len, | ||
301 | struct strbuf *out, const char *prefix) | ||
302 | { | ||
303 | int needquote; | ||
304 | |||
305 | if (len < 0) | ||
306 | len = strlen(in); | ||
307 | |||
308 | /* "../" prefix itself does not need quoting, but "in" might. */ | ||
309 | needquote = next_quote_pos(in, len) < len; | ||
310 | strbuf_setlen(out, 0); | ||
311 | strbuf_grow(out, len); | ||
312 | |||
313 | if (needquote) | ||
314 | strbuf_addch(out, '"'); | ||
315 | if (prefix) { | ||
316 | int off = 0; | ||
317 | while (prefix[off] && off < len && prefix[off] == in[off]) | ||
318 | if (prefix[off] == '/') { | ||
319 | prefix += off + 1; | ||
320 | in += off + 1; | ||
321 | len -= off + 1; | ||
322 | off = 0; | ||
323 | } else | ||
324 | off++; | ||
325 | |||
326 | for (; *prefix; prefix++) | ||
327 | if (*prefix == '/') | ||
328 | strbuf_addstr(out, "../"); | ||
329 | } | ||
330 | |||
331 | quote_c_style_counted (in, len, out, NULL, 1); | ||
332 | |||
333 | if (needquote) | ||
334 | strbuf_addch(out, '"'); | ||
335 | if (!out->len) | ||
336 | strbuf_addstr(out, "./"); | ||
337 | |||
338 | return out->buf; | ||
339 | } | ||
340 | |||
341 | /* | ||
342 | * C-style name unquoting. | ||
343 | * | ||
344 | * Quoted should point at the opening double quote. | ||
345 | * + Returns 0 if it was able to unquote the string properly, and appends the | ||
346 | * result in the strbuf `sb'. | ||
347 | * + Returns -1 in case of error, and doesn't touch the strbuf. Though note | ||
348 | * that this function will allocate memory in the strbuf, so calling | ||
349 | * strbuf_release is mandatory whichever result unquote_c_style returns. | ||
350 | * | ||
351 | * Updates endp pointer to point at one past the ending double quote if given. | ||
352 | */ | ||
353 | int unquote_c_style(struct strbuf *sb, const char *quoted, const char **endp) | ||
354 | { | ||
355 | size_t oldlen = sb->len, len; | ||
356 | int ch, ac; | ||
357 | |||
358 | if (*quoted++ != '"') | ||
359 | return -1; | ||
360 | |||
361 | for (;;) { | ||
362 | len = strcspn(quoted, "\"\\"); | ||
363 | strbuf_add(sb, quoted, len); | ||
364 | quoted += len; | ||
365 | |||
366 | switch (*quoted++) { | ||
367 | case '"': | ||
368 | if (endp) | ||
369 | *endp = quoted; | ||
370 | return 0; | ||
371 | case '\\': | ||
372 | break; | ||
373 | default: | ||
374 | goto error; | ||
375 | } | ||
376 | |||
377 | switch ((ch = *quoted++)) { | ||
378 | case 'a': ch = '\a'; break; | ||
379 | case 'b': ch = '\b'; break; | ||
380 | case 'f': ch = '\f'; break; | ||
381 | case 'n': ch = '\n'; break; | ||
382 | case 'r': ch = '\r'; break; | ||
383 | case 't': ch = '\t'; break; | ||
384 | case 'v': ch = '\v'; break; | ||
385 | |||
386 | case '\\': case '"': | ||
387 | break; /* verbatim */ | ||
388 | |||
389 | /* octal values with first digit over 4 overflow */ | ||
390 | case '0': case '1': case '2': case '3': | ||
391 | ac = ((ch - '0') << 6); | ||
392 | if ((ch = *quoted++) < '0' || '7' < ch) | ||
393 | goto error; | ||
394 | ac |= ((ch - '0') << 3); | ||
395 | if ((ch = *quoted++) < '0' || '7' < ch) | ||
396 | goto error; | ||
397 | ac |= (ch - '0'); | ||
398 | ch = ac; | ||
399 | break; | ||
400 | default: | ||
401 | goto error; | ||
402 | } | ||
403 | strbuf_addch(sb, ch); | ||
404 | } | ||
405 | |||
406 | error: | ||
407 | strbuf_setlen(sb, oldlen); | ||
408 | return -1; | ||
409 | } | ||
410 | |||
411 | /* quoting as a string literal for other languages */ | ||
412 | |||
413 | void perl_quote_print(FILE *stream, const char *src) | ||
414 | { | ||
415 | const char sq = '\''; | ||
416 | const char bq = '\\'; | ||
417 | char c; | ||
418 | |||
419 | fputc(sq, stream); | ||
420 | while ((c = *src++)) { | ||
421 | if (c == sq || c == bq) | ||
422 | fputc(bq, stream); | ||
423 | fputc(c, stream); | ||
424 | } | ||
425 | fputc(sq, stream); | ||
426 | } | ||
427 | |||
428 | void python_quote_print(FILE *stream, const char *src) | ||
429 | { | ||
430 | const char sq = '\''; | ||
431 | const char bq = '\\'; | ||
432 | const char nl = '\n'; | ||
433 | char c; | ||
434 | |||
435 | fputc(sq, stream); | ||
436 | while ((c = *src++)) { | ||
437 | if (c == nl) { | ||
438 | fputc(bq, stream); | ||
439 | fputc('n', stream); | ||
440 | continue; | ||
441 | } | ||
442 | if (c == sq || c == bq) | ||
443 | fputc(bq, stream); | ||
444 | fputc(c, stream); | ||
445 | } | ||
446 | fputc(sq, stream); | ||
447 | } | ||
448 | |||
449 | void tcl_quote_print(FILE *stream, const char *src) | ||
450 | { | ||
451 | char c; | ||
452 | |||
453 | fputc('"', stream); | ||
454 | while ((c = *src++)) { | ||
455 | switch (c) { | ||
456 | case '[': case ']': | ||
457 | case '{': case '}': | ||
458 | case '$': case '\\': case '"': | ||
459 | fputc('\\', stream); | ||
460 | default: | ||
461 | fputc(c, stream); | ||
462 | break; | ||
463 | case '\f': | ||
464 | fputs("\\f", stream); | ||
465 | break; | ||
466 | case '\r': | ||
467 | fputs("\\r", stream); | ||
468 | break; | ||
469 | case '\n': | ||
470 | fputs("\\n", stream); | ||
471 | break; | ||
472 | case '\t': | ||
473 | fputs("\\t", stream); | ||
474 | break; | ||
475 | case '\v': | ||
476 | fputs("\\v", stream); | ||
477 | break; | ||
478 | } | ||
479 | } | ||
480 | fputc('"', stream); | ||
481 | } | ||