add data_file.ml

This commit is contained in:
swrup 2025-11-29 14:03:37 +01:00
parent 760e6de1fb
commit bc91f7357c
8 changed files with 207 additions and 101 deletions

View file

@ -175,17 +175,6 @@ module Coin = struct
} }
let all_coins = List.map parse_coin coin_sections let all_coins = List.map parse_coin coin_sections
let get_coin name =
match List.find_opt (fun v -> v.section_name = name) all_coins with
| None ->
fail_with
(Fmt.str "section `[coin_%s]` not found, coin `%s` is not defined"
name name)
| Some v -> v
let kudo_1 = get_coin "kudo_1"
let kudo_2 = get_coin "kudo_2"
end end
module Exchange_secmod_rsa = struct module Exchange_secmod_rsa = struct

View file

@ -165,6 +165,27 @@ module RsaPublicKey = struct
let jsont = Jsont.of_of_string ~kind:"RsaPublicKey" of_b32 ~enc:to_b32 let jsont = Jsont.of_of_string ~kind:"RsaPublicKey" of_b32 ~enc:to_b32
end end
module RsaPrivateKey = struct
open Mirage_crypto_pk.Rsa
type t = priv
let pub_of_priv = pub_of_priv
(* TODO rsa *)
let to_octets _t : string = assert false
let of_octets _t : t = assert false
(*let bin = Bin.map (Bin.bytes 32) of_octets to_octets*)
let of_b32 s =
let open Syntax in
let* octets = B32.decode s in
Ok (of_octets octets)
let to_b32 t = B32.encode (to_octets t)
let jsont = Jsont.of_of_string ~kind:"EddsaPrivateKey" of_b32 ~enc:to_b32
end
module RsaSignature : sig module RsaSignature : sig
type t type t

93
src/data_file.ml Normal file
View file

@ -0,0 +1,93 @@
open Bos.OS
open Syntax
open Crypto
let read fname =
let* b = File.exists fname in
match b with
| false -> Ok None
| true ->
let+ content = File.read fname in
Some content
let read_eddsa fname =
let+ content_opt = read fname in
Option.map EddsaPrivateKey.of_octets content_opt
let read_rsa fname =
let+ content_opt = read fname in
Option.map RsaPrivateKey.of_octets content_opt
let write_eddsa fname priv = EddsaPrivateKey.to_octets priv |> File.write fname
let write_rsa fname priv = RsaPrivateKey.to_octets priv |> File.write fname
let load_signkey conn fname =
let* opt = read_eddsa fname in
match opt with
| None -> Ok None
| Some priv -> (
let pub = EddsaPrivateKey.pub_of_priv priv in
let* opt = Pg.lookup_signing_key conn pub in
match opt with
| None ->
Fmt.error_msg
"load_signkey error no associated metadata found in database for \
signkey `%s`."
(Fpath.to_string fname)
| Some (stamp_start, stamp_expire, stamp_end) ->
(* TODO master_sig *)
let master_sig = None in
let v =
Signkey.
{ pub; priv; stamp_start; stamp_expire; stamp_end; master_sig }
in
Ok (Some v))
let load_denom conn ~section_name fname =
let* opt = read_rsa fname in
match opt with
| None -> Ok None
| Some priv -> (
let pub = RsaPrivateKey.pub_of_priv priv in
let h_pub = Bin_type.DenominationHash.hash (RsaPublicKey.to_octets pub) in
let* opt = Pg.lookup_denomination_key conn h_pub in
match opt with
| None ->
Fmt.error_msg
"load_denom error no associated metadata found in database for \
denom `%s`."
(Fpath.to_string fname)
| Some
( stamp_start,
stamp_expire_withdraw,
stamp_expire_deposit,
stamp_expire_legal,
value,
fee_withdraw,
fee_deposit,
fee_refresh,
fee_refund,
age_mask ) ->
(* TODO master_sig *)
let master_sig = None in
let v =
Denomination.
{
pub;
priv;
section_name;
value;
stamp_start;
stamp_expire_withdraw;
stamp_expire_deposit;
stamp_expire_legal;
fee_withdraw;
fee_deposit;
fee_refresh;
fee_refund;
age_mask;
h_pub;
master_sig;
}
in
Ok (Some v))

View file

