MTE unikernel with Mkernel+Mnet+Vifu
This commit is contained in:
parent
76662bf23c
commit
a23e59dfca
31 changed files with 535 additions and 376 deletions
|
|
@ -6,6 +6,8 @@ module Log = (val Logs.src_log src : Logs.LOG)
|
|||
open Syntax
|
||||
open Crypto
|
||||
open Time
|
||||
module Sfn = Mfat.Sfn
|
||||
module Spath = Mfat.Spath
|
||||
|
||||
module Cfg = struct
|
||||
open Config
|
||||
|
|
@ -48,87 +50,71 @@ type key = {
|
|||
}
|
||||
|
||||
type t = {
|
||||
sm_key_priv: EddsaPrivateKey.t;
|
||||
fs: Fat.t;
|
||||
sm_priv: EddsaPrivateKey.t;
|
||||
sm_pub: EddsaPublicKey.t;
|
||||
ht: (DenominationHash.t, key) Hashtbl.t;
|
||||
mutable last_change: Timestamp.t;
|
||||
}
|
||||
|
||||
let parse_filename =
|
||||
let scan_filename s =
|
||||
Scanf.sscanf_opt s "%Lu-%Lu" (fun t1 t2 ->
|
||||
(TimeAbsolute.of_s t1, TimeAbsolute.of_s t2))
|
||||
let key_bin =
|
||||
let open Bin in
|
||||
record (fun t1 t2 section_name priv ->
|
||||
let pub = RsaPrivateKey.pub_of_priv priv in
|
||||
let h_pub = DenominationHash.hash_of_rsa pub in
|
||||
{ section_name; t1; t2; priv; pub; h_pub })
|
||||
|+ field TimeAbsolute.bin (fun t -> t.t1)
|
||||
|+ field TimeAbsolute.bin (fun t -> t.t2)
|
||||
|+ field cstring (fun t -> t.section_name)
|
||||
|+ field RsaPrivateKey.bin (fun t -> t.priv)
|
||||
|> sealr
|
||||
|
||||
let key_spath k =
|
||||
let sfn_res =
|
||||
String.sub (DenominationHash.to_b32 k.h_pub) 0 8 |> Sfn.of_string
|
||||
in
|
||||
fun fpath -> scan_filename (Fpath.filename fpath)
|
||||
match sfn_res with
|
||||
| Error _ -> failwith "not possible"
|
||||
| Ok sfn -> Spath.(Cfg.key_dir / sfn)
|
||||
|
||||
let pp_filename =
|
||||
let to_int64 abs =
|
||||
abs |> Timestamp.of_absolute |> Timestamp.to_s |> function
|
||||
| None ->
|
||||
(* (= `never`) this should not happen given resonable config value *)
|
||||
Fmt.failwith "encountered timestamp with value `never`"
|
||||
| Some i -> i
|
||||
in
|
||||
fun ppf (t1, t2) -> Fmt.pf ppf "%Lu-%Lu" (to_int64 t1) (to_int64 t2)
|
||||
let read_eddsa fs spath =
|
||||
Log.debug (fun m -> m "reading key file `%a`" Spath.pp spath);
|
||||
let* data = Fat.read fs spath |> unwrap_msg in
|
||||
let* priv = EddsaPrivateKey.of_octets data in
|
||||
let pub = EddsaPrivateKey.pub_of_priv priv in
|
||||
Ok (priv, pub)
|
||||
|
||||
let key_fpath k =
|
||||
let fname = Fmt.str "%a" pp_filename (k.t1, k.t2) in
|
||||
Fpath.(v Cfg.key_dir / k.section_name / fname)
|
||||
|
||||
(* -- IO -- *)
|
||||
|
||||
let read_eddsa fpath =
|
||||
Log.debug (fun m -> m "reading key file `%a`" Fpath.pp fpath);
|
||||
let* data = Bos.OS.File.read fpath |> unwrap_msg in
|
||||
EddsaPrivateKey.of_octets data
|
||||
|
||||
let read_rsa fpath =
|
||||
Log.debug (fun m -> m "reading key file `%a`" Fpath.pp fpath);
|
||||
let* data = Bos.OS.File.read fpath |> unwrap_msg in
|
||||
RsaPrivateKey.of_octets data
|
||||
|
||||
let write_eddsa fpath priv =
|
||||
Log.debug (fun m -> m "writing key file `%a`" Fpath.pp fpath);
|
||||
let write_eddsa fs spath priv =
|
||||
Log.debug (fun m -> m "writing key file `%a`" Spath.pp spath);
|
||||
let data = EddsaPrivateKey.to_octets priv in
|
||||
Bos.OS.File.write fpath data |> unwrap_msg
|
||||
Fat.write fs spath data |> unwrap_msg
|
||||
|
||||
let write_rsa fpath priv =
|
||||
let data = RsaPrivateKey.to_octets priv in
|
||||
Bos.OS.File.write fpath data |> unwrap_msg
|
||||
let read_key fs spath =
|
||||
Log.debug (fun m -> m "reading key file `%a`" Spath.pp spath);
|
||||
let* data = Fat.read fs spath |> unwrap_msg in
|
||||
let k = Bin.decode key_bin data (ref 0) in
|
||||
Ok k
|
||||
|
||||
let write_key k = write_rsa (key_fpath k) k.priv
|
||||
let write_key fs k =
|
||||
let spath = key_spath k in
|
||||
Log.debug (fun m -> m "writing key file `%a`" Spath.pp spath);
|
||||
let data = Bin.to_string key_bin k in
|
||||
Fat.write fs spath data |> unwrap_msg
|
||||
|
||||
let delete_file fpath =
|
||||
(* check fpath just to be safe *)
|
||||
let () =
|
||||
let root = Fpath.v Cfg.key_dir in
|
||||
if not @@ Fpath.is_rooted ~root fpath then
|
||||
Fmt.failwith
|
||||
"delete_file failure: file `%a` is not contained in secmod directory"
|
||||
Fpath.pp fpath
|
||||
in
|
||||
Log.debug (fun m -> m "delete key file `%a`" Fpath.pp fpath);
|
||||
let+ () = Bos.OS.File.delete ~must_exist:true fpath |> unwrap_msg in
|
||||
let delete_file fs spath =
|
||||
Log.debug (fun m -> m "delete key file `%a`" Spath.pp spath);
|
||||
let+ () = Fat.remove fs spath |> unwrap_msg in
|
||||
()
|
||||
|
||||
let get_key_dir_contents dir_fpath =
|
||||
let* b = Bos.OS.Dir.create ~mode:0o700 dir_fpath |> unwrap_msg in
|
||||
if b then Log.info (fun m -> m "created directory `%a`" Fpath.pp dir_fpath);
|
||||
let+ l =
|
||||
Bos.OS.Dir.contents ~dotfiles:false ~rel:false dir_fpath |> unwrap_msg
|
||||
in
|
||||
List.map Fpath.normalize l
|
||||
|
||||
(* -- *)
|
||||
|
||||
let gen_key ~section_name t1 t2 =
|
||||
let bits = Cfg.rsa_keysize ~section_name in
|
||||
let priv, pub = RsaPrivateKey.generate ~bits () in
|
||||
let h_pub = DenominationHash.hash_of_rsa pub in
|
||||
let k = { section_name; priv; pub; h_pub; t1; t2 } in
|
||||
Log.debug (fun m ->
|
||||
m "generated key (%a):@,`%s`" pp_filename (t1, t2)
|
||||
(DenominationHash.to_octets h_pub |> B32.encode));
|
||||
{ section_name; priv; pub; h_pub; t1; t2 }
|
||||
m "generated key %s `%s`" section_name (DenominationHash.to_b32 k.h_pub));
|
||||
k
|
||||
|
||||
let sort_keys l = List.sort (fun a b -> TimeAbsolute.compare a.t2 b.t2) l
|
||||
|
||||
|
|
@ -147,8 +133,6 @@ let split_in_periodes ~start ~end_ ~duration_withdraw =
|
|||
go acc start end_
|
||||
|
||||
let gen_additional_keys_until_lookahead ~now ~section_name l =
|
||||
(* try to _not_ generate keys with validity start in the past
|
||||
(probably not important) *)
|
||||
let start =
|
||||
match List.rev (sort_keys l) with
|
||||
| [] -> now
|
||||
|
|
@ -164,58 +148,39 @@ let gen_additional_keys_until_lookahead ~now ~section_name l =
|
|||
in
|
||||
new_keys
|
||||
|
||||
let sm_key_fpath =
|
||||
Result.get_ok
|
||||
@@
|
||||
let+ fpath = Fpath.of_string Cfg.sm_priv_key |> unwrap_msg in
|
||||
Fpath.normalize fpath
|
||||
|
||||
(* we load sm_key separately
|
||||
we don't accept non-key files in key_dir *)
|
||||
let load_key ~section_name fpath =
|
||||
match parse_filename fpath with
|
||||
| None -> Fmt.error "invalid file `%a`" Fpath.pp fpath
|
||||
| Some (t1, t2) ->
|
||||
let+ priv = read_rsa fpath in
|
||||
let pub = RsaPrivateKey.pub_of_priv priv in
|
||||
let h_pub = DenominationHash.hash_of_rsa pub in
|
||||
{ section_name; priv; pub; h_pub; t1; t2 }
|
||||
|
||||
let load_section section_name =
|
||||
let section_fpath = Fpath.(v Cfg.key_dir / section_name) in
|
||||
let* l = get_key_dir_contents section_fpath in
|
||||
let l = List.filter (fun fpath -> not @@ Fpath.equal fpath sm_key_fpath) l in
|
||||
let* keys = list_map (load_key ~section_name) l in
|
||||
Ok keys
|
||||
|
||||
let load () =
|
||||
let* keys_l = list_map load_section Cfg.sections in
|
||||
let keys = List.concat keys_l in
|
||||
let load fs =
|
||||
let* () =
|
||||
if Fat.exists fs Cfg.key_dir then Ok ()
|
||||
else Fat.mkdir fs Cfg.key_dir |> unwrap_msg
|
||||
in
|
||||
let* l = Fat.ls fs Cfg.key_dir |> unwrap_msg in
|
||||
let l = List.map (fun entry -> entry.Fat.name) l in
|
||||
let l = List.map (Spath.add Cfg.key_dir) l in
|
||||
let l =
|
||||
List.filter (fun spath -> not @@ Spath.equal spath Cfg.sm_priv_key) l
|
||||
in
|
||||
let* keys = list_map (read_key fs) l in
|
||||
match keys with
|
||||
| [] -> Ok None
|
||||
| _l ->
|
||||
let* sm_key_priv = read_eddsa sm_key_fpath in
|
||||
let sm_pub = EddsaPrivateKey.pub_of_priv sm_key_priv in
|
||||
let* sm_priv, sm_pub = read_eddsa fs Cfg.sm_priv_key in
|
||||
let ht = Hashtbl.create 0xff in
|
||||
let () = List.iter (fun k -> Hashtbl.replace ht k.h_pub k) keys in
|
||||
(* TODO list_issue_date *)
|
||||
let last_change = Timestamp.never in
|
||||
Ok (Some { sm_key_priv; sm_pub; ht; last_change })
|
||||
Ok (Some { fs; sm_priv; sm_pub; ht })
|
||||
|
||||
let init () =
|
||||
let* opt = load () in
|
||||
let now = TimeAbsolute.of_ptime (Ptime_clock.now ()) in
|
||||
let init fs =
|
||||
let* opt = load fs in
|
||||
let now = TimeAbsolute.of_ptime (Mirage_ptime.now ()) in
|
||||
let* t =
|
||||
match opt with
|
||||
| Some t -> Ok t
|
||||
| None ->
|
||||
let sm_key_priv, sm_pub = EddsaPrivateKey.generate () in
|
||||
let sm_priv, sm_pub = EddsaPrivateKey.generate () in
|
||||
Log.debug (fun m ->
|
||||
m "generated secmod key: `%s`" (EddsaPublicKey.to_b32 sm_pub));
|
||||
let* () = write_eddsa sm_key_fpath sm_key_priv in
|
||||
let* () = write_eddsa fs Cfg.sm_priv_key sm_priv in
|
||||
let ht = Hashtbl.create 0xff in
|
||||
let last_change = Timestamp.of_absolute now in
|
||||
Ok { sm_key_priv; sm_pub; ht; last_change }
|
||||
Ok { fs; sm_priv; sm_pub; ht }
|
||||
in
|
||||
let all_keys = List.of_seq @@ Hashtbl.to_seq_values t.ht in
|
||||
let new_keys_l =
|
||||
|
|
@ -229,12 +194,12 @@ let init () =
|
|||
in
|
||||
let new_keys = List.concat new_keys_l in
|
||||
List.iter (fun k -> Hashtbl.replace t.ht k.h_pub k) new_keys;
|
||||
let+ () = list_iter write_key new_keys in
|
||||
let+ () = list_iter (write_key fs) new_keys in
|
||||
t
|
||||
|
||||
module Make () = struct
|
||||
module Make (Fs : Fat.FS) = struct
|
||||
let t =
|
||||
match init () with
|
||||
match init Fs.t with
|
||||
| Error e -> Fmt.failwith "secmod_rsa initialization failure: %s." e
|
||||
| Ok t -> t
|
||||
|
||||
|
|
@ -244,7 +209,7 @@ module Make () = struct
|
|||
let delete h_pub =
|
||||
let* k = find h_pub in
|
||||
Hashtbl.remove t.ht h_pub;
|
||||
delete_file (key_fpath k)
|
||||
delete_file t.fs (key_spath k)
|
||||
|
||||
let _delete_outdated ~now =
|
||||
Hashtbl.to_seq t.ht
|
||||
|
|
@ -254,20 +219,17 @@ module Make () = struct
|
|||
|> list_iter delete
|
||||
|
||||
let add section_name t1 t2 =
|
||||
let last_change = Timestamp.of_ptime (Ptime_clock.now ()) in
|
||||
let k = gen_key ~section_name t1 t2 in
|
||||
let+ () = write_key k in
|
||||
let+ () = write_key t.fs k in
|
||||
Hashtbl.replace t.ht k.h_pub k;
|
||||
t.last_change <- last_change;
|
||||
()
|
||||
|
||||
let last_change () = t.last_change
|
||||
let sm_pub = t.sm_pub
|
||||
let sign_secmod s = EddsaSignature.sign ~key:t.sm_key_priv s
|
||||
let sign_secmod s = EddsaSignature.sign ~key:t.sm_priv s
|
||||
|
||||
let sign h_pub msg =
|
||||
let+ k = find h_pub in
|
||||
let data = Fdh_rsa.sign ~key:k.priv msg in
|
||||
let data = RsaSignature.sign ~key:k.priv msg in
|
||||
data
|
||||
|
||||
let revoke h_pub =
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue