/*
 * Copyright (c) 1998,1999,2000
 *	Traakan, Inc., Los Altos, CA
 *	All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice unmodified, this list of conditions, and the following
 *    disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 *
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
 * SUCH DAMAGE.
 */

/*
 * Project:  NDMJOB
 * Ident:    $Id: $
 *
 * Description:
 *
 */


#include "ndmjob.h"
#include "ndmlib.h"


#ifndef NDMOS_OPTION_NO_NDMP4
#define MAX_PROTOCOL_VERSION	NDMP4VER
#else /* !NDMOS_OPTION_NO_NDMP4 */
#ifndef NDMOS_OPTION_NO_NDMP3
#define MAX_PROTOCOL_VERSION	NDMP3VER
#else /* !NDMOS_OPTION_NO_NDMP3 */
#ifndef NDMOS_OPTION_NO_NDMP2
#define MAX_PROTOCOL_VERSION	NDMP2VER
#else /* !NDMOS_OPTION_NO_NDMP2 */
#define MAX_PROTOCOL_VERSION	0
#endif /* !NDMOS_OPTION_NO_NDM2 */
#endif /* !NDMOS_OPTION_NO_NDMP3 */
#endif /* !NDMOS_OPTION_NO_NDMP4 */



/*
 * INITIALIZE AND DESTRUCT
 ****************************************************************
 *
 * Initialize an ndmconn. This pretty much amounts to
 * initializing the underlying ndmchan and stuffing
 * the function pointers.
 */

struct ndmconn *
ndmconn_initialize (struct ndmconn *aconn, char *name)
{
	struct ndmconn *	conn = aconn;

	if (!conn) {
		conn = NDMOS_MACRO_NEW(struct ndmconn);
		if (!conn)
			return 0;
	}

	NDMOS_MACRO_ZEROFILL(conn);

	if (!name) name = "#?";				/* default */

	ndmchan_initialize (&conn->chan, name);
	conn->was_allocated = aconn == 0;
	conn->next_sequence = 1;
	xdrrec_create (&conn->xdrs, 0, 0, (void*) conn,
				(void*)ndmconn_readit,
				(void*)ndmconn_writeit);
	conn->unexpected = ndmconn_unexpected;

	conn->call = ndmconn_call;

	conn->time_limit = 0;

	return conn;
}

/*
 * Get rid of an ndmconn.
 */
void
ndmconn_destruct (struct ndmconn *conn)
{
	if (conn->chan.fd >= 0) {
		close (conn->chan.fd);
		conn->chan.fd = -1;
	}

	xdr_destroy (&conn->xdrs);

	if (conn->was_allocated) {
		NDMOS_API_FREE (conn);
		conn = 0;
	}
}

/*
 * ESTABLISH CONNECTION
 ****************************************************************
 *
 * The following four routines establish the TCP/IP connection
 * between agents.
 *
 * ndmconn_connect_agent()
 *     make a connection per an ndmagent, uses ..._host_port()
 * ndmconn_connect_host_port ()
 *     make a connection per a hostname and port, uses ..._sockaddr_in()
 * ndmconn_connect_sockaddr_in()
 *     make a connection per sockaddr_in, performs NDMP_CONNECT_
 *     sequences, but no authentication
 * ndmconn_accept()
 *     make a connection (receive it really) from a file descriptor
 *     already accept()ed.
 */

int
ndmconn_connect_agent (struct ndmconn *conn, struct ndmagent *agent)
{
	if (agent->conn_type == NDMCONN_TYPE_RESIDENT) {
		conn->conn_type = NDMCONN_TYPE_RESIDENT;
		conn->protocol_version = agent->protocol_version;
		if (conn->protocol_version == 0) {
			/* Let's negotiate......MAX */
			conn->protocol_version = MAX_PROTOCOL_VERSION;
		}
		ndmchan_start_resident (&conn->chan);
		return 0;
	}

	if (agent->port == 0) agent->port = NDMPPORT;

	return ndmconn_connect_host_port (conn,
			agent->host, agent->port, agent->protocol_version);
}


