compy/proxy/mitmlistener.go

package proxy

import (
	"crypto/tls"
	"net"
)

type mitmListener struct {
	c  chan net.Conn
	cf *certFaker
}

func newMitmListener(cf *certFaker) *mitmListener {
	return &mitmListener{
		c:  make(chan net.Conn),
		cf: cf,
	}
}

func (l *mitmListener) Accept() (net.Conn, error) {
	return <-l.c, nil
}

func (l *mitmListener) Close() error {
	return nil
}

func (l *mitmListener) Addr() net.Addr {
	return nil
}

func (l *mitmListener) Serve(conn net.Conn, host string) (net.Conn, error) {
	sconn, err := tls.Dial("tcp", host, nil)
	if err != nil {
		return nil, err
	}
	fakeCert, err := l.cf.FakeCert(sconn.ConnectionState().PeerCertificates[0])
	if err != nil {
		sconn.Close()
		return nil, err
	}
	tlsconf := &tls.Config{Certificates: []tls.Certificate{*fakeCert}}
	l.c <- tls.Server(conn, tlsconf)
	return sconn, nil
}