forked from github/multus-cni
ac21a96804
gomodule is still in progress to migrate for now, hence multus team decide to keep vendor directory to support build without gomodule.
239 lines
6.7 KiB
Go
239 lines
6.7 KiB
Go
package netlink
|
|
|
|
import (
|
|
"fmt"
|
|
"net"
|
|
"strings"
|
|
"syscall"
|
|
|
|
"github.com/vishvananda/netlink/nl"
|
|
)
|
|
|
|
type PDP struct {
|
|
Version uint32
|
|
TID uint64
|
|
PeerAddress net.IP
|
|
MSAddress net.IP
|
|
Flow uint16
|
|
NetNSFD uint32
|
|
ITEI uint32
|
|
OTEI uint32
|
|
}
|
|
|
|
func (pdp *PDP) String() string {
|
|
elems := []string{}
|
|
elems = append(elems, fmt.Sprintf("Version: %d", pdp.Version))
|
|
if pdp.Version == 0 {
|
|
elems = append(elems, fmt.Sprintf("TID: %d", pdp.TID))
|
|
} else if pdp.Version == 1 {
|
|
elems = append(elems, fmt.Sprintf("TEI: %d/%d", pdp.ITEI, pdp.OTEI))
|
|
}
|
|
elems = append(elems, fmt.Sprintf("MS-Address: %s", pdp.MSAddress))
|
|
elems = append(elems, fmt.Sprintf("Peer-Address: %s", pdp.PeerAddress))
|
|
return fmt.Sprintf("{%s}", strings.Join(elems, " "))
|
|
}
|
|
|
|
func (p *PDP) parseAttributes(attrs []syscall.NetlinkRouteAttr) error {
|
|
for _, a := range attrs {
|
|
switch a.Attr.Type {
|
|
case nl.GENL_GTP_ATTR_VERSION:
|
|
p.Version = native.Uint32(a.Value)
|
|
case nl.GENL_GTP_ATTR_TID:
|
|
p.TID = native.Uint64(a.Value)
|
|
case nl.GENL_GTP_ATTR_PEER_ADDRESS:
|
|
p.PeerAddress = net.IP(a.Value)
|
|
case nl.GENL_GTP_ATTR_MS_ADDRESS:
|
|
p.MSAddress = net.IP(a.Value)
|
|
case nl.GENL_GTP_ATTR_FLOW:
|
|
p.Flow = native.Uint16(a.Value)
|
|
case nl.GENL_GTP_ATTR_NET_NS_FD:
|
|
p.NetNSFD = native.Uint32(a.Value)
|
|
case nl.GENL_GTP_ATTR_I_TEI:
|
|
p.ITEI = native.Uint32(a.Value)
|
|
case nl.GENL_GTP_ATTR_O_TEI:
|
|
p.OTEI = native.Uint32(a.Value)
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func parsePDP(msgs [][]byte) ([]*PDP, error) {
|
|
pdps := make([]*PDP, 0, len(msgs))
|
|
for _, m := range msgs {
|
|
attrs, err := nl.ParseRouteAttr(m[nl.SizeofGenlmsg:])
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
pdp := &PDP{}
|
|
if err := pdp.parseAttributes(attrs); err != nil {
|
|
return nil, err
|
|
}
|
|
pdps = append(pdps, pdp)
|
|
}
|
|
return pdps, nil
|
|
}
|
|
|
|
func (h *Handle) GTPPDPList() ([]*PDP, error) {
|
|
f, err := h.GenlFamilyGet(nl.GENL_GTP_NAME)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
msg := &nl.Genlmsg{
|
|
Command: nl.GENL_GTP_CMD_GETPDP,
|
|
Version: nl.GENL_GTP_VERSION,
|
|
}
|
|
req := h.newNetlinkRequest(int(f.ID), syscall.NLM_F_DUMP)
|
|
req.AddData(msg)
|
|
msgs, err := req.Execute(syscall.NETLINK_GENERIC, 0)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return parsePDP(msgs)
|
|
}
|
|
|
|
func GTPPDPList() ([]*PDP, error) {
|
|
return pkgHandle.GTPPDPList()
|
|
}
|
|
|
|
func gtpPDPGet(req *nl.NetlinkRequest) (*PDP, error) {
|
|
msgs, err := req.Execute(syscall.NETLINK_GENERIC, 0)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
pdps, err := parsePDP(msgs)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if len(pdps) != 1 {
|
|
return nil, fmt.Errorf("invalid reqponse for GENL_GTP_CMD_GETPDP")
|
|
}
|
|
return pdps[0], nil
|
|
}
|
|
|
|
func (h *Handle) GTPPDPByTID(link Link, tid int) (*PDP, error) {
|
|
f, err := h.GenlFamilyGet(nl.GENL_GTP_NAME)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
msg := &nl.Genlmsg{
|
|
Command: nl.GENL_GTP_CMD_GETPDP,
|
|
Version: nl.GENL_GTP_VERSION,
|
|
}
|
|
req := h.newNetlinkRequest(int(f.ID), 0)
|
|
req.AddData(msg)
|
|
req.AddData(nl.NewRtAttr(nl.GENL_GTP_ATTR_VERSION, nl.Uint32Attr(0)))
|
|
req.AddData(nl.NewRtAttr(nl.GENL_GTP_ATTR_LINK, nl.Uint32Attr(uint32(link.Attrs().Index))))
|
|
req.AddData(nl.NewRtAttr(nl.GENL_GTP_ATTR_TID, nl.Uint64Attr(uint64(tid))))
|
|
return gtpPDPGet(req)
|
|
}
|
|
|
|
func GTPPDPByTID(link Link, tid int) (*PDP, error) {
|
|
return pkgHandle.GTPPDPByTID(link, tid)
|
|
}
|
|
|
|
func (h *Handle) GTPPDPByITEI(link Link, itei int) (*PDP, error) {
|
|
f, err := h.GenlFamilyGet(nl.GENL_GTP_NAME)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
msg := &nl.Genlmsg{
|
|
Command: nl.GENL_GTP_CMD_GETPDP,
|
|
Version: nl.GENL_GTP_VERSION,
|
|
}
|
|
req := h.newNetlinkRequest(int(f.ID), 0)
|
|
req.AddData(msg)
|
|
req.AddData(nl.NewRtAttr(nl.GENL_GTP_ATTR_VERSION, nl.Uint32Attr(1)))
|
|
req.AddData(nl.NewRtAttr(nl.GENL_GTP_ATTR_LINK, nl.Uint32Attr(uint32(link.Attrs().Index))))
|
|
req.AddData(nl.NewRtAttr(nl.GENL_GTP_ATTR_I_TEI, nl.Uint32Attr(uint32(itei))))
|
|
return gtpPDPGet(req)
|
|
}
|
|
|
|
func GTPPDPByITEI(link Link, itei int) (*PDP, error) {
|
|
return pkgHandle.GTPPDPByITEI(link, itei)
|
|
}
|
|
|
|
func (h *Handle) GTPPDPByMSAddress(link Link, addr net.IP) (*PDP, error) {
|
|
f, err := h.GenlFamilyGet(nl.GENL_GTP_NAME)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
msg := &nl.Genlmsg{
|
|
Command: nl.GENL_GTP_CMD_GETPDP,
|
|
Version: nl.GENL_GTP_VERSION,
|
|
}
|
|
req := h.newNetlinkRequest(int(f.ID), 0)
|
|
req.AddData(msg)
|
|
req.AddData(nl.NewRtAttr(nl.GENL_GTP_ATTR_VERSION, nl.Uint32Attr(0)))
|
|
req.AddData(nl.NewRtAttr(nl.GENL_GTP_ATTR_LINK, nl.Uint32Attr(uint32(link.Attrs().Index))))
|
|
req.AddData(nl.NewRtAttr(nl.GENL_GTP_ATTR_MS_ADDRESS, []byte(addr.To4())))
|
|
return gtpPDPGet(req)
|
|
}
|
|
|
|
func GTPPDPByMSAddress(link Link, addr net.IP) (*PDP, error) {
|
|
return pkgHandle.GTPPDPByMSAddress(link, addr)
|
|
}
|
|
|
|
func (h *Handle) GTPPDPAdd(link Link, pdp *PDP) error {
|
|
f, err := h.GenlFamilyGet(nl.GENL_GTP_NAME)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
msg := &nl.Genlmsg{
|
|
Command: nl.GENL_GTP_CMD_NEWPDP,
|
|
Version: nl.GENL_GTP_VERSION,
|
|
}
|
|
req := h.newNetlinkRequest(int(f.ID), syscall.NLM_F_EXCL|syscall.NLM_F_ACK)
|
|
req.AddData(msg)
|
|
req.AddData(nl.NewRtAttr(nl.GENL_GTP_ATTR_VERSION, nl.Uint32Attr(pdp.Version)))
|
|
req.AddData(nl.NewRtAttr(nl.GENL_GTP_ATTR_LINK, nl.Uint32Attr(uint32(link.Attrs().Index))))
|
|
req.AddData(nl.NewRtAttr(nl.GENL_GTP_ATTR_PEER_ADDRESS, []byte(pdp.PeerAddress.To4())))
|
|
req.AddData(nl.NewRtAttr(nl.GENL_GTP_ATTR_MS_ADDRESS, []byte(pdp.MSAddress.To4())))
|
|
|
|
switch pdp.Version {
|
|
case 0:
|
|
req.AddData(nl.NewRtAttr(nl.GENL_GTP_ATTR_TID, nl.Uint64Attr(pdp.TID)))
|
|
req.AddData(nl.NewRtAttr(nl.GENL_GTP_ATTR_FLOW, nl.Uint16Attr(pdp.Flow)))
|
|
case 1:
|
|
req.AddData(nl.NewRtAttr(nl.GENL_GTP_ATTR_I_TEI, nl.Uint32Attr(pdp.ITEI)))
|
|
req.AddData(nl.NewRtAttr(nl.GENL_GTP_ATTR_O_TEI, nl.Uint32Attr(pdp.OTEI)))
|
|
default:
|
|
return fmt.Errorf("unsupported GTP version: %d", pdp.Version)
|
|
}
|
|
_, err = req.Execute(syscall.NETLINK_GENERIC, 0)
|
|
return err
|
|
}
|
|
|
|
func GTPPDPAdd(link Link, pdp *PDP) error {
|
|
return pkgHandle.GTPPDPAdd(link, pdp)
|
|
}
|
|
|
|
func (h *Handle) GTPPDPDel(link Link, pdp *PDP) error {
|
|
f, err := h.GenlFamilyGet(nl.GENL_GTP_NAME)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
msg := &nl.Genlmsg{
|
|
Command: nl.GENL_GTP_CMD_DELPDP,
|
|
Version: nl.GENL_GTP_VERSION,
|
|
}
|
|
req := h.newNetlinkRequest(int(f.ID), syscall.NLM_F_EXCL|syscall.NLM_F_ACK)
|
|
req.AddData(msg)
|
|
req.AddData(nl.NewRtAttr(nl.GENL_GTP_ATTR_VERSION, nl.Uint32Attr(pdp.Version)))
|
|
req.AddData(nl.NewRtAttr(nl.GENL_GTP_ATTR_LINK, nl.Uint32Attr(uint32(link.Attrs().Index))))
|
|
|
|
switch pdp.Version {
|
|
case 0:
|
|
req.AddData(nl.NewRtAttr(nl.GENL_GTP_ATTR_TID, nl.Uint64Attr(pdp.TID)))
|
|
case 1:
|
|
req.AddData(nl.NewRtAttr(nl.GENL_GTP_ATTR_I_TEI, nl.Uint32Attr(pdp.ITEI)))
|
|
default:
|
|
return fmt.Errorf("unsupported GTP version: %d", pdp.Version)
|
|
}
|
|
_, err = req.Execute(syscall.NETLINK_GENERIC, 0)
|
|
return err
|
|
}
|
|
|
|
func GTPPDPDel(link Link, pdp *PDP) error {
|
|
return pkgHandle.GTPPDPDel(link, pdp)
|
|
}
|