genetlink_linux.go 3.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168
  1. package netlink
  2. import (
  3. "fmt"
  4. "syscall"
  5. "github.com/vishvananda/netlink/nl"
  6. "golang.org/x/sys/unix"
  7. )
  8. type GenlOp struct {
  9. ID uint32
  10. Flags uint32
  11. }
  12. type GenlMulticastGroup struct {
  13. ID uint32
  14. Name string
  15. }
  16. type GenlFamily struct {
  17. ID uint16
  18. HdrSize uint32
  19. Name string
  20. Version uint32
  21. MaxAttr uint32
  22. Ops []GenlOp
  23. Groups []GenlMulticastGroup
  24. }
  25. func parseOps(b []byte) ([]GenlOp, error) {
  26. attrs, err := nl.ParseRouteAttr(b)
  27. if err != nil {
  28. return nil, err
  29. }
  30. ops := make([]GenlOp, 0, len(attrs))
  31. for _, a := range attrs {
  32. nattrs, err := nl.ParseRouteAttr(a.Value)
  33. if err != nil {
  34. return nil, err
  35. }
  36. var op GenlOp
  37. for _, na := range nattrs {
  38. switch na.Attr.Type {
  39. case nl.GENL_CTRL_ATTR_OP_ID:
  40. op.ID = native.Uint32(na.Value)
  41. case nl.GENL_CTRL_ATTR_OP_FLAGS:
  42. op.Flags = native.Uint32(na.Value)
  43. }
  44. }
  45. ops = append(ops, op)
  46. }
  47. return ops, nil
  48. }
  49. func parseMulticastGroups(b []byte) ([]GenlMulticastGroup, error) {
  50. attrs, err := nl.ParseRouteAttr(b)
  51. if err != nil {
  52. return nil, err
  53. }
  54. groups := make([]GenlMulticastGroup, 0, len(attrs))
  55. for _, a := range attrs {
  56. nattrs, err := nl.ParseRouteAttr(a.Value)
  57. if err != nil {
  58. return nil, err
  59. }
  60. var g GenlMulticastGroup
  61. for _, na := range nattrs {
  62. switch na.Attr.Type {
  63. case nl.GENL_CTRL_ATTR_MCAST_GRP_NAME:
  64. g.Name = nl.BytesToString(na.Value)
  65. case nl.GENL_CTRL_ATTR_MCAST_GRP_ID:
  66. g.ID = native.Uint32(na.Value)
  67. }
  68. }
  69. groups = append(groups, g)
  70. }
  71. return groups, nil
  72. }
  73. func (f *GenlFamily) parseAttributes(attrs []syscall.NetlinkRouteAttr) error {
  74. for _, a := range attrs {
  75. switch a.Attr.Type {
  76. case nl.GENL_CTRL_ATTR_FAMILY_NAME:
  77. f.Name = nl.BytesToString(a.Value)
  78. case nl.GENL_CTRL_ATTR_FAMILY_ID:
  79. f.ID = native.Uint16(a.Value)
  80. case nl.GENL_CTRL_ATTR_VERSION:
  81. f.Version = native.Uint32(a.Value)
  82. case nl.GENL_CTRL_ATTR_HDRSIZE:
  83. f.HdrSize = native.Uint32(a.Value)
  84. case nl.GENL_CTRL_ATTR_MAXATTR:
  85. f.MaxAttr = native.Uint32(a.Value)
  86. case nl.GENL_CTRL_ATTR_OPS:
  87. ops, err := parseOps(a.Value)
  88. if err != nil {
  89. return err
  90. }
  91. f.Ops = ops
  92. case nl.GENL_CTRL_ATTR_MCAST_GROUPS:
  93. groups, err := parseMulticastGroups(a.Value)
  94. if err != nil {
  95. return err
  96. }
  97. f.Groups = groups
  98. }
  99. }
  100. return nil
  101. }
  102. func parseFamilies(msgs [][]byte) ([]*GenlFamily, error) {
  103. families := make([]*GenlFamily, 0, len(msgs))
  104. for _, m := range msgs {
  105. attrs, err := nl.ParseRouteAttr(m[nl.SizeofGenlmsg:])
  106. if err != nil {
  107. return nil, err
  108. }
  109. family := &GenlFamily{}
  110. if err := family.parseAttributes(attrs); err != nil {
  111. return nil, err
  112. }
  113. families = append(families, family)
  114. }
  115. return families, nil
  116. }
  117. func (h *Handle) GenlFamilyList() ([]*GenlFamily, error) {
  118. msg := &nl.Genlmsg{
  119. Command: nl.GENL_CTRL_CMD_GETFAMILY,
  120. Version: nl.GENL_CTRL_VERSION,
  121. }
  122. req := h.newNetlinkRequest(nl.GENL_ID_CTRL, unix.NLM_F_DUMP)
  123. req.AddData(msg)
  124. msgs, err := req.Execute(unix.NETLINK_GENERIC, 0)
  125. if err != nil {
  126. return nil, err
  127. }
  128. return parseFamilies(msgs)
  129. }
  130. func GenlFamilyList() ([]*GenlFamily, error) {
  131. return pkgHandle.GenlFamilyList()
  132. }
  133. func (h *Handle) GenlFamilyGet(name string) (*GenlFamily, error) {
  134. msg := &nl.Genlmsg{
  135. Command: nl.GENL_CTRL_CMD_GETFAMILY,
  136. Version: nl.GENL_CTRL_VERSION,
  137. }
  138. req := h.newNetlinkRequest(nl.GENL_ID_CTRL, 0)
  139. req.AddData(msg)
  140. req.AddData(nl.NewRtAttr(nl.GENL_CTRL_ATTR_FAMILY_NAME, nl.ZeroTerminated(name)))
  141. msgs, err := req.Execute(unix.NETLINK_GENERIC, 0)
  142. if err != nil {
  143. return nil, err
  144. }
  145. families, err := parseFamilies(msgs)
  146. if len(families) != 1 {
  147. return nil, fmt.Errorf("invalid response for GENL_CTRL_CMD_GETFAMILY")
  148. }
  149. return families[0], nil
  150. }
  151. func GenlFamilyGet(name string) (*GenlFamily, error) {
  152. return pkgHandle.GenlFamilyGet(name)
  153. }