int
ndmconn_connect_host_port (struct ndmconn *conn,
  char * hostname, int port, unsigned want_protocol_version)
{
	struct sockaddr_in	sin;
	char *			err = "???";

	if (conn->chan.fd >= 0) {
		err = "already-connected";
		return ndmconn_set_err_msg (conn, err);
	}

	if (ndmhost_lookup (hostname, &sin) != 0) {
		err = "bad-host-name";
		return ndmconn_set_err_msg (conn, err);
	}

	if (port == 0) port = NDMPPORT;

	sin.sin_port = htons(port);

	return ndmconn_connect_sockaddr_in (conn, &sin, want_protocol_version);
}

int
ndmconn_connect_sockaddr_in (struct ndmconn *conn,
  struct sockaddr_in *sin, unsigned want_protocol_version)
{
	int			fd = -1;
	int			rc;
	char *			err = "???";
	unsigned		max_protocol_version = MAX_PROTOCOL_VERSION;

	if (conn->chan.fd >= 0) {
		err = "already-connected";
		return ndmconn_set_err_msg (conn, err);
	}

	fd = socket (AF_INET, SOCK_STREAM, 0);
	if (fd < 0) {
		err = malloc(1024);
		snprintf(err, 1023, "open a socket failed: %s", strerror(errno));
		goto error_out;
	}

	/* reserved port? */
	if (connect (fd, (struct sockaddr *)sin, sizeof *sin) < 0) {
		// leak memory
		err = malloc(1024);
		snprintf(err, 1023, "connect failed: %s", strerror(errno));
		goto error_out;
	}

	ndmchan_start_readchk (&conn->chan, fd);
	conn->conn_type = NDMCONN_TYPE_REMOTE;

	/*
	 * Await the NDMP_NOTIFY_CONNECTED request (no reply)
	 * Don't get confused that this client-side is awaiting
	 * a "request" from the server.
	 */
	NDMC_WITH_NO_REPLY(ndmp0_notify_connected,0)
		rc = ndmconn_recv_nmb(conn, &xa->request);
		if (rc != 0) {
			err = "recv-notify-connected";
			goto error_out;
		}
		if (xa->request.header.message_type != NDMP0_MESSAGE_REQUEST
		 || xa->request.header.message != NDMP0_NOTIFY_CONNECTED) {
			err = "msg-not-notify-connected";
			goto error_out;
		}

		if (request->reason != NDMP0_CONNECTED) {
			err = "notify-connected-not-connected";
			goto error_out;
		}

		if (max_protocol_version > request->protocol_version) {
			max_protocol_version = request->protocol_version;
		}
	NDMC_ENDWITH

	if (want_protocol_version == 0) {
		want_protocol_version = max_protocol_version;
	} else if (want_protocol_version > max_protocol_version) {
		err = "connect-want/max-version-mismatch";
		goto error_out;
	}

	/*
	 * Send the OPEN request
	 */
	NDMC_WITH(ndmp0_connect_open,0)
		request->protocol_version = want_protocol_version;
		rc = NDMC_CALL(conn);
		if (rc) {
			err = "connect-open-failed";
			goto error_out;
		}
	NDMC_ENDWITH

	/* GOOD! */

	conn->protocol_version = want_protocol_version;

	return 0;

  error_out:
	if (fd >= 0) {
		close (fd);
		fd = -1;
	}
	conn->chan.fd = -1;
	conn->chan.mode = NDMCHAN_MODE_IDLE;
	conn->conn_type = NDMCONN_TYPE_NONE;

	return ndmconn_set_err_msg (conn, err);
}

int
ndmconn_try_open (struct ndmconn *conn, unsigned protocol_version)
{
	int		rc;

	/*
	 * Send the OPEN request
	 */
	NDMC_WITH(ndmp0_connect_open,0)
		request->protocol_version = protocol_version;
		rc = NDMC_CALL(conn);
		if (rc) {
			ndmconn_set_err_msg (conn, "connect-open-failed");
		}
	NDMC_ENDWITH

	return rc;
}

