Merge branch 'vendor/OPENSSH'
[dragonfly.git] / contrib / tcsh-6 / tc.str.c
1 /* $Header: /p/tcsh/cvsroot/tcsh/tc.str.c,v 3.30 2009/06/25 21:27:38 christos Exp $ */
2 /*
3  * tc.str.c: Short string package
4  *           This has been a lesson of how to write buggy code!
5  */
6 /*-
7  * Copyright (c) 1980, 1991 The Regents of the University of California.
8  * All rights reserved.
9  *
10  * Redistribution and use in source and binary forms, with or without
11  * modification, are permitted provided that the following conditions
12  * are met:
13  * 1. Redistributions of source code must retain the above copyright
14  *    notice, this list of conditions and the following disclaimer.
15  * 2. Redistributions in binary form must reproduce the above copyright
16  *    notice, this list of conditions and the following disclaimer in the
17  *    documentation and/or other materials provided with the distribution.
18  * 3. Neither the name of the University nor the names of its contributors
19  *    may be used to endorse or promote products derived from this software
20  *    without specific prior written permission.
21  *
22  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
23  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
24  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
25  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
26  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
27  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
28  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
29  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
30  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
31  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
32  * SUCH DAMAGE.
33  */
34 #include "sh.h"
35
36 #include <limits.h>
37
38 RCSID("$tcsh: tc.str.c,v 3.30 2009/06/25 21:27:38 christos Exp $")
39
40 #define MALLOC_INCR     128
41 #ifdef WIDE_STRINGS
42 #define MALLOC_SURPLUS  MB_LEN_MAX /* Space for one multibyte character */
43 #else
44 #define MALLOC_SURPLUS  0
45 #endif
46
47 #ifdef WIDE_STRINGS
48 size_t
49 one_mbtowc(wchar_t *pwc, const char *s, size_t n)
50 {
51     int len;
52
53     len = rt_mbtowc(pwc, s, n);
54     if (len == -1) {
55         reset_mbtowc();
56         *pwc = (unsigned char)*s | INVALID_BYTE;
57     }
58     if (len <= 0)
59         len = 1;
60     return len;
61 }
62
63 size_t
64 one_wctomb(char *s, wchar_t wchar)
65 {
66     int len;
67
68     if (wchar & INVALID_BYTE) {
69         s[0] = wchar & 0xFF;
70         len = 1;
71     } else {
72         len = wctomb(s, wchar);
73         if (len == -1)
74             s[0] = wchar;
75         if (len <= 0)
76             len = 1;
77     }
78     return len;
79 }
80
81 int
82 rt_mbtowc(wchar_t *pwc, const char *s, size_t n)
83 {
84     int ret;
85     char back[MB_LEN_MAX];
86
87     ret = mbtowc(pwc, s, n);
88     if (ret > 0 && (wctomb(back, *pwc) != ret || memcmp(s, back, ret) != 0))
89         ret = -1;
90     return ret;
91 }
92 #endif
93
94 #ifdef SHORT_STRINGS
95 Char  **
96 blk2short(char **src)
97 {
98     size_t     n;
99     Char **sdst, **dst;
100
101     /*
102      * Count
103      */
104     for (n = 0; src[n] != NULL; n++)
105         continue;
106     sdst = dst = xmalloc((n + 1) * sizeof(Char *));
107
108     for (; *src != NULL; src++)
109         *dst++ = SAVE(*src);
110     *dst = NULL;
111     return (sdst);
112 }
113
114 char  **
115 short2blk(Char **src)
116 {
117     size_t     n;
118     char **sdst, **dst;
119
120     /*
121      * Count
122      */
123     for (n = 0; src[n] != NULL; n++)
124         continue;
125     sdst = dst = xmalloc((n + 1) * sizeof(char *));
126
127     for (; *src != NULL; src++)
128         *dst++ = strsave(short2str(*src));
129     *dst = NULL;
130     return (sdst);
131 }
132
133 Char   *
134 str2short(const char *src)
135 {
136     static struct Strbuf buf; /* = Strbuf_INIT; */
137
138     if (src == NULL)
139         return (NULL);
140
141     buf.len = 0;
142     while (*src) {
143         Char wc;
144
145         src += one_mbtowc(&wc, src, MB_LEN_MAX);
146         Strbuf_append1(&buf, wc);
147     }
148     Strbuf_terminate(&buf);
149     return buf.s;
150 }
151
152 char   *
153 short2str(const Char *src)
154 {
155     static char *sdst = NULL;
156     static size_t dstsize = 0;
157     char *dst, *edst;
158
159     if (src == NULL)
160         return (NULL);
161
162     if (sdst == NULL) {
163         dstsize = MALLOC_INCR;
164         sdst = xmalloc((dstsize + MALLOC_SURPLUS) * sizeof(char));
165     }
166     dst = sdst;
167     edst = &dst[dstsize];
168     while (*src) {
169         dst += one_wctomb(dst, *src & CHAR);
170         src++;
171         if (dst >= edst) {
172             char *wdst = dst;
173             char *wedst = edst;
174
175             dstsize += MALLOC_INCR;
176             sdst = xrealloc(sdst, (dstsize + MALLOC_SURPLUS) * sizeof(char));
177             edst = &sdst[dstsize];
178             dst = &edst[-MALLOC_INCR];
179             while (wdst > wedst) {
180                 dst++;
181                 wdst--;
182             }
183         }
184     }
185     *dst = 0;
186     return (sdst);
187 }
188
189 #ifndef WIDE_STRINGS
190 Char   *
191 s_strcpy(Char *dst, const Char *src)
192 {
193     Char *sdst;
194
195     sdst = dst;
196     while ((*dst++ = *src++) != '\0')
197         continue;
198     return (sdst);
199 }
200
201 Char   *
202 s_strncpy(Char *dst, const Char *src, size_t n)
203 {
204     Char *sdst;
205
206     if (n == 0)
207         return(dst);
208
209     sdst = dst;
210     do 
211         if ((*dst++ = *src++) == '\0') {
212             while (--n != 0)
213                 *dst++ = '\0';
214             return(sdst);
215         }
216     while (--n != 0);
217     return (sdst);
218 }
219
220 Char   *
221 s_strcat(Char *dst, const Char *src)
222 {
223     Strcpy(Strend(dst), src);
224     return dst;
225 }
226
227 #ifdef NOTUSED
228 Char   *
229 s_strncat(Char *dst, const Char *src, size_t n)
230 {
231     Char *sdst;
232
233     if (n == 0) 
234         return (dst);
235
236     sdst = dst;
237
238     while (*dst)
239         dst++;
240
241     do 
242         if ((*dst++ = *src++) == '\0')
243             return(sdst);
244     while (--n != 0)
245         continue;
246
247     *dst = '\0';
248     return (sdst);
249 }
250
251 #endif
252
253 Char   *
254 s_strchr(const Char *str, int ch)
255 {
256     do
257         if (*str == ch)
258             return ((Char *)(intptr_t)str);
259     while (*str++);
260     return (NULL);
261 }
262
263 Char   *
264 s_strrchr(const Char *str, int ch)
265 {
266     const Char *rstr;
267
268     rstr = NULL;
269     do
270         if (*str == ch)
271             rstr = str;
272     while (*str++);
273     return ((Char *)(intptr_t)rstr);
274 }
275
276 size_t
277 s_strlen(const Char *str)
278 {
279     size_t n;
280
281     for (n = 0; *str++; n++)
282         continue;
283     return (n);
284 }
285
286 int
287 s_strcmp(const Char *str1, const Char *str2)
288 {
289     for (; *str1 && *str1 == *str2; str1++, str2++)
290         continue;
291     /*
292      * The following case analysis is necessary so that characters which look
293      * negative collate low against normal characters but high against the
294      * end-of-string NUL.
295      */
296     if (*str1 == '\0' && *str2 == '\0')
297         return (0);
298     else if (*str1 == '\0')
299         return (-1);
300     else if (*str2 == '\0')
301         return (1);
302     else
303         return (*str1 - *str2);
304 }
305
306 int
307 s_strncmp(const Char *str1, const Char *str2, size_t n)
308 {
309     if (n == 0)
310         return (0);
311     do {
312         if (*str1 != *str2) {
313             /*
314              * The following case analysis is necessary so that characters 
315              * which look negative collate low against normal characters
316              * but high against the end-of-string NUL.
317              */
318             if (*str1 == '\0')
319                 return (-1);
320             else if (*str2 == '\0')
321                 return (1);
322             else
323                 return (*str1 - *str2);
324         }
325         if (*str1 == '\0')
326             return(0);
327         str1++, str2++;
328     } while (--n != 0);
329     return(0);
330 }
331 #endif /* not WIDE_STRINGS */
332
333 int
334 s_strcasecmp(const Char *str1, const Char *str2)
335 {
336 #ifdef WIDE_STRINGS
337     wchar_t l1 = 0, l2 = 0;
338     for (; *str1 && ((*str1 == *str2 && (l1 = l2 = 0) == 0) || 
339         (l1 = towlower(*str1)) == (l2 = towlower(*str2))); str1++, str2++)
340         continue;
341     
342 #else
343     unsigned char c1, c2, l1 = 0, l2 = 0;
344     for (; *str1 && ((*str1 == *str2 && (l1 = l2 = 0) == 0) || 
345         ((c1 = (unsigned char)*str1) == *str1 &&
346          (c2 = (unsigned char)*str2) == *str2 &&
347         (l1 = tolower(c1)) == (l2 = tolower(c2)))); str1++, str2++)
348         continue;
349 #endif
350     /*
351      * The following case analysis is necessary so that characters which look
352      * negative collate low against normal characters but high against the
353      * end-of-string NUL.
354      */
355     if (*str1 == '\0' && *str2 == '\0')
356         return (0);
357     else if (*str1 == '\0')
358         return (-1);
359     else if (*str2 == '\0')
360         return (1);
361     else if (l1 == l2)  /* They are zero when they are equal */
362         return (*str1 - *str2);
363     else
364         return (l1 - l2);
365 }
366
367 Char   *
368 s_strnsave(const Char *s, size_t len)
369 {
370     Char *n;
371
372     n = xmalloc((len + 1) * sizeof (*n));
373     memcpy(n, s, len * sizeof (*n));
374     n[len] = '\0';
375     return n;
376 }
377
378 Char   *
379 s_strsave(const Char *s)
380 {
381     Char   *n;
382     size_t size;
383
384     if (s == NULL)
385         s = STRNULL;
386     size = (Strlen(s) + 1) * sizeof(*n);
387     n = xmalloc(size);
388     memcpy(n, s, size);
389     return (n);
390 }
391
392 Char   *
393 s_strspl(const Char *cp, const Char *dp)
394 {
395     Char *res, *ep;
396     const Char *p, *q;
397
398     if (!cp)
399         cp = STRNULL;
400     if (!dp)
401         dp = STRNULL;
402     for (p = cp; *p++;)
403         continue;
404     for (q = dp; *q++;)
405         continue;
406     res = xmalloc(((p - cp) + (q - dp) - 1) * sizeof(Char));
407     for (ep = res, q = cp; (*ep++ = *q++) != '\0';)
408         continue;
409     for (ep--, q = dp; (*ep++ = *q++) != '\0';)
410         continue;
411     return (res);
412 }
413
414 Char   *
415 s_strend(const Char *cp)
416 {
417     if (!cp)
418         return ((Char *)(intptr_t) cp);
419     while (*cp)
420         cp++;
421     return ((Char *)(intptr_t) cp);
422 }
423
424 Char   *
425 s_strstr(const Char *s, const Char *t)
426 {
427     do {
428         const Char *ss = s;
429         const Char *tt = t;
430
431         do
432             if (*tt == '\0')
433                 return ((Char *)(intptr_t) s);
434         while (*ss++ == *tt++);
435     } while (*s++ != '\0');
436     return (NULL);
437 }
438
439 #else /* !SHORT_STRINGS */
440 char *
441 caching_strip(const char *s)
442 {
443     static char *buf = NULL;
444     static size_t buf_size = 0;
445     size_t size;
446
447     if (s == NULL)
448       return NULL;
449     size = strlen(s) + 1;
450     if (buf_size < size) {
451         buf = xrealloc(buf, size);
452         buf_size = size;
453     }
454     memcpy(buf, s, size);
455     strip(buf);
456     return buf;
457 }
458 #endif
459
460 char   *
461 short2qstr(const Char *src)
462 {
463     static char *sdst = NULL;
464     static size_t dstsize = 0;
465     char *dst, *edst;
466
467     if (src == NULL)
468         return (NULL);
469
470     if (sdst == NULL) {
471         dstsize = MALLOC_INCR;
472         sdst = xmalloc((dstsize + MALLOC_SURPLUS) * sizeof(char));
473     }
474     dst = sdst;
475     edst = &dst[dstsize];
476     while (*src) {
477         if (*src & QUOTE) {
478             *dst++ = '\\';
479             if (dst == edst) {
480                 dstsize += MALLOC_INCR;
481                 sdst = xrealloc(sdst,
482                                 (dstsize + MALLOC_SURPLUS) * sizeof(char));
483                 edst = &sdst[dstsize];
484                 dst = &edst[-MALLOC_INCR];
485             }
486         }
487         dst += one_wctomb(dst, *src & CHAR);
488         src++;
489         if (dst >= edst) {
490             ptrdiff_t i = dst - edst;
491             dstsize += MALLOC_INCR;
492             sdst = xrealloc(sdst, (dstsize + MALLOC_SURPLUS) * sizeof(char));
493             edst = &sdst[dstsize];
494             dst = &edst[-MALLOC_INCR + i];
495         }
496     }
497     *dst = 0;
498     return (sdst);
499 }
500
501 struct blk_buf *
502 bb_alloc()
503 {
504     return xcalloc(1, sizeof(struct blk_buf));
505 }
506
507 static void
508 bb_store(struct blk_buf *bb, Char *str)
509 {
510     if (bb->len == bb->size) { /* Keep space for terminating NULL */
511         if (bb->size == 0)
512             bb->size = 16; /* Arbitrary */
513         else
514             bb->size *= 2;
515         bb->vec = xrealloc(bb->vec, bb->size * sizeof (*bb->vec));
516     }
517     bb->vec[bb->len] = str;
518 }
519
520 void
521 bb_append(struct blk_buf *bb, Char *str)
522 {
523     bb_store(bb, str);
524     bb->len++;
525 }
526
527 void
528 bb_cleanup(void *xbb)
529 {
530     struct blk_buf *bb;
531     size_t i;
532
533     bb = xbb;
534     for (i = 0; i < bb->len; i++)
535         xfree(bb->vec[i]);
536     xfree(bb->vec);
537 }
538
539 void
540 bb_free(void *bb)
541 {
542     bb_cleanup(bb);
543     xfree(bb);
544 }
545
546 Char **
547 bb_finish(struct blk_buf *bb)
548 {
549     bb_store(bb, NULL);
550     return xrealloc(bb->vec, (bb->len + 1) * sizeof (*bb->vec));
551 }
552
553 #define DO_STRBUF(STRBUF, CHAR, STRLEN)                         \
554                                                                 \
555 struct STRBUF *                                                 \
556 STRBUF##_alloc(void)                                            \
557 {                                                               \
558     return xcalloc(1, sizeof(struct STRBUF));                   \
559 }                                                               \
560                                                                 \
561 static void                                                     \
562 STRBUF##_store1(struct STRBUF *buf, CHAR c)                     \
563 {                                                               \
564     if (buf->size == buf->len) {                                \
565         if (buf->size == 0)                                     \
566             buf->size = 64; /* Arbitrary */                     \
567         else                                                    \
568             buf->size *= 2;                                     \
569         buf->s = xrealloc(buf->s, buf->size * sizeof(*buf->s)); \
570     }                                                           \
571     buf->s[buf->len] = c;                                       \
572 }                                                               \
573                                                                 \
574 /* Like strbuf_append1(buf, '\0'), but don't advance len */     \
575 void                                                            \
576 STRBUF##_terminate(struct STRBUF *buf)                          \
577 {                                                               \
578     STRBUF##_store1(buf, '\0');                                 \
579 }                                                               \
580                                                                 \
581 void                                                            \
582 STRBUF##_append1(struct STRBUF *buf, CHAR c)                    \
583 {                                                               \
584     STRBUF##_store1(buf, c);                                    \
585     buf->len++;                                                 \
586 }                                                               \
587                                                                 \
588 void                                                            \
589 STRBUF##_appendn(struct STRBUF *buf, const CHAR *s, size_t len) \
590 {                                                               \
591     if (buf->size < buf->len + len) {                           \
592         if (buf->size == 0)                                     \
593             buf->size = 64; /* Arbitrary */                     \
594         while (buf->size < buf->len + len)                      \
595             buf->size *= 2;                                     \
596         buf->s = xrealloc(buf->s, buf->size * sizeof(*buf->s)); \
597     }                                                           \
598     memcpy(buf->s + buf->len, s, len * sizeof(*buf->s));        \
599     buf->len += len;                                            \
600 }                                                               \
601                                                                 \
602 void                                                            \
603 STRBUF##_append(struct STRBUF *buf, const CHAR *s)              \
604 {                                                               \
605     STRBUF##_appendn(buf, s, STRLEN(s));                        \
606 }                                                               \
607                                                                 \
608 CHAR *                                                          \
609 STRBUF##_finish(struct STRBUF *buf)                             \
610 {                                                               \
611     STRBUF##_append1(buf, 0);                                   \
612     return xrealloc(buf->s, buf->len * sizeof(*buf->s));        \
613 }                                                               \
614                                                                 \
615 void                                                            \
616 STRBUF##_cleanup(void *xbuf)                                    \
617 {                                                               \
618     struct STRBUF *buf;                                         \
619                                                                 \
620     buf = xbuf;                                                 \
621     xfree(buf->s);                                              \
622 }                                                               \
623                                                                 \
624 void                                                            \
625 STRBUF##_free(void *xbuf)                                       \
626 {                                                               \
627     STRBUF##_cleanup(xbuf);                                     \
628     xfree(xbuf);                                                \
629 }                                                               \
630                                                                 \
631 const struct STRBUF STRBUF##_init /* = STRBUF##_INIT; */
632
633 DO_STRBUF(strbuf, char, strlen);
634 DO_STRBUF(Strbuf, Char, Strlen);