Yet another PRNG, versalite RC4 implementation for multi-purpose and multi-platform usage

src/Makefile.am

@@ -6,5 +6,5 @@ AM_LDFLAGS =
 # this lists the binaries to produce, the (non-PHONY, binary) targets in
 # the previous manual Makefile
 bin_PROGRAMS = nwipe
-nwipe_SOURCES = context.h logging.h options.h prng.h version.h temperature.h nwipe.c gui.c method.h pass.c device.c gui.h isaac_rand/isaac_standard.h isaac_rand/isaac_rand.h isaac_rand/isaac_rand.c isaac_rand/isaac64.h isaac_rand/isaac64.c mt19937ar-cok/mt19937ar-cok.c nwipe.h mt19937ar-cok/mt19937ar-cok.h alfg/add_lagg_fibonacci_prng.h alfg/add_lagg_fibonacci_prng.c xor/xoroshiro256_prng.h xor/xoroshiro256_prng.c pass.h device.h logging.c method.c options.c prng.c version.c temperature.c PDFGen/pdfgen.h PDFGen/pdfgen.c create_pdf.c create_pdf.h embedded_images/shred_db.jpg.c embedded_images/shred_db.jpg.h  embedded_images/tick_erased.jpg.c embedded_images/tick_erased.jpg.h embedded_images/redcross.c embedded_images/redcross.h hpa_dco.h hpa_dco.c miscellaneous.h miscellaneous.c embedded_images/nwipe_exclamation.jpg.h embedded_images/nwipe_exclamation.jpg.c conf.h conf.c customers.h customers.c hddtemp_scsi/hddtemp.h hddtemp_scsi/scsi.h hddtemp_scsi/scsicmds.h hddtemp_scsi/get_scsi_temp.c hddtemp_scsi/scsi.c hddtemp_scsi/scsicmds.c
+nwipe_SOURCES = context.h logging.h options.h prng.h version.h temperature.h nwipe.c gui.c method.h pass.c device.c gui.h rc4/rc4_prng.h rc4/rc4_prng.c isaac_rand/isaac_standard.h isaac_rand/isaac_rand.h isaac_rand/isaac_rand.c isaac_rand/isaac64.h isaac_rand/isaac64.c mt19937ar-cok/mt19937ar-cok.c nwipe.h mt19937ar-cok/mt19937ar-cok.h alfg/add_lagg_fibonacci_prng.h alfg/add_lagg_fibonacci_prng.c xor/xoroshiro256_prng.h xor/xoroshiro256_prng.c pass.h device.h logging.c method.c options.c prng.c version.c temperature.c PDFGen/pdfgen.h PDFGen/pdfgen.c create_pdf.c create_pdf.h embedded_images/shred_db.jpg.c embedded_images/shred_db.jpg.h  embedded_images/tick_erased.jpg.c embedded_images/tick_erased.jpg.h embedded_images/redcross.c embedded_images/redcross.h hpa_dco.h hpa_dco.c miscellaneous.h miscellaneous.c embedded_images/nwipe_exclamation.jpg.h embedded_images/nwipe_exclamation.jpg.c conf.h conf.c customers.h customers.c hddtemp_scsi/hddtemp.h hddtemp_scsi/scsi.h hddtemp_scsi/scsicmds.h hddtemp_scsi/get_scsi_temp.c hddtemp_scsi/scsi.c hddtemp_scsi/scsicmds.c
 nwipe_LDADD = $(PARTED_LIBS) $(LIBCONFIG)

src/gui.c

@@ -1616,11 +1616,12 @@ void nwipe_gui_prng( void )
     extern nwipe_prng_t nwipe_aes_ctr_prng;
     extern nwipe_prng_t nwipe_xoroshiro256_prng;
     extern nwipe_prng_t nwipe_add_lagg_fibonacci_prng;
+    extern nwipe_prng_t nwipe_rc4_prng;
 
     extern int terminate_signal;
 
     /* The number of implemented PRNGs. */
-    const int count = 5;
+    const int count = 6;
 
     /* The first tabstop. */
     const int tab1 = 2;
@@ -1662,6 +1663,10 @@ void nwipe_gui_prng( void )
     {
         focus = 4;
     }
