tools
/
out-of-tree
1
0
Fork 0
Code Releases Activity

Rename qemu package

timestamps
dump_stack() 2019-02-02 21:33:27 +00:00
parent 144a8547bc
commit b5965e8374
Signed by: dump_stack
GPG Key ID: D90D0727C1FEBEAB
3 changed files with 40 additions and 40 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
qemu/qemu-kernel.go
View File

@ -2,7 +2,7 @@
// Use of this source code is governed by a AGPLv3 license // Use of this source code is governed by a AGPLv3 license
// (or later) that can be found in the LICENSE file. // (or later) that can be found in the LICENSE file.
package qemukernel package qemu
import ( import (
"bytes" "bytes"

76
qemu/qemu-kernel_test.go
View File

@ -2,7 +2,7 @@
// Use of this source code is governed by a AGPLv3 license // Use of this source code is governed by a AGPLv3 license
// (or later) that can be found in the LICENSE file. // (or later) that can be found in the LICENSE file.
package qemukernel package qemu
import ( import (
"crypto/sha512" "crypto/sha512"
@ -50,16 +50,16 @@ func TestQemuSystemNew(t *testing.T) {
func TestQemuSystemStart(t *testing.T) { func TestQemuSystemStart(t *testing.T) {
kernel := Kernel{Name: "Test kernel", KernelPath: testConfigVmlinuz} kernel := Kernel{Name: "Test kernel", KernelPath: testConfigVmlinuz}
qemu, err := NewQemuSystem(X86_64, kernel, "/bin/sh") q, err := NewQemuSystem(X86_64, kernel, "/bin/sh")
if err != nil { if err != nil {
t.Fatal(err) t.Fatal(err)
} }
if err = qemu.Start(); err != nil { if err = q.Start(); err != nil {
t.Fatal(err) t.Fatal(err)
} }
qemu.Stop() q.Stop()
} }
func TestGetFreeAddrPort(t *testing.T) { func TestGetFreeAddrPort(t *testing.T) {
@ -74,24 +74,24 @@ func TestGetFreeAddrPort(t *testing.T) {
func TestQemuSystemStart_Timeout(t *testing.T) { func TestQemuSystemStart_Timeout(t *testing.T) {
t.Parallel() t.Parallel()
kernel := Kernel{Name: "Test kernel", KernelPath: testConfigVmlinuz} kernel := Kernel{Name: "Test kernel", KernelPath: testConfigVmlinuz}
qemu, err := NewQemuSystem(X86_64, kernel, "/bin/sh") q, err := NewQemuSystem(X86_64, kernel, "/bin/sh")
if err != nil { if err != nil {
t.Fatal(err) t.Fatal(err)
} }
qemu.Timeout = time.Second q.Timeout = time.Second
if err = qemu.Start(); err != nil { if err = q.Start(); err != nil {
t.Fatal(err) t.Fatal(err)
} }
time.Sleep(2 * time.Second) time.Sleep(2 * time.Second)
if !qemu.Died { if !q.Died {
t.Fatal("qemu does not died :c") t.Fatal("qemu does not died :c")
} }
if !qemu.KilledByTimeout { if !q.KilledByTimeout {
t.Fatal("qemu died not because of timeout O_o") t.Fatal("qemu died not because of timeout O_o")
} }
} }
@ -120,13 +120,13 @@ func startTestQemu(t *testing.T, timeout time.Duration) (q *QemuSystem, err erro
} }
func TestQemuSystemCommand(t *testing.T) { func TestQemuSystemCommand(t *testing.T) {
qemu, err := startTestQemu(t, 0) q, err := startTestQemu(t, 0)
if err != nil { if err != nil {
t.Fatal(err) t.Fatal(err)
} }
defer qemu.Stop() defer q.Stop()
output, err := qemu.Command("root", "cat /etc/shadow") output, err := q.Command("root", "cat /etc/shadow")
if err != nil { if err != nil {
t.Fatal(err) t.Fatal(err)
} }
@ -134,7 +134,7 @@ func TestQemuSystemCommand(t *testing.T) {
t.Fatal("Wrong output from `cat /etc/shadow` by root") t.Fatal("Wrong output from `cat /etc/shadow` by root")
} }
output, err = qemu.Command("user", "cat /etc/passwd") output, err = q.Command("user", "cat /etc/passwd")
if err != nil { if err != nil {
t.Fatal(err) t.Fatal(err)
} }
@ -142,7 +142,7 @@ func TestQemuSystemCommand(t *testing.T) {
t.Fatal("Wrong output from `cat /etc/passwd` by user") t.Fatal("Wrong output from `cat /etc/passwd` by user")
} }
_, err = qemu.Command("user", "cat /etc/shadow") _, err = q.Command("user", "cat /etc/shadow")
// unsuccessful is good because user must not read /etc/shadow // unsuccessful is good because user must not read /etc/shadow
if err == nil { if err == nil {
t.Fatal("User have rights for /etc/shadow. WAT?!") t.Fatal("User have rights for /etc/shadow. WAT?!")
@ -150,11 +150,11 @@ func TestQemuSystemCommand(t *testing.T) {
} }
func TestQemuSystemCopyFile(t *testing.T) { func TestQemuSystemCopyFile(t *testing.T) {
qemu, err := startTestQemu(t, 0) q, err := startTestQemu(t, 0)
if err != nil { if err != nil {
t.Fatal(err) t.Fatal(err)
} }
defer qemu.Stop() defer q.Stop()
localPath := "/bin/sh" localPath := "/bin/sh"
@ -165,12 +165,12 @@ func TestQemuSystemCopyFile(t *testing.T) {
shaLocal := fmt.Sprintf("%x", sha512.Sum512(content)) shaLocal := fmt.Sprintf("%x", sha512.Sum512(content))
err = qemu.CopyFile("user", localPath, "/tmp/test") err = q.CopyFile("user", localPath, "/tmp/test")
if err != nil { if err != nil {
t.Fatal(err) t.Fatal(err)
} }
shaRemote, err := qemu.Command("user", "sha512sum /tmp/test") shaRemote, err := q.Command("user", "sha512sum /tmp/test")
if err != nil { if err != nil {
t.Fatal(err) t.Fatal(err)
} }
@ -183,11 +183,11 @@ func TestQemuSystemCopyFile(t *testing.T) {
} }
func TestQemuSystemCopyAndRun(t *testing.T) { func TestQemuSystemCopyAndRun(t *testing.T) {
qemu, err := startTestQemu(t, 0) q, err := startTestQemu(t, 0)
if err != nil { if err != nil {
t.Fatal(err) t.Fatal(err)
} }
defer qemu.Stop() defer q.Stop()
randStr := fmt.Sprintf("%d", rand.Int()) randStr := fmt.Sprintf("%d", rand.Int())
content := []byte("#!/bin/sh\n echo -n " + randStr + "\n") content := []byte("#!/bin/sh\n echo -n " + randStr + "\n")
@ -205,7 +205,7 @@ func TestQemuSystemCopyAndRun(t *testing.T) {
t.Fatal(err) t.Fatal(err)
} }
output, err := qemu.CopyAndRun("user", tmpfile.Name()) output, err := q.CopyAndRun("user", tmpfile.Name())
if err != nil { if err != nil {
t.Fatal(err) t.Fatal(err)
} }
@ -217,23 +217,23 @@ func TestQemuSystemCopyAndRun(t *testing.T) {
} }
func TestQemuSystemCopyAndInsmod(t *testing.T) { func TestQemuSystemCopyAndInsmod(t *testing.T) {
qemu, err := startTestQemu(t, 0) q, err := startTestQemu(t, 0)
if err != nil { if err != nil {
t.Fatal(err) t.Fatal(err)
} }
defer qemu.Stop() defer q.Stop()
lsmodBefore, err := qemu.Command("root", "lsmod | wc -l") lsmodBefore, err := q.Command("root", "lsmod | wc -l")
if err != nil { if err != nil {
t.Fatal(err) t.Fatal(err)
} }
_, err = qemu.CopyAndInsmod(testConfigSampleKo) _, err = q.CopyAndInsmod(testConfigSampleKo)
if err != nil { if err != nil {
t.Fatal(err) t.Fatal(err)
} }
lsmodAfter, err := qemu.Command("root", "lsmod | wc -l") lsmodAfter, err := q.Command("root", "lsmod | wc -l")
if err != nil { if err != nil {
t.Fatal(err) t.Fatal(err)
} }
@ -244,20 +244,20 @@ func TestQemuSystemCopyAndInsmod(t *testing.T) {
} }
func TestQemuSystemKernelPanic(t *testing.T) { func TestQemuSystemKernelPanic(t *testing.T) {
qemu, err := startTestQemu(t, 5*time.Minute) q, err := startTestQemu(t, 5*time.Minute)
if err != nil { if err != nil {
t.Fatal(err) t.Fatal(err)
} }
defer qemu.Stop() defer q.Stop()
// Enable sysrq // Enable sysrq
_, err = qemu.Command("root", "echo 1 > /proc/sys/kernel/sysrq") _, err = q.Command("root", "echo 1 > /proc/sys/kernel/sysrq")
if err != nil { if err != nil {
t.Fatal(err) t.Fatal(err)
} }
// Trigger kernel panic // Trigger kernel panic
err = qemu.AsyncCommand("root", "sleep 1s && echo c > /proc/sysrq-trigger") err = q.AsyncCommand("root", "sleep 1s && echo c > /proc/sysrq-trigger")
if err != nil { if err != nil {
t.Fatal(err) t.Fatal(err)
} }
@ -265,40 +265,40 @@ func TestQemuSystemKernelPanic(t *testing.T) {
// Wait for panic watcher timeout // Wait for panic watcher timeout
time.Sleep(5 * time.Second) time.Sleep(5 * time.Second)
if qemu.KilledByTimeout { if q.KilledByTimeout {
t.Fatal("qemu is killed by timeout, not because of panic") t.Fatal("qemu is killed by timeout, not because of panic")
} }
if !qemu.Died { if !q.Died {
t.Fatal("qemu is not killed after kernel panic") t.Fatal("qemu is not killed after kernel panic")
} }
if !qemu.KernelPanic { if !q.KernelPanic {
t.Fatal("qemu is died but there's no information about panic") t.Fatal("qemu is died but there's no information about panic")
} }
} }
func TestQemuSystemRun(t *testing.T) { func TestQemuSystemRun(t *testing.T) {
qemu, err := startTestQemu(t, 0) q, err := startTestQemu(t, 0)
if err != nil { if err != nil {
t.Fatal(err) t.Fatal(err)
} }
defer qemu.Stop() defer q.Stop()
for { for {
_, err := qemu.Command("root", "echo") _, err := q.Command("root", "echo")
if err == nil { if err == nil {
break break
} }
} }
start := time.Now() start := time.Now()
err = qemu.AsyncCommand("root", "sleep 1m") err = q.AsyncCommand("root", "sleep 1m")
if err != nil { if err != nil {
t.Fatal(err) t.Fatal(err)
} }
if time.Since(start) > 10*time.Second { if time.Since(start) > 10*time.Second {
t.Fatalf("qemu.AsyncCommand does not async (waited %s)", t.Fatalf("q.AsyncCommand does not async (waited %s)",
time.Since(start)) time.Since(start))
} }

2
qemu/test.config.go
View File

@ -2,7 +2,7 @@
// Use of this source code is governed by a AGPLv3 license // Use of this source code is governed by a AGPLv3 license
// (or later) that can be found in the LICENSE file. // (or later) that can be found in the LICENSE file.
package qemukernel package qemu
const testConfigVmlinuz = "../tools/qemu-debian-img/ubuntu1804.vmlinuz" const testConfigVmlinuz = "../tools/qemu-debian-img/ubuntu1804.vmlinuz"
const testConfigInitrd = "../tools/qemu-debian-img/ubuntu1804.initrd" const testConfigInitrd = "../tools/qemu-debian-img/ubuntu1804.initrd"