update

examples/custom_dns_resolver/main.go

@@ -0,0 +1,132 @@
+package main
+
+import (
+	"context"
+	"flag"
+	"log/slog"
+	"net/http"
+	"os"
+	"os/signal"
+	"syscall"
+	"time"
+
+	"github.com/darkit/goproxy"
+	"github.com/darkit/goproxy/pkg/dns"
+)
+
+func main() {
+	// 解析命令行参数
+	var (
+		listenAddr  = flag.String("listen", ":8080", "代理服务器监听地址")
+		targetAddr  = flag.String("target", "http://example.com", "目标服务器地址（反向代理模式下使用）")
+		proxyMode   = flag.String("mode", "reverse", "代理模式: forward, reverse, transparent")
+		enableHTTPS = flag.Bool("https", false, "是否启用HTTPS拦截")
+		verbosity   = flag.Int("v", 0, "日志详细级别 (0-3)")
+		host        = flag.String("dns-host", "example.com", "要解析的主机名")
+		ip          = flag.String("dns-ip", "127.0.0.1", "解析的IP地址")
+	)
+	flag.Parse()
+
+	// 配置日志
+	logLevel := slog.LevelInfo
+	switch *verbosity {
+	case 1:
+		logLevel = slog.LevelDebug
+	case 2:
+		logLevel = slog.LevelInfo
+	case 3:
+		logLevel = slog.LevelDebug - 3 // 更详细的调试级别
+	}
+
+	logger := slog.New(slog.NewTextHandler(os.Stdout, &slog.HandlerOptions{
+		Level: logLevel,
+	}))
+	slog.SetDefault(logger)
+
+	// 创建自定义DNS解析器
+	resolver := dns.NewResolver(
+		dns.WithFallback(true),     // 如果找不到自定义规则，回退到系统DNS
+		dns.WithTTL(5*time.Minute), // 设置缓存TTL
+	)
+
+	// 添加自定义DNS解析规则
+	if err := resolver.Add(*host, *ip); err != nil {
+		slog.Error("添加DNS解析规则失败", "error", err)
+		return
+	}
+
+	// 添加通配符DNS解析规则示例
+	if err := resolver.AddWildcard("*.example.org", "192.168.1.2"); err != nil {
+		slog.Error("添加通配符DNS解析规则失败", "error", err)
+		return
+	}
+
+	slog.Info("已添加DNS解析规则", "host", *host, "ip", *ip)
+
+	// 创建代理选项
+	var (
+		proxy goproxy.UnifiedProxy
+		err   error
+	)
+
+	// 根据代理模式创建不同类型的代理
+	switch *proxyMode {
+	case "forward":
+		proxy, err = goproxy.NewForwardProxy(*listenAddr, goproxy.WithUnifiedDNSResolver(resolver))
+	case "reverse":
+		proxy, err = goproxy.NewReverseProxy(*listenAddr, *targetAddr, goproxy.WithUnifiedDNSResolver(resolver))
+	case "transparent":
+		proxy, err = goproxy.NewTransparentProxy(*listenAddr, goproxy.WithUnifiedDNSResolver(resolver))
+	default:
+		slog.Error("不支持的代理模式", "mode", *proxyMode)
+		return
+	}
+
+	if err != nil {
+		slog.Error("创建代理失败", "error", err)
+		return
+	}
+
+	// 设置信号处理
+	sigCh := make(chan os.Signal, 1)
+	signal.Notify(sigCh, syscall.SIGINT, syscall.SIGTERM)
+
+	// 打印启动信息
+	slog.Info("代理服务器已启动",
+		"mode", *proxyMode,
+		"listen", *listenAddr,
+		"target", *targetAddr,
+		"https", *enableHTTPS)
+
+	// 启动HTTP服务器
+	server := &http.Server{
+		Addr:    *listenAddr,
+		Handler: proxy,
+	}
+
+	// 监听和服务连接
+	go func() {
+		if err := server.ListenAndServe(); err != nil && err != http.ErrServerClosed {
+			slog.Error("服务器运行失败", "error", err)
+			os.Exit(1)
+		}
+	}()
+
+	// 等待中断信号
+	<-sigCh
+	slog.Info("接收到信号，正在关闭...")
+
+	// 创建关闭上下文
+	ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
+	defer cancel()
+
+	// 优雅关闭HTTP服务器
+	if err := server.Shutdown(ctx); err != nil {
+		slog.Error("服务器关闭失败", "error", err)
+	}
+
+	// 关闭代理
+	if err := proxy.Close(); err != nil {
+		slog.Error("关闭代理时出错", "error", err)
+	}
+}

