add rsa priv_to_octets
This commit is contained in:
parent
bc91f7357c
commit
ad648178ab
3 changed files with 137 additions and 62 deletions
|
|
@ -97,67 +97,17 @@ end = struct
|
|||
end
|
||||
|
||||
module RsaPublicKey = struct
|
||||
module Binary = struct
|
||||
(* https://www.gnupg.org/documentation/manuals/gcrypt/MPI-formats.html
|
||||
format: { uint16_be: n size; uint16_be: e size; n; e}
|
||||
|
||||
integer in big-endian format (MSB first)
|
||||
leading zeroes are stripped unless they are required to keep a value positive
|
||||
no 0-termination *)
|
||||
|
||||
let rev_string len s = String.init len (fun i -> s.[len - 1 - i])
|
||||
|
||||
(* todo: need to strip leading zeros or something? *)
|
||||
(* reverse bytes because Z.of_bits reads bytes in little endian *)
|
||||
let z_of_bits_be src pos len =
|
||||
String.sub src pos len |> rev_string len |> Z.of_bits
|
||||
|
||||
let to_octets ({ n; e } : Mirage_crypto_pk.Rsa.pub) =
|
||||
let n = Z.to_bits n in
|
||||
let e = Z.to_bits e in
|
||||
let n_len = String.length n in
|
||||
let e_len = String.length e in
|
||||
let len = 4 + n_len + e_len in
|
||||
let b = Bytes.make len '\x00' in
|
||||
Bytes.set_uint16_be b 0 n_len;
|
||||
Bytes.set_uint16_be b 2 e_len;
|
||||
let n = rev_string n_len n in
|
||||
let e = rev_string e_len e in
|
||||
Bytes.blit_string n 0 b 4 n_len;
|
||||
Bytes.blit_string e 0 b (4 + n_len) e_len;
|
||||
Bytes.to_string b
|
||||
|
||||
let of_octets =
|
||||
let check = function
|
||||
| false -> Error "RsaPublicKey.of_octets: invalid data"
|
||||
| true -> Ok ()
|
||||
in
|
||||
fun s ->
|
||||
let open Syntax in
|
||||
let len = String.length s in
|
||||
let* () = check (len >= 4) in
|
||||
let n_len = String.get_uint16_be s 0 in
|
||||
let e_len = String.get_uint16_be s 2 in
|
||||
let* () = check (len = n_len + e_len + 4) in
|
||||
let n = z_of_bits_be s 4 n_len in
|
||||
let e = z_of_bits_be s (4 + n_len) e_len in
|
||||
Mirage_crypto_pk.Rsa.pub ~n ~e |> unwrap_err_msg
|
||||
|
||||
let of_octets s =
|
||||
match of_octets s with
|
||||
| Error e -> Fmt.failwith "RsaPublicKey.of_octets failure: %s@." e
|
||||
| Ok v -> v
|
||||
end
|
||||
|
||||
include Binary
|
||||
open Mirage_crypto_pk
|
||||
|
||||
type t = Rsa.pub
|
||||
|
||||
let of_octets = Util.Bin_rsa.pub_of_octets
|
||||
let to_octets = Util.Bin_rsa.pub_to_octets
|
||||
|
||||
let of_b32 s =
|
||||
let open Syntax in
|
||||
let* s = B32.decode s in
|
||||
let v = of_octets s in
|
||||
let v = Util.Bin_rsa.pub_of_octets s in
|
||||
let+ v = Rsa.pub ~n:v.n ~e:v.e |> unwrap_err_msg in
|
||||
v
|
||||
|
||||
|
|
@ -171,19 +121,16 @@ module RsaPrivateKey = struct
|
|||
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_octets = Util.Bin_rsa.priv_of_octets
|
||||
let to_octets = Util.Bin_rsa.priv_to_octets
|
||||
|
||||
let of_b32 s =
|
||||
let open Syntax in
|
||||
let* octets = B32.decode s in
|
||||
Ok (of_octets octets)
|
||||
let* s = B32.decode s in
|
||||
Ok (of_octets s)
|
||||
|
||||
let to_b32 t = B32.encode (to_octets t)
|
||||
let jsont = Jsont.of_of_string ~kind:"EddsaPrivateKey" of_b32 ~enc:to_b32
|
||||
let jsont = Jsont.of_of_string ~kind:"RsaPrivateKey" of_b32 ~enc:to_b32
|
||||
end
|
||||
|
||||
module RsaSignature : sig
|
||||
|
|
|
|||
111
src/util.ml
111
src/util.ml
|
|
@ -3,3 +3,114 @@
|
|||
let bin_of_string bin s =
|
||||
let v = Bin.decode bin s (ref 0) in
|
||||
Ok v
|
||||
|
||||
module Bin_rsa = struct
|
||||
(* TODO
|
||||
- need to strip leading zeros?
|
||||
- endianess ok?
|
||||
- tests *)
|
||||
(* RSA public key binary format
|
||||
https://www.gnupg.org/documentation/manuals/gcrypt/MPI-formats.html
|
||||
:= { uint16_be: n size; uint16_be: e size; n; e}
|
||||
|
||||
integer in big-endian format (MSB first)
|
||||
leading zeroes are stripped unless they are required to keep a value positive
|
||||
no 0-termination *)
|
||||
(* RSA private key custom format is inspired by the public rsa key format
|
||||
used by secmod to save private key to file *)
|
||||
open Syntax
|
||||
|
||||
module Internal = struct
|
||||
let rev_string len s = String.init len (fun i -> s.[len - 1 - i])
|
||||
|
||||
(* we need reverse bytes because Z.of_bits reads bytes in little endian *)
|
||||
let z_of_bits_be src pos len =
|
||||
String.sub src pos len |> rev_string len |> Z.of_bits
|
||||
|
||||
let z_to_bits_be z =
|
||||
let bits = Z.to_bits z in
|
||||
rev_string (String.length bits) bits
|
||||
|
||||
let check = function false -> Error (`Msg "invalid data") | true -> Ok ()
|
||||
|
||||
let z_array_to_octets (arr : Z.t array) =
|
||||
let nb = Array.length arr in
|
||||
(*Fmt.epr "z_array_to_octets, nb=%d@." nb;*)
|
||||
let bits_arr = Array.map z_to_bits_be arr in
|
||||
let len_arr = Array.map String.length bits_arr in
|
||||
let len = (2 * nb) + Array.fold_left ( + ) 0 len_arr in
|
||||
let b = Bytes.make len '\x00' in
|
||||
let pos = ref 0 in
|
||||
Array.iter
|
||||
(fun len ->
|
||||
Bytes.set_uint16_be b !pos len;
|
||||
pos := !pos + 2)
|
||||
len_arr;
|
||||
Array.iteri
|
||||
(fun i bits ->
|
||||
let len = len_arr.(i) in
|
||||
Bytes.blit_string bits 0 b !pos len;
|
||||
pos := !pos + len)
|
||||
bits_arr;
|
||||
Bytes.unsafe_to_string b
|
||||
|
||||
let z_array_of_octets ~nb s =
|
||||
(*Fmt.epr "z_array_of_octets, nb=%d@." nb;*)
|
||||
let s_len = String.length s in
|
||||
let* () = check (s_len > 2 * nb) in
|
||||
let pos = ref 0 in
|
||||
let len_arr =
|
||||
Array.init nb (fun _i ->
|
||||
let len = String.get_uint16_be s !pos in
|
||||
pos := !pos + 2;
|
||||
len)
|
||||
in
|
||||
let* () =
|
||||
let len = (2 * nb) + Array.fold_left ( + ) 0 len_arr in
|
||||
check (s_len = len)
|
||||
in
|
||||
let z_arr =
|
||||
Array.init nb (fun i ->
|
||||
let len = len_arr.(i) in
|
||||
let z = z_of_bits_be s !pos len in
|
||||
pos := !pos + len;
|
||||
z)
|
||||
in
|
||||
Ok z_arr
|
||||
end
|
||||
|
||||
open Internal
|
||||
|
||||
let pub_to_octets ({ n; e } : Mirage_crypto_pk.Rsa.pub) =
|
||||
z_array_to_octets [| n; e |]
|
||||
|
||||
let pub_of_octets s =
|
||||
let pub_of_octets s =
|
||||
let* arr = z_array_of_octets ~nb:2 s in
|
||||
match arr with
|
||||
| [| n; e |] ->
|
||||
let* pub = Mirage_crypto_pk.Rsa.pub ~n ~e in
|
||||
Ok pub
|
||||
| _ -> assert false
|
||||
in
|
||||
match pub_of_octets s with
|
||||
| Error (`Msg e) -> Fmt.failwith "rsa pub_of_octets failure: %s@." e
|
||||
| Ok v -> v
|
||||
|
||||
let priv_to_octets ({ e; d; n; p; q; dp; dq; q' } : Mirage_crypto_pk.Rsa.priv)
|
||||
=
|
||||
z_array_to_octets [| e; d; n; p; q; dp; dq; q' |]
|
||||
|
||||
let priv_of_octets s =
|
||||
let priv_of_octets s =
|
||||
let* arr = z_array_of_octets ~nb:8 s in
|
||||
match arr with
|
||||
| [| e; d; n; p; q; dp; dq; q' |] ->
|
||||
let* priv = Mirage_crypto_pk.Rsa.priv ~e ~d ~n ~p ~q ~dp ~dq ~q' in
|
||||
Ok priv
|
||||
| _ -> assert false
|
||||
in
|
||||
match priv_of_octets s with
|
||||
| Error (`Msg e) -> Fmt.failwith "rsa priv_of_octets failure: %s@." e
|
||||
| Ok v -> v
|
||||
end
|
||||
|
|
|
|||
17
test/test.ml
17
test/test.ml
|
|
@ -1,3 +1,20 @@
|
|||
let () = Mirage_crypto_rng_unix.use_default ()
|
||||
|
||||
let () =
|
||||
let priv = Mirage_crypto_pk.Rsa.generate ~bits:2048 () in
|
||||
let pub = Mirage_crypto_pk.Rsa.pub_of_priv priv in
|
||||
let () =
|
||||
let open Crypto.RsaPrivateKey in
|
||||
let priv' = priv |> to_octets |> of_octets in
|
||||
assert (to_octets priv = to_octets priv')
|
||||
in
|
||||
let () =
|
||||
let open Crypto.RsaPublicKey in
|
||||
let pub' = pub |> to_octets |> of_octets in
|
||||
assert (to_octets pub = to_octets pub')
|
||||
in
|
||||
()
|
||||
|
||||
let () =
|
||||
let open Api in
|
||||
let check jsont s =
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue