umgeher/utools
0
Fork 0
Code Issues Pull requests Activity

new fns: hex_encode and sha256

Browse source
This commit is contained in:
Umgeher Torgersen 2023-01-08 16:52:51 +00:00
parent 831c37a26b
commit ceafc8ccb7
1 changed files with 45 additions and 33 deletions
Download patch file Download diff file Expand all files Collapse all files

78
src/utools.erl
View file

@ -2,66 +2,75 @@
-export([b32_decode/1]). -export([b32_decode/1]).
-export([b32_encode/1]). -export([b32_encode/1]).
-export([hex_encode/1]).
-export([hotp/2]). -export([hotp/2]).
-export([rand_bytes/0, rand_bytes/1]). -export([rand_bytes/0, rand_bytes/1]).
-export([rand_chars/1]). -export([rand_chars/1]).
-export([rand_hash/0]). -export([rand_hash/0]).
-export([sha256/1]).
-export([totp/1, totp/2]). -export([totp/1, totp/2]).
-export([totp_check/2]). -export([totp_check/2]).
-export([totp_generate/0]). -export([totp_generate/0]).
b32_decode({<<V, "======">>, Bits}) -> b32_decode({<<V, "======">>, Bits}) ->
<<Bits/bits, (b32_std_dec(V) bsr 2):3>>; <<Bits/bits, (b32_std_dec(V) bsr 2):3>>;
b32_decode({<<V, "====">>, Bits}) -> b32_decode({<<V, "====">>, Bits}) ->
<<Bits/bits, (b32_std_dec(V) bsr 4):1>>; <<Bits/bits, (b32_std_dec(V) bsr 4):1>>;
b32_decode({<<V, "===">>, Bits}) -> b32_decode({<<V, "===">>, Bits}) ->
<<Bits/bits, (b32_std_dec(V) bsr 1):4>>; <<Bits/bits, (b32_std_dec(V) bsr 1):4>>;
b32_decode({<<V, "=">>, Bits}) -> b32_decode({<<V, "=">>, Bits}) ->
<<Bits/bits, (b32_std_dec(V) bsr 3):2>>; <<Bits/bits, (b32_std_dec(V) bsr 3):2>>;
b32_decode({<<V, R/binary>>, Bits}) -> b32_decode({<<V, R/binary>>, Bits}) ->
b32_decode({R, <<Bits/bits, (b32_std_dec(V)):5>>}); b32_decode({R, <<Bits/bits, (b32_std_dec(V)):5>>});
b32_decode({<<>>, Bits}) -> b32_decode({<<>>, Bits}) ->
Bits; Bits;
b32_decode(V) when is_list(V) -> b32_decode(V) when is_list(V) ->
b32_decode(list_to_binary(V)); b32_decode(list_to_binary(V));
b32_decode(V) when is_binary(V) -> b32_decode(V) when is_binary(V) ->
b32_decode({V, <<>>}). b32_decode({V, <<>>}).
b32_encode({body, V}) -> b32_encode({body, V}) ->
O = 5 * (byte_size(V) div 5), O = 5 * (byte_size(V) div 5),
<<Body:O/binary, R/binary>> = V, <<Body:O/binary, R/binary>> = V,
{<< <<(b32_std_enc(X))>> || <<X:5>> <= Body>>, R}; {<< <<(b32_std_enc(X))>> || <<X:5>> <= Body>>, R};
b32_encode({rest, V}) -> b32_encode({rest, V}) ->
O = 5 * (bit_size(V) div 5), O = 5 * (bit_size(V) div 5),
<<Body:O/bits, Rest/bits>> = V, <<Body:O/bits, Rest/bits>> = V,
B0 = << <<(b32_std_enc(X))>> || <<X:5>> <= Body>>, B0 = << <<(b32_std_enc(X))>> || <<X:5>> <= Body>>,
{B1, Pad} = case Rest of {B1, Pad} = case Rest of
<<I:3>> -> {<<(b32_std_enc(I bsl 2))>>, 6}; <<I:3>> -> {<<(b32_std_enc(I bsl 2))>>, 6};
<<I:1>> -> {<<(b32_std_enc(I bsl 4))>>, 4}; <<I:1>> -> {<<(b32_std_enc(I bsl 4))>>, 4};
<<I:4>> -> {<<(b32_std_enc(I bsl 1))>>, 3}; <<I:4>> -> {<<(b32_std_enc(I bsl 1))>>, 3};
<<I:2>> -> {<<(b32_std_enc(I bsl 3))>>, 1}; <<I:2>> -> {<<(b32_std_enc(I bsl 3))>>, 1};
<<>> -> {<<>>, 0} <<>> -> {<<>>, 0}
end, end,
{<<B0/binary, B1/binary>>, Pad}; {<<B0/binary, B1/binary>>, Pad};
b32_encode(V) when is_list(V) -> b32_encode(V) when is_list(V) ->
b32_encode(list_to_binary(V)); b32_encode(list_to_binary(V));
b32_encode(V) when is_binary(V) -> b32_encode(V) when is_binary(V) ->
{EB, _R} = b32_encode({body, V}), {EB, _R} = b32_encode({body, V}),
{ER, P} = b32_encode({rest, V}), {ER, P} = b32_encode({rest, V}),
Padding = list_to_binary(lists:duplicate(P, $=)), Padding = list_to_binary(lists:duplicate(P, $=)),
<<EB/binary, ER/binary, Padding/binary>>. <<EB/binary, ER/binary, Padding/binary>>.
b32_std_dec(I) when I >= $2 andalso I =< $7 -> b32_std_dec(I) when I >= $2 andalso I =< $7 ->
I - 24; I - 24;
b32_std_dec(I) when I >= $a andalso I =< $z -> b32_std_dec(I) when I >= $a andalso I =< $z ->
I - $a; I - $a;
b32_std_dec(I) when I >= $A andalso I =< $Z -> b32_std_dec(I) when I >= $A andalso I =< $Z ->
I - $A. I - $A.
b32_std_enc(I) when is_integer(I) andalso I >= 26 andalso I =< 31 -> b32_std_enc(I) when is_integer(I) andalso I >= 26 andalso I =< 31 ->
I + 24; I + 24;
b32_std_enc(I) when is_integer(I) andalso I >= 0 andalso I =< 25 -> b32_std_enc(I) when is_integer(I) andalso I >= 0 andalso I =< 25 ->
I + $a. I + $a.
hex_encode(<<X:128/big-unsigned-integer>>) ->
binary:list_to_bin(lists:flatten(io_lib:format("~32.16.0b", [X])));
hex_encode(<<X:160/big-unsigned-integer>>) ->
binary:list_to_bin(lists:flatten(io_lib:format("~40.16.0b", [X])));
hex_encode(<<X:256/big-unsigned-integer>>) ->
binary:list_to_bin(lists:flatten(io_lib:format("~64.16.0b", [X]))).
hotp(Token, Time) -> hotp(Token, Time) ->
K = b32_decode(Token), K = b32_decode(Token),
@ -92,7 +101,10 @@ rand_chars(Size) when Size > 0 ->
rand_hash() -> rand_hash() ->
{ok, R} = rand_chars(64), {ok, R} = rand_chars(64),
{ok, binary:encode_hex(crypto:hash(sha256, <<"utools:rand_hash(", R/binary, ")">>))}. {ok, sha256(<<"utools:rand_hash(", R/binary, ")">>)}.
sha256(<<Data/binary>>) ->
hex_encode(crypto:hash(sha256, Data)).
totp(<<Secret/binary>>) -> totp(<<Secret/binary>>) ->
totp(Secret, erlang:timestamp()). totp(Secret, erlang:timestamp()).