proxy.go

package tcpdam

import (
	"io"
	"sync"

	"github.com/op/go-logging"
)

type Proxy struct {
	Lconn, Rconn io.ReadWriteCloser
	Dam          *Dam
	Logger       *logging.Logger
}

func (p *Proxy) Flush() error {
	p.Logger.Debug("Flush connection")
	defer p.Lconn.Close()
	defer p.Dam.Flushed(p)
	Rconn, err := p.Dam.Dial()
	if err != nil {
		return err
	}
	p.Rconn = Rconn

	defer p.Rconn.Close()

	var wg sync.WaitGroup

	wg.Add(1)
	go func() {
		io.Copy(p.Lconn, p.Rconn)
		p.Lconn.Close()
		wg.Done()
	}()

	wg.Add(1)
	go func() {
		io.Copy(p.Rconn, p.Lconn)
		p.Rconn.Close()
		wg.Done()
	}()

	wg.Wait()
	p.Logger.Debug("Flushing connection done")
	return nil
}

func (p *Proxy) Start() {
	// Should slowly read and buffer Lconn to not timeout
}