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) return &str[i];
 148         }
 149         return NULL;
 150 }
 151
 152 /**
 153  * \fn EXPORT char *strstr(char *str1, const char *str2)
 154  * \brief Search a \a str1 for the first occurance of \a str2
 155  */
 156 EXPORT char *strstr(char *str1, const char *str2)
 157 {
 158         const char      *test = str2;
 159
 160         for(;*str1;str1++)
 161         {
 162                 if(*test == '\0')       return str1;
 163                 if(*str1 == *test)      test++;
 164                 else    test = str2;
 165         }
 166         return NULL;
 167 }
 168
 169 // --- Memory ---
 170 /**
 171  * \fn EXPORT void *memset(void *dest, int val, size_t num)
 172  * \brief Clear memory with the specified value
 173  */
 174 EXPORT void *memset(void *dest, int val, size_t num)
 175 {
 176         unsigned char *p = dest;
 177         while(num--)    *p++ = val;
 178         return dest;
 179 }
 180
 181 /**
 182  * \fn EXPORT void *memcpy(void *dest, const void *src, size_t count)
 183  * \brief Copy one memory area to another
 184  */
 185 EXPORT void *memcpy(void *dest, const void *src, size_t count)
 186 {
 187     char *sp = (char *)src;
 188     char *dp = (char *)dest;
 189     for(;count--;) *dp++ = *sp++;
 190     return dest;
 191 }
 192
 193 /**
 194  * \fn EXPORT void *memmove(void *dest, const void *src, size_t count)
 195  * \brief Copy data in memory, avoiding overlap problems
 196  */
 197 EXPORT void *memmove(void *dest, const void *src, size_t count)
 198 {
 199     char *sp = (char *)src;
 200     char *dp = (char *)dest;
 201         // Check if corruption will happen
 202         if( (intptr_t)dest > (intptr_t)src && (intptr_t)dest < (intptr_t)src+count )
 203                 for(;count--;) dp[count] = sp[count];
 204         else
 205         for(;count--;) *dp++ = *sp++;
 206     return dest;
 207 }
 208
 209 /**
 210  * \fn EXPORT int memcmp(const void *mem1, const void *mem2, size_t count)
 211  * \brief Compare two regions of memory
 212  * \param mem1  Region 1
 213  * \param mem2  Region 2
 214  * \param count Number of bytes to check
 215  */
 216 EXPORT int memcmp(const void *mem1, const void *mem2, size_t count)
 217 {
 218         while(count--)
 219         {
 220                 if( *(unsigned char*)mem1 != *(unsigned char*)mem2 )
 221                         return *(unsigned char*)mem1 - *(unsigned char*)mem2;
 222                 mem1 ++;
 223                 mem2 ++;
 224         }
 225         return 0;
 226 }
 227
 228 /**
 229  * \fn EXPORT void *memchr(void *ptr, int value, size_t num)
 230  * \brief Locates the first occurence of \a value starting at \a ptr
 231  * \param ptr   Starting memory location
 232  * \param value Value to find
 233  * \param num   Size of memory area to check
 234  */
 235 EXPORT void *memchr(const void *ptr, int value, size_t num)
 236 {
 237         while(num--)
 238         {
 239                 if( *(const unsigned char*)ptr == (unsigned char)value )
 240                         return ptr;
 241                 ptr ++;
 242         }
 243         return NULL;
 244 }