实现内网穿透(二)

项目地址：https://github.com/fzdwx/burst

今天主要使用Go实现 echo client，并与server成功连接。

websocket框架选型

使用在Go语言中比较流行的实现：https://github.com/gorilla/websocket

项目搭建

mkdri go && cd go
go mod init demo
go get github.com/gorilla/websocket

与客户端建立连接

func main() {

    // 定义客户端的地址
    u := url.URL{Scheme: "ws", Host: "locaalhost:999", Path: "/connect"}

    // 与客户端建立连接
	c, _, err := websocket.DefaultDialer.Dial(u.String(), nil)
	if err != nil {
		log.Fatal("dial:", err)
	}
	defer c.Close()

    // 启动一个协程，读取从服务端发送过来的数据
	go func() {
		for {
			_, message, _ := c.ReadMessage()
			fmt.Println(string(message))
		}
	}()


    // 阻塞主线程
    down := make(chan byte)
	for {
		<-down
	}
}

启动客户端

成功接收到来自服务端的消息。

实现echo handle

只需要我们从客户端发送消息就ok了，然后服务端就会把我们发送的数据返回给我们。

go func() {
	for {
		var input string
		fmt.Scanln(&input)
		c.WriteMessage(websocket.TextMessage, []byte(input))
	}
}()

1. 启动一个协程，读取从控制台输入的数据，然后发送到服务端。

效果

最终代码

package main

import (
	"flag"
	"fmt"
	"github.com/gorilla/websocket"
	"log"
	"net/url"
)

var addr = flag.String("addr", "localhost:9999", "proxy server addr")

func main() {

	u := url.URL{Scheme: "ws", Host: *addr, Path: "/connect"}

	c, _, err := websocket.DefaultDialer.Dial(u.String(), nil)
	if err != nil {
		log.Fatal("dial:", err)
	}
	defer c.Close()

	down := make(chan byte)
	go func() {
		for {
			_, message, _ := c.ReadMessage()
			fmt.Println("服务端发送:" + string(message))
		}
	}()

	go func() {
		for {
			var input string
			fmt.Scanln(&input)
			c.WriteMessage(websocket.TextMessage, []byte(input))
		}
	}()

	for {
		<-down
	}
}

明日目标

明天就正式进入具体的代码实现了。

1. 定义交互协议（暂定为json,主要为了快速开发）
2. 3个阶段
   1. 服务端与客户端连接初始化（确定客户端与服务端的端口映射关系）
   2. 客户连接，通知客户端，监听本地端口
   3. 数据转发。