balsoft/ligo
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Baker: add client-side validation

Browse Source
This commit is contained in:
Vincent Botbol 2018-06-14 12:16:17 +02:00 committed by Grégoire Henry
parent 0512ff55ce
commit ea9b3ae68c
14 changed files with 209 additions and 72 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/proto_alpha/lib_client/jbuild
View File

@ -9,6 +9,7 @@
tezos-shell-services
tezos-client-base
tezos-rpc
tezos-storage
tezos-signer-backends
bip39))
(library_flags (:standard -linkall))
@ -17,6 +18,7 @@
-open Tezos_base__TzPervasives
-open Tezos_shell_services
-open Tezos_client_base
-open Tezos_storage
-open Tezos_rpc))))
(alias

5
src/proto_alpha/lib_client/proto_alpha.ml
View File

@ -8,9 +8,12 @@
(**************************************************************************)
module Name = struct let name = "alpha" end
module Alpha_environment = Tezos_protocol_environment_faked.MakeV1(Name)()
module T = Tezos_protocol_environment.Make(Tezos_storage.Context)
module Alpha_environment = T.MakeV1(Name)()
module Proto = Tezos_protocol_alpha.Functor.Make(Alpha_environment)
module Alpha_block_services = Block_services.Make(Proto)(Proto)
include Proto
class type rpc_context = object

6
src/proto_alpha/lib_delegate/client_baking_blocks.ml
View File

@ -19,6 +19,7 @@ type block_info = {
protocol: Protocol_hash.t ;
next_protocol: Protocol_hash.t ;
level: Raw_level.t ;
context : Context_hash.t ;
}
let raw_info cctxt ?(chain = `Main) hash shell_header =
@ -27,12 +28,13 @@ let raw_info cctxt ?(chain = `Main) hash shell_header =
Shell_services.Blocks.protocols
cctxt ~chain ~block () >>=? fun { current_protocol = protocol ;
next_protocol } ->
let { Tezos_base.Block_header.predecessor ; fitness ; timestamp ; level ; _ } =
let { Tezos_base.Block_header.predecessor ; fitness ;
timestamp ; level ; context ; _ } =
shell_header in
match Raw_level.of_int32 level with
| Ok level ->
return { hash ; chain_id ; predecessor ; fitness ;
timestamp ; protocol ; next_protocol ; level }
timestamp ; protocol ; next_protocol ; level ; context }
| Error _ ->
failwith "Cannot convert level into int32"

1
src/proto_alpha/lib_delegate/client_baking_blocks.mli
View File

@ -19,6 +19,7 @@ type block_info = {
protocol: Protocol_hash.t ;
next_protocol: Protocol_hash.t ;
level: Raw_level.t ;
context : Context_hash.t ;
}
val info:

84
src/proto_alpha/lib_delegate/client_baking_forge.ml
View File

