1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
|
//
// RandomStream.cpp
//
// Library: Foundation
// Package: Crypt
// Module: RandomStream
//
// Copyright (c) 2004-2006, Applied Informatics Software Engineering GmbH.
// and Contributors.
//
// SPDX-License-Identifier: BSL-1.0
//
#include "Poco/RandomStream.h"
#include "Poco/Random.h"
#include "Poco/SHA1Engine.h"
#if defined(POCO_OS_FAMILY_WINDOWS)
#include "Poco/UnWindows.h"
#include <wincrypt.h>
#elif defined(POCO_OS_FAMILY_UNIX)
#include <fcntl.h>
#include <unistd.h>
#endif
#include <ctime>
namespace Poco {
RandomBuf::RandomBuf(): BufferedStreamBuf(256, std::ios::in)
{
}
RandomBuf::~RandomBuf()
{
}
int RandomBuf::readFromDevice(char* buffer, std::streamsize length)
{
int n = 0;
#if defined(POCO_OS_FAMILY_WINDOWS)
HCRYPTPROV hProvider = 0;
CryptAcquireContext(&hProvider, 0, 0, PROV_RSA_FULL, CRYPT_VERIFYCONTEXT);
CryptGenRandom(hProvider, (DWORD) length, (BYTE*) buffer);
CryptReleaseContext(hProvider, 0);
n = static_cast<int>(length);
#else
#if defined(POCO_OS_FAMILY_UNIX)
int fd = open("/dev/urandom", O_RDONLY, 0);
if (fd >= 0)
{
n = read(fd, buffer, length);
close(fd);
}
#endif
if (n <= 0)
{
// x is here as a source of randomness, so it does not make
// much sense to protect it with a Mutex.
static UInt32 x = 0;
Random rnd1(256);
Random rnd2(64);
x += rnd1.next();
n = 0;
SHA1Engine engine;
UInt32 t = (UInt32) std::time(NULL);
engine.update(&t, sizeof(t));
void* p = this;
engine.update(&p, sizeof(p));
engine.update(buffer, length);
UInt32 junk[32];
engine.update(junk, sizeof(junk));
while (n < length)
{
for (int i = 0; i < 100; ++i)
{
UInt32 r = rnd2.next();
engine.update(&r, sizeof(r));
engine.update(&x, sizeof(x));
x += rnd1.next();
}
DigestEngine::Digest d = engine.digest();
for (DigestEngine::Digest::const_iterator it = d.begin(); it != d.end() && n < length; ++it, ++n)
{
engine.update(*it);
*buffer++ = *it++;
}
}
}
#endif
return n;
}
RandomIOS::RandomIOS()
{
poco_ios_init(&_buf);
}
RandomIOS::~RandomIOS()
{
}
RandomBuf* RandomIOS::rdbuf()
{
return &_buf;
}
RandomInputStream::RandomInputStream(): std::istream(&_buf)
{
}
RandomInputStream::~RandomInputStream()
{
}
} // namespace Poco
|
