Ejemplo n.º 1
0
 def test_sizeof(self):
     self.assertEqual(sizeof('int'), 4)
     define('INIT_THREAD_SIZE', 2048 * sizeof('long'))
     self.assertEqual(cstruct.DEFINES['INIT_THREAD_SIZE'], 16384)
     self.assertEqual(sizeof('struct Position'), 3)
     self.assertEqual(sizeof('struct Position'), len(Position))
     self.assertEqual(sizeof(Position), 3)
     self.assertRaises(KeyError, lambda: sizeof('bla'))
     self.assertRaises(KeyError, lambda: sizeof('struct Bla'))
Ejemplo n.º 2
0
from cstruct import define, typedef, CStruct
from keyplus.constants import *

class CStructWithBytes(CStruct):
    def to_bytes(self):
        return bytearray(self.pack())

typedef('uint8', 'uint8_t')
typedef('uint16', 'uint16_t')
typedef('uint32', 'uint32_t')

typedef('int8', 'int8_t')
typedef('int16', 'int16_t')
typedef('int32', 'int32_t')

define('MAX_NUM_KEYBOARDS', 64)
define('MAX_NUM_DEVICES', 64)
define('AES_KEY_LEN', 16)
define('NRF_ADDR_LEN', 5)

def make_bit_field_variables(class_obj, field_list):
    """
    This function uses the python built-in `property()` to create attributes
    to access the bit fields of a function directly.

    The field_list should contain a list of properties where each property is
    defined as a tuple of the form:
        (bit_mask_variable, bit_field_name, bit_field_mask)

    After this function has been called, you should be able to get and set
    the bit fields using the `bit_field_name`.
Ejemplo n.º 3
0
 def test_define(self):
     define('A', 10)
     self.assertEqual(cstruct.DEFINES['A'], 10)
     undef('A')
     self.assertRaises(KeyError, lambda: cstruct.DEFINES['A'])
Ejemplo n.º 4
0
# all copies or substantial portions of the Software.
#
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL THE
# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
# FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
# IN THE SOFTWARE.
#
#*****************************************************************************

from unittest import TestCase, main
from cstruct import (sizeof, typedef, define, CStruct, NATIVE_ORDER)

define("UT_NAMESIZE", 32)
define("UT_LINESIZE", 32)
define("UT_HOSTSIZE", 256)

typedef("int", "pid_t")
typedef("long", "time_t")


class ExitStatus(CStruct):
    __def__ = """
        struct {
            short   e_termination;      /* Process termination status.  */
            short   e_exit;             /* Process exit status.  */
        }
    """
Ejemplo n.º 5
0
import struct

class TestUnion(cstruct.CStruct):
    __byte_order__ = cstruct.LITTLE_ENDIAN
    __def__ = """
        union {
            uint8   a;
            uint8   a1;
            uint16  b;
            uint32  c;
            struct Partition d;
            struct Partition e[4];
        }
    """

define('INIT_THREAD_SIZE', 2048 * sizeof('long'))

# class TaskUnion(cstruct.CStruct):
#     __def__ = """
#         union {
#             struct task_struct task;
#             unsigned long stack[INIT_TASK_SIZE/sizeof(long)];
#         }
#     """

class TestCaseUnion(TestCase):

    # def test_sizeof(self):
    #     self.assertEqual(cstruct.DEFINES['INIT_THREAD_SIZE'], 16384)
    #     self.assertEqual(sizeof('struct TestUnion'), 64)
    #