4 files changed, 128 insertions, 1 deletions

diff --git a/tests/atf_python/sys/netlink/attrs.py b/tests/atf_python/sys/netlink/attrs.py
index f6fe9ee43c98..58fbab7fc8db 100644
--- a/tests/atf_python/sys/netlink/attrs.py
+++ b/tests/atf_python/sys/netlink/attrs.py
@@ -7,6 +7,8 @@ from atf_python.sys.netlink.utils import enum_or_int
 
 
 class NlAttr(object):
+    HDR_LEN = 4  # sizeof(struct nlattr)
+
     def __init__(self, nla_type, data):
         if isinstance(nla_type, Enum):
             self._nla_type = nla_type.value
diff --git a/tests/atf_python/sys/netlink/base_headers.py b/tests/atf_python/sys/netlink/base_headers.py
index 759d8827fb3c..71771a249b3d 100644
--- a/tests/atf_python/sys/netlink/base_headers.py
+++ b/tests/atf_python/sys/netlink/base_headers.py
@@ -15,6 +15,13 @@ class Nlmsghdr(Structure):
     ]
 
 
+class Nlattr(Structure):
+    _fields_ = [
+        ("nla_len", c_ushort),
+        ("nla_type", c_ushort),
+    ]
+
+
 class NlMsgType(Enum):
     NLMSG_NOOP = 1
     NLMSG_ERROR = 2
diff --git a/tests/atf_python/sys/netlink/netlink.py b/tests/atf_python/sys/netlink/netlink.py
index f813727d55b4..4bdefc2d5014 100644
--- a/tests/atf_python/sys/netlink/netlink.py
+++ b/tests/atf_python/sys/netlink/netlink.py
@@ -22,8 +22,8 @@ from atf_python.sys.netlink.message import BaseNetlinkMessage
 from atf_python.sys.netlink.message import NlMsgCategory
 from atf_python.sys.netlink.message import NlMsgProps
 from atf_python.sys.netlink.message import StdNetlinkMessage
-from atf_python.sys.netlink.netlink_generic import GenlCtrlMsgType
 from atf_python.sys.netlink.netlink_generic import GenlCtrlAttrType
+from atf_python.sys.netlink.netlink_generic import GenlCtrlMsgType
 from atf_python.sys.netlink.netlink_generic import handler_classes as genl_classes
 from atf_python.sys.netlink.netlink_route import handler_classes as rt_classes
 from atf_python.sys.netlink.utils import align4
diff --git a/tests/atf_python/sys/netlink/netlink_generic.py b/tests/atf_python/sys/netlink/netlink_generic.py
index ee75d5bf37f3..06dc8704fe07 100644
--- a/tests/atf_python/sys/netlink/netlink_generic.py
+++ b/tests/atf_python/sys/netlink/netlink_generic.py
@@ -1,10 +1,16 @@
 #!/usr/local/bin/python3
+from ctypes import c_int64
+from ctypes import c_long
 from ctypes import sizeof
+from ctypes import Structure
 from enum import Enum
+import struct
 
+from atf_python.sys.netlink.attrs import NlAttr
 from atf_python.sys.netlink.attrs import NlAttrStr
 from atf_python.sys.netlink.attrs import NlAttrU16
 from atf_python.sys.netlink.attrs import NlAttrU32
+from atf_python.sys.netlink.attrs import NlAttrU8
 from atf_python.sys.netlink.base_headers import GenlMsgHdr
 from atf_python.sys.netlink.message import NlMsgCategory
 from atf_python.sys.netlink.message import NlMsgProps
@@ -105,6 +111,118 @@ class NetlinkGenlCtrlMessage(NetlinkGenlMessage):
     family_name = GenlCtrlFamilyName
 
 
