policy_manager/random_provider.cc

// Copyright (c) 2014 The Chromium OS Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include <stdio.h>
#include <unistd.h>

#include <base/file_path.h>
#include <base/file_util.h>
#include <base/stringprintf.h>

#include "policy_manager/random_provider.h"

using std::string;

namespace {

// The device providing randomness.
const char* kRandomDevice = "/dev/urandom";

}  // namespace

namespace chromeos_policy_manager {

// The random variable class implementation.
class RandomVariable : public Variable<uint64_t> {
 public:
  RandomVariable(FILE* fp) : fp_(fp) {}
  virtual ~RandomVariable() {}

 protected:
  virtual const uint64_t* GetValue(base::TimeDelta /* timeout */,
                                   string* errmsg) {
    uint64_t result;
    // Aliasing via char pointer abides by the C/C++ strict-aliasing rules.
    char* const buf = reinterpret_cast<char*>(&result);
    unsigned int buf_rd = 0;

    while (buf_rd < sizeof(result)) {
      int rd = fread(buf + buf_rd, 1, sizeof(result) - buf_rd, fp_.get());
      if (rd == 0 || ferror(fp_.get())) {
        // Either EOF on fp or read failed.
        if (errmsg) {
          *errmsg = StringPrintf("Error reading from the random device: %s",
                                 kRandomDevice);
        }
        return NULL;
      }
      buf_rd += rd;
    }

    return new uint64_t(result);
  }

 private:
  DISALLOW_COPY_AND_ASSIGN(RandomVariable);

  file_util::ScopedFILE fp_;
};


// RandomProvider implementation.

bool RandomProvider::DoInit(void) {
  FILE* fp = fopen(kRandomDevice, "r");
  if (!fp)
    return false;
  var_random_seed = new RandomVariable(fp);
  return true;
}

}  // namespace chromeos_policy_manager