cool! created higher-order function table for agent redirection.

(This used to be commit 7bb2e55d0d3a00a97bba99ebed722a4adf84093c)

1 files changed, 242 insertions, 0 deletions

diff --git a/source3/lib/vagent.c b/source3/lib/vagent.c
new file mode 100644
index 0000000000..96df03e2f3
--- /dev/null
+++ b/source3/lib/vagent.c
@@ -0,0 +1,242 @@
+/* 
+   Unix SMB/Netbios implementation.
+   Version 2
+   SMB agent/socket plugin
+   Copyright (C) Andrew Tridgell 1999
+   
+   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.
+   
+   You should have received a copy of the GNU General Public License
+   along with this program; if not, write to the Free Software
+   Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
+*/
+
+#include "includes.h"
+#include "smb.h"
+
+extern int DEBUGLEVEL;
+
+
+/****************************************************************************
+terminate socket connection
+****************************************************************************/
+static void sock_redir_free(struct vagent_ops *va, struct sock_redir *sock)
+{
+	if (sock->c != -1)
+	{
+		close(sock->c);
+		sock->c = -1;
+	}
+	if (sock->n != NULL)
+	{
+		va->free_sock(sock->n);
+		sock->n = NULL;
+	}
+	free(sock);
+}
+
+/****************************************************************************
+free a sockent array
+****************************************************************************/
+static void free_sock_array(struct vagent_ops*va)
+{
+	void(*fn)(void*) = (void(*)(void*))&va->free_sock;
+	free_void_array(va->num_socks, (void**)va->socks, *fn);
+}
+
+/****************************************************************************
+add a sockent state to the array
+****************************************************************************/
+static struct sock_redir* add_sock_to_array(uint32 *len,
+				struct sock_redir ***array,
+				struct sock_redir *sock)
+{
+	int i;
+	for (i = 0; i < (*len); i++)
+	{
+		if ((*array)[i] == NULL)
+		{
+			(*array)[i] = sock;
+			return sock;
+		}
+	}
+
+	return (struct sock_redir*)add_item_to_array(len,
+	                     (void***)array, (void*)sock);
+				
+}
+
+/****************************************************************************
+initiate sockent array
+****************************************************************************/
+void init_sock_redir(struct vagent_ops*va)
+{
+	va->socks = NULL;
+	va->num_socks = 0;
+}
+
+/****************************************************************************
+terminate sockent array
+****************************************************************************/
+void free_sock_redir(struct vagent_ops*va)
+{
+	free_sock_array(va);
+	init_sock_redir(va);
+}
+
+/****************************************************************************
+create a new sockent state from user credentials
+****************************************************************************/
+static struct sock_redir *sock_redir_get(struct vagent_ops *va, int fd)
+{
+	struct sock_redir *sock = (struct sock_redir*)malloc(sizeof(*sock));
+
+	if (sock == NULL)
+	{
+		return NULL;
+	}
+
+	ZERO_STRUCTP(sock);
+
+	sock->c = fd;
+	sock->n = NULL;
+
+	DEBUG(10,("sock_redir_get:\tfd:\t%d\n", fd));
+
+	return sock;
+}
+/****************************************************************************
+init sock state
+****************************************************************************/
+static void sock_add(struct vagent_ops *va, int fd)
+{
+	struct sock_redir *sock;
+	sock = sock_redir_get(va, fd);
+	if (sock != NULL)
+	{
+		add_sock_to_array(&va->num_socks, &va->socks, sock);
+	}
+}
+
+/****************************************************************************
+delete a sockent state
+****************************************************************************/
+static BOOL sock_del(struct vagent_ops *va, int fd)
+{
+	int i;
+
+	for (i = 0; i < va->num_socks; i++)
+	{
+		if (va->socks[i] == NULL) continue;
+		if (va->socks[i]->c == fd) 
+		{
+			sock_redir_free(va, va->socks[i]);
+			va->socks[i] = NULL;
+			return True;
+		}
+	}
+
+	return False;
+}
+
+void start_agent(struct vagent_ops *va)
+{
+	int s, c;
+
+	s = va->get_agent_sock(va->id);
+
+	while (1)
+	{
+		int i;
+		fd_set fds;
+		int num;
+		struct sockaddr_un addr;
+		int in_addrlen = sizeof(addr);
+		int maxfd = s;
+		
+		FD_ZERO(&fds);
+		FD_SET(s, &fds);
+
+		for (i = 0; i < va->num_socks; i++)
+		{
+			if (va->socks[i] != NULL)
+			{
+				int fd = va->socks[i]->c;
+				FD_SET(fd, &fds);
+				maxfd = MAX(maxfd, fd);
+
+				if (va->socks[i]->n != NULL)
+				{
+					fd = va->socks[i]->s;
+					FD_SET(fd, &fds);
+					maxfd = MAX(fd, maxfd);
+				}
+			}
+		}
+
+		dbgflush();
+		num = sys_select(maxfd+1,&fds,NULL, NULL);
+
+		if (num <= 0)
+		{
+			continue;
+		}
+
+		if (FD_ISSET(s, &fds))
+		{
+			FD_CLR(s, &fds);
+			c = accept(s, (struct sockaddr*)&addr, &in_addrlen);
+			if (c != -1)
+			{
+				sock_add(va, c);
+			}
+		}
+
+		for (i = 0; i < va->num_socks; i++)
+		{
+			if (va->socks[i] == NULL)
+			{
+				continue;
+			}
+			if (FD_ISSET(va->socks[i]->c, &fds))
+			{
+				FD_CLR(va->socks[i]->c, &fds);
+				if (!va->process_cli_sock(va->socks,
+				                          va->num_socks,
+				                          va->socks[i]))
+				{
+					sock_redir_free(va, va->socks[i]);
+					va->socks[i] = NULL;
+				}
+			}
+			if (va->socks[i] == NULL)
+			{
+				continue;
+			}
+			if (va->socks[i]->n == NULL)
+			{
+				continue;
+			}
+			if (FD_ISSET(va->socks[i]->s, &fds))
+			{
+				FD_CLR(va->socks[i]->s, &fds);
+				if (!va->process_srv_sock(va->socks,
+				                          va->num_socks,
+				                          va->socks[i]->s))
+				{
+					sock_redir_free(va, va->socks[i]);
+					va->socks[i] = NULL;
+				}
+			}
+		}
+	}
+}
+