Golang 语言基础(下) | 青训营笔记

186 阅读14分钟

这是我参与「第五营 」笔记创作活动的的第2天

简单项目实战

一、猜谜游戏

简述

在这个游戏里面,程序首先会生成一个介于1100之间的随机整数,然后提示玩家进行猜测。玩家每次输入一个数字,程序会告诉玩家这个猜测的值是高于还是低于那个秘密的随机数,并且让玩家再次猜测。如果猜对了,就告诉玩家胜利并且退出程序。

image.png

PS:运气不错,三次猜出来了。

思路分析

  1. 随机数生成:使用math/rand包,通过rand.Intn来生成一个介于1100之间的随机整数。在这之前需要设置随机数种子,否则的话每一个都会生成相同的随机数序列。这里我们可以用当前的时间戳time.Now().UnixNano()来初始化随机数种子。
  2. 读取用户输入:每个程序执行的时候都会打开几个文件,stdinstdoutstderr等,stdin文件可以用os.Stdin来得到。直接操作这个文件很不方便,我们会用bufio.NewReader把一个文件转换成一个reader变量。reader变量上会有很多用来操作一个流的操作,我们可以用它的ReadString方法来读取一行。如果失败了的话,我们会打印错误并能退出。ReadString返回的结果包含结尾的换行符,我们可以用Trim把它去掉,再转换成数字,如果转换失败,我们同样打印错误并退出。
  3. 实现判断逻辑:我们来比较这两个值的大小,如果是用户输入的值比那个秘密的值要大的话,就告诉用户你猜的值太大了,请再试一次。如果是小了也同理,如果是相等的话,那么我们就告诉用户赢了。
  4. 实现游戏循环:通过一个for循环,在用户猜错时可以继续猜,猜对时退出循环。

Go 代码

package main

import (
    "bufio"
    "fmt"
    "math/rand"
    "os"
    "strconv"
    "strings"
    "time"
)

func main() {
    maxNum := 100
    rand.Seed(time.Now().UnixNano())  // 用当前的时间戳来初始化随机数种子
    secretNumber := rand.Intn(maxNum) // 生成1到100之间的随机整数

    fmt.Println("Please input your guess")
    reader := bufio.NewReader(os.Stdin)
    for {
        input, err := reader.ReadString('\n') // 读取一行输入
        if err != nil {
            fmt.Println("An error occured while reading input. Please try again", err)
            continue
        }

        input = strings.Trim(input, "\r\n") // 去掉换行符

        guess, err := strconv.Atoi(input) // 转换成数字
        if err != nil {
            fmt.Println("Invalid input. Please enter an integer value")
            continue
        }
        fmt.Println("You guess is", guess)
        if guess > secretNumber {
            fmt.Println("Your guess is bigger than the secret number. Please try again")
        } else if guess < secretNumber {
            fmt.Println("Your guess is smaller than the secret number. Please try again")
        } else {
            fmt.Println("Correct, you Legend!")
            break
        }
    }
}

二、在线词典

简述

用户可以在命令行里面查询一个单词。我们能通过调用第三方API的查询到单词的翻译并打印出来。这个例子里面,我们会学习如何用Golang来来发送HTTP请求、解析JSON,还会学习如何使用代码生成来提高开发效率。

image.png

思路分析

  1. 抓包:以彩云小译提供的在线翻译为例。右键检查或F12打开浏览器的开发者工具,此时我们点一下翻译按钮,浏览器会发送一系列请求。这是一个HTTPPOST的请求,请求的header的相当的复杂,请求头是一个json里面有两个字段,一个是代表你要从什么语言转化成什么语言,source就是你要查询的单词。API的返回结果里面会有Wikidictionary两个字段。我们需要用的结果主要在dictionary.Explanations字段里面。其他有些字段里面还包括音标等信息。
