2005-01-31 07:15:58 +03:00
/*
Unix SMB / CIFS implementation .
answer node status queries
Copyright ( C ) Andrew Tridgell 2005
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-10 06:07:03 +04:00
the Free Software Foundation ; either version 3 of the License , or
2005-01-31 07:15:58 +03:00
( 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
2007-07-10 06:07:03 +04:00
along with this program . If not , see < http : //www.gnu.org/licenses/>.
2005-01-31 07:15:58 +03:00
*/
# include "includes.h"
2008-10-11 23:31:42 +04:00
# include "../lib/util/dlinklist.h"
2005-01-31 07:15:58 +03:00
# include "system/network.h"
# include "nbt_server/nbt_server.h"
2006-01-10 01:12:53 +03:00
# include "lib/socket/socket.h"
2006-03-16 03:23:11 +03:00
# include "librpc/gen_ndr/ndr_nbt.h"
2005-01-31 07:15:58 +03:00
/*
send a name status reply
*/
2005-02-04 04:39:10 +03:00
static void nbtd_node_status_reply ( struct nbt_name_socket * nbtsock ,
struct nbt_name_packet * request_packet ,
2006-01-10 01:12:53 +03:00
struct socket_address * src ,
2005-02-04 04:39:10 +03:00
struct nbt_name * name ,
2005-02-06 11:25:53 +03:00
struct nbtd_interface * iface )
2005-01-31 07:15:58 +03:00
{
struct nbt_name_packet * packet ;
uint32_t name_count ;
2005-02-06 11:25:53 +03:00
struct nbtd_iface_name * iname ;
2005-07-10 12:41:02 +04:00
struct nbtd_server * nbtsrv = iface - > nbtsrv ;
2005-01-31 07:15:58 +03:00
/* work out how many names to send */
name_count = 0 ;
for ( iname = iface - > names ; iname ; iname = iname - > next ) {
if ( ( iname - > nb_flags & NBT_NM_ACTIVE ) & &
strcmp ( iname - > name . name , " * " ) ! = 0 ) {
name_count + + ;
}
}
packet = talloc_zero ( nbtsock , struct nbt_name_packet ) ;
if ( packet = = NULL ) return ;
packet - > name_trn_id = request_packet - > name_trn_id ;
packet - > ancount = 1 ;
packet - > operation = NBT_OPCODE_QUERY | NBT_FLAG_REPLY | NBT_FLAG_AUTHORITIVE ;
packet - > answers = talloc_array ( packet , struct nbt_res_rec , 1 ) ;
if ( packet - > answers = = NULL ) goto failed ;
packet - > answers [ 0 ] . name = * name ;
packet - > answers [ 0 ] . rr_type = NBT_QTYPE_STATUS ;
packet - > answers [ 0 ] . rr_class = NBT_QCLASS_IP ;
packet - > answers [ 0 ] . ttl = 0 ;
packet - > answers [ 0 ] . rdata . status . num_names = name_count ;
packet - > answers [ 0 ] . rdata . status . names = talloc_array ( packet - > answers ,
struct nbt_status_name , name_count ) ;
if ( packet - > answers [ 0 ] . rdata . status . names = = NULL ) goto failed ;
name_count = 0 ;
for ( iname = iface - > names ; iname ; iname = iname - > next ) {
if ( ( iname - > nb_flags & NBT_NM_ACTIVE ) & &
strcmp ( iname - > name . name , " * " ) ! = 0 ) {
struct nbt_status_name * n = & packet - > answers [ 0 ] . rdata . status . names [ name_count ] ;
n - > name = talloc_asprintf ( packet - > answers , " %-15s " , iname - > name . name ) ;
if ( n - > name = = NULL ) goto failed ;
n - > type = iname - > name . type ;
n - > nb_flags = iname - > nb_flags ;
name_count + + ;
}
}
/* we deliberately don't fill in the statistics structure as
it could lead to giving attackers too much information */
ZERO_STRUCT ( packet - > answers [ 0 ] . rdata . status . statistics ) ;
2005-02-12 04:00:15 +03:00
DEBUG ( 7 , ( " Sending node status reply for %s to %s:%d \n " ,
2005-10-14 16:22:15 +04:00
nbt_name_string ( packet , name ) , src - > addr , src - > port ) ) ;
2005-01-31 07:15:58 +03:00
2005-07-10 12:41:02 +04:00
nbtsrv - > stats . total_sent + + ;
2005-10-14 16:22:15 +04:00
nbt_name_reply_send ( nbtsock , src , packet ) ;
2005-01-31 07:15:58 +03:00
failed :
talloc_free ( packet ) ;
}
/*
answer a node status query
*/
2005-02-04 04:39:10 +03:00
void nbtd_query_status ( struct nbt_name_socket * nbtsock ,
struct nbt_name_packet * packet ,
2006-01-10 01:12:53 +03:00
struct socket_address * src )
2005-01-31 07:15:58 +03:00
{
struct nbt_name * name ;
2005-02-06 11:25:53 +03:00
struct nbtd_iface_name * iname ;
2008-09-23 11:02:16 +04:00
struct nbtd_interface * iface = talloc_get_type ( nbtsock - > incoming . private_data ,
2005-02-06 11:25:53 +03:00
struct nbtd_interface ) ;
2005-01-31 07:15:58 +03:00
2005-10-14 18:02:47 +04:00
NBTD_ASSERT_PACKET ( packet , src , packet - > qdcount = = 1 ) ;
NBTD_ASSERT_PACKET ( packet , src , packet - > questions [ 0 ] . question_type = = NBT_QTYPE_STATUS ) ;
NBTD_ASSERT_PACKET ( packet , src , packet - > questions [ 0 ] . question_class = = NBT_QCLASS_IP ) ;
2005-01-31 07:15:58 +03:00
/* see if we have the requested name on this interface */
name = & packet - > questions [ 0 ] . name ;
2005-02-04 04:39:10 +03:00
iname = nbtd_find_iname ( iface , name , NBT_NM_ACTIVE ) ;
2005-01-31 07:15:58 +03:00
if ( iname = = NULL ) {
2005-02-12 04:00:15 +03:00
DEBUG ( 7 , ( " Node status query for %s from %s - not found on %s \n " ,
2005-10-14 16:22:15 +04:00
nbt_name_string ( packet , name ) , src - > addr , iface - > ip_address ) ) ;
2005-01-31 07:15:58 +03:00
return ;
}
2005-10-14 16:22:15 +04:00
nbtd_node_status_reply ( nbtsock , packet , src ,
2005-02-04 04:39:10 +03:00
& iname - > name , iface ) ;
2005-01-31 07:15:58 +03:00
}