Add partial support for non-blocking mode of operation.