random.c

Go to the documentation of this file.

 
#include "config.h"
#include <errno.h>
#include <stddef.h>
#include <stdint.h>
#include <stdio.h>
#include <string.h>
#include <sys/types.h>
#include "random.h"
#include "exit.h"
#include "file.h"
#include "logging2.h"
#include "message.h"
#ifdef HAVE_SYS_RANDOM_H
#include <sys/random.h>
#endif

static FILE *FpRandom = NULL;

static const unsigned char Base32[] = "abcdefghijklmnopqrstuvwxyz234567";

int mutt_randbuf(void *buf, size_t buflen)
{
  if (buflen > 1048576)
  {
    mutt_error(_("mutt_randbuf buflen=%zu"), buflen);
    return -1;
  }
 
#ifdef HAVE_GETRANDOM
  ssize_t rc;
  ssize_t count = 0;
  do
  {
    // getrandom() can return less than requested if there's insufficient
    // entropy or it's interrupted by a signal.
    rc = getrandom((char *) buf + count, buflen - count, 0);
    if (rc > 0)
      count += rc;
  } while (((rc >= 0) && (count < buflen)) || ((rc == -1) && (errno == EINTR)));
  if (count == buflen)
    return 0;
#endif
  /* let's try urandom in case we're on an old kernel, or the user has
   * configured selinux, seccomp or something to not allow getrandom */
  if (!FpRandom)
  {
    FpRandom = mutt_file_fopen("/dev/urandom", "r");
    if (!FpRandom)
    {
      mutt_error(_("open /dev/urandom: %s"), strerror(errno));
      return -1;
    }
    setvbuf(FpRandom, NULL, _IONBF, 0);
  }
  if (fread(buf, 1, buflen, FpRandom) != buflen)
  {
    mutt_error(_("read /dev/urandom: %s"), strerror(errno));
    return -1;
  }
 
  return 0;
}

void mutt_rand_base32(char *buf, size_t buflen)
{
  if (!buf || (buflen == 0))
    return;
 
  uint8_t *p = (uint8_t *) buf;
 
  if (mutt_randbuf(p, buflen) < 0)
    mutt_exit(1); // LCOV_EXCL_LINE
  for (size_t pos = 0; pos < buflen; pos++)
    p[pos] = Base32[p[pos] % 32];
}

uint64_t mutt_rand64(void)
{
  uint64_t num = 0;
 
  if (mutt_randbuf(&num, sizeof(num)) < 0)
    mutt_exit(1); // LCOV_EXCL_LINE
  return num;
}