image.png
  1. 代码生成:我们需要在Golang里面去发送这个请求。因为这个请求比较复杂,用代码构造很麻烦,实际上我们有一种非常简单的方式来生成代码,我们可以右键浏览器里面的Copy as cURL(bash)复制curl请求。将curl请求粘贴到Convert curl to Go生成Golang代码,我们直接把它copy到我们的编辑器里面。
image.png

生成代码

package main

import (
    "fmt"
    "io/ioutil"
    "log"
    "net/http"
    "strings"
)

func main() {
    client := &http.Client{}
    var data = strings.NewReader(`{"trans_type":"en2zh","source":"good"}`)
    req, err := http.NewRequest("POST", "https://api.interpreter.caiyunai.com/v1/dict", data)
    if err != nil {
        log.Fatal(err)
    }
    req.Header.Set("authority", "api.interpreter.caiyunai.com")
    req.Header.Set("accept", "application/json, text/plain, */*")
    req.Header.Set("accept-language", "zh-CN,zh;q=0.9,en-GB;q=0.8,en;q=0.7,en-US;q=0.6")
    req.Header.Set("app-name", "xy")
    req.Header.Set("content-type", "application/json;charset=UTF-8")
    req.Header.Set("device-id", "")
    req.Header.Set("origin", "https://fanyi.caiyunapp.com")
    req.Header.Set("os-type", "web")
    req.Header.Set("os-version", "")
    req.Header.Set("referer", "https://fanyi.caiyunapp.com/")
    req.Header.Set("sec-ch-ua", `"Not_A Brand";v="99", "Microsoft Edge";v="109", "Chromium";v="109"`)
    req.Header.Set("sec-ch-ua-mobile", "?0")
    req.Header.Set("sec-ch-ua-platform", `"Windows"`)
    req.Header.Set("sec-fetch-dest", "empty")
    req.Header.Set("sec-fetch-mode", "cors")
    req.Header.Set("sec-fetch-site", "cross-site")
    req.Header.Set("user-agent", "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/109.0.0.0 Safari/537.36 Edg/109.0.1518.52")
    req.Header.Set("x-authorization", "token:qgemv4jr1y38jyq6vhvi")
    resp, err := client.Do(req)
    if err != nil {
        log.Fatal(err)
    }
    defer resp.Body.Close()
    bodyText, err := ioutil.ReadAll(resp.Body)
    if err != nil {
        log.Fatal(err)
    }
    fmt.Printf("%s\n", bodyText)
}

首先我们创建了一个HTTP client,创建的时候可以指定很多参数,包括比如请求的超时是否使用cookie等。接下来是构造一个HTTP请求,这是一个POST请求,然后会用到http.NewRequest,第一个参数是HTTP方法POST,第二个参数是URL,最后一个参数是bodybody因为可能很大,为了支持流式发送,是一个只读流。我们用了strings.NewReader来把字符串转换成一个流。这样我们就成功构造了一个HTTPrequest

接下来我们需要对这个HTTPrequest设置header。然后调用client.Do发起请求,就能得到response,如果请求失败的话,那么这个error会返回非nil,会打印错误并且退出进程。response有它的HTTP状态码,response headerbodybody同样是一个流,在Golang里面,为了避免资源泄露,你需要加一个defer来手动关闭这个流,这个defer会在这个函数运行结束之后去执行。接下来我们是用ioutl.ReadAll来读取这个流,能得到整个JSON对象body

  1. 生成 request body:在Golang里面。我们需要生成一段JSON,常用的方式是我们先构造出来一个结构体,这个结构体和我们需要生成的JSON的结构是一一对应的。我们再来定义一个变量,初始化每个结构体成员,再调用JSON.Marshal来得到这个序列化之后的字符串。不同于之前这里是个字符串,我们这里是个字节数组。所以我们需要用bytes.newReaeder来构造那个request上的body接下来代码不变。然后我们就能成功地进入一个变量来发送HTTP请求。
