Skip to content

Instantly share code, notes, and snippets.

@wallrat
Created July 6, 2012 11:25
Show Gist options
  • Save wallrat/3059639 to your computer and use it in GitHub Desktop.
Save wallrat/3059639 to your computer and use it in GitHub Desktop.
Simple GO TCP proxy
package main
import (
"net"
"fmt"
"io"
"os"
)
func main() {
if len(os.Args) != 3 {
fatal("usage: netfwd local remote")
}
localAddr := os.Args[1]
remoteAddr := os.Args[2]
local, err := net.Listen("tcp", localAddr)
if local == nil {
fatal("cannot listen: %v", err)
}
for {
conn, err := local.Accept()
if conn == nil {
fatal("accept failed: %v", err)
}
go forward(conn, remoteAddr)
}
}
func forward(local net.Conn, remoteAddr string) {
remote, err := net.Dial("tcp", "", remoteAddr)
if remote == nil {
fmt.Fprintf(os.Stderr, "remote dial failed: %v\n", err)
return
}
go io.Copy(local, remote)
go io.Copy(remote, local)
}
func fatal(s string, a ... interface{}) {
fmt.Fprintf(os.Stderr, "netfwd: %s\n", fmt.Sprintf(s, a))
os.Exit(2)
}
@fftlxyz
Copy link

fftlxyz commented Jan 24, 2015

awesome

@YihaoPeng
Copy link

YihaoPeng commented Oct 31, 2017

a small change, local and remote will Close() at the same time if one part disconnected.

package main
import (
        "net"
        "fmt"
        "io"
        "os"
)
func main() {
        if len(os.Args) != 3 {
                fatal("usage: netfwd local remote")
        }
        localAddr := os.Args[1]
        remoteAddr := os.Args[2]
        local, err := net.Listen("tcp", localAddr)
        if local == nil {
                fatal("cannot listen: %v", err)
        }
        for {
                conn, err := local.Accept()
                if conn == nil {
                        fatal("accept failed: %v", err)
                }
                go forward(conn, remoteAddr)
        }
}

func forward(local net.Conn, remoteAddr string) {
        remote, err := net.Dial("tcp", remoteAddr)
        if remote == nil {
                fmt.Fprintf(os.Stderr, "remote dial failed: %v\n", err)
                return
        }

        go func() {
            io.Copy(local, remote)
            local.Close()
            remote.Close()
        }()

        go func() {
            io.Copy(remote, local)
            remote.Close()
            local.Close()
        }()
}

func fatal(s string, a ... interface{}) {
        fmt.Fprintf(os.Stderr, "netfwd: %s\n", fmt.Sprintf(s, a))
        os.Exit(2)
}

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