Пример #1
0
func handleCreateTunnel(session *yamux.Session, port int) (url string, err error) {
	// TODO: Attempt to bind to local port
	addr := fmt.Sprintf("127.0.0.1:%v", port)
	log.WithFields(logrus.Fields{
		"server": addr,
	}).Info("attempting to connect to server")
	local, err := net.Dial("tcp", addr)
	if err != nil {
		return "", errors.New("Could not establish connection to host port.")
	}
	log.WithFields(logrus.Fields{
		"server": addr,
	}).Info("connected to server")

	// Establish stream
	stream, err := session.Open()
	if err != nil {
		return "", errors.New("Could not create stream.")
	}

	// TODO: send control message
	log.WithFields(logrus.Fields{
		"server": addr,
	}).Info("sending TunnelRequest")
	req := &msg.TunnelRequest{
		Port: port,
	}
	if err := msg.WriteMsg(stream, req); err != nil {
		return "", err
	}

	// TODO: wait for ack w/ url
	log.WithFields(logrus.Fields{
		"server": addr,
	}).Info("waiting for TunnelReply")

	var res msg.TunnelReply
	if err := msg.ReadMsgInto(stream, &res); err != nil {
		return "", err
	}
	log.WithFields(logrus.Fields{
		"server": addr,
		"port":   port,
		"uri":    res.URI,
	}).Info("received TunnelReply")
	url = res.URI

	// TODO: Update global state for later control messages

	// TODO: launch goroutine to proxy data
	go proxy.Proxy(stream, local)

	return url, nil
}
Пример #2
0
func tunnel(sconn net.Conn) {
	// buff := make([]byte, 0xff)

	// read control message to figure out what port is being forwarded
	var req msg.TunnelRequest
	if err := msg.ReadMsgInto(sconn, &req); err != nil {
		log.Error(err.Error())
	}

	log.WithFields(logrus.Fields{
		"client":     sconn.RemoteAddr(),
		"proxy_port": req.Port,
	}).Info("received TunnelRequest")

	// bind to available local port to proxy
	local, _ := net.Listen("tcp", ":0")
	defer local.Close()
	log.Info(local.Addr())

	// send url back in ack message
	hostname, _ := os.Hostname()
	port := local.Addr().(*net.TCPAddr).Port
	res := &msg.TunnelReply{
		URI: fmt.Sprintf("tcp://%s:%d", hostname, port),
	}
	if err := msg.WriteMsg(sconn, res); err != nil {
		log.Error(err.Error())
		return
	}

	// TODO: start proxying
	for {
		conn, err := local.Accept()
		if err != nil {
			// handle error
			log.Error("err")
		}
		log.Info("new conn")
		go proxy.Proxy(conn, sconn)
		// _, err := sconn.Read(buff)
		// if err != nil {
		// 	if err == io.EOF {
		// 		break
		// 	}
		// 	log.Printf("Stream read error: %s", err)
		// 	break
		// }
		// // log.Printf("stream sent %d bytes: %s", n, buff[:n])
		// sconn.Write([]byte("GET /\n\n"))
	}
}