Go - 实现简单的Socks5代理

概述

这是一个使用 Go 语言实现的 SOCKS5 代理服务器，支持 TCP 和 UDP 转发。它基于 github.com/armon/go-socks5 库，并手动实现了 UDP 关联功能。

代码

package main

import (
	"context"
	"fmt"
	"log"
	"net"
	"github.com/armon/go-socks5"
)

// UDPRelay handles UDP forwarding
func UDPRelay(udpAddr *net.UDPAddr) {
	conn, err := net.ListenUDP("udp", udpAddr)
	if err != nil {
		log.Fatalf("Failed to listen on UDP: %v", err)
	}
	defer conn.Close()

	buffer := make([]byte, 65535)
	for {
		n, addr, err := conn.ReadFromUDP(buffer)
		if err != nil {
			log.Printf("UDP read error: %v", err)
			continue
		}

		// Forwarding UDP packets (you may add rules here)
		log.Printf("Received UDP packet from %v, size: %d", addr, n)
		_, err = conn.WriteToUDP(buffer[:n], addr)
		if err != nil {
			log.Printf("UDP write error: %v", err)
		}
	}
}

func main() {
	// Create a SOCKS5 server
	srv, err := socks5.New(&socks5.Config{
		Logger: log.Default(),
		Rules:  socks5.PermitAll(),
	})
	if err != nil {
		log.Fatalf("Failed to create SOCKS5 server: %v", err)
	}

	// Start UDP relay in a separate goroutine
	udpAddr, _ := net.ResolveUDPAddr("udp", ":1081")
	go UDPRelay(udpAddr)

	// Start the SOCKS5 server
	log.Println("Starting SOCKS5 proxy on :1080")
	if err := srv.ListenAndServe("tcp", ":1080"); err != nil {
		log.Fatalf("Failed to start SOCKS5 server: %v", err)
	}
}

TCP 代理：

• 使用 github.com/armon/go-socks5 处理 TCP 代理请求。

UDP 关联：

• 在 :1081 端口监听 UDP 数据包。
• 直接回写数据包以支持 UDP 转发（可扩展为完整的 UDP 代理）。

启动方式：

• 运行 SOCKS5 服务器监听 :1080 端口（TCP）。
• 运行 UDP 关联监听 :1081 端口（UDP）。