Support for Message flushing upon disconnection

Signed-off-by: Eyal Salomon <esalomon@mellanox.com>