arblant
firefox-vpn-client / core / internal / service / serversetup_test.go
serversetup_test.go65 lines · 2.4 KB
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
package service

import (
	"regexp"
	"strings"
	"testing"

	"golang.org/x/crypto/bcrypt"
)

// A new server's dashboard and proxy open to the network, each behind its
// own random password; the ports stay.
func TestSetupServer(t *testing.T) {
	cfg := newTestDaemon(t).config()
	cfg.Dashboard.Listen, cfg.Proxy.Listen = "127.0.0.1:9898", "127.0.0.1:9899"
	login, err := SetupServer(&cfg, " ops ")
	if err != nil {
		t.Fatal(err)
	}
	if cfg.Dashboard.Listen != "0.0.0.0:9898" || cfg.Proxy.Listen != "0.0.0.0:9899" {
		t.Fatalf("listen addresses: dashboard %q, proxy %q", cfg.Dashboard.Listen, cfg.Proxy.Listen)
	}
	if login.DashboardPort != "9898" || login.ProxyPort != "9899" {
		t.Fatalf("ports: %+v", login)
	}
	if login.DashboardUsername != "ops" || cfg.Dashboard.Username != "ops" || !regexp.MustCompile(`^proxy[2-9]{4}$`).MatchString(login.ProxyUsername) || cfg.Proxy.Username != login.ProxyUsername {
		t.Fatalf("usernames: %+v", login)
	}
	if len(login.DashboardPassword) != 16 || len(login.ProxyPassword) != 16 || login.DashboardPassword == login.ProxyPassword {
		t.Fatalf("passwords must be 16 characters and differ: %+v", login)
	}
	if bcrypt.CompareHashAndPassword([]byte(cfg.Dashboard.PasswordHash), []byte(login.DashboardPassword)) != nil {
		t.Fatal("the stored hash does not match the dashboard password")
	}
	if cfg.Proxy.Password != login.ProxyPassword {
		t.Fatal("the proxy password was not stored")
	}

	empty := newTestDaemon(t).config()
	if login, err := SetupServer(&empty, "-"); err != nil || !regexp.MustCompile(`^panel[2-9]{4}$`).MatchString(login.DashboardUsername) {
		t.Fatalf("an empty username becomes admin: %+v, %v", login, err)
	}
	for _, bad := range []string{"a b", "a:b", strings.Repeat("x", 65)} {
		cfg := newTestDaemon(t).config()
		if _, err := SetupServer(&cfg, bad); err == nil {
			t.Fatalf("username %q accepted", bad)
		}
	}
}

// The dashboard never opens to other devices without a login: settings
// that would do it are refused.
func TestDashboardNeedsLoginToOpen(t *testing.T) {
	d := newTestDaemon(t)
	cfg := d.config()
	cfg.Dashboard.Listen = "0.0.0.0:10820"
	if _, err := d.updateSettings(cfg, nil); err == nil || !strings.Contains(err.Error(), "username and password") {
		t.Fatalf("opening the dashboard without a login must be refused, got %v", err)
	}
	cfg.Dashboard.Username = "admin"
	password := "a long password"
	if _, err := d.updateSettings(cfg, &password); err != nil {
		t.Fatalf("with a login it may open: %v", err)
	}
}