nmbd_nameregister.c [plain text]
#include "includes.h"
extern fstring global_myworkgroup;
static void register_name_response(struct subnet_record *subrec,
struct response_record *rrec, struct packet_struct *p)
{
struct nmb_packet *nmb = &p->packet.nmb;
BOOL bcast = nmb->header.nm_flags.bcast;
BOOL success = True;
struct nmb_name *question_name = &rrec->packet->packet.nmb.question.question_name;
struct nmb_name *answer_name = &nmb->answers->rr_name;
int ttl = 0;
uint16 nb_flags = 0;
struct in_addr registered_ip;
if(!question_name || !answer_name)
{
DEBUG(0,("register_name_response: malformed response (%s is NULL).\n",
question_name ? "question_name" : "answer_name" ));
return;
}
if(!nmb_name_equal(question_name, answer_name))
{
DEBUG(0,("register_name_response: Answer name %s differs from question \
name %s.\n", nmb_namestr(answer_name), nmb_namestr(question_name)));
return;
}
if(bcast)
{
#if 1
if((nmb->header.rcode == ACT_ERR) && strequal(global_myworkgroup, answer_name->name) &&
(answer_name->name_type == 0x1b))
{
rrec->num_msgs--;
DEBUG(5,("register_name_response: Ignoring broadcast response to \
registration of name %s due to old Samba server bug.\n", nmb_namestr(answer_name)));
return;
}
#endif
DEBUG(1,("register_name_response: Failed to register \
name %s on subnet %s via broadcast. Error code was %d. Reject came from IP %s\n",
nmb_namestr(answer_name),
subrec->subnet_name, nmb->header.rcode, inet_ntoa(p->ip)));
success = False;
}
else
{
if(nmb->header.rcode != 0)
{
success = False;
DEBUG(0,("register_name_response: server at IP %s rejected our \
name registration of %s with error code %d.\n", inet_ntoa(p->ip),
nmb_namestr(answer_name), nmb->header.rcode));
}
else if(nmb->header.opcode == NMB_WACK_OPCODE)
{
DEBUG(5,("register_name_response: WACK from WINS server %s in registering \
name %s on subnet %s.\n", inet_ntoa(p->ip), nmb_namestr(answer_name), subrec->subnet_name));
rrec->repeat_count = 0;
rrec->repeat_time = p->timestamp + nmb->answers->ttl;
rrec->num_msgs--;
return;
}
else
{
success = True;
nb_flags = get_nb_flags(nmb->answers->rdata);
putip((char*)®istered_ip,&nmb->answers->rdata[2]);
ttl = nmb->answers->ttl;
}
}
DEBUG(5,("register_name_response: %s in registering name %s on subnet %s.\n",
success ? "success" : "failure", nmb_namestr(answer_name), subrec->subnet_name));
if(success)
{
standard_success_register(subrec, rrec->userdata, answer_name, nb_flags, ttl, registered_ip);
if( rrec->success_fn)
(*(register_name_success_function)rrec->success_fn)(subrec, rrec->userdata, answer_name, nb_flags, ttl, registered_ip);
}
else
{
if( rrec->fail_fn)
(*(register_name_fail_function)rrec->fail_fn)(subrec, rrec, question_name);
standard_fail_register( subrec, rrec, question_name);
}
remove_response_record(subrec, rrec);
}
static void register_name_timeout_response(struct subnet_record *subrec,
struct response_record *rrec)
{
struct nmb_packet *sent_nmb = &rrec->packet->packet.nmb;
BOOL bcast = sent_nmb->header.nm_flags.bcast;
BOOL success = False;
struct nmb_name *question_name = &sent_nmb->question.question_name;
uint16 nb_flags = 0;
int ttl = 0;
struct in_addr registered_ip;
if(bcast)
{
if(rrec->num_msgs == 0)
{
success = True;
nb_flags = get_nb_flags(sent_nmb->additional->rdata);
ttl = sent_nmb->additional->ttl;
putip(®istered_ip,&sent_nmb->additional->rdata[2]);
}
}
else
{
if(rrec->num_msgs == 0)
{
DEBUG(2,("register_name_timeout_response: WINS server at address %s is not \
responding.\n", inet_ntoa(rrec->packet->ip)));
rrec->repeat_count = 3;
rrec->repeat_interval *= 2;
if(rrec->repeat_interval > (5 * 60))
rrec->repeat_interval = (5 * 60);
rrec->repeat_time = time(NULL) + rrec->repeat_interval;
rrec->in_expiration_processing = False;
DEBUG(5,("register_name_timeout_response: increasing WINS timeout to %d seconds.\n",
(int)rrec->repeat_interval));
return;
}
}
DEBUG(5,("register_name_timeout_response: %s in registering name %s on subnet %s.\n",
success ? "success" : "failure", nmb_namestr(question_name), subrec->subnet_name));
if(success)
{
standard_success_register(subrec, rrec->userdata, question_name, nb_flags, ttl, registered_ip);
if( rrec->success_fn)
(*(register_name_success_function)rrec->success_fn)(subrec, rrec->userdata, question_name, nb_flags, ttl, registered_ip);
}
else
{
if( rrec->fail_fn)
(*(register_name_fail_function)rrec->fail_fn)(subrec, rrec, question_name);
standard_fail_register( subrec, rrec, question_name);
}
remove_response_record(subrec, rrec);
}
static BOOL multihomed_register_name( struct nmb_name *nmbname, uint16 nb_flags,
register_name_success_function success_fn,
register_name_fail_function fail_fn,
struct userdata_struct *userdata)
{
int num_ips=0;
int i;
struct in_addr *ip_list = NULL;
struct subnet_record *subrec;
for(subrec = FIRST_SUBNET; subrec; subrec = NEXT_SUBNET_EXCLUDING_UNICAST(subrec) )
num_ips++;
if((ip_list = (struct in_addr *)malloc(num_ips * sizeof(struct in_addr)))==NULL)
{
DEBUG(0,("multihomed_register_name: malloc fail !\n"));
return True;
}
for( subrec = FIRST_SUBNET, i = 0;
subrec;
subrec = NEXT_SUBNET_EXCLUDING_UNICAST(subrec), i++ )
ip_list[i] = subrec->myip;
(void)add_name_to_subnet( unicast_subnet, nmbname->name, nmbname->name_type,
nb_flags, lp_max_ttl(), SELF_NAME,
num_ips, ip_list);
for( i = 0; i < num_ips; i++)
{
if(queue_register_multihomed_name( unicast_subnet,
register_name_response,
register_name_timeout_response,
(i == num_ips - 1) ? success_fn : NULL,
(i == num_ips - 1) ? fail_fn : NULL,
(i == num_ips - 1) ? userdata : NULL,
nmbname,
nb_flags,
ip_list[i]) == NULL)
{
DEBUG(0,("multihomed_register_name: Failed to send packet trying to \
register name %s IP %s\n", nmb_namestr(nmbname), inet_ntoa(ip_list[i]) ));
SAFE_FREE(ip_list);
return True;
}
}
SAFE_FREE(ip_list);
return False;
}
BOOL register_name(struct subnet_record *subrec,
char *name, int type, uint16 nb_flags,
register_name_success_function success_fn,
register_name_fail_function fail_fn,
struct userdata_struct *userdata)
{
struct nmb_name nmbname;
make_nmb_name(&nmbname, name, type);
nb_flags |= NB_ACTIVE;
if( (subrec == unicast_subnet) && we_are_multihomed())
return multihomed_register_name(&nmbname, nb_flags,
success_fn, fail_fn,
userdata);
if(queue_register_name( subrec,
register_name_response,
register_name_timeout_response,
success_fn,
fail_fn,
userdata,
&nmbname,
nb_flags) == NULL)
{
DEBUG(0,("register_name: Failed to send packet trying to register name %s\n",
nmb_namestr(&nmbname)));
return True;
}
return False;
}
BOOL refresh_name(struct subnet_record *subrec, struct name_record *namerec,
refresh_name_success_function success_fn,
refresh_name_fail_function fail_fn,
struct userdata_struct *userdata)
{
int i;
for( i = 0; i < namerec->data.num_ips; i++)
{
if(queue_refresh_name( subrec,
register_name_response,
register_name_timeout_response,
(i == (namerec->data.num_ips - 1)) ? success_fn : NULL,
(i == (namerec->data.num_ips - 1)) ? fail_fn : NULL,
(i == (namerec->data.num_ips - 1)) ? userdata : NULL,
namerec,
namerec->data.ip[i]) == NULL)
{
DEBUG(0,("refresh_name: Failed to send packet trying to refresh name %s\n",
nmb_namestr(&namerec->name)));
return True;
}
}
return False;
}