tordam

commit 649e647cc468fbbedc35679dc2a821c9cd3debea
parent 6b40d0567149017082560dce7d7aebcd8949eb28
Author: parazyd <parazyd@dyne.org>
Date:   Mon, 11 Dec 2017 16:06:08 +0100

Add very primitive tests for dam-dir.

These tests try to crash or panic the directory daemon. They should fail
if such a thing occurs.

Diffstat:
A
cmd/dam-dir/main_test.go
 | 
177
+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++

1 file changed, 177 insertions(+), 0 deletions(-)
diff --git a/cmd/dam-dir/main_test.go b/cmd/dam-dir/main_test.go
@@ -0,0 +1,177 @@
+package main
+
+// See LICENSE file for copyright and license details.
+
+import (
+	"encoding/base64"
+	"encoding/json"
+	"net/http"
+	"os"
+	"os/exec"
+	"syscall"
+	"testing"
+	"time"
+
+	lib "github.com/parazyd/tor-dam/pkg/damlib"
+)
+
+type msgStruct struct {
+	Secret string
+}
+
+func postReq(data map[string]string) (*http.Response, error) {
+	msg, err := json.Marshal(data)
+	if err != nil {
+		return nil, err
+	}
+	resp, err := lib.HTTPPost("http://localhost:49371/announce", msg)
+	if err != nil {
+		return nil, err
+	}
+	return resp, nil
+}
+
+func getRespText(resp *http.Response) (msgStruct, error) {
+	var m msgStruct
+
+	decoder := json.NewDecoder(resp.Body)
+	err := decoder.Decode(&m)
+	if err != nil {
+		return m, err
+	}
+
+	return m, nil
+}
+
+func TestValidFirstHandshake(t *testing.T) {
+	// Valid 1/2 handshake request
+	req := map[string]string{
+		"nodetype":  "node",
+		"address":   "22mobp7vrb7a4gt2.onion",
+		"message":   "I am a DAM node!",
+		"signature": "BuB/Dv8E44CLzUX88K2Ab0lUNS9A0GSkHPtrFNNWZMihPMWN0ORhwMZBRnMJ8woPO3wSONBvEvaCXA2hvsVrUJTa+hnevQNyQXCRhdTVVuVXEpjyFzkMamxb6InrGqbsGGkEUqGMSr9aaQ85N02MMrM6T6JuyqSSssFg2xuO+P4=",
+		"secret":    "",
+	}
+	resp, err := postReq(req)
+	if err != nil {
+		t.Error(err)
+	}
+
+	if resp.StatusCode != 200 {
+		t.Error("Server did not respond with HTTP 200")
+	}
+
+	m, err := getRespText(resp)
+	if err != nil {
+		t.Error(err)
+	}
+	decodedSecret, err := base64.StdEncoding.DecodeString(m.Secret)
+	if err != nil {
+		t.Error(err)
+	}
+
+	if len(decodedSecret) != 128 {
+		t.Error("decodedSecret is not of correct length.")
+	}
+
+	t.Log("Server replied:", m.Secret)
+}
+
+func TestInvalidFirstHandshake(t *testing.T) {
+	// Invalid 1/2 handshake request
+	var req map[string]string
+
+	// We don't actually care about the validity. We are rather trying to crash
+	// the directory daemon.
+
+	// Invalid: nodetype
+	req = map[string]string{
+		"nodetype":  "foobar",
+		"address":   "22mobp7vrb7a4gt2.onion",
+		"message":   "I am a DAM node!",
+		"signature": "BuB/Dv8E44CLzUX88K2Ab0lUNS9A0GSkHPtrFNNWZMihPMWN0ORhwMZBRnMJ8woPO3wSONBvEvaCXA2hvsVrUJTa+hnevQNyQXCRhdTVVuVXEpjyFzkMamxb6InrGqbsGGkEUqGMSr9aaQ85N02MMrM6T6JuyqSSssFg2xuO+P4=",
+		"secret":    "",
+	}
+	t.Log("Testing Invalid: nodetype")
+	resp, err := postReq(req)
+	if err != nil {
+		t.Error(err)
+	}
+	m, err := getRespText(resp)
+	if err != nil {
+		t.Error(err)
+	} else {
+		t.Log("Server replied:", m.Secret)
+	}
+
+	// Invalid: address
+	req = map[string]string{
+		"nodetype":  "node",
+		"address":   "11moup7v3b7a4gt20onion",
+		"message":   "I am a DAM node!",
+		"signature": "BuB/Dv8E44CLzUX88K2Ab0lUNS9A0GSkHPtrFNNWZMihPMWN0ORhwMZBRnMJ8woPO3wSONBvEvaCXA2hvsVrUJTa+hnevQNyQXCRhdTVVuVXEpjyFzkMamxb6InrGqbsGGkEUqGMSr9aaQ85N02MMrM6T6JuyqSSssFg2xuO+P4=",
+		"secret":    "",
+	}
+	t.Log("Testing Invalid: address")
+	resp, err = postReq(req)
+	if err != nil {
+		t.Error(err)
+	}
+	m, err = getRespText(resp)
+	if err != nil {
+		t.Error(err)
+	} else {
+		t.Log("Server replied:", m.Secret)
+	}
+
+	// Invalid: message vs. signature
+	req = map[string]string{
+		"nodetype":  "node",
+		"address":   "11moup7v3b7a4gt20onion",
+		"message":   "I am a weird DAM node!",
+		"signature": "BuB/Dv8E44CLzUX88K2Ab0lUNS9A0GSkHPtrFNNWZMihPMWN0ORhwMZBRnMJ8woPO3wSONBvEvaCXA2hvsVrUJTa+hnevQNyQXCRhdTVVuVXEpjyFzkMamxb6InrGqbsGGkEUqGMSr9aaQ85N02MMrM6T6JuyqSSssFg2xuO+P4=",
+		"secret":    "",
+	}
+	t.Log("Testing Invalid: message vs. signature")
+	resp, err = postReq(req)
+	if err != nil {
+		t.Error(err)
+	}
+	m, err = getRespText(resp)
+	if err != nil {
+		t.Error(err)
+	} else {
+		t.Log("Server replied:", m.Secret)
+	}
+
+	// Invalid: signature format
+	req = map[string]string{
+		"nodetype":  "node",
+		"address":   "11moup7v3b7a4gt20onion",
+		"message":   "I am a DAM node!",
+		"signature": "this is not base64",
+		"secret":    "",
+	}
+	t.Log("Testing Invalid: signature format")
+	resp, err = postReq(req)
+	if err != nil {
+		t.Error(err)
+	}
+	m, err = getRespText(resp)
+	if err != nil {
+		t.Error(err)
+	} else {
+		t.Log("Server replied:", m.Secret)
+	}
+}
+
+func TestMain(m *testing.M) {
+	cmd := exec.Command("./dam-dir")
+	cmd.SysProcAttr = &syscall.SysProcAttr{Setpgid: true}
+	cmd.Start()
+	time.Sleep(1000 * time.Millisecond)
+
+	ex := m.Run()
+	syscall.Kill(-cmd.Process.Pid, syscall.SIGKILL)
+	os.Exit(ex)
+}