1

/*

     2

    Copyright Edouard TISSERANT 2024

     3

    See COPYING for details

     4

*/

     5

     6

#if !defined(_PLCObject_hpp_)

     7

#define _PLCObject_hpp_

     8

     9

#include <map>

    10

#include <vector>

    11

#include <mutex>

    12

#include <thread>

    13

    14

#include "blob.hpp"

    15

    16

#include "erpc_PLCObject_interface.hpp"

    17

    18

using namespace erpcShim;

    19

    20

#define FOR_EACH_PLC_SYMBOLS_DO(ACTION) \

    21

    ACTION(PLC_ID)\

    22

    ACTION(startPLC)\

    23

    ACTION(stopPLC)\

    24

    ACTION(ResetDebugVariables)\

    25

    ACTION(RegisterDebugVariable)\

    26

    ACTION(FreeDebugData)\

    27

    ACTION(GetDebugData)\

    28

    ACTION(suspendDebug)\

    29

    ACTION(resumeDebug)\

    30

    ACTION(ResetLogCount)\

    31

    ACTION(GetLogCount)\

    32

    ACTION(LogMessage)\

    33

    ACTION(GetLogMessage)

    34

    35

extern "C" {   

    36

    typedef struct s_PLCSyms{

    37

        uint8_t *PLC_ID;

    38

        int (*startPLC)(int argc,char **argv);

    39

        int (*stopPLC)(void);

    40

        void (*ResetDebugVariables)(void);

    41

        int (*RegisterDebugVariable)(unsigned int idx, void* force, size_t force_size);

    42

        void (*FreeDebugData)(void);

    43

        int (*GetDebugData)(unsigned int *tick, unsigned int *size, void **buffer);

    44

        int (*suspendDebug)(int disable);

    45

        void (*resumeDebug)(void);

    46

        void (*ResetLogCount)(void);

    47

        uint32_t (*GetLogCount)(uint8_t level);

    48

        int (*LogMessage)(uint8_t level, char* buf, uint32_t size);

    49

        uint32_t (*GetLogMessage)(uint8_t level, uint32_t msgidx, char* buf, uint32_t max_size, uint32_t* tick, uint32_t* tv_sec, uint32_t* tv_nsec);

    50

    } PLCSyms;

    51

}

    52

class PLCObject : public BeremizPLCObjectService_interface

    53

{

    54

    public:

    55

    56

        PLCObject(void);

    57

        ~PLCObject(void);

    58

    59

        // ERPC interface

    60

        uint32_t AppendChunkToBlob(const binary_t * data, const binary_t * blobID, binary_t * newBlobID);

    61

        uint32_t GetLogMessage(uint8_t level, uint32_t msgID, log_message * message);

    62

        uint32_t GetPLCID(PSKID * plcID);

    63

        uint32_t GetPLCstatus(PLCstatus * status);

    64

        uint32_t GetTraceVariables(uint32_t debugToken, TraceVariables * traces);

    65

        uint32_t MatchMD5(const char * MD5, bool * match);

    66

        uint32_t NewPLC(const char * md5sum, const binary_t * plcObjectBlobID, const list_extra_file_1_t * extrafiles, bool * success);

    67

        uint32_t PurgeBlobs(void);

    68

        uint32_t RepairPLC(void);

    69

        uint32_t ResetLogCount(void);

    70

        uint32_t SeedBlob(const binary_t * seed, binary_t * blobID);

    71

        uint32_t SetTraceVariablesList(const list_trace_order_1_t * orders, int32_t * debugtoken);

    72

        uint32_t StartPLC(void);

    73

        uint32_t StopPLC(bool * success);

    74

        uint32_t AutoLoad();

    75

    76

    private:

    77

        // A map of all the blobs

    78

        std::map<std::vector<uint8_t>, Blob*> m_mapBlobIDToBlob;

    79

    80

        // PLC object library handle

    81

        void * m_handle;

    82

    83

        // Shared object mutex

    84

        std::mutex m_PLClibMutex;

    85

    86

        // Symbols resolved from the PLC object

    87

        PLCSyms m_PLCSyms;

    88

    89

        // argc and argv for the PLC object

    90

        int m_argc;

    91

        char ** m_argv;

    92

    93

        // PLC status

    94

        PLCstatus m_status;

    95

    96

        // PSK

    97

        std::string m_PSK_ID;

    98

        std::string m_PSK_secret;

    99

   100

        // Debug token, used for consistency check of traces

   101

        uint32_t m_debugToken;

   102

   103

        // Trace thread

   104

        std::thread m_traceThread;

   105

   106

        // Trace thread mutex

   107

        std::mutex m_tracesMutex;

   108

   109

        // Trace double buffer

   110

        std::vector<trace_sample> m_traces;

   111

   112

        uint32_t BlobAsFile(const binary_t * BlobID, std::filesystem::path filename);

   113

        uint32_t LoadPLC(void);

   114

        uint32_t UnLoadPLC(void);

   115

        uint32_t LogMessage(uint8_t level, std::string message);

   116

        uint32_t PurgePLC(void);

   117

        void PurgeTraceBuffer(void);

   118

        void TraceThreadProc(void);

   119

};

   120

   121

#endif