diff options
Diffstat (limited to 'source3/nmbd/nmbd_namequery.c')
| -rw-r--r-- | source3/nmbd/nmbd_namequery.c | 234 |
1 files changed, 234 insertions, 0 deletions
diff --git a/source3/nmbd/nmbd_namequery.c b/source3/nmbd/nmbd_namequery.c new file mode 100644 index 00000000000..5d98935fec7 --- /dev/null +++ b/source3/nmbd/nmbd_namequery.c @@ -0,0 +1,234 @@ +/* + Unix SMB/Netbios implementation. + Version 1.9. + NBT netbios routines and daemon - version 2 + Copyright (C) Andrew Tridgell 1994-1997 + Copyright (C) Luke Kenneth Casson Leighton 1994-1997 + Copyright (C) Jeremy Allison 1994-1997 + + 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. + +*/ + +#include "includes.h" + +extern int DEBUGLEVEL; + +extern pstring scope; + +/**************************************************************************** + Deal with a response packet when querying a name. +****************************************************************************/ + +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; + + /* Ensure we don't retry the query but leave the response record cleanup + to the timeout code. We may get more answer responses in which case + we should mark the name in conflict.. */ + rrec->repeat_count = 0; + + if(rrec->num_msgs == 1) + { + /* This is the first response. */ + + if(nmb->header.opcode == NMB_WACK_OPCODE) + { + /* WINS server is telling us to wait. Pretend we didn't get + the response but don't send out any more query requests. */ + + DEBUG(5,("query_name_response: WACK from WINS server %s in querying \ +name %s on subnet %s.\n", inet_ntoa(p->ip), namestr(question_name), subrec->subnet_name)); + + rrec->repeat_count = 0; + /* How long we should wait for. */ + rrec->repeat_time = p->timestamp + nmb->answers->ttl; + rrec->num_msgs--; + return; + } + else if(nmb->header.rcode != 0) + { + success = False; + + DEBUG(5,("query_name_response: On subnet %s - negative response \ +from IP %s for name %s. Error code was %d.\n", subrec->subnet_name, inet_ntoa(p->ip), + namestr(question_name), nmb->header.rcode)); + } + else + { + success = True; + + putip((char *)&answer_ip,&nmb->answers->rdata[2]); + DEBUG(5,("query_name_response: On subnet %s - positive response from IP %s\ +for name %s. IP of that name is %s\n", subrec->subnet_name, inet_ntoa(p->ip), + namestr(question_name), inet_ntoa(answer_ip))); + + /* Interestingly, we could add these names to our namelists, and + change nmbd to a model that checked its own name cache first, + before sending out a query. This is a task for another day, though. + */ + } + } + else if( rrec->num_msgs > 1) + { + DEBUG(0,("query_name_response: Multiple (%d) responses received for a query on \ +subnet %s for name %s. This response was from IP %s\n", + rrec->num_msgs, subrec->subnet_name, namestr(question_name), + inet_ntoa(rrec->packet->ip) )); + + /* We have already called the success or fail function, so we + don't call again here. Leave the response record around in + case we get more responses. */ + + return; + } + + if(success && rrec->success_fn) + (*rrec->success_fn)(subrec, rrec->userdata, question_name, answer_ip, nmb->answers); + else if( rrec->fail_fn) + (*rrec->fail_fn)(subrec, rrec, question_name, nmb->header.rcode); + +} + +/**************************************************************************** + Deal with a timeout when querying a name. +****************************************************************************/ + +static void query_name_timeout_response(struct subnet_record *subrec, + struct response_record *rrec) +{ + struct nmb_packet *sent_nmb = &rrec->packet->packet.nmb; + /* We can only fail here, never succeed. */ + BOOL failed = True; + struct nmb_name *question_name = &sent_nmb->question.question_name; + + if(rrec->num_msgs != 0) + { + /* We got at least one response, and have called the success/fail + function already. */ + + failed = False; + } + + if(failed) + { + DEBUG(5,("query_name_timeout_response: No response to querying name %s on subnet %s.\n", + namestr(question_name), subrec->subnet_name)); + + if(rrec->fail_fn) + (*rrec->fail_fn)(subrec, rrec, question_name, 0); + } + + remove_response_record(subrec, rrec); +} + +/**************************************************************************** + Lookup a name on our local namelists. We check the lmhosts file first. If the + name is not there we look for the name on the given subnet. +****************************************************************************/ + +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->source == SELF_NAME) || + (namerec->source == LMHOSTS_NAME)) ) + { + *namerecp = namerec; + return True; + } + return False; +} + +/**************************************************************************** + Try and query for a name. +****************************************************************************/ + +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, scope); + + /* + * We need to check our local namelists first. + * It may be an magic name, lmhosts name or just + * a name we have registered. + */ + + if(query_local_namelists(subrec, &nmbname, &namerec) == True) + { + struct res_rec rrec; + int i; + + bzero((char *)&rrec, sizeof(struct res_rec)); + + /* Fake up the needed res_rec just in case it's used. */ + rrec.rr_name = nmbname; + rrec.rr_type = RR_TYPE_NB; + rrec.rr_class = RR_CLASS_IN; + rrec.ttl = PERMANENT_TTL; + rrec.rdlength = namerec->num_ips * 6; + if(rrec.rdlength > MAX_DGRAM_SIZE) + { + DEBUG(0,("query_name: nmbd internal error - there are %d ip addresses for name %s.\n", + namerec->num_ips, namestr(&nmbname) )); + return False; + } + + for( i = 0; i < namerec->num_ips; i++) + { + set_nb_flags( &rrec.rdata[i*6], namerec->nb_flags ); + putip( &rrec.rdata[(i*6) + 2], (char *)&namerec->ip[i]); + } + + /* Call the success function directly. */ + if(success_fn) + (*success_fn)(subrec, userdata, &nmbname, namerec->ip[0], &rrec); + return False; + } + + if(queue_query_name( subrec, + query_name_response, + query_name_timeout_response, + success_fn, + fail_fn, + userdata, + &nmbname) == NULL) + { + DEBUG(0,("query_name: Failed to send packet trying to query name %s\n", + namestr(&nmbname))); + return True; + } + return False; +} |