aboutsummaryrefslogtreecommitdiff
path: root/src/randoms.erl
blob: a64012ca34308f699e6d919fcbfd74560b1783bc (plain) (blame)
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
%%%----------------------------------------------------------------------
%%% File    : randoms.erl
%%% Author  : Alexey Shchepin <alexey@process-one.net>
%%% Purpose : Random generation number wrapper
%%% Created : 13 Dec 2002 by Alexey Shchepin <alexey@process-one.net>
%%%
%%%
%%% ejabberd, Copyright (C) 2002-2018   ProcessOne
%%%
%%% This program is free software; you can redistribute it and/or
%%% modify it under the terms of the GNU General Public License as
%%% published by the Free Software Foundation; either version 2 of the
%%% License, or (at your option) any later version.
%%%
%%% This program 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
%%% General Public License for more details.
%%%
%%% You should have received a copy of the GNU General Public License along
%%% with this program; if not, write to the Free Software Foundation, Inc.,
%%% 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
%%%
%%%----------------------------------------------------------------------

-module(randoms).

-author('alexey@process-one.net').

-export([get_string/0, uniform/0, uniform/1, uniform/2, bytes/1,
	 round_robin/1]).

-define(THRESHOLD, 16#10000000000000000).

-ifdef(RAND_UNIFORM).
get_string() ->
    R = rand:uniform(?THRESHOLD),
    integer_to_binary(R).

uniform() ->
    rand:uniform().

uniform(N) ->
    rand:uniform(N).

uniform(N, M) ->
    rand:uniform(M-N+1) + N-1.
-else.
get_string() ->
    R = crypto:rand_uniform(0, ?THRESHOLD),
    integer_to_binary(R).

uniform() ->
    crypto:rand_uniform(0, ?THRESHOLD)/?THRESHOLD.

uniform(N) ->
    crypto:rand_uniform(1, N+1).

uniform(N, M) ->
    crypto:rand_uniform(N, M+1).
-endif.

-ifdef(STRONG_RAND_BYTES).
bytes(N) ->
    crypto:strong_rand_bytes(N).
-else.
bytes(N) ->
    crypto:rand_bytes(N).
-endif.

-spec round_robin(pos_integer()) -> non_neg_integer().
round_robin(N) ->
    p1_time_compat:unique_integer([monotonic, positive]) rem N.