from pyroute2.netlink import nla
from pyroute2.netlink import nlmsg_atoms


class ipoib(nla):
    prefix = 'IFLA_IPOIB_'

    nla_map = (('IFLA_IPOIB_UNSPEC', 'none'),
               ('IFLA_IPOIB_PKEY', 'uint16'),
               ('IFLA_IPOIB_MODE', 'mode'),
               ('IFLA_IPOIB_UMCAST', 'uint16'))

    class mode(nlmsg_atoms.uint16):
        value_map = {0: 'datagram',
                     1: 'connected'}