+KtestFamilyName = "ktest"
+
+
+class KtestMsgType(Enum):
+    KTEST_CMD_UNSPEC = 0
+    KTEST_CMD_LIST = 1
+    KTEST_CMD_RUN = 2
+    KTEST_CMD_NEWTEST = 3
+    KTEST_CMD_NEWMESSAGE = 4
+
+
+class KtestAttrType(Enum):
+    KTEST_ATTR_MOD_NAME = 1
+    KTEST_ATTR_TEST_NAME = 2
+    KTEST_ATTR_TEST_DESCR = 3
+    KTEST_ATTR_TEST_META = 4
+
+
+class KtestLogMsgType(Enum):
+    KTEST_MSG_START = 1
+    KTEST_MSG_END = 2
+    KTEST_MSG_LOG = 3
+    KTEST_MSG_FAIL = 4
+
+
+class KtestMsgAttrType(Enum):
+    KTEST_MSG_ATTR_TS = 1
+    KTEST_MSG_ATTR_FUNC = 2
+    KTEST_MSG_ATTR_FILE = 3
+    KTEST_MSG_ATTR_LINE = 4
+    KTEST_MSG_ATTR_TEXT = 5
+    KTEST_MSG_ATTR_LEVEL = 6
+    KTEST_MSG_ATTR_META = 7
+
+
+class timespec(Structure):
+    _fields_ = [
+        ("tv_sec", c_int64),
+        ("tv_nsec", c_long),
+    ]
+
+
+class NlAttrTS(NlAttr):
+    DATA_LEN = sizeof(timespec)
+
+    def __init__(self, nla_type, val):
+        self.ts = val
+        super().__init__(nla_type, b"")
+
+    @property
+    def nla_len(self):
+        return NlAttr.HDR_LEN + self.DATA_LEN
+
+    def _print_attr_value(self):
+        return " tv_sec={} tv_nsec={}".format(self.ts.tv_sec, self.ts.tv_nsec)
+
+    @staticmethod
+    def _validate(data):
+        assert len(data) == NlAttr.HDR_LEN + NlAttrTS.DATA_LEN
+        nla_len, nla_type = struct.unpack("@HH", data[:NlAttr.HDR_LEN])
+        assert nla_len == NlAttr.HDR_LEN + NlAttrTS.DATA_LEN
+
+    @classmethod
+    def _parse(cls, data):
+        nla_len, nla_type = struct.unpack("@HH", data[:NlAttr.HDR_LEN])
+        val = timespec.from_buffer_copy(data[NlAttr.HDR_LEN:])
+        return cls(nla_type, val)
+
+    def __bytes__(self):
+        return self._to_bytes(bytes(self.ts))
+
+
+ktest_info_attrs = prepare_attrs_map(
+    [
+        AttrDescr(KtestAttrType.KTEST_ATTR_MOD_NAME, NlAttrStr),
+        AttrDescr(KtestAttrType.KTEST_ATTR_TEST_NAME, NlAttrStr),
+        AttrDescr(KtestAttrType.KTEST_ATTR_TEST_DESCR, NlAttrStr),
+    ]
+)
+
+
+ktest_msg_attrs = prepare_attrs_map(
+    [
+        AttrDescr(KtestMsgAttrType.KTEST_MSG_ATTR_FUNC, NlAttrStr),
+        AttrDescr(KtestMsgAttrType.KTEST_MSG_ATTR_FILE, NlAttrStr),
+        AttrDescr(KtestMsgAttrType.KTEST_MSG_ATTR_LINE, NlAttrU32),
+        AttrDescr(KtestMsgAttrType.KTEST_MSG_ATTR_TEXT, NlAttrStr),
+        AttrDescr(KtestMsgAttrType.KTEST_MSG_ATTR_LEVEL, NlAttrU8),
+        AttrDescr(KtestMsgAttrType.KTEST_MSG_ATTR_TS, NlAttrTS),
+    ]
+)
+
+
+class KtestInfoMessage(NetlinkGenlMessage):
+    messages = [
+        NlMsgProps(KtestMsgType.KTEST_CMD_LIST, NlMsgCategory.GET),
+        NlMsgProps(KtestMsgType.KTEST_CMD_RUN, NlMsgCategory.NEW),
+        NlMsgProps(KtestMsgType.KTEST_CMD_NEWTEST, NlMsgCategory.NEW),
+    ]
+    nl_attrs_map = ktest_info_attrs
+    family_name = KtestFamilyName
+
+
+class KtestMsgMessage(NetlinkGenlMessage):
+    messages = [
+        NlMsgProps(KtestMsgType.KTEST_CMD_NEWMESSAGE, NlMsgCategory.NEW),
+    ]
+    nl_attrs_map = ktest_msg_attrs
+    family_name = KtestFamilyName
+
+
 handler_classes = {
     GenlCtrlFamilyName: [NetlinkGenlCtrlMessage],
+    KtestFamilyName: [KtestInfoMessage, KtestMsgMessage],
 }