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

P2p: missing Lwt_utils.protect in the reader worker.

Browse Source
This commit is contained in:
Grégoire Henry 2017-11-08 16:10:40 +01:00 committed by Benjamin Canou
parent c9fe53eb10
commit cb1e5cc396
1 changed files with 8 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
src/node/net/p2p_connection.ml
View File

@ -261,16 +261,19 @@ module Reader = struct
let rec worker_loop st init_mbytes =
Lwt_unix.yield () >>= fun () ->
begin
read_message st init_mbytes >>=? fun msg ->
match msg with
| None ->
Lwt_pipe.push st.messages (Error [Decoding_error]) >>= fun () ->
return None
Lwt_utils.protect ~canceler:st.canceler begin fun () ->
Lwt_pipe.push st.messages (Error [Decoding_error]) >>= fun () ->
return None
end
| Some (msg, size, rem_mbytes) ->
Lwt_pipe.push st.messages (Ok (size, msg)) >>= fun () ->
return (Some rem_mbytes)
Lwt_utils.protect ~canceler:st.canceler begin fun () ->
Lwt_pipe.push st.messages (Ok (size, msg)) >>= fun () ->
return (Some rem_mbytes)
end
end >>= function
| Ok Some rem_mbytes ->
worker_loop st rem_mbytes