From 8d0660607f2c2d95e1319e04d0d573d9115c4dc0 Mon Sep 17 00:00:00 2001 From: Luke Leighton Date: Mon, 29 Nov 1999 23:56:09 +0000 Subject: this is going to sound _really_ weird, ok, but i had to implement equivalents of NetUseAdd and NetUseDel! (This used to be commit 86f4b1d3cc3887c4bb7bd6433f5f932f7db1b88e) --- source3/rpc_client/cli_use.c | 330 +++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 330 insertions(+) create mode 100644 source3/rpc_client/cli_use.c (limited to 'source3/rpc_client/cli_use.c') diff --git a/source3/rpc_client/cli_use.c b/source3/rpc_client/cli_use.c new file mode 100644 index 0000000000..d9c40139bc --- /dev/null +++ b/source3/rpc_client/cli_use.c @@ -0,0 +1,330 @@ +/* + 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 cli_use +{ + struct cli_state *cli; + uint32 num_users; +}; + +static struct cli_use **clis = NULL; +uint32 num_clis = 0; + +/**************************************************************************** +terminate client connection +****************************************************************************/ +static void cli_use_free(struct cli_use *cli) +{ + cli_ulogoff(cli->cli); + cli_shutdown(cli->cli); + memset(cli->cli, 0, sizeof(*cli->cli)); + free(cli->cli); + + free(cli); +} + +/**************************************************************************** +free a client array +****************************************************************************/ +static void free_cli_array(uint32 num_entries, struct cli_use **entries) +{ + void(*fn)(void*) = (void(*)(void*))&cli_use_free; + free_void_array(num_entries, (void**)entries, *fn); +} + +/**************************************************************************** +add a client state to the array +****************************************************************************/ +static struct cli_use* add_cli_to_array(uint32 *len, + struct cli_use ***array, + struct cli_use *cli) +{ + int i; + for (i = 0; i < num_clis; i++) + { + if (clis[i] == NULL) + { + clis[i] = cli; + return cli; + } + } + + return (struct cli_use*)add_item_to_array(len, + (void***)array, (void*)cli); + +} + +/**************************************************************************** +initiate client array +****************************************************************************/ +void init_client_use(void) +{ + clis = NULL; + num_clis = 0; +} + +/**************************************************************************** +terminate client array +****************************************************************************/ +void free_cli_use(void) +{ + free_cli_array(num_clis, clis); + init_client_use(); +} + +/**************************************************************************** +find client state. server name, user name, domain name and password must all +match. +****************************************************************************/ +static struct cli_use *cli_find(const char* srv_name, + const struct user_credentials *usr_creds) +{ + int i; + const char *sv_name = srv_name; + if (strnequal("\\\\", sv_name, 2)) + { + sv_name = &sv_name[2]; + } + + for (i = 0; i < num_clis; i++) + { + uchar ntpw[16], clintpw[16]; + char *cli_name = NULL; + + if (clis[i] == NULL) continue; + + cli_name = clis[i]->cli->desthost; + if (strnequal("\\\\", cli_name, 2)) + { + cli_name = &cli_name[2]; + } + + if (!strequal(cli_name, sv_name)) continue; + + pwd_get_lm_nt_16(&usr_creds->pwd, NULL, ntpw); + pwd_get_lm_nt_16(&clis[i]->cli->usr.pwd, NULL, clintpw); + + if (strequal(usr_creds->user_name, clis[i]->cli->usr.user_name) && + strequal(usr_creds->domain, clis[i]->cli->usr.domain) && + memcmp(ntpw, clintpw, 16) == 0) + { + return clis[i]; + } + } + + return NULL; +} + +/**************************************************************************** +create a new client state from user credentials +****************************************************************************/ +static struct cli_use *cli_use_get(const char* srv_name, + const struct user_credentials *usr_creds) + +{ + struct cli_use *cli = (struct cli_use*)malloc(sizeof(*cli)); + + if (cli == NULL) + { + return NULL; + } + + memset(cli, 0, sizeof(*cli)); + + cli->cli = cli_initialise(NULL); + + if (cli->cli == NULL) + { + return NULL; + } + + cli->cli->capabilities |= CAP_NT_SMBS | CAP_STATUS32; + cli_init_creds(cli->cli, usr_creds); + + cli->cli->use_ntlmv2 = lp_client_ntlmv2(); + + add_cli_to_array(&num_clis, &clis, cli); + + return cli; +} + +/**************************************************************************** +init client state +****************************************************************************/ +struct cli_state *cli_net_use_addlist(char* servers, + const struct user_credentials *usr_creds) +{ + struct cli_use *cli = cli_find(servers, usr_creds); + + if (cli != NULL) + { + cli->num_users++; + return cli->cli; + } + + /* + * allocate + */ + + cli = cli_use_get(servers, usr_creds); + + if (cli == NULL) + { + return NULL; + } + + if (!cli_connect_serverlist(cli->cli, servers)) + { + DEBUG(0,("cli_use_init: connection failed\n")); + cli_use_free(cli); + return NULL; + } + + cli->cli->ntlmssp_cli_flgs = 0x0; + + cli->num_users++; + + return cli->cli; +} + +/**************************************************************************** +init client state +****************************************************************************/ +struct cli_state *cli_net_use_add(const char* srv_name, + const struct user_credentials *usr_creds) +{ + struct nmb_name calling; + struct nmb_name called; + struct in_addr *dest_ip = NULL; + fstring dest_host; + struct in_addr ip; + + struct cli_use *cli = cli_find(srv_name, usr_creds); + + if (cli != NULL) + { + cli->num_users++; + return cli->cli; + } + + /* + * allocate + */ + + cli = cli_use_get(srv_name, usr_creds); + + if (resolve_srv_name(srv_name, dest_host, &ip)) + { + dest_ip = &ip; + } + else + { + cli_use_free(cli); + return NULL; + } + + make_nmb_name(&called , dns_to_netbios_name(dest_host ), 32, scope); + make_nmb_name(&calling, dns_to_netbios_name(global_myname), 0, scope); + + /* + * connect + */ + + if (!cli_establish_connection(cli->cli, + dest_host, dest_ip, + &calling, &called, + "IPC$", "IPC", + False, True)) + { + DEBUG(0,("cli_use_init: connection failed\n")); + cli_use_free(cli); + return NULL; + } + + cli->cli->ntlmssp_cli_flgs = 0x0; + + cli->num_users++; + + return cli->cli; +} + +/**************************************************************************** +delete a client state +****************************************************************************/ +BOOL cli_net_use_del(const char* srv_name, + const struct user_credentials *usr_creds, + BOOL force_close, + BOOL *connection_closed) +{ + int i; + const char *sv_name = srv_name; + if (strnequal("\\\\", sv_name, 2)) + { + sv_name = &sv_name[2]; + } + + *connection_closed = False; + + for (i = 0; i < num_clis; i++) + { + char *cli_name = NULL; + + if (clis[i] == NULL) continue; + if (clis[i]->cli == NULL) continue; + + cli_name = clis[i]->cli->desthost; + if (strnequal("\\\\", cli_name, 2)) + { + cli_name = &cli_name[2]; + } + + if (!strequal(cli_name, sv_name)) continue; + + if (strequal(usr_creds->user_name, + clis[i]->cli->usr.user_name) && + strequal(usr_creds->domain, + clis[i]->cli->usr.domain)) + { + /* decrement number of users */ + clis[i]->num_users--; + if (force_close || clis[i]->num_users == 0) + { + cli_use_free(clis[i]); + clis[i] = NULL; + *connection_closed = True; + } + return True; + } + } + + return False; +} + -- cgit