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

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

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