git.sesse.net Git - vlc/blob - src/misc/rand.c

   1 /*****************************************************************************
   2  * rand.c : non-predictible random bytes generator
   3  *****************************************************************************
   4  * Copyright © 2007 Rémi Denis-Courmont
   5  * $Id$
   6  *
   7  * This program is free software; you can redistribute it and/or modify
   8  * it under the terms of the GNU General Public License as published by
   9  * the Free Software Foundation; either version 2 of the License, or
  10  * (at your option) any later version.
  11  *
  12  * This program is distributed in the hope that it will be useful,
  13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15  * GNU General Public License for more details.
  16  *
  17  * You should have received a copy of the GNU General Public License
  18  * along with this program; if not, write to the Free Software
  19  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
  20  *****************************************************************************/
  21
  22 #include <vlc/vlc.h>
  23 #include <vlc_rand.h>
  24
  25 #ifndef WIN32
  26 #include <stdint.h>
  27 #include <string.h>
  28 #include <stdlib.h>
  29
  30 #include <sys/types.h>
  31 #include <fcntl.h>
  32 #include <unistd.h>
  33 #include <pthread.h>
  34
  35 #include <vlc_md5.h>
  36
  37 /*
  38  * Pseudo-random number generator using a HMAC-MD5 in counter mode.
  39  * Probably not very secure (expert patches welcome) but definitely
  40  * better than rand() which is defined to be reproducible...
  41  */
  42 #define BLOCK_SIZE 64
  43
  44 static uint8_t okey[BLOCK_SIZE], ikey[BLOCK_SIZE];
  45
  46 static void vlc_rand_init (void)
  47 {
  48 #if defined (__OpenBSD__) || defined (__OpenBSD_kernel__)
  49     static const char randfile[] = "/dev/random";
  50 #else
  51     static const char randfile[] = "/dev/urandom";
  52 #endif
  53     uint8_t key[BLOCK_SIZE];
  54
  55     /* Get non-predictible value as key for HMAC */
  56     int fd = open (randfile, O_RDONLY);
  57     if (fd == -1)
  58         return; /* Uho! */
  59
  60     for (size_t i = 0; i < sizeof (key);)
  61     {
  62          ssize_t val = read (fd, key + i, sizeof (key) - i);
  63          if (val > 0)
  64              i += val;
  65     }
  66
  67     /* Precompute outer and inner keys for HMAC */
  68     for (size_t i = 0; i < sizeof (key); i++)
  69     {
  70         okey[i] = key[i] ^ 0x5c;
  71         ikey[i] = key[i] ^ 0x36;
  72     }
  73
  74     close (fd);
  75 }
  76
  77
  78 void vlc_rand_bytes (void *buf, size_t len)
  79 {
  80     static pthread_mutex_t lock = PTHREAD_MUTEX_INITIALIZER;
  81     static uint64_t counter = 0;
  82
  83     struct md5_s md;
  84     uint64_t stamp = NTPtime64 ();
  85
  86     while (len > 0)
  87     {
  88         uint64_t val;
  89
  90         pthread_mutex_lock (&lock);
  91         if (counter == 0)
  92             vlc_rand_init ();
  93         val = counter++;
  94         pthread_mutex_unlock (&lock);
  95
  96         InitMD5 (&md);
  97         AddMD5 (&md, ikey, sizeof (ikey));
  98         AddMD5 (&md, &stamp, sizeof (stamp));
  99         AddMD5 (&md, &val, sizeof (val));
 100         EndMD5 (&md);
 101
 102         if (len < sizeof (md.p_digest))
 103         {
 104             memcpy (buf, md.p_digest, len);
 105             break;
 106         }
 107
 108         memcpy (buf, md.p_digest, sizeof (md.p_digest));
 109         len -= sizeof (md.p_digest);
 110         buf = ((uint8_t *)buf) + sizeof (md.p_digest);
 111     }
 112 }
 113
 114 #else /* WIN32 */
 115 #define _CRT_RAND_S
 116 #include <stdlib.h>
 117
 118 void vlc_rand_bytes (void *buf, size_t len)
 119 {
 120     while (len > 0)
 121     {
 122         unsigned int val;
 123         rand_s (&val);
 124
 125         if (len < sizeof (val))
 126         {
 127             memcpy (buf, &val, len);
 128             break;
 129         }
 130
 131         memcpy (buf, &val, sizeof (val));
 132     }
 133 }
 134 #endif