int
ndmconn_accept (struct ndmconn *conn, int sock)
{
	char *			err = "???";

	if (conn->chan.fd >= 0) {
		err = "already-connected";
		return ndmconn_set_err_msg (conn, err);
	}

	ndmchan_start_readchk (&conn->chan, sock);
	conn->conn_type = NDMCONN_TYPE_REMOTE;

	/*
	 * Send the NDMP_NOTIFY_CONNECTED message, no reply
	 * The connect()er is waiting for it.
	 */
	NDMC_WITH_NO_REPLY(ndmp0_notify_connected,0)
		request->reason = NDMP0_CONNECTED;
		request->protocol_version = MAX_PROTOCOL_VERSION;
		request->text_reason = "Hello";
		NDMC_SEND(conn);
	NDMC_ENDWITH

	/* assume connection is running in offered protocol_version */
	conn->protocol_version = MAX_PROTOCOL_VERSION;

	return 0;
}




/*
 * TERMINATE CONNECTION
 ****************************************************************
 *
 * These two routines are about terminating a connection.
 * They are incomplete.
 */

/* hangup */
int
ndmconn_abort (struct ndmconn *conn)
{
	return 0;
}

/* orderly close */
int
ndmconn_close (struct ndmconn *conn)
{
	return 0;
}


/*
 * Return the underlying fd of the ndmconn.
 * This is no longer used since the ndmchan stuff was done.
 */

int
ndmconn_fileno (struct ndmconn *conn)
{
	return conn->chan.fd;
}





/*
 * AUTHENTICATION
 *
 * The following three routines do the NDMP_CONNECT_AUTH sequences.
 */

int
ndmconn_auth_agent (struct ndmconn *conn, struct ndmagent *agent)
{
	int		rc;

	if (conn->conn_type == NDMCONN_TYPE_RESIDENT)
		return 0;

	switch (agent->auth_type) {
	case 'n':		/* NDMP_AUTH_NONE */
		rc = ndmconn_auth_none (conn);
		break;

	case 't':		/* NDMP_AUTH_TEXT */
		rc = ndmconn_auth_text (conn, agent->account, agent->password);
		break;

	case 'm':		/* NDMP_AUTH_MD5 */
		rc = ndmconn_auth_md5 (conn, agent->account, agent->password);
		break;

	case 'v':		/* void (don't auth) */
		rc = 0;
		break;

	default:
		ndmconn_set_err_msg (conn, "connect-auth-unknown");
		rc = -1;
		break;
	}

	return rc;
}

int
ndmconn_auth_none (struct ndmconn *conn)
{
	int		rc;

	switch (conn->protocol_version) {
	default:
		ndmconn_set_err_msg (conn, "connect-auth-none-vers-botch");
		return -1;

#ifndef NDMOS_OPTION_NO_NDMP2
	case NDMP2VER:
	    NDMC_WITH(ndmp2_connect_client_auth, NDMP2VER)
		request->auth_data.auth_type = NDMP2_AUTH_NONE;
		rc = NDMC_CALL(conn);
	    NDMC_ENDWITH
	    break;
#endif /* !NDMOS_OPTION_NO_NDMP2 */

#ifndef NDMOS_OPTION_NO_NDMP3
	case NDMP3VER:
	    NDMC_WITH(ndmp3_connect_client_auth, NDMP3VER)
		request->auth_data.auth_type = NDMP3_AUTH_NONE;
		rc = NDMC_CALL(conn);
	    NDMC_ENDWITH
	    break;
#endif /* !NDMOS_OPTION_NO_NDMP3 */

#ifndef NDMOS_OPTION_NO_NDMP4
	case NDMP4VER:
	    NDMC_WITH(ndmp4_connect_client_auth, NDMP4VER)
		request->auth_data.auth_type = NDMP4_AUTH_NONE;
		rc = NDMC_CALL(conn);
	    NDMC_ENDWITH
	    break;
#endif /* !NDMOS_OPTION_NO_NDMP4 */
	}

	if (rc) {
		ndmconn_set_err_msg (conn, "connect-auth-none-failed");
		return -1;
	}

	return 0;
}

