2002-12-13 21:58:27 +01:00
|
|
|
%%%----------------------------------------------------------------------
|
|
|
|
%%% File : randoms.erl
|
2007-12-24 13:58:05 +01:00
|
|
|
%%% Author : Alexey Shchepin <alexey@process-one.net>
|
|
|
|
%%% Purpose : Random generation number wrapper
|
|
|
|
%%% Created : 13 Dec 2002 by Alexey Shchepin <alexey@process-one.net>
|
|
|
|
%%%
|
|
|
|
%%%
|
2018-01-05 21:18:58 +01:00
|
|
|
%%% ejabberd, Copyright (C) 2002-2018 ProcessOne
|
2007-12-24 13:58:05 +01:00
|
|
|
%%%
|
|
|
|
%%% 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.
|
2009-01-12 15:44:42 +01:00
|
|
|
%%%
|
2014-02-22 11:27:40 +01:00
|
|
|
%%% 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.
|
2007-12-24 13:58:05 +01:00
|
|
|
%%%
|
2002-12-13 21:58:27 +01:00
|
|
|
%%%----------------------------------------------------------------------
|
|
|
|
|
|
|
|
-module(randoms).
|
2013-03-14 10:33:02 +01:00
|
|
|
|
2007-12-24 13:58:05 +01:00
|
|
|
-author('alexey@process-one.net').
|
2002-12-13 21:58:27 +01:00
|
|
|
|
2017-04-23 10:54:56 +02:00
|
|
|
-export([get_string/0, uniform/0, uniform/1, uniform/2, bytes/1,
|
2018-02-23 18:23:54 +01:00
|
|
|
round_robin/1, get_alphanum_string/1]).
|
2002-12-13 21:58:27 +01:00
|
|
|
|
2016-10-18 07:17:21 +02:00
|
|
|
-define(THRESHOLD, 16#10000000000000000).
|
|
|
|
|
2017-08-17 18:32:15 +02:00
|
|
|
-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.
|
2002-12-13 21:58:27 +01:00
|
|
|
get_string() ->
|
2016-10-18 07:17:21 +02:00
|
|
|
R = crypto:rand_uniform(0, ?THRESHOLD),
|
2016-11-13 08:44:53 +01:00
|
|
|
integer_to_binary(R).
|
2016-01-04 11:55:25 +01:00
|
|
|
|
2016-10-18 07:17:21 +02:00
|
|
|
uniform() ->
|
|
|
|
crypto:rand_uniform(0, ?THRESHOLD)/?THRESHOLD.
|
|
|
|
|
|
|
|
uniform(N) ->
|
2016-10-18 07:35:47 +02:00
|
|
|
crypto:rand_uniform(1, N+1).
|
2016-10-18 07:17:21 +02:00
|
|
|
|
2017-04-14 12:57:52 +02:00
|
|
|
uniform(N, M) ->
|
|
|
|
crypto:rand_uniform(N, M+1).
|
2017-08-17 18:32:15 +02:00
|
|
|
-endif.
|
2017-04-14 12:57:52 +02:00
|
|
|
|
2016-10-18 07:17:21 +02:00
|
|
|
-ifdef(STRONG_RAND_BYTES).
|
|
|
|
bytes(N) ->
|
|
|
|
crypto:strong_rand_bytes(N).
|
|
|
|
-else.
|
|
|
|
bytes(N) ->
|
|
|
|
crypto:rand_bytes(N).
|
|
|
|
-endif.
|
2017-04-23 10:54:56 +02:00
|
|
|
|
|
|
|
-spec round_robin(pos_integer()) -> non_neg_integer().
|
|
|
|
round_robin(N) ->
|
2017-04-24 23:51:01 +02:00
|
|
|
p1_time_compat:unique_integer([monotonic, positive]) rem N.
|
2018-02-23 18:23:54 +01:00
|
|
|
|
|
|
|
-spec get_alphanum_string(non_neg_integer()) -> binary().
|
|
|
|
get_alphanum_string(Length) ->
|
|
|
|
list_to_binary(get_alphanum_string([], Length)).
|
|
|
|
|
|
|
|
-spec get_alphanum_string(string(), non_neg_integer()) -> string().
|
|
|
|
get_alphanum_string(S, 0) -> S;
|
|
|
|
get_alphanum_string(S, N) ->
|
|
|
|
get_alphanum_string([make_rand_char() | S], N - 1).
|
|
|
|
|
|
|
|
-spec make_rand_char() -> char().
|
|
|
|
make_rand_char() ->
|
|
|
|
map_int_to_char(uniform(0, 61)).
|
|
|
|
|
|
|
|
-spec map_int_to_char(0..61) -> char().
|
|
|
|
map_int_to_char(N) when N =< 9 -> N + 48; % Digit.
|
|
|
|
map_int_to_char(N) when N =< 35 -> N + 55; % Upper-case character.
|
|
|
|
map_int_to_char(N) when N =< 61 -> N + 61. % Lower-case character.
|