--- /dev/null
+/*
+ * 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) {
+ 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;
+}
projects / debian / amanda / blobdiff