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

diff --git a/functions/string/memset.c b/functions/string/memset.c
index 48477a7..753a333 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[] = "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