Instantly share code, notes, and snippets.

Embed
What would you like to do?
Simple TCP proxy in Golang
package main
import (
"bytes"
"encoding/hex"
"flag"
"fmt"
"io"
"log"
"net"
)
var localAddr *string = flag.String("l", "localhost:9999", "local address")
var remoteAddr *string = flag.String("r", "localhost:80", "remote address")
func proxyConn(conn *net.TCPConn) {
rAddr, err := net.ResolveTCPAddr("tcp", *remoteAddr)
if err != nil {
panic(err)
}
rConn, err := net.DialTCP("tcp", nil, rAddr)
if err != nil {
panic(err)
}
defer rConn.Close()
buf := &bytes.Buffer{}
for {
data := make([]byte, 256)
n, err := conn.Read(data)
if err != nil {
panic(err)
}
buf.Write(data[:n])
if data[0] == 13 && data[1] == 10 {
break
}
}
if _, err := rConn.Write(buf.Bytes()); err != nil {
panic(err)
}
log.Printf("sent:\n%v", hex.Dump(buf.Bytes()))
data := make([]byte, 1024)
n, err := rConn.Read(data)
if err != nil {
if err != io.EOF {
panic(err)
} else {
log.Printf("received err: %v", err)
}
}
log.Printf("received:\n%v", hex.Dump(data[:n]))
}
func handleConn(in <-chan *net.TCPConn, out chan<- *net.TCPConn) {
for conn := range in {
proxyConn(conn)
out <- conn
}
}
func closeConn(in <-chan *net.TCPConn) {
for conn := range in {
conn.Close()
}
}
func main() {
flag.Parse()
fmt.Printf("Listening: %v\nProxying: %v\n\n", *localAddr, *remoteAddr)
addr, err := net.ResolveTCPAddr("tcp", *localAddr)
if err != nil {
panic(err)
}
listener, err := net.ListenTCP("tcp", addr)
if err != nil {
panic(err)
}
pending, complete := make(chan *net.TCPConn), make(chan *net.TCPConn)
for i := 0; i < 5; i++ {
go handleConn(pending, complete)
}
go closeConn(complete)
for {
conn, err := listener.AcceptTCP()
if err != nil {
panic(err)
}
pending <- conn
}
}
@elimisteve

This comment has been minimized.

elimisteve commented Apr 27, 2013

You may want to replace

if data[0] == 13 && data[1] == 10

with

if data[0] == '\r' && data[1] == '\n'

to make it less magical :-)

@jpillora

This comment has been minimized.

jpillora commented Sep 30, 2014

@helinwang

This comment has been minimized.

helinwang commented Jun 23, 2015

Thanks for the code.
come of my comments

  • serializing read then write does not seem to be necessary. You can do something like
go io.Copy(src, dst)
go io.Copy(dst, src)
  • write only contents before "\r\n" seems strange to me. io.EOF seems a better ending signal.
@tscolari

This comment has been minimized.

tscolari commented Nov 16, 2015

I'm trying to create a simple proxy that redirects based on the request hostname.
Using this approach, is there a way to figure out which hostname, instead of ip, the local connection arrived at?

@abdel-aziz

This comment has been minimized.

abdel-aziz commented Jun 9, 2016

answer to tscolari's question: you can do something like that

req, err := http.ReadRequest(bufio.NewReader(conn))
            if err != nil {
                log.Error("Failed to get request from conn: ", err)
            } 
                host, _, err := net.SplitHostPort(req.Host)
                if err != nil {
                    log.Error("Failed to SplitHostPort: ", err)
                }
                ...
@apratheek

This comment has been minimized.

apratheek commented Jun 23, 2016

Hi, I'm trying to run this gist, and it seems to me that Line 31 is blocking. It keeps waiting for data to be available on the connection, instead of immediately returning an EOF. I'm pretty sure conn.Read(data) has never blocked previously, so I'm trying to figure out if this is a bug on my computer. The only part of the code that I've modified is on Line 14, where I've replaced localhost:80 with google.com:80. If it helps, I'm on OS X 10.11.4, and on Golang 1.6.2.

I've also tried running it on an ubuntu image, inside a docker container on my Mac, but the problem seems to persist. Can anyone help me out here?

@samsulbasri

This comment has been minimized.

samsulbasri commented Sep 11, 2017

Bagai mana cara gratis internet open vpn xl

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment