tcp_stream_msg.h

Go to the documentation of this file.

// Copyright (C) 2022 Internet Systems Consortium, Inc. ("ISC")
//
// This Source Code Form is subject to the terms of the Mozilla Public
// License, v. 2.0. If a copy of the MPL was not distributed with this
// file, You can obtain one at http://mozilla.org/MPL/2.0/.
 
#ifndef TCP_STREAM_MSG_H
#define TCP_STREAM_MSG_H
 
#include <tcp/tcp_connection.h>
#include <boost/shared_ptr.hpp>
 
#include <array>
#include <functional>
#include <string>
 
namespace isc {
namespace tcp {
 
class TcpStreamRequest : public TcpRequest {
public:
    TcpStreamRequest() : expected_size_(0) {
    }
 
    virtual ~TcpStreamRequest() {
    }
 
    virtual size_t postBuffer(const void* buf, const size_t nbytes);
 
    virtual bool needData() const;
 
    virtual std::string logFormatRequest(const size_t limit = 0) const;
 
    virtual void unpack();
 
    size_t getRequestSize() const {
        return (request_.size());
    }
 
    const uint8_t* getRequest() const {
        if (request_.empty()) {
            isc_throw(InvalidOperation, "TcpStreamRequest::getRequest()"
                                        " - cannot access empty request");
        }
 
        return (request_.data());
    }
 
    std::string getRequestString() const {
        return (std::string(request_.begin(), request_.end()));
    };
 
protected:
    std::vector<uint8_t> request_;
 
private:
    size_t expected_size_;
};
 
typedef boost::shared_ptr<TcpStreamRequest> TcpStreamRequestPtr;
 
class TcpStreamResponse : public TcpResponse {
public:
    TcpStreamResponse() {};
 
    virtual ~TcpStreamResponse() {};
 
    virtual void setResponseData(const uint8_t* data, size_t length);
 
    virtual void appendResponseData(const uint8_t* data, size_t length);
 
    virtual void setResponseData(const std::string& str);
 
    virtual void appendResponseData(const std::string& str);
 
    virtual void pack();
 
    std::string getResponseString() const {
        return (std::string(response_.begin(), response_.end()));
    };
 
private:
    std::vector<uint8_t> response_;
 
};
 
typedef boost::shared_ptr<TcpStreamResponse> TcpStreamResponsePtr;
 
} // end of namespace isc::tcp
} // end of namespace isc
 
#endif  // TCP_STREAM_MSG_H