int
ndmconn_auth_text (struct ndmconn *conn, char *id, char *pw)
{
	int		rc;

	switch (conn->protocol_version) {
	default:
		ndmconn_set_err_msg (conn, "connect-auth-text-vers-botch");
		return -1;

#ifndef NDMOS_OPTION_NO_NDMP2
	case NDMP2VER:
	    NDMC_WITH(ndmp2_connect_client_auth, NDMP2VER)
		struct ndmp2_auth_text *at;

		request->auth_data.auth_type = NDMP2_AUTH_TEXT;
		at = &request->auth_data.ndmp2_auth_data_u.auth_text;
		at->auth_id = id;
		at->auth_password = pw;
		rc = NDMC_CALL(conn);
	    NDMC_ENDWITH
	    break;
#endif /* !NDMOS_OPTION_NO_NDMP2 */

#ifndef NDMOS_OPTION_NO_NDMP3
	case NDMP3VER:
	    NDMC_WITH(ndmp3_connect_client_auth, NDMP3VER)
		struct ndmp3_auth_text *at;

		request->auth_data.auth_type = NDMP3_AUTH_TEXT;
		at = &request->auth_data.ndmp3_auth_data_u.auth_text;
		at->auth_id = id;
		at->auth_password = pw;
		rc = NDMC_CALL(conn);
	    NDMC_ENDWITH
	    break;
#endif /* !NDMOS_OPTION_NO_NDMP3 */

#ifndef NDMOS_OPTION_NO_NDMP4
	case NDMP4VER:
	    NDMC_WITH(ndmp4_connect_client_auth, NDMP4VER)
		struct ndmp4_auth_text *at;

		request->auth_data.auth_type = NDMP4_AUTH_TEXT;
		at = &request->auth_data.ndmp4_auth_data_u.auth_text;
		at->auth_id = id;
		at->auth_password = pw;
		rc = NDMC_CALL(conn);
	    NDMC_ENDWITH
	    break;
#endif /* !NDMOS_OPTION_NO_NDMP4 */
	}

	if (rc) {
		ndmconn_set_err_msg (conn, "connect-auth-text-failed");
		return -1;
	}

	return 0;
}

