uORBFastRpcChannel.hpp

/****************************************************************************
 *
 * Copyright (C) 2015 Mark Charlebois. All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 *
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in
 *    the documentation and/or other materials provided with the
 *    distribution.
 * 3. Neither the name PX4 nor the names of its contributors may be
 *    used to endorse or promote products derived from this software
 *    without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
 * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
 * COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
 * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS
 * OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
 * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
 * ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
 * POSSIBILITY OF SUCH DAMAGE.
 *
 ****************************************************************************/
#ifndef _uORBFastRpcChannel_hpp_
#define _uORBFastRpcChannel_hpp_

#include <stdint.h>
#include <string>
#include <list>
#include "uORB/uORBCommunicator.hpp"
#include <semaphore.h>
#include <set>
#include <px4_sem.h>

namespace uORB
{
class FastRpcChannel;
}

class uORB::FastRpcChannel : public uORBCommunicator::IChannel
{
public:
    static uORB::FastRpcChannel *GetInstance()
    {
        return &(_Instance);
    }

    virtual int16_t topic_advertised(const char *messageName);

    virtual int16_t topic_unadvertised(const char *messageName);

    virtual int16_t add_subscription(const char *messageName, int32_t msgRateInHz);


    virtual int16_t remove_subscription(const char *messageName);

    virtual int16_t register_handler(uORBCommunicator::IChannelRxHandler *handler);


    //=========================================================================
    //     INTERFACES FOR Data messages
    //=========================================================================

    virtual int16_t send_message(const char *messageName, int32_t length, uint8_t *data);

    //Function to return the data to krait.
    int16_t get_data
    (
        int32_t *msg_type,
        char *topic_name,
        int32_t topic_name_len,
        uint8_t *data,
        int32_t data_len_in_bytes,
        int32_t *bytes_returned
    );

    int16_t get_bulk_data(uint8_t *buffer, int32_t max_size_in_bytes, int32_t *returned_bytes, int32_t *topic_count);

    // function to check if there are subscribers for a topic on adsp.
    int16_t is_subscriber_present(const char *messageName, int32_t *status);

    // function to release the blocking semaphore for get_data method.
    int16_t unblock_get_data_method();

    uORBCommunicator::IChannelRxHandler *GetRxHandler()
    {
        return _RxHandler;
    }

    void AddRemoteSubscriber(const std::string &messageName)
    {
        _RemoteSubscribers.insert(messageName);
    }
    void RemoveRemoteSubscriber(const std::string &messageName)
    {
        _RemoteSubscribers.erase(messageName);
    }

private: // data members
    static uORB::FastRpcChannel _Instance;
    uORBCommunicator::IChannelRxHandler *_RxHandler;

    static const int32_t _MAX_MSG_QUEUE_SIZE = 100;
    static const int32_t _CONTROL_MSG_TYPE_ADD_SUBSCRIBER = 1;
    static const int32_t _CONTROL_MSG_TYPE_REMOVE_SUBSCRIBER = 2;
    static const int32_t _DATA_MSG_TYPE = 3;
    static const int32_t _CONTROL_MSG_TYPE_ADVERTISE = 4;
    static const int32_t _CONTROL_MSG_TYPE_UNADVERTISE = 5;

    static const int32_t _PACKET_FIELD_TOPIC_NAME_LEN_SIZE_IN_BYTES = 2;
    static const int32_t _PACKET_FIELD_DATA_LEN_IN_BYTES = 2;
    static const int32_t _PACKET_HEADER_SIZE = 1 + //first byte is the MSG Type
            _PACKET_FIELD_TOPIC_NAME_LEN_SIZE_IN_BYTES + _PACKET_FIELD_DATA_LEN_IN_BYTES;

    struct FastRpcDataMsg {
        int32_t     _MaxBufferSize;
        int32_t     _Length;
        uint8_t    *_Buffer;
        std::string _MsgName;
    };

    struct FastRpcControlMsg {
        int32_t _Type;
        std::string _MsgName;
    };

    struct BulkTransferHeader {
        uint16_t _MsgType;
        uint16_t _MsgNameLen;
        uint16_t _DataLen;
    };


    struct FastRpcDataMsg _DataMsgQueue[ _MAX_MSG_QUEUE_SIZE ];
    int32_t _DataQInIndex;
    int32_t _DataQOutIndex;

    struct FastRpcControlMsg _ControlMsgQueue[ _MAX_MSG_QUEUE_SIZE ];
    int32_t _ControlQInIndex;
    int32_t _ControlQOutIndex;

    std::list<std::string> _Subscribers;

    //utility classes
    class Mutex
    {
    public:
        Mutex()
        {
            sem_init(&_Sem, 0, 1);
        }
        ~Mutex()
        {
            sem_destroy(&_Sem);
        }
        void lock()
        {
            sem_wait(&_Sem);
        }
        void unlock()
        {
            sem_post(&_Sem);
        }
    private:
        sem_t _Sem;

        Mutex(const Mutex &);

        Mutex &operator=(const Mutex &);
    };

    class Semaphore
    {
    public:
        Semaphore()
        {
            sem_init(&_Sem, 0, 0);
            /* _Sem use case is a signal */
            px4_sem_setprotocol(&_Sem, SEM_PRIO_NONE);
        }
        ~Semaphore()
        {
            sem_destroy(&_Sem);
        }
        void post()
        {
            sem_post(&_Sem);
        }
        void wait()
        {
            sem_wait(&_Sem);
        }
    private:
        sem_t _Sem;
        Semaphore(const Semaphore &);
        Semaphore &operator=(const Semaphore &);

    };

    Mutex _QueueMutex;
    Semaphore _DataAvailableSemaphore;

private://class members.
    FastRpcChannel();

    void check_and_expand_data_buffer(int32_t index, int32_t length);

    bool IsControlQFull();
    bool IsControlQEmpty();
    bool IsDataQFull();
    bool IsDataQEmpty();
    int32_t DataQSize();
    int32_t ControlQSize();

    int32_t get_msg_size_at(bool isData, int32_t index);
    int32_t copy_msg_to_buffer(bool isData, int32_t src_index, uint8_t *dst_buffer, int32_t offset, int32_t dst_buffer_len);
    int16_t control_msg_queue_add(int32_t msgtype, const char *messageName);

    std::set<std::string> _RemoteSubscribers;
};

#endif /* _uORBFastRpcChannel_hpp_ */