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
|
%%%----------------------------------------------------------------------
%%% File : sha.erl
%%% Author : Alexey Shchepin <alexey@process-one.net>
%%% Purpose :
%%% Created : 20 Dec 2002 by Alexey Shchepin <alexey@process-one.net>
%%%
%%%
%%% ejabberd, Copyright (C) 2002-2013 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., 59 Temple Place, Suite 330, Boston, MA
%%% 02111-1307 USA
%%%
%%%----------------------------------------------------------------------
-module(sha).
-author('alexey@process-one.net').
-export([start/0, sha/1, sha1/1, sha224/1, sha256/1,
sha384/1, sha512/1, to_hexlist/1]).
-ifdef(HAVE_MD2).
-export([md2/1]).
-endif.
-include("ejabberd.hrl").
-define(DRIVER, sha_drv).
start() ->
crypto:start(),
Res = case erl_ddll:load_driver(ejabberd:get_so_path(),
?DRIVER)
of
ok -> ok;
{error, already_loaded} -> ok;
Err -> Err
end,
case Res of
ok ->
Port = open_port({spawn, atom_to_list(?DRIVER)},
[binary]),
register(?DRIVER, Port);
{error, Reason} ->
?CRITICAL_MSG("unable to load driver '~s': ~s",
[driver_path(), erl_ddll:format_error(Reason)])
end.
digit_to_xchar(D) when (D >= 0) and (D < 10) -> D + 48;
digit_to_xchar(D) -> D + 87.
-spec sha(binary()) -> binary().
sha(Text) ->
Bin = crypto:sha(Text),
to_hexlist(Bin).
-spec to_hexlist(binary()) -> binary().
to_hexlist(Bin) ->
iolist_to_binary(lists:reverse(ints_to_rxstr(binary_to_list(Bin), []))).
ints_to_rxstr([], Res) -> Res;
ints_to_rxstr([N | Ns], Res) ->
ints_to_rxstr(Ns,
[digit_to_xchar(N rem 16), digit_to_xchar(N div 16)
| Res]).
-spec sha1(binary()) -> binary().
-spec sha224(binary()) -> binary().
-spec sha256(binary()) -> binary().
-spec sha384(binary()) -> binary().
-spec sha512(binary()) -> binary().
sha1(Text) -> crypto:sha(Text).
sha224(Text) -> erlang:port_control(?DRIVER, 224, Text).
sha256(Text) -> erlang:port_control(?DRIVER, 256, Text).
sha384(Text) -> erlang:port_control(?DRIVER, 384, Text).
sha512(Text) -> erlang:port_control(?DRIVER, 512, Text).
-ifdef(HAVE_MD2).
-spec md2(binary()) -> binary().
md2(Text) -> erlang:port_control(?DRIVER, 2, Text).
-endif.
driver_path() ->
Suffix = case os:type() of
{win32, _} -> ".dll";
_ -> ".so"
end,
filename:join(ejabberd:get_so_path(),
atom_to_list(?DRIVER) ++ Suffix).
|