+    if( nwipe_options.prng == &nwipe_rc4_prng )
+    {
+        focus = 5;
+    }
     do
     {
         /* Clear the main window. */
@@ -1678,6 +1683,7 @@ void nwipe_gui_prng( void )
         mvwprintw( main_window, yy++, tab1, "  %s", nwipe_isaac64.label );
         mvwprintw( main_window, yy++, tab1, "  %s", nwipe_add_lagg_fibonacci_prng.label );
         mvwprintw( main_window, yy++, tab1, "  %s", nwipe_xoroshiro256_prng.label );
+        mvwprintw( main_window, yy++, tab1, "  %s", nwipe_rc4_prng.label );
         yy++;
 
         /* Print the cursor. */
@@ -1852,6 +1858,38 @@ void nwipe_gui_prng( void )
                            tab1,
                            "especially for legacy systems, due to its efficiency and minimal demands.  " );
                 break;
+
+            case 5:
+
+                mvwprintw( main_window,
+                           yy++,
+                           tab1,
+                           "RC4, originally designed by Ron Rivest, is a widely used symmetric stream cipher       " );
+                mvwprintw( main_window,
+                           yy++,
+                           tab1,
+                           "algorithm that can also function as a pseudo-random number generator (PRNG).          " );
+                mvwprintw( main_window,
+                           yy++,
+                           tab1,
+                           "Although it was primarily intended for encryption, RC4 has been adapted for various    " );
+                mvwprintw( main_window,
+                           yy++,
+                           tab1,
+                           "applications that require random number generation. The algorithm features a variable  " );
+                mvwprintw( main_window,
+                           yy++,
+                           tab1,
+                           "key length and generates numbers in a fast, byte-oriented manner. It is suitable for   " );
+                mvwprintw( main_window,
+                           yy++,
+                           tab1,
+                           "scenarios requiring simplicity and speed, but newer PRNGs may offer better randomness  " );
+                mvwprintw( main_window,
+                           yy++,
+                           tab1,
+                           "for cryptographic purposes.                                                           " );
+                break;
         }
 
         /* switch */
@@ -1922,6 +1960,10 @@ void nwipe_gui_prng( void )
                 {
                     nwipe_options.prng = &nwipe_xoroshiro256_prng;
                 }
+                if( focus == 5 )
+                {
+                    nwipe_options.prng = &nwipe_rc4_prng;
+                }
                 return;
 
             case KEY_BACKSPACE:

src/options.c

@@ -44,6 +44,7 @@ int nwipe_options_parse( int argc, char** argv )
     extern nwipe_prng_t nwipe_isaac64;
     extern nwipe_prng_t nwipe_add_lagg_fibonacci_prng;
     extern nwipe_prng_t nwipe_xoroshiro256_prng;
+    extern nwipe_prng_t nwipe_rc4_prng;
 
     /* The getopt() result holder. */
     int nwipe_opt;
@@ -503,6 +504,11 @@ int nwipe_options_parse( int argc, char** argv )
                     nwipe_options.prng = &nwipe_xoroshiro256_prng;
                     break;
                 }
+                if( strcmp( optarg, "rc4_prng" ) == 0 )
+                {
+                    nwipe_options.prng = &nwipe_rc4_prng;
+                    break;
+                }
 
                 /* Else we do not know this PRNG. */
                 fprintf( stderr, "Error: Unknown prng '%s'.\n", optarg );
@@ -554,6 +560,7 @@ void nwipe_options_log( void )
     extern nwipe_prng_t nwipe_isaac64;
     extern nwipe_prng_t nwipe_add_lagg_fibonacci_prng;
     extern nwipe_prng_t nwipe_xoroshiro256_prng;
