first version of msrpc agent redirector code. client-side only.

(This used to be commit 5e5a1dceee0b6088822697284d3e0af04d197477)

1 files changed, 321 insertions, 0 deletions

diff --git a/source3/lib/msrpc_use.c b/source3/lib/msrpc_use.c
new file mode 100644
index 0000000000..a8fb009df0
--- /dev/null
+++ b/source3/lib/msrpc_use.c
@@ -0,0 +1,321 @@
+/* 
+   Unix SMB/Netbios implementation.
+   Version 1.9.
+   SMB client generic functions
+   Copyright (C) Andrew Tridgell 1994-1999
+   Copyright (C) Luke Kenneth Casson Leighton 1996-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.
+*/
+
+#define NO_SYSLOG
+
+#include "includes.h"
+#include "trans2.h"
+
+extern int DEBUGLEVEL;
+extern pstring scope;
+extern pstring global_myname;
+
+struct msrpc_use
+{
+	struct msrpc_state *cli;
+	uint32 num_users;
+};
+
+static struct msrpc_use **msrpcs = NULL;
+uint32 num_msrpcs = 0;
+
+/****************************************************************************
+terminate client connection
+****************************************************************************/
+static void msrpc_use_free(struct msrpc_use *cli)
+{
+	if (cli->cli != NULL)
+	{
+		if (cli->cli->initialised)
+		{
+			msrpc_shutdown(cli->cli);
+		}
+		free(cli->cli);
+	}
+
+	free(cli);
+}
+
+/****************************************************************************
+free a client array
+****************************************************************************/
+static void free_msrpc_array(uint32 num_entries, struct msrpc_use **entries)
+{
+	void(*fn)(void*) = (void(*)(void*))&msrpc_use_free;
+	free_void_array(num_entries, (void**)entries, *fn);
+}
+
+/****************************************************************************
+add a client state to the array
+****************************************************************************/
+static struct msrpc_use* add_msrpc_to_array(uint32 *len,
+				struct msrpc_use ***array,
+				struct msrpc_use *cli)
+{
+	int i;
+	for (i = 0; i < num_msrpcs; i++)
+	{
+		if (msrpcs[i] == NULL)
+		{
+			msrpcs[i] = cli;
+			return cli;
+		}
+	}
+
+	return (struct msrpc_use*)add_item_to_array(len,
+	                     (void***)array, (void*)cli);
+				
+}
+
+/****************************************************************************
+initiate client array
+****************************************************************************/
+void init_msrpc_use(void)
+{
+	msrpcs = NULL;
+	num_msrpcs = 0;
+}
+
+/****************************************************************************
+terminate client array
+****************************************************************************/
+void free_msrpc_use(void)
+{
+	free_msrpc_array(num_msrpcs, msrpcs);
+	init_msrpc_use();
+}
+
+/****************************************************************************
+find client state.  server name, user name, domain name and password must all
+match.
+****************************************************************************/
+static struct msrpc_use *msrpc_find(const char* pipe_name,
+				const struct user_credentials *usr_creds)
+{
+	int i;
+	struct user_credentials null_usr;
+
+	copy_user_creds(&null_usr, usr_creds);
+	usr_creds = &null_usr;
+		
+	DEBUG(10,("msrpc_find: %s %s %s\n",
+			pipe_name,
+			usr_creds->user_name,
+			usr_creds->domain));
+
+	for (i = 0; i < num_msrpcs; i++)
+	{
+		char *msrpc_name = NULL;
+		struct msrpc_use *c = msrpcs[i];
+
+		if (c == NULL) continue;
+
+		msrpc_name = c->cli->pipe_name;
+
+		DEBUG(10,("msrpc_find[%d]: %s %s %s\n",
+				i, msrpc_name,
+		                c->cli->usr.user_name,
+				c->cli->usr.domain));
+				
+		if (!strequal(msrpc_name, pipe_name))
+		{
+			continue;
+		}
+		if (!strequal(usr_creds->user_name, c->cli->usr.user_name))
+		{
+			continue;
+		}
+		if (!usr_creds->reuse &&
+		    !pwd_compare(&usr_creds->pwd, &c->cli->usr.pwd))
+		{
+			DEBUG(100,("password doesn't match\n"));
+			continue;
+		}
+		if (usr_creds->domain[0] == 0)
+		{
+			return c;
+		}
+		if (strequal(usr_creds->domain, c->cli->usr.domain))
+		{
+			return c;
+		}
+	}
+
+	return NULL;
+}
+
+/****************************************************************************
+create a new client state from user credentials
+****************************************************************************/
+static struct msrpc_use *msrpc_use_get(const char* pipe_name,
+				const struct user_credentials *usr_creds)
+{
+	struct msrpc_use *cli = (struct msrpc_use*)malloc(sizeof(*cli));
+
+	if (cli == NULL)
+	{
+		return NULL;
+	}
+
+	memset(cli, 0, sizeof(*cli));
+
+	cli->cli = msrpc_initialise(NULL);
+
+	if (cli->cli == NULL)
+	{
+		return NULL;
+	}
+
+	msrpc_init_creds(cli->cli, usr_creds);
+
+	return cli;
+}
+
+/****************************************************************************
+init client state
+****************************************************************************/
+struct msrpc_state *msrpc_use_add(const char* pipe_name,
+				const struct user_credentials *usr_creds,
+				BOOL redir)
+{
+	struct msrpc_use *cli = msrpc_find(pipe_name, usr_creds); 
+
+	if (cli != NULL)
+	{
+		cli->num_users++;
+		return cli->cli;
+	}
+
+	/* reuse an existing connection requested, and one was not found */
+	if (usr_creds != NULL && usr_creds->reuse && !redir)
+	{
+		return False;
+	}
+
+	/*
+	 * allocate
+	 */
+
+	cli = msrpc_use_get(pipe_name, usr_creds);
+	cli->cli->redirect = redir;
+
+	if (!msrpc_establish_connection(cli->cli, pipe_name))
+	{
+		DEBUG(0,("msrpc_net_use_add: connection failed\n"));
+		cli->cli = NULL;
+		msrpc_use_free(cli);
+		return NULL;
+	}
+
+	add_msrpc_to_array(&num_msrpcs, &msrpcs, cli);
+	cli->num_users++;
+
+	return cli->cli;
+}
+
+/****************************************************************************
+delete a client state
+****************************************************************************/
+BOOL msrpc_use_del(const char* pipe_name,
+				const struct user_credentials *usr_creds,
+				BOOL force_close,
+				BOOL *connection_closed)
+{
+	int i;
+
+	DEBUG(10,("msrpc_net_use_del: %s. force close: %s\n",
+	           pipe_name, BOOLSTR(force_close)));
+
+	if (connection_closed != NULL)
+	{
+		*connection_closed = False;
+	}
+
+	for (i = 0; i < num_msrpcs; i++)
+	{
+		char *msrpc_name = NULL;
+
+		if (msrpcs[i] == NULL) continue;
+		if (msrpcs[i]->cli == NULL) continue;
+
+		msrpc_name = msrpcs[i]->cli->pipe_name;
+
+		if (!strequal(msrpc_name, pipe_name)) continue;
+
+		if (strequal(usr_creds->user_name,
+                             msrpcs[i]->cli->usr.user_name) &&
+		    strequal(usr_creds->domain,
+		             msrpcs[i]->cli->usr.domain))
+		{
+			/* decrement number of users */
+			msrpcs[i]->num_users--;
+
+			DEBUG(10,("idx: %i num_users now: %d\n",
+				   i, msrpcs[i]->num_users));
+
+			if (force_close || msrpcs[i]->num_users == 0)
+			{
+				msrpc_use_free(msrpcs[i]);
+				msrpcs[i] = NULL;
+				if (connection_closed != NULL)
+				{
+					*connection_closed = True;
+				}
+			}
+			return True;
+		}
+	}
+
+	return False;
+}
+
+/****************************************************************************
+enumerate client states
+****************************************************************************/
+void msrpc_net_use_enum(uint32 *num_cons, struct use_info ***use)
+{
+	int i;
+
+	*num_cons = 0;
+	*use = NULL;
+
+	for (i = 0; i < num_msrpcs; i++)
+	{
+		struct use_info item;
+
+		ZERO_STRUCT(item);
+
+		if (msrpcs[i] == NULL) continue;
+
+		item.connected = msrpcs[i]->cli != NULL ? True : False;
+	
+		if (item.connected)
+		{
+			item.srv_name  = msrpcs[i]->cli->pipe_name;
+			item.user_name = msrpcs[i]->cli->usr.user_name;
+			item.domain    = msrpcs[i]->cli->usr.domain;
+		}
+
+		add_use_info_to_array(num_cons, use, &item);
+	}
+}
+