1998-03-15 05:37:52 +03:00
/*
2002-01-30 09:08:46 +03:00
Unix SMB / CIFS implementation .
1998-03-15 05:37:52 +03:00
connection claim routines
Copyright ( C ) Andrew Tridgell 1998
2008-10-10 19:50:09 +04:00
1998-03-15 05:37:52 +03:00
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
2007-07-09 23:25:36 +04:00
the Free Software Foundation ; either version 3 of the License , or
1998-03-15 05:37:52 +03:00
( at your option ) any later version .
2008-10-10 19:50:09 +04:00
1998-03-15 05:37:52 +03:00
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 .
2008-10-10 19:50:09 +04:00
1998-03-15 05:37:52 +03:00
You should have received a copy of the GNU General Public License
2007-07-10 04:52:41 +04:00
along with this program . If not , see < http : //www.gnu.org/licenses/>.
1998-03-15 05:37:52 +03:00
*/
# include "includes.h"
2011-03-22 18:57:01 +03:00
# include "smbd/smbd.h"
2010-07-04 20:15:36 +04:00
# include "smbd/globals.h"
2010-08-18 20:59:23 +04:00
# include "dbwrap.h"
2011-03-24 15:46:20 +03:00
# include "auth.h"
1998-03-15 05:37:52 +03:00
1999-12-21 07:54:30 +03:00
/****************************************************************************
2001-05-14 10:15:46 +04:00
Delete a connection record .
1999-12-21 07:54:30 +03:00
* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
2001-05-14 10:15:46 +04:00
2007-10-19 04:40:25 +04:00
bool yield_connection ( connection_struct * conn , const char * name )
1999-12-21 07:54:30 +03:00
{
2007-05-28 15:38:42 +04:00
struct db_record * rec ;
NTSTATUS status ;
1998-03-15 05:37:52 +03:00
1999-12-21 07:54:30 +03:00
DEBUG ( 3 , ( " Yielding connection to %s \n " , name ) ) ;
1998-03-15 05:37:52 +03:00
2010-02-26 19:53:34 +03:00
rec = connections_fetch_entry ( talloc_tos ( ) , conn , name ) ;
if ( rec = = NULL ) {
2007-05-28 15:38:42 +04:00
DEBUG ( 0 , ( " connections_fetch_entry failed \n " ) ) ;
return False ;
}
1999-12-21 07:54:30 +03:00
2007-05-28 15:38:42 +04:00
status = rec - > delete_rec ( rec ) ;
if ( ! NT_STATUS_IS_OK ( status ) ) {
DEBUG ( NT_STATUS_EQUAL ( status , NT_STATUS_NOT_FOUND ) ? 3 : 0 ,
( " deleting connection record returned %s \n " ,
nt_errstr ( status ) ) ) ;
2001-06-09 00:26:46 +04:00
}
2000-01-12 06:09:17 +03:00
2007-05-29 22:14:54 +04:00
TALLOC_FREE ( rec ) ;
2007-05-28 15:38:42 +04:00
return NT_STATUS_IS_OK ( status ) ;
1998-03-15 05:37:52 +03:00
}
2001-05-14 10:15:46 +04:00
struct count_stat {
int curr_connections ;
2006-07-11 22:01:26 +04:00
const char * name ;
2007-10-19 04:40:25 +04:00
bool Clear ;
2001-05-14 10:15:46 +04:00
} ;
/****************************************************************************
Count the entries belonging to a service in the connection db .
* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
2007-05-28 15:38:42 +04:00
static int count_fn ( struct db_record * rec ,
const struct connections_key * ckey ,
const struct connections_data * crec ,
void * udp )
2001-05-14 10:15:46 +04:00
{
struct count_stat * cs = ( struct count_stat * ) udp ;
2008-10-10 19:50:09 +04:00
2007-05-28 15:38:42 +04:00
if ( crec - > cnum = = - 1 ) {
2001-05-14 10:15:46 +04:00
return 0 ;
2007-04-21 01:09:44 +04:00
}
2001-05-14 10:15:46 +04:00
2001-05-17 04:24:34 +04:00
/* If the pid was not found delete the entry from connections.tdb */
2007-05-28 15:38:42 +04:00
if ( cs - > Clear & & ! process_exists ( crec - > pid ) & & ( errno = = ESRCH ) ) {
NTSTATUS status ;
2005-09-30 21:13:37 +04:00
DEBUG ( 2 , ( " pid %s doesn't exist - deleting connections %d [%s] \n " ,
2007-05-28 15:38:42 +04:00
procid_str_static ( & crec - > pid ) , crec - > cnum ,
crec - > servicename ) ) ;
status = rec - > delete_rec ( rec ) ;
if ( ! NT_STATUS_IS_OK ( status ) ) {
DEBUG ( 0 , ( " count_fn: tdb_delete failed with error %s \n " ,
nt_errstr ( status ) ) ) ;
}
2001-05-14 10:15:46 +04:00
return 0 ;
}
2007-10-11 00:34:30 +04:00
if ( strequal ( crec - > servicename , cs - > name ) )
2001-05-14 10:15:46 +04:00
cs - > curr_connections + + ;
return 0 ;
}
1998-03-15 05:37:52 +03:00
1999-12-21 07:54:30 +03:00
/****************************************************************************
2001-05-14 10:15:46 +04:00
Claim an entry in the connections database .
1998-03-15 05:37:52 +03:00
* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
2001-05-14 10:15:46 +04:00
2007-10-19 04:40:25 +04:00
int count_current_connections ( const char * sharename , bool clear )
1998-03-15 05:37:52 +03:00
{
2006-07-11 22:01:26 +04:00
struct count_stat cs ;
2011-06-14 19:50:41 +04:00
int ret ;
2002-09-25 19:19:00 +04:00
2006-07-11 22:01:26 +04:00
cs . curr_connections = 0 ;
cs . name = sharename ;
cs . Clear = clear ;
2001-05-14 10:15:46 +04:00
/*
2006-07-11 22:01:26 +04:00
* This has a race condition , but locking the chain before hand is worse
* as it leads to deadlock .
2001-05-14 10:15:46 +04:00
*/
2011-06-14 19:52:42 +04:00
/*
* become_root ( ) because we might have to open connections . tdb
* via ctdb , which is not possible without root .
*/
become_root ( ) ;
2011-06-14 19:50:41 +04:00
ret = connections_forall ( count_fn , & cs ) ;
2011-06-14 19:52:42 +04:00
unbecome_root ( ) ;
2011-06-20 13:10:31 +04:00
if ( ret < 0 ) {
2007-05-28 15:38:42 +04:00
DEBUG ( 0 , ( " count_current_connections: traverse of "
" connections.tdb failed \n " ) ) ;
2011-06-14 19:50:41 +04:00
return 0 ;
2006-07-11 22:01:26 +04:00
}
2007-04-21 01:09:44 +04:00
2006-07-11 22:01:26 +04:00
return cs . curr_connections ;
}
2001-05-14 10:15:46 +04:00
2006-07-11 22:01:26 +04:00
/****************************************************************************
Claim an entry in the connections database .
* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
2001-05-14 10:15:46 +04:00
2010-07-04 22:45:43 +04:00
bool claim_connection ( connection_struct * conn , const char * name )
2006-07-11 22:01:26 +04:00
{
2007-05-28 15:38:42 +04:00
struct db_record * rec ;
2006-07-11 22:01:26 +04:00
struct connections_data crec ;
2007-05-28 15:38:42 +04:00
TDB_DATA dbuf ;
NTSTATUS status ;
2001-09-19 11:06:34 +04:00
2007-05-28 15:38:42 +04:00
DEBUG ( 5 , ( " claiming [%s] \n " , name ) ) ;
2008-01-06 16:21:25 +03:00
if ( ! ( rec = connections_fetch_entry ( talloc_tos ( ) , conn , name ) ) ) {
2007-05-28 15:38:42 +04:00
DEBUG ( 0 , ( " connections_fetch_entry failed \n " ) ) ;
2007-05-08 17:44:36 +04:00
return False ;
2006-07-11 22:01:26 +04:00
}
1998-03-15 05:37:52 +03:00
/* fill in the crec */
1999-12-21 07:54:30 +03:00
ZERO_STRUCT ( crec ) ;
1998-03-15 05:37:52 +03:00
crec . magic = 0x280267 ;
2010-07-04 20:15:36 +04:00
crec . pid = sconn_server_id ( conn - > sconn ) ;
2010-07-04 20:12:24 +04:00
crec . cnum = conn - > cnum ;
2011-02-21 12:25:52 +03:00
crec . uid = conn - > session_info - > utok . uid ;
crec . gid = conn - > session_info - > utok . gid ;
2010-07-04 20:12:24 +04:00
strlcpy ( crec . servicename , lp_servicename ( SNUM ( conn ) ) ,
sizeof ( crec . servicename ) ) ;
1998-03-15 05:37:52 +03:00
crec . start = time ( NULL ) ;
2008-10-10 19:50:09 +04:00
2007-05-28 15:38:42 +04:00
strlcpy ( crec . machine , get_remote_machine_name ( ) , sizeof ( crec . machine ) ) ;
2010-08-16 11:39:29 +04:00
strlcpy ( crec . addr , conn - > sconn - > client_id . addr , sizeof ( crec . addr ) ) ;
1998-03-15 05:37:52 +03:00
2007-03-29 13:35:51 +04:00
dbuf . dptr = ( uint8 * ) & crec ;
1999-12-21 07:54:30 +03:00
dbuf . dsize = sizeof ( crec ) ;
2007-05-28 15:38:42 +04:00
status = rec - > store ( rec , dbuf , TDB_REPLACE ) ;
TALLOC_FREE ( rec ) ;
if ( ! NT_STATUS_IS_OK ( status ) ) {
2001-05-24 01:33:43 +04:00
DEBUG ( 0 , ( " claim_connection: tdb_store failed with error %s. \n " ,
2007-05-28 15:38:42 +04:00
nt_errstr ( status ) ) ) ;
2001-09-19 11:06:34 +04:00
return False ;
2001-05-24 01:33:43 +04:00
}
1999-12-21 07:54:30 +03:00
2001-09-19 11:06:34 +04:00
return True ;
2001-05-14 22:43:49 +04:00
}