@ -2,6 +2,7 @@ open Crypto
type t = { type t = {
pub: RsaPublicKey.t; pub: RsaPublicKey.t;
priv: RsaPrivateKey.t;
section_name: string; section_name: string;
value: Amount.t; value: Amount.t;
stamp_start: Timestamp.t; stamp_start: Timestamp.t;
@ -14,7 +15,6 @@ type t = {
fee_refund: Amount.t; fee_refund: Amount.t;
age_mask: int; age_mask: int;
h_pub: Bin_type.DenominationHash.t; h_pub: Bin_type.DenominationHash.t;
sign: string -> RsaSignature.t;
master_sig: EddsaSignature.t option; master_sig: EddsaSignature.t option;
} }
@ -51,10 +51,10 @@ let make
let priv = generate ~bits:rsa_keysize () in let priv = generate ~bits:rsa_keysize () in
let pub = pub_of_priv priv in let pub = pub_of_priv priv in
let h_pub = Bin_type.DenominationHash.hash (RsaPublicKey.to_octets pub) in let h_pub = Bin_type.DenominationHash.hash (RsaPublicKey.to_octets pub) in
let sign = RsaSignature.sign ~key:priv in
let master_sig = None in let master_sig = None in
{ {
pub; pub;
priv;
section_name; section_name;
value; value;
stamp_start; stamp_start;
@ -67,6 +67,5 @@ let make
fee_refund; fee_refund;
age_mask= 0; age_mask= 0;
h_pub; h_pub;
sign;
master_sig; master_sig;
} }

View file

@ -1,3 +1,5 @@
open Syntax
type env = { type env = {
caqti_switch: Caqti_miou.Switch.t; caqti_switch: Caqti_miou.Switch.t;
db_uri: Uri.t; db_uri: Uri.t;
@ -16,9 +18,11 @@ let db_connection : (env, Caqti_miou.connection) Vif.Device.device =
Fmt.failwith "Database preflight failure: %a." Caqti_error.pp err Fmt.failwith "Database preflight failure: %a." Caqti_error.pp err
| Ok () -> conn) | Ok () -> conn)
(* TODO KV store *)
module Secmod_signkey = struct module Secmod_signkey = struct
(* TODO
- key rotation
- how many signkey to use?
we just use 1 for now *)
type t = { type t = {
sm_key: Signkey.t; sm_key: Signkey.t;
keys: Signkey.t list; keys: Signkey.t list;
@ -26,83 +30,34 @@ module Secmod_signkey = struct
let dir = Fpath.(v "data" / "secmod_signkey") let dir = Fpath.(v "data" / "secmod_signkey")
let read_eddsa_key_files fname = let store_secmod_data t =
let open Syntax in let* () = Data_file.write_eddsa Fpath.(dir / "sm_key") t.sm_key.priv in
let open Bos.OS in t.keys
let open Crypto in |> List.mapi (fun i key ->
let pub = Fpath.(dir / fname) |> Fpath.set_ext "pub" in let fname = Fpath.(dir / string_of_int i) in
let priv = Fpath.(dir / fname) in (fname, key.Signkey.priv))
let* b0 = File.exists pub in |> list_iter (fun (fname, key) -> Data_file.write_eddsa fname key)
let* b1 = File.exists priv in
match (b0, b1) with
| true, true ->
let* pub = File.read pub in
let* priv = File.read priv in
let pub = EddsaPublicKey.of_octets pub in
let priv = EddsaPrivateKey.of_octets priv in
Ok (Some (pub, priv))
| false, false -> Ok None
| _, _ -> Error (`Msg "secmod_signkey: invalid local file state.")
let write_eddsa_key_files fname signkey = let load conn =
let open Syntax in let error_invalid_state =
let open Bos.OS in Fmt.error_msg "secmod_signkey load error: invalid store state."
let open Crypto in
let pub_file = Fpath.(dir / fname) |> Fpath.set_ext "pub" in
let priv_file = Fpath.(dir / fname) in
let pub = EddsaPublicKey.to_octets signkey.Signkey.pub in
let priv = EddsaPrivateKey.to_octets signkey.priv in
let* () = File.write pub_file pub in
let* () = File.write priv_file priv in
Ok ()
let write_local_keys t =
let open Syntax in
let* () = write_eddsa_key_files "eddsa_sm_key" t.sm_key in
let keys =
List.mapi (fun i key -> ("eddsa_" ^ string_of_int i, key)) t.keys
in in
let* () = list_iter (fun (s, key) -> write_eddsa_key_files s key) keys in let* sm_key = Data_file.load_signkey conn Fpath.(dir / "sm_key") in
Ok () let* keys =
let l = List.init 1 (fun i -> Fpath.(dir / string_of_int i)) in
let load_local_keys conn = let* l =
let open Syntax in Syntax.list_map (fun fname -> Data_file.load_signkey conn fname) l
let load s = in
let* opt = read_eddsa_key_files s in match Syntax.opt_list l with
match opt with | Error () -> error_invalid_state
| None -> Ok None | Ok opt -> Ok opt
| Some (pub, priv) -> (
let* opt = Pg.lookup_signing_key conn pub in
match opt with
| None ->
Fmt.error_msg
"secmod_signkey error: no associtaed metadata found in \
database for key `%s`."
s
| Some (stamp_start, stamp_expire, stamp_end) ->
let master_sig = None in
Ok
(Some
Signkey.
{
pub;
priv;
stamp_start;
stamp_expire;
stamp_end;
master_sig;
}))
in in
let* sm_key = load "sm_key" in match (sm_key, keys) with
let* key_0 = load "0" in
match (sm_key, key_0) with
| None, None -> Ok None | None, None -> Ok None
| Some sm_key, Some key_0 -> | Some sm_key, Some keys -> Ok (Some { sm_key; keys })
let keys = [ key_0 ] in | _, _ -> error_invalid_state
Ok (Some { sm_key; keys })
| _, _ -> Error (`Msg "secmod_signkey: invalid local file state.")
let generate_fresh_keys () = let generate_fresh_secmod_data () =
let sm_key = Signkey.generate () in let sm_key = Signkey.generate () in
let keys = [ Signkey.generate () ] in let keys = [ Signkey.generate () ] in
{ sm_key; keys } { sm_key; keys }
@ -112,17 +67,13 @@ module Secmod_signkey = struct
Vif.Device.v ~name:"secmod_signkey" ~finally Vif.Device.v ~name:"secmod_signkey" ~finally
[ Vif.Device.value db_connection ] [ Vif.Device.value db_connection ]
@@ fun conn (_env : env) -> @@ fun conn (_env : env) ->
match load_local_keys conn with match load conn with
| Ok None -> | Ok None ->
Fmt.pr "secmod_signkey: generate_fresh_keys@."; let t = generate_fresh_secmod_data () in
let t = generate_fresh_keys () in
(*write_local_keys t |> Result.get_ok;*)
t t
| Ok (Some v) -> | Ok (Some v) -> v
Fmt.pr "secmod_signkey: loaded local keys@.";
v
| Error _ -> | Error _ ->
(* TODO pp error *) (* TODO error: pretty print *)
Fmt.failwith "secmod_signkey init failure." Fmt.failwith "secmod_signkey init failure."
end end
@ -132,12 +83,57 @@ module Secmod_denom = struct
keys: Denomination.t list; keys: Denomination.t list;
} }
let v = let dir = Fpath.(v "data" / "secmod_signkey")
let finally _key = () in
Vif.Device.v ~name:"secmod_denom" ~finally [] @@ fun (_env : env) -> let store_secmod_data t =
let* () = Data_file.write_eddsa Fpath.(dir / "sm_key") t.sm_key.priv in
t.keys
|> List.mapi (fun i key ->
let fname = Fpath.(dir / string_of_int i) in
(fname, key.Denomination.priv))
|> list_iter (fun (fname, key) -> Data_file.write_rsa fname key)
let load conn =
let error_invalid_state =
Fmt.error_msg "secmod_denom load error: invalid store state."
in
let* sm_key = Data_file.load_signkey conn Fpath.(dir / "sm_key") in
let* keys =
let* l =
let open Config.Coin in
Syntax.list_map
(fun coin ->
let section_name = coin.section_name in
let fname = Fpath.(dir / section_name) in
(* todo: could check that coin config match db values *)
Data_file.load_denom conn ~section_name fname)
all_coins
in
match Syntax.opt_list l with
| Error () -> error_invalid_state
| Ok opt -> Ok opt
in
match (sm_key, keys) with
| None, None -> Ok None
| Some sm_key, Some keys -> Ok (Some { sm_key; keys })
| _, _ -> error_invalid_state
let generate_fresh_secmod_data () =
let sm_key = Signkey.generate () in let sm_key = Signkey.generate () in
let keys = List.map Denomination.make Config.Coin.all_coins in let keys = List.map Denomination.make Config.Coin.all_coins in
{ sm_key; keys } { sm_key; keys }
let v =
let finally _key = () in
Vif.Device.v ~name:"secmod_denom" ~finally
[ Vif.Device.value db_connection ]
@@ fun conn (_env : env) ->
match load conn with
| Ok None ->
let t = generate_fresh_secmod_data () in
t
| Ok (Some v) -> v
| Error _ -> Fmt.failwith "secmod_denom init failure."
end end
let secmod_signkey = Secmod_signkey.v let secmod_signkey = Secmod_signkey.v

View file

@ -4,6 +4,7 @@ open Devices
let mk_future_denom denom_key_signf let mk_future_denom denom_key_signf
({ ({
pub; pub;
priv= _;
section_name; section_name;
value; value;
stamp_start; stamp_start;
@ -16,7 +17,6 @@ let mk_future_denom denom_key_signf
fee_refund; fee_refund;
age_mask; age_mask;
h_pub; h_pub;
sign= _;
master_sig= _; master_sig= _;
} : } :
Denomination.t) = Denomination.t) =

View file

@ -128,6 +128,7 @@ let add_denomination_key =
Denomination. Denomination.
{ {
pub; pub;
priv= _;
section_name= _; section_name= _;
value; value;
stamp_start; stamp_start;
@ -140,7 +141,6 @@ let add_denomination_key =
fee_refund; fee_refund;
age_mask; age_mask;
h_pub; h_pub;
sign= _;
master_sig; master_sig;
} }
-> ->

View file

@ -38,3 +38,11 @@ let list_fold_left f acc l =
let* acc = acc in let* acc = acc in
f acc v) f acc v)
(Ok acc) l (Ok acc) l
let opt_list l =
match (List.for_all Option.is_none l, List.for_all Option.is_some l) with
| _, true ->
let l = List.map Option.get l in
Ok (Some l)
| true, _ -> Ok None
| _, _ -> Error ()