WebSocketStream.h

Go to the documentation of this file.

/*
 * SPDX-FileCopyrightText: 2006-2021 Istituto Italiano di Tecnologia (IIT)
 * SPDX-License-Identifier: BSD-3-Clause
 */
 
#ifndef YARP_WEBSOCKSTREAM_H
#define YARP_WEBSOCKSTREAM_H
 
#include "WebSocket/WebSocket.h"
 
#include <yarp/os/ManagedBytes.h>
#include <yarp/os/TwoWayStream.h>
 
 
class WebSocketStream :
        public yarp::os::TwoWayStream,
        public yarp::os::InputStream,
        public yarp::os::OutputStream
{
public:
    WebSocketStream(TwoWayStream* delegate);
 
    ~WebSocketStream() override;
 
    InputStream& getInputStream() override;
    OutputStream& getOutputStream() override;
 
    const yarp::os::Contact& getLocalAddress() const override;
    const yarp::os::Contact& getRemoteAddress() const override;
 
    void interrupt() override;
 
    void close() override;
 
    using yarp::os::InputStream::read;
    yarp::conf::ssize_t read(yarp::os::Bytes& bytesToRead) override;
 
    using yarp::os::OutputStream::write;
    void write(const yarp::os::Bytes& bytesToWrite) override;
 
    bool isOk() const override;
 
    void reset() override;
 
    void beginPacket() override;
    void endPacket() override;
 
private:
    yarp::os::TwoWayStream* delegate;
    yarp::os::Contact local, remote;
    yarp::os::ManagedBytes buffer;
    size_t currentHead {0};
 
    WebSocketFrameType getFrame(yarp::os::ManagedBytes& payload);
 
    static void makeFrame(WebSocketFrameType frame_type,
                   const yarp::os::Bytes& payload,
                   yarp::os::ManagedBytes& frame);
};
 
#endif // YARP_WEBSOCKSTREAM_H