+    extern nwipe_prng_t nwipe_rc4_prng;
 
     /**
      *  Prints a manifest of options to the log.
@@ -623,6 +630,11 @@ void nwipe_options_log( void )
                 {
                     nwipe_log( NWIPE_LOG_NOTICE, "  prng     = Isaac" );
                 }
+                if( nwipe_options.prng == &nwipe_rc4_prng )
+                {
+                    nwipe_log( NWIPE_LOG_NOTICE, "  prng     = RC4" );
+                }
+
                 else
                 {
                     if( nwipe_options.prng == &nwipe_isaac64 )
@@ -714,7 +726,7 @@ void display_help()
     puts( "  -l, --logfile=FILE      Filename to log to. Default is STDOUT\n" );
     puts( "  -P, --PDFreportpath=PATH Path to write PDF reports to. Default is \".\"" );
     puts( "                           If set to \"noPDF\" no PDF reports are written.\n" );
-    puts( "  -p, --prng=METHOD       PRNG option (mersenne|twister|isaac|isaac64|add_lagg_fibonacci_prng)\n" );
+    puts( "  -p, --prng=METHOD       PRNG option (mersenne|twister|isaac|isaac64|add_lagg_fibonacci_prng|rc4_prng)\n" );
     puts( "  -q, --quiet             Anonymize logs and the GUI by removing unique data, i.e." );
     puts( "                          serial numbers, LU WWN Device ID, and SMBIOS/DMI data" );
     puts( "                          XXXXXX = S/N exists, ????? = S/N not obtainable\n" );

src/prng.c

@@ -21,12 +21,14 @@
 #include "prng.h"
 #include "context.h"
 #include "logging.h"
+#include <stdio.h>
 
 #include "mt19937ar-cok/mt19937ar-cok.h"
 #include "isaac_rand/isaac_rand.h"
 #include "isaac_rand/isaac64.h"
 #include "alfg/add_lagg_fibonacci_prng.h"  //Lagged Fibonacci generator prototype
 #include "xor/xoroshiro256_prng.h"  //XORoshiro-256 prototype
+#include "rc4/rc4_prng.h"  //RC4 protoype
 
 nwipe_prng_t nwipe_twister = { "Mersenne Twister (mt19937ar-cok)", nwipe_twister_init, nwipe_twister_read };
 
@@ -40,6 +42,86 @@ nwipe_prng_t nwipe_add_lagg_fibonacci_prng = { "Lagged Fibonacci generator",
 /* XOROSHIRO-256 PRNG Structure */
 nwipe_prng_t nwipe_xoroshiro256_prng = { "XORoshiro-256", nwipe_xoroshiro256_prng_init, nwipe_xoroshiro256_prng_read };
 
+/* RC4 PRNG Structure */
+nwipe_prng_t nwipe_rc4_prng = { "RC4", nwipe_rc4_prng_init, nwipe_rc4_prng_read };
+
+// Function to check CPUID and test if SSE4.1 is supported
+#if defined( _MSC_VER )
+#include <intrin.h>  // For MSVC compilers to use __cpuid
+#else
+#include <cpuid.h>  // For GCC/Clang compilers to use __cpuid
+#endif
+
+/**
+ * Function to check if the CPU supports SSE 4.1.
+ *
+ * @return 1 if SSE 4.1 is supported, 0 otherwise.
+ */
+int cpu_supports_sse41()
+{
+    unsigned int eax, ebx, ecx, edx;
+
+#if defined( _MSC_VER )
+    int cpuInfo[4];
+    // Call __cpuid with eax = 1 to get feature information
+    __cpuid( cpuInfo, 1 );
+    ecx = cpuInfo[2];  // ECX register contains feature bits
+#else
+    // Use GCC/Clang __cpuid function
+    __cpuid( 1, eax, ebx, ecx, edx );
+#endif
+
+    // SSE 4.1 is indicated by bit 19 in the ECX register.
+    // If this bit is set, the CPU supports SSE 4.1.
+    return ( ecx & ( 1 << 19 ) ) != 0;
+}
+
+/**
+ * Function to check if the CPU supports AVX2.
+ *
+ * @return 1 if AVX2 is supported, 0 otherwise.
+ */
+
+/*
+int cpu_supports_avx2()
+{
+   unsigned int eax, ebx, ecx, edx;
+
+#if defined( _MSC_VER )
+   int cpuInfo[4];
+   // Call __cpuid with eax = 1 to get basic feature information
+   __cpuid( cpuInfo, 1 );
+   ecx = cpuInfo[2];  // ECX contains basic feature bits
+
+   // Check if the CPU supports AVX (bit 28 of ECX in leaf 1)
+   if( !( ecx & ( 1 << 28 ) ) )
+   {
+       return 0;  // AVX is not supported, so AVX2 can't be supported
+   }
+
+   // Call __cpuid with eax = 7 and ecx = 0 to get extended features
+   __cpuid( cpuInfo, 7 );
+   ebx = cpuInfo[1];  // EBX contains extended feature bits
+#else
+   // Use GCC/Clang __cpuid for basic feature information (leaf 1)
+   __cpuid( 1, eax, ebx, ecx, edx );
+
+   // Check if the CPU supports AVX (bit 28 of ECX in leaf 1)
+   if( !( ecx & ( 1 << 28 ) ) )
+   {
+       return 0;  // AVX is not supported, so AVX2 can't be supported
+   }
+
+   // Use GCC/Clang __cpuid for extended feature information (leaf 7)
+   __cpuid_count( 7, 0, eax, ebx, ecx, edx );
+#endif
+
+   // AVX2 is indicated by bit 5 in the EBX register of CPUID leaf 7.
+   return ( ebx & ( 1 << 5 ) ) != 0;
+}
+
+*/
+
 /* Print given number of bytes from unsigned integer number to a byte stream buffer starting with low-endian. */
 static inline void u32_to_buffer( u8* restrict buffer, u32 val, const int len )
 {
@@ -340,3 +422,80 @@ int nwipe_xoroshiro256_prng_read( NWIPE_PRNG_READ_SIGNATURE )
 
     return 0;  // Success
 }