type DictRequest struct {
    TransType string `json:"trans_type"`
    Source    string `json:"source"`
    UserID    string `json:"user_id"`
}
  1. 解析 request body:此时有一个小技巧的是,网上有对应的代码生成工具,我们可以打开JSON转Golang Struct,把JSON字符串粘贴进去,这样我们就能够生成对应结构体。在某些时刻,我们如果不需要对这个返回结果做很多精细的操作,我们可以选择转换嵌套,能让生成的代码更加紧凑。
image.png

这样我们就得到了一个response结构体。接下来我们修改代码,我们先定一个response结构体的对象,然后我们用json.Unmarshalbody反序列化到这个结构体里面,再打印出response里面的特定字段。

image.png

观察JSON可以看出我们需要的结果是在Dictionary.explanations中。我们用for range循环来迭代它,然后直接打印结构,参照一些词典的显示方式,我们可以在那个前面打印出这个单词和它的音标。这里有英式音标和美式音标。

  1. 完善代码:现在我们的程序的输入还是写死的。我们把代码的主体改成一个query函数,查询的单词作为参数传递进来。

Go 代码

package main

import (
	"bytes"
	"encoding/json"
	"fmt"
	"io/ioutil"
	"log"
	"net/http"
	"os"
)

type DictRequest struct {
	TransType string `json:"trans_type"`
	Source    string `json:"source"`
	UserID    string `json:"user_id"`
}

type DictResponse struct {
	Rc   int `json:"rc"`
	Wiki struct {
		KnownInLaguages int `json:"known_in_laguages"`
		Description     struct {
			Source string      `json:"source"`
			Target interface{} `json:"target"`
		} `json:"description"`
		ID   string `json:"id"`
		Item struct {
			Source string `json:"source"`
			Target string `json:"target"`
		} `json:"item"`
		ImageURL  string `json:"image_url"`
		IsSubject string `json:"is_subject"`
		Sitelink  string `json:"sitelink"`
	} `json:"wiki"`
	Dictionary struct {
		Prons struct {
			EnUs string `json:"en-us"`
			En   string `json:"en"`
		} `json:"prons"`
		Explanations []string      `json:"explanations"`
		Synonym      []string      `json:"synonym"`
		Antonym      []string      `json:"antonym"`
		WqxExample   [][]string    `json:"wqx_example"`
		Entry        string        `json:"entry"`
		Type         string        `json:"type"`
		Related      []interface{} `json:"related"`
		Source       string        `json:"source"`
	} `json:"dictionary"`
}

func query(word string) {
	client := &http.Client{}
	request := DictRequest{TransType: "en2zh", Source: word}
	buf, err := json.Marshal(request)
	if err != nil {
		log.Fatal(err)
	}
	var data = bytes.NewReader(buf)
	req, err := http.NewRequest("POST", "https://api.interpreter.caiyunai.com/v1/dict", data)
	if err != nil {
		log.Fatal(err)
	}
	req.Header.Set("Connection", "keep-alive")
	req.Header.Set("DNT", "1")
	req.Header.Set("os-version", "")
	req.Header.Set("sec-ch-ua-mobile", "?0")
	req.Header.Set("User-Agent", "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/99.0.4844.51 Safari/537.36")
	req.Header.Set("app-name", "xy")
	req.Header.Set("Content-Type", "application/json;charset=UTF-8")
	req.Header.Set("Accept", "application/json, text/plain, */*")
	req.Header.Set("device-id", "")
	req.Header.Set("os-type", "web")
	req.Header.Set("X-Authorization", "token:qgemv4jr1y38jyq6vhvi")
	req.Header.Set("Origin", "https://fanyi.caiyunapp.com")
	req.Header.Set("Sec-Fetch-Site", "cross-site")
	req.Header.Set("Sec-Fetch-Mode", "cors")
	req.Header.Set("Sec-Fetch-Dest", "empty")
	req.Header.Set("Referer", "https://fanyi.caiyunapp.com/")
	req.Header.Set("Accept-Language", "zh-CN,zh;q=0.9")
	req.Header.Set("Cookie", "_ym_uid=16456948721020430059; _ym_d=1645694872")
	resp, err := client.Do(req)
	if err != nil {
		log.Fatal(err)
	}
	defer resp.Body.Close()
	bodyText, err := ioutil.ReadAll(resp.Body)
	if err != nil {
		log.Fatal(err)
	}
	if resp.StatusCode != 200 {
		log.Fatal("bad StatusCode:", resp.StatusCode, "body", string(bodyText))
	}
	var dictResponse DictResponse
	err = json.Unmarshal(bodyText, &dictResponse)
	if err != nil {
		log.Fatal(err)
	}
	fmt.Println(word, "UK:", dictResponse.Dictionary.Prons.En, "US:", dictResponse.Dictionary.Prons.EnUs)
	for _, item := range dictResponse.Dictionary.Explanations {
		fmt.Println(item)
	}
}

