config.go

package main

import (
	"errors"
	"net"

	"snet/config"
	"snet/proxy"
	"snet/proxy/http"
	"snet/proxy/socks5"
	"snet/proxy/ss"
	"snet/proxy/ss2"
	"snet/proxy/tls"
)

func genConfigByType(c *config.Config, proxyType string) (proxy.Config, error) {
	switch proxyType {
	case "ss":
		ip, err := resolvHostIP(c.SSHost)
		if err != nil {
			return nil, err
		}
		var cipher string
		// backwards compatibility for typo "ss-cphier-method"
		if c.SSCipherMethod != "" {
			cipher = c.SSCipherMethod
		} else {
			cipher = c.SSChpierMethod
		}
		return &ss.Config{Host: ip, Port: c.SSPort, CipherMethod: cipher, Password: c.SSPasswd}, nil
	case "ss2":
		ip, err := resolvHostIP(c.SS2Host)
		if err != nil {
			return nil, err
		}
		return &ss2.Config{Host: ip, Port: c.SS2Port, CipherMethod: c.SS2CipherMethod, Password: c.SS2Passwd, Key: c.SS2Key}, nil
	case "http":
		ip, err := resolvHostIP(c.HTTPProxyHost)
		if err != nil {
			return nil, err
		}
		return &http.Config{Host: ip, Port: c.HTTPProxyPort, AuthUser: c.HTTPProxyAuthUser, AuthPassword: c.HTTPProxyAuthPassword}, nil
	case "tls":
		ip, err := resolvHostIP(c.TLSHost)
		if err != nil {
			return nil, err
		}
		return &tls.Config{Host: ip, Port: c.TLSPort, Token: c.TLSToken}, nil
	case "socks5":
		ip, err := resolvHostIP(c.SOCKS5Host)
		if err != nil {
			return nil, err
		}
		return &socks5.Config{Host: ip, Port: c.SOCKS5Port, AuthUser: c.SOCKS5AuthUser, AuthPassword: c.SOCKS5AuthPassword}, nil
	}
	return nil, nil
}

func resolvHostIP(host string) (net.IP, error) {
	ips, err := net.LookupIP(host)
	if err != nil {
		return nil, err
	}
	if len(ips) == 0 {
		return nil, errors.New("No ip found for " + host)
	}
	return ips[0], nil
}