190 Star 792 Fork 219

骁龙/chuantou

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
该仓库未声明开源许可证文件(LICENSE),使用请关注具体项目描述及其代码上游依赖。
克隆/下载
client.go 5.10 KB
一键复制 编辑 原始数据 按行查看 历史
骁龙 提交于 2017-03-29 10:23 . 版本v1.2,新增自定义接口功能
//client端,运行在家里有网站的电脑中
package main
import (
"flag"
"fmt"
"net"
"os"
"runtime"
"strconv"
"strings"
"time"
)
var host *string = flag.String("host", "127.0.0.1", "请输入服务器ip")
var remotePort *string = flag.String("remotePort", "20012", "服务器地址端口")
var localPort *string = flag.String("localPort", "80", "本地端口")
//与browser相关的conn
type browser struct {
conn net.Conn
er chan bool
writ chan bool
recv chan []byte
send chan []byte
}
//读取browser过来的数据
func (self browser) read() {
for {
var recv []byte = make([]byte, 10240)
n, err := self.conn.Read(recv)
if err != nil {
self.writ <- true
self.er <- true
//fmt.Println("读取browser失败", err)
break
}
self.recv <- recv[:n]
}
}
//把数据发送给browser
func (self browser) write() {
for {
var send []byte = make([]byte, 10240)
select {
case send = <-self.send:
self.conn.Write(send)
case <-self.writ:
//fmt.Println("写入browser进程关闭")
break
}
}
}
//与server相关的conn
type server struct {
conn net.Conn
er chan bool
writ chan bool
recv chan []byte
send chan []byte
}
//读取server过来的数据
func (self *server) read() {
//isheart与timeout共同判断是不是自己设定的SetReadDeadline
var isheart bool = false
//20秒发一次心跳包
self.conn.SetReadDeadline(time.Now().Add(time.Second * 20))
for {
var recv []byte = make([]byte, 10240)
n, err := self.conn.Read(recv)
if err != nil {
if strings.Contains(err.Error(), "timeout") && !isheart {
//fmt.Println("发送心跳包")
self.conn.Write([]byte("hh"))
//4秒时间收心跳包
self.conn.SetReadDeadline(time.Now().Add(time.Second * 4))
isheart = true
continue
}
//浏览器有可能连接上不发消息就断开,此时就发一个0,为了与服务器一直有一条tcp通路
self.recv <- []byte("0")
self.er <- true
self.writ <- true
//fmt.Println("没收到心跳包或者server关闭,关闭此条tcp", err)
break
}
//收到心跳包
if recv[0] == 'h' && recv[1] == 'h' {
//fmt.Println("收到心跳包")
self.conn.SetReadDeadline(time.Now().Add(time.Second * 20))
isheart = false
continue
}
self.recv <- recv[:n]
}
}
//把数据发送给server
func (self server) write() {
for {
var send []byte = make([]byte, 10240)
select {
case send = <-self.send:
self.conn.Write(send)
case <-self.writ:
//fmt.Println("写入server进程关闭")
break
}
}
}
func main() {
runtime.GOMAXPROCS(runtime.NumCPU())
flag.Parse()
if flag.NFlag() != 3 {
flag.PrintDefaults()
os.Exit(1)
}
local, _ := strconv.Atoi(*localPort)
remote, _ := strconv.Atoi(*remotePort)
if !(local >= 0 && local < 65536) {
fmt.Println("端口设置错误")
os.Exit(1)
}
if !(remote >= 0 && remote < 65536) {
fmt.Println("端口设置错误")
os.Exit(1)
}
target := net.JoinHostPort(*host, *remotePort)
for {
//链接端口
serverconn := dail(target)
recv := make(chan []byte)
send := make(chan []byte)
//1个位置是为了防止两个读取线程一个退出后另一个永远卡住
er := make(chan bool, 1)
writ := make(chan bool)
next := make(chan bool)
server := &server{serverconn, er, writ, recv, send}
go server.read()
go server.write()
go handle(server, next)
<-next
}
}
//显示错误
func log(err error) {
if err != nil {
fmt.Printf("出现错误: %v\n", err)
}
}
//显示错误并退出
func logExit(err error) {
if err != nil {
fmt.Printf("出现错误,退出线程: %v\n", err)
runtime.Goexit()
}
}
//显示错误并关闭链接,退出线程
func logClose(err error, conn net.Conn) {
if err != nil {
//fmt.Println("对方已关闭", err)
runtime.Goexit()
}
}
//链接端口
func dail(hostport string) net.Conn {
conn, err := net.Dial("tcp", hostport)
logExit(err)
return conn
}
//两个socket衔接相关处理
func handle(server *server, next chan bool) {
var serverrecv = make([]byte, 10240)
//阻塞这里等待server传来数据再链接browser
fmt.Println("等待server发来消息")
serverrecv = <-server.recv
//连接上,下一个tcp连上服务器
next <- true
//fmt.Println("开始新的tcp链接,发来的消息是:", string(serverrecv))
var browse *browser
//server发来数据,链接本地80端口
serverconn := dail("127.0.0.1:" + *localPort)
recv := make(chan []byte)
send := make(chan []byte)
er := make(chan bool, 1)
writ := make(chan bool)
browse = &browser{serverconn, er, writ, recv, send}
go browse.read()
go browse.write()
browse.send <- serverrecv
for {
var serverrecv = make([]byte, 10240)
var browserrecv = make([]byte, 10240)
select {
case serverrecv = <-server.recv:
if serverrecv[0] != '0' {
browse.send <- serverrecv
}
case browserrecv = <-browse.recv:
server.send <- browserrecv
case <-server.er:
//fmt.Println("server关闭了,关闭server与browse")
server.conn.Close()
browse.conn.Close()
runtime.Goexit()
case <-browse.er:
//fmt.Println("browse关闭了,关闭server与browse")
server.conn.Close()
browse.conn.Close()
runtime.Goexit()
}
}
}
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
Go
1
https://gitee.com/wapai/chuantou.git
git@gitee.com:wapai/chuantou.git
wapai
chuantou
chuantou
master

搜索帮助