@ -18,9 +18,21 @@ let generate_proof_of_work_nonce () =
let generate_seed_nonce () =
match Nonce.of_bytes @@
Rand.generate Constants.nonce_length with
| Error _ -> assert false
| Error _errs -> assert false
| Ok nonce -> nonce
let rec retry_call f ?(msg="Call error") ?(n=5) () =
f () >>= function
| Ok r -> return r
| (Error errs) as x ->
if n > 0 then
begin
lwt_log_error "%s\n%a\nRetrying..."
msg pp_print_error errs >>= fun () ->
Lwt_unix.sleep 1. >>= retry_call f ~msg ~n:(n-1)
end
else
Lwt.return x
let forge_block_header
(cctxt : #Proto_alpha.full)
@ -370,14 +382,16 @@ let rec insert_baking_slot slot = function
type state = {
genesis: Block_hash.t ;
index : Context.index ;
mutable delegates: public_key_hash list ;
mutable best: Client_baking_blocks.block_info ;
mutable future_slots:
(Time.t * (Client_baking_blocks.block_info * int * public_key_hash)) list ;
}
let create_state genesis delegates best =
let create_state genesis index delegates best =
{ genesis ;
index ;
delegates ;
best ;
future_slots = [] ;
@ -491,6 +505,42 @@ let pop_baking_slots state =
state.future_slots <- future_slots ;
slots
let filter_invalid_operations (cctxt : #full) state block_info (operations : packed_operation list list) =
let open Client_baking_simulator in
lwt_debug "Starting client-side validation" >>= fun () ->
begin_construction cctxt state.index block_info >>=? fun initial_inc ->
let endorsements = List.nth operations 0 in
let votes = List.nth operations 1 in
let anonymous = List.nth operations 2 in
let managers = List.nth operations 3 in
(* TODO log *)
let validate_operation inc op =
add_operation inc op >>= function
| Error _ -> return None
| Ok inc -> return (Some inc)
in
let filter_valid_operations inc ops =
fold_left_s (fun (inc, acc) op ->
validate_operation inc op >>=? function
| None -> return (inc, acc)
| Some inc -> return (inc, op :: acc)
) (inc, []) ops
in
let is_valid_endorsement endorsement =
validate_operation initial_inc endorsement >>=? function
| None -> return None
| Some inc -> finalize_construction inc >>= begin function
| Ok _ -> return (Some endorsement)
| Error _ -> return None
end
in
filter_map_s is_valid_endorsement endorsements >>=? fun _endorsements ->
filter_valid_operations initial_inc votes >>=? fun (inc, votes) ->
filter_valid_operations inc anonymous >>=? fun (inc, anonymous) ->
filter_valid_operations inc managers >>=? fun (_, managers) ->
return @@ List.map List.rev [ endorsements ; votes ; anonymous ; managers ]
let bake_slot
cctxt
state
@ -525,9 +575,27 @@ let bake_slot
let protocol_data =
forge_faked_protocol_data ~priority ~seed_nonce_hash in
let operations = classify_operations operations in
(* Don't validate if current block is genesis *)
begin
(* Don't load an alpha context if still in genesis *)
if Protocol_hash.(bi.protocol = bi.next_protocol) then
filter_invalid_operations cctxt state bi operations
else
return operations
end >>= function
| Error errs ->
lwt_log_error "Error while filtering invalid operations (client-side) :@\n%a"
pp_print_error
errs >>= fun () ->
return None
| Ok operations ->
retry_call
(fun () ->
Alpha_block_services.Helpers.Preapply.block
cctxt ~chain ~block
~timestamp ~sort:true ~protocol_data operations >>= function
~timestamp ~sort:true ~protocol_data operations)
~msg:"Error while prevalidating operations" ()
>>= function
| Error errs ->
lwt_log_error "Error while prevalidating operations:@\n%a"
pp_print_error
@ -635,6 +703,7 @@ let check_error p =
let create
(cctxt : #Proto_alpha.full)
?max_priority
~(context_path: string)
(delegates: public_key_hash list)
(block_stream: Client_baking_blocks.block_info tzresult Lwt_stream.t)
(bi: Client_baking_blocks.block_info) =
@ -651,7 +720,9 @@ let create
last_get_block := Some t ;
t
| Some t -> t in
let state = create_state genesis_hash delegates bi in
lwt_debug "Opening shell context" >>= fun () ->
Client_baking_simulator.load_context ~context_path >>= fun index ->
let state = create_state genesis_hash index delegates bi in
check_error @@ insert_block cctxt ?max_priority state bi >>= fun () ->
(* main loop *)
@ -694,16 +765,15 @@ let create
cctxt#message "Starting the baker" >>= fun () ->
worker_loop ()
(* Wrapper around previous [create] function that handles the case of
unavailable blocks (empty block chain). *)
let create
(cctxt : #Proto_alpha.full)
?max_priority
~(context_path: string)
(delegates: public_key_hash list)
(block_stream: Client_baking_blocks.block_info tzresult Lwt_stream.t) =
Client_baking_scheduling.wait_for_first_block
~info:cctxt#message
block_stream
(create cctxt ?max_priority delegates block_stream)
(create cctxt ?max_priority ~context_path delegates block_stream)

1
src/proto_alpha/lib_delegate/client_baking_forge.mli
View File

@ -79,6 +79,7 @@ end
val create:
#Proto_alpha.full ->
?max_priority: int ->
context_path: string ->
public_key_hash list ->
Client_baking_blocks.block_info tzresult Lwt_stream.t ->
unit tzresult Lwt.t

28
src/proto_alpha/lib_delegate/client_baking_lib.ml
View File

@ -101,31 +101,3 @@ let reveal_nonces cctxt () =
Client_baking_forge.get_unrevealed_nonces
cctxt cctxt#block >>=? fun nonces ->
do_reveal cctxt cctxt#block nonces
let run_daemon cctxt ?max_priority ~endorsement_delay delegates ~endorsement ~baking ~denunciation =
let endorser =
if endorsement then
Client_daemon.Endorser.run cctxt
~delay:endorsement_delay
~min_date:((Time.add (Time.now ()) (Int64.neg 1800L)))
delegates >>=? fun () -> return ()
else return ()
in
let baker =
if baking then
Client_daemon.Baker.run cctxt
?max_priority
~min_date:((Time.add (Time.now ()) (Int64.neg 1800L)))
delegates >>=? fun () -> return ()
else return ()
in
let accuser =
if denunciation then
Client_daemon.Accuser.run cctxt >>=? fun () -> return ()
else
return ()
in
endorser >>=? fun () ->
baker >>=? fun () ->
accuser >>=? fun () ->
return ()

11
src/proto_alpha/lib_delegate/client_baking_lib.mli
View File

@ -45,14 +45,3 @@ val reveal_nonces :
#Proto_alpha.full ->
unit ->
unit Error_monad.tzresult Lwt.t
(** Initialize the baking daemon *)
val run_daemon:
#Proto_alpha.full ->
?max_priority:int ->
endorsement_delay:int ->
public_key_hash list ->
endorsement:bool ->
baking:bool ->
denunciation:bool ->
unit Error_monad.tzresult Lwt.t

64
src/proto_alpha/lib_delegate/client_baking_simulator.ml Normal file
View File

@ -0,0 +1,64 @@
(**************************************************************************)
(* *)
(* Copyright (c) 2014 - 2018. *)
(* Dynamic Ledger Solutions, Inc. <contact@tezos.com> *)
(* *)
(* All rights reserved. No warranty, explicit or implicit, provided. *)
(* *)
(**************************************************************************)
open Proto_alpha
open Alpha_context
module Main = Alpha_environment.Lift(Main)
type incremental = {
predecessor: Client_baking_blocks.block_info ;
context : Context.t ;
state: Main.validation_state ;
rev_operations: Operation.packed list ;
header: Tezos_base.Block_header.shell_header ;
}
let load_context ~context_path =
Context.init ~readonly:true context_path
let begin_construction (_cctxt : #Proto_alpha.full) index predecessor =
let { Client_baking_blocks.context } = predecessor in
Context.checkout_exn index context >>= fun context ->
let timestamp = Time.now () in
let predecessor_hash = predecessor.hash in
(* Shell_services.Blocks.header cctxt ~chain:`Main ~block:(`Hash (predecessor_hash, 0)) ()
* >>=? fun { shell ; _ } -> *)
let header : Tezos_base.Block_header.shell_header = Tezos_base.Block_header.{
predecessor = predecessor_hash ;
proto_level = 0 (* shell.proto_level *) ;
validation_passes = 0 (* shell.validation_passes *) ;
fitness = predecessor.fitness (* shell.fitness *) ;
timestamp ;
level = 0l (* shell.level *) ;
context = Context_hash.zero ;
operations_hash = Operation_list_list_hash.zero ;
} in
Main.begin_construction
~predecessor_context: context
~predecessor_timestamp: header.timestamp
~predecessor_fitness: header.fitness
~predecessor_level: header.level
~predecessor:predecessor_hash
~timestamp
() >>=? fun state ->
return {
predecessor ;
context ;
state ;
rev_operations = [] ;
header ;
}
let add_operation st ( op : Operation.packed ) =
Main.apply_operation st.state op >>=? fun (state, _) ->
return { st with state ; rev_operations = op :: st.rev_operations }
let finalize_construction inc =
Main.finalize_block inc.state >>=? fun _ -> return ()

27
src/proto_alpha/lib_delegate/client_baking_simulator.mli Normal file
View File

@ -0,0 +1,27 @@
(**************************************************************************)
(* *)
(* Copyright (c) 2014 - 2018. *)
(* Dynamic Ledger Solutions, Inc. <contact@tezos.com> *)
(* *)
(* All rights reserved. No warranty, explicit or implicit, provided. *)
(* *)
(**************************************************************************)
open Proto_alpha
open Alpha_context
type incremental = {
predecessor: Client_baking_blocks.block_info ;
context : Context.t ;
state: Main.validation_state ;
rev_operations: Operation.packed list ;
header: Tezos_base.Block_header.shell_header ;
}
val load_context : context_path:string -> Context.index Lwt.t
val begin_construction : #Proto_alpha.full -> Context.index -> Client_baking_blocks.block_info -> incremental tzresult Lwt.t
val add_operation : incremental -> Operation.packed -> incremental tzresult Lwt.t
val finalize_construction : incremental -> unit tzresult Lwt.t

5
src/proto_alpha/lib_delegate/client_daemon.ml
View File

@ -18,14 +18,13 @@ module Endorser = struct
end
module Baker = struct
let run (cctxt : #Proto_alpha.full) ?max_priority ?min_date delegates =
let run (cctxt : #Proto_alpha.full) ?max_priority ?min_date ~context_path delegates =
Client_baking_blocks.monitor_heads
cctxt `Main >>=? fun block_stream ->
Client_baking_forge.create cctxt
?max_priority delegates block_stream >>=? fun () ->
?max_priority ~context_path delegates block_stream >>=? fun () ->
ignore min_date;
return ()

1
src/proto_alpha/lib_delegate/client_daemon.mli
View File

@ -23,6 +23,7 @@ module Baker : sig
#Proto_alpha.full ->
?max_priority: int ->
?min_date: Time.t ->
context_path: string ->
public_key_hash list -> unit tzresult Lwt.t
end

10
src/proto_alpha/lib_delegate/delegate_commands.ml
View File

@ -54,14 +54,18 @@ let baker_commands () =
title = "Commands related to the baker daemon." }
in
[
command ~group ~desc: "Launch a daemon that handles delegate operations."
command ~group ~desc: "Launch the baker daemon."
(args1 max_priority_arg)
(prefixes [ "launch" ]
(prefixes [ "launch" ; "with" ; "context" ]
@@ string
~name:"Context path"
~desc:"Path to the shell context"
@@ seq_of_param Client_keys.Public_key_hash.alias_param)
(fun max_priority delegates cctxt ->
(fun max_priority context_path delegates cctxt ->
Client_daemon.Baker.run cctxt
?max_priority
~min_date:((Time.add (Time.now ()) (Int64.neg 1800L)))
~context_path
(List.map snd delegates)
)
]

2
src/proto_alpha/lib_delegate/jbuild
View File

@ -10,6 +10,7 @@
tezos-client-base
tezos-client-alpha
tezos-client-commands
tezos-storage
tezos-rpc))
(library_flags (:standard -linkall))
(modules (:standard \
@ -22,6 +23,7 @@
-open Tezos_client_base
-open Tezos_client_alpha
-open Tezos_client_commands
-open Tezos_storage
-open Tezos_rpc))))
(library