1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108
|
package main
import (
"fmt"
"net"
"runtime"
"github.com/mndrix/tap-go"
"github.com/opencontainers/runtime-tools/cgroups"
"github.com/opencontainers/runtime-tools/validation/util"
)
func testNetworkCgroups() error {
t := tap.New()
t.Header(0)
defer t.AutoPlan()
loIfName := ""
ethIfName := ""
// we try to get the first 2 interfaces on the system, to assign
// the first one to loIfName, the second to ethIfName.
l, err := net.Interfaces()
if err != nil {
// fall back to the default list only with lo
loIfName = "lo"
ethIfName = "eth0"
} else {
loIfName = l[0].Name
ethIfName = l[1].Name
}
cases := []struct {
classid uint32
prio uint32
ifName string
withNetNs bool
withUserNs bool
}{
{255, 10, loIfName, true, true},
{255, 10, loIfName, true, false},
{255, 10, loIfName, false, true},
{255, 10, loIfName, false, false},
{255, 10, ethIfName, true, true},
{255, 10, ethIfName, true, false},
{255, 10, ethIfName, false, true},
{255, 10, ethIfName, false, false},
{255, 30, loIfName, true, true},
{255, 30, loIfName, true, false},
{255, 30, loIfName, false, true},
{255, 30, loIfName, false, false},
{255, 30, ethIfName, true, true},
{255, 30, ethIfName, true, false},
{255, 30, ethIfName, false, true},
{255, 30, ethIfName, false, false},
{550, 10, loIfName, true, true},
{550, 10, loIfName, true, false},
{550, 10, loIfName, false, true},
{550, 10, loIfName, false, false},
{550, 10, ethIfName, true, true},
{550, 10, ethIfName, true, false},
{550, 10, ethIfName, false, true},
{550, 10, ethIfName, false, false},
{550, 30, loIfName, true, true},
{550, 30, loIfName, true, false},
{550, 30, loIfName, false, true},
{550, 30, loIfName, false, false},
{550, 30, ethIfName, true, true},
{550, 30, ethIfName, true, false},
{550, 30, ethIfName, false, true},
{550, 30, ethIfName, false, false},
}
for _, c := range cases {
g, err := util.GetDefaultGenerator()
if err != nil {
return err
}
g.SetLinuxCgroupsPath(cgroups.AbsCgroupPath)
g.SetLinuxResourcesNetworkClassID(c.classid)
g.AddLinuxResourcesNetworkPriorities(c.ifName, c.prio)
if !c.withNetNs {
g.RemoveLinuxNamespace("network")
}
if !c.withUserNs {
g.RemoveLinuxNamespace("user")
}
err = util.RuntimeOutsideValidate(g, t, util.ValidateLinuxResourcesNetwork)
if err != nil {
return err
}
}
return nil
}
func main() {
if "linux" != runtime.GOOS {
util.Fatal(fmt.Errorf("linux-specific cgroup test"))
}
if err := testNetworkCgroups(); err != nil {
util.Fatal(err)
}
}
|