X-Git-Url: https://pd.if.org/git/?a=blobdiff_plain;ds=sidebyside;f=functions%2Fstring%2Fmemset.c;h=735a8e60623dc1f9ebeb2cbf8e7f0a218e2efc7e;hb=1a0d9f7faca36f149db3b85c09a72528e0c73a83;hp=17d96f5d2c073a29f835e2f282055bbefd83c051;hpb=1d9d92ba957a0b8307c9a65c35867fde68e6533b;p=pdclib

diff --git a/functions/string/memset.c b/functions/string/memset.c
index 17d96f5..735a8e6 100644
--- a/functions/string/memset.c
+++ b/functions/string/memset.c
@@ -1,19 +1,39 @@
-/* ----------------------------------------------------------------------------
- * $Id$
- * ----------------------------------------------------------------------------
- * Public Domain C Library - http://pdclib.sourceforge.net
- * This code is Public Domain. Use, modify, and redistribute at will.
- * --------------------------------------------------------------------------*/
+/* $Id$ */
 
-#include <__size_t.h>
+/* Release $Name$ */
 
-void * memset( void * dest, int c, size_t n )
+/* memset( void *, int, size_t )
+
+   This file is part of the Public Domain C Library (PDCLib).
+   Permission is granted to use, modify, and / or redistribute at will.
+*/
+
+#include <string.h>
+
+void * memset( void * s, int c, size_t n )
 {
-    unsigned char ch = (unsigned char) c;
-    unsigned char * dest_p = (unsigned char *) dest;
+    unsigned char * p = (unsigned char *) s;
     while ( n-- )
     {
-        *dest_p++ = ch;
+        *p++ = (unsigned char) c;
     }
-    return dest;
+    return s;
+}
+
+#ifdef TEST
+#include <_PDCLIB_test.h>
+
+int main()
+{
+    char s[10] = "xxxxxxxxx";
+    BEGIN_TESTS;
+    TESTCASE( memset( s, 'o', 10 ) == s );
+    TESTCASE( s[9] == 'o' );
+    TESTCASE( memset( s, '_', 0 ) == s );
+    TESTCASE( s[0] == 'o' );
+    TESTCASE( memset( s, '_', 1 ) == s );
+    TESTCASE( s[0] == '_' );
+    TESTCASE( s[1] == 'o' );
+    return TEST_RESULTS;
 }
+#endif