Add strcspn() and strndup()
[people/indolent/gpxe.git/.git] / src / core / string.c
1 /*
2  *  Copyright (C) 1991, 1992  Linus Torvalds
3  *  Copyright (C) 2004 Tobias Lorenz
4  *
5  *  string handling functions
6  *  based on linux/lib/string.c
7  *
8  * This program is free software; you can redistribute it and/or modify
9  * it under the terms of the GNU General Public License version 2 as
10  * published by the Free Software Foundation.
11  */
12
13 /*
14  * stupid library routines.. The optimized versions should generally be found
15  * as inline code in <asm-xx/string.h>
16  *
17  * These are buggy as well..
18  *
19  * * Fri Jun 25 1999, Ingo Oeser <ioe@informatik.tu-chemnitz.de>
20  * -  Added strsep() which will replace strtok() soon (because strsep() is
21  *    reentrant and should be faster). Use only strsep() in new code, please.
22  */
23  
24 #include <stdint.h>
25 #include <stdlib.h>
26 #include <string.h>
27 #include <ctype.h>
28
29 /* *** FROM string.c *** */
30
31 #ifndef __HAVE_ARCH_STRNICMP
32 /**
33  * strnicmp - Case insensitive, length-limited string comparison
34  * @s1: One string
35  * @s2: The other string
36  * @len: the maximum number of characters to compare
37  */
38 int strnicmp(const char *s1, const char *s2, size_t len)
39 {
40         /* Yes, Virginia, it had better be unsigned */
41         unsigned char c1, c2;
42
43         c1 = 0; c2 = 0;
44         if (len) {
45                 do {
46                         c1 = *s1; c2 = *s2;
47                         s1++; s2++;
48                         if (!c1)
49                                 break;
50                         if (!c2)
51                                 break;
52                         if (c1 == c2)
53                                 continue;
54                         c1 = tolower(c1);
55                         c2 = tolower(c2);
56                         if (c1 != c2)
57                                 break;
58                 } while (--len);
59         }
60         return (int)c1 - (int)c2;
61 }
62 #endif
63
64 char * ___strtok;
65
66 #ifndef __HAVE_ARCH_STRCPY
67 /**
68  * strcpy - Copy a %NUL terminated string
69  * @dest: Where to copy the string to
70  * @src: Where to copy the string from
71  */
72 char * strcpy(char * dest,const char *src)
73 {
74         char *tmp = dest;
75
76         while ((*dest++ = *src++) != '\0')
77                 /* nothing */;
78         return tmp;
79 }
80 #endif
81
82 #ifndef __HAVE_ARCH_STRNCPY
83 /**
84  * strncpy - Copy a length-limited, %NUL-terminated string
85  * @dest: Where to copy the string to
86  * @src: Where to copy the string from
87  * @count: The maximum number of bytes to copy
88  *
89  * Note that unlike userspace strncpy, this does not %NUL-pad the buffer.
90  * However, the result is not %NUL-terminated if the source exceeds
91  * @count bytes.
92  */
93 char * strncpy(char * dest,const char *src,size_t count)
94 {
95         char *tmp = dest;
96
97         while (count-- && (*dest++ = *src++) != '\0')
98                 /* nothing */;
99
100         return tmp;
101 }
102 #endif
103
104 #ifndef __HAVE_ARCH_STRCAT
105 /**
106  * strcat - Append one %NUL-terminated string to another
107  * @dest: The string to be appended to
108  * @src: The string to append to it
109  */
110 char * strcat(char * dest, const char * src)
111 {
112         char *tmp = dest;
113
114         while (*dest)
115                 dest++;
116         while ((*dest++ = *src++) != '\0')
117                 ;
118
119         return tmp;
120 }
121 #endif
122
123 #ifndef __HAVE_ARCH_STRNCAT
124 /**
125  * strncat - Append a length-limited, %NUL-terminated string to another
126  * @dest: The string to be appended to
127  * @src: The string to append to it
128  * @count: The maximum numbers of bytes to copy
129  *
130  * Note that in contrast to strncpy, strncat ensures the result is
131  * terminated.
132  */
133 char * strncat(char *dest, const char *src, size_t count)
134 {
135         char *tmp = dest;
136
137         if (count) {
138                 while (*dest)
139                         dest++;
140                 while ((*dest++ = *src++)) {
141                         if (--count == 0) {
142                                 *dest = '\0';
143                                 break;
144                         }
145                 }
146         }
147
148         return tmp;
149 }
150 #endif
151
152 #ifndef __HAVE_ARCH_STRCMP
153 /**
154  * strcmp - Compare two strings
155  * @cs: One string
156  * @ct: Another string
157  */
158 int strcmp(const char * cs,const char * ct)
159 {
160         register signed char __res;
161
162         while (1) {
163                 if ((__res = *cs - *ct++) != 0 || !*cs++)
164                         break;
165         }
166
167         return __res;
168 }
169 #endif
170
171 #ifndef __HAVE_ARCH_STRNCMP
172 /**
173  * strncmp - Compare two length-limited strings
174  * @cs: One string
175  * @ct: Another string
176  * @count: The maximum number of bytes to compare
177  */
178 int strncmp(const char * cs,const char * ct,size_t count)
179 {
180         register signed char __res = 0;
181
182         while (count) {
183                 if ((__res = *cs - *ct++) != 0 || !*cs++)
184                         break;
185                 count--;
186         }
187
188         return __res;
189 }
190 #endif
191
192 #ifndef __HAVE_ARCH_STRCHR
193 /**
194  * strchr - Find the first occurrence of a character in a string
195  * @s: The string to be searched
196  * @c: The character to search for
197  */
198 char * strchr(const char * s, int c)
199 {
200         for(; *s != (char) c; ++s)
201                 if (*s == '\0')
202                         return NULL;
203         return (char *) s;
204 }
205 #endif
206
207 #ifndef __HAVE_ARCH_STRRCHR
208 /**
209  * strrchr - Find the last occurrence of a character in a string
210  * @s: The string to be searched
211  * @c: The character to search for
212  */
213 char * strrchr(const char * s, int c)
214 {
215        const char *p = s + strlen(s);
216        do {
217            if (*p == (char)c)
218                return (char *)p;
219        } while (--p >= s);
220        return NULL;
221 }
222 #endif
223
224 #ifndef __HAVE_ARCH_STRLEN
225 /**
226  * strlen - Find the length of a string
227  * @s: The string to be sized
228  */
229 size_t strlen(const char * s)
230 {
231         const char *sc;
232
233         for (sc = s; *sc != '\0'; ++sc)
234                 /* nothing */;
235         return sc - s;
236 }
237 #endif
238
239 #ifndef __HAVE_ARCH_STRNLEN
240 /**
241  * strnlen - Find the length of a length-limited string
242  * @s: The string to be sized
243  * @count: The maximum number of bytes to search
244  */
245 size_t strnlen(const char * s, size_t count)
246 {
247         const char *sc;
248
249         for (sc = s; count-- && *sc != '\0'; ++sc)
250                 /* nothing */;
251         return sc - s;
252 }
253 #endif
254
255 #ifndef __HAVE_ARCH_STRSPN
256 /**
257  * strspn - Calculate the length of the initial substring of @s which only
258  *      contain letters in @accept
259  * @s: The string to be searched
260  * @accept: The string to search for
261  */
262 size_t strspn(const char *s, const char *accept)
263 {
264         const char *p;
265         const char *a;
266         size_t count = 0;
267
268         for (p = s; *p != '\0'; ++p) {
269                 for (a = accept; *a != '\0'; ++a) {
270                         if (*p == *a)
271                                 break;
272                 }
273                 if (*a == '\0')
274                         return count;
275                 ++count;
276         }
277
278         return count;
279 }
280 #endif
281
282 #ifndef __HAVE_ARCH_STRCSPN
283 /**
284  * strcspn - Calculate the length of the initial substring of @s which only
285  *      contain letters not in @reject
286  * @s: The string to be searched
287  * @accept: The string to search for
288  */
289 size_t strcspn(const char *s, const char *reject)
290 {
291         const char *p;
292         const char *r;
293         size_t count = 0;
294
295         for (p = s; *p != '\0'; ++p) {
296                 for (r = reject; *r != '\0'; ++r) {
297                         if (*p == *r)
298                                 return count;
299                 }
300                 ++count;
301         }
302
303         return count;
304 }
305 #endif
306
307 #ifndef __HAVE_ARCH_STRPBRK
308 /**
309  * strpbrk - Find the first occurrence of a set of characters
310  * @cs: The string to be searched
311  * @ct: The characters to search for
312  */
313 char * strpbrk(const char * cs,const char * ct)
314 {
315         const char *sc1,*sc2;
316
317         for( sc1 = cs; *sc1 != '\0'; ++sc1) {
318                 for( sc2 = ct; *sc2 != '\0'; ++sc2) {
319                         if (*sc1 == *sc2)
320                                 return (char *) sc1;
321                 }
322         }
323         return NULL;
324 }
325 #endif
326
327 #ifndef __HAVE_ARCH_STRTOK
328 /**
329  * strtok - Split a string into tokens
330  * @s: The string to be searched
331  * @ct: The characters to search for
332  *
333  * WARNING: strtok is deprecated, use strsep instead.
334  */
335 char * strtok(char * s,const char * ct)
336 {
337         char *sbegin, *send;
338
339         sbegin  = s ? s : ___strtok;
340         if (!sbegin) {
341                 return NULL;
342         }
343         sbegin += strspn(sbegin,ct);
344         if (*sbegin == '\0') {
345                 ___strtok = NULL;
346                 return( NULL );
347         }
348         send = strpbrk( sbegin, ct);
349         if (send && *send != '\0')
350                 *send++ = '\0';
351         ___strtok = send;
352         return (sbegin);
353 }
354 #endif
355
356 #ifndef __HAVE_ARCH_STRSEP
357 /**
358  * strsep - Split a string into tokens
359  * @s: The string to be searched
360  * @ct: The characters to search for
361  *
362  * strsep() updates @s to point after the token, ready for the next call.
363  *
364  * It returns empty tokens, too, behaving exactly like the libc function
365  * of that name. In fact, it was stolen from glibc2 and de-fancy-fied.
366  * Same semantics, slimmer shape. ;)
367  */
368 char * strsep(char **s, const char *ct)
369 {
370         char *sbegin = *s, *end;
371
372         if (sbegin == NULL)
373                 return NULL;
374
375         end = strpbrk(sbegin, ct);
376         if (end)
377                 *end++ = '\0';
378         *s = end;
379
380         return sbegin;
381 }
382 #endif
383
384 #ifndef __HAVE_ARCH_MEMSET
385 /**
386  * memset - Fill a region of memory with the given value
387  * @s: Pointer to the start of the area.
388  * @c: The byte to fill the area with
389  * @count: The size of the area.
390  *
391  * Do not use memset() to access IO space, use memset_io() instead.
392  */
393 void * memset(void * s,int c,size_t count)
394 {
395         char *xs = (char *) s;
396
397         while (count--)
398                 *xs++ = c;
399
400         return s;
401 }
402 #endif
403
404 #ifndef __HAVE_ARCH_BCOPY
405 /**
406  * bcopy - Copy one area of memory to another
407  * @src: Where to copy from
408  * @dest: Where to copy to
409  * @count: The size of the area.
410  *
411  * Note that this is the same as memcpy(), with the arguments reversed.
412  * memcpy() is the standard, bcopy() is a legacy BSD function.
413  *
414  * You should not use this function to access IO space, use memcpy_toio()
415  * or memcpy_fromio() instead.
416  */
417 char * bcopy(const char * src, char * dest, int count)
418 {
419         char *tmp = dest;
420
421         while (count--)
422                 *tmp++ = *src++;
423
424         return dest;
425 }
426 #endif
427
428 #ifndef __HAVE_ARCH_MEMCPY
429 /**
430  * memcpy - Copy one area of memory to another
431  * @dest: Where to copy to
432  * @src: Where to copy from
433  * @count: The size of the area.
434  *
435  * You should not use this function to access IO space, use memcpy_toio()
436  * or memcpy_fromio() instead.
437  */
438 void * memcpy(void * dest,const void *src,size_t count)
439 {
440         char *tmp = (char *) dest, *s = (char *) src;
441
442         while (count--)
443                 *tmp++ = *s++;
444
445         return dest;
446 }
447 #endif
448
449 #ifndef __HAVE_ARCH_MEMMOVE
450 /**
451  * memmove - Copy one area of memory to another
452  * @dest: Where to copy to
453  * @src: Where to copy from
454  * @count: The size of the area.
455  *
456  * Unlike memcpy(), memmove() copes with overlapping areas.
457  */
458 void * memmove(void * dest,const void *src,size_t count)
459 {
460         char *tmp, *s;
461
462         if (dest <= src) {
463                 tmp = (char *) dest;
464                 s = (char *) src;
465                 while (count--)
466                         *tmp++ = *s++;
467                 }
468         else {
469                 tmp = (char *) dest + count;
470                 s = (char *) src + count;
471                 while (count--)
472                         *--tmp = *--s;
473                 }
474
475         return dest;
476 }
477 #endif
478
479 #ifndef __HAVE_ARCH_MEMCMP
480 /**
481  * memcmp - Compare two areas of memory
482  * @cs: One area of memory
483  * @ct: Another area of memory
484  * @count: The size of the area.
485  */
486 int memcmp(const void * cs,const void * ct,size_t count)
487 {
488         const unsigned char *su1, *su2;
489         int res = 0;
490
491         for( su1 = cs, su2 = ct; 0 < count; ++su1, ++su2, count--)
492                 if ((res = *su1 - *su2) != 0)
493                         break;
494         return res;
495 }
496 #endif
497
498 #ifndef __HAVE_ARCH_MEMSCAN
499 /**
500  * memscan - Find a character in an area of memory.
501  * @addr: The memory area
502  * @c: The byte to search for
503  * @size: The size of the area.
504  *
505  * returns the address of the first occurrence of @c, or 1 byte past
506  * the area if @c is not found
507  */
508 void * memscan(void * addr, int c, size_t size)
509 {
510         unsigned char * p = (unsigned char *) addr;
511
512         while (size) {
513                 if (*p == c)
514                         return (void *) p;
515                 p++;
516                 size--;
517         }
518         return (void *) p;
519 }
520 #endif
521
522 #ifndef __HAVE_ARCH_STRSTR
523 /**
524  * strstr - Find the first substring in a %NUL terminated string
525  * @s1: The string to be searched
526  * @s2: The string to search for
527  */
528 char * strstr(const char * s1,const char * s2)
529 {
530         int l1, l2;
531
532         l2 = strlen(s2);
533         if (!l2)
534                 return (char *) s1;
535         l1 = strlen(s1);
536         while (l1 >= l2) {
537                 l1--;
538                 if (!memcmp(s1,s2,l2))
539                         return (char *) s1;
540                 s1++;
541         }
542         return NULL;
543 }
544 #endif
545
546 #ifndef __HAVE_ARCH_MEMCHR
547 /**
548  * memchr - Find a character in an area of memory.
549  * @s: The memory area
550  * @c: The byte to search for
551  * @n: The size of the area.
552  *
553  * returns the address of the first occurrence of @c, or %NULL
554  * if @c is not found
555  */
556 void * memchr(const void *s, int c, size_t n)
557 {
558         const unsigned char *p = s;
559         while (n-- != 0) {
560                 if ((unsigned char)c == *p++) {
561                         return (void *)(p-1);
562                 }
563         }
564         return NULL;
565 }
566
567 #endif
568
569 char * strndup(const char *s, size_t n)
570 {
571         size_t len = strlen(s);
572         char *new;
573
574         if (len>n)
575                 len = n;
576         new = malloc(len+1);
577         if (new) {
578                 new[len] = '\0';
579                 memcpy(new,s,len);
580         }
581         return new;
582 }
583
584 char * strdup(const char *s) {
585         return strndup(s, ~((size_t)0));
586 }