go-gitlooseleaf/net_mirror.go

// copy this file to modules/graceful/net_anon.go before building gitea
package graceful

import (
	"fmt"
	"net"
	"net/http"
	"net/url"
	"os"
	"strings"

	"github.com/go-i2p/go-meta-listener/mirror"
)

// This implements the GetListener function for TLS, I2P, and Onion. Note the exemption for Unix sockets.
func MultiGetListener(network, address string) (net.Listener, error) {
	EMAIL, err := os.Getenv("EMAIL")
	if err != nil {
		return nil, err
	}
	if EMAIL == "" {
		return nil, fmt.Errorf("EMAIL environment variable not set, TLS not possible")
	}
	// Add a deferral to say that we've tried to grab a listener
	defer GetManager().InformCleanup()
	switch network {
	case "unix", "unixpacket":
		// I2P isn't really a replacement for the stuff you use Unix sockets for and it's also not an anonymity risk, so treat them normally
		unixAddr, err := ResolveUnixAddr(network, address)
		if err != nil {
			return nil, err
		}
		return GetListenerUnixWrapper(network, unixAddr)
	default:
		return mirror.Listen(address, EMAIL, "./certs", true)
	}
}

// We use `init() to ensure that the appropriate Listeners and Dialers are correctly placed at runtime
func init() {
	GetListener = MultiGetListener
	httpClient := &http.Client{
		Transport: &http.Transport{
			Dial: Dial,
		},
	}

	http.DefaultClient = httpClient
	http.DefaultTransport = httpClient.Transport
}

func Dial(network, addr string) (net.Conn, error) {
	// convert the addr to a URL
	url, err := url.Parse(addr)
	if err != nil {
		return nil, err
	}
	// get the domain name
	domain := url.Hostname()
	// get the top-level domain
	fr := strings.Split(domain, ".")
	tld := fr[len(fr)-1]
	switch tld {
	case "i2p":
		// I2P is a special case, we need to use the garlic dialer
		return Garlic.Dial(addr)
	case "onion":
		// Onion is a special case, we need to use the onion dialer
		return Onion.Dial(addr)
	default:
		// For everything else, we can use the default dialer
		return net.Dial(network, addr)
	}
}