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

diff --git a/functions/string/memset.c b/functions/string/memset.c
index 48477a7..4946ec4 100644
--- a/functions/string/memset.c
+++ b/functions/string/memset.c
@@ -1,19 +1,38 @@
-// ----------------------------------------------------------------------------
-// $Id$
-// ----------------------------------------------------------------------------
-// Public Domain C Library - http://pdclib.sourceforge.net
-// This code is Public Domain. Use, modify, and redistribute at will.
-// ----------------------------------------------------------------------------
+/* memset( void *, int, size_t )
 
-#include <__size_t.h>
+   This file is part of the Public Domain C Library (PDCLib).
+   Permission is granted to use, modify, and / or redistribute at will.
+*/
 
-void * memset( void * dest, int c, size_t n )
+#include <string.h>
+
+#ifndef REGTEST
+
+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;
+}
+
+#endif
+
+#ifdef TEST
+#include "_PDCLIB_test.h"
+
+int main( void )
+{
+    char s[] = "xxxxxxxxx";
+    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