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

reject invalid protocols

Browse Source
This commit is contained in:
Vincent Bernardoff 2016-10-24 14:10:17 +02:00
parent 488373551b
commit ec79241adb
2 changed files with 9 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
src/node/shell/node.ml
View File

@ -25,11 +25,16 @@ let inject_operation validator ?force bytes =
Lwt.return (hash, t)
let inject_protocol state ?force:_ proto =
(* TODO: Validate the protocol *)
let proto_bytes = Store.Protocol.to_bytes proto in
let hash = Protocol_hash.hash_bytes [proto_bytes] in
let t = State.Protocol.store state proto_bytes >>|? ignore in
Lwt.return (hash, t)
let validation = Updater.compile hash proto >>= function
| false -> Lwt.fail_with (Format.asprintf "Invalid protocol %a: compilation failed" Protocol_hash.pp_short hash)
| true ->
State.Protocol.store state proto_bytes >>= function
| Ok None -> Lwt.fail_with "Previously registred protocol"
| t -> t >|? ignore |> Lwt.return
in
Lwt.return (hash, validation)
let process_operation state validator bytes =
State.Operation.store state bytes >>= function

2
src/node/updater/updater.mli
View File

@ -70,7 +70,7 @@ module type REGISTRED_PROTOCOL = sig
and type 'a tzresult := 'a tzresult
end
type component = {
type component = Store.component = {
name : string ;
interface : string option ;
implementation : string ;