func main() {
	if len(os.Args) != 2 {
		fmt.Fprintf(os.Stderr, `usage: simpleDict WORD
example: simpleDict hello
		`)
		os.Exit(1)
	}
	word := os.Args[1]
	query(word)
}

三、SOCKS5 代理

简述

SOCKS5 是一个代理协议,它在使用TCP/IP协议通讯的前端机器和服务器机器之间扮演一个中介角色,使得内部网中的前端机器变得能够访问Internet网中的服务器,或者使通讯更加安全。SOCKS5 服务器通过将前端发来的请求转发给真正的目标服务器, 模拟了一个前端的行为。在这里,前端和SOCKS5之间也是通过TCP/IP协议进行通讯,前端将原本要发送给真正服务器的请求发送给SOCKS5服务器,然后SOCKS5服务器将请求转发给真正的服务器。

⭐ 原理 image.png

思路分析

  1. TCP echo server:第一步,先在Golang里面写一个简单的TCP echo server。为了方便测试,server的工作逻辑很简单,你给他发送啥,他就回复啥。
package main

import (
    "bufio"
    "log"
    "net"
)

func main() {
    server, err := net.Listen("tcp", "127.0.0.1:1080")
    if err != nil {
            panic(err)
    }
    for {
            client, err := server.Accept()
            if err != nil {
                    log.Printf("Accept failed %v", err)
                    continue
            }
            go process(client)
    }
}

func process(conn net.Conn) {
    defer conn.Close()
    reader := bufio.NewReader(conn)
    for {
            b, err := reader.ReadByte()
            if err != nil {
                    break
            }
            _, err = conn.Write([]byte{b})
            if err != nil {
                    break
            }
    }
}
  1. auth:首先是认证阶段,浏览器会给代理服务器发送一个包,这个包有三个字段(version协议版本号,固定是5,methods认证的方法数目,每个methods的编码,0代表不需要认证,2代表用户名密码认证)。
  • 先用read bytes来把版本号读出来,如果版本号不是socket 5的话直接返回报错,接下来再读取method size,也是一个字节。然后需要去make一个相应长度的一个slice,用io.ReadFull把它去填充进去。现在把获取到的版本号和认证方式打印一下。此时,代理服务器还需要返回一个response,返回包括两个字段,一个是version,一个是method
const socks5Ver = 0x05
const cmdBind = 0x01
const atypIPV4 = 0x01
const atypeHOST = 0x03
const atypeIPV6 = 0x04

func process(conn net.Conn) {
    defer conn.Close()
    reader := bufio.NewReader(conn)
    err := auth(reader, conn)
    if err != nil {
            log.Printf("client %v auth failed:%v", conn.RemoteAddr(), err)
            return
    }
    log.Println("auth success")
}

