| // Copyright (C) 2010 David Sugar, Tycho Softworks. |
| // |
| // This file is part of GNU uCommon C++. |
| // |
| // GNU uCommon C++ is free software: you can redistribute it and/or modify |
| // it under the terms of the GNU Lesser General Public License as published |
| // by the Free Software Foundation, either version 3 of the License, or |
| // (at your option) any later version. |
| // |
| // GNU uCommon C++ is distributed in the hope that it will be useful, |
| // but WITHOUT ANY WARRANTY; without even the implied warranty of |
| // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
| // GNU Lesser General Public License for more details. |
| // |
| // You should have received a copy of the GNU Lesser General Public License |
| // along with GNU uCommon C++. If not, see <http://www.gnu.org/licenses/>. |
| |
| #include "local.h" |
| |
| #ifndef _MSWINDOWS_ |
| #include <fcntl.h> |
| #endif |
| |
| void Random::seed(void) |
| { |
| time_t now; |
| |
| time(&now); |
| srand((int)now); |
| } |
| |
| bool Random::seed(const unsigned char *buf, size_t size) |
| { |
| #ifdef _MSWINDOWS_ |
| return false; |
| #else |
| int fd = open("/dev/random", O_WRONLY); |
| bool result = false; |
| |
| if(fd > -1) { |
| if(write(fd, buf, size) > 0) |
| result = true; |
| close(fd); |
| } |
| return result; |
| #endif |
| } |
| |
| size_t Random::key(unsigned char *buf, size_t size) |
| { |
| #ifdef _MSWINDOWS_ |
| if((_handle != (HCRYPTPROV)NULL) && CryptGenRandom(_handle, size, buf)) |
| return size; |
| return 0; |
| #else |
| int fd = open("/dev/random", O_RDONLY); |
| ssize_t result = 0; |
| |
| if(fd > -1) { |
| result = read(fd, buf, size); |
| close(fd); |
| } |
| |
| if(result < 0) |
| result = 0; |
| |
| return (size_t)result; |
| #endif |
| } |
| |
| size_t Random::fill(unsigned char *buf, size_t size) |
| { |
| #ifdef _MSWINDOWS_ |
| return key(buf, size); |
| #else |
| int fd = open("/dev/urandom", O_RDONLY); |
| ssize_t result = 0; |
| |
| if(fd > -1) { |
| result = read(fd, buf, size); |
| close(fd); |
| } |
| // ugly...would not trust it |
| else { |
| result = 0; |
| while(result++ < (ssize_t)size) |
| *(buf++) = rand() & 0xff; |
| } |
| |
| if(result < 0) |
| result = 0; |
| |
| return (size_t)result; |
| #endif |
| } |
| |
| bool Random::status(void) |
| { |
| #ifdef _MSWINDOWS_ |
| return true; |
| #else |
| if(fsys::is_file("/dev/random")) |
| return true; |
| |
| return false; |
| #endif |
| } |
| |