examples/dns_resolver_comparison/main.go

@@ -0,0 +1,112 @@
+package main
+
+import (
+	"flag"
+	"log/slog"
+	"net"
+	"net/http"
+	"net/url"
+	"os"
+	"time"
+
+	"github.com/darkit/goproxy"
+	"github.com/darkit/goproxy/pkg/dns"
+)
+
+func main() {
+	// 解析命令行参数
+	var (
+		listenAddr     = flag.String("listen", ":8080", "代理服务器监听地址")
+		targetAddr     = flag.String("target", "http://example.com", "目标服务器地址")
+		useDNSResolver = flag.Bool("custom-dns", false, "是否使用自定义DNS解析")
+		dnsHost        = flag.String("dns-host", "example.com", "要解析的主机名")
+		dnsIP          = flag.String("dns-ip", "127.0.0.1", "解析的IP地址")
+		verbosity      = flag.Int("v", 0, "日志详细级别 (0-3)")
+	)
+	flag.Parse()
+
+	// 配置日志
+	logLevel := slog.LevelInfo
+	switch *verbosity {
+	case 1:
+		logLevel = slog.LevelDebug
+	case 2:
+		logLevel = slog.LevelInfo
+	case 3:
+		logLevel = slog.LevelDebug - 3 // 更详细的调试级别
+	}
+
+	logger := slog.New(slog.NewTextHandler(os.Stdout, &slog.HandlerOptions{
+		Level: logLevel,
+	}))
+	slog.SetDefault(logger)
+
+	// 创建反向代理实例
+	var proxy goproxy.UnifiedProxy
+	var err error
+
+	if *useDNSResolver {
+		// 使用自定义DNS解析器
+		slog.Info("使用自定义DNS解析器")
+
+		// 创建自定义DNS解析器
+		resolver := dns.NewResolver(
+			dns.WithFallback(true),     // 如果找不到自定义规则，回退到系统DNS
+			dns.WithTTL(5*time.Minute), // 设置缓存TTL
+		)
+
+		// 添加自定义DNS解析规则
+		if err := resolver.Add(*dnsHost, *dnsIP); err != nil {
+			slog.Error("添加DNS解析规则失败", "error", err)
+			return
+		}
+
+		slog.Info("添加DNS解析规则", "host", *dnsHost, "ip", *dnsIP)
+
+		// 创建带自定义DNS解析器的反向代理
+		proxy, err = goproxy.NewReverseProxy(*listenAddr, *targetAddr, goproxy.WithUnifiedDNSResolver(resolver))
+	} else {
+		// 使用系统DNS解析
+		slog.Info("使用系统DNS解析")
+		proxy, err = goproxy.NewReverseProxy(*listenAddr, *targetAddr)
+	}
+
+	if err != nil {
+		slog.Error("创建代理服务器失败", "error", err)
+		return
+	}
+
+	// 使用系统DNS解析一次目标地址，打印结果供比较
+	targetHost := *targetAddr
+	// 尝试解析URL
+	if u, err := url.Parse(targetHost); err == nil && u.Host != "" {
+		targetHost = u.Host
+	}
+	// 如果还包含端口，移除端口
+	if host, _, err := net.SplitHostPort(targetHost); err == nil {
+		targetHost = host
+	}
+	ips, err := net.LookupIP(targetHost)
+	if err != nil {
+		slog.Error("系统DNS解析失败", "host", targetHost, "error", err)
+	} else {
+		slog.Info("系统DNS解析结果", "host", targetHost, "ips", ips)
+	}
+
+	// 创建HTTP服务器
+	server := &http.Server{
+		Addr:    *listenAddr,
+		Handler: proxy,
+	}
+
+	// 启动服务器
+	slog.Info("代理服务器已启动",
+		"listen", *listenAddr,
+		"target", *targetAddr,
+		"custom_dns", *useDNSResolver)
+
+	// 监听和服务连接
+	if err := server.ListenAndServe(); err != nil {
+		slog.Error("服务器运行失败", "error", err)
+	}
+}

examples/rewriter/http_server.go

@@ -90,7 +90,7 @@ func RewriteMiddleware(rw *rewriter.Rewriter, next http.Handler) http.Handler {
 	})
 }
 
-func main() {
+func main1() {
 	// 创建重写器
 	rw := rewriter.NewRewriter()
 

examples/rewriter/main.go

@@ -10,7 +10,7 @@ import (
 	"github.com/darkit/goproxy/pkg/rewriter"
 )
 
-func main() {
+func main2() {
 	// 创建URL重写器
 	rw := rewriter.NewRewriter()