Usermode/Libraries/libc.so_src/string.c

   1 /*
   2  * AcessOS Basic C Library
   3  * string.c
   4  */
   5 #include <acess/sys.h>
   6 #include <stdlib.h>
   7 #include <stdio.h>
   8 #include "lib.h"
   9
  10 /**
  11  * \fn EXPORT int strcmp(const char *s1, const char *s2)
  12  * \brief Compare two strings
  13  */
  14 EXPORT int strcmp(const char *s1, const char *s2)
  15 {
  16         while(*s1 == *s2 && *s1 != '\0' && *s2 != '\0') {
  17                 s1++; s2++;
  18         }
  19         return (int)*s1 - (int)*s2;
  20 }
  21
  22 /**
  23  * \fn EXPORT int strncmp(const char *s1, const char *s2)
  24  * \brief Compare two strings
  25  */
  26 EXPORT int strncmp(const char *s1, const char *s2, size_t n)
  27 {
  28         if( n == 0 )    return 0;
  29         while(n -- && *s1 == *s2 && *s1 != '\0' && *s2 != '\0') {
  30                 s1++; s2++;
  31         }
  32         return (int)*s1 - (int)*s2;
  33 }
  34
  35 /**
  36  * \fn EXPORT char *strcpy(char *dst, const char *src)
  37  * \brief Copy a string to another
  38  */
  39 EXPORT char *strcpy(char *dst, const char *src)
  40 {
  41         char *_dst = dst;
  42         while(*src) {
  43                 *dst = *src;
  44                 src++; dst++;
  45         }
  46         *dst = '\0';
  47         return _dst;
  48 }
  49
  50 /**
  51  * \fn EXPORT char *strncpy(char *dst, const char *src)
  52  * \brief Copy at most \a num characters from \a src to \a dst
  53  * \return \a dst
  54  */
  55 EXPORT char *strncpy(char *dst, const char *src, size_t num)
  56 {
  57         char *to = dst;
  58         while(*src && num--)    *to++ = *src++;
  59         *to = '\0';
  60         return dst;
  61 }
  62
  63 /**
  64  * \fn EXPORT char *strcat(char *dst, const char *src)
  65  * \brief Append a string onto another
  66  */
  67 EXPORT char *strcat(char *dst, const char *src)
  68 {
  69         char    *to = dst;
  70         // Find the end
  71         while(*to)      to++;
  72         // Copy
  73         while(*src)     *to++ = *src++;
  74         // End string
  75         *to = '\0';
  76         return dst;
  77 }
  78
  79 /**
  80  * \fn EXPORT int strlen(const char *str)
  81  * \brief Get the length of a string
  82  */
  83 EXPORT int strlen(const char *str)
  84 {
  85         int retval;
  86         for(retval = 0; *str != '\0'; str++)
  87                 retval++;
  88         return retval;
  89 }
  90
  91 /**
  92  * \fn EXPORT char *strdup(const char *str)
  93  * \brief Duplicate a string using heap memory
  94  * \note Defined in POSIX Spec, not C spec
  95  */
  96 EXPORT char *strdup(const char *str)
  97 {
  98         size_t  len = strlen(str);
  99         char    *ret = malloc(len+1);
 100         if(ret == NULL) return NULL;
 101         strcpy(ret, str);
 102         return ret;
 103 }
 104
 105 /**
 106  * \fn EXPORT char *strndup(const char *str, size_t maxlen)
 107  * \brief Duplicate a string into the heap with a maximum length
 108  * \param str   Input string buffer
 109  * \param maxlen        Maximum valid size of the \a str buffer
 110  * \return Heap string with the same value of \a str
 111  */
 112 EXPORT char *strndup(const char *str, size_t maxlen)
 113 {
 114         size_t  len;
 115         char    *ret;
 116         for( len = 0; len < maxlen && str[len]; len ++) ;
 117         ret = malloc( len + 1);
 118         memcpy( ret, str, len );
 119         ret[len] = '\0';
 120         return ret;
 121 }
 122
 123 /**
 124  * \fn EXPORT char *strchr(char *str, int character)
 125  * \brief Locate a character in a string
 126  */
 127 EXPORT char *strchr(const char *str, int character)
 128 {
 129         for(;*str;str++)
 130         {
 131                 if(*str == character)
 132                         return (char*)str;
 133         }
 134         return NULL;
 135 }
 136
 137 /**
 138  * \fn EXPORT char *strrchr(char *str, int character)
 139  * \brief Locate the last occurance of a character in a string
 140  */
 141 EXPORT char *strrchr(const char *str, int character)
 142 {
 143          int    i;
 144         i = strlen(str)-1;
 145         while(i--)
 146         {
 147                 if(str[i] == character)
 148                         return (void*)&str[i];
 149         }
 150         return NULL;
 151 }
 152
 153 /**
 154  * \fn EXPORT char *strstr(char *str1, const char *str2)
 155  * \brief Search a \a str1 for the first occurance of \a str2
 156  */
 157 EXPORT char *strstr(char *str1, const char *str2)
 158 {
 159         const char      *test = str2;
 160
 161         for(;*str1;str1++)
 162         {
 163                 if(*test == '\0')       return str1;
 164                 if(*str1 == *test)      test++;
 165                 else    test = str2;
 166         }
 167         return NULL;
 168 }
 169
 170 // --- Memory ---
 171 /**
 172  * \fn EXPORT void *memset(void *dest, int val, size_t num)
 173  * \brief Clear memory with the specified value
 174  */
 175 EXPORT void *memset(void *dest, int val, size_t num)
 176 {
 177         unsigned char *p = dest;
 178         while(num--)    *p++ = val;
 179         return dest;
 180 }
 181
 182 /**
 183  * \fn EXPORT void *memcpy(void *dest, const void *src, size_t count)
 184  * \brief Copy one memory area to another
 185  */
 186 EXPORT void *memcpy(void *dest, const void *src, size_t count)
 187 {
 188     char *sp = (char *)src;
 189     char *dp = (char *)dest;
 190     for(;count--;) *dp++ = *sp++;
 191     return dest;
 192 }
 193
 194 /**
 195  * \fn EXPORT void *memmove(void *dest, const void *src, size_t count)
 196  * \brief Copy data in memory, avoiding overlap problems
 197  */
 198 EXPORT void *memmove(void *dest, const void *src, size_t count)
 199 {
 200     char *sp = (char *)src;
 201     char *dp = (char *)dest;
 202         // Check if corruption will happen
 203         if( (intptr_t)dest > (intptr_t)src && (intptr_t)dest < (intptr_t)src+count )
 204                 for(;count--;) dp[count] = sp[count];
 205         else
 206         for(;count--;) *dp++ = *sp++;
 207     return dest;
 208 }
 209
 210 /**
 211  * \fn EXPORT int memcmp(const void *mem1, const void *mem2, size_t count)
 212  * \brief Compare two regions of memory
 213  * \param mem1  Region 1
 214  * \param mem2  Region 2
 215  * \param count Number of bytes to check
 216  */
 217 EXPORT int memcmp(const void *mem1, const void *mem2, size_t count)
 218 {
 219         while(count--)
 220         {
 221                 if( *(unsigned char*)mem1 != *(unsigned char*)mem2 )
 222                         return *(unsigned char*)mem1 - *(unsigned char*)mem2;
 223                 mem1 ++;
 224                 mem2 ++;
 225         }
 226         return 0;
 227 }
 228
 229 /**
 230  * \fn EXPORT void *memchr(void *ptr, int value, size_t num)
 231  * \brief Locates the first occurence of \a value starting at \a ptr
 232  * \param ptr   Starting memory location
 233  * \param value Value to find
 234  * \param num   Size of memory area to check
 235  */
 236 EXPORT void *memchr(const void *ptr, int value, size_t num)
 237 {
 238         while(num--)
 239         {
 240                 if( *(const unsigned char*)ptr == (unsigned char)value )
 241                         return (void*)ptr;
 242                 ptr ++;
 243         }
 244         return NULL;
 245 }