int
ndmconn_auth_md5 (struct ndmconn *conn, char *id, char *pw)
{
	int		rc;
	char		challenge[NDMP_MD5_CHALLENGE_LENGTH];
	char		digest[NDMP_MD5_DIGEST_LENGTH];

	switch (conn->protocol_version) {
	default:
		ndmconn_set_err_msg (conn, "connect-auth-md5-vers-botch");
		return -1;

#ifndef NDMOS_OPTION_NO_NDMP2
	case NDMP2VER:
	    NDMC_WITH(ndmp2_config_get_auth_attr, NDMP2VER)
		request->auth_type = NDMP2_AUTH_MD5;
		rc = NDMC_CALL(conn);
		if (rc == 0) {
			if (reply->server_attr.auth_type != NDMP2_AUTH_MD5) {
				ndmconn_set_err_msg (conn,
					"connect-auth-md5-attr-type-botch");
				return -1;
			}
			NDMOS_API_BCOPY (
				reply->server_attr.ndmp2_auth_attr_u.challenge,
				challenge, sizeof challenge);
		}
	    NDMC_ENDWITH
	    break;
#endif /* !NDMOS_OPTION_NO_NDMP2 */

#ifndef NDMOS_OPTION_NO_NDMP3
	case NDMP3VER:
	    NDMC_WITH(ndmp3_config_get_auth_attr, NDMP3VER)
		request->auth_type = NDMP3_AUTH_MD5;
		rc = NDMC_CALL(conn);
		if (rc == 0) {
			if (reply->server_attr.auth_type != NDMP3_AUTH_MD5) {
				ndmconn_set_err_msg (conn,
					"connect-auth-md5-attr-type-botch");
				return -1;
			}
			NDMOS_API_BCOPY (
				reply->server_attr.ndmp3_auth_attr_u.challenge,
				challenge, sizeof challenge);
		}
	    NDMC_ENDWITH
	    break;
#endif /* !NDMOS_OPTION_NO_NDMP3 */

#ifndef NDMOS_OPTION_NO_NDMP4
	case NDMP4VER:
	    NDMC_WITH(ndmp4_config_get_auth_attr, NDMP4VER)
		request->auth_type = NDMP4_AUTH_MD5;
		rc = NDMC_CALL(conn);
		if (rc == 0) {
			if (reply->server_attr.auth_type != NDMP4_AUTH_MD5) {
				ndmconn_set_err_msg (conn,
					"connect-auth-md5-attr-type-botch");
				return -1;
			}
			NDMOS_API_BCOPY (
				reply->server_attr.ndmp4_auth_attr_u.challenge,
				challenge, sizeof challenge);
		}
	    NDMC_ENDWITH
	    break;
#endif /* !NDMOS_OPTION_NO_NDMP4 */
	}

	if (rc) {
		ndmconn_set_err_msg (conn, "connect-auth-md5-attr-failed");
		return -1;
	}

	ndmmd5_digest (challenge, pw, digest);

	switch (conn->protocol_version) {
	default:
		ndmconn_set_err_msg (conn, "connect-auth-text-vers-botch");
		return -1;

#ifndef NDMOS_OPTION_NO_NDMP2
	case NDMP2VER:
	    NDMC_WITH(ndmp2_connect_client_auth, NDMP2VER)
		struct ndmp2_auth_md5 *am;

		request->auth_data.auth_type = NDMP2_AUTH_MD5;
		am = &request->auth_data.ndmp2_auth_data_u.auth_md5;
		am->auth_id = id;
		NDMOS_API_BCOPY (digest, am->auth_digest, sizeof digest);
		rc = NDMC_CALL(conn);
	    NDMC_ENDWITH
	    break;
#endif /* !NDMOS_OPTION_NO_NDMP2 */

#ifndef NDMOS_OPTION_NO_NDMP3
	case NDMP3VER:
	    NDMC_WITH(ndmp3_connect_client_auth, NDMP3VER)
		struct ndmp3_auth_md5 *am;

		request->auth_data.auth_type = NDMP3_AUTH_MD5;
		am = &request->auth_data.ndmp3_auth_data_u.auth_md5;
		am->auth_id = id;
		NDMOS_API_BCOPY (digest, am->auth_digest, sizeof digest);
		rc = NDMC_CALL(conn);
	    NDMC_ENDWITH
	    break;
#endif /* !NDMOS_OPTION_NO_NDMP3 */

#ifndef NDMOS_OPTION_NO_NDMP4
	case NDMP4VER:
	    NDMC_WITH(ndmp4_connect_client_auth, NDMP4VER)
		struct ndmp4_auth_md5 *am;

		request->auth_data.auth_type = NDMP4_AUTH_MD5;
		am = &request->auth_data.ndmp4_auth_data_u.auth_md5;
		am->auth_id = id;
		NDMOS_API_BCOPY (digest, am->auth_digest, sizeof digest);
		rc = NDMC_CALL(conn);
	    NDMC_ENDWITH
	    break;
#endif /* !NDMOS_OPTION_NO_NDMP4 */
	}

	if (rc) {
		ndmconn_set_err_msg (conn, "connect-auth-md5-failed");
		return -1;
	}

	return 0;
}




/*
 * CALL (REQUEST/REPLY), SEND, and RECEIVE
 ****************************************************************
 */

