X-Git-Url: https://pd.if.org/git/?a=blobdiff_plain;f=functions%2Fstring%2Fmemset.c;h=735a8e60623dc1f9ebeb2cbf8e7f0a218e2efc7e;hb=0c09ab83dad5042cd2a78cebe8273728feb2f45a;hp=48477a71778934b15d517391f5f1530c93967516;hpb=b06bbb1e2a100095da21e18139e4f960604d0d02;p=pdclib

diff --git a/functions/string/memset.c b/functions/string/memset.c
index 48477a7..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