X-Git-Url: https://git.ucc.asn.au/?a=blobdiff_plain;f=Usermode%2FLibraries%2Flibc.so_src%2Fstdlib.c;h=c91106c4070c72bf1008599ce0dc980338ce7902;hb=58716e08fd3ce62861636a300879e10e930b177b;hp=733cb919a8cfadd09f217301b3940e5a5ee899c6;hpb=7941d6b368acb0abc17e6a77ffaf7b4c306b67ab;p=tpg%2Facess2.git

diff --git a/Usermode/Libraries/libc.so_src/stdlib.c b/Usermode/Libraries/libc.so_src/stdlib.c
index 733cb919..c91106c4 100644
--- a/Usermode/Libraries/libc.so_src/stdlib.c
+++ b/Usermode/Libraries/libc.so_src/stdlib.c
@@ -2,17 +2,19 @@
  * AcessOS Basic C Library
  * stdlib.c
  */
-/**
- * \todo Move half of these to stdio
- */
 #include <acess/sys.h>
 #include <stdlib.h>
 #include <stdio.h>
+#include <errno.h>
+#include <ctype.h>
+#include <limits.h>
 #include "lib.h"
 
 #define _stdout	1
 #define _stdin	0
 
+extern void	*_crt0_exit_handler;
+
 // === PROTOTYPES ===
 EXPORT int	atoi(const char *str);
 EXPORT void	exit(int status);
@@ -24,6 +26,8 @@ void	(*g_stdlib_exithandler)(void);
 void atexit(void (*__func)(void))
 {
 	g_stdlib_exithandler = __func;
+	// TODO: Replace with meta-function to allow multiple atexit() handlers
+	_crt0_exit_handler = __func;	
 }
 
 /**
@@ -32,6 +36,8 @@ void atexit(void (*__func)(void))
  */
 EXPORT void exit(int status)
 {
+	if( g_stdlib_exithandler )
+		g_stdlib_exithandler();
 	_exit(status);
 }
 
@@ -65,66 +71,96 @@ EXPORT void qsort(void *base, size_t nmemb, size_t size, int(*compar)(const void
 	}
 }
 
-/**
- * \fn EXPORT int atoi(const char *str)
- * \brief Convert a string to an integer
- */
-EXPORT int atoi(const char *str)
+EXPORT long long strtoll(const char *str, char **end, int base)
 {
 	 int	neg = 0;
-	 int	ret = 0;
-	
-	// NULL Check
-	if(!str)	return 0;
+	long long	ret = 0;
 	
-	while(*str == ' ' || *str == '\t')	str++;
+	if( !str || base < 0 || base > 36 || base == 1 ) {
+		if(end)
+			*end = (char*)str;
+		errno = EINVAL;
+		return 0;
+	}
+
+	while( isspace(*str) )
+		str++;
 	
-	// Check for negative
-	if(*str == '-') {
-		neg = 1;
+	// Check for negative (or positive) sign
+	if(*str == '-' || *str == '+') {
+		neg = (*str == '-');
 		str++;
 	}
 	
-	if(*str == '0') {
+	if( base == 0 || base == 16 ) {
+		if( *str == '0' && str[1] == 'x' ) {
+			str += 2;
+			base = 16;
+		}
+	}
+	
+	if( base == 0 && *str == '0' ) {
 		str ++;
-		if(*str == 'x') {
-			str++;
-			// Hex
-			while( ('0' <= *str && *str <= '9')
-				|| ('A' <= *str && *str <= 'F' )
-				|| ('a' <= *str && *str <= 'f' )
-				)
-			{
-				ret *= 16;
-				if(*str <= '9') {
-					ret += *str - '0';
-				} else if (*str <= 'F') {
-					ret += *str - 'A' + 10;
-				} else {
-					ret += *str - 'a' + 10;
-				}
-				str++;
-			}
-		} else {
-			// Octal
-			while( '0' <= *str && *str <= '7' )
-			{
-				ret *= 8;
-				ret += *str - '0';
-				str++;
-			}
+		base = 8;
+	}
+
+	if( base == 0 )
+		base = 10;
+
+	while( *str )
+	{
+		 int	next = -1;
+		if( base <= 10 ) {
+			if( '0' <= *str && *str <= '0'+base-1 )
+				next = *str - '0';
 		}
-	} else {
-		// Decimal
-		while( '0' <= *str && *str <= '9' )
-		{
-			ret *= 10;
-			ret += *str - '0';
-			str++;
+		else {
+			if( '0' <= *str && *str <= '9' )
+				next = *str - '0';
+			if( 'A' <= *str && *str <= 'A'+base-10-1 )
+				next = *str - 'A';
+			if( 'a' <= *str && *str <= 'a'+base-10-1 )
+				next = *str - 'a';
 		}
+		if( next < 0 )
+			break;
+		ret *= base;
+		ret += next;
+		str ++;
 	}
-	
-	// Negate if needed
-	if(neg)	ret = -ret;
+
+	if( neg )
+		ret = -ret;	
+
+	if(end)
+		*end = (char*)str;
 	return ret;
 }
+
+EXPORT long strtol(const char *str, char **end, int base)
+{
+	long long tmp = strtoll(str, end, base);
+	if( tmp > LONG_MAX || tmp < LONG_MIN ) {
+		errno = ERANGE;
+		return (tmp > LONG_MAX) ? LONG_MAX : LONG_MIN;
+	}
+	return tmp;
+}
+
+/**
+ * \fn EXPORT int atoi(const char *str)
+ * \brief Convert a string to an integer
+ */
+EXPORT int atoi(const char *str)
+{
+	long long	tmp = strtoll(str, NULL, 0);
+	if( tmp > INT_MAX || tmp < INT_MIN ) {
+		errno = ERANGE;
+		return (tmp > INT_MAX) ? INT_MAX : INT_MIN;
+	}
+	return tmp;
+}
+
+int abs(int j) { return j < 0 ? -j : j; }
+long int labs(long int j) { return j < 0 ? -j : j; }
+