handler_http.go

package ztp

import (
	"bufio"
	"io"
	"net"
	"net/http"
)

type HTTPProxyHandler struct{}

func NewHTTPProxyHandler() *HTTPProxyHandler {
	return &HTTPProxyHandler{}
}

var httpDial = net.Dial

func (_ *HTTPProxyHandler) Protocol() Protocol { return ProtocolHTTP }

func (_ *HTTPProxyHandler) Handle(client net.Conn, svc *Service, id *Identity) error {
	buf := bufio.NewReader(client)
	req, err := http.ReadRequest(buf)
	if err != nil {
		return err
	}

	backend, err := httpDial("tcp", svc.Address)
	if err != nil {
		return err
	}
	defer backend.Close()

	if err = req.Write(backend); err == nil {
		errCh := make(chan error, 1)
		go func() {
			_, err = io.Copy(backend, buf)
			errCh <- err
		}()

		_, copyErr := io.Copy(client, backend)
		_ = <-errCh

		if copyErr != nil {
			return copyErr
		}
	}

	return nil
}