int
ndmconn_call (struct ndmconn *conn, struct ndmp_xa_buf *xa)
{
	unsigned		protocol_version = conn->protocol_version;
	unsigned		msg = xa->request.header.message;
	int			rc;
	struct ndmp_xdr_message_table *	xmte;

	conn->last_message = msg;
	conn->last_call_status = NDMCONN_CALL_STATUS_BOTCH;
	conn->last_header_error = -1;	/* invalid */
	conn->last_reply_error = -1;	/* invalid */

	if (protocol_version != xa->request.protocol_version) {
		ndmconn_set_err_msg (conn, "protocol-version-mismatch");
		return NDMCONN_CALL_STATUS_BOTCH;
	}

	xmte = ndmp_xmt_lookup (protocol_version, msg);
	if (!xmte) {
		ndmconn_set_err_msg (conn, "no-xdr-found");
		return NDMCONN_CALL_STATUS_BOTCH;
	}

	xa->request.header.message_type = NDMP0_MESSAGE_REQUEST;

	if (!xmte->xdr_reply) {
		/* no reply expected, just a send (eg NOTIFY) */
		return ndmconn_send_nmb (conn, &xa->request);
	}

	rc = ndmconn_exchange_nmb (conn, &xa->request, &xa->reply);
	if (rc) {
		ndmconn_set_err_msg (conn, "exchange-failed");
		return NDMCONN_CALL_STATUS_BOTCH;
	}

	if (xa->reply.header.message != msg) {
		ndmconn_set_err_msg (conn, "msg-mismatch");
		return NDMCONN_CALL_STATUS_BOTCH;
	}

	/* TODO: this should be converted ndmp_xto9_error(....) */
	conn->last_header_error = xa->reply.header.error;

	if (xa->reply.header.error) {
		conn->last_call_status = NDMCONN_CALL_STATUS_HDR_ERROR;
		ndmconn_set_err_msg (conn, "reply-error-hdr");
		return NDMCONN_CALL_STATUS_HDR_ERROR;
	}

	conn->last_reply_error = ndmnmb_get_reply_error (&xa->reply);

	if (conn->last_reply_error != NDMP9_NO_ERR) {
		conn->last_call_status = NDMCONN_CALL_STATUS_REPLY_ERROR;
		ndmconn_set_err_msg (conn, "reply-error");
		return NDMCONN_CALL_STATUS_REPLY_ERROR;
	}

	return NDMCONN_CALL_STATUS_OK;
}

int
ndmconn_exchange_nmb (struct ndmconn *conn,
  struct ndmp_msg_buf *request_nmb,
  struct ndmp_msg_buf *reply_nmb)
{
	int			rc;

	if ((rc = ndmconn_send_nmb (conn, request_nmb)) != 0)
		return rc;
	conn->received_time = 0;
	conn->sent_time = time(0);

	for (;;) {
		if ((rc = ndmconn_recv_nmb (conn, reply_nmb)) != 0)
			return rc;

		if (reply_nmb->header.message_type == NDMP0_MESSAGE_REPLY
		 && reply_nmb->header.reply_sequence
		    == request_nmb->header.sequence) {
			conn->received_time = time(0);
			return 0;
		}

		(*conn->unexpected)(conn, reply_nmb);
	}
}

int
ndmconn_send_nmb (struct ndmconn *conn, struct ndmp_msg_buf *nmb)
{
	return ndmconn_xdr_nmb (conn, nmb, XDR_ENCODE);
}

int
ndmconn_recv_nmb (struct ndmconn *conn, struct ndmp_msg_buf *nmb)
{
	NDMOS_MACRO_ZEROFILL (nmb);
	nmb->protocol_version = conn->protocol_version;

	return ndmconn_xdr_nmb (conn, nmb, XDR_DECODE);
}

void
ndmconn_free_nmb (struct ndmconn *conn, struct ndmp_msg_buf *nmb)
{
	ndmnmb_free (nmb);
}


int
ndmconn_xdr_nmb (struct ndmconn *conn,
  struct ndmp_msg_buf *nmb,
  enum xdr_op x_op)
{
	xdrproc_t xdr_body = 0;

	assert (conn->conn_type == NDMCONN_TYPE_REMOTE);

	if (conn->chan.fd < 0) {
		return ndmconn_set_err_msg (conn, "not-open");
	}

	conn->xdrs.x_op = x_op;

	if (x_op == XDR_ENCODE) {
		xdr_body = ndmnmb_find_xdrproc (nmb);

		if (nmb->header.error == NDMP0_NO_ERR && !xdr_body) {
			return ndmconn_set_err_msg (conn, "unknown-body");
		}
		nmb->header.sequence = conn->next_sequence++;
		nmb->header.time_stamp = time(0);
		ndmconn_snoop_nmb (conn, nmb, "Send");
	}
	if (x_op == XDR_DECODE) {
		if (!xdrrec_skiprecord (&conn->xdrs)) {
			return ndmconn_set_err_msg (conn, "xdr-get-next");
		}
	}

