github/falco
1
0
Fork 0
mirror of https://github.com/falcosecurity/falco.git synced 2025-07-04 18:36:48 +00:00
Code Issues Projects Releases Wiki Activity

new(userspace/falco): request stream context process and end handling

Browse Source 
Co-authored-by: Lorenzo Fontana <lo@linux.com>
Signed-off-by: Leonardo Di Donato <leodidonato@gmail.com>
This commit is contained in:
Leonardo Di Donato 2019-09-03 16:19:51 +00:00 committed by Leo Di Donato
parent 13f5a76b97
commit fbe4e34a57
1 changed files with 45 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

45
userspace/falco/grpc_server.cpp
View File

@ -44,11 +44,56 @@ void request_stream_context<falco_output_request, falco_output_response>::start(
template<>
void request_stream_context<falco_output_request, falco_output_response>::process(falco_grpc_server* srv)
{
// When it is the 1st process call
if(m_state == request_context_base::REQUEST)
{
m_state = request_context_base::WRITE;
m_stream_ctx.reset(new stream_context(m_srv_ctx.get()));
}
// Processing
falco_output_response res;
(srv->*m_process_func)(*m_stream_ctx, m_req, res);
// When there still are more responses to stream
if(m_stream_ctx->m_has_more)
{
m_res_writer->Write(res, this);
}
// No more responses to stream
else
{
// Communicate to the gRPC runtime that we have finished.
// The memory address of `this` instance uniquely identifies the event.
m_state = request_context_base::FINISH;
m_res_writer->Finish(grpc::Status::OK, this);
}
}
template<>
void request_stream_context<falco_output_request, falco_output_response>::end(falco_grpc_server* srv, bool isError)
{
if(m_stream_ctx)
{
m_stream_ctx->m_status = stream_context::SUCCESS;
if(isError)
{
m_stream_ctx->m_status = stream_context::ERROR;
// todo > log error
}
// Complete the processing
falco_output_response res;
(srv->*m_process_func)(*m_stream_ctx, m_req, res);
}
else
{
// Handle the edge case when `m_request_func` event failed
// which means `m_stream_ctx` was not set
// todo > log error
}
start(srv);
}
bool falco_grpc_server_impl::is_running()