diff options
3 files changed, 9 insertions, 0 deletions
diff --git a/libtransport/src/hicn/transport/interfaces/socket.h b/libtransport/src/hicn/transport/interfaces/socket.h index f0194880a..4c9bda7df 100644 --- a/libtransport/src/hicn/transport/interfaces/socket.h +++ b/libtransport/src/hicn/transport/interfaces/socket.h @@ -78,6 +78,8 @@ class Socket { virtual void connect() = 0; + virtual bool isRunning() = 0; + protected: virtual ~Socket(){}; diff --git a/libtransport/src/hicn/transport/interfaces/socket_consumer.h b/libtransport/src/hicn/transport/interfaces/socket_consumer.h index eceee2d34..0f83fd38f 100644 --- a/libtransport/src/hicn/transport/interfaces/socket_consumer.h +++ b/libtransport/src/hicn/transport/interfaces/socket_consumer.h @@ -159,6 +159,11 @@ class ConsumerSocket : public BaseSocket { void connect() override; /** + * @brief Check whether consumer socket is active or not. + */ + bool isRunning() override { return transport_protocol_->isRunning(); } + + /** * Retrieve a content using the protocol specified in the constructor. * This function blocks until the whole content is downloaded. * For monitoring the status of the download, the application MUST set the diff --git a/libtransport/src/hicn/transport/interfaces/socket_producer.h b/libtransport/src/hicn/transport/interfaces/socket_producer.h index 1f1acd401..83d0f73f3 100644 --- a/libtransport/src/hicn/transport/interfaces/socket_producer.h +++ b/libtransport/src/hicn/transport/interfaces/socket_producer.h @@ -51,6 +51,8 @@ class ProducerSocket : public Socket<BasePortal>, void connect() override; + bool isRunning() override { return !io_service_.stopped(); }; + uint32_t produce(Name content_name, const uint8_t *buffer, size_t buffer_size, bool is_last = true, uint32_t start_offset = 0) { return produce(content_name, utils::MemBuf::copyBuffer(buffer, buffer_size), |