roughing in tunnel frontend (#170)

endpoints/proxyFrontend/config.go

@@ -1,17 +0,0 @@
-package proxyFrontend
-
-import "github.com/openziti/zrok/endpoints"
-
-type Config struct {
-	IdentityName string
-	ShrToken     string
-	Address      string
-	RequestsChan chan *endpoints.Request
-}
-
-func DefaultConfig(identityName string) *Config {
-	return &Config{
-		IdentityName: identityName,
-		Address:      "0.0.0.0:8080",
-	}
-}

endpoints/proxyFrontend/http.go

@@ -19,6 +19,20 @@ import (
 	"time"
 )
 
+type Config struct {
+	IdentityName string
+	ShrToken     string
+	Address      string
+	RequestsChan chan *endpoints.Request
+}
+
+func DefaultConfig(identityName string) *Config {
+	return &Config{
+		IdentityName: identityName,
+		Address:      "0.0.0.0:8080",
+	}
+}
+
 type httpFrontend struct {
 	cfg      *Config
 	zCtx     ziti.Context

endpoints/tunnelFrontend/frontend.go

@@ -0,0 +1,78 @@
+package tunnelFrontend
+
+import (
+	"github.com/openziti/sdk-golang/ziti"
+	"github.com/openziti/sdk-golang/ziti/config"
+	"github.com/openziti/transport/v2"
+	"github.com/openziti/zrok/model"
+	"github.com/openziti/zrok/zrokdir"
+	"github.com/pkg/errors"
+	"github.com/sirupsen/logrus"
+	"io"
+)
+
+type Config struct {
+	BindAddress  string
+	IdentityName string
+	ShrToken     string
+}
+
+type Frontend struct {
+	cfg      *Config
+	zCtx     ziti.Context
+	listener transport.Address
+	closer   io.Closer
+}
+
+func NewFrontend(cfg *Config) (*Frontend, error) {
+	addr, err := transport.ParseAddress(cfg.BindAddress)
+	if err != nil {
+		return nil, errors.Wrapf(err, "error parsing '%v'", cfg.BindAddress)
+	}
+	zCfgPath, err := zrokdir.ZitiIdentityFile(cfg.IdentityName)
+	if err != nil {
+		return nil, errors.Wrapf(err, "error getting ziti identity '%v' from zrokdir", cfg.IdentityName)
+	}
+	zCfg, err := config.NewFromFile(zCfgPath)
+	if err != nil {
+		return nil, errors.Wrap(err, "error loading config")
+	}
+	zCfg.ConfigTypes = []string{model.ZrokProxyConfig}
+	zCtx := ziti.NewContextWithConfig(zCfg)
+	return &Frontend{
+		cfg:      cfg,
+		zCtx:     zCtx,
+		listener: addr,
+	}, nil
+}
+
+func (f *Frontend) Run() error {
+	closer, err := f.listener.Listen(f.cfg.ShrToken, nil, f.accept, nil)
+	if err != nil {
+		return err
+	}
+	f.closer = closer
+	return nil
+}
+
+func (f *Frontend) Stop() {
+	if f.closer != nil {
+		_ = f.closer.Close()
+	}
+}
+
+func (f *Frontend) accept(conn transport.Conn) {
+	buf := make([]byte, 10240)
+	for {
+		n, err := conn.Read(buf)
+		if err != nil {
+			logrus.Errorf("error reading: %v", err)
+			return
+		}
+		n, err = conn.Write(buf[:n])
+		if err != nil {
+			logrus.Errorf("error writing: %v", err)
+			return
+		}
+	}
+}