transport_telnet.go

package main

import (
	"fmt"
	"strings"

	// "github.com/nemith/go-netconf/v2/transport"
	"github.com/nemith/netconf/transport"
	"github.com/ziutek/telnet"
)

const (
	// telnetDefaultPort sets the default port for use by Telnet
	telnetDefaultPort = 23
)

// VendorIOProc is the interface used when establishing a telnet NETCONF session
type VendorIOProc interface {
	Login(*TransportTelnet, string, string) error
	StartNetconf(*TransportTelnet) error
}

type framer = transport.Framer
type TransportTelnet = Transport

// TransportTelnet is used to define what makes up a Telnet Transport layer for
// NETCONF
// type TransportTelnet struct {
type Transport struct {
	telnetConn *telnet.Conn
	*framer
}

// Dial is used to create a TCP Telnet connection to the remote host returning
// only an error if it is unable to dial the remote host.
func (t *TransportTelnet) Dial(target string, username string, password string, vendor VendorIOProc) error {
	if !strings.Contains(target, ":") {
		target = fmt.Sprintf("%s:%d", target, telnetDefaultPort)
	}

	tn, err := telnet.Dial("tcp", target)
	if err != nil {
		return err
	}
	tn.SetUnixWriteMode(true)

	t.telnetConn = tn
	// t.ReadWriteCloser = tn
	// t.FramedTransport = transport.NewFramedTransport(tn, tn)
	t.framer = transport.NewFramer(tn, tn)

	// vendor.Login(t, username, password)
	// vendor.StartNetconf(t)

	return nil
}

func (t *TransportTelnet) Close() error {
	return t.telnetConn.Close()
}

// DialTelnet dials and returns the usable telnet session.
func DialTelnet(target string, username string, password string, vendor VendorIOProc) (transport.Transport, error) {
	var t *TransportTelnet = &TransportTelnet{}
	if err := t.Dial(target, username, password, vendor); err != nil {
		return nil, err
	}
	return t, nil
}