gtp_linux.go 6.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239
  1. package netlink
  2. import (
  3. "fmt"
  4. "net"
  5. "strings"
  6. "syscall"
  7. "github.com/vishvananda/netlink/nl"
  8. "golang.org/x/sys/unix"
  9. )
  10. type PDP struct {
  11. Version uint32
  12. TID uint64
  13. PeerAddress net.IP
  14. MSAddress net.IP
  15. Flow uint16
  16. NetNSFD uint32
  17. ITEI uint32
  18. OTEI uint32
  19. }
  20. func (pdp *PDP) String() string {
  21. elems := []string{}
  22. elems = append(elems, fmt.Sprintf("Version: %d", pdp.Version))
  23. if pdp.Version == 0 {
  24. elems = append(elems, fmt.Sprintf("TID: %d", pdp.TID))
  25. } else if pdp.Version == 1 {
  26. elems = append(elems, fmt.Sprintf("TEI: %d/%d", pdp.ITEI, pdp.OTEI))
  27. }
  28. elems = append(elems, fmt.Sprintf("MS-Address: %s", pdp.MSAddress))
  29. elems = append(elems, fmt.Sprintf("Peer-Address: %s", pdp.PeerAddress))
  30. return fmt.Sprintf("{%s}", strings.Join(elems, " "))
  31. }
  32. func (p *PDP) parseAttributes(attrs []syscall.NetlinkRouteAttr) error {
  33. for _, a := range attrs {
  34. switch a.Attr.Type {
  35. case nl.GENL_GTP_ATTR_VERSION:
  36. p.Version = native.Uint32(a.Value)
  37. case nl.GENL_GTP_ATTR_TID:
  38. p.TID = native.Uint64(a.Value)
  39. case nl.GENL_GTP_ATTR_PEER_ADDRESS:
  40. p.PeerAddress = net.IP(a.Value)
  41. case nl.GENL_GTP_ATTR_MS_ADDRESS:
  42. p.MSAddress = net.IP(a.Value)
  43. case nl.GENL_GTP_ATTR_FLOW:
  44. p.Flow = native.Uint16(a.Value)
  45. case nl.GENL_GTP_ATTR_NET_NS_FD:
  46. p.NetNSFD = native.Uint32(a.Value)
  47. case nl.GENL_GTP_ATTR_I_TEI:
  48. p.ITEI = native.Uint32(a.Value)
  49. case nl.GENL_GTP_ATTR_O_TEI:
  50. p.OTEI = native.Uint32(a.Value)
  51. }
  52. }
  53. return nil
  54. }
  55. func parsePDP(msgs [][]byte) ([]*PDP, error) {
  56. pdps := make([]*PDP, 0, len(msgs))
  57. for _, m := range msgs {
  58. attrs, err := nl.ParseRouteAttr(m[nl.SizeofGenlmsg:])
  59. if err != nil {
  60. return nil, err
  61. }
  62. pdp := &PDP{}
  63. if err := pdp.parseAttributes(attrs); err != nil {
  64. return nil, err
  65. }
  66. pdps = append(pdps, pdp)
  67. }
  68. return pdps, nil
  69. }
  70. func (h *Handle) GTPPDPList() ([]*PDP, error) {
  71. f, err := h.GenlFamilyGet(nl.GENL_GTP_NAME)
  72. if err != nil {
  73. return nil, err
  74. }
  75. msg := &nl.Genlmsg{
  76. Command: nl.GENL_GTP_CMD_GETPDP,
  77. Version: nl.GENL_GTP_VERSION,
  78. }
  79. req := h.newNetlinkRequest(int(f.ID), unix.NLM_F_DUMP)
  80. req.AddData(msg)
  81. msgs, err := req.Execute(unix.NETLINK_GENERIC, 0)
  82. if err != nil {
  83. return nil, err
  84. }
  85. return parsePDP(msgs)
  86. }
  87. func GTPPDPList() ([]*PDP, error) {
  88. return pkgHandle.GTPPDPList()
  89. }
  90. func gtpPDPGet(req *nl.NetlinkRequest) (*PDP, error) {
  91. msgs, err := req.Execute(unix.NETLINK_GENERIC, 0)
  92. if err != nil {
  93. return nil, err
  94. }
  95. pdps, err := parsePDP(msgs)
  96. if err != nil {
  97. return nil, err
  98. }
  99. if len(pdps) != 1 {
  100. return nil, fmt.Errorf("invalid reqponse for GENL_GTP_CMD_GETPDP")
  101. }
  102. return pdps[0], nil
  103. }
  104. func (h *Handle) GTPPDPByTID(link Link, tid int) (*PDP, error) {
  105. f, err := h.GenlFamilyGet(nl.GENL_GTP_NAME)
  106. if err != nil {
  107. return nil, err
  108. }
  109. msg := &nl.Genlmsg{
  110. Command: nl.GENL_GTP_CMD_GETPDP,
  111. Version: nl.GENL_GTP_VERSION,
  112. }
  113. req := h.newNetlinkRequest(int(f.ID), 0)
  114. req.AddData(msg)
  115. req.AddData(nl.NewRtAttr(nl.GENL_GTP_ATTR_VERSION, nl.Uint32Attr(0)))
  116. req.AddData(nl.NewRtAttr(nl.GENL_GTP_ATTR_LINK, nl.Uint32Attr(uint32(link.Attrs().Index))))
  117. req.AddData(nl.NewRtAttr(nl.GENL_GTP_ATTR_TID, nl.Uint64Attr(uint64(tid))))
  118. return gtpPDPGet(req)
  119. }
  120. func GTPPDPByTID(link Link, tid int) (*PDP, error) {
  121. return pkgHandle.GTPPDPByTID(link, tid)
  122. }
  123. func (h *Handle) GTPPDPByITEI(link Link, itei int) (*PDP, error) {
  124. f, err := h.GenlFamilyGet(nl.GENL_GTP_NAME)
  125. if err != nil {
  126. return nil, err
  127. }
  128. msg := &nl.Genlmsg{
  129. Command: nl.GENL_GTP_CMD_GETPDP,
  130. Version: nl.GENL_GTP_VERSION,
  131. }
  132. req := h.newNetlinkRequest(int(f.ID), 0)
  133. req.AddData(msg)
  134. req.AddData(nl.NewRtAttr(nl.GENL_GTP_ATTR_VERSION, nl.Uint32Attr(1)))
  135. req.AddData(nl.NewRtAttr(nl.GENL_GTP_ATTR_LINK, nl.Uint32Attr(uint32(link.Attrs().Index))))
  136. req.AddData(nl.NewRtAttr(nl.GENL_GTP_ATTR_I_TEI, nl.Uint32Attr(uint32(itei))))
  137. return gtpPDPGet(req)
  138. }
  139. func GTPPDPByITEI(link Link, itei int) (*PDP, error) {
  140. return pkgHandle.GTPPDPByITEI(link, itei)
  141. }
  142. func (h *Handle) GTPPDPByMSAddress(link Link, addr net.IP) (*PDP, error) {
  143. f, err := h.GenlFamilyGet(nl.GENL_GTP_NAME)
  144. if err != nil {
  145. return nil, err
  146. }
  147. msg := &nl.Genlmsg{
  148. Command: nl.GENL_GTP_CMD_GETPDP,
  149. Version: nl.GENL_GTP_VERSION,
  150. }
  151. req := h.newNetlinkRequest(int(f.ID), 0)
  152. req.AddData(msg)
  153. req.AddData(nl.NewRtAttr(nl.GENL_GTP_ATTR_VERSION, nl.Uint32Attr(0)))
  154. req.AddData(nl.NewRtAttr(nl.GENL_GTP_ATTR_LINK, nl.Uint32Attr(uint32(link.Attrs().Index))))
  155. req.AddData(nl.NewRtAttr(nl.GENL_GTP_ATTR_MS_ADDRESS, []byte(addr.To4())))
  156. return gtpPDPGet(req)
  157. }
  158. func GTPPDPByMSAddress(link Link, addr net.IP) (*PDP, error) {
  159. return pkgHandle.GTPPDPByMSAddress(link, addr)
  160. }
  161. func (h *Handle) GTPPDPAdd(link Link, pdp *PDP) error {
  162. f, err := h.GenlFamilyGet(nl.GENL_GTP_NAME)
  163. if err != nil {
  164. return err
  165. }
  166. msg := &nl.Genlmsg{
  167. Command: nl.GENL_GTP_CMD_NEWPDP,
  168. Version: nl.GENL_GTP_VERSION,
  169. }
  170. req := h.newNetlinkRequest(int(f.ID), unix.NLM_F_EXCL|unix.NLM_F_ACK)
  171. req.AddData(msg)
  172. req.AddData(nl.NewRtAttr(nl.GENL_GTP_ATTR_VERSION, nl.Uint32Attr(pdp.Version)))
  173. req.AddData(nl.NewRtAttr(nl.GENL_GTP_ATTR_LINK, nl.Uint32Attr(uint32(link.Attrs().Index))))
  174. req.AddData(nl.NewRtAttr(nl.GENL_GTP_ATTR_PEER_ADDRESS, []byte(pdp.PeerAddress.To4())))
  175. req.AddData(nl.NewRtAttr(nl.GENL_GTP_ATTR_MS_ADDRESS, []byte(pdp.MSAddress.To4())))
  176. switch pdp.Version {
  177. case 0:
  178. req.AddData(nl.NewRtAttr(nl.GENL_GTP_ATTR_TID, nl.Uint64Attr(pdp.TID)))
  179. req.AddData(nl.NewRtAttr(nl.GENL_GTP_ATTR_FLOW, nl.Uint16Attr(pdp.Flow)))
  180. case 1:
  181. req.AddData(nl.NewRtAttr(nl.GENL_GTP_ATTR_I_TEI, nl.Uint32Attr(pdp.ITEI)))
  182. req.AddData(nl.NewRtAttr(nl.GENL_GTP_ATTR_O_TEI, nl.Uint32Attr(pdp.OTEI)))
  183. default:
  184. return fmt.Errorf("unsupported GTP version: %d", pdp.Version)
  185. }
  186. _, err = req.Execute(unix.NETLINK_GENERIC, 0)
  187. return err
  188. }
  189. func GTPPDPAdd(link Link, pdp *PDP) error {
  190. return pkgHandle.GTPPDPAdd(link, pdp)
  191. }
  192. func (h *Handle) GTPPDPDel(link Link, pdp *PDP) error {
  193. f, err := h.GenlFamilyGet(nl.GENL_GTP_NAME)
  194. if err != nil {
  195. return err
  196. }
  197. msg := &nl.Genlmsg{
  198. Command: nl.GENL_GTP_CMD_DELPDP,
  199. Version: nl.GENL_GTP_VERSION,
  200. }
  201. req := h.newNetlinkRequest(int(f.ID), unix.NLM_F_EXCL|unix.NLM_F_ACK)
  202. req.AddData(msg)
  203. req.AddData(nl.NewRtAttr(nl.GENL_GTP_ATTR_VERSION, nl.Uint32Attr(pdp.Version)))
  204. req.AddData(nl.NewRtAttr(nl.GENL_GTP_ATTR_LINK, nl.Uint32Attr(uint32(link.Attrs().Index))))
  205. switch pdp.Version {
  206. case 0:
  207. req.AddData(nl.NewRtAttr(nl.GENL_GTP_ATTR_TID, nl.Uint64Attr(pdp.TID)))
  208. case 1:
  209. req.AddData(nl.NewRtAttr(nl.GENL_GTP_ATTR_I_TEI, nl.Uint32Attr(pdp.ITEI)))
  210. default:
  211. return fmt.Errorf("unsupported GTP version: %d", pdp.Version)
  212. }
  213. _, err = req.Execute(unix.NETLINK_GENERIC, 0)
  214. return err
  215. }
  216. func GTPPDPDel(link Link, pdp *PDP) error {
  217. return pkgHandle.GTPPDPDel(link, pdp)
  218. }