goadb/wire/dialer.go

package wire

import (
	"errors"
	"fmt"
	"net"
	"runtime"
)

/*
Dialer knows how to create connections to an adb server.
*/
type Dialer interface {
	Dial() (*Conn, error)
}

type netDialer struct {
	Host string
	Port int
}

func NewDialer(host string, port int) Dialer {
	return &netDialer{host, port}
}

func (d *netDialer) String() string {
	return fmt.Sprintf("netDialer(%s:%d)", d.Host, d.Port)
}

// Dial connects to the adb server on the host and port set on the netDialer.
// The zero-value will connect to the default, localhost:5037.
func (d *netDialer) Dial() (*Conn, error) {
	host := d.Host
	if host == "" {
		return nil, errors.New("Must specify adb hostname (cannot be empty).")
	}

	port := d.Port
	if port == 0 {
		return nil, errors.New("Must specify port (cannot be 0).")
	}

	netConn, err := net.Dial("tcp", fmt.Sprintf("%s:%d", host, port))
	if err != nil {
		return nil, err
	}

	conn := &Conn{
		Scanner: NewScanner(netConn),
		Sender:  NewSender(netConn),
	}

	// Prevent leaking the network connection, not sure if TCPConn does this itself.
	runtime.SetFinalizer(netConn, func(conn *net.TCPConn) {
		conn.Close()
	})

	return conn, nil
}

func RoundTripSingleResponse(d Dialer, req string) ([]byte, error) {
	conn, err := d.Dial()
	if err != nil {
		return nil, err
	}
	defer conn.Close()

	return conn.RoundTripSingleResponse([]byte(req))
}