Feat: AllowProxyToLocal config

internal/conf/proxy.go

@@ -3,11 +3,13 @@ package conf
 type ProxyConfig struct {
 	MovieProxy        bool `yaml:"movie_proxy" env:"PROXY_MOVIE"`
 	LiveProxy         bool `yaml:"live_proxy" env:"PROXY_LIVE"`
+	AllowProxyToLocal bool `yaml:"allow_proxy_to_local" env:"PROXY_ALLOW_TO_LOCAL"`
 }
 
 func DefaultProxyConfig() ProxyConfig {
 	return ProxyConfig{
 		MovieProxy:        true,
 		LiveProxy:         true,
+		AllowProxyToLocal: false,
 	}
 }

internal/model/movie.go

@@ -67,7 +67,7 @@ func (m *BaseMovie) Validate() error {
 		if err != nil {
 			return err
 		}
-		if utils.IsLocalIP(u.Host) {
+		if !conf.Conf.Proxy.AllowProxyToLocal && utils.IsLocalIP(u.Host) {
 			return errors.New("local ip is not allowed")
 		}
 		switch u.Scheme {
@@ -89,7 +89,7 @@ func (m *BaseMovie) Validate() error {
 		if err != nil {
 			return err
 		}
-		if utils.IsLocalIP(u.Host) {
+		if !conf.Conf.Proxy.AllowProxyToLocal && utils.IsLocalIP(u.Host) {
 			return errors.New("local ip is not allowed")
 		}
 		if u.Scheme != "http" && u.Scheme != "https" {

server/handlers/movie.go

@@ -499,8 +499,12 @@ func ProxyMovie(ctx *gin.Context) {
 }
 
 func proxyURL(ctx *gin.Context, u string, headers map[string]string) error {
-	if l, err := utils.ParseURLIsLocalIP(u); err != nil || l {
+	if !conf.Conf.Proxy.AllowProxyToLocal {
+		if l, err := utils.ParseURLIsLocalIP(u); err != nil {
 			return err
+		} else if l {
+			return errors.New("not allow proxy to local")
+		}
 	}
 	r, err := http.NewRequestWithContext(ctx, http.MethodGet, u, nil)
 	if err != nil {