(change_name): Check that the NAME is not too long.
[gnupg.git] / jnlib / stringhelp.c
1 /* stringhelp.c -  standard string helper functions
2  * Copyright (C) 1998, 1999, 2000, 2001, 2003 Free Software Foundation, Inc.
3  *
4  * This file is part of GnuPG.
5  *
6  * GnuPG is free software; you can redistribute it and/or modify
7  * it under the terms of the GNU General Public License as published by
8  * the Free Software Foundation; either version 2 of the License, or
9  * (at your option) any later version.
10  *
11  * GnuPG is distributed in the hope that it will be useful,
12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14  * GNU General Public License for more details.
15  *
16  * You should have received a copy of the GNU General Public License
17  * along with this program; if not, write to the Free Software
18  * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA
19  */
20
21 #include <config.h>
22 #include <stdlib.h>
23 #include <string.h>
24 #include <stdarg.h>
25 #include <ctype.h>
26
27 #include "libjnlib-config.h"
28 #include "utf8conv.h"
29 #include "stringhelp.h"
30
31
32 /****************
33  * look for the substring SUB in buffer and return a pointer to that
34  * substring in BUF or NULL if not found.
35  * Comparison is case-insensitive.
36  */
37 const char *
38 memistr( const char *buf, size_t buflen, const char *sub )
39 {
40     const byte *t, *s ;
41     size_t n;
42
43     for( t=buf, n=buflen, s=sub ; n ; t++, n-- )
44         if( toupper(*t) == toupper(*s) ) {
45             for( buf=t++, buflen = n--, s++;
46                  n && toupper(*t) == toupper(*s); t++, s++, n-- )
47                 ;
48             if( !*s )
49                 return buf;
50             t = buf; n = buflen; s = sub ;
51         }
52
53     return NULL ;
54 }
55
56 const char *
57 ascii_memistr( const char *buf, size_t buflen, const char *sub )
58 {
59     const byte *t, *s ;
60     size_t n;
61
62     for( t=buf, n=buflen, s=sub ; n ; t++, n-- )
63         if( ascii_toupper(*t) == ascii_toupper(*s) ) {
64             for( buf=t++, buflen = n--, s++;
65                  n && ascii_toupper(*t) == ascii_toupper(*s); t++, s++, n-- )
66                 ;
67             if( !*s )
68                 return buf;
69             t = buf; n = buflen; s = sub ;
70         }
71
72     return NULL ;
73 }
74
75 /****************
76  * Wie strncpy(), aber es werden maximal n-1 zeichen kopiert und ein
77  * '\0' angehängt. Ist n = 0, so geschieht nichts, ist Destination
78  * gleich NULL, so wird via jnlib_xmalloc Speicher besorgt, ist dann nicht
79  * genügend Speicher vorhanden, so bricht die funktion ab.
80  */
81 char *
82 mem2str( char *dest , const void *src , size_t n )
83 {
84     char *d;
85     const char *s;
86
87     if( n ) {
88         if( !dest )
89             dest = jnlib_xmalloc( n ) ;
90         d = dest;
91         s = src ;
92         for(n--; n && *s; n-- )
93             *d++ = *s++;
94         *d = '\0' ;
95     }
96
97     return dest ;
98 }
99
100
101 /****************
102  * remove leading and trailing white spaces
103  */
104 char *
105 trim_spaces( char *str )
106 {
107     char *string, *p, *mark;
108
109     string = str;
110     /* find first non space character */
111     for( p=string; *p && isspace( *(byte*)p ) ; p++ )
112         ;
113     /* move characters */
114     for( (mark = NULL); (*string = *p); string++, p++ )
115         if( isspace( *(byte*)p ) ) {
116             if( !mark )
117                 mark = string ;
118         }
119         else
120             mark = NULL ;
121     if( mark )
122         *mark = '\0' ;  /* remove trailing spaces */
123
124     return str ;
125 }
126
127 /****************
128  * remove trailing white spaces
129  */
130 char *
131 trim_trailing_spaces( char *string )
132 {
133     char *p, *mark;
134
135     for( mark = NULL, p = string; *p; p++ ) {
136         if( isspace( *(byte*)p ) ) {
137             if( !mark )
138                 mark = p;
139         }
140         else
141             mark = NULL;
142     }
143     if( mark )
144         *mark = '\0' ;
145
146     return string ;
147 }
148
149
150 unsigned
151 trim_trailing_chars( byte *line, unsigned len, const char *trimchars )
152 {
153     byte *p, *mark;
154     unsigned n;
155
156     for(mark=NULL, p=line, n=0; n < len; n++, p++ ) {
157         if( strchr(trimchars, *p ) ) {
158             if( !mark )
159                 mark = p;
160         }
161         else
162             mark = NULL;
163     }
164
165     if( mark ) {
166         *mark = 0;
167         return mark - line;
168     }
169     return len;
170 }
171
172 /****************
173  * remove trailing white spaces and return the length of the buffer
174  */
175 unsigned
176 trim_trailing_ws( byte *line, unsigned len )
177 {
178     return trim_trailing_chars( line, len, " \t\r\n" );
179 }
180
181 size_t
182 length_sans_trailing_chars (const unsigned char *line, size_t len,
183                             const char *trimchars )
184 {
185   const unsigned char *p, *mark;
186   size_t n;
187   
188   for( mark=NULL, p=line, n=0; n < len; n++, p++ )
189     {
190       if (strchr (trimchars, *p ))
191         {
192           if( !mark )
193             mark = p;
194         }
195       else
196         mark = NULL;
197     }
198   
199   if (mark) 
200     return mark - line;
201   return len;
202 }
203
204 /****************
205  * remove trailing white spaces and return the length of the buffer
206  */
207 size_t
208 length_sans_trailing_ws (const unsigned char *line, size_t len)
209 {
210   return length_sans_trailing_chars (line, len, " \t\r\n");
211 }
212
213
214
215 /***************
216  * Extract from a given path the filename component.
217  *
218  */
219 char *
220 make_basename(const char *filepath)
221 {
222     char *p;
223
224     if ( !(p=strrchr(filepath, '/')) )
225       #ifdef HAVE_DRIVE_LETTERS
226         if ( !(p=strrchr(filepath, '\\')) )
227             if ( !(p=strrchr(filepath, ':')) )
228       #endif
229               {
230                 return jnlib_xstrdup(filepath);
231               }
232
233     return jnlib_xstrdup(p+1);
234 }
235
236
237
238 /***************
239  * Extract from a given filename the path prepended to it.
240  * If their isn't a path prepended to the filename, a dot
241  * is returned ('.').
242  *
243  */
244 char *
245 make_dirname(const char *filepath)
246 {
247     char *dirname;
248     int  dirname_length;
249     char *p;
250
251     if ( !(p=strrchr(filepath, '/')) )
252       #ifdef HAVE_DRIVE_LETTERS
253         if ( !(p=strrchr(filepath, '\\')) )
254             if ( !(p=strrchr(filepath, ':')) )
255       #endif
256               {
257                 return jnlib_xstrdup(".");
258               }
259
260     dirname_length = p-filepath;
261     dirname = jnlib_xmalloc(dirname_length+1);
262     strncpy(dirname, filepath, dirname_length);
263     dirname[dirname_length] = 0;
264
265     return dirname;
266 }
267
268
269
270 /****************
271  * Construct a filename from the NULL terminated list of parts.
272  * Tilde expansion is done here.
273  */
274 char *
275 make_filename( const char *first_part, ... )
276 {
277     va_list arg_ptr ;
278     size_t n;
279     const char *s;
280     char *name, *home, *p;
281
282     va_start( arg_ptr, first_part ) ;
283     n = strlen(first_part)+1;
284     while( (s=va_arg(arg_ptr, const char *)) )
285         n += strlen(s) + 1;
286     va_end(arg_ptr);
287
288     home = NULL;
289     if( *first_part == '~' && first_part[1] == '/'
290                            && (home = getenv("HOME")) && *home )
291         n += strlen(home);
292
293     name = jnlib_xmalloc(n);
294     p = home ? stpcpy(stpcpy(name,home), first_part+1)
295              : stpcpy(name, first_part);
296     va_start( arg_ptr, first_part ) ;
297     while( (s=va_arg(arg_ptr, const char *)) )
298         p = stpcpy(stpcpy(p,"/"), s);
299     va_end(arg_ptr);
300
301     return name;
302 }
303
304
305 int
306 compare_filenames( const char *a, const char *b )
307 {
308     /* ? check whether this is an absolute filename and
309      * resolve symlinks?
310      */
311 #ifdef HAVE_DRIVE_LETTERS
312     return stricmp(a,b);
313 #else
314     return strcmp(a,b);
315 #endif
316 }
317
318 /* Print a BUFFER to stream FP while replacing all control characters
319    and the character DELIM with standard C escape sequences.  Returns
320    the number of characters printed. */
321 size_t 
322 print_sanitized_buffer (FILE *fp, const void *buffer, size_t length,
323                         int delim)
324 {
325   const unsigned char *p = buffer;
326   size_t count = 0;
327
328   for (; length; length--, p++, count++)
329     {
330       if (*p < 0x20 || *p == 0x7f || *p == delim)
331         {
332           putc ('\\', fp);
333           count++;
334           if (*p == '\n')
335             putc ('n', fp);
336           else if (*p == '\r')
337             putc ('r', fp);
338           else if (*p == '\f')
339             putc ('f', fp);
340           else if (*p == '\v')
341             putc ('v', fp);
342           else if (*p == '\b')
343             putc ('b', fp);
344           else if (!*p)
345             putc('0', fp);
346           else
347             {
348               fprintf (fp, "x%02x", *p);
349               count += 2;
350             }
351         }
352       else
353         putc (*p, fp);
354     }
355
356   return count;
357 }
358
359 size_t 
360 print_sanitized_utf8_buffer (FILE *fp, const void *buffer,
361                              size_t length, int delim)
362 {
363   const char *p = buffer;
364   size_t i;
365
366   /* We can handle plain ascii simpler, so check for it first. */
367   for (i=0; i < length; i++ ) 
368     {
369       if ( (p[i] & 0x80) )
370         break;
371     }
372   if (i < length)
373     {
374         char *buf = utf8_to_native (p, length, delim);
375         /*(utf8 conversion already does the control character quoting)*/
376         i = strlen (buf);
377         fputs (buf, fp);
378         jnlib_free (buf);
379         return i;
380     }
381   else
382     return print_sanitized_buffer (fp, p, length, delim);
383 }
384
385
386 size_t 
387 print_sanitized_string (FILE *fp, const char *string, int delim)
388 {
389   return string? print_sanitized_buffer (fp, string, strlen (string), delim):0;
390 }
391
392 size_t 
393 print_sanitized_utf8_string (FILE *fp, const char *string, int delim)
394 {
395   /* FIXME: convert to local characterset */
396   return print_sanitized_string (fp, string, delim);
397 }
398
399 /* Create a string from the buffer P of length N which is suitable for
400    printing.  Caller must release the created string using xfree. */
401 char *
402 sanitize_buffer (const unsigned char *p, size_t n, int delim)
403 {
404   size_t save_n, buflen;
405   const byte *save_p;
406   char *buffer, *d;
407
408   /* first count length */
409   for (save_n = n, save_p = p, buflen=1 ; n; n--, p++ ) 
410     {
411       if ( *p < 0x20 || *p == 0x7f || *p == delim  || (delim && *p=='\\'))
412         {
413           if ( *p=='\n' || *p=='\r' || *p=='\f'
414                || *p=='\v' || *p=='\b' || !*p )
415             buflen += 2;
416           else
417             buflen += 4;
418         }
419       else
420         buflen++;
421     }
422   p = save_p;
423   n = save_n;
424   /* and now make the string */
425   d = buffer = jnlib_xmalloc( buflen );
426   for ( ; n; n--, p++ )
427     {
428       if (*p < 0x20 || *p == 0x7f || *p == delim || (delim && *p=='\\')) {
429         *d++ = '\\';
430         if( *p == '\n' )
431           *d++ = 'n';
432         else if( *p == '\r' )
433           *d++ = 'r';
434         else if( *p == '\f' )
435           *d++ = 'f';
436         else if( *p == '\v' )
437           *d++ = 'v';
438         else if( *p == '\b' )
439           *d++ = 'b';
440         else if( !*p )
441           *d++ = '0';
442         else {
443           sprintf(d, "x%02x", *p );
444           d += 2;
445         }
446       }
447       else
448         *d++ = *p;
449     }
450   *d = 0;
451   return buffer;
452 }
453
454 /****************************************************
455  ******** locale insensitive ctype functions ********
456  ****************************************************/
457 /* FIXME: replace them by a table lookup and macros */
458 int
459 ascii_isupper (int c)
460 {
461     return c >= 'A' && c <= 'Z';
462 }
463
464 int
465 ascii_islower (int c)
466 {
467     return c >= 'a' && c <= 'z';
468 }
469
470 int 
471 ascii_toupper (int c)
472 {
473     if (c >= 'a' && c <= 'z')
474         c &= ~0x20;
475     return c;
476 }
477
478 int 
479 ascii_tolower (int c)
480 {
481     if (c >= 'A' && c <= 'Z')
482         c |= 0x20;
483     return c;
484 }
485
486
487 int
488 ascii_strcasecmp( const char *a, const char *b )
489 {
490     if (a == b)
491         return 0;
492
493     for (; *a && *b; a++, b++) {
494         if (*a != *b && ascii_toupper(*a) != ascii_toupper(*b))
495             break;
496     }
497     return *a == *b? 0 : (ascii_toupper (*a) - ascii_toupper (*b));
498 }
499
500 int 
501 ascii_strncasecmp (const char *a, const char *b, size_t n)
502 {
503   const unsigned char *p1 = (const unsigned char *)a;
504   const unsigned char *p2 = (const unsigned char *)b;
505   unsigned char c1, c2;
506
507   if (p1 == p2 || !n )
508     return 0;
509
510   do
511     {
512       c1 = ascii_tolower (*p1);
513       c2 = ascii_tolower (*p2);
514
515       if ( !--n || c1 == '\0')
516         break;
517
518       ++p1;
519       ++p2;
520     }
521   while (c1 == c2);
522   
523   return c1 - c2;
524 }
525
526
527 int
528 ascii_memcasecmp( const char *a, const char *b, size_t n )
529 {
530     if (a == b)
531         return 0;
532     for ( ; n; n--, a++, b++ ) {
533         if( *a != *b  && ascii_toupper (*a) != ascii_toupper (*b) )
534             return *a == *b? 0 : (ascii_toupper (*a) - ascii_toupper (*b));
535     }
536     return 0;
537 }
538
539 int
540 ascii_strcmp( const char *a, const char *b )
541 {
542     if (a == b)
543         return 0;
544
545     for (; *a && *b; a++, b++) {
546         if (*a != *b )
547             break;
548     }
549     return *a == *b? 0 : (*(signed char *)a - *(signed char *)b);
550 }
551
552
553 void *
554 ascii_memcasemem (const void *haystack, size_t nhaystack,
555                   const void *needle, size_t nneedle)
556 {
557
558   if (!nneedle)
559     return (void*)haystack; /* finding an empty needle is really easy */
560   if (nneedle <= nhaystack)
561     {
562       const unsigned char *a = haystack;
563       const unsigned char *b = a + nhaystack - nneedle;
564       
565       for (; a <= b; a++)
566         {
567           if ( !ascii_memcasecmp (a, needle, nneedle) )
568             return (void *)a;
569         }
570     }
571   return NULL;
572 }
573
574 /*********************************************
575  ********** missing string functions *********
576  *********************************************/
577
578 #ifndef HAVE_STPCPY
579 char *
580 stpcpy(char *a,const char *b)
581 {
582     while( *b )
583         *a++ = *b++;
584     *a = 0;
585
586     return (char*)a;
587 }
588 #endif
589
590 #ifndef HAVE_STRLWR
591 char *
592 strlwr(char *s)
593 {
594     char *p;
595     for(p=s; *p; p++ )
596         *p = tolower(*p);
597     return s;
598 }
599 #endif
600
601
602 #ifndef HAVE_STRCASECMP
603 int
604 strcasecmp( const char *a, const char *b )
605 {
606     for( ; *a && *b; a++, b++ ) {
607         if( *a != *b && toupper(*a) != toupper(*b) )
608             break;
609     }
610     return *(const byte*)a - *(const byte*)b;
611 }
612 #endif
613
614
615 /****************
616  * mingw32/cpd has a memicmp()
617  */
618 #ifndef HAVE_MEMICMP
619 int
620 memicmp( const char *a, const char *b, size_t n )
621 {
622     for( ; n; n--, a++, b++ )
623         if( *a != *b  && toupper(*(const byte*)a) != toupper(*(const byte*)b) )
624             return *(const byte *)a - *(const byte*)b;
625     return 0;
626 }
627 #endif