diff options
Diffstat (limited to 'source3/rpc_client')
-rw-r--r-- | source3/rpc_client/cli_connect.c | 190 | ||||
-rw-r--r-- | source3/rpc_client/cli_use.c | 448 | ||||
-rw-r--r-- | source3/rpc_client/ncacn_np_use.c | 173 | ||||
-rw-r--r-- | source3/rpc_client/ncalrpc_l_use.c | 127 |
4 files changed, 938 insertions, 0 deletions
diff --git a/source3/rpc_client/cli_connect.c b/source3/rpc_client/cli_connect.c new file mode 100644 index 0000000000..ded3e50354 --- /dev/null +++ b/source3/rpc_client/cli_connect.c @@ -0,0 +1,190 @@ +/* + Unix SMB/Netbios implementation. + Version 1.9. + SMB client generic functions + Copyright (C) Andrew Tridgell 1994-2000 + Copyright (C) Luke Kenneth Casson Leighton 1996-2000 + + 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 "rpc_parse.h" +#include "rpc_client.h" + +enum +{ MSRPC_NONE, MSRPC_LOCAL, MSRPC_SMB }; + +struct cli_connection +{ + uint32 num_connections; + char *srv_name; + char *pipe_name; + struct user_creds usr_creds; + + int type; + + union + { + struct ncacn_np *smb; + struct msrpc_local *local; + void *cli; + } msrpc; + + cli_auth_fns *auth; + void *auth_info; + void *auth_creds; +}; + +static struct cli_connection **con_list = NULL; +static uint32 num_cons = 0; + +struct user_creds *usr_creds = NULL; +vuser_key *user_key = NULL; + +extern int DEBUGLEVEL; + + +void init_connections(void) +{ + con_list = NULL; + num_cons = 0; + + init_cli_use(); +} + +static void free_con_array(uint32 num_entries, + struct cli_connection **entries) +{ + void (*fn) (void *) = (void (*)(void *))&cli_connection_free; + free_void_array(num_entries, (void **)entries, *fn); +} + + +void free_connections(void) +{ + DEBUG(3, ("free_connections: closing all MSRPC connections\n")); + free_con_array(num_cons, con_list); + free_cli_use(); + + init_connections(); +} + +/**************************************************************************** +terminate client connection +****************************************************************************/ +void cli_connection_free(struct cli_connection *con) +{ + BOOL closed = False; + void *oldcli = NULL; + int i; + + DEBUG(10, ("cli_connection_free: %d\n", __LINE__)); + + if (con->msrpc.cli != NULL) + { + switch (con->type) + { + case MSRPC_LOCAL: + { + DEBUG(10, ("msrpc local connection\n")); + ncalrpc_l_use_del(con->pipe_name, + &con->msrpc.local->nt.key, + False, &closed); + oldcli = con->msrpc.local; + con->msrpc.local = NULL; + break; + } + case MSRPC_SMB: + { + DEBUG(10, ("msrpc smb connection\n")); + ncacn_np_use_del(con->srv_name, + con->pipe_name, + &con->msrpc.smb->smb->key, + False, &closed); + oldcli = con->msrpc.local; + con->msrpc.smb = NULL; + break; + } + } + } + + DEBUG(10, ("cli_connection_free: closed: %s\n", BOOLSTR(closed))); + + if (closed) + { + for (i = 0; i < num_cons; i++) + { + struct cli_connection *c = con_list[i]; + if (c != NULL && con != c && c->msrpc.cli == oldcli) + { + /* WHOOPS! fnum already open: too bad!!! + get rid of all other connections that + were using that connection + */ + switch (c->type) + { + case MSRPC_LOCAL: + { + c->msrpc.local = NULL; + break; + } + case MSRPC_SMB: + { + c->msrpc.smb = NULL; + break; + } + } + } + } + } + + if (con->msrpc.cli != NULL) + { + free(con->msrpc.cli); + } + con->msrpc.cli = NULL; + + if (con->srv_name != NULL) + { + free(con->srv_name); + con->srv_name = NULL; + } + if (con->pipe_name != NULL) + { + free(con->pipe_name); + con->pipe_name = NULL; + } + + if (con->auth_info != NULL) + { + free(con->auth_info); + con->auth_info = NULL; + } + + memset(&con->usr_creds, 0, sizeof(con->usr_creds)); + + for (i = 0; i < num_cons; i++) + { + if (con == con_list[i]) + { + con_list[i] = NULL; + } + } + + free(con); +} diff --git a/source3/rpc_client/cli_use.c b/source3/rpc_client/cli_use.c new file mode 100644 index 0000000000..10997f028f --- /dev/null +++ b/source3/rpc_client/cli_use.c @@ -0,0 +1,448 @@ +/* + 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" +#include "rpc_client.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; +static uint32 num_clis = 0; + +/**************************************************************************** +terminate client connection +****************************************************************************/ +static void cli_use_free(struct cli_use *cli) +{ + if (cli->cli != NULL) + { + if (cli->cli->initialised) + { + cli_ulogoff(cli->cli); + cli_shutdown(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_cli_use(void) +{ + clis = NULL; + num_clis = 0; +} + +/**************************************************************************** +terminate client array +****************************************************************************/ +void free_cli_use(void) +{ + free_cli_array(num_clis, clis); + init_cli_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 ntuser_creds *usr_creds, + BOOL reuse) +{ + int i; + const char *sv_name = srv_name; + struct ntuser_creds null_usr; + + if (usr_creds == NULL) + { + copy_nt_creds(&null_usr, usr_creds); + usr_creds = &null_usr; + } + + if (strnequal("\\\\", sv_name, 2)) + { + sv_name = &sv_name[2]; + } + + DEBUG(10, ("cli_find: %s %s %s reuse: %s\n", + srv_name, usr_creds->user_name, usr_creds->domain, + BOOLSTR(reuse))); + + + for (i = 0; i < num_clis; i++) + { + char *cli_name = NULL; + struct cli_use *c = clis[i]; + + if (c == NULL || !c->cli->initialised || c->cli->fd == -1) + { + continue; + } + + cli_name = c->cli->desthost; + + DEBUG(10, ("cli_find[%d]: %s %s %s\n", + i, cli_name, + c->cli->user_name, c->cli->domain)); + + if (strnequal("\\\\", cli_name, 2)) + { + cli_name = &cli_name[2]; + } + + if (!strequal(cli_name, sv_name)) + { + continue; + } + if (strequal(usr_creds->user_name, "") && + strequal(usr_creds->domain, "") && + pwd_is_nullpwd(&usr_creds->pwd)) + { + return c; + } + if (!strequal(usr_creds->user_name, c->cli->user_name)) + { + continue; + } + if (!reuse && !pwd_compare(&usr_creds->pwd, &c->cli->pwd)) + { + DEBUG(100, ("password doesn't match\n")); + continue; + } + if (usr_creds->domain[0] == 0) + { + return c; + } + if (strequal(usr_creds->domain, c->cli->domain)) + { + return c; + } + } + + return NULL; +} + +/**************************************************************************** +create a new client state from user credentials +****************************************************************************/ +static struct cli_use *cli_use_get(const char *srv_name, + const struct ntuser_creds *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_init_creds(cli->cli, usr_creds); + + return cli; +} + +/**************************************************************************** +init client state +****************************************************************************/ +struct cli_state *cli_net_use_add(const char *srv_name, + const struct ntuser_creds *usr_creds, + BOOL reuse, BOOL *is_new) +{ + 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; + + DEBUG(10, ("cli_net_use_add\n")); + + cli = cli_find(srv_name, usr_creds, reuse); + + if (cli != NULL) + { + cli->num_users++; + DEBUG(10, + ("cli_net_use_add: num_users: %d\n", cli->num_users)); + (*is_new) = False; + return cli->cli; + } + + /* reuse an existing connection requested, and one was not found */ + if (usr_creds != NULL && reuse) + { + return False; + } + + /* + * 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), 0x20); + make_nmb_name(&calling, dns_to_netbios_name(global_myname), 0); + + /* + * connect + */ + + if (!cli_establish_connection(cli->cli, + dest_host, dest_ip, + &calling, &called, + "IPC$", "IPC", False, True)) + { + DEBUG(0, ("cli_net_use_add: connection failed\n")); + cli_use_free(cli); + return NULL; + } + + add_cli_to_array(&num_clis, &clis, cli); + cli->num_users++; + + DEBUG(10, ("cli_net_use_add: num_users: %d\n", cli->num_users)); + + (*is_new) = True; + + return cli->cli; +} + +/**************************************************************************** +delete a client state +****************************************************************************/ +BOOL cli_net_use_del(const char *srv_name, + const struct ntuser_creds *usr_creds, + BOOL force_close, BOOL *connection_closed) +{ + int i; + const char *sv_name = srv_name; + + DEBUG(10, ("cli_net_use_del: %s. %s. %s. force close: %s\n", + srv_name, + usr_creds ? usr_creds->user_name : "", + usr_creds ? usr_creds->domain : "", BOOLSTR(force_close))); + + if (strnequal("\\\\", sv_name, 2)) + { + sv_name = &sv_name[2]; + } + + if (connection_closed != NULL) + { + *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; + + DEBUG(10, ("connection: %s %s %s\n", cli_name, + clis[i]->cli->user_name, + clis[i]->cli->domain)); + + if (strnequal("\\\\", cli_name, 2)) + { + cli_name = &cli_name[2]; + } + + if (!strequal(cli_name, sv_name)) + continue; + + if (strequal(usr_creds ? usr_creds->user_name : "", + clis[i]->cli->user_name) && + strequal(usr_creds ? usr_creds->domain : "", + clis[i]->cli->domain)) + { + /* decrement number of users */ + clis[i]->num_users--; + + DEBUG(10, ("idx: %i num_users now: %d\n", + i, clis[i]->num_users)); + + if (force_close || clis[i]->num_users == 0) + { + cli_use_free(clis[i]); + clis[i] = NULL; + if (connection_closed != NULL) + { + *connection_closed = True; + } + } + return True; + } + } + + return False; +} + +/**************************************************************************** +enumerate client states +****************************************************************************/ +void cli_net_use_enum(uint32 *num_cons, struct use_info ***use) +{ + int i; + + *num_cons = 0; + *use = NULL; + + for (i = 0; i < num_clis; i++) + { + struct use_info item; + + ZERO_STRUCT(item); + + if (clis[i] == NULL) + continue; + + item.connected = clis[i]->cli != NULL ? True : False; + + if (item.connected) + { + item.srv_name = clis[i]->cli->desthost; + item.user_name = clis[i]->cli->user_name; + item.key = clis[i]->cli->key; + item.domain = clis[i]->cli->domain; + } + + add_use_info_to_array(num_cons, use, &item); + } +} + + +/**************************************************************************** +wait for keyboard activity, swallowing network packets on all client states. +****************************************************************************/ +void cli_use_wait_keyboard(void) +{ + fd_set fds; + struct timeval timeout; + + while (1) + { + int i; + int maxfd = fileno(stdin); + FD_ZERO(&fds); + FD_SET(fileno(stdin), &fds); + for (i = 0; i < num_clis; i++) + { + if (clis[i] != NULL && clis[i]->cli != NULL) + { + int fd = clis[i]->cli->fd; + FD_SET(fd, &fds); + maxfd = MAX(fd, maxfd); + } + } + + timeout.tv_sec = 20; + timeout.tv_usec = 0; + sys_select(maxfd + 1, &fds, &timeout); + + if (FD_ISSET(fileno(stdin), &fds)) + return; + + /* We deliberately use receive_smb instead of + client_receive_smb as we want to receive + session keepalives and then drop them here. + */ + for (i = 0; i < num_clis; i++) + { + int fd; + if (clis[i] == NULL || clis[i]->cli == NULL) + continue; + fd = clis[i]->cli->fd; + if (FD_ISSET(fd, &fds)) + receive_smb(fd, clis[i]->cli->inbuf, 0); + } + } +} diff --git a/source3/rpc_client/ncacn_np_use.c b/source3/rpc_client/ncacn_np_use.c new file mode 100644 index 0000000000..e658edbd63 --- /dev/null +++ b/source3/rpc_client/ncacn_np_use.c @@ -0,0 +1,173 @@ +/* + Unix SMB/Netbios implementation. + Version 1.9. + SMB client generic functions + Copyright (C) Andrew Tridgell 1994-2000 + Copyright (C) Luke Kenneth Casson Leighton 1996-2000 + + 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 "rpc_parse.h" +#include "rpc_client.h" +#include "trans2.h" + +extern int DEBUGLEVEL; +extern pstring global_myname; + +struct ncacn_np_use +{ + struct ncacn_np *cli; + uint32 num_users; +}; + +static struct ncacn_np_use **msrpcs = NULL; +static uint32 num_msrpcs = 0; + +/**************************************************************************** +terminate client connection +****************************************************************************/ +static void ncacn_np_shutdown(struct ncacn_np *cli) +{ + if (cli != NULL) + { + if (cli->smb != NULL) + { + if (cli->smb->initialised) + { + /* cli_nt_session_close(cli->smb, cli->fnum); JERRY */ + cli_nt_session_close(cli->smb); + } +#if 0 /* commented out by JERRY */ + cli_net_use_del(cli->smb->desthost, + &cli->smb->usr, False, False); +#endif + } + } +} + + +/**************************************************************************** +terminate client connection +****************************************************************************/ +static void ncacn_np_use_free(struct ncacn_np_use *cli) +{ + if (cli->cli != NULL) + { + if (cli->cli->initialised) + { + ncacn_np_shutdown(cli->cli); + } + ZERO_STRUCTP(cli->cli); + free(cli->cli); + } + ZERO_STRUCTP(cli); + free(cli); +} + + +/**************************************************************************** +delete a client state +****************************************************************************/ +BOOL ncacn_np_use_del(const char *srv_name, const char *pipe_name, + const vuser_key * key, + BOOL force_close, BOOL *connection_closed) +{ + int i; + DEBUG(10, ("ncacn_np_net_use_del: %s. force close: %s ", + pipe_name, BOOLSTR(force_close))); + if (key != NULL) + { + DEBUG(10, ("[%d,%x]", key->pid, key->vuid)); + } + DEBUG(10, ("\n")); + + if (connection_closed != NULL) + { + *connection_closed = False; + } + + if (strnequal("\\PIPE\\", pipe_name, 6)) + { + pipe_name = &pipe_name[6]; + } + + if (strnequal("\\\\", srv_name, 2)) + { + srv_name = &srv_name[6]; + } + + for (i = 0; i < num_msrpcs; i++) + { + char *ncacn_np_name = NULL; + char *ncacn_np_srv_name = NULL; + struct ncacn_np_use *c = msrpcs[i]; + vuser_key k; + + if (c == NULL || c->cli == NULL || c->cli->smb == NULL) + continue; + + ncacn_np_name = c->cli->pipe_name; + ncacn_np_srv_name = c->cli->smb->desthost; + + k = c->cli->smb->key; + + DEBUG(10, ("use_del[%d]: %s %s %s %s [%d,%x]\n", + i, ncacn_np_name, ncacn_np_srv_name, + c->cli->smb->user_name, + c->cli->smb->domain, k.pid, k.vuid)); + + if (strnequal("\\PIPE\\", ncacn_np_name, 6)) + { + ncacn_np_name = &ncacn_np_name[6]; + } + if (!strequal(ncacn_np_srv_name, srv_name)) + { + continue; + } + if (strnequal("\\\\", ncacn_np_srv_name, 2)) + { + ncacn_np_srv_name = &ncacn_np_srv_name[6]; + } + if (!strequal(ncacn_np_name, pipe_name)) + { + continue; + } + if (key->pid != k.pid || key->vuid != k.vuid) + { + continue; + } + /* decrement number of users */ + c->num_users--; + DEBUG(10, ("idx: %i num_users now: %d\n", + i, c->num_users)); + if (force_close || c->num_users == 0) + { + ncacn_np_use_free(c); + msrpcs[i] = NULL; + if (connection_closed != NULL) + { + *connection_closed = True; + } + } + return True; + } + + return False; +} + diff --git a/source3/rpc_client/ncalrpc_l_use.c b/source3/rpc_client/ncalrpc_l_use.c new file mode 100644 index 0000000000..81ade8e1a6 --- /dev/null +++ b/source3/rpc_client/ncalrpc_l_use.c @@ -0,0 +1,127 @@ +/* + 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 "rpc_parse.h" + +extern int DEBUGLEVEL; + +struct ncalrpc_use +{ + struct msrpc_local *cli; + uint32 num_users; +}; + +static struct ncalrpc_use **clis = NULL; +static uint32 num_clis = 0; + +/**************************************************************************** +terminate client connection +****************************************************************************/ +static void ncalrpc_use_free(struct ncalrpc_use *cli) +{ + if (cli->cli != NULL) + { + if (cli->cli->initialised) + { + ncalrpc_l_shutdown(cli->cli); + } + free(cli->cli); + } + + free(cli); +} + +/**************************************************************************** +delete a client state +****************************************************************************/ +BOOL ncalrpc_l_use_del(const char *pipe_name, + const vuser_key * key, + BOOL force_close, BOOL *connection_closed) +{ + int i; + + if (strnequal("\\PIPE\\", pipe_name, 6)) + { + pipe_name = &pipe_name[6]; + } + + DEBUG(10, ("ncalrpc_l_use_del: %s. [%d,%x] force close: %s\n", + pipe_name, key->pid, key->vuid, BOOLSTR(force_close))); + + if (connection_closed != NULL) + { + *connection_closed = False; + } + + for (i = 0; i < num_clis; i++) + { + char *ncalrpc_name = NULL; + + if (clis[i] == NULL) + continue; + if (clis[i]->cli == NULL) + continue; + + ncalrpc_name = clis[i]->cli->pipe_name; + + if (strnequal("\\PIPE\\", pipe_name, 6)) + { + ncalrpc_name = &ncalrpc_name[6]; + } + + DEBUG(10, ("connection: %s [%d,%x]", ncalrpc_name, + clis[i]->cli->nt.key.pid, + clis[i]->cli->nt.key.vuid)); + + if (!strequal(ncalrpc_name, pipe_name)) + continue; + + if (key->pid != clis[i]->cli->nt.key.pid || + key->vuid != clis[i]->cli->nt.key.vuid) + { + continue; + } + /* decrement number of users */ + clis[i]->num_users--; + + DEBUG(10, ("idx: %i num_users now: %d\n", + i, clis[i]->num_users)); + + if (force_close || clis[i]->num_users == 0) + { + ncalrpc_use_free(clis[i]); + clis[i] = NULL; + if (connection_closed != NULL) + { + *connection_closed = True; + } + } + return True; + } + + return False; +} + + |