Mask body of runOps to avoid a heap corruption (#89)

This commit is contained in:
Moritz Kiefer 2019-08-29 18:15:48 +02:00 committed by Gabriel Gonzalez
parent 01b17286b6
commit be70fc49b0
1 changed files with 8 additions and 1 deletions

View File

@ -208,7 +208,14 @@ runOps :: C.Call
-> IO (Either GRPCIOError [OpRecvResult])
runOps call cq ops =
let l = length ops in
withOpArrayAndCtxts ops $ \(opArray, contexts) -> do
-- It is crucial to mask exceptions here. If we dont do this, we can
-- run into the following situation:
--
-- 1. We allocate an OpContext, e.g., OpRecvMessageContext and the corresponding ByteBuffer.
-- 2. We pass the buffer to gRPC in startBatch.
-- 3. If we now get an exception we will free the ByteBuffer.
-- 4. gRPC can now end up writing to the freed ByteBuffer and we get a heap corruption.
withOpArrayAndCtxts ops $ \(opArray, contexts) -> mask_ $ do
grpcDebug $ "runOps: allocated op contexts: " ++ show contexts
tag <- newTag cq
grpcDebug $ "runOps: tag: " ++ show tag