+
+int nwipe_rc4_prng_init( NWIPE_PRNG_INIT_SIGNATURE )
+{
+    nwipe_log( NWIPE_LOG_NOTICE, "Initialising RC4 PRNG" );
+
+    if( *state == NULL )
+    {
+        /* This is the first time that we have been called. */
+        *state = malloc( sizeof( rc4_state_t ) );
+    }
+    rc4_init( (rc4_state_t*) *state, (uint64_t*) ( seed->s ), seed->length / sizeof( uint64_t ) );
+
+    return 0;
+}
+
+int nwipe_rc4_prng_read( NWIPE_PRNG_READ_SIGNATURE )
+{
+    u8* restrict bufpos = buffer;
+    size_t words = count / SIZE_OF_RC4_PRNG;  // SIZE_OF_RC4_PRNG is 4096 bytes
+
+    // Check if the CPU supports AVX2 first
+    /*
+        int use_avx2 = cpu_supports_avx2();
+    */
+
+    // Check if the CPU supports SSE4.1
+    int use_sse4 = cpu_supports_sse41();
+
+    /* Loop to fill the buffer with blocks directly from the RC4 algorithm */
+    for( size_t ii = 0; ii < words; ++ii )
+    {
+        /*if( use_avx2 )
+        {
+            // Use AVX2-optimized version
+            rc4_avx2_genrand( (rc4_state_t*) *state, bufpos );
+        }*/
+        if( use_sse4 )
+        {
+            // Use SSE4-optimized version
+            rc4_sse4_genrand( (rc4_state_t*) *state, bufpos );
+        }
+        else
+        {
+            // Fallback to generic version
+            rc4_genrand_4096_to_buf( (rc4_state_t*) *state, bufpos );
+        }
+        bufpos += SIZE_OF_RC4_PRNG;  // Move to the next block
+    }
+
+    /* Handle remaining bytes if count is not a multiple of SIZE_OF_RC4_PRNG */
+    const size_t remain = count % SIZE_OF_RC4_PRNG;
+    if( remain > 0 )
+    {
+        unsigned char temp_output[SIZE_OF_RC4_PRNG];  // Temporary buffer for the last block
+
+        /*if( use_avx2 )
+        {
+            // Use AVX2-optimized version
+            rc4_avx2_genrand( (rc4_state_t*) *state, temp_output );
+        }*/
+        if( use_sse4 )
+        {
+            // Use SSE4-optimized version
+            rc4_sse4_genrand( (rc4_state_t*) *state, temp_output );
+        }
+        else
+        {
+            // Fallback to generic version
+            rc4_genrand_4096_to_buf( (rc4_state_t*) *state, temp_output );
+        }
+
+        // Copy the remaining bytes
+        memcpy( bufpos, temp_output, remain );
+    }
+
+    return 0;  // Success
+}

src/prng.h

@@ -63,6 +63,10 @@ int nwipe_add_lagg_fibonacci_prng_read( NWIPE_PRNG_READ_SIGNATURE );
 int nwipe_xoroshiro256_prng_init( NWIPE_PRNG_INIT_SIGNATURE );
 int nwipe_xoroshiro256_prng_read( NWIPE_PRNG_READ_SIGNATURE );
 
+/* RC4 prototypes. */
+int nwipe_rc4_prng_init( NWIPE_PRNG_INIT_SIGNATURE );
+int nwipe_rc4_prng_read( NWIPE_PRNG_READ_SIGNATURE );
+
 /* Size of the twister is not derived from the architecture, but it is strictly 4 bytes */
 #define SIZE_OF_TWISTER 4
 
@@ -76,4 +80,7 @@ int nwipe_xoroshiro256_prng_read( NWIPE_PRNG_READ_SIGNATURE );
 /* Size of the XOROSHIRO-256 is not derived from the architecture, but it is strictly 32 bytes */
 #define SIZE_OF_XOROSHIRO256_PRNG 32
 
+/* Size of the RC4 is not derived from the architecture, but it is strictly 4096 bytes */
+#define SIZE_OF_RC4_PRNG 4096
+
 #endif /* PRNG_H_ */