Add TestIP that uses cpu on 4 Linux architectures #316
There are no files selected for viewing
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
|
Collaborator
There was a problem hiding this comment. do you want both the cpio and the cpio.gz? |
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,240 @@ | ||
| // Copyright 2025 the u-root Authors. All rights reserved | ||
| // Use of this source code is governed by a BSD-style | ||
| // license that can be found in the LICENSE file. | ||
|
|
||
| //go:build !race && amd64 | ||
rminnich marked this conversation as resolved.
Show resolved
Hide resolved
|
||
|
|
||
| package vm_test | ||
|
|
||
| import ( | ||
| "context" | ||
| "os" | ||
| "strings" | ||
| "testing" | ||
| "time" | ||
|
|
||
| "github.com/u-root/cpu/client" | ||
| "github.com/u-root/cpu/vm" | ||
| ) | ||
|
|
||
| func count(s string, t []string) int { | ||
| var i int | ||
| for _, n := range t { | ||
| if strings.Contains(s, n) { | ||
| i++ | ||
| } | ||
| } | ||
| return i | ||
| } | ||
|
|
||
| func all(s string, t []string) bool { | ||
| return count(s, t) == len(t) | ||
| } | ||
|
|
||
| func some(s string, t []string) bool { | ||
| return count(s, t) > 0 | ||
| } | ||
|
|
||
| func none(s string, t []string) bool { | ||
| return count(s, t) == 0 | ||
| } | ||
|
|
||
| // TestIP tests creation and removal of addresses, tunnels, and | ||
| // ARP entries with the u-root ip command. | ||
| func TestIP(t *testing.T) { | ||
| d := t.TempDir() | ||
| for _, arch := range []string{"amd64", "arm", "arm64", "riscv64"} { | ||
| i, err := vm.New("linux", arch) | ||
| if err != nil { | ||
| t.Fatalf("Testing kernel=linux arch=%s: got %v, want nil", arch, err) | ||
| } | ||
|
|
||
| ctx, cancel := context.WithCancel(context.Background()) | ||
| defer cancel() | ||
|
|
||
| t.Logf("image:%s", i) | ||
| n, err := i.Uroot(d) | ||
| if err != nil { | ||
| t.Skipf("skipping this test as we have no uroot command") | ||
| } | ||
|
|
||
| c, err := i.CommandContext(ctx, d, n) | ||
| if err != nil { | ||
| t.Fatalf("starting VM: got %v, want nil", err) | ||
| } | ||
| t.Logf("Start %v", c.Args) | ||
|
|
||
| // For debug. It's been needed, but we do not want this spew in | ||
| // CI logs, so leave it off. | ||
| if false { | ||
| c.Stdout, c.Stderr = os.Stdout, os.Stderr | ||
| } | ||
| if err := i.StartVM(c); err != nil { | ||
| t.Fatalf("starting VM: got %v, want nil", err) | ||
| } | ||
|
|
||
| type iptest struct { | ||
| cmd any | ||
| delay time.Duration | ||
| failok bool | ||
| includes []string | ||
| excludes []string | ||
| } | ||
| // This is a slice of iptest slices. | ||
| // The intent is that if the first one fails, and it has failok set to false, the rest of the tests | ||
| // in that slice are skipped. | ||
| for _, iptest := range [][]iptest{ | ||
| { | ||
| {cmd: "ip link set eth1 down", delay: 3 * time.Second}, | ||
| {cmd: "cat /sys/class/net/eth1/operstate", includes: []string{"down"}}, | ||
| {cmd: "ip link set eth1 up", delay: 3 * time.Second}, | ||
| {cmd: "cat /sys/class/net/eth1/operstate", includes: []string{"up"}}, | ||
| {cmd: []string{"ip", "addr", "add", "192.168.241.1/24", "dev", "eth1"}}, | ||
| {cmd: []string{"cat", "/proc/net/fib_trie"}, includes: []string{"192.168.241.1"}}, | ||
| {cmd: "ip route add 192.168.242.0/24 via 192.168.241.1 dev eth1"}, | ||
| {cmd: "cat /proc/net/route", includes: []string{"00F2A8C0", "01F1A8C0"}}, | ||
| {cmd: "ip route del 192.168.242.0/24"}, | ||
| {cmd: "cat /proc/net/route", excludes: []string{"00F2A8C0", "01F1A8C0"}}, | ||
| {cmd: "ip tunnel add my_test_tunnel mode sit remote 192.168.242.1 local 192.168.241.1 ttl 64"}, | ||
| {cmd: "cat /proc/net/dev", includes: []string{"my_test_tunnel"}}, | ||
| {cmd: "ip tunnel del my_test_tunnel"}, | ||
| {cmd: "cat /proc/net/dev", excludes: []string{"my_test_tunnel"}}, | ||
| {cmd: "ip tunnel add my_test_tunnel mode sit remote 192.168.242.1 local 192.168.241.1 ttl 64"}, | ||
| {cmd: "ip tunnel show my_test_tunnel", includes: []string{"my_test_tunnel", "remote 192.168.242.1", "local 192.168.241.1", "ttl 64"}}, | ||
| {cmd: "ip tunnel del my_test_tunnel"}, | ||
| {cmd: "cat /proc/net/dev", excludes: []string{"my_test_tunnel"}}, | ||
| }, | ||
| { | ||
| // Various tunnel tests. | ||
| // Add a GRE tunnel with key and tos options | ||
| {cmd: "ip tunnel add gre_tunnel mode gre remote 192.168.242.2 local 192.168.241.1 ttl 128 key 1234 tos 10", failok: false, delay: time.Second}, | ||
| {cmd: "cat /proc/net/dev", includes: []string{"gre_tunnel"}}, | ||
| // Verify GRE tunnel parameters | ||
| {cmd: "ip tunnel show gre_tunnel", includes: []string{"gre_tunnel:", "remote 192.168.242.2", "local 192.168.241.1", "ttl 128", "key 1234", "tos 0xa"}}, | ||
| {cmd: "ip link set gre_tunnel up"}, | ||
| {cmd: "ip addr add 10.0.0.1/24 dev gre_tunnel"}, | ||
| {cmd: []string{"cat", "/proc/net/fib_trie"}, includes: []string{"10.0.0.1"}}, | ||
| {cmd: "ip link set gre_tunnel down"}, | ||
| {cmd: "ip tunnel del gre_tunnel"}, | ||
| {cmd: "cat /proc/net/dev", excludes: []string{"gre_tunnel"}}, | ||
| }, | ||
| { | ||
| {cmd: "ip tunnel add vti_tunnel mode vti remote 192.168.242.3 local 192.168.241.1 key 5678", failok: false}, | ||
|
|
||
| // Verify VTI tunnel exists in /proc/net/dev | ||
| {cmd: "cat /proc/net/dev", includes: []string{"vti_tunnel"}}, | ||
|
|
||
| // Verify VTI tunnel parameters | ||
| {cmd: "ip tunnel show vti_tunnel", includes: []string{"vti_tunnel:", "remote 192.168.242.3", "local 192.168.241.1", "key 5678"}}, | ||
| {cmd: "ip link set vti_tunnel up"}, | ||
| {cmd: "ip addr add 172.16.0.1/30 dev vti_tunnel"}, | ||
| {cmd: []string{"cat", "/proc/net/fib_trie"}, includes: []string{"172.16.0.1"}}, | ||
| {cmd: "ip link set vti_tunnel down"}, | ||
| {cmd: "ip tunnel del vti_tunnel"}, | ||
| {cmd: "cat /proc/net/dev", excludes: []string{"vti_tunnel"}}, | ||
| }, | ||
| { | ||
| {cmd: "ip tunnel add ipip_tunnel mode ipip remote 192.168.243.1 local 192.168.241.1 ttl 64", failok: false}, | ||
|
|
||
| // Verify IPIP tunnel exists in /proc/net/dev | ||
| {cmd: "cat /proc/net/dev", includes: []string{"ipip_tunnel"}}, | ||
|
|
||
| // Verify IPIP tunnel parameters | ||
| {cmd: "ip tunnel show ipip_tunnel", includes: []string{"ipip_tunnel:", "remote 192.168.243.1", "local 192.168.241.1", "ttl 64"}}, | ||
| {cmd: "ip link set ipip_tunnel up"}, | ||
| {cmd: "ip addr add 172.17.0.1/30 dev ipip_tunnel"}, | ||
| {cmd: []string{"cat", "/proc/net/fib_trie"}, includes: []string{"172.17.0.1"}}, | ||
| {cmd: "ip link set ipip_tunnel down"}, | ||
| {cmd: "ip tunnel del ipip_tunnel"}, | ||
| {cmd: "cat /proc/net/dev", excludes: []string{"ipip_tunnel"}}, | ||
| }, | ||
| { | ||
| // ARP tests | ||
| {cmd: "ip neigh add 192.168.241.2 lladdr 00:11:22:33:44:55 dev eth1"}, | ||
| {cmd: "cat /proc/net/arp", includes: []string{"192.168.241.2"}}, | ||
|
|
||
| // Verify the neighbor entry | ||
| {cmd: "ip neigh show dev eth1", includes: []string{"192.168.241.2", "192.168.241.2 dev eth1 lladdr 00:11:22:33:44:55 PERMANENT"}}, | ||
|
|
||
| // Replace the entry with another hwaddress, nud state and router flag | ||
| {cmd: "ip neigh replace 192.168.241.2 lladdr 11:22:33:44:55:66 dev eth1 nud stale router", includes: []string{}}, | ||
|
|
||
| // Verify the modified flags | ||
| {cmd: "ip neigh show dev eth1", includes: []string{"192.168.241.2", "192.168.241.2 dev eth1 lladdr 11:22:33:44:55:66 router STALE"}}, | ||
|
|
||
| // Delete the neighbor | ||
| {cmd: "ip neigh del 192.168.241.2 dev eth1", includes: []string{}}, | ||
| {cmd: "cat /proc/net/arp", excludes: []string{"192.168.241.2"}}, | ||
|
|
||
| // Test IP Neighbor flush capability | ||
| // Add 3 neighbors | ||
| {cmd: "ip neigh add 192.168.241.5 lladdr aa:bb:cc:dd:ee:ff nud stale dev eth1"}, | ||
| {cmd: "ip neigh add 192.168.241.6 lladdr aa:bb:cc:11:22:33 nud stale dev eth1"}, | ||
| {cmd: "ip neigh add 192.168.241.7 lladdr aa:bb:cc:44:55:66 dev eth1"}, | ||
|
|
||
| // Verify all entries exist | ||
| {cmd: "cat /proc/net/arp", includes: []string{"192.168.241.5", "192.168.241.6", "192.168.241.7"}}, | ||
|
|
||
| // Flush the 2 stale neighbors from the table for eth1 | ||
| {cmd: "ip neigh flush dev eth1", includes: []string{}}, | ||
|
|
||
| // Verify the 2 stale entries are gone, the permanent one remains | ||
| {cmd: "cat /proc/net/arp", includes: []string{"192.168.241.7"}, excludes: []string{"192.168.241.5", "192.168.241.6"}}, | ||
|
|
||
| // Delete the IP address from eth1 | ||
| {cmd: "ip addr del 192.168.241.1/24 dev eth1"}, | ||
| {cmd: []string{"cat", "/proc/net/fib_trie"}, excludes: []string{"192.168.241.1"}}, | ||
| // Bring the eth1 interface down | ||
| {cmd: "ip link set eth1 down", delay: 2 * time.Second}, | ||
| {cmd: "cat /sys/class/net/eth1/operstate", includes: []string{"down"}}, | ||
| }, | ||
| } { | ||
|
|
||
| for _, tt := range iptest { | ||
| var cmd []string | ||
| switch c := tt.cmd.(type) { | ||
| case []string: | ||
| cmd = c | ||
| case string: | ||
| cmd = strings.Fields(c) | ||
| default: | ||
| t.Fatalf("tt.cmd.Type(): type %T, want string or []string", tt.cmd) | ||
| } | ||
| t.Logf("Run %s", cmd) | ||
| cpu, err := i.CPUCommand(cmd[0], cmd[1:]...) | ||
| if err != nil { | ||
| t.Errorf("CPUCommand: got %v, want nil", err) | ||
| continue | ||
| } | ||
| if false { | ||
| client.SetVerbose(t.Logf) | ||
| } | ||
|
|
||
| b, err := cpu.CombinedOutput() | ||
|
|
||
| if false { // Only enable this if you want to see console output/errors. | ||
| t.Logf("%s %v", string(b), err) | ||
| } | ||
|
|
||
| if err != nil { | ||
| if tt.failok { | ||
| t.Logf("%s: got %v, want nil, skipping rest of tests in slice", cmd, err) | ||
| break | ||
| } | ||
| t.Fatalf("%s: got %v, want nil", cmd, err) | ||
| } | ||
| //t.Logf("%q, includes %s?", string(b), tt.includes) | ||
| if !all(string(b), tt.includes) { | ||
| t.Fatalf("%s: got %s, does not contain all of %s", cmd, string(b), tt.includes) | ||
| } | ||
| //t.Logf("%q, excludes %s?", string(b), tt.includes) | ||
| if some(string(b), tt.excludes) { | ||
| t.Fatalf("%s:got %s, contains some of %s and should not", cmd, string(b), tt.excludes) | ||
| } | ||
| if tt.delay > 0 { | ||
| time.Sleep(tt.delay) | ||
| } | ||
| } | ||
| } | ||
| } | ||
| } | ||
Binary file not shown.
Binary file not shown.
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.
Oops, something went wrong.
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
Uh oh!
There was an error while loading. Please reload this page.