func auth(reader *bufio.Reader, conn net.Conn) (err error) {
    // +----+----------+----------+
    // |VER | NMETHODS | METHODS  |
    // +----+----------+----------+
    // | 1  |    1     | 1 to 255 |
    // +----+----------+----------+
    // VER: 协议版本,socks5为0x05
    // NMETHODS: 支持认证的方法数量
    // METHODS: 对应NMETHODS,NMETHODS的值为多少,METHODS就有多少个字节。RFC预定义了一些值的含义,内容如下:
    // X’00’ NO AUTHENTICATION REQUIRED
    // X’02’ USERNAME/PASSWORD

    ver, err := reader.ReadByte()
    if err != nil {
            return fmt.Errorf("read ver failed:%w", err)
    }
    if ver != socks5Ver {
            return fmt.Errorf("not supported ver:%v", ver)
    }
    methodSize, err := reader.ReadByte()
    if err != nil {
            return fmt.Errorf("read methodSize failed:%w", err)
    }
    method := make([]byte, methodSize)
    _, err = io.ReadFull(reader, method)
    if err != nil {
            return fmt.Errorf("read method failed:%w", err)
    }
    log.Println("ver", ver, "method", method)
    // +----+--------+
    // |VER | METHOD |
    // +----+--------+
    // | 1  |   1    |
    // +----+--------+
    _, err = conn.Write([]byte{socks5Ver, 0x00})
    if err != nil {
            return fmt.Errorf("write failed:%w", err)
    }
    return nil
}
  1. 请求阶段:请求阶段,浏览器会发送一个包,包里面包含如下6个字段,version版本号,command请求的类型,RSV保留字段,atype就是目标地址类型,可能是 IPV4IPV6或者域名,addr地址,port端口号,然后需要逐个去读取这些字段。
  • 试图读取到携带URL或者IP地址+端口的包,然后把它打印出来。实现一个和auth函数类似的connect函数,同样在process里面去调用。
  • 接下来把这个地址和端口打印出来用于调试,收到浏览器的这个请求包之后,需要返回一个包。
func process(conn net.Conn) {
    defer conn.Close()
    reader := bufio.NewReader(conn)
    err := auth(reader, conn)
    if err != nil {
            log.Printf("client %v auth failed:%v", conn.RemoteAddr(), err)
            return
    }
    err = connect(reader, conn)
    if err != nil {
            log.Printf("client %v auth failed:%v", conn.RemoteAddr(), err)
            return
    }
}

func connect(reader *bufio.Reader, conn net.Conn) (err error) {
    // +----+-----+-------+------+----------+----------+
    // |VER | CMD |  RSV  | ATYP | DST.ADDR | DST.PORT |
    // +----+-----+-------+------+----------+----------+
    // | 1  |  1  | X'00' |  1   | Variable |    2     |
    // +----+-----+-------+------+----------+----------+
    // VER 版本号,socks5的值为0x05
    // CMD 0x01表示CONNECT请求
    // RSV 保留字段,值为0x00
    // ATYP 目标地址类型,DST.ADDR的数据对应这个字段的类型。
    //   0x01表示IPv4地址,DST.ADDR为4个字节
    //   0x03表示域名,DST.ADDR是一个可变长度的域名
    // DST.ADDR 一个可变长度的值
    // DST.PORT 目标端口,固定2个字节

    buf := make([]byte, 4)
    _, err = io.ReadFull(reader, buf)
    if err != nil {
            return fmt.Errorf("read header failed:%w", err)
    }
    ver, cmd, atyp := buf[0], buf[1], buf[3]
    if ver != socks5Ver {
            return fmt.Errorf("not supported ver:%v", ver)
    }
    if cmd != cmdBind {
            return fmt.Errorf("not supported cmd:%v", ver)
    }
    addr := ""
    switch atyp {
    case atypIPV4:
            _, err = io.ReadFull(reader, buf)
            if err != nil {
                    return fmt.Errorf("read atyp failed:%w", err)
            }
            addr = fmt.Sprintf("%d.%d.%d.%d", buf[0], buf[1], buf[2], buf[3])
    case atypeHOST:
            hostSize, err := reader.ReadByte()
            if err != nil {
                    return fmt.Errorf("read hostSize failed:%w", err)
            }
            host := make([]byte, hostSize)
            _, err = io.ReadFull(reader, host)
            if err != nil {
                    return fmt.Errorf("read host failed:%w", err)
            }
            addr = string(host)
    case atypeIPV6:
            return errors.New("IPv6: no supported yet")
    default:
            return errors.New("invalid atyp")
    }
    _, err = io.ReadFull(reader, buf[:2])
    if err != nil {
            return fmt.Errorf("read port failed:%w", err)
    }
    port := binary.BigEndian.Uint16(buf[:2])

    log.Println("dial", addr, port)

    // +----+-----+-------+------+----------+----------+
    // |VER | REP |  RSV  | ATYP | BND.ADDR | BND.PORT |
    // +----+-----+-------+------+----------+----------+
    // | 1  |  1  | X'00' |  1   | Variable |    2     |
    // +----+-----+-------+------+----------+----------+
    // VER socks版本,这里为0x05
    // REP Relay field,内容取值如下 X’00’ succeeded
    // RSV 保留字段
    // ATYPE 地址类型
    // BND.ADDR 服务绑定的地址
    // BND.PORT 服务绑定的端口DST.PORT
    _, err = conn.Write([]byte{0x05, 0x00, 0x00, 0x01, 0, 0, 0, 0, 0, 0})
    if err != nil {
            return fmt.Errorf("write failed: %w", err)
    }
    return nil
}
  1. relay 阶段:直接用net.dial建立一个TCP连接,建立完连接之后,要加一个defer来关闭连接。
  • 接下来需要建立浏览器和下游服务器的双向数据转发。标准库的io.copy可以实现一个单向数据转发,双向转发需要启动两个goroutinue
  • 这里可以使用到标准库里面的一个context机制,用contextwith cancel来创建一个context,在最后等待ctx.Done(),只要cancel被调用,ctx.Done就会立刻返回,然后在上面的两个goroutinue里面 调用一次cancel 即可。

