Changeset 135 for trunk/src

Timestamp:

Oct 27, 2007, 12:51:09 AM (17 years ago)

Author:

rainer

Message:

Reentrant sh_entropy() function.

File:

• trunk/src/sh_entropy.c (modified) (19 diffs)

trunk/src/sh_entropy.c

@@ -117 +117 @@
 int sh_entropy(int getbytes, char * nbuf)
 {
-    static int fd = -1;
+    int fd = -1;
     int n;
     byte buffer[256+2];
@@ -148 +148 @@
         int addr_len;
 
-      #ifdef EGD_SOCKET_NAME
+#ifdef EGD_SOCKET_NAME
         bname = EGD_SOCKET_NAME;
-      #endif
+#endif
         if ( !bname || !*bname )
             bname = _("=entropy");
@@ -191 +191 @@
                              _("sh_entropy") ); 
             SH_FREE(name);
+            close(fd);
             SL_RETURN( -1, _("sh_entropy") );
           }
@@ -206 +207 @@
             sh_error_handle ((-1), FIL__, __LINE__, myerror, MSG_E_SUBGEN,
                              _("cannot write to EGD"),
-                             _("sh_entropy") ); 
+                             _("sh_entropy") );
+            close(fd);
             SL_RETURN( -1, _("sh_entropy") );
           }
@@ -247 +249 @@
             sh_error_handle ((-1), FIL__, __LINE__, myerror, MSG_E_SUBGEN,
                              _("cannot write to EGD"),
-                             _("sh_entropy") ); 
+                             _("sh_entropy") );
+            close(fd);
             SL_RETURN( -1, _("sh_entropy") );
           }
@@ -267 +270 @@
     }
     memset(buffer, 0, sizeof(buffer) );
-
+    close(fd);
     SL_RETURN( 0, _("sh_entropy") ); /* success */
 }
@@ -275 +278 @@
 
 #if defined (HAVE_URANDOM)
+
+#include "sh_pthread.h"
 
 int read_mbytes(int timeout_val, char * path, char * nbuf, int nbytes)
@@ -373 +378 @@
       /* -- Give out nbytes bytes from the new pool. --
        */
+      SH_MUTEX_LOCK_UNSAFE(mutex_skey);
       for (i = 0; i < KEY_BYT; ++i)
         {
@@ -379 +385 @@
             nbuf[i] = keybuf[i];
         }
+      SH_MUTEX_UNLOCK_UNSAFE(mutex_skey);
       memset (keybuf, '\0', KEY_BYT);
       
@@ -408 +415 @@
 
 #include "sh_static.h"
+#include "sh_pthread.h"
 
 static
@@ -432 +440 @@
 
 static
-sourcetable_t source[] = {
+sourcetable_t source_template[] = {
   { N_("w"),
     N_("w"),
@@ -712 +720 @@
   int    imax, selcount;
   char errbuf[SH_ERRBUF_SIZE];
+
+  sourcetable_t  *source = NULL;
   
   SL_ENTER(_("sh_entropy"));
@@ -723 +733 @@
   /* --- If there is entropy in the pool, return it. ---
    */
+  SH_MUTEX_LOCK_UNSAFE(mutex_skey);
   if (skey->poolc >= nbytes)
     {
@@ -731 +742 @@
           --skey->poolc;
         }
+      SH_MUTEX_UNLOCK_UNSAFE(mutex_skey); /* alternative path */
       SL_RETURN(0, _("sh_entropy"));
     }
+  SH_MUTEX_UNLOCK_UNSAFE(mutex_skey);
 
 
@@ -746 +759 @@
                       _("sl_get_cap_sub"));
     }
+
+  while (source_template[i].command != NULL) {
+    ++i;
+  }
+  source = SH_ALLOC(i * sizeof(sourcetable_t));
+  for (j = 0; j < i;++j)
+    memcpy(&source[j], &source_template[j], sizeof(sourcetable_t));
+  i = 0;
 
   while (source[i].command != NULL) {
@@ -877 +898 @@
     }
   buffer[bufcount] = '\0';
+  
+  SH_FREE(source);
 
   if (0 != (caperr = sl_drop_cap_sub()))
@@ -906 +929 @@
       /* store in system pool
        */
+      SH_MUTEX_LOCK_UNSAFE(mutex_skey);
       for (i = 0; i < KEY_BYT; ++i)
         skey->poolv[i] = keybuf[i];
       skey->poolc = KEY_BYT;
+      SH_MUTEX_UNLOCK_UNSAFE(mutex_skey);
       memset (buffer, '\0', BUF_ENT+2);
       memset (keybuf, '\0', KEY_BYT);
@@ -921 +946 @@
   /* give out nbytes Bytes from the entropy pool
    */
+  SH_MUTEX_LOCK_UNSAFE(mutex_skey);
   for (i = 0; i < nbytes; ++i)
     {
@@ -926 +952 @@
       --skey->poolc;
     }
+  SH_MUTEX_UNLOCK_UNSAFE(mutex_skey);
 
   SL_RETURN(0, _("sh_entropy"));