mirror of
https://github.com/processone/ejabberd.git
synced 2024-11-26 16:26:24 +01:00
More account support(Update/Info)
This commit is contained in:
parent
88365ed507
commit
ddb043aa71
232
src/mod_acme.erl
232
src/mod_acme.erl
@ -9,13 +9,13 @@
|
||||
, directory/2
|
||||
, new_nonce/2
|
||||
%% Account
|
||||
, new_account/2
|
||||
, new_reg/2
|
||||
, update_account/2
|
||||
, account_info/2 %% TODO: Maybe change to get_account
|
||||
, account_key_change/2
|
||||
, deactivate_account/2
|
||||
%% Orders/Certificates
|
||||
, new_order/2
|
||||
, new_cert/2
|
||||
, new_authz/2
|
||||
, get_certificate/2
|
||||
, get_authz/2
|
||||
@ -39,14 +39,18 @@
|
||||
% -define(CA_URL, "http://localhost:4000").
|
||||
|
||||
-define(DEFAULT_DIRECTORY, ?CA_URL ++ "/directory").
|
||||
|
||||
-define(DEFAULT_NEW_NONCE, ?CA_URL ++ "/acme/new_nonce").
|
||||
-define(DEFAULT_ACCOUNT, "2273801").
|
||||
|
||||
-define(DEFAULT_KEY_FILE, "private_key_temporary").
|
||||
-define(DEFAULT_TOS, <<"https://letsencrypt.org/documents/LE-SA-v1.1.1-August-1-2016.pdf">>).
|
||||
|
||||
-record(state, {
|
||||
ca_url = ?CA_URL :: list(),
|
||||
dir_url = ?DEFAULT_DIRECTORY :: list(),
|
||||
dirs = maps:new(),
|
||||
nonce = ""
|
||||
nonce = "",
|
||||
account = none
|
||||
}).
|
||||
|
||||
%% This will be initially just be filled with stub functions
|
||||
@ -64,14 +68,15 @@ directory(Pid, Options) ->
|
||||
new_nonce(Pid, Options) ->
|
||||
gen_server:call(Pid, ?FUNCTION_NAME).
|
||||
|
||||
new_account(Pid, Options) ->
|
||||
new_reg(Pid, Options) ->
|
||||
gen_server:call(Pid, ?FUNCTION_NAME).
|
||||
|
||||
update_account(Pid, Options) ->
|
||||
ok.
|
||||
update_account(Pid, AccountId) ->
|
||||
%% TODO: This has to have more info ofcourse
|
||||
gen_server:call(Pid, {?FUNCTION_NAME, AccountId}).
|
||||
|
||||
account_info(Pid, Options) ->
|
||||
ok.
|
||||
account_info(Pid, AccountId) ->
|
||||
gen_server:call(Pid, {?FUNCTION_NAME, AccountId}).
|
||||
|
||||
account_key_change(Pid, Options) ->
|
||||
ok.
|
||||
@ -79,11 +84,11 @@ account_key_change(Pid, Options) ->
|
||||
deactivate_account(Pid, Options) ->
|
||||
ok.
|
||||
|
||||
new_order(Pid, Options) ->
|
||||
ok.
|
||||
new_cert(Pid, Options) ->
|
||||
gen_server:call(Pid, ?FUNCTION_NAME).
|
||||
|
||||
new_authz(Pid, Options) ->
|
||||
ok.
|
||||
gen_server:call(Pid, ?FUNCTION_NAME).
|
||||
|
||||
get_certificate(Pid, Options) ->
|
||||
ok.
|
||||
@ -129,7 +134,7 @@ handle_call(directory, _From, S = #state{dir_url=Url, dirs=Dirs}) ->
|
||||
|
||||
% Find and save the replay nonce
|
||||
% io:format("Directory Head Response: ~p~n", [Head]),
|
||||
{"replay-nonce", Nonce} = proplists:lookup("replay-nonce", Head),
|
||||
Nonce = get_nonce(Head),
|
||||
|
||||
%% Update the directories in state
|
||||
%% TODO: Get the merge of the old and the new dictionary
|
||||
@ -143,30 +148,146 @@ handle_call(new_nonce, _From, S = #state{dirs=Dirs}) ->
|
||||
{ok, {Status, Head, []}} =
|
||||
httpc:request(head, {Url, []}, [], []),
|
||||
{reply, {ok, {Status, Head}}, S};
|
||||
handle_call(new_account, _From, S = #state{ca_url = Ca, dirs=Dirs, nonce = Nonce}) ->
|
||||
handle_call(new_reg, _From, S = #state{ca_url = Ca, dirs=Dirs, nonce = Nonce}) ->
|
||||
%% Get url from all directories
|
||||
#{"new-reg" := Url} = Dirs,
|
||||
|
||||
%% Make the request body
|
||||
ReqBody = jiffy:encode({
|
||||
[ { <<"contact">>,
|
||||
[
|
||||
<<"mailto:cert-admin@example.com">>
|
||||
]
|
||||
}
|
||||
[ { <<"contact">>, [<<"mailto:cert-admin@example.com">>]}
|
||||
, { <<"resource">>, <<"new-reg">>}
|
||||
]}),
|
||||
|
||||
%% Generate a key for the first time use
|
||||
Key = generate_key(),
|
||||
|
||||
%% Write the key to a file
|
||||
jose_jwk:to_file(?DEFAULT_KEY_FILE, Key),
|
||||
|
||||
%% Jose
|
||||
{_, SignedBody} = sign_a_json_object_using_jose(ReqBody, Url, Nonce),
|
||||
{_, SignedBody} = sign_a_json_object_using_jose(Key, ReqBody, Url, Nonce),
|
||||
io:format("Signed Body: ~p~n", [SignedBody]),
|
||||
|
||||
%% Encode the Signed body with jiffy
|
||||
FinalBody = jiffy:encode(SignedBody),
|
||||
|
||||
%% Post request
|
||||
{ok, {Status, Head, Body}} =
|
||||
httpc:request(post, {Url, [], "application/jose+json", FinalBody}, [], []),
|
||||
{reply, {ok, {Status, Head, Body}}, S};
|
||||
|
||||
%% Get and save the new nonce
|
||||
NewNonce = get_nonce(Head),
|
||||
|
||||
{reply, {ok, {Status, Head, Body}}, S#state{nonce=NewNonce}};
|
||||
handle_call({account_info, AccountId}, _From, S = #state{ca_url = Ca, dirs=Dirs, nonce = Nonce}) ->
|
||||
%% Get url from accountId
|
||||
Url = Ca ++ "/acme/reg/" ++ AccountId,
|
||||
|
||||
%% Make the request body
|
||||
ReqBody = jiffy:encode({[
|
||||
{ <<"resource">>, <<"reg">>}
|
||||
]}),
|
||||
|
||||
%% Get the key from a file
|
||||
Key = jose_jwk:from_file(?DEFAULT_KEY_FILE),
|
||||
|
||||
%% Jose
|
||||
{_, SignedBody} = sign_a_json_object_using_jose(Key, ReqBody, Url, Nonce),
|
||||
io:format("Signed Body: ~p~n", [SignedBody]),
|
||||
|
||||
%% Encode the Signed body with jiffy
|
||||
FinalBody = jiffy:encode(SignedBody),
|
||||
|
||||
%% Post request
|
||||
{ok, {Status, Head, Body}} =
|
||||
httpc:request(post, {Url, [], "application/jose+json", FinalBody}, [], []),
|
||||
|
||||
% Get and save the new nonce
|
||||
NewNonce = get_nonce(Head),
|
||||
|
||||
{reply, {ok, {Status, Head, Body}}, S#state{nonce=NewNonce}};
|
||||
handle_call({update_account, AccountId}, _From, S = #state{ca_url = Ca, dirs=Dirs, nonce = Nonce}) ->
|
||||
%% Get url from accountId
|
||||
Url = Ca ++ "/acme/reg/" ++ AccountId,
|
||||
|
||||
%% Make the request body
|
||||
ReqBody = jiffy:encode({[
|
||||
{ <<"resource">>, <<"reg">>},
|
||||
{ <<"agreement">>, ?DEFAULT_TOS}
|
||||
]}),
|
||||
|
||||
%% Get the key from a file
|
||||
Key = jose_jwk:from_file(?DEFAULT_KEY_FILE),
|
||||
|
||||
%% Jose
|
||||
{_, SignedBody} = sign_a_json_object_using_jose(Key, ReqBody, Url, Nonce),
|
||||
io:format("Signed Body: ~p~n", [SignedBody]),
|
||||
|
||||
%% Encode the Signed body with jiffy
|
||||
FinalBody = jiffy:encode(SignedBody),
|
||||
|
||||
%% Post request
|
||||
{ok, {Status, Head, Body}} =
|
||||
httpc:request(post, {Url, [], "application/jose+json", FinalBody}, [], []),
|
||||
|
||||
% Get and save the new nonce
|
||||
NewNonce = get_nonce(Head),
|
||||
|
||||
{reply, {ok, {Status, Head, Body}}, S#state{nonce=NewNonce}};
|
||||
handle_call(new_cert, _From, S = #state{ca_url = Ca, dirs=Dirs, nonce = Nonce}) ->
|
||||
%% Get url from all directories
|
||||
#{"new-cert" := Url} = Dirs,
|
||||
|
||||
%% Make the request body
|
||||
ReqBody = jiffy:encode({[
|
||||
{ <<"resource">>, <<"new-cert">>}
|
||||
]}),
|
||||
|
||||
%% Get the key from a file
|
||||
Key = jose_jwk:from_file(?DEFAULT_KEY_FILE),
|
||||
|
||||
%% Jose
|
||||
{_, SignedBody} = sign_a_json_object_using_jose(Key, ReqBody, Url, Nonce),
|
||||
io:format("Signed Body: ~p~n", [SignedBody]),
|
||||
|
||||
%% Encode the Signed body with jiffy
|
||||
FinalBody = jiffy:encode(SignedBody),
|
||||
|
||||
%% Post request
|
||||
{ok, {Status, Head, Body}} =
|
||||
httpc:request(post, {Url, [], "application/jose+json", FinalBody}, [], []),
|
||||
|
||||
% Get and save the new nonce
|
||||
NewNonce = get_nonce(Head),
|
||||
|
||||
{reply, {ok, {Status, Head, Body}}, S#state{nonce=NewNonce}};
|
||||
handle_call(new_authz, _From, S = #state{ca_url = Ca, dirs=Dirs, nonce = Nonce}) ->
|
||||
%% Get url from all directories
|
||||
#{"new-authz" := Url} = Dirs,
|
||||
|
||||
%% Make the request body
|
||||
ReqBody = jiffy:encode({[
|
||||
{ <<"resource">>, <<"new-authz">>}
|
||||
]}),
|
||||
|
||||
%% Get the key from a file
|
||||
Key = jose_jwk:from_file(?DEFAULT_KEY_FILE),
|
||||
|
||||
%% Jose
|
||||
{_, SignedBody} = sign_a_json_object_using_jose(Key, ReqBody, Url, Nonce),
|
||||
io:format("Signed Body: ~p~n", [SignedBody]),
|
||||
|
||||
%% Encode the Signed body with jiffy
|
||||
FinalBody = jiffy:encode(SignedBody),
|
||||
|
||||
%% Post request
|
||||
{ok, {Status, Head, Body}} =
|
||||
httpc:request(post, {Url, [], "application/jose+json", FinalBody}, [], []),
|
||||
|
||||
% Get and save the new nonce
|
||||
NewNonce = get_nonce(Head),
|
||||
|
||||
{reply, {ok, {Status, Head, Body}}, S#state{nonce=NewNonce}};
|
||||
handle_call(stop, _From, State) ->
|
||||
{stop, normal, ok, State}.
|
||||
|
||||
@ -189,20 +310,26 @@ final_url(Urls) ->
|
||||
Joined = lists:join("/", Urls),
|
||||
lists:flatten(Joined).
|
||||
|
||||
get_nonce(Head) ->
|
||||
{"replay-nonce", Nonce} = proplists:lookup("replay-nonce", Head),
|
||||
Nonce.
|
||||
|
||||
|
||||
%% Test
|
||||
|
||||
sign_a_json_object_using_jose(Json, Url, Nonce) ->
|
||||
generate_key() ->
|
||||
% Generate a key for now
|
||||
Key = jose_jwk:generate_key({ec, secp256r1}),
|
||||
io:format("Key: ~p~n", [Key]),
|
||||
|
||||
Key.
|
||||
|
||||
sign_a_json_object_using_jose(Key, Json, Url, Nonce) ->
|
||||
% Generate a public key
|
||||
PubKey = jose_jwk:to_public(Key),
|
||||
io:format("Public Key: ~p~n", [PubKey]),
|
||||
% io:format("Public Key: ~p~n", [PubKey]),
|
||||
{_, BinaryPubKey} = jose_jwk:to_binary(PubKey),
|
||||
io:format("Public Key: ~p~n", [BinaryPubKey]),
|
||||
% io:format("Public Key: ~p~n", [BinaryPubKey]),
|
||||
PubKeyJson = jiffy:decode(BinaryPubKey),
|
||||
io:format("Public Key: ~p~n", [PubKeyJson]),
|
||||
|
||||
@ -218,31 +345,14 @@ sign_a_json_object_using_jose(Json, Url, Nonce) ->
|
||||
%% Im not sure if it is needed
|
||||
% , <<"b64">> => true
|
||||
, <<"jwk">> => PubKeyJson
|
||||
% , <<"url">> => list_to_bitstring(Url)
|
||||
, <<"nonce">> => list_to_bitstring(Nonce)
|
||||
}),
|
||||
io:format("Jws: ~p~n", [JwsObj]),
|
||||
% io:format("Jws: ~p~n", [JwsObj]),
|
||||
|
||||
% ProtectedObj = jose_jws:signing_input(Json,
|
||||
% #{ <<"alg">> => <<"HS256">>
|
||||
% %% Im not sure if it is needed
|
||||
% , <<"jwk">> => PubKeyJson
|
||||
% , <<"url">> => Url
|
||||
% , <<"nonce">> => Nonce
|
||||
% }, JwsObj),
|
||||
% io:format("ProtectedObj: ~p~n", [ProtectedObj]),
|
||||
|
||||
% {Modules, ProtectedBinary} = to_binary(JwsObj),
|
||||
% io:format("ProtectedObj: ~p~n", [ProtectedObj]),
|
||||
% Protected = base64url:encode(ProtectedBinary),
|
||||
% Payload = base64url:encode(PlainText),
|
||||
% SigningInput = signing_input(PlainText, Protected, NewJWS),
|
||||
% Signature = base64url:encode(ALGModule:sign(Key, SigningInput, NewALG)),
|
||||
% {Modules, maps:put(<<"payload">>, Payload, signature_to_map(Protected, Header, Key, Signature))};
|
||||
|
||||
%% Signed Message
|
||||
Signed = jose_jws:sign(Key, Json, JwsObj),
|
||||
io:format("Signed: ~p~n", [Signed]),
|
||||
% io:format("Signed: ~p~n", [Signed]),
|
||||
|
||||
%% Peek protected
|
||||
Protected = jose_jws:peek_protected(Signed),
|
||||
@ -253,15 +363,8 @@ sign_a_json_object_using_jose(Json, Url, Nonce) ->
|
||||
io:format("Payload: ~p~n", [jiffy:decode(Payload)]),
|
||||
|
||||
%% Verify
|
||||
io:format("Verify: ~p~n", [jose_jws:verify(Key, Signed)]),
|
||||
|
||||
% %% To binary
|
||||
% Binary = jose_jws:to_binary(Signed),
|
||||
% io:format("Binary: ~p~n", [jose_jws:to_binary(Signed)]),
|
||||
|
||||
% %% To map
|
||||
% Map = jose_jws:to_map(Signed),
|
||||
% io:format("Map: ~p~n", [jose_jws:to_map(Signed)]),
|
||||
% {true, _} = jose_jws:verify(Key, Signed),
|
||||
% io:format("Verify: ~p~n", [jose_jws:verify(Key, Signed)]),
|
||||
|
||||
Signed.
|
||||
|
||||
@ -272,7 +375,28 @@ scenario() ->
|
||||
{ok, Result} = directory(Pid, []),
|
||||
io:format("Directory result: ~p~n", [Result]),
|
||||
|
||||
{ok, Result1} = new_account(Pid, []),
|
||||
io:format("New account result: ~p~n", [Result1]),
|
||||
% %% Request the creation of a new account
|
||||
% {ok, {Status, Head, Body}} = new_reg(Pid, []),
|
||||
% io:format("New account~nHead: ~p~nBody: ~p~n", [{Status, Head}, jiffy:decode(Body)]),
|
||||
|
||||
%% Get the info of an existing account
|
||||
% {ok, {Status1, Head1, Body1}} = account_info(Pid, ?DEFAULT_ACCOUNT),
|
||||
% io:format("Account: ~p~nHead: ~p~nBody: ~p~n",
|
||||
% [?DEFAULT_ACCOUNT, {Status1, Head1}, jiffy:decode(Body1)]),
|
||||
|
||||
%% Update the account to agree to terms and services
|
||||
{ok, {Status1, Head1, Body1}} = update_account(Pid, ?DEFAULT_ACCOUNT),
|
||||
io:format("Account: ~p~nHead: ~p~nBody: ~p~n",
|
||||
[?DEFAULT_ACCOUNT, {Status1, Head1}, jiffy:decode(Body1)]),
|
||||
|
||||
%% New certification
|
||||
% {ok, {Status2, Head2, Body2}} = new_cert(Pid, []),
|
||||
% io:format("New Cert~nHead: ~p~nBody: ~p~n",
|
||||
% [{Status2, Head2}, jiffy:decode(Body2)]),
|
||||
|
||||
%% New authorization
|
||||
{ok, {Status2, Head2, Body2}} = new_authz(Pid, []),
|
||||
io:format("New Authz~nHead: ~p~nBody: ~p~n",
|
||||
[{Status2, Head2}, jiffy:decode(Body2)]),
|
||||
ok.
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user