Go 代码

package main

import (
    "bufio"
    "context"
    "encoding/binary"
    "errors"
    "fmt"
    "io"
    "log"
    "net"
)

const socks5Ver = 0x05
const cmdBind = 0x01
const atypIPV4 = 0x01
const atypeHOST = 0x03
const atypeIPV6 = 0x04

func main() {
    server, err := net.Listen("tcp", "127.0.0.1:1080")
    if err != nil {
            panic(err)
    }
    for {
            client, err := server.Accept()
            if err != nil {
                    log.Printf("Accept failed %v", err)
                    continue
            }
            go process(client)
    }
}

func process(conn net.Conn) {
    defer conn.Close()
    reader := bufio.NewReader(conn)
    err := auth(reader, conn)
    if err != nil {
            log.Printf("client %v auth failed:%v", conn.RemoteAddr(), err)
            return
    }
    err = connect(reader, conn)
    if err != nil {
            log.Printf("client %v auth failed:%v", conn.RemoteAddr(), err)
            return
    }
}

func auth(reader *bufio.Reader, conn net.Conn) (err error) {
    // +----+----------+----------+
    // |VER | NMETHODS | METHODS  |
    // +----+----------+----------+
    // | 1  |    1     | 1 to 255 |
    // +----+----------+----------+
    // VER: 协议版本,socks5为0x05
    // NMETHODS: 支持认证的方法数量
    // METHODS: 对应NMETHODS,NMETHODS的值为多少,METHODS就有多少个字节。RFC预定义了一些值的含义,内容如下:
    // X’00’ NO AUTHENTICATION REQUIRED
    // X’02’ USERNAME/PASSWORD

    ver, err := reader.ReadByte()
    if err != nil {
            return fmt.Errorf("read ver failed:%w", err)
    }
    if ver != socks5Ver {
            return fmt.Errorf("not supported ver:%v", ver)
    }
    methodSize, err := reader.ReadByte()
    if err != nil {
            return fmt.Errorf("read methodSize failed:%w", err)
    }
    method := make([]byte, methodSize)
    _, err = io.ReadFull(reader, method)
    if err != nil {
            return fmt.Errorf("read method failed:%w", err)
    }

    // +----+--------+
    // |VER | METHOD |
    // +----+--------+
    // | 1  |   1    |
    // +----+--------+
    _, err = conn.Write([]byte{socks5Ver, 0x00})
    if err != nil {
            return fmt.Errorf("write failed:%w", err)
    }
    return nil
}

