transport/dialer/dialer_windows.go

// Licensed to Elasticsearch B.V. under one or more contributor
// license agreements. See the NOTICE file distributed with
// this work for additional information regarding copyright
// ownership. Elasticsearch B.V. licenses this file to you under
// the Apache License, Version 2.0 (the "License"); you may
// not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing,
// software distributed under the License is distributed on an
// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
// KIND, either express or implied.  See the License for the
// specific language governing permissions and limitations
// under the License.

//go:build windows

package dialer

import (
	"context"
	"errors"
	"net"
	"strings"
	"time"

	winio "github.com/Microsoft/go-winio"

	"github.com/elastic/elastic-agent-libs/api/npipe"
	"github.com/elastic/elastic-agent-libs/transport"
)

// UnixDialerBuilder creates a builder to dial over a unix domain socket.
type UnixDialerBuilder struct {
	Path string
}

// Make creates a dialer.
func (t *UnixDialerBuilder) Make(_ time.Duration) (transport.Dialer, error) {
	return nil, errors.New(
		"cannot use the URI, unix sockets are not supported on Windows, use npipe instead",
	)
}

func (t *UnixDialerBuilder) String() string {
	return "Unix: " + t.Path
}

// NpipeDialerBuilder creates a builder to dial over a named pipe.
type NpipeDialerBuilder struct {
	Path string
}

func (t *NpipeDialerBuilder) String() string {
	return "Npipe: " + t.Path
}

// Make creates a dialer.
func (t *NpipeDialerBuilder) Make(timeout time.Duration) (transport.Dialer, error) {
	to := timeout
	return transport.DialerFunc(
		func(ctx context.Context, _, _ string) (net.Conn, error) {
			ctx, cancel := context.WithTimeout(ctx, to)
			defer cancel()
			return winio.DialPipeContext(
				ctx,
				strings.TrimSuffix(npipe.TransformString(t.Path), "/"),
			)
		},
	), nil
}