1

import os

     2

import subprocess

     3

import sys

     4

import ctypes

     5

from threading import Thread

     6

import ctypes

     7

import time

     8

import re

     9

from targets.typemapping import LogLevelsDict

    10

    11

SDOAnswered = PLCBinary.SDOAnswered

    12

SDOAnswered.restype = None

    13

SDOAnswered.argtypes = []

    14

    15

SDOThread = None

    16

SDOProc = None

    17

Result = None

    18

    19

    20

def SDOThreadProc(*params):

    21

    global Result, SDOProc

    22

    if params[0] == "upload":

    23

        cmdfmt = "ethercat upload -p %d -t %s 0x%.4x 0x%.2x"

    24

    else:

    25

        cmdfmt = "ethercat download -p %d -t %s 0x%.4x 0x%.2x %s"

    26

    27

    command = cmdfmt % params[1:]

    28

    SDOProc = subprocess.Popen(command, stdout=subprocess.PIPE, shell=True)

    29

    res = SDOProc.wait()

    30

    output = SDOProc.communicate()[0]

    31

    32

    if params[0] == "upload":

    33

        Result = None

    34

        if res == 0:

    35

            if params[2] in ["float", "double"]:

    36

                Result = float(output)

    37

            elif params[2] in ["string", "octet_string", "unicode_string"]:

    38

                Result = output

    39

            else:

    40

                hex_value, dec_value = output.split()

    41

                if int(hex_value, 16) == int(dec_value):

    42

                    Result = int(dec_value)

    43

    else:

    44

        Result = res == 0

    45

    46

    SDOAnswered()

    47

    if res != 0:

    48

        PLCObject.LogMessage(

    49

            LogLevelsDict["WARNING"],

    50

            "%s : %s" % (command, output))

    51

    52

    53

def EthercatSDOUpload(pos, index, subindex, var_type):

    54

    global SDOThread

    55

    SDOThread = Thread(target=SDOThreadProc, args=["upload", pos, var_type, index, subindex])

    56

    SDOThread.start()

    57

    58

    59

def EthercatSDODownload(pos, index, subindex, var_type, value):

    60

    global SDOThread

    61

    SDOThread = Thread(target=SDOThreadProc, args=["download", pos, var_type, index, subindex, value])

    62

    SDOThread.start()

    63

    64

    65

def GetResult():

    66

    global Result

    67

    return Result

    68

    69

    70

KMSGPollThread = None

    71

StopKMSGThread = False

    72

    73

    74

def KMSGPollThreadProc():

    75

    """

    76

    Logs Kernel messages starting with EtherCAT

    77

    Uses GLibc wrapper to Linux syscall "klogctl"

    78

    Last 4 KB are polled, and lines compared to last

    79

    captured line to detect new lines

    80

    """

    81

    global StopKMSGThread

    82

    libc = ctypes.CDLL("libc.so.6")

    83

    klog = libc.klogctl

    84

    klog.argtypes = [ctypes.c_int, ctypes.c_char_p, ctypes.c_int]

    85

    klog.restype = ctypes.c_int

    86

    s = ctypes.create_string_buffer(4*1024)

    87

    last = None

    88

    while not StopKMSGThread:

    89

        bytes_to_read = klog(3, s, len(s)-1)

    90

        log = s.value[:bytes_to_read-1]

    91

        if last:

    92

            log = log.rpartition(last)[2]

    93

        if log:

    94

            last = log.rpartition('\n')[2]

    95

            for lvl, msg in re.findall(

    96

                            r'<(\d)>\[\s*\d*\.\d*\]\s*(EtherCAT\s*.*)$',

    97

                            log, re.MULTILINE):

    98

                PLCObject.LogMessage(

    99

                    LogLevelsDict[{

   100

                        "4": "WARNING",

   101

                        "3": "CRITICAL"}.get(lvl, "DEBUG")],

   102

                    msg)

   103

        time.sleep(0.5)

   104

   105

   106

def _runtime_etherlab_init():

   107

    global KMSGPollThread, StopKMSGThread

   108

    StopKMSGThread = False

   109

    KMSGPollThread = Thread(target=KMSGPollThreadProc)

   110

    KMSGPollThread.start()

   111

   112

   113

def _runtime_etherlab_cleanup():

   114

    global KMSGPollThread, StopKMSGThread, SDOProc, SDOThread

   115

    try:

   116

        os.kill(SDOProc.pid, SIGTERM)

   117

    except Exception:

   118

        pass

   119

    SDOThread = None

   120

    StopKMSGThread = True

   121

    KMSGPollThread = None