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
|
/*************************************************
* Unix EntropySource Source File *
* (C) 1999-2008 Jack Lloyd *
*************************************************/
#include <botan/es_unix.h>
#include <botan/unix_cmd.h>
#include <botan/parsing.h>
#include <algorithm>
#include <sys/time.h>
#include <sys/stat.h>
#include <sys/resource.h>
#include <unistd.h>
namespace Botan {
namespace {
/*************************************************
* Sort ordering by priority *
*************************************************/
bool Unix_Program_Cmp(const Unix_Program& a, const Unix_Program& b)
{ return (a.priority < b.priority); }
}
/*************************************************
* Unix_EntropySource Constructor *
*************************************************/
Unix_EntropySource::Unix_EntropySource(const std::vector<std::string>& path) :
PATH(path)
{
add_default_sources(sources);
}
/*************************************************
* Add sources to the list *
*************************************************/
void Unix_EntropySource::add_sources(const Unix_Program srcs[], u32bit count)
{
sources.insert(sources.end(), srcs, srcs + count);
std::sort(sources.begin(), sources.end(), Unix_Program_Cmp);
}
/*************************************************
* Unix Fast Poll *
*************************************************/
void Unix_EntropySource::do_fast_poll()
{
const char* STAT_TARGETS[] = { "/", "/tmp", ".", "..", 0 };
for(u32bit j = 0; STAT_TARGETS[j]; j++)
{
struct stat statbuf;
clear_mem(&statbuf, 1);
::stat(STAT_TARGETS[j], &statbuf);
add_bytes(&statbuf, sizeof(statbuf));
}
add_bytes(::getpid());
add_bytes(::getppid());
add_bytes(::getuid());
add_bytes(::getgid());
add_bytes(::geteuid());
add_bytes(::getegid());
add_bytes(::getpgrp());
add_bytes(::getsid(0));
struct ::rusage usage;
clear_mem(&usage, 1);
::getrusage(RUSAGE_SELF, &usage);
add_bytes(&usage, sizeof(usage));
::getrusage(RUSAGE_CHILDREN, &usage);
add_bytes(&usage, sizeof(usage));
}
/*************************************************
* Unix Slow Poll *
*************************************************/
void Unix_EntropySource::do_slow_poll()
{
const u32bit TRY_TO_GET = 16 * 1024;
const u32bit MINIMAL_WORKING = 32;
u32bit got = 0;
for(u32bit j = 0; j != sources.size(); j++)
{
DataSource_Command pipe(sources[j].name_and_args, PATH);
SecureVector<byte> buffer(DEFAULT_BUFFERSIZE);
u32bit got_from_src = 0;
while(!pipe.end_of_data())
{
u32bit this_loop = pipe.read(buffer, buffer.size());
add_bytes(buffer, this_loop);
got_from_src += this_loop;
}
sources[j].working = (got_from_src >= MINIMAL_WORKING) ? true : false;
got += got_from_src;
if(got >= TRY_TO_GET)
break;
}
}
}
|
