pike.git/ src/ builtin.cmod

2016-01-17

2016-01-17 16:04:02 by Martin Nilsson <nilsson@fastmail.com>

• ffaa5a3cb321837cf357a3524e4f9aa0517a5449 (37 lines) (+12/-25) ^{[ Show | Annotate ]}
  Branch: 8.1

Make RandomSystem use a global random source between objects.

2202:      #ifdef __NT__   #include <wincrypt.h> + static HCRYPTPROV crypto_handle;   PIKECLASS RandomSystem   {    INHERIT RandomInterface; -  CVAR HCRYPTPROV handle; +     -  INIT -  { -  THIS->handle = NULL; -  } -  +     PIKEFUN string(8bit) random_string(int len)    {    if( len<1 )    RETURN empty_pike_string;    if(!crypto_handle)    { -  if( !CryptAcquireContext(&THIS->handle, 0, 0, PROV_RSA_FULL, +  if( !CryptAcquireContext(&crypto_handle, 0, 0, PROV_RSA_FULL,    CRYPT_VERIFYCONTEXT|CRYPT_SILENT) )    Pike_error("Failed to set up Crypto Service.\n");    }       struct pike_string *res = begin_shared_string(size); -  if( !CryptGenRandom(THIS->handle, size, (BYTE*)res->str) ) +  if( !CryptGenRandom(crypto_handle, size, (BYTE*)res->str) )    {    do_free_unlinked_pike_string (res);    Pike_error("Failed to create random data.\n");
2234:    }   }   #else + static int random_fd;   PIKECLASS RandomSystem   {    INHERIT RandomInterface; -  CVAR int fd; +     -  INIT -  { -  THIS->fd = -1; -  } -  -  EXIT -  { -  close(THIS->fd); -  } -  +     PIKEFUN string(8bit) random_string(int len)    {    if( len<1 )    RETURN empty_pike_string;    -  if( THIS->fd==-1 ) +  if( random_fd==-1 )    { -  THIS->fd = open("/dev/urandom", O_RDONLY); -  if( THIS->fd==-1 ) +  random_fd = open("/dev/urandom", O_RDONLY); +  if( random_fd==-1 )    Pike_error("Failed to open /dev/urandom.\n");    }   
2265:    char* str = ret->str;    while( len )    { -  int sz = read(THIS->fd, str, len); +  int sz = read(random_fd, str, len);    str += sz;    len -= sz;    }    -  pop_n_elems(args); -  push_string(end_shared_string(ret)); +  RETURN end_shared_string(ret);    }   }   #endif
6073:    SET_SVAL(gc_destruct_cb, PIKE_T_INT, NUMBER_NUMBER, integer, 0);    SET_SVAL(gc_done_cb, PIKE_T_INT, NUMBER_NUMBER, integer, 0);    INIT; +  random_fd = -1;   }      void exit_builtin(void)
6095:   #ifndef USE_SETENV    if (env_allocs) free_mapping (env_allocs);   #endif +  if (random_fd!=-1) close(random_fd);   }