feat: enable shared envd host routing

Author: matthewlouisbrockman

packages/client-proxy/internal/proxy/proxy.go

@@ -16,7 +16,6 @@ import (
 	"google.golang.org/grpc/codes"
 	"google.golang.org/grpc/status"
 
-	"github.com/e2b-dev/infra/packages/shared/pkg/env"
 	"github.com/e2b-dev/infra/packages/shared/pkg/featureflags"
 	proxygrpc "github.com/e2b-dev/infra/packages/shared/pkg/grpc/proxy"
 	"github.com/e2b-dev/infra/packages/shared/pkg/logger"
@@ -63,6 +62,17 @@ func normalizeNodeIP(nodeIP string) (string, error) {
 	return nodeIP, nil
 }
 
+func legacySandboxHostForEnvdRequest(host string, sandboxID string, port uint64) *string {
+	hostname := strings.Split(host, ":")[0]
+	domain, ok := strings.CutPrefix(hostname, "envd.")
+	if !ok || domain == "" {
+		return nil
+	}
+
+	legacyHost := fmt.Sprintf("%d-%s.%s", port, sandboxID, domain)
+	return &legacyHost
+}
+
 func catalogResolution(ctx context.Context, sandboxId string, sandboxPort uint64, trafficAccessToken string, envdAccessToken string, c catalog.SandboxesCatalog, pausedChecker PausedSandboxResumer, featureFlags *featureflags.Client) (string, error) {
 	s, err := c.GetSandbox(ctx, sandboxId)
 	if err != nil {
@@ -133,7 +143,7 @@ func handlePausedSandbox(
 }
 
 func NewClientProxy(meterProvider metric.MeterProvider, serviceName string, port uint16, catalog catalog.SandboxesCatalog, pausedSandboxResumer PausedSandboxResumer, featureFlagsClient *featureflags.Client) (*reverseproxy.Proxy, error) {
-	getTargetFromRequest := reverseproxy.GetTargetFromRequest(env.IsLocal())
+	getTargetFromRequest := reverseproxy.GetTargetFromRequest(true)
 	proxy := reverseproxy.New(
 		port,
 		// Retries that are needed to handle port forwarding delays in sandbox envd are handled by the orchestrator proxy
@@ -198,6 +208,11 @@ func NewClientProxy(meterProvider metric.MeterProvider, serviceName string, port
 				SandboxPort:   port,
 				ConnectionKey: pool.ClientProxyConnectionKey,
 				Url:           url,
+				MaskRequestHost: legacySandboxHostForEnvdRequest(
+					r.Host,
+					sandboxId,
+					port,
+				),
 			}, nil
 		},
 		nil,

packages/shared/pkg/proxy/host.go

@@ -11,7 +11,7 @@ import (
 
 func GetTargetFromRequest(processHeaders bool) func(r *http.Request) (sandboxId string, port uint64, err error) {
 	return func(r *http.Request) (sandboxId string, port uint64, err error) {
-		if processHeaders {
+		if processHeaders && shouldParseHeaders(r.Host) {
 			var ok bool
 			sandboxId, port, ok, err = parseHeaders(r.Header)
 			if err != nil {
@@ -38,6 +38,11 @@ func GetTargetFromRequest(processHeaders bool) func(r *http.Request) (sandboxId
 	}
 }
 
+func shouldParseHeaders(host string) bool {
+	host = strings.Split(host, ":")[0]
+	return host == "localhost" || strings.HasPrefix(host, "envd.")
+}
+
 func parseHost(host string) (sandboxID string, port uint64, err error) {
 	dot := strings.Index(host, ".")
 

packages/shared/pkg/proxy/host_test.go

@@ -142,6 +142,26 @@ func TestGetTargetFromRequest(t *testing.T) { //nolint:tparallel // cannot call
 			},
 			wantErrIs: MissingHeaderError{Header: headerSandboxPort},
 		},
+		{
+			name: "headers: envd shared host",
+			host: "envd.e2b.app",
+			headers: http.Header{
+				headerSandboxID:   []string{"isv6ril5xadwn1k9t2jye"},
+				headerSandboxPort: []string{"49983"},
+			},
+			wantID:   "isv6ril5xadwn1k9t2jye",
+			wantPort: 49983,
+		},
+		{
+			name: "headers: ignored on regular sandbox host",
+			host: "49983-isv6ril5xadwn1k9t2jye.e2b.app",
+			headers: http.Header{
+				headerSandboxID:   []string{"iother5b5aiixd410phsjv"},
+				headerSandboxPort: []string{"3000"},
+			},
+			wantID:   "isv6ril5xadwn1k9t2jye",
+			wantPort: 49983,
+		},
 	}
 
 	for _, tt := range tests {