#include "includes.h"
extern struct in_addr ipzero;
static void query_name_response( struct subnet_record *subrec,
struct response_record *rrec,
struct packet_struct *p)
{
struct nmb_packet *nmb = &p->packet.nmb;
BOOL success = False;
struct nmb_name *question_name =
&rrec->packet->packet.nmb.question.question_name;
struct in_addr answer_ip = ipzero;
rrec->repeat_count = 0;
if(rrec->num_msgs == 1)
{
if(nmb->header.opcode == NMB_WACK_OPCODE)
{
if( DEBUGLVL( 5 ) )
{
dbgtext( "query_name_response: " );
dbgtext( "WACK from WINS server %s ", inet_ntoa(p->ip) );
dbgtext( "in querying name %s ", nmb_namestr(question_name) );
dbgtext( "on subnet %s.\n", subrec->subnet_name );
}
rrec->repeat_count = 0;
rrec->repeat_time = p->timestamp + nmb->answers->ttl;
rrec->num_msgs--;
return;
}
else if(nmb->header.rcode != 0)
{
success = False;
if( DEBUGLVL( 5 ) )
{
dbgtext( "query_name_response: On subnet %s ", subrec->subnet_name );
dbgtext( "- negative response from IP %s ", inet_ntoa(p->ip) );
dbgtext( "for name %s. ", nmb_namestr(question_name) );
dbgtext( "Error code was %d.\n", nmb->header.rcode );
}
}
else
{
success = True;
putip((char *)&answer_ip,&nmb->answers->rdata[2]);
if( DEBUGLVL( 5 ) )
{
dbgtext( "query_name_response: On subnet %s ", subrec->subnet_name );
dbgtext( "- positive response from IP %s ", inet_ntoa(p->ip) );
dbgtext( "for name %s. ", nmb_namestr(question_name) );
dbgtext( "IP of that name is %s\n", inet_ntoa(answer_ip) );
}
}
}
else if( rrec->num_msgs > 1)
{
if( DEBUGLVL( 0 ) )
{
putip( (char *)&answer_ip, &nmb->answers->rdata[2] );
dbgtext( "query_name_response: " );
dbgtext( "Multiple (%d) responses ", rrec->num_msgs );
dbgtext( "received for a query on subnet %s ", subrec->subnet_name );
dbgtext( "for name %s.\nThis response ", nmb_namestr(question_name) );
dbgtext( "was from IP %s, reporting", inet_ntoa(p->ip) );
dbgtext( "an IP address of %s.\n", inet_ntoa(answer_ip) );
}
return;
}
if(success && rrec->success_fn)
(*(query_name_success_function)rrec->success_fn)(subrec, rrec->userdata, question_name, answer_ip, nmb->answers);
else if( rrec->fail_fn)
(*(query_name_fail_function)rrec->fail_fn)(subrec, rrec, question_name, nmb->header.rcode);
}
static void query_name_timeout_response(struct subnet_record *subrec,
struct response_record *rrec)
{
struct nmb_packet *sent_nmb = &rrec->packet->packet.nmb;
BOOL failed = True;
struct nmb_name *question_name = &sent_nmb->question.question_name;
if(rrec->num_msgs != 0)
{
failed = False;
}
if(failed)
{
if( DEBUGLVL( 5 ) )
{
dbgtext( "query_name_timeout_response: No response to " );
dbgtext( "query for name %s ", nmb_namestr(question_name) );
dbgtext( "on subnet %s.\n", subrec->subnet_name );
}
if(rrec->fail_fn)
(*(query_name_fail_function)rrec->fail_fn)(subrec, rrec, question_name, 0);
}
remove_response_record(subrec, rrec);
}
static BOOL query_local_namelists(struct subnet_record *subrec, struct nmb_name *nmbname,
struct name_record **namerecp)
{
struct name_record *namerec;
*namerecp = NULL;
if(find_name_in_lmhosts(nmbname, namerecp))
return True;
if((namerec = find_name_on_subnet(subrec, nmbname, FIND_ANY_NAME))==NULL)
return False;
if( NAME_IS_ACTIVE(namerec)
&& ( (namerec->data.source == SELF_NAME)
|| (namerec->data.source == LMHOSTS_NAME) ) )
{
*namerecp = namerec;
return True;
}
return False;
}
BOOL query_name(struct subnet_record *subrec, char *name, int type,
query_name_success_function success_fn,
query_name_fail_function fail_fn,
struct userdata_struct *userdata)
{
struct nmb_name nmbname;
struct name_record *namerec;
make_nmb_name(&nmbname, name, type);
if(query_local_namelists(subrec, &nmbname, &namerec) == True)
{
struct res_rec rrec;
int i;
memset((char *)&rrec, '\0', sizeof(struct res_rec));
rrec.rr_name = nmbname;
rrec.rr_type = RR_TYPE_NB;
rrec.rr_class = RR_CLASS_IN;
rrec.ttl = PERMANENT_TTL;
rrec.rdlength = namerec->data.num_ips * 6;
if(rrec.rdlength > MAX_DGRAM_SIZE)
{
if( DEBUGLVL( 0 ) )
{
dbgtext( "query_name: nmbd internal error - " );
dbgtext( "there are %d ip addresses ", namerec->data.num_ips );
dbgtext( "for name %s.\n", nmb_namestr(&nmbname) );
}
return False;
}
for( i = 0; i < namerec->data.num_ips; i++)
{
set_nb_flags( &rrec.rdata[i*6], namerec->data.nb_flags );
putip( &rrec.rdata[(i*6) + 2], (char *)&namerec->data.ip[i]);
}
if(success_fn)
(*(query_name_success_function)success_fn)(subrec, userdata, &nmbname, namerec->data.ip[0], &rrec);
return False;
}
if(queue_query_name( subrec,
query_name_response,
query_name_timeout_response,
success_fn,
fail_fn,
userdata,
&nmbname) == NULL)
{
if( DEBUGLVL( 0 ) )
{
dbgtext( "query_name: Failed to send packet " );
dbgtext( "trying to query name %s\n", nmb_namestr(&nmbname) );
}
return True;
}
return False;
}
BOOL query_name_from_wins_server(struct in_addr ip_to,
char *name, int type,
query_name_success_function success_fn,
query_name_fail_function fail_fn,
struct userdata_struct *userdata)
{
struct nmb_name nmbname;
make_nmb_name(&nmbname, name, type);
if(queue_query_name_from_wins_server( ip_to,
query_name_response,
query_name_timeout_response,
success_fn,
fail_fn,
userdata,
&nmbname) == NULL)
{
if( DEBUGLVL( 0 ) )
{
dbgtext( "query_name_from_wins_server: Failed to send packet " );
dbgtext( "trying to query name %s\n", nmb_namestr(&nmbname) );
}
return True;
}
return False;
}