oragono/vendor/github.com/ergochat/go-ident/client.go

// Package ident implements an RFC 1413 client
package ident

import (
	"bufio"
	"fmt"
	"net"
	"strings"
	"time"
)

// Response is a successful answer to our query to the identd server.
type Response struct {
	OS         string
	Charset    string
	Identifier string
}

// ResponseError indicates that the identd server returned an error rather than an
// identifying string.
type ResponseError struct {
	Type string
}

func (e ResponseError) Error() string {
	return fmt.Sprintf("Ident error: %s", e.Type)
}

// ProtocolError indicates that an error occurred with the protocol itself, that the response
// could not be successfully parsed or was malformed.
type ProtocolError struct {
	Line string
}

func (e ProtocolError) Error() string {
	return fmt.Sprintf("Unexpected response from server: %s", e.Line)
}

// Query makes an Ident query, if timeout is >0 the query is timed out after that many seconds.
func Query(ip string, portOnServer, portOnClient int, timeout time.Duration) (response Response, err error) {
	// if a timeout is set, respect it from the beginning of the query, including the dial time
	var deadline time.Time
	if timeout > 0 {
		deadline = time.Now().Add(timeout)
	}

	var conn net.Conn
	if timeout > 0 {
		conn, err = net.DialTimeout("tcp", net.JoinHostPort(ip, "113"), timeout)
	} else {
		conn, err = net.Dial("tcp", net.JoinHostPort(ip, "113"))
	}
	if err != nil {
		return
	}
	defer conn.Close()

	// if timeout is 0, `deadline` is the empty time.Time{} which means no deadline:
	conn.SetDeadline(deadline)

	_, err = conn.Write([]byte(fmt.Sprintf("%d, %d\r\n", portOnClient, portOnServer)))
	if err != nil {
		return
	}

	r := bufio.NewReaderSize(conn, 1024)
	respBytes, err := r.ReadSlice('\n')
	if err != nil {
		return
	}
	resp := string(respBytes)

	fields := strings.SplitN(resp, ":", 4)
	if len(fields) < 3 {
		return response, ProtocolError{resp}
	}
	for i, field := range fields {
		fields[i] = strings.TrimSpace(field)
	}

	switch fields[1] {
	case "USERID":
		if len(fields) != 4 {
			return response, ProtocolError{resp}
		}

		var os, charset string
		osAndCharset := strings.SplitN(fields[2], ",", 2)
		if len(osAndCharset) == 2 {
			os = osAndCharset[0]
			charset = osAndCharset[1]
		} else {
			os = osAndCharset[0]
			charset = "US-ASCII"
		}

		return Response{
			OS:         os,
			Charset:    charset,
			Identifier: fields[3],
		}, nil
	case "ERROR":
		if len(fields) != 3 {
			return response, ProtocolError{resp}
		}

		return response, ResponseError{fields[2]}
	default:
		err = ProtocolError{resp}
	}
	return
}