139 lines
2.8 KiB
Go
139 lines
2.8 KiB
Go
|
package netlink
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
)
|
||
|
|
|
||
|
|
const (
|
||
|
|
HANDLE_NONE = 0
|
||
|
|
HANDLE_INGRESS = 0xFFFFFFF1
|
||
|
|
HANDLE_ROOT = 0xFFFFFFFF
|
||
|
|
PRIORITY_MAP_LEN = 16
|
||
|
|
)
|
||
|
|
|
||
|
|
type Qdisc interface {
|
||
|
|
Attrs() *QdiscAttrs
|
||
|
|
Type() string
|
||
|
|
}
|
||
|
|
|
||
|
|
// Qdisc represents a netlink qdisc. A qdisc is associated with a link,
|
||
|
|
// has a handle, a parent and a refcnt. The root qdisc of a device should
|
||
|
|
// have parent == HANDLE_ROOT.
|
||
|
|
type QdiscAttrs struct {
|
||
|
|
LinkIndex int
|
||
|
|
Handle uint32
|
||
|
|
Parent uint32
|
||
|
|
Refcnt uint32 // read only
|
||
|
|
}
|
||
|
|
|
||
|
|
func (q QdiscAttrs) String() string {
|
||
|
|
return fmt.Sprintf("{LinkIndex: %d, Handle: %s, Parent: %s, Refcnt: %s}", q.LinkIndex, HandleStr(q.Handle), HandleStr(q.Parent), q.Refcnt)
|
||
|
|
}
|
||
|
|
|
||
|
|
func MakeHandle(major, minor uint16) uint32 {
|
||
|
|
return (uint32(major) << 16) | uint32(minor)
|
||
|
|
}
|
||
|
|
|
||
|
|
func MajorMinor(handle uint32) (uint16, uint16) {
|
||
|
|
return uint16((handle & 0xFFFF0000) >> 16), uint16(handle & 0x0000FFFFF)
|
||
|
|
}
|
||
|
|
|
||
|
|
func HandleStr(handle uint32) string {
|
||
|
|
switch handle {
|
||
|
|
case HANDLE_NONE:
|
||
|
|
return "none"
|
||
|
|
case HANDLE_INGRESS:
|
||
|
|
return "ingress"
|
||
|
|
case HANDLE_ROOT:
|
||
|
|
return "root"
|
||
|
|
default:
|
||
|
|
major, minor := MajorMinor(handle)
|
||
|
|
return fmt.Sprintf("%x:%x", major, minor)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// PfifoFast is the default qdisc created by the kernel if one has not
|
||
|
|
// been defined for the interface
|
||
|
|
type PfifoFast struct {
|
||
|
|
QdiscAttrs
|
||
|
|
Bands uint8
|
||
|
|
PriorityMap [PRIORITY_MAP_LEN]uint8
|
||
|
|
}
|
||
|
|
|
||
|
|
func (qdisc *PfifoFast) Attrs() *QdiscAttrs {
|
||
|
|
return &qdisc.QdiscAttrs
|
||
|
|
}
|
||
|
|
|
||
|
|
func (qdisc *PfifoFast) Type() string {
|
||
|
|
return "pfifo_fast"
|
||
|
|
}
|
||
|
|
|
||
|
|
// Prio is a basic qdisc that works just like PfifoFast
|
||
|
|
type Prio struct {
|
||
|
|
QdiscAttrs
|
||
|
|
Bands uint8
|
||
|
|
PriorityMap [PRIORITY_MAP_LEN]uint8
|
||
|
|
}
|
||
|
|
|
||
|
|
func NewPrio(attrs QdiscAttrs) *Prio {
|
||
|
|
return &Prio{
|
||
|
|
QdiscAttrs: attrs,
|
||
|
|
Bands: 3,
|
||
|
|
PriorityMap: [PRIORITY_MAP_LEN]uint8{1, 2, 2, 2, 1, 2, 0, 0, 1, 1, 1, 1, 1, 1, 1, 1},
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (qdisc *Prio) Attrs() *QdiscAttrs {
|
||
|
|
return &qdisc.QdiscAttrs
|
||
|
|
}
|
||
|
|
|
||
|
|
func (qdisc *Prio) Type() string {
|
||
|
|
return "prio"
|
||
|
|
}
|
||
|
|
|
||
|
|
// Tbf is a classful qdisc that rate limits based on tokens
|
||
|
|
type Tbf struct {
|
||
|
|
QdiscAttrs
|
||
|
|
// TODO: handle 64bit rate properly
|
||
|
|
Rate uint64
|
||
|
|
Limit uint32
|
||
|
|
Buffer uint32
|
||
|
|
// TODO: handle other settings
|
||
|
|
}
|
||
|
|
|
||
|
|
func (qdisc *Tbf) Attrs() *QdiscAttrs {
|
||
|
|
return &qdisc.QdiscAttrs
|
||
|
|
}
|
||
|
|
|
||
|
|
func (qdisc *Tbf) Type() string {
|
||
|
|
return "tbf"
|
||
|
|
}
|
||
|
|
|
||
|
|
// Ingress is a qdisc for adding ingress filters
|
||
|
|
type Ingress struct {
|
||
|
|
QdiscAttrs
|
||
|
|
}
|
||
|
|
|
||
|
|
func (qdisc *Ingress) Attrs() *QdiscAttrs {
|
||
|
|
return &qdisc.QdiscAttrs
|
||
|
|
}
|
||
|
|
|
||
|
|
func (qdisc *Ingress) Type() string {
|
||
|
|
return "ingress"
|
||
|
|
}
|
||
|
|
|
||
|
|
// GenericQdisc qdiscs represent types that are not currently understood
|
||
|
|
// by this netlink library.
|
||
|
|
type GenericQdisc struct {
|
||
|
|
QdiscAttrs
|
||
|
|
QdiscType string
|
||
|
|
}
|
||
|
|
|
||
|
|
func (qdisc *GenericQdisc) Attrs() *QdiscAttrs {
|
||
|
|
return &qdisc.QdiscAttrs
|
||
|
|
}
|
||
|
|
|
||
|
|
func (qdisc *GenericQdisc) Type() string {
|
||
|
|
return qdisc.QdiscType
|
||
|
|
}
|