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
|
% Licensed under the Apache License, Version 2.0 (the "License"); you may not
% use this file except in compliance with the License. You may obtain a copy of
% the License at
%
% http://www.apache.org/licenses/LICENSE-2.0
%
% Unless required by applicable law or agreed to in writing, software
% distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
% WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
% License for the specific language governing permissions and limitations under
% the License.
-module(aegis).
-include_lib("fabric/include/fabric2.hrl").
-define(WRAPPED_KEY, {?DB_AEGIS, 1}).
-export([
init_db/2,
open_db/1,
decrypt/2,
decrypt/3,
encrypt/3,
wrap_fold_fun/2
]).
init_db(#{} = Db, Options) ->
Db#{
is_encrypted => aegis_server:init_db(Db, Options)
}.
open_db(#{} = Db) ->
Db#{
is_encrypted => aegis_server:open_db(Db)
}.
encrypt(#{} = _Db, _Key, <<>>) ->
<<>>;
encrypt(#{is_encrypted := false}, _Key, Value) when is_binary(Value) ->
Value;
encrypt(#{is_encrypted := true} = Db, Key, Value)
when is_binary(Key), is_binary(Value) ->
aegis_server:encrypt(Db, Key, Value).
decrypt(#{} = Db, Rows) when is_list(Rows) ->
lists:map(fun({Key, Value}) ->
{Key, decrypt(Db, Key, Value)}
end, Rows).
decrypt(#{} = _Db, _Key, <<>>) ->
<<>>;
decrypt(#{is_encrypted := false}, _Key, Value) when is_binary(Value) ->
Value;
decrypt(#{is_encrypted := true} = Db, Key, Value)
when is_binary(Key), is_binary(Value) ->
aegis_server:decrypt(Db, Key, Value).
wrap_fold_fun(Db, Fun) when is_function(Fun, 2) ->
fun({Key, Value}, Acc) ->
Fun({Key, decrypt(Db, Key, Value)}, Acc)
end.
|