	if (!xdr_ndmp0_header (&conn->xdrs, &nmb->header)) {
		ndmconn_abort (conn);
		if (x_op == XDR_DECODE
		 && conn->chan.eof && !conn->chan.error) {
			return ndmconn_set_err_msg (conn, "EOF");
		} else {
			return ndmconn_set_err_msg (conn, "xdr-hdr");
		}
	}

	if (x_op == XDR_DECODE) {
		xdr_body = ndmnmb_find_xdrproc (nmb);

		if (nmb->header.error == NDMP0_NO_ERR && !xdr_body) {
			return ndmconn_set_err_msg (conn, "unknown-body");
		}
	}
	if (nmb->header.error == NDMP0_NO_ERR) {
		if (!(*xdr_body) (&conn->xdrs, &nmb->body)) {
			ndmconn_abort (conn);
			return ndmconn_set_err_msg (conn, "xdr-body");
		}
	}

	if (x_op == XDR_ENCODE) {
		if (!xdrrec_endofrecord(&conn->xdrs, 1)) {
			ndmconn_abort (conn);
			return ndmconn_set_err_msg (conn, "xdr-send");
		}
	}
	if (x_op == XDR_DECODE) {
		ndmconn_snoop_nmb (conn, nmb, "Recv");
	}

	return 0;
}




/*
 * XDR READ/WRITE CALLBACKS
 ****************************************************************
 *
 * ndmconn_readit() and ndmconn_writeit() are the XDR callbacks
 * used by xdrrec_create(). They are fundamentally wrappers
 * around read() and write(), and have very similar parameters.
 * See the xdr(3) manual page (or try "man xdrrec_create").
 *
 * ndmconn_readit() tracks the XDR record marks, and never
 * reads across a record boundary. This keeps select() an
 * indicator of when there is a (single) request pending.
 * Otherwise, we have to check buffers internal to XDR
 * as well as the file descriptor (via select) to determine
 * if a request is pending.
 */

int
ndmconn_readit (void *a_conn, char *buf, int len)
{
	struct ndmconn *conn = (struct ndmconn *)a_conn;
	int		rc, i, c;

	/* could impose timeout here */
	if (conn->chan.fd < 0 || conn->chan.eof)
		return -1;

	ndmconn_snoop (conn, 8,
		"frag_resid=%d fhb_off=%d", conn->frag_resid, conn->fhb_off);

	if (conn->frag_resid == 0) {
		i = 0;
		while (i < 4) {
			c = 4 - i;

			rc = ndmconn_sys_read (conn, (void *)(conn->frag_hdr_buf+i), c);
			if (rc <= 0) {
				return rc;
			}
			i += rc;
		}
		conn->frag_resid = conn->frag_hdr_buf[0] << 24;
		conn->frag_resid |= conn->frag_hdr_buf[1] << 16;
		conn->frag_resid |= conn->frag_hdr_buf[2] << 8;
		conn->frag_resid |= conn->frag_hdr_buf[3];
		conn->frag_resid &= 0xFFFFFF;
		conn->fhb_off = 0;
	}
	if (conn->fhb_off < 4) {
		i = 0;
		while (conn->fhb_off < 4 && len > 0) {
			buf[i++] = conn->frag_hdr_buf[conn->fhb_off++];
			len--;
		}
		return i;
	}

	if ((unsigned int)len > conn->frag_resid)
		len = (unsigned int)conn->frag_resid;

	rc = ndmconn_sys_read (conn, buf, len);

	if (rc > 0) {
		conn->frag_resid -= rc;
	}

	return rc;
}

int
ndmconn_writeit (void *a_conn, char *buf, int len)
{
	struct ndmconn *conn = (struct ndmconn *)a_conn;

	/* could impose timeout here */
	if (conn->chan.fd < 0)
		return -1;

	return ndmconn_sys_write (conn, buf, len);
}

/*
 * ndmconn_sys_read() and ndmconn_sys_write() are simply
 * wrappers around read() and write(). They implement
 * the low-level snooping.
 */

