Debian 3.9.10debian/3.9.10debian

1 files changed, 225 insertions, 0 deletions

diff --git a/src/pmdas/summary/mainloop.c b/src/pmdas/summary/mainloop.c
new file mode 100644
index 0000000..e59a467
--- /dev/null
+++ b/src/pmdas/summary/mainloop.c
@@ -0,0 +1,225 @@
+/*
+ * Copyright (c) 1995 Silicon Graphics, Inc.  All Rights Reserved.
+ * 
+ * This program is free software; you can redistribute it and/or modify it
+ * under the terms of the GNU General Public License as published by the
+ * Free Software Foundation; either version 2 of the License, or (at your
+ * option) any later version.
+ * 
+ * This program is distributed in the hope that it will be useful, but
+ * WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
+ * or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
+ * for more details.
+ */
+
+#include "pmapi.h"
+#include "impl.h"
+#include "pmda.h"
+#include "summary.h"
+
+static void (*freeResultCallback)(pmResult *) = __pmFreeResultValues;
+
+void
+mainLoopFreeResultCallback(void (*callback)(pmResult *res))
+{
+    freeResultCallback = callback;
+}
+
+void
+summaryMainLoop(char *pmdaname, int clientfd, pmdaInterface *dtp)
+{
+    __pmPDU		*pb_pmcd;
+    __pmPDU		*pb_client;
+    int			sts;
+    pmID		pmid;
+    pmDesc		desc;
+    int			npmids;
+    pmID		*pmidlist;
+    pmResult		*result;
+    int			ctxnum;
+    __pmTimeval		when;
+    int			ident;
+    int			type;
+    pmInDom		indom;
+    int			inst;
+    char		*name;
+    __pmInResult		*inres;
+    char		*buffer;
+    __pmProfile		*profile;
+    __pmProfile		*saveprofile = NULL;
+    static fd_set	readFds;
+    int			maxfd;
+    int			clientReady, pmcdReady;
+    int infd, outfd;
+
+    if (dtp->comm.pmda_interface != PMDA_INTERFACE_2) {
+	__pmNotifyErr(LOG_CRIT, 
+		     "summaryMainLoop supports PMDA protocol version 2 only, "
+		     "not %d\n", dtp->comm.pmda_interface);
+	exit(1);
+    } else {
+	infd = dtp->version.two.ext->e_infd;
+	outfd = dtp->version.two.ext->e_outfd;
+    }
+
+    maxfd = infd+1;
+    if (clientfd >= maxfd)
+	maxfd = clientfd+1;
+
+    for ( ;; ) {
+	FD_ZERO(&readFds);
+	FD_SET(infd, &readFds);
+	FD_SET(clientfd, &readFds);
+
+	/* select here : block if nothing to do */
+	sts = select(maxfd, &readFds, NULL, NULL, NULL);
+
+	clientReady = FD_ISSET(clientfd, &readFds);
+	pmcdReady = FD_ISSET(infd, &readFds);
+
+	if (sts < 0)
+	    break;
+	if (sts == 0)
+	    continue;
+
+	if (clientReady) {
+	    /*
+	     * Service the command/client
+	     */
+	    sts = __pmGetPDU(clientfd, ANY_SIZE, TIMEOUT_NEVER, &pb_client);
+	    if (sts < 0)
+		__pmNotifyErr(LOG_ERR, "client __pmGetPDU: %s\n", pmErrStr(sts));
+	    if (sts <= 0)
+		/* End of File or error */
+		goto done;
+
+	    service_client(pb_client);
+	    __pmUnpinPDUBuf(pb_client);
+	}
+
+	if (pmcdReady) {
+	    /* service pmcd */
+	    sts = __pmGetPDU(infd, ANY_SIZE, TIMEOUT_NEVER, &pb_pmcd);
+
+	    if (sts < 0)
+		__pmNotifyErr(LOG_ERR, "__pmGetPDU: %s\n", pmErrStr(sts));
+	    if (sts <= 0)
+		/* End of File or error */
+		goto done;
+
+	    switch (sts) {
+
+		case PDU_PROFILE:
+		    /*
+		     * can ignore ctxnum, since pmcd has already used this to send
+		     * the correct profile, if required
+		     */
+		    if ((sts = __pmDecodeProfile(pb_pmcd, &ctxnum, &profile)) >= 0)
+			sts = dtp->version.two.profile(profile,
+                                                       dtp->version.two.ext);
+		    if (sts < 0)
+			__pmSendError(outfd, FROM_ANON, sts);
+		    else {
+			if (saveprofile != NULL)
+			    free(saveprofile);
+			/*
+			 * need to keep the last valid one around, as the DSO
+			 * routine just remembers the address
+			 */
+			saveprofile = profile;
+		    }
+		    break;
+
+		case PDU_FETCH:
+		    /*
+		     * can ignore ctxnum, since pmcd has already used this to send
+		     * the correct profile, if required
+		     */
+		    sts = __pmDecodeFetch(pb_pmcd, &ctxnum, &when, &npmids, &pmidlist);
+
+		    /* Ignore "when"; pmcd should intercept archive log requests */
+		    if (sts >= 0) {
+			sts = dtp->version.two.fetch(npmids, pmidlist, &result,
+						     dtp->version.two.ext);
+			__pmUnpinPDUBuf(pmidlist);
+		    }
+		    if (sts < 0)
+			__pmSendError(outfd, FROM_ANON, sts);
+		    else {
+			int st;
+			st =__pmSendResult(outfd, FROM_ANON, result);
+			if (st < 0) {
+			    __pmNotifyErr(LOG_ERR, 
+					  "Cannot send fetch result: %s\n",
+					  pmErrStr(st));
+			}
+			(*freeResultCallback)(result);
+		    }
+		    break;
+
+		case PDU_DESC_REQ:
+		    if ((sts = __pmDecodeDescReq(pb_pmcd, &pmid)) >= 0) {
+			sts = dtp->version.two.desc(pmid, &desc,
+						    dtp->version.two.ext);
+		    }
+		    if (sts < 0)
+			__pmSendError(outfd, FROM_ANON, sts);
+		    else
+			__pmSendDesc(outfd, FROM_ANON, &desc);
+		    break;
+
+		case PDU_INSTANCE_REQ:
+		    if ((sts = __pmDecodeInstanceReq(pb_pmcd, &when, &indom, &inst, &name)) >= 0) {
+			/*
+			 * Note: when is ignored.
+			 *		If we get this far, we are _only_ dealing
+			 *		with current data (pmcd handles the other
+			 *		cases).
+			 */
+			sts = dtp->version.two.instance(indom, inst, name,
+                                                        &inres,
+						        dtp->version.two.ext);
+		    }
+		    if (sts < 0)
+			__pmSendError(outfd, FROM_ANON, sts);
+		    else {
+			__pmSendInstance(outfd, FROM_ANON, inres);
+			__pmFreeInResult(inres);
+		    }
+		    break;
+
+		case PDU_TEXT_REQ:
+		    if ((sts = __pmDecodeTextReq(pb_pmcd, &ident, &type)) >= 0) {
+			sts = dtp->version.two.text(ident, type, &buffer,
+						    dtp->version.two.ext);
+		    }
+		    if (sts < 0)
+			__pmSendError(outfd, FROM_ANON, sts);
+		    else
+			__pmSendText(outfd, FROM_ANON, ident, buffer);
+		    break;
+
+		case PDU_RESULT:
+		    if ((sts = __pmDecodeResult(pb_pmcd, &result)) >= 0)
+			sts = dtp->version.two.store(result,
+						     dtp->version.two.ext);
+		    __pmSendError(outfd, FROM_ANON, sts);
+		    pmFreeResult(result);
+		    break;
+
+		case PDU_ERROR:
+		    /* end of context from PMCD ... we don't care */
+		    break;
+
+		default:
+		    fprintf(stderr, "%s: bogus pdu type: 0x%0x?\n", pmdaname, sts);
+		    __pmSendError(outfd, FROM_ANON, PM_ERR_NYI);
+		    break;
+	    }
+	    __pmUnpinPDUBuf(pb_pmcd);
+	}
+    }
+
+done:
+    return;
+}