go/libkb/env_test.go - master - githubmirror/client

go/libkb/env_test.go
178 строк5 KB
zoom-ua
gofumpt (#28710)
17 дек 2025, 17:16
17 дек 2025, 17:16cda510b
// Copyright 2015 Keybase, Inc. All rights reserved. Use of
// this source code is governed by the included BSD license.

package libkb

import (
	"os"
	"path/filepath"
	"testing"

	"github.com/stretchr/testify/require"
)

func TestEnvDarwin(t *testing.T) {
	env := newEnv(nil, nil, "darwin", makeLogGetter(t))

	sockFile, err := env.GetSocketBindFile()
	if err != nil {
		t.Fatal(err)
	}

	cacheDir := env.GetSandboxCacheDir()
	expectedSockFile := filepath.Join(cacheDir, "keybased.sock")
	if sockFile != expectedSockFile {
		t.Fatalf("Clients expect sock file to be %s", expectedSockFile)
	}
}

type MockedConfigReader struct {
	NullConfiguration
}

var globalTorMode = TorNone

func (nc MockedConfigReader) GetTorMode() (TorMode, error) {
	return globalTorMode, nil
}

var globalProxyType = ""

func (nc MockedConfigReader) GetProxyType() string {
	return globalProxyType
}

var globalProxyAddress = ""

func (nc MockedConfigReader) GetProxy() string {
	return globalProxyAddress
}

var globalIsCertPinningEnabled = true

func (nc MockedConfigReader) IsCertPinningEnabled() bool {
	return globalIsCertPinningEnabled
}

func resetGlobals() {
	globalTorMode = TorNone
	globalProxyType = ""
	globalProxyAddress = ""
	globalIsCertPinningEnabled = true
}

func TestTorMode(t *testing.T) {
	resetGlobals()

	mockedEnv := NewEnv(MockedConfigReader{}, MockedConfigReader{}, makeLogGetter(t))

	// Test that when tor mode is enabled, a Socks proxy is properly configured
	require.Equal(t, NoProxy, mockedEnv.GetProxyType())
	require.Equal(t, "", mockedEnv.GetProxy())

	globalTorMode = TorLeaky
	require.Equal(t, Socks, mockedEnv.GetProxyType())
	require.Equal(t, "localhost:9050", mockedEnv.GetProxy())

	globalTorMode = TorStrict
	require.Equal(t, Socks, mockedEnv.GetProxyType())
	require.Equal(t, "localhost:9050", mockedEnv.GetProxy())

	// Test that tor mode overrides proxy settings
	globalProxyType = "http"
	globalProxyAddress = "localhost:8080"
	require.Equal(t, Socks, mockedEnv.GetProxyType())
	require.Equal(t, "localhost:9050", mockedEnv.GetProxy())
}

func TestGetProxyType(t *testing.T) {
	resetGlobals()

	defaultEnv := NewEnv(nil, nil, makeLogGetter(t))
	require.Equal(t, NoProxy, defaultEnv.GetProxyType())

	mockedEnv := NewEnv(MockedConfigReader{}, MockedConfigReader{}, makeLogGetter(t))
	require.Equal(t, NoProxy, mockedEnv.GetProxyType())

	globalProxyType = "Socks"
	require.Equal(t, Socks, mockedEnv.GetProxyType())
	globalProxyType = "SOCKS"
	require.Equal(t, Socks, mockedEnv.GetProxyType())
	globalProxyType = "SoCkS"
	require.Equal(t, Socks, mockedEnv.GetProxyType())

	globalProxyType = "http_connect"
	require.Equal(t, HTTPConnect, mockedEnv.GetProxyType())
	globalProxyType = "HTTP_CONNECT"
	require.Equal(t, HTTPConnect, mockedEnv.GetProxyType())

	globalProxyType = "BOGUS"
	require.Equal(t, NoProxy, mockedEnv.GetProxyType())

	resetGlobals()
	require.Equal(t, NoProxy, mockedEnv.GetProxyType())

	orig := os.Getenv("PROXY_TYPE")

	os.Setenv("PROXY_TYPE", "socks")
	require.Equal(t, Socks, mockedEnv.GetProxyType())
	os.Setenv("PROXY_TYPE", "http_connect")
	require.Equal(t, HTTPConnect, mockedEnv.GetProxyType())

	os.Setenv("PROXY_TYPE", orig)
}

func TestGetProxy(t *testing.T) {
	resetGlobals()

	defaultEnv := NewEnv(nil, nil, makeLogGetter(t))
	require.Equal(t, "", defaultEnv.GetProxy())

	mockedEnv := NewEnv(MockedConfigReader{}, MockedConfigReader{}, makeLogGetter(t))
	require.Equal(t, "", mockedEnv.GetProxy())

	globalProxyAddress = "localhost:8090"
	require.Equal(t, "localhost:8090", mockedEnv.GetProxy())

	resetGlobals()
	require.Equal(t, "", defaultEnv.GetProxy())

	orig := os.Getenv("PROXY")
	os.Setenv("PROXY", "localhost:8080")
	require.Equal(t, "localhost:8080", mockedEnv.GetProxy())
	os.Setenv("PROXY", orig)

	orig = os.Getenv("HTTP_PROXY")
	os.Setenv("HTTP_PROXY", "localhost:8081")
	require.Equal(t, "localhost:8081", mockedEnv.GetProxy())
	os.Setenv("HTTP_PROXY", orig)

	orig = os.Getenv("HTTPS_PROXY")
	os.Setenv("HTTPS_PROXY", "localhost:8082")
	require.Equal(t, "localhost:8082", mockedEnv.GetProxy())
	os.Setenv("HTTPS_PROXY", orig)
}

func TestIsCertPinningEnabled(t *testing.T) {
	resetGlobals()

	defaultEnv := NewEnv(nil, nil, makeLogGetter(t))
	require.Equal(t, true, defaultEnv.IsCertPinningEnabled())

	mockedEnv := NewEnv(MockedConfigReader{}, MockedConfigReader{}, makeLogGetter(t))
	require.Equal(t, true, mockedEnv.IsCertPinningEnabled())

	globalIsCertPinningEnabled = false
	require.Equal(t, false, mockedEnv.IsCertPinningEnabled())

	globalIsCertPinningEnabled = true
	require.Equal(t, true, mockedEnv.IsCertPinningEnabled())

	orig := os.Getenv("DISABLE_SSL_PINNING")
	os.Setenv("DISABLE_SSL_PINNING", "true")
	require.Equal(t, false, mockedEnv.IsCertPinningEnabled())
	os.Setenv("DISABLE_SSL_PINNING", "no")
	require.Equal(t, true, mockedEnv.IsCertPinningEnabled())
	os.Setenv("DISABLE_SSL_PINNING", orig)
	require.Equal(t, true, mockedEnv.IsCertPinningEnabled())
}