int
ndmconn_sys_read (struct ndmconn *conn, char *buf, unsigned len)
{
	int		rc;

	ndmconn_snoop (conn, 9, "reading %d ...", len);

	rc = read (conn->chan.fd, buf, len);

	ndmconn_snoop (conn, 8, "read=%d len=%d", rc, len);
	ndmconn_hex_dump (conn, buf, rc);

	if (rc <= 0) {
		conn->chan.eof = 1;
		if (rc < 0)
			conn->chan.error = 1;
	}

	return rc;
}

int
ndmconn_sys_write (struct ndmconn *conn, char *buf, unsigned len)
{
	int		rc;

	ndmconn_snoop (conn, 9, "writing %d ...", len);
	ndmconn_hex_dump (conn, buf, len);

	rc = write (conn->chan.fd, buf, len);

	ndmconn_snoop (conn, 8, "write=%d len=%d", rc, len);

	if (rc !=(int)len) {
		conn->chan.eof = 1;
		conn->chan.error = 1;
	}

	return rc;
}




/*
 * UNEXPECTED
 ****************************************************************
 *
 * The default unexpected() handler for a connection. It is
 * called when ndmconn_exchange_nmb() receives something
 * other than the reply for which it is waiting.
 * This default routine silently dumps the message.
 */

void
ndmconn_unexpected (struct ndmconn *conn, struct ndmp_msg_buf *nmb)
{
	xdrproc_t xdr_body = ndmnmb_find_xdrproc (nmb);

	if (xdr_body) {
		xdr_free (xdr_body, (void*) &nmb->body);
	}
}




/*
 * SNOOP
 ****************************************************************
 *
 * The ndmconn snoop stuff. The cool part. This pretty prints
 * NDMP messages as they go flying by this end-point.
 */

int
ndmconn_set_snoop (struct ndmconn *conn, struct ndmlog *log, int level)
{
	conn->snoop_log = log;
	conn->snoop_level = level;
	return 0;
}

void
ndmconn_clear_snoop (struct ndmconn *conn)
{
	conn->snoop_log = 0;
	conn->snoop_level = 0;
}

void
ndmconn_snoop_nmb (struct ndmconn *conn,
  struct ndmp_msg_buf *nmb,
  char *whence)
{
	if (!conn->snoop_log) {
		return;
	}

	ndmnmb_snoop (conn->snoop_log, conn->chan.name, conn->snoop_level,
				nmb, whence);
}

void
ndmconn_snoop (struct ndmconn *conn, int level, char *fmt, ...)
{
	va_list		ap;

	if (conn->snoop_log && conn->snoop_level >= level) {
		va_start (ap, fmt);
		ndmlogfv (conn->snoop_log, conn->chan.name, level, fmt, ap);
		va_end (ap);
	}
}

/* used by ndmconn_sys_read() and ndmconn_sys_write() to show low-level */
void
ndmconn_hex_dump (struct ndmconn *conn, char *buf, unsigned len)
{
	struct ndmlog *	log = conn->snoop_log;
	char *		tag = conn->chan.name;
	char		linebuf[16*3+3];
	char *		p = linebuf;
	int		b;
	unsigned	i;

	if (log && conn->snoop_level > 8) {
		for (i = 0; i < len; i++) {
			b = buf[i] & 0xFF;
			sprintf (p, " %02x", b);
			while (*p) p++;
			if ((i&0xF) == 0xF) {
				ndmlogf (log,tag,9,"%s",linebuf+1);
				p = linebuf;
			}
		}
		if (p > linebuf) {
			ndmlogf (log,tag,9,"%s",linebuf+1);
		}
	}
}




/*
 * ERRORS
 ****************************************************************
 *
 * Possible errors for ndmconn are not enumerated.
 * Instead, errors are indicated by a -1 return, and
 * a simple string error message is available for details.
 * Appologies for the english-centric design, but it
 * is quick and easy, and better than using printf().
 */

int
ndmconn_set_err_msg (struct ndmconn *conn, char *err_msg)
{
	conn->last_err_msg = err_msg;
	ndmconn_snoop (conn, 4, "ERR=%s", err_msg);
	return -1;
}

char *
ndmconn_get_err_msg (struct ndmconn *conn)
{
	if (!conn->last_err_msg)
		return "-no-error-";
	else
		return conn->last_err_msg;
}