frp-panel/conf/helper.go

package conf

import (
	"context"
	"crypto/x509"
	"crypto/x509/pkix"
	"fmt"
	"math/big"
	"net"
	"net/url"
	"time"

	"github.com/VaalaCat/frp-panel/defs"
	"github.com/VaalaCat/frp-panel/utils"
	"github.com/VaalaCat/frp-panel/utils/logger"
	v1 "github.com/fatedier/frp/pkg/config/v1"
)

func RPCListenAddr(cfg Config) string {
	return fmt.Sprintf(":%d", cfg.Master.RPCPort)
}

func rpcCallAddr(cfg Config) string {
	return fmt.Sprintf("%s:%d", cfg.Master.RPCHost, cfg.Master.RPCPort)
}

func JWTSecret(cfg Config) string {
	return utils.SHA1(fmt.Sprintf("%s:%d:%s", cfg.Master.APIHost, cfg.Master.APIPort, cfg.App.GlobalSecret))
}

func MasterAPIListenAddr(cfg Config) string {
	return fmt.Sprintf(":%d", cfg.Master.APIPort)
}

func ServerAPIListenAddr(cfg Config) string {
	return fmt.Sprintf(":%d", cfg.Server.APIPort)
}

func FRPsAuthOption(cfg Config, isDefault bool) v1.HTTPPluginOptions {
	var port int
	if isDefault {
		port = cfg.Master.APIPort
	} else {
		port = cfg.Master.InternalFRPAuthServerPort
	}
	authUrl, err := url.Parse(fmt.Sprintf("http://%s:%d%s",
		cfg.Master.InternalFRPAuthServerHost,
		port,
		cfg.Master.InternalFRPAuthServerPath))
	if err != nil {
		logger.Logger(context.Background()).WithError(err).Fatalf("parse auth url error")
	}
	return v1.HTTPPluginOptions{
		Name: "multiuser",
		Ops:  []string{"Login"},
		Addr: authUrl.Host,
		Path: authUrl.Path,
	}
}

func GetJWTWithPayload(cfg Config, uid int, payload map[string]interface{}) (string, error) {
	payload[defs.UserIDKey] = uid
	return utils.GetJwtTokenFromMap(JWTSecret(cfg),
		time.Now().Unix(),
		int64(cfg.App.CookieAge),
		payload)
}

func GetJWTWithAllPermission(cfg Config, uid int) string {
	token, _ := GetJWTWithPayload(cfg, uid, map[string]interface{}{
		defs.TokenPayloadKey_Permissions: AllPermission(),
	})
	return token
}

func AllPermission() []defs.APIPermission {
	return []defs.APIPermission{{
		Method: "*",
		Path:   "*",
	}}
}

func GetCommonJWT(cfg Config, uid int) string {
	token, _ := GetJWTWithPayload(cfg, uid, map[string]interface{}{})
	return token
}

func GetCommonJWTWithExpireTime(cfg Config, uid int64, expSec int) string {
	token, _ := utils.GetJwtTokenFromMap(JWTSecret(cfg),
		time.Now().Unix(),
		int64(expSec),
		map[string]interface{}{defs.UserIDKey: uid})
	return token
}

func GetAPIURL(cfg Config) string {

	if len(cfg.Client.APIUrl) != 0 {
		return cfg.Client.APIUrl
	}

	return fmt.Sprintf("%s://%s:%d", cfg.Master.APIScheme, cfg.Master.APIHost, cfg.Master.APIPort)
}

func GetCertTemplate(cfg Config) *x509.Certificate {
	now := time.Now()
	return &x509.Certificate{
		SerialNumber: big.NewInt(now.Unix()),
		Subject: pkix.Name{
			CommonName:         cfg.Master.APIHost,
			Country:            []string{"CN"},
			Organization:       []string{"frp-panel"},
			OrganizationalUnit: []string{"frp-panel"},
		},
		SignatureAlgorithm:    x509.SHA512WithRSA,
		DNSNames:              []string{cfg.Master.APIHost},
		IPAddresses:           []net.IP{net.ParseIP("127.0.0.1"), net.ParseIP("::1")},
		NotBefore:             now,
		NotAfter:              now.AddDate(10, 0, 0),
		SubjectKeyId:          []byte{102, 114, 112, 45, 112, 97, 110, 101, 108},
		BasicConstraintsValid: true,
		IsCA:                  true,
		ExtKeyUsage:           []x509.ExtKeyUsage{x509.ExtKeyUsageServerAuth, x509.ExtKeyUsageClientAuth},
		KeyUsage: x509.KeyUsageKeyEncipherment |
			x509.KeyUsageDigitalSignature | x509.KeyUsageCertSign | x509.KeyUsageKeyAgreement |
			x509.KeyUsageDataEncipherment,
	}
}

type LisOpt struct {
	MuxLis net.Listener
	ApiLis net.Listener
	RunAPI bool
}

func GetListener(c context.Context, cfg Config) LisOpt {
	runAPI := RPCListenAddr(cfg) != MasterAPIListenAddr(cfg)

	muxLis, err := net.Listen("tcp", RPCListenAddr(cfg))
	if err != nil {
		logger.Logger(c).WithError(err).Fatalf("failed to listen: %v", RPCListenAddr(cfg))
	}

	opt := LisOpt{
		MuxLis: muxLis,
		RunAPI: runAPI,
	}

	if runAPI {
		apiLis, err := net.Listen("tcp", MasterAPIListenAddr(cfg))
		if err != nil {
			logger.Logger(c).WithError(err).Warnf("failed to listen: %v, but mux server can handle http api", MasterAPIListenAddr(cfg))
		}
		opt.ApiLis = apiLis
	}

	if !runAPI {
		opt.ApiLis = nil
	}

	return opt
}

type ConnInfo struct {
	Host   string
	Scheme Scheme
}

type Scheme string

const (
	GRPC Scheme = "grpc"
	WS   Scheme = "ws"
	WSS  Scheme = "wss"
)

func GetRPCConnInfo(cfg Config) ConnInfo {
	rpcUrl := cfg.Client.RPCUrl

	if len(rpcUrl) == 0 {
		return ConnInfo{
			Host:   rpcCallAddr(cfg),
			Scheme: GRPC,
		}
	}

	parsedUrl, err := url.Parse(rpcUrl)
	if err != nil {
		logger.Logger(context.Background()).WithError(err).Fatalf("parse rpc url error")
	}

	return ConnInfo{
		Host:   parsedUrl.Host,
		Scheme: Scheme(parsedUrl.Scheme),
	}
}