func connect(reader *bufio.Reader, conn net.Conn) (err error) {
    // +----+-----+-------+------+----------+----------+
    // |VER | CMD |  RSV  | ATYP | DST.ADDR | DST.PORT |
    // +----+-----+-------+------+----------+----------+
    // | 1  |  1  | X'00' |  1   | Variable |    2     |
    // +----+-----+-------+------+----------+----------+
    // VER 版本号,socks5的值为0x05
    // CMD 0x01表示CONNECT请求
    // RSV 保留字段,值为0x00
    // ATYP 目标地址类型,DST.ADDR的数据对应这个字段的类型。
    //   0x01表示IPv4地址,DST.ADDR为4个字节
    //   0x03表示域名,DST.ADDR是一个可变长度的域名
    // DST.ADDR 一个可变长度的值
    // DST.PORT 目标端口,固定2个字节

    buf := make([]byte, 4)
    _, err = io.ReadFull(reader, buf)
    if err != nil {
            return fmt.Errorf("read header failed:%w", err)
    }
    ver, cmd, atyp := buf[0], buf[1], buf[3]
    if ver != socks5Ver {
            return fmt.Errorf("not supported ver:%v", ver)
    }
    if cmd != cmdBind {
            return fmt.Errorf("not supported cmd:%v", ver)
    }
    addr := ""
    switch atyp {
    case atypIPV4:
            _, err = io.ReadFull(reader, buf)
            if err != nil {
                    return fmt.Errorf("read atyp failed:%w", err)
            }
            addr = fmt.Sprintf("%d.%d.%d.%d", buf[0], buf[1], buf[2], buf[3])
    case atypeHOST:
            hostSize, err := reader.ReadByte()
            if err != nil {
                    return fmt.Errorf("read hostSize failed:%w", err)
            }
            host := make([]byte, hostSize)
            _, err = io.ReadFull(reader, host)
            if err != nil {
                    return fmt.Errorf("read host failed:%w", err)
            }
            addr = string(host)
    case atypeIPV6:
            return errors.New("IPv6: no supported yet")
    default:
            return errors.New("invalid atyp")
    }
    _, err = io.ReadFull(reader, buf[:2])
    if err != nil {
            return fmt.Errorf("read port failed:%w", err)
    }
    port := binary.BigEndian.Uint16(buf[:2])

    dest, err := net.Dial("tcp", fmt.Sprintf("%v:%v", addr, port))
    if err != nil {
            return fmt.Errorf("dial dst failed:%w", err)
    }
    defer dest.Close()
    log.Println("dial", addr, port)

    // +----+-----+-------+------+----------+----------+
    // |VER | REP |  RSV  | ATYP | BND.ADDR | BND.PORT |
    // +----+-----+-------+------+----------+----------+
    // | 1  |  1  | X'00' |  1   | Variable |    2     |
    // +----+-----+-------+------+----------+----------+
    // VER socks版本,这里为0x05
    // REP Relay field,内容取值如下 X’00’ succeeded
    // RSV 保留字段
    // ATYPE 地址类型
    // BND.ADDR 服务绑定的地址
    // BND.PORT 服务绑定的端口DST.PORT
    _, err = conn.Write([]byte{0x05, 0x00, 0x00, 0x01, 0, 0, 0, 0, 0, 0})
    if err != nil {
            return fmt.Errorf("write failed: %w", err)
    }
    ctx, cancel := context.WithCancel(context.Background())
    defer cancel()

    go func() {
            _, _ = io.Copy(dest, reader)
            cancel()
    }()
    go func() {
            _, _ = io.Copy(conn, dest)
            cancel()
    }()

    <-ctx.Done()
    return nil
}