summaryrefslogtreecommitdiff
path: root/src/pmproxy/client.c
diff options
context:
space:
mode:
Diffstat (limited to 'src/pmproxy/client.c')
-rw-r--r--src/pmproxy/client.c260
1 files changed, 260 insertions, 0 deletions
diff --git a/src/pmproxy/client.c b/src/pmproxy/client.c
new file mode 100644
index 0000000..f45534b
--- /dev/null
+++ b/src/pmproxy/client.c
@@ -0,0 +1,260 @@
+/*
+ * Copyright (c) 2012-2013 Red Hat.
+ * Copyright (c) 1995-2002 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 "pmproxy.h"
+
+#define MIN_CLIENTS_ALLOC 8
+
+ClientInfo *client;
+int nClients; /* Number in array, (not all in use) */
+int maxReqPortFd; /* highest request port fd */
+int maxSockFd; /* largest fd for a client */
+__pmFdSet sockFds; /* for client select() */
+
+static int
+NewClient(void)
+{
+ int i;
+ static int clientSize;
+
+ for (i = 0; i < nClients; i++)
+ if (!client[i].status.connected)
+ break;
+
+ if (i == clientSize) {
+ int j, sz;
+
+ clientSize = clientSize ? clientSize * 2 : MIN_CLIENTS_ALLOC;
+ sz = sizeof(ClientInfo) * clientSize;
+ client = (ClientInfo *) realloc(client, sz);
+ if (client == NULL) {
+ __pmNoMem("NewClient", sz, PM_RECOV_ERR);
+ Shutdown();
+ exit(1);
+ }
+ for (j = i; j < clientSize; j++) {
+ client[j].addr = NULL;
+ }
+ }
+ client[i].addr = __pmSockAddrAlloc();
+ if (client[i].addr == NULL) {
+ __pmNoMem("NewClient", __pmSockAddrSize(), PM_RECOV_ERR);
+ Shutdown();
+ exit(1);
+ }
+ if (i >= nClients)
+ nClients = i + 1;
+ return i;
+}
+
+/* MY_BUFLEN needs to big enough to hold "hostname port" */
+#define MY_BUFLEN (MAXHOSTNAMELEN+10)
+#define MY_VERSION "pmproxy-server 1\n"
+
+/* Establish a new socket connection to a client */
+ClientInfo *
+AcceptNewClient(int reqfd)
+{
+ int i;
+ int fd;
+ __pmSockLen addrlen;
+ int ok = 0;
+ char buf[MY_BUFLEN];
+ char *bp;
+ char *endp;
+ char *abufp;
+
+ i = NewClient();
+ addrlen = __pmSockAddrSize();
+ fd = __pmAccept(reqfd, client[i].addr, &addrlen);
+ if (fd == -1) {
+ __pmNotifyErr(LOG_ERR, "AcceptNewClient(%d) __pmAccept failed: %s",
+ reqfd, netstrerror());
+ Shutdown();
+ exit(1);
+ }
+ __pmSetSocketIPC(fd);
+ if (fd > maxSockFd)
+ maxSockFd = fd;
+ __pmFD_SET(fd, &sockFds);
+
+ client[i].fd = fd;
+ client[i].pmcd_fd = -1;
+ client[i].status.connected = 1;
+ client[i].pmcd_hostname = NULL;
+
+ /*
+ * version negotiation (converse to negotiate_proxy() logic in
+ * libpcp
+ *
+ * __pmRecv client version message
+ * __pmSend my server version message
+ * __pmRecv pmcd hostname and pmcd port
+ */
+ for (bp = buf; bp < &buf[MY_BUFLEN]; bp++) {
+ if (__pmRecv(fd, bp, 1, 0) != 1) {
+ *bp = '\0'; /* null terminate what we have */
+ bp = &buf[MY_BUFLEN]; /* flag error */
+ break;
+ }
+ /* end of line means no more ... */
+ if (*bp == '\n' || *bp == '\r') {
+ *bp = '\0';
+ break;
+ }
+ }
+ if (bp < &buf[MY_BUFLEN]) {
+ /* looks OK so far ... is this a version we can support? */
+ if (strcmp(buf, "pmproxy-client 1") == 0) {
+ client[i].version = 1;
+ ok = 1;
+ }
+ }
+
+ if (!ok) {
+ abufp = __pmSockAddrToString(client[i].addr);
+ __pmNotifyErr(LOG_WARNING, "Bad version string from client at %s", abufp);
+ free(abufp);
+ fprintf(stderr, "AcceptNewClient: bad version string was \"");
+ for (bp = buf; *bp && bp < &buf[MY_BUFLEN]; bp++)
+ fputc(*bp & 0xff, stderr);
+ fprintf(stderr, "\"\n");
+ DeleteClient(&client[i]);
+ return NULL;
+ }
+
+ if (__pmSend(fd, MY_VERSION, strlen(MY_VERSION), 0) != strlen(MY_VERSION)) {
+ abufp = __pmSockAddrToString(client[i].addr);
+ __pmNotifyErr(LOG_WARNING, "AcceptNewClient: failed to send version "
+ "string (%s) to client at %s\n", MY_VERSION, abufp);
+ free(abufp);
+ DeleteClient(&client[i]);
+ return NULL;
+ }
+
+ for (bp = buf; bp < &buf[MY_BUFLEN]; bp++) {
+ if (__pmRecv(fd, bp, 1, 0) != 1) {
+ *bp = '\0'; /* null terminate what we have */
+ bp = &buf[MY_BUFLEN]; /* flag error */
+ break;
+ }
+ /* end of line means no more ... */
+ if (*bp == '\n' || *bp == '\r') {
+ *bp = '\0';
+ break;
+ }
+ }
+ if (bp < &buf[MY_BUFLEN]) {
+ /* looks OK so far ... get hostname and port */
+ for (bp = buf; *bp && *bp != ' '; bp++)
+ ;
+ if (bp != buf) {
+ *bp = '\0';
+ client[i].pmcd_hostname = strdup(buf);
+ if (client[i].pmcd_hostname == NULL)
+ __pmNoMem("PMCD.hostname", strlen(buf), PM_FATAL_ERR);
+ bp++;
+ client[i].pmcd_port = (int)strtoul(bp, &endp, 10);
+ if (*endp != '\0') {
+ abufp = __pmSockAddrToString(client[i].addr);
+ __pmNotifyErr(LOG_WARNING, "AcceptNewClient: bad pmcd port "
+ "\"%s\" from client at %s", bp, abufp);
+ free(abufp);
+ DeleteClient(&client[i]);
+ return NULL;
+ }
+ }
+ /* error, fall through */
+ }
+
+ if (client[i].pmcd_hostname == NULL) {
+ abufp = __pmSockAddrToString(client[i].addr);
+ __pmNotifyErr(LOG_WARNING, "AcceptNewClient: failed to get PMCD "
+ "hostname (%s) from client at %s", buf, abufp);
+ free(abufp);
+ DeleteClient(&client[i]);
+ return NULL;
+ }
+
+#ifdef PCP_DEBUG
+ if (pmDebug & DBG_TRACE_CONTEXT) {
+ /*
+ * note error message gets appended to once pmcd connection is
+ * made in ClientLoop()
+ */
+ abufp = __pmSockAddrToString(client[i].addr);
+ fprintf(stderr, "AcceptNewClient [%d] fd=%d from %s to %s (port %s)",
+ i, fd, abufp, client[i].pmcd_hostname, bp);
+ free(abufp);
+ }
+#endif
+
+ return &client[i];
+}
+
+void
+DeleteClient(ClientInfo *cp)
+{
+ int i;
+
+ for (i = 0; i < nClients; i++)
+ if (cp == &client[i])
+ break;
+
+ if (i == nClients) {
+ fprintf(stderr, "DeleteClient: Botch: tried to delete non-existent client @" PRINTF_P_PFX "%p\n", cp);
+ return;
+ }
+
+#ifdef PCP_DEBUG
+ if (pmDebug & DBG_TRACE_CONTEXT)
+ fprintf(stderr, "DeleteClient [%d]\n", i);
+#endif
+
+ if (cp->fd >= 0) {
+ __pmFD_CLR(cp->fd, &sockFds);
+ __pmCloseSocket(cp->fd);
+ }
+ if (cp->pmcd_fd >= 0) {
+ __pmFD_CLR(cp->pmcd_fd, &sockFds);
+ __pmCloseSocket(cp->pmcd_fd);
+ }
+ if (i == nClients-1) {
+ i--;
+ while (i >= 0 && !client[i].status.connected)
+ i--;
+ nClients = (i >= 0) ? i + 1 : 0;
+ }
+ if (cp->fd == maxSockFd || cp->pmcd_fd == maxSockFd) {
+ maxSockFd = maxReqPortFd;
+ for (i = 0; i < nClients; i++) {
+ if (cp->status.connected == 0)
+ continue;
+ if (client[i].fd > maxSockFd)
+ maxSockFd = client[i].fd;
+ if (client[i].pmcd_fd > maxSockFd)
+ maxSockFd = client[i].pmcd_fd;
+ }
+ }
+ __pmSockAddrFree(cp->addr);
+ cp->addr = NULL;
+ cp->status.connected = 0;
+ cp->fd = -1;
+ cp->pmcd_fd = -1;
+ if (cp->pmcd_hostname != NULL) {
+ free(cp->pmcd_hostname);
+ cp->pmcd_hostname = NULL;
+ }
+}
generated by cgit v1.2.3 (git 2.39.1) at 2024-06-10 01:39:23 +0000