1

#!/usr/bin/env python

     2

# -*- coding: utf-8 -*-

     3

#

     4

# See COPYING.Runtime file for copyrights details.

     5

#

     6

     7

import ctypes

     8

from ctypes import *

     9

from datetime import timedelta as td

    10

    13

    14

    15

class IEC_STRING(Structure):

    16

    """

    17

    Must be changed according to changes in iec_types.h

    18

    """

    19

    _fields_ = [("len", c_uint8),

    20

                ("body", c_char * 126)]

    21

    22

    23

class IEC_TIME(Structure):

    24

    """

    25

    Must be changed according to changes in iec_types.h

    26

    """

    27

    _fields_ = [("s", c_long),   # tv_sec

    28

                ("ns", c_long)]  # tv_nsec

    29

    30

    31

def _t(t, u=lambda x: x.value, p=lambda t, x: t(x)):

    32

    return (t, u, p)

    33

    34

    35

def _ttime():

    36

    return (IEC_TIME,

    37

            lambda x: td(0, x.s, x.ns/1000.0),

    38

            lambda t, x: t(x.days * 24 * 3600 + x.seconds, x.microseconds*1000))

    39

    40

    41

SameEndianessTypeTranslator = {

    42

    "BOOL":       _t(c_uint8, lambda x: x.value != 0),

    43

    "STEP":       _t(c_uint8),

    44

    "TRANSITION": _t(c_uint8),

    45

    "ACTION":     _t(c_uint8),

    46

    "SINT":       _t(c_int8),

    47

    "USINT":      _t(c_uint8),

    48

    "BYTE":       _t(c_uint8),

    49

    "STRING":     (IEC_STRING,

    50

                   lambda x: x.body[:x.len],

    51

                   lambda t, x: t(len(x), x)),

    52

    "INT":        _t(c_int16),

    53

    "UINT":       _t(c_uint16),

    54

    "WORD":       _t(c_uint16),

    55

    "DINT":       _t(c_int32),

    56

    "UDINT":      _t(c_uint32),

    57

    "DWORD":      _t(c_uint32),

    58

    "LINT":       _t(c_int64),

    59

    "ULINT":      _t(c_uint64),

    60

    "LWORD":      _t(c_uint64),

    61

    "REAL":       _t(c_float),

    62

    "LREAL":      _t(c_double),

    63

    "TIME":       _ttime(),

    64

    "TOD":        _ttime(),

    65

    "DATE":       _ttime(),

    66

    "DT":         _ttime(),

    67

    }

    68

    69

SwapedEndianessTypeTranslator = {

    70

    # TODO

    71

    }

    72

    73

TypeTranslator = SameEndianessTypeTranslator

    74

    75

# Construct debugger natively supported types

    76

DebugTypesSize = dict([(key, sizeof(t)) for key, (t, p, u) in SameEndianessTypeTranslator.items() if t is not None])

    77

    78

    79

def UnpackDebugBuffer(buff, indexes):

    80

    res = []

    81

    buffoffset = 0

    82

    buffsize = len(buff)

    83

    buffptr = cast(ctypes.pythonapi.PyString_AsString(id(buff)), c_void_p).value

    84

    for iectype in indexes:

    85

        c_type, unpack_func, _pack_func = \

    86

            TypeTranslator.get(iectype, (None, None, None))

    87

    88

        cursor = c_void_p(buffptr + buffoffset)

    89

        if iectype == "STRING":

    90

            # strlen is stored in c_uint8 and sizeof(c_uint8) is 1

    91

            # first check we can read size

    92

            if (buffoffset + 1) <= buffsize:

    93

                size = 1 + cast(cursor,POINTER(c_type)).contents.len

    94

            else:

    95

                return None

    96

        else:

    97

            size = sizeof(c_type)

    98

    99

        if c_type is not None and (buffoffset + size) <= buffsize:

   100

            value = unpack_func(cast(cursor,

   101

                                     POINTER(c_type)).contents)

   102

            buffoffset += size

   103

            res.append(value)

   104

        else:

   105

            return None

   106

    if buffoffset and buffoffset == buffsize:

   107

        return res

   108

    return None