mirror of
https://github.com/samba-team/samba.git
synced 2024-12-23 17:34:34 +03:00
c2f3ed48c5
Add a simple "processed packet queue" cache to stop nmbd responding to packets received on the broadcast and non-broadcast socket (which it has opened when "nmbd bind explicit broadcast = yes"). This is a very simple packet queue - it only keeps the packets processed during a single call to listen_for_packets() (i.e. one select call). This means that if the delivery notification for a packet received on both broadcast and non-broadcast addresses is done in two different select calls, the packet will still be processed twice. This is a very rare occurrance and we can just live with it when it does as the protocol is stateless. If this is ever flagged as a repeatable problem then we can add a longer lived cache, using timeout processing to clear etc. etc. But without storing all packets processed we can never be *sure* we've eliminated the race condition so I'm going to go with this simple solution until someone proves a more complex one is needed :-). Jeremy.
2120 lines
65 KiB
C
2120 lines
65 KiB
C
/*
|
|
Unix SMB/CIFS implementation.
|
|
NBT netbios routines and daemon - version 2
|
|
Copyright (C) Andrew Tridgell 1994-1998
|
|
Copyright (C) Luke Kenneth Casson Leighton 1994-1998
|
|
Copyright (C) Jeremy Allison 1994-2003
|
|
|
|
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 3 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, see <http://www.gnu.org/licenses/>.
|
|
|
|
*/
|
|
|
|
#include "includes.h"
|
|
|
|
extern int ClientNMB;
|
|
extern int ClientDGRAM;
|
|
extern int global_nmb_port;
|
|
|
|
extern int num_response_packets;
|
|
|
|
bool rescan_listen_set = False;
|
|
|
|
|
|
/*******************************************************************
|
|
The global packet linked-list. Incoming entries are
|
|
added to the end of this list. It is supposed to remain fairly
|
|
short so we won't bother with an end pointer.
|
|
******************************************************************/
|
|
|
|
static struct packet_struct *packet_queue = NULL;
|
|
|
|
/***************************************************************************
|
|
Utility function to find the specific fd to send a packet out on.
|
|
**************************************************************************/
|
|
|
|
static int find_subnet_fd_for_address( struct in_addr local_ip )
|
|
{
|
|
struct subnet_record *subrec;
|
|
|
|
for( subrec = FIRST_SUBNET; subrec; subrec = NEXT_SUBNET_EXCLUDING_UNICAST(subrec))
|
|
if(ip_equal_v4(local_ip, subrec->myip))
|
|
return subrec->nmb_sock;
|
|
|
|
return ClientNMB;
|
|
}
|
|
|
|
/***************************************************************************
|
|
Utility function to find the specific fd to send a mailslot packet out on.
|
|
**************************************************************************/
|
|
|
|
static int find_subnet_mailslot_fd_for_address( struct in_addr local_ip )
|
|
{
|
|
struct subnet_record *subrec;
|
|
|
|
for( subrec = FIRST_SUBNET; subrec; subrec = NEXT_SUBNET_EXCLUDING_UNICAST(subrec))
|
|
if(ip_equal_v4(local_ip, subrec->myip))
|
|
return subrec->dgram_sock;
|
|
|
|
return ClientDGRAM;
|
|
}
|
|
|
|
/***************************************************************************
|
|
Get/Set problematic nb_flags as network byte order 16 bit int.
|
|
**************************************************************************/
|
|
|
|
uint16 get_nb_flags(char *buf)
|
|
{
|
|
return ((((uint16)*buf)&0xFFFF) & NB_FLGMSK);
|
|
}
|
|
|
|
void set_nb_flags(char *buf, uint16 nb_flags)
|
|
{
|
|
*buf++ = ((nb_flags & NB_FLGMSK) & 0xFF);
|
|
*buf = '\0';
|
|
}
|
|
|
|
/***************************************************************************
|
|
Dumps out the browse packet data.
|
|
**************************************************************************/
|
|
|
|
static void debug_browse_data(char *outbuf, int len)
|
|
{
|
|
int i,j;
|
|
|
|
DEBUG( 4, ( "debug_browse_data():\n" ) );
|
|
for (i = 0; i < len; i+= 16) {
|
|
DEBUGADD( 4, ( "%3x char ", i ) );
|
|
|
|
for (j = 0; j < 16; j++) {
|
|
unsigned char x;
|
|
if (i+j >= len)
|
|
break;
|
|
|
|
x = outbuf[i+j];
|
|
if (x < 32 || x > 127)
|
|
x = '.';
|
|
|
|
DEBUGADD( 4, ( "%c", x ) );
|
|
}
|
|
|
|
DEBUGADD( 4, ( "%*s hex", 16-j, "" ) );
|
|
|
|
for (j = 0; j < 16; j++) {
|
|
if (i+j >= len)
|
|
break;
|
|
DEBUGADD( 4, ( " %02x", (unsigned char)outbuf[i+j] ) );
|
|
}
|
|
|
|
DEBUGADD( 4, ("\n") );
|
|
}
|
|
}
|
|
|
|
/***************************************************************************
|
|
Generates the unique transaction identifier
|
|
**************************************************************************/
|
|
|
|
static uint16 name_trn_id=0;
|
|
|
|
static uint16 generate_name_trn_id(void)
|
|
{
|
|
if (!name_trn_id) {
|
|
name_trn_id = ((unsigned)time(NULL)%(unsigned)0x7FFF) + ((unsigned)sys_getpid()%(unsigned)100);
|
|
}
|
|
name_trn_id = (name_trn_id+1) % (unsigned)0x7FFF;
|
|
return name_trn_id;
|
|
}
|
|
|
|
/***************************************************************************
|
|
Either loops back or sends out a completed NetBIOS packet.
|
|
**************************************************************************/
|
|
|
|
static bool send_netbios_packet(struct packet_struct *p)
|
|
{
|
|
bool loopback_this_packet = False;
|
|
|
|
/* Check if we are sending to or from ourselves as a WINS server. */
|
|
if(ismyip_v4(p->ip) && (p->port == global_nmb_port))
|
|
loopback_this_packet = True;
|
|
|
|
if(loopback_this_packet) {
|
|
struct packet_struct *lo_packet = NULL;
|
|
DEBUG(5,("send_netbios_packet: sending packet to ourselves.\n"));
|
|
if((lo_packet = copy_packet(p)) == NULL)
|
|
return False;
|
|
queue_packet(lo_packet);
|
|
} else if (!send_packet(p)) {
|
|
DEBUG(0,("send_netbios_packet: send_packet() to IP %s port %d failed\n",
|
|
inet_ntoa(p->ip),p->port));
|
|
return False;
|
|
}
|
|
|
|
return True;
|
|
}
|
|
|
|
/***************************************************************************
|
|
Sets up the common elements of an outgoing NetBIOS packet.
|
|
|
|
Note: do not attempt to rationalise whether rec_des should be set or not
|
|
in a particular situation. Just follow rfc_1002 or look at examples from WinXX.
|
|
It does NOT follow the rule that requests to the wins server always have
|
|
rec_des true. See for example name releases and refreshes
|
|
**************************************************************************/
|
|
|
|
static struct packet_struct *create_and_init_netbios_packet(struct nmb_name *nmbname,
|
|
bool bcast, bool rec_des,
|
|
struct in_addr to_ip)
|
|
{
|
|
struct packet_struct *packet = NULL;
|
|
struct nmb_packet *nmb = NULL;
|
|
|
|
/* Allocate the packet_struct we will return. */
|
|
if((packet = SMB_MALLOC_P(struct packet_struct)) == NULL) {
|
|
DEBUG(0,("create_and_init_netbios_packet: malloc fail (1) for packet struct.\n"));
|
|
return NULL;
|
|
}
|
|
|
|
memset((char *)packet,'\0',sizeof(*packet));
|
|
|
|
nmb = &packet->packet.nmb;
|
|
|
|
nmb->header.name_trn_id = generate_name_trn_id();
|
|
nmb->header.response = False;
|
|
nmb->header.nm_flags.recursion_desired = rec_des;
|
|
nmb->header.nm_flags.recursion_available = False;
|
|
nmb->header.nm_flags.trunc = False;
|
|
nmb->header.nm_flags.authoritative = False;
|
|
nmb->header.nm_flags.bcast = bcast;
|
|
|
|
nmb->header.rcode = 0;
|
|
nmb->header.qdcount = 1;
|
|
nmb->header.ancount = 0;
|
|
nmb->header.nscount = 0;
|
|
|
|
nmb->question.question_name = *nmbname;
|
|
nmb->question.question_type = QUESTION_TYPE_NB_QUERY;
|
|
nmb->question.question_class = QUESTION_CLASS_IN;
|
|
|
|
packet->ip = to_ip;
|
|
packet->port = NMB_PORT;
|
|
packet->recv_fd = -1;
|
|
packet->send_fd = ClientNMB;
|
|
packet->timestamp = time(NULL);
|
|
packet->packet_type = NMB_PACKET;
|
|
packet->locked = False;
|
|
|
|
return packet; /* Caller must free. */
|
|
}
|
|
|
|
/***************************************************************************
|
|
Sets up the common elements of register, refresh or release packet.
|
|
**************************************************************************/
|
|
|
|
static bool create_and_init_additional_record(struct packet_struct *packet,
|
|
uint16 nb_flags,
|
|
const struct in_addr *register_ip)
|
|
{
|
|
struct nmb_packet *nmb = &packet->packet.nmb;
|
|
|
|
if((nmb->additional = SMB_MALLOC_P(struct res_rec)) == NULL) {
|
|
DEBUG(0,("create_and_init_additional_record: malloc fail for additional record.\n"));
|
|
return False;
|
|
}
|
|
|
|
memset((char *)nmb->additional,'\0',sizeof(struct res_rec));
|
|
|
|
nmb->additional->rr_name = nmb->question.question_name;
|
|
nmb->additional->rr_type = RR_TYPE_NB;
|
|
nmb->additional->rr_class = RR_CLASS_IN;
|
|
|
|
/* See RFC 1002, sections 5.1.1.1, 5.1.1.2 and 5.1.1.3 */
|
|
if (nmb->header.nm_flags.bcast)
|
|
nmb->additional->ttl = PERMANENT_TTL;
|
|
else
|
|
nmb->additional->ttl = lp_max_ttl();
|
|
|
|
nmb->additional->rdlength = 6;
|
|
|
|
set_nb_flags(nmb->additional->rdata,nb_flags);
|
|
|
|
/* Set the address for the name we are registering. */
|
|
putip(&nmb->additional->rdata[2], register_ip);
|
|
|
|
/*
|
|
it turns out that Jeremys code was correct, we are supposed
|
|
to send registrations from the IP we are registering. The
|
|
trick is what to do on timeouts! When we send on a
|
|
non-routable IP then the reply will timeout, and we should
|
|
treat this as success, not failure. That means we go into
|
|
our standard refresh cycle for that name which copes nicely
|
|
with disconnected networks.
|
|
*/
|
|
packet->recv_fd = -1;
|
|
packet->send_fd = find_subnet_fd_for_address(*register_ip);
|
|
|
|
return True;
|
|
}
|
|
|
|
/***************************************************************************
|
|
Sends out a name query.
|
|
**************************************************************************/
|
|
|
|
static bool initiate_name_query_packet( struct packet_struct *packet)
|
|
{
|
|
struct nmb_packet *nmb = NULL;
|
|
|
|
nmb = &packet->packet.nmb;
|
|
|
|
nmb->header.opcode = NMB_NAME_QUERY_OPCODE;
|
|
nmb->header.arcount = 0;
|
|
|
|
nmb->header.nm_flags.recursion_desired = True;
|
|
|
|
DEBUG(4,("initiate_name_query_packet: sending query for name %s (bcast=%s) to IP %s\n",
|
|
nmb_namestr(&nmb->question.question_name),
|
|
BOOLSTR(nmb->header.nm_flags.bcast), inet_ntoa(packet->ip)));
|
|
|
|
return send_netbios_packet( packet );
|
|
}
|
|
|
|
/***************************************************************************
|
|
Sends out a name query - from a WINS server.
|
|
**************************************************************************/
|
|
|
|
static bool initiate_name_query_packet_from_wins_server( struct packet_struct *packet)
|
|
{
|
|
struct nmb_packet *nmb = NULL;
|
|
|
|
nmb = &packet->packet.nmb;
|
|
|
|
nmb->header.opcode = NMB_NAME_QUERY_OPCODE;
|
|
nmb->header.arcount = 0;
|
|
|
|
nmb->header.nm_flags.recursion_desired = False;
|
|
|
|
DEBUG(4,("initiate_name_query_packet_from_wins_server: sending query for name %s (bcast=%s) to IP %s\n",
|
|
nmb_namestr(&nmb->question.question_name),
|
|
BOOLSTR(nmb->header.nm_flags.bcast), inet_ntoa(packet->ip)));
|
|
|
|
return send_netbios_packet( packet );
|
|
}
|
|
|
|
/***************************************************************************
|
|
Sends out a name register.
|
|
**************************************************************************/
|
|
|
|
static bool initiate_name_register_packet( struct packet_struct *packet,
|
|
uint16 nb_flags, const struct in_addr *register_ip)
|
|
{
|
|
struct nmb_packet *nmb = &packet->packet.nmb;
|
|
|
|
nmb->header.opcode = NMB_NAME_REG_OPCODE;
|
|
nmb->header.arcount = 1;
|
|
|
|
nmb->header.nm_flags.recursion_desired = True;
|
|
|
|
if(create_and_init_additional_record(packet, nb_flags, register_ip) == False)
|
|
return False;
|
|
|
|
DEBUG(4,("initiate_name_register_packet: sending registration for name %s (bcast=%s) to IP %s\n",
|
|
nmb_namestr(&nmb->additional->rr_name),
|
|
BOOLSTR(nmb->header.nm_flags.bcast), inet_ntoa(packet->ip)));
|
|
|
|
return send_netbios_packet( packet );
|
|
}
|
|
|
|
/***************************************************************************
|
|
Sends out a multihomed name register.
|
|
**************************************************************************/
|
|
|
|
static bool initiate_multihomed_name_register_packet(struct packet_struct *packet,
|
|
uint16 nb_flags, struct in_addr *register_ip)
|
|
{
|
|
struct nmb_packet *nmb = &packet->packet.nmb;
|
|
fstring second_ip_buf;
|
|
|
|
fstrcpy(second_ip_buf, inet_ntoa(packet->ip));
|
|
|
|
nmb->header.opcode = NMB_NAME_MULTIHOMED_REG_OPCODE;
|
|
nmb->header.arcount = 1;
|
|
|
|
nmb->header.nm_flags.recursion_desired = True;
|
|
|
|
if(create_and_init_additional_record(packet, nb_flags, register_ip) == False)
|
|
return False;
|
|
|
|
DEBUG(4,("initiate_multihomed_name_register_packet: sending registration \
|
|
for name %s IP %s (bcast=%s) to IP %s\n",
|
|
nmb_namestr(&nmb->additional->rr_name), inet_ntoa(*register_ip),
|
|
BOOLSTR(nmb->header.nm_flags.bcast), second_ip_buf ));
|
|
|
|
return send_netbios_packet( packet );
|
|
}
|
|
|
|
/***************************************************************************
|
|
Sends out a name refresh.
|
|
**************************************************************************/
|
|
|
|
static bool initiate_name_refresh_packet( struct packet_struct *packet,
|
|
uint16 nb_flags, struct in_addr *refresh_ip)
|
|
{
|
|
struct nmb_packet *nmb = &packet->packet.nmb;
|
|
|
|
nmb->header.opcode = NMB_NAME_REFRESH_OPCODE_8;
|
|
nmb->header.arcount = 1;
|
|
|
|
nmb->header.nm_flags.recursion_desired = False;
|
|
|
|
if(create_and_init_additional_record(packet, nb_flags, refresh_ip) == False)
|
|
return False;
|
|
|
|
DEBUG(4,("initiate_name_refresh_packet: sending refresh for name %s (bcast=%s) to IP %s\n",
|
|
nmb_namestr(&nmb->additional->rr_name),
|
|
BOOLSTR(nmb->header.nm_flags.bcast), inet_ntoa(packet->ip)));
|
|
|
|
return send_netbios_packet( packet );
|
|
}
|
|
|
|
/***************************************************************************
|
|
Sends out a name release.
|
|
**************************************************************************/
|
|
|
|
static bool initiate_name_release_packet( struct packet_struct *packet,
|
|
uint16 nb_flags, struct in_addr *release_ip)
|
|
{
|
|
struct nmb_packet *nmb = &packet->packet.nmb;
|
|
|
|
nmb->header.opcode = NMB_NAME_RELEASE_OPCODE;
|
|
nmb->header.arcount = 1;
|
|
|
|
nmb->header.nm_flags.recursion_desired = False;
|
|
|
|
if(create_and_init_additional_record(packet, nb_flags, release_ip) == False)
|
|
return False;
|
|
|
|
DEBUG(4,("initiate_name_release_packet: sending release for name %s (bcast=%s) to IP %s\n",
|
|
nmb_namestr(&nmb->additional->rr_name),
|
|
BOOLSTR(nmb->header.nm_flags.bcast), inet_ntoa(packet->ip)));
|
|
|
|
return send_netbios_packet( packet );
|
|
}
|
|
|
|
/***************************************************************************
|
|
Sends out a node status.
|
|
**************************************************************************/
|
|
|
|
static bool initiate_node_status_packet( struct packet_struct *packet )
|
|
{
|
|
struct nmb_packet *nmb = &packet->packet.nmb;
|
|
|
|
nmb->header.opcode = NMB_NAME_QUERY_OPCODE;
|
|
nmb->header.arcount = 0;
|
|
|
|
nmb->header.nm_flags.recursion_desired = False;
|
|
|
|
nmb->question.question_type = QUESTION_TYPE_NB_STATUS;
|
|
|
|
DEBUG(4,("initiate_node_status_packet: sending node status request for name %s to IP %s\n",
|
|
nmb_namestr(&nmb->question.question_name),
|
|
inet_ntoa(packet->ip)));
|
|
|
|
return send_netbios_packet( packet );
|
|
}
|
|
|
|
/****************************************************************************
|
|
Simplification functions for queuing standard packets.
|
|
These should be the only publicly callable functions for sending
|
|
out packets.
|
|
****************************************************************************/
|
|
|
|
/****************************************************************************
|
|
Assertion - we should never be sending nmbd packets on the remote
|
|
broadcast subnet.
|
|
****************************************************************************/
|
|
|
|
static bool assert_check_subnet(struct subnet_record *subrec)
|
|
{
|
|
if( subrec == remote_broadcast_subnet) {
|
|
DEBUG(0,("assert_check_subnet: Attempt to send packet on remote broadcast subnet. \
|
|
This is a bug.\n"));
|
|
return True;
|
|
}
|
|
return False;
|
|
}
|
|
|
|
/****************************************************************************
|
|
Queue a register name packet to the broadcast address of a subnet.
|
|
****************************************************************************/
|
|
|
|
struct response_record *queue_register_name( struct subnet_record *subrec,
|
|
response_function resp_fn,
|
|
timeout_response_function timeout_fn,
|
|
register_name_success_function success_fn,
|
|
register_name_fail_function fail_fn,
|
|
struct userdata_struct *userdata,
|
|
struct nmb_name *nmbname,
|
|
uint16 nb_flags)
|
|
{
|
|
struct packet_struct *p;
|
|
struct response_record *rrec;
|
|
struct sockaddr_storage ss;
|
|
const struct sockaddr_storage *pss = NULL;
|
|
if(assert_check_subnet(subrec))
|
|
return NULL;
|
|
|
|
/* note that all name registration requests have RD set (rfc1002 - section 4.2.2 */
|
|
if ((p = create_and_init_netbios_packet(nmbname, (subrec != unicast_subnet), True,
|
|
subrec->bcast_ip)) == NULL)
|
|
return NULL;
|
|
|
|
in_addr_to_sockaddr_storage(&ss, subrec->bcast_ip);
|
|
pss = iface_ip((struct sockaddr *)&ss);
|
|
if (!pss || pss->ss_family != AF_INET) {
|
|
p->locked = False;
|
|
free_packet(p);
|
|
return NULL;
|
|
}
|
|
|
|
if(initiate_name_register_packet(p, nb_flags,
|
|
&((const struct sockaddr_in *)pss)->sin_addr) == False) {
|
|
p->locked = False;
|
|
free_packet(p);
|
|
return NULL;
|
|
}
|
|
|
|
if((rrec = make_response_record(subrec, /* subnet record. */
|
|
p, /* packet we sent. */
|
|
resp_fn, /* function to call on response. */
|
|
timeout_fn, /* function to call on timeout. */
|
|
(success_function)success_fn, /* function to call on operation success. */
|
|
(fail_function)fail_fn, /* function to call on operation fail. */
|
|
userdata)) == NULL) {
|
|
p->locked = False;
|
|
free_packet(p);
|
|
return NULL;
|
|
}
|
|
|
|
return rrec;
|
|
}
|
|
|
|
/****************************************************************************
|
|
Queue a refresh name packet to the broadcast address of a subnet.
|
|
****************************************************************************/
|
|
|
|
void queue_wins_refresh(struct nmb_name *nmbname,
|
|
response_function resp_fn,
|
|
timeout_response_function timeout_fn,
|
|
uint16 nb_flags,
|
|
struct in_addr refresh_ip,
|
|
const char *tag)
|
|
{
|
|
struct packet_struct *p;
|
|
struct response_record *rrec;
|
|
struct in_addr wins_ip;
|
|
struct userdata_struct *userdata;
|
|
fstring ip_str;
|
|
|
|
wins_ip = wins_srv_ip_tag(tag, refresh_ip);
|
|
|
|
if ((p = create_and_init_netbios_packet(nmbname, False, False, wins_ip)) == NULL) {
|
|
return;
|
|
}
|
|
|
|
if (!initiate_name_refresh_packet(p, nb_flags, &refresh_ip)) {
|
|
p->locked = False;
|
|
free_packet(p);
|
|
return;
|
|
}
|
|
|
|
fstrcpy(ip_str, inet_ntoa(refresh_ip));
|
|
|
|
DEBUG(6,("Refreshing name %s IP %s with WINS server %s using tag '%s'\n",
|
|
nmb_namestr(nmbname), ip_str, inet_ntoa(wins_ip), tag));
|
|
|
|
userdata = (struct userdata_struct *)SMB_MALLOC(sizeof(*userdata) + strlen(tag) + 1);
|
|
if (!userdata) {
|
|
p->locked = False;
|
|
free_packet(p);
|
|
DEBUG(0,("Failed to allocate userdata structure!\n"));
|
|
return;
|
|
}
|
|
ZERO_STRUCTP(userdata);
|
|
userdata->userdata_len = strlen(tag) + 1;
|
|
strlcpy(userdata->data, tag, userdata->userdata_len);
|
|
|
|
if ((rrec = make_response_record(unicast_subnet,
|
|
p,
|
|
resp_fn, timeout_fn,
|
|
NULL,
|
|
NULL,
|
|
userdata)) == NULL) {
|
|
p->locked = False;
|
|
free_packet(p);
|
|
return;
|
|
}
|
|
|
|
free(userdata);
|
|
|
|
/* we don't want to repeat refresh packets */
|
|
rrec->repeat_count = 0;
|
|
}
|
|
|
|
|
|
/****************************************************************************
|
|
Queue a multihomed register name packet to a given WINS server IP
|
|
****************************************************************************/
|
|
|
|
struct response_record *queue_register_multihomed_name( struct subnet_record *subrec,
|
|
response_function resp_fn,
|
|
timeout_response_function timeout_fn,
|
|
register_name_success_function success_fn,
|
|
register_name_fail_function fail_fn,
|
|
struct userdata_struct *userdata,
|
|
struct nmb_name *nmbname,
|
|
uint16 nb_flags,
|
|
struct in_addr register_ip,
|
|
struct in_addr wins_ip)
|
|
{
|
|
struct packet_struct *p;
|
|
struct response_record *rrec;
|
|
bool ret;
|
|
|
|
/* Sanity check. */
|
|
if(subrec != unicast_subnet) {
|
|
DEBUG(0,("queue_register_multihomed_name: should only be done on \
|
|
unicast subnet. subnet is %s\n.", subrec->subnet_name ));
|
|
return NULL;
|
|
}
|
|
|
|
if(assert_check_subnet(subrec))
|
|
return NULL;
|
|
|
|
if ((p = create_and_init_netbios_packet(nmbname, False, True, wins_ip)) == NULL)
|
|
return NULL;
|
|
|
|
if (nb_flags & NB_GROUP)
|
|
ret = initiate_name_register_packet( p, nb_flags, ®ister_ip);
|
|
else
|
|
ret = initiate_multihomed_name_register_packet(p, nb_flags, ®ister_ip);
|
|
|
|
if (ret == False) {
|
|
p->locked = False;
|
|
free_packet(p);
|
|
return NULL;
|
|
}
|
|
|
|
if ((rrec = make_response_record(subrec, /* subnet record. */
|
|
p, /* packet we sent. */
|
|
resp_fn, /* function to call on response. */
|
|
timeout_fn, /* function to call on timeout. */
|
|
(success_function)success_fn, /* function to call on operation success. */
|
|
(fail_function)fail_fn, /* function to call on operation fail. */
|
|
userdata)) == NULL) {
|
|
p->locked = False;
|
|
free_packet(p);
|
|
return NULL;
|
|
}
|
|
|
|
return rrec;
|
|
}
|
|
|
|
/****************************************************************************
|
|
Queue a release name packet to the broadcast address of a subnet.
|
|
****************************************************************************/
|
|
|
|
struct response_record *queue_release_name( struct subnet_record *subrec,
|
|
response_function resp_fn,
|
|
timeout_response_function timeout_fn,
|
|
release_name_success_function success_fn,
|
|
release_name_fail_function fail_fn,
|
|
struct userdata_struct *userdata,
|
|
struct nmb_name *nmbname,
|
|
uint16 nb_flags,
|
|
struct in_addr release_ip,
|
|
struct in_addr dest_ip)
|
|
{
|
|
struct packet_struct *p;
|
|
struct response_record *rrec;
|
|
|
|
if(assert_check_subnet(subrec))
|
|
return NULL;
|
|
|
|
if ((p = create_and_init_netbios_packet(nmbname, (subrec != unicast_subnet), False, dest_ip)) == NULL)
|
|
return NULL;
|
|
|
|
if(initiate_name_release_packet( p, nb_flags, &release_ip) == False) {
|
|
p->locked = False;
|
|
free_packet(p);
|
|
return NULL;
|
|
}
|
|
|
|
if((rrec = make_response_record(subrec, /* subnet record. */
|
|
p, /* packet we sent. */
|
|
resp_fn, /* function to call on response. */
|
|
timeout_fn, /* function to call on timeout. */
|
|
(success_function)success_fn, /* function to call on operation success. */
|
|
(fail_function)fail_fn, /* function to call on operation fail. */
|
|
userdata)) == NULL) {
|
|
p->locked = False;
|
|
free_packet(p);
|
|
return NULL;
|
|
}
|
|
|
|
/*
|
|
* For a broadcast release packet, only send once.
|
|
* This will cause us to remove the name asap. JRA.
|
|
*/
|
|
|
|
if (subrec != unicast_subnet) {
|
|
rrec->repeat_count = 0;
|
|
rrec->repeat_time = 0;
|
|
}
|
|
|
|
return rrec;
|
|
}
|
|
|
|
/****************************************************************************
|
|
Queue a query name packet to the broadcast address of a subnet.
|
|
****************************************************************************/
|
|
|
|
struct response_record *queue_query_name( struct subnet_record *subrec,
|
|
response_function resp_fn,
|
|
timeout_response_function timeout_fn,
|
|
query_name_success_function success_fn,
|
|
query_name_fail_function fail_fn,
|
|
struct userdata_struct *userdata,
|
|
struct nmb_name *nmbname)
|
|
{
|
|
struct packet_struct *p;
|
|
struct response_record *rrec;
|
|
struct in_addr to_ip;
|
|
|
|
if(assert_check_subnet(subrec))
|
|
return NULL;
|
|
|
|
to_ip = subrec->bcast_ip;
|
|
|
|
/* queries to the WINS server turn up here as queries to IP 0.0.0.0
|
|
These need to be handled a bit differently */
|
|
if (subrec->type == UNICAST_SUBNET && is_zero_ip_v4(to_ip)) {
|
|
/* What we really need to do is loop over each of our wins
|
|
* servers and wins server tags here, but that just doesn't
|
|
* fit our architecture at the moment (userdata may already
|
|
* be used when we get here). For now we just query the first
|
|
* active wins server on the first tag.
|
|
*/
|
|
char **tags = wins_srv_tags();
|
|
if (!tags) {
|
|
return NULL;
|
|
}
|
|
to_ip = wins_srv_ip_tag(tags[0], to_ip);
|
|
wins_srv_tags_free(tags);
|
|
}
|
|
|
|
if(( p = create_and_init_netbios_packet(nmbname,
|
|
(subrec != unicast_subnet),
|
|
(subrec == unicast_subnet),
|
|
to_ip)) == NULL)
|
|
return NULL;
|
|
|
|
if(lp_bind_interfaces_only()) {
|
|
int i;
|
|
|
|
DEBUG(10,("queue_query_name: bind_interfaces_only is set, looking for suitable source IP\n"));
|
|
for(i = 0; i < iface_count(); i++) {
|
|
const struct in_addr *ifip = iface_n_ip_v4(i);
|
|
|
|
if (ifip == NULL) {
|
|
DEBUG(0,("queue_query_name: interface %d has NULL IP address !\n", i));
|
|
continue;
|
|
}
|
|
|
|
if (is_loopback_ip_v4(*ifip)) {
|
|
DEBUG(5,("queue_query_name: ignoring loopback interface (%d)\n", i));
|
|
continue;
|
|
}
|
|
|
|
DEBUG(10,("queue_query_name: using source IP %s\n",inet_ntoa(*ifip)));
|
|
p->send_fd = find_subnet_fd_for_address( *ifip );
|
|
break;
|
|
}
|
|
}
|
|
|
|
if(initiate_name_query_packet( p ) == False) {
|
|
p->locked = False;
|
|
free_packet(p);
|
|
return NULL;
|
|
}
|
|
|
|
if((rrec = make_response_record(subrec, /* subnet record. */
|
|
p, /* packet we sent. */
|
|
resp_fn, /* function to call on response. */
|
|
timeout_fn, /* function to call on timeout. */
|
|
(success_function)success_fn, /* function to call on operation success. */
|
|
(fail_function)fail_fn, /* function to call on operation fail. */
|
|
userdata)) == NULL) {
|
|
p->locked = False;
|
|
free_packet(p);
|
|
return NULL;
|
|
}
|
|
|
|
return rrec;
|
|
}
|
|
|
|
/****************************************************************************
|
|
Queue a query name packet to a given address from the WINS subnet.
|
|
****************************************************************************/
|
|
|
|
struct response_record *queue_query_name_from_wins_server( struct in_addr to_ip,
|
|
response_function resp_fn,
|
|
timeout_response_function timeout_fn,
|
|
query_name_success_function success_fn,
|
|
query_name_fail_function fail_fn,
|
|
struct userdata_struct *userdata,
|
|
struct nmb_name *nmbname)
|
|
{
|
|
struct packet_struct *p;
|
|
struct response_record *rrec;
|
|
|
|
if ((p = create_and_init_netbios_packet(nmbname, False, False, to_ip)) == NULL)
|
|
return NULL;
|
|
|
|
if(initiate_name_query_packet_from_wins_server( p ) == False) {
|
|
p->locked = False;
|
|
free_packet(p);
|
|
return NULL;
|
|
}
|
|
|
|
if((rrec = make_response_record(wins_server_subnet, /* subnet record. */
|
|
p, /* packet we sent. */
|
|
resp_fn, /* function to call on response. */
|
|
timeout_fn, /* function to call on timeout. */
|
|
(success_function)success_fn, /* function to call on operation success. */
|
|
(fail_function)fail_fn, /* function to call on operation fail. */
|
|
userdata)) == NULL) {
|
|
p->locked = False;
|
|
free_packet(p);
|
|
return NULL;
|
|
}
|
|
|
|
return rrec;
|
|
}
|
|
|
|
/****************************************************************************
|
|
Queue a node status packet to a given name and address.
|
|
****************************************************************************/
|
|
|
|
struct response_record *queue_node_status( struct subnet_record *subrec,
|
|
response_function resp_fn,
|
|
timeout_response_function timeout_fn,
|
|
node_status_success_function success_fn,
|
|
node_status_fail_function fail_fn,
|
|
struct userdata_struct *userdata,
|
|
struct nmb_name *nmbname,
|
|
struct in_addr send_ip)
|
|
{
|
|
struct packet_struct *p;
|
|
struct response_record *rrec;
|
|
|
|
/* Sanity check. */
|
|
if(subrec != unicast_subnet) {
|
|
DEBUG(0,("queue_register_multihomed_name: should only be done on \
|
|
unicast subnet. subnet is %s\n.", subrec->subnet_name ));
|
|
return NULL;
|
|
}
|
|
|
|
if(assert_check_subnet(subrec))
|
|
return NULL;
|
|
|
|
if(( p = create_and_init_netbios_packet(nmbname, False, False, send_ip)) == NULL)
|
|
return NULL;
|
|
|
|
if(initiate_node_status_packet(p) == False) {
|
|
p->locked = False;
|
|
free_packet(p);
|
|
return NULL;
|
|
}
|
|
|
|
if((rrec = make_response_record(subrec, /* subnet record. */
|
|
p, /* packet we sent. */
|
|
resp_fn, /* function to call on response. */
|
|
timeout_fn, /* function to call on timeout. */
|
|
(success_function)success_fn, /* function to call on operation success. */
|
|
(fail_function)fail_fn, /* function to call on operation fail. */
|
|
userdata)) == NULL) {
|
|
p->locked = False;
|
|
free_packet(p);
|
|
return NULL;
|
|
}
|
|
|
|
return rrec;
|
|
}
|
|
|
|
/****************************************************************************
|
|
Reply to a netbios name packet. see rfc1002.txt
|
|
****************************************************************************/
|
|
|
|
void reply_netbios_packet(struct packet_struct *orig_packet,
|
|
int rcode, enum netbios_reply_type_code rcv_code, int opcode,
|
|
int ttl, char *data,int len)
|
|
{
|
|
struct packet_struct packet;
|
|
struct nmb_packet *nmb = NULL;
|
|
struct res_rec answers;
|
|
struct nmb_packet *orig_nmb = &orig_packet->packet.nmb;
|
|
bool loopback_this_packet = False;
|
|
int rr_type = RR_TYPE_NB;
|
|
const char *packet_type = "unknown";
|
|
|
|
/* Check if we are sending to or from ourselves. */
|
|
if(ismyip_v4(orig_packet->ip) && (orig_packet->port == global_nmb_port))
|
|
loopback_this_packet = True;
|
|
|
|
nmb = &packet.packet.nmb;
|
|
|
|
/* Do a partial copy of the packet. We clear the locked flag and
|
|
the resource record pointers. */
|
|
packet = *orig_packet; /* Full structure copy. */
|
|
packet.locked = False;
|
|
nmb->answers = NULL;
|
|
nmb->nsrecs = NULL;
|
|
nmb->additional = NULL;
|
|
|
|
switch (rcv_code) {
|
|
case NMB_STATUS:
|
|
packet_type = "nmb_status";
|
|
nmb->header.nm_flags.recursion_desired = False;
|
|
nmb->header.nm_flags.recursion_available = False;
|
|
rr_type = RR_TYPE_NBSTAT;
|
|
break;
|
|
case NMB_QUERY:
|
|
packet_type = "nmb_query";
|
|
nmb->header.nm_flags.recursion_desired = True;
|
|
nmb->header.nm_flags.recursion_available = True;
|
|
if (rcode) {
|
|
rr_type = RR_TYPE_NULL;
|
|
}
|
|
break;
|
|
case NMB_REG:
|
|
case NMB_REG_REFRESH:
|
|
packet_type = "nmb_reg";
|
|
nmb->header.nm_flags.recursion_desired = True;
|
|
nmb->header.nm_flags.recursion_available = True;
|
|
break;
|
|
case NMB_REL:
|
|
packet_type = "nmb_rel";
|
|
nmb->header.nm_flags.recursion_desired = False;
|
|
nmb->header.nm_flags.recursion_available = False;
|
|
break;
|
|
case NMB_WAIT_ACK:
|
|
packet_type = "nmb_wack";
|
|
nmb->header.nm_flags.recursion_desired = False;
|
|
nmb->header.nm_flags.recursion_available = False;
|
|
rr_type = RR_TYPE_NULL;
|
|
break;
|
|
case WINS_REG:
|
|
packet_type = "wins_reg";
|
|
nmb->header.nm_flags.recursion_desired = True;
|
|
nmb->header.nm_flags.recursion_available = True;
|
|
break;
|
|
case WINS_QUERY:
|
|
packet_type = "wins_query";
|
|
nmb->header.nm_flags.recursion_desired = True;
|
|
nmb->header.nm_flags.recursion_available = True;
|
|
if (rcode) {
|
|
rr_type = RR_TYPE_NULL;
|
|
}
|
|
break;
|
|
default:
|
|
DEBUG(0,("reply_netbios_packet: Unknown packet type: %s %s to ip %s\n",
|
|
packet_type, nmb_namestr(&orig_nmb->question.question_name),
|
|
inet_ntoa(packet.ip)));
|
|
return;
|
|
}
|
|
|
|
DEBUG(4,("reply_netbios_packet: sending a reply of packet type: %s %s to ip %s \
|
|
for id %hu\n", packet_type, nmb_namestr(&orig_nmb->question.question_name),
|
|
inet_ntoa(packet.ip), orig_nmb->header.name_trn_id));
|
|
|
|
nmb->header.name_trn_id = orig_nmb->header.name_trn_id;
|
|
nmb->header.opcode = opcode;
|
|
nmb->header.response = True;
|
|
nmb->header.nm_flags.bcast = False;
|
|
nmb->header.nm_flags.trunc = False;
|
|
nmb->header.nm_flags.authoritative = True;
|
|
|
|
nmb->header.rcode = rcode;
|
|
nmb->header.qdcount = 0;
|
|
nmb->header.ancount = 1;
|
|
nmb->header.nscount = 0;
|
|
nmb->header.arcount = 0;
|
|
|
|
memset((char*)&nmb->question,'\0',sizeof(nmb->question));
|
|
|
|
nmb->answers = &answers;
|
|
memset((char*)nmb->answers,'\0',sizeof(*nmb->answers));
|
|
|
|
nmb->answers->rr_name = orig_nmb->question.question_name;
|
|
nmb->answers->rr_type = rr_type;
|
|
nmb->answers->rr_class = RR_CLASS_IN;
|
|
nmb->answers->ttl = ttl;
|
|
|
|
if (data && len) {
|
|
if (len < 0 || len > sizeof(nmb->answers->rdata)) {
|
|
DEBUG(5,("reply_netbios_packet: "
|
|
"invalid packet len (%d)\n",
|
|
len ));
|
|
return;
|
|
}
|
|
nmb->answers->rdlength = len;
|
|
memcpy(nmb->answers->rdata, data, len);
|
|
}
|
|
|
|
packet.packet_type = NMB_PACKET;
|
|
packet.recv_fd = -1;
|
|
/* Ensure we send out on the same fd that the original
|
|
packet came in on to give the correct source IP address. */
|
|
if (orig_packet->send_fd != -1) {
|
|
packet.send_fd = orig_packet->send_fd;
|
|
} else {
|
|
packet.send_fd = orig_packet->recv_fd;
|
|
}
|
|
packet.timestamp = time(NULL);
|
|
|
|
debug_nmb_packet(&packet);
|
|
|
|
if(loopback_this_packet) {
|
|
struct packet_struct *lo_packet;
|
|
DEBUG(5,("reply_netbios_packet: sending packet to ourselves.\n"));
|
|
if((lo_packet = copy_packet(&packet)) == NULL)
|
|
return;
|
|
queue_packet(lo_packet);
|
|
} else if (!send_packet(&packet)) {
|
|
DEBUG(0,("reply_netbios_packet: send_packet to IP %s port %d failed\n",
|
|
inet_ntoa(packet.ip),packet.port));
|
|
}
|
|
}
|
|
|
|
/*******************************************************************
|
|
Queue a packet into a packet queue
|
|
******************************************************************/
|
|
|
|
void queue_packet(struct packet_struct *packet)
|
|
{
|
|
struct packet_struct *p;
|
|
|
|
if (!packet_queue) {
|
|
packet->prev = NULL;
|
|
packet->next = NULL;
|
|
packet_queue = packet;
|
|
return;
|
|
}
|
|
|
|
/* find the bottom */
|
|
for (p=packet_queue;p->next;p=p->next)
|
|
;
|
|
|
|
p->next = packet;
|
|
packet->next = NULL;
|
|
packet->prev = p;
|
|
}
|
|
|
|
/****************************************************************************
|
|
Try and find a matching subnet record for a datagram port 138 packet.
|
|
****************************************************************************/
|
|
|
|
static struct subnet_record *find_subnet_for_dgram_browse_packet(struct packet_struct *p)
|
|
{
|
|
struct subnet_record *subrec;
|
|
|
|
/* Go through all the broadcast subnets and see if the mask matches. */
|
|
for (subrec = FIRST_SUBNET; subrec ; subrec = NEXT_SUBNET_EXCLUDING_UNICAST(subrec)) {
|
|
if(same_net_v4(p->ip, subrec->bcast_ip, subrec->mask_ip))
|
|
return subrec;
|
|
}
|
|
|
|
/* If the subnet record is the remote announce broadcast subnet,
|
|
hack it here to be the first subnet. This is really gross and
|
|
is needed due to people turning on port 137/138 broadcast
|
|
forwarding on their routers. May fire and brimstone rain
|
|
down upon them...
|
|
*/
|
|
|
|
return FIRST_SUBNET;
|
|
}
|
|
|
|
/****************************************************************************
|
|
Dispatch a browse frame from port 138 to the correct processing function.
|
|
****************************************************************************/
|
|
|
|
static void process_browse_packet(struct packet_struct *p, char *buf,int len)
|
|
{
|
|
struct dgram_packet *dgram = &p->packet.dgram;
|
|
int command = CVAL(buf,0);
|
|
struct subnet_record *subrec = find_subnet_for_dgram_browse_packet(p);
|
|
char scope[64];
|
|
unstring src_name;
|
|
|
|
/* Drop the packet if it's a different NetBIOS scope, or the source is from one of our names. */
|
|
pull_ascii(scope, dgram->dest_name.scope, 64, 64, STR_TERMINATE);
|
|
if (!strequal(scope, global_scope())) {
|
|
DEBUG(7,("process_browse_packet: Discarding datagram from IP %s. Scope (%s) \
|
|
mismatch with our scope (%s).\n", inet_ntoa(p->ip), scope, global_scope()));
|
|
return;
|
|
}
|
|
|
|
pull_ascii_nstring(src_name, sizeof(src_name), dgram->source_name.name);
|
|
if (is_myname(src_name)) {
|
|
DEBUG(7,("process_browse_packet: Discarding datagram from IP %s. Source name \
|
|
%s is one of our names !\n", inet_ntoa(p->ip), nmb_namestr(&dgram->source_name)));
|
|
return;
|
|
}
|
|
|
|
switch (command) {
|
|
case ANN_HostAnnouncement:
|
|
debug_browse_data(buf, len);
|
|
process_host_announce(subrec, p, buf+1);
|
|
break;
|
|
case ANN_DomainAnnouncement:
|
|
debug_browse_data(buf, len);
|
|
process_workgroup_announce(subrec, p, buf+1);
|
|
break;
|
|
case ANN_LocalMasterAnnouncement:
|
|
debug_browse_data(buf, len);
|
|
process_local_master_announce(subrec, p, buf+1);
|
|
break;
|
|
case ANN_AnnouncementRequest:
|
|
debug_browse_data(buf, len);
|
|
process_announce_request(subrec, p, buf+1);
|
|
break;
|
|
case ANN_Election:
|
|
debug_browse_data(buf, len);
|
|
process_election(subrec, p, buf+1);
|
|
break;
|
|
case ANN_GetBackupListReq:
|
|
debug_browse_data(buf, len);
|
|
process_get_backup_list_request(subrec, p, buf+1);
|
|
break;
|
|
case ANN_GetBackupListResp:
|
|
debug_browse_data(buf, len);
|
|
/* We never send ANN_GetBackupListReq so we should never get these. */
|
|
DEBUG(0,("process_browse_packet: Discarding GetBackupListResponse \
|
|
packet from %s IP %s\n", nmb_namestr(&dgram->source_name), inet_ntoa(p->ip)));
|
|
break;
|
|
case ANN_ResetBrowserState:
|
|
debug_browse_data(buf, len);
|
|
process_reset_browser(subrec, p, buf+1);
|
|
break;
|
|
case ANN_MasterAnnouncement:
|
|
/* Master browser datagrams must be processed on the unicast subnet. */
|
|
subrec = unicast_subnet;
|
|
|
|
debug_browse_data(buf, len);
|
|
process_master_browser_announce(subrec, p, buf+1);
|
|
break;
|
|
case ANN_BecomeBackup:
|
|
/*
|
|
* We don't currently implement this. Log it just in case.
|
|
*/
|
|
debug_browse_data(buf, len);
|
|
DEBUG(10,("process_browse_packet: On subnet %s ignoring browse packet \
|
|
command ANN_BecomeBackup from %s IP %s to %s\n", subrec->subnet_name, nmb_namestr(&dgram->source_name),
|
|
inet_ntoa(p->ip), nmb_namestr(&dgram->dest_name)));
|
|
break;
|
|
default:
|
|
debug_browse_data(buf, len);
|
|
DEBUG(0,("process_browse_packet: On subnet %s ignoring browse packet \
|
|
command code %d from %s IP %s to %s\n", subrec->subnet_name, command, nmb_namestr(&dgram->source_name),
|
|
inet_ntoa(p->ip), nmb_namestr(&dgram->dest_name)));
|
|
break;
|
|
}
|
|
}
|
|
|
|
/****************************************************************************
|
|
Dispatch a LanMan browse frame from port 138 to the correct processing function.
|
|
****************************************************************************/
|
|
|
|
static void process_lanman_packet(struct packet_struct *p, char *buf,int len)
|
|
{
|
|
struct dgram_packet *dgram = &p->packet.dgram;
|
|
int command = SVAL(buf,0);
|
|
struct subnet_record *subrec = find_subnet_for_dgram_browse_packet(p);
|
|
char scope[64];
|
|
unstring src_name;
|
|
|
|
/* Drop the packet if it's a different NetBIOS scope, or the source is from one of our names. */
|
|
|
|
pull_ascii(scope, dgram->dest_name.scope, 64, 64, STR_TERMINATE);
|
|
if (!strequal(scope, global_scope())) {
|
|
DEBUG(7,("process_lanman_packet: Discarding datagram from IP %s. Scope (%s) \
|
|
mismatch with our scope (%s).\n", inet_ntoa(p->ip), scope, global_scope()));
|
|
return;
|
|
}
|
|
|
|
pull_ascii_nstring(src_name, sizeof(src_name), dgram->source_name.name);
|
|
if (is_myname(src_name)) {
|
|
DEBUG(0,("process_lanman_packet: Discarding datagram from IP %s. Source name \
|
|
%s is one of our names !\n", inet_ntoa(p->ip), nmb_namestr(&dgram->source_name)));
|
|
return;
|
|
}
|
|
|
|
switch (command) {
|
|
case ANN_HostAnnouncement:
|
|
debug_browse_data(buf, len);
|
|
process_lm_host_announce(subrec, p, buf+1, len > 1 ? len-1 : 0);
|
|
break;
|
|
case ANN_AnnouncementRequest:
|
|
process_lm_announce_request(subrec, p, buf+1, len > 1 ? len-1 : 0);
|
|
break;
|
|
default:
|
|
DEBUG(0,("process_lanman_packet: On subnet %s ignoring browse packet \
|
|
command code %d from %s IP %s to %s\n", subrec->subnet_name, command, nmb_namestr(&dgram->source_name),
|
|
inet_ntoa(p->ip), nmb_namestr(&dgram->dest_name)));
|
|
break;
|
|
}
|
|
}
|
|
|
|
/****************************************************************************
|
|
Determine if a packet is for us on port 138. Note that to have any chance of
|
|
being efficient we need to drop as many packets as possible at this
|
|
stage as subsequent processing is expensive.
|
|
****************************************************************************/
|
|
|
|
static bool listening(struct packet_struct *p,struct nmb_name *nbname)
|
|
{
|
|
struct subnet_record *subrec = NULL;
|
|
|
|
for (subrec = FIRST_SUBNET; subrec ; subrec = NEXT_SUBNET_EXCLUDING_UNICAST(subrec)) {
|
|
if(same_net_v4(p->ip, subrec->bcast_ip, subrec->mask_ip))
|
|
break;
|
|
}
|
|
|
|
if(subrec == NULL)
|
|
subrec = unicast_subnet;
|
|
|
|
return (find_name_on_subnet(subrec, nbname, FIND_SELF_NAME) != NULL);
|
|
}
|
|
|
|
/****************************************************************************
|
|
Process udp 138 datagrams
|
|
****************************************************************************/
|
|
|
|
static void process_dgram(struct packet_struct *p)
|
|
{
|
|
char *buf;
|
|
char *buf2;
|
|
int len;
|
|
struct dgram_packet *dgram = &p->packet.dgram;
|
|
|
|
/* If we aren't listening to the destination name then ignore the packet */
|
|
if (!listening(p,&dgram->dest_name)) {
|
|
unexpected_packet(p);
|
|
DEBUG(5,("process_dgram: ignoring dgram packet sent to name %s from %s\n",
|
|
nmb_namestr(&dgram->dest_name), inet_ntoa(p->ip)));
|
|
return;
|
|
}
|
|
|
|
if (dgram->header.msg_type != 0x10 && dgram->header.msg_type != 0x11 && dgram->header.msg_type != 0x12) {
|
|
unexpected_packet(p);
|
|
/* Don't process error packets etc yet */
|
|
DEBUG(5,("process_dgram: ignoring dgram packet sent to name %s from IP %s as it is \
|
|
an error packet of type %x\n", nmb_namestr(&dgram->dest_name), inet_ntoa(p->ip), dgram->header.msg_type));
|
|
return;
|
|
}
|
|
|
|
/* Ensure we have a large enough packet before looking inside. */
|
|
if (dgram->datasize < (smb_vwv12 - 2)) {
|
|
/* That's the offset minus the 4 byte length + 2 bytes of offset. */
|
|
DEBUG(0,("process_dgram: ignoring too short dgram packet (%u) sent to name %s from IP %s\n",
|
|
(unsigned int)dgram->datasize,
|
|
nmb_namestr(&dgram->dest_name),
|
|
inet_ntoa(p->ip) ));
|
|
return;
|
|
}
|
|
|
|
buf = &dgram->data[0];
|
|
buf -= 4; /* XXXX for the pseudo tcp length - someday I need to get rid of this */
|
|
|
|
if (CVAL(buf,smb_com) != SMBtrans)
|
|
return;
|
|
|
|
len = SVAL(buf,smb_vwv11);
|
|
buf2 = smb_base(buf) + SVAL(buf,smb_vwv12);
|
|
|
|
if (len <= 0 || len > dgram->datasize) {
|
|
DEBUG(0,("process_dgram: ignoring malformed1 (datasize = %d, len = %d) datagram \
|
|
packet sent to name %s from IP %s\n",
|
|
dgram->datasize,
|
|
len,
|
|
nmb_namestr(&dgram->dest_name),
|
|
inet_ntoa(p->ip) ));
|
|
return;
|
|
}
|
|
|
|
if (buf2 < dgram->data || (buf2 >= dgram->data + dgram->datasize)) {
|
|
DEBUG(0,("process_dgram: ignoring malformed2 (datasize = %d, len=%d, off=%d) datagram \
|
|
packet sent to name %s from IP %s\n",
|
|
dgram->datasize,
|
|
len,
|
|
(int)PTR_DIFF(buf2, dgram->data),
|
|
nmb_namestr(&dgram->dest_name),
|
|
inet_ntoa(p->ip) ));
|
|
return;
|
|
}
|
|
|
|
if ((buf2 + len < dgram->data) || (buf2 + len > dgram->data + dgram->datasize)) {
|
|
DEBUG(0,("process_dgram: ignoring malformed3 (datasize = %d, len=%d, off=%d) datagram \
|
|
packet sent to name %s from IP %s\n",
|
|
dgram->datasize,
|
|
len,
|
|
(int)PTR_DIFF(buf2, dgram->data),
|
|
nmb_namestr(&dgram->dest_name),
|
|
inet_ntoa(p->ip) ));
|
|
return;
|
|
}
|
|
|
|
DEBUG(4,("process_dgram: datagram from %s to %s IP %s for %s of type %d len=%d\n",
|
|
nmb_namestr(&dgram->source_name),nmb_namestr(&dgram->dest_name),
|
|
inet_ntoa(p->ip), smb_buf(buf),CVAL(buf2,0),len));
|
|
|
|
/* Datagram packet received for the browser mailslot */
|
|
if (strequal(smb_buf(buf),BROWSE_MAILSLOT)) {
|
|
process_browse_packet(p,buf2,len);
|
|
return;
|
|
}
|
|
|
|
/* Datagram packet received for the LAN Manager mailslot */
|
|
if (strequal(smb_buf(buf),LANMAN_MAILSLOT)) {
|
|
process_lanman_packet(p,buf2,len);
|
|
return;
|
|
}
|
|
|
|
/* Datagram packet received for the domain logon mailslot */
|
|
if (strequal(smb_buf(buf),NET_LOGON_MAILSLOT)) {
|
|
process_logon_packet(p,buf2,len,NET_LOGON_MAILSLOT);
|
|
return;
|
|
}
|
|
|
|
/* Datagram packet received for the NT domain logon mailslot */
|
|
if (strequal(smb_buf(buf),NT_LOGON_MAILSLOT)) {
|
|
process_logon_packet(p,buf2,len,NT_LOGON_MAILSLOT);
|
|
return;
|
|
}
|
|
|
|
unexpected_packet(p);
|
|
}
|
|
|
|
/****************************************************************************
|
|
Validate a response nmb packet.
|
|
****************************************************************************/
|
|
|
|
static bool validate_nmb_response_packet( struct nmb_packet *nmb )
|
|
{
|
|
bool ignore = False;
|
|
|
|
switch (nmb->header.opcode) {
|
|
case NMB_NAME_REG_OPCODE:
|
|
case NMB_NAME_REFRESH_OPCODE_8: /* ambiguity in rfc1002 about which is correct. */
|
|
case NMB_NAME_REFRESH_OPCODE_9: /* WinNT uses 8 by default. */
|
|
if (nmb->header.ancount == 0) {
|
|
DEBUG(0,("validate_nmb_response_packet: Bad REG/REFRESH Packet. "));
|
|
ignore = True;
|
|
}
|
|
break;
|
|
|
|
case NMB_NAME_QUERY_OPCODE:
|
|
if ((nmb->header.ancount != 0) && (nmb->header.ancount != 1)) {
|
|
DEBUG(0,("validate_nmb_response_packet: Bad QUERY Packet. "));
|
|
ignore = True;
|
|
}
|
|
break;
|
|
|
|
case NMB_NAME_RELEASE_OPCODE:
|
|
if (nmb->header.ancount == 0) {
|
|
DEBUG(0,("validate_nmb_response_packet: Bad RELEASE Packet. "));
|
|
ignore = True;
|
|
}
|
|
break;
|
|
|
|
case NMB_WACK_OPCODE:
|
|
/* Check WACK response here. */
|
|
if (nmb->header.ancount != 1) {
|
|
DEBUG(0,("validate_nmb_response_packet: Bad WACK Packet. "));
|
|
ignore = True;
|
|
}
|
|
break;
|
|
default:
|
|
DEBUG(0,("validate_nmb_response_packet: Ignoring packet with unknown opcode %d.\n",
|
|
nmb->header.opcode));
|
|
return True;
|
|
}
|
|
|
|
if(ignore)
|
|
DEBUG(0,("Ignoring response packet with opcode %d.\n", nmb->header.opcode));
|
|
|
|
return ignore;
|
|
}
|
|
|
|
/****************************************************************************
|
|
Validate a request nmb packet.
|
|
****************************************************************************/
|
|
|
|
static bool validate_nmb_packet( struct nmb_packet *nmb )
|
|
{
|
|
bool ignore = False;
|
|
|
|
switch (nmb->header.opcode) {
|
|
case NMB_NAME_REG_OPCODE:
|
|
case NMB_NAME_REFRESH_OPCODE_8: /* ambiguity in rfc1002 about which is correct. */
|
|
case NMB_NAME_REFRESH_OPCODE_9: /* WinNT uses 8 by default. */
|
|
case NMB_NAME_MULTIHOMED_REG_OPCODE:
|
|
if (nmb->header.qdcount==0 || nmb->header.arcount==0) {
|
|
DEBUG(0,("validate_nmb_packet: Bad REG/REFRESH Packet. "));
|
|
ignore = True;
|
|
}
|
|
break;
|
|
|
|
case NMB_NAME_QUERY_OPCODE:
|
|
if ((nmb->header.qdcount == 0) || ((nmb->question.question_type != QUESTION_TYPE_NB_QUERY) &&
|
|
(nmb->question.question_type != QUESTION_TYPE_NB_STATUS))) {
|
|
DEBUG(0,("validate_nmb_packet: Bad QUERY Packet. "));
|
|
ignore = True;
|
|
}
|
|
break;
|
|
|
|
case NMB_NAME_RELEASE_OPCODE:
|
|
if (nmb->header.qdcount==0 || nmb->header.arcount==0) {
|
|
DEBUG(0,("validate_nmb_packet: Bad RELEASE Packet. "));
|
|
ignore = True;
|
|
}
|
|
break;
|
|
default:
|
|
DEBUG(0,("validate_nmb_packet: Ignoring packet with unknown opcode %d.\n",
|
|
nmb->header.opcode));
|
|
return True;
|
|
}
|
|
|
|
if(ignore)
|
|
DEBUG(0,("validate_nmb_packet: Ignoring request packet with opcode %d.\n", nmb->header.opcode));
|
|
|
|
return ignore;
|
|
}
|
|
|
|
/****************************************************************************
|
|
Find a subnet (and potentially a response record) for a packet.
|
|
****************************************************************************/
|
|
|
|
static struct subnet_record *find_subnet_for_nmb_packet( struct packet_struct *p,
|
|
struct response_record **pprrec)
|
|
{
|
|
struct nmb_packet *nmb = &p->packet.nmb;
|
|
struct response_record *rrec = NULL;
|
|
struct subnet_record *subrec = NULL;
|
|
|
|
if(pprrec != NULL)
|
|
*pprrec = NULL;
|
|
|
|
if(nmb->header.response) {
|
|
/* It's a response packet. Find a record for it or it's an error. */
|
|
|
|
rrec = find_response_record( &subrec, nmb->header.name_trn_id);
|
|
if(rrec == NULL) {
|
|
DEBUG(3,("find_subnet_for_nmb_packet: response record not found for response id %hu\n",
|
|
nmb->header.name_trn_id));
|
|
unexpected_packet(p);
|
|
return NULL;
|
|
}
|
|
|
|
if(subrec == NULL) {
|
|
DEBUG(0,("find_subnet_for_nmb_packet: subnet record not found for response id %hu\n",
|
|
nmb->header.name_trn_id));
|
|
return NULL;
|
|
}
|
|
|
|
if(pprrec != NULL)
|
|
*pprrec = rrec;
|
|
return subrec;
|
|
}
|
|
|
|
/* Try and see what subnet this packet belongs to. */
|
|
|
|
/* WINS server ? */
|
|
if(packet_is_for_wins_server(p))
|
|
return wins_server_subnet;
|
|
|
|
/* If it wasn't a broadcast packet then send to the UNICAST subnet. */
|
|
if(nmb->header.nm_flags.bcast == False)
|
|
return unicast_subnet;
|
|
|
|
/* Go through all the broadcast subnets and see if the mask matches. */
|
|
for (subrec = FIRST_SUBNET; subrec ; subrec = NEXT_SUBNET_EXCLUDING_UNICAST(subrec)) {
|
|
if(same_net_v4(p->ip, subrec->bcast_ip, subrec->mask_ip))
|
|
return subrec;
|
|
}
|
|
|
|
/* If none match it must have been a directed broadcast - assign the remote_broadcast_subnet. */
|
|
return remote_broadcast_subnet;
|
|
}
|
|
|
|
/****************************************************************************
|
|
Process a nmb request packet - validate the packet and route it.
|
|
****************************************************************************/
|
|
|
|
static void process_nmb_request(struct packet_struct *p)
|
|
{
|
|
struct nmb_packet *nmb = &p->packet.nmb;
|
|
struct subnet_record *subrec = NULL;
|
|
|
|
debug_nmb_packet(p);
|
|
|
|
/* Ensure we have a good packet. */
|
|
if(validate_nmb_packet(nmb))
|
|
return;
|
|
|
|
/* Allocate a subnet to this packet - if we cannot - fail. */
|
|
if((subrec = find_subnet_for_nmb_packet(p, NULL))==NULL)
|
|
return;
|
|
|
|
switch (nmb->header.opcode) {
|
|
case NMB_NAME_REG_OPCODE:
|
|
if(subrec == wins_server_subnet)
|
|
wins_process_name_registration_request(subrec, p);
|
|
else
|
|
process_name_registration_request(subrec, p);
|
|
break;
|
|
|
|
case NMB_NAME_REFRESH_OPCODE_8: /* ambiguity in rfc1002 about which is correct. */
|
|
case NMB_NAME_REFRESH_OPCODE_9:
|
|
if(subrec == wins_server_subnet)
|
|
wins_process_name_refresh_request(subrec, p);
|
|
else
|
|
process_name_refresh_request(subrec, p);
|
|
break;
|
|
|
|
case NMB_NAME_MULTIHOMED_REG_OPCODE:
|
|
if(subrec == wins_server_subnet) {
|
|
wins_process_multihomed_name_registration_request(subrec, p);
|
|
} else {
|
|
DEBUG(0,("process_nmb_request: Multihomed registration request must be \
|
|
directed at a WINS server.\n"));
|
|
}
|
|
break;
|
|
|
|
case NMB_NAME_QUERY_OPCODE:
|
|
switch (nmb->question.question_type) {
|
|
case QUESTION_TYPE_NB_QUERY:
|
|
if(subrec == wins_server_subnet)
|
|
wins_process_name_query_request(subrec, p);
|
|
else
|
|
process_name_query_request(subrec, p);
|
|
break;
|
|
case QUESTION_TYPE_NB_STATUS:
|
|
if(subrec == wins_server_subnet) {
|
|
DEBUG(0,("process_nmb_request: NB_STATUS request directed at WINS server is \
|
|
not allowed.\n"));
|
|
break;
|
|
} else {
|
|
process_node_status_request(subrec, p);
|
|
}
|
|
break;
|
|
}
|
|
break;
|
|
|
|
case NMB_NAME_RELEASE_OPCODE:
|
|
if(subrec == wins_server_subnet)
|
|
wins_process_name_release_request(subrec, p);
|
|
else
|
|
process_name_release_request(subrec, p);
|
|
break;
|
|
}
|
|
}
|
|
|
|
/****************************************************************************
|
|
Process a nmb response packet - validate the packet and route it.
|
|
to either the WINS server or a normal response.
|
|
****************************************************************************/
|
|
|
|
static void process_nmb_response(struct packet_struct *p)
|
|
{
|
|
struct nmb_packet *nmb = &p->packet.nmb;
|
|
struct subnet_record *subrec = NULL;
|
|
struct response_record *rrec = NULL;
|
|
|
|
debug_nmb_packet(p);
|
|
|
|
if(validate_nmb_response_packet(nmb))
|
|
return;
|
|
|
|
if((subrec = find_subnet_for_nmb_packet(p, &rrec))==NULL)
|
|
return;
|
|
|
|
if(rrec == NULL) {
|
|
DEBUG(0,("process_nmb_response: response packet received but no response record \
|
|
found for id = %hu. Ignoring packet.\n", nmb->header.name_trn_id));
|
|
return;
|
|
}
|
|
|
|
/* Increment the number of responses received for this record. */
|
|
rrec->num_msgs++;
|
|
/* Ensure we don't re-send the request. */
|
|
rrec->repeat_count = 0;
|
|
|
|
/* Call the response received function for this packet. */
|
|
(*rrec->resp_fn)(subrec, rrec, p);
|
|
}
|
|
|
|
/*******************************************************************
|
|
Run elements off the packet queue till its empty
|
|
******************************************************************/
|
|
|
|
void run_packet_queue(void)
|
|
{
|
|
struct packet_struct *p;
|
|
|
|
while ((p = packet_queue)) {
|
|
packet_queue = p->next;
|
|
if (packet_queue)
|
|
packet_queue->prev = NULL;
|
|
p->next = p->prev = NULL;
|
|
|
|
switch (p->packet_type) {
|
|
case NMB_PACKET:
|
|
if(p->packet.nmb.header.response)
|
|
process_nmb_response(p);
|
|
else
|
|
process_nmb_request(p);
|
|
break;
|
|
|
|
case DGRAM_PACKET:
|
|
process_dgram(p);
|
|
break;
|
|
}
|
|
free_packet(p);
|
|
}
|
|
}
|
|
|
|
/*******************************************************************
|
|
Retransmit or timeout elements from all the outgoing subnet response
|
|
record queues. NOTE that this code must also check the WINS server
|
|
subnet for response records to timeout as the WINS server code
|
|
can send requests to check if a client still owns a name.
|
|
(Patch from Andrey Alekseyev <fetch@muffin.arcadia.spb.ru>).
|
|
******************************************************************/
|
|
|
|
void retransmit_or_expire_response_records(time_t t)
|
|
{
|
|
struct subnet_record *subrec;
|
|
|
|
for (subrec = FIRST_SUBNET; subrec; subrec = get_next_subnet_maybe_unicast_or_wins_server(subrec)) {
|
|
struct response_record *rrec, *nextrrec;
|
|
|
|
restart:
|
|
|
|
for (rrec = subrec->responselist; rrec; rrec = nextrrec) {
|
|
nextrrec = rrec->next;
|
|
|
|
if (rrec->repeat_time <= t) {
|
|
if (rrec->repeat_count > 0) {
|
|
/* Resend while we have a non-zero repeat_count. */
|
|
if(!send_packet(rrec->packet)) {
|
|
DEBUG(0,("retransmit_or_expire_response_records: Failed to resend packet id %hu \
|
|
to IP %s on subnet %s\n", rrec->response_id, inet_ntoa(rrec->packet->ip), subrec->subnet_name));
|
|
}
|
|
rrec->repeat_time = t + rrec->repeat_interval;
|
|
rrec->repeat_count--;
|
|
} else {
|
|
DEBUG(4,("retransmit_or_expire_response_records: timeout for packet id %hu to IP %s \
|
|
on subnet %s\n", rrec->response_id, inet_ntoa(rrec->packet->ip), subrec->subnet_name));
|
|
|
|
/*
|
|
* Check the flag in this record to prevent recursion if we end
|
|
* up in this function again via the timeout function call.
|
|
*/
|
|
|
|
if(!rrec->in_expiration_processing) {
|
|
|
|
/*
|
|
* Set the recursion protection flag in this record.
|
|
*/
|
|
|
|
rrec->in_expiration_processing = True;
|
|
|
|
/* Call the timeout function. This will deal with removing the
|
|
timed out packet. */
|
|
if(rrec->timeout_fn) {
|
|
(*rrec->timeout_fn)(subrec, rrec);
|
|
} else {
|
|
/* We must remove the record ourself if there is
|
|
no timeout function. */
|
|
remove_response_record(subrec, rrec);
|
|
}
|
|
/* We have changed subrec->responselist,
|
|
* restart from the beginning of this list. */
|
|
goto restart;
|
|
} /* !rrec->in_expitation_processing */
|
|
} /* rrec->repeat_count > 0 */
|
|
} /* rrec->repeat_time <= t */
|
|
} /* end for rrec */
|
|
} /* end for subnet */
|
|
}
|
|
|
|
/****************************************************************************
|
|
Create an fd_set containing all the sockets in the subnet structures,
|
|
plus the broadcast sockets.
|
|
***************************************************************************/
|
|
|
|
static bool create_listen_fdset(fd_set **ppset, int **psock_array, int *listen_number, int *maxfd)
|
|
{
|
|
int *sock_array = NULL;
|
|
struct subnet_record *subrec = NULL;
|
|
int count = 0;
|
|
int num = 0;
|
|
fd_set *pset = SMB_MALLOC_P(fd_set);
|
|
|
|
if(pset == NULL) {
|
|
DEBUG(0,("create_listen_fdset: malloc fail !\n"));
|
|
return True;
|
|
}
|
|
|
|
/* The Client* sockets */
|
|
count++;
|
|
|
|
/* Check that we can add all the fd's we need. */
|
|
for (subrec = FIRST_SUBNET; subrec; subrec = NEXT_SUBNET_EXCLUDING_UNICAST(subrec))
|
|
count++;
|
|
|
|
/* each interface gets 4 sockets */
|
|
count *= 4;
|
|
|
|
if(count > FD_SETSIZE) {
|
|
DEBUG(0,("create_listen_fdset: Too many file descriptors needed (%d). We can \
|
|
only use %d.\n", count, FD_SETSIZE));
|
|
SAFE_FREE(pset);
|
|
return True;
|
|
}
|
|
|
|
if((sock_array = SMB_MALLOC_ARRAY(int, count)) == NULL) {
|
|
DEBUG(0,("create_listen_fdset: malloc fail for socket array. size %d\n", count));
|
|
SAFE_FREE(pset);
|
|
return True;
|
|
}
|
|
|
|
FD_ZERO(pset);
|
|
|
|
/* Add in the lp_socket_address() interface on 137. */
|
|
FD_SET(ClientNMB,pset);
|
|
sock_array[num++] = ClientNMB;
|
|
*maxfd = MAX( *maxfd, ClientNMB);
|
|
|
|
/* the lp_socket_address() interface has only one socket */
|
|
sock_array[num++] = -1;
|
|
|
|
/* Add in the 137 sockets on all the interfaces. */
|
|
for (subrec = FIRST_SUBNET; subrec; subrec = NEXT_SUBNET_EXCLUDING_UNICAST(subrec)) {
|
|
FD_SET(subrec->nmb_sock,pset);
|
|
sock_array[num++] = subrec->nmb_sock;
|
|
*maxfd = MAX( *maxfd, subrec->nmb_sock);
|
|
|
|
sock_array[num++] = subrec->nmb_bcast;
|
|
if (subrec->nmb_bcast != -1) {
|
|
FD_SET(subrec->nmb_bcast,pset);
|
|
*maxfd = MAX( *maxfd, subrec->nmb_bcast);
|
|
}
|
|
}
|
|
|
|
/* Add in the lp_socket_address() interface on 138. */
|
|
FD_SET(ClientDGRAM,pset);
|
|
sock_array[num++] = ClientDGRAM;
|
|
*maxfd = MAX( *maxfd, ClientDGRAM);
|
|
|
|
/* the lp_socket_address() interface has only one socket */
|
|
sock_array[num++] = -1;
|
|
|
|
/* Add in the 138 sockets on all the interfaces. */
|
|
for (subrec = FIRST_SUBNET; subrec; subrec = NEXT_SUBNET_EXCLUDING_UNICAST(subrec)) {
|
|
FD_SET(subrec->dgram_sock,pset);
|
|
sock_array[num++] = subrec->dgram_sock;
|
|
*maxfd = MAX( *maxfd, subrec->dgram_sock);
|
|
|
|
sock_array[num++] = subrec->dgram_bcast;
|
|
if (subrec->dgram_bcast != -1) {
|
|
FD_SET(subrec->dgram_bcast,pset);
|
|
*maxfd = MAX( *maxfd, subrec->dgram_bcast);
|
|
}
|
|
}
|
|
|
|
*listen_number = count;
|
|
|
|
SAFE_FREE(*ppset);
|
|
SAFE_FREE(*psock_array);
|
|
|
|
*ppset = pset;
|
|
*psock_array = sock_array;
|
|
|
|
return False;
|
|
}
|
|
|
|
/****************************************************************************
|
|
List of packets we're processing this select.
|
|
***************************************************************************/
|
|
|
|
struct processed_packet {
|
|
struct processed_packet *next;
|
|
struct processed_packet *prev;
|
|
enum packet_type packet_type;
|
|
struct in_addr ip;
|
|
int packet_id;
|
|
};
|
|
|
|
/****************************************************************************
|
|
Have we seen this before ?
|
|
***************************************************************************/
|
|
|
|
static bool is_processed_packet(struct processed_packet *processed_packet_list,
|
|
struct packet_struct *packet)
|
|
{
|
|
struct processed_packet *p = NULL;
|
|
|
|
for (p = processed_packet_list; p; p = p->next) {
|
|
if (ip_equal_v4(p->ip, packet->ip) && p->packet_type == packet->packet_type) {
|
|
if ((p->packet_type == NMB_PACKET) &&
|
|
(p->packet_id ==
|
|
packet->packet.nmb.header.name_trn_id)) {
|
|
return true;
|
|
} else if ((p->packet_type == DGRAM_PACKET) &&
|
|
(p->packet_id ==
|
|
packet->packet.dgram.header.dgm_id)) {
|
|
return true;
|
|
}
|
|
}
|
|
}
|
|
return false;
|
|
}
|
|
|
|
/****************************************************************************
|
|
Keep a list of what we've seen before.
|
|
***************************************************************************/
|
|
|
|
static bool store_processed_packet(struct processed_packet **pp_processed_packet_list,
|
|
struct packet_struct *packet)
|
|
{
|
|
struct processed_packet *p = SMB_MALLOC_P(struct processed_packet);
|
|
if (!p) {
|
|
return false;
|
|
}
|
|
p->packet_type = packet->packet_type;
|
|
p->ip = packet->ip;
|
|
if (packet->packet_type == NMB_PACKET) {
|
|
p->packet_id = packet->packet.nmb.header.name_trn_id;
|
|
} else if (packet->packet_type == DGRAM_PACKET) {
|
|
p->packet_id = packet->packet.dgram.header.dgm_id;
|
|
} else {
|
|
return false;
|
|
}
|
|
|
|
DLIST_ADD(*pp_processed_packet_list, p);
|
|
return true;
|
|
}
|
|
|
|
/****************************************************************************
|
|
Throw away what we've seen before.
|
|
***************************************************************************/
|
|
|
|
static void free_processed_packet_list(struct processed_packet **pp_processed_packet_list)
|
|
{
|
|
struct processed_packet *p = NULL, *next = NULL;
|
|
|
|
for (p = *pp_processed_packet_list; p; p = next) {
|
|
next = p->next;
|
|
DLIST_REMOVE(*pp_processed_packet_list, p);
|
|
SAFE_FREE(p);
|
|
}
|
|
}
|
|
|
|
/****************************************************************************
|
|
Listens for NMB or DGRAM packets, and queues them.
|
|
return True if the socket is dead
|
|
***************************************************************************/
|
|
|
|
bool listen_for_packets(bool run_election)
|
|
{
|
|
static fd_set *listen_set = NULL;
|
|
static int listen_number = 0;
|
|
static int *sock_array = NULL;
|
|
int i;
|
|
static int maxfd = 0;
|
|
|
|
fd_set r_fds;
|
|
fd_set w_fds;
|
|
int selrtn;
|
|
struct timeval timeout;
|
|
#ifndef SYNC_DNS
|
|
int dns_fd;
|
|
#endif
|
|
struct processed_packet *processed_packet_list = NULL;
|
|
|
|
if(listen_set == NULL || rescan_listen_set) {
|
|
if(create_listen_fdset(&listen_set, &sock_array, &listen_number, &maxfd)) {
|
|
DEBUG(0,("listen_for_packets: Fatal error. unable to create listen set. Exiting.\n"));
|
|
return True;
|
|
}
|
|
rescan_listen_set = False;
|
|
}
|
|
|
|
memcpy((char *)&r_fds, (char *)listen_set, sizeof(fd_set));
|
|
FD_ZERO(&w_fds);
|
|
|
|
#ifndef SYNC_DNS
|
|
dns_fd = asyncdns_fd();
|
|
if (dns_fd != -1) {
|
|
FD_SET(dns_fd, &r_fds);
|
|
maxfd = MAX( maxfd, dns_fd);
|
|
}
|
|
#endif
|
|
|
|
/* Process a signal and timer events now... */
|
|
if (run_events(nmbd_event_context(), 0, NULL, NULL)) {
|
|
return False;
|
|
}
|
|
|
|
/*
|
|
* During elections and when expecting a netbios response packet we
|
|
* need to send election packets at tighter intervals.
|
|
* Ideally it needs to be the interval (in ms) between time now and
|
|
* the time we are expecting the next netbios packet.
|
|
*/
|
|
|
|
timeout.tv_sec = (run_election||num_response_packets) ? 1 : NMBD_SELECT_LOOP;
|
|
timeout.tv_usec = 0;
|
|
|
|
{
|
|
struct timeval now = timeval_current();
|
|
event_add_to_select_args(nmbd_event_context(), &now,
|
|
&r_fds, &w_fds, &timeout, &maxfd);
|
|
}
|
|
|
|
selrtn = sys_select(maxfd+1,&r_fds,&w_fds,NULL,&timeout);
|
|
|
|
if (run_events(nmbd_event_context(), selrtn, &r_fds, &w_fds)) {
|
|
return False;
|
|
}
|
|
|
|
if (selrtn == -1) {
|
|
return False;
|
|
}
|
|
|
|
#ifndef SYNC_DNS
|
|
if (dns_fd != -1 && FD_ISSET(dns_fd,&r_fds)) {
|
|
run_dns_queue();
|
|
}
|
|
#endif
|
|
|
|
for(i = 0; i < listen_number; i++) {
|
|
enum packet_type packet_type;
|
|
struct packet_struct *packet;
|
|
const char *packet_name;
|
|
int client_fd;
|
|
int client_port;
|
|
|
|
if (sock_array[i] == -1) {
|
|
continue;
|
|
}
|
|
|
|
if (!FD_ISSET(sock_array[i],&r_fds)) {
|
|
continue;
|
|
}
|
|
|
|
if (i < (listen_number/2)) {
|
|
/* Port 137 */
|
|
packet_type = NMB_PACKET;
|
|
packet_name = "nmb";
|
|
client_fd = ClientNMB;
|
|
client_port = global_nmb_port;
|
|
} else {
|
|
/* Port 138 */
|
|
packet_type = DGRAM_PACKET;
|
|
packet_name = "dgram";
|
|
client_fd = ClientDGRAM;
|
|
client_port = DGRAM_PORT;
|
|
}
|
|
|
|
packet = read_packet(sock_array[i], packet_type);
|
|
if (!packet) {
|
|
continue;
|
|
}
|
|
|
|
/*
|
|
* If we got a packet on the broadcast socket and interfaces
|
|
* only is set then check it came from one of our local nets.
|
|
*/
|
|
if (lp_bind_interfaces_only() &&
|
|
(sock_array[i] == client_fd) &&
|
|
(!is_local_net_v4(packet->ip))) {
|
|
DEBUG(7,("discarding %s packet sent to broadcast socket from %s:%d\n",
|
|
packet_name, inet_ntoa(packet->ip), packet->port));
|
|
free_packet(packet);
|
|
continue;
|
|
}
|
|
|
|
if ((is_loopback_ip_v4(packet->ip) || ismyip_v4(packet->ip)) &&
|
|
packet->port == client_port)
|
|
{
|
|
if (client_port == DGRAM_PORT) {
|
|
DEBUG(7,("discarding own dgram packet from %s:%d\n",
|
|
inet_ntoa(packet->ip),packet->port));
|
|
free_packet(packet);
|
|
continue;
|
|
}
|
|
|
|
if (packet->packet.nmb.header.nm_flags.bcast) {
|
|
DEBUG(7,("discarding own nmb bcast packet from %s:%d\n",
|
|
inet_ntoa(packet->ip),packet->port));
|
|
free_packet(packet);
|
|
continue;
|
|
}
|
|
}
|
|
|
|
|
|
if (is_processed_packet(processed_packet_list, packet)) {
|
|
DEBUG(7,("discarding duplicate packet from %s:%d\n",
|
|
inet_ntoa(packet->ip),packet->port));
|
|
free_packet(packet);
|
|
continue;
|
|
}
|
|
|
|
store_processed_packet(&processed_packet_list, packet);
|
|
|
|
/*
|
|
* 0,2,4,... are unicast sockets
|
|
* 1,3,5,... are broadcast sockets
|
|
*
|
|
* on broadcast socket we only receive packets
|
|
* and send replies via the unicast socket.
|
|
*
|
|
* 0,1 and 2,3 and ... belong together.
|
|
*/
|
|
if ((i % 2) != 0) {
|
|
/* this is a broadcast socket */
|
|
packet->send_fd = sock_array[i-1];
|
|
} else {
|
|
/* this is already a unicast socket */
|
|
packet->send_fd = sock_array[i];
|
|
}
|
|
|
|
queue_packet(packet);
|
|
}
|
|
|
|
free_processed_packet_list(&processed_packet_list);
|
|
return False;
|
|
}
|
|
|
|
/****************************************************************************
|
|
Construct and send a netbios DGRAM.
|
|
**************************************************************************/
|
|
|
|
bool send_mailslot(bool unique, const char *mailslot,char *buf, size_t len,
|
|
const char *srcname, int src_type,
|
|
const char *dstname, int dest_type,
|
|
struct in_addr dest_ip,struct in_addr src_ip,
|
|
int dest_port)
|
|
{
|
|
bool loopback_this_packet = False;
|
|
struct packet_struct p;
|
|
struct dgram_packet *dgram = &p.packet.dgram;
|
|
char *ptr,*p2;
|
|
char tmp[4];
|
|
|
|
memset((char *)&p,'\0',sizeof(p));
|
|
|
|
if(ismyip_v4(dest_ip) && (dest_port == DGRAM_PORT)) /* Only if to DGRAM_PORT */
|
|
loopback_this_packet = True;
|
|
|
|
/* generate_name_trn_id(); */ /* Not used, so gone, RJS */
|
|
|
|
/* DIRECT GROUP or UNIQUE datagram. */
|
|
dgram->header.msg_type = unique ? 0x10 : 0x11;
|
|
dgram->header.flags.node_type = M_NODE;
|
|
dgram->header.flags.first = True;
|
|
dgram->header.flags.more = False;
|
|
dgram->header.dgm_id = generate_name_trn_id();
|
|
dgram->header.source_ip = src_ip;
|
|
dgram->header.source_port = DGRAM_PORT;
|
|
dgram->header.dgm_length = 0; /* Let build_dgram() handle this. */
|
|
dgram->header.packet_offset = 0;
|
|
|
|
make_nmb_name(&dgram->source_name,srcname,src_type);
|
|
make_nmb_name(&dgram->dest_name,dstname,dest_type);
|
|
|
|
ptr = &dgram->data[0];
|
|
|
|
/* Setup the smb part. */
|
|
ptr -= 4; /* XXX Ugliness because of handling of tcp SMB length. */
|
|
memcpy(tmp,ptr,4);
|
|
|
|
if (smb_size + 17*2 + strlen(mailslot) + 1 + len > MAX_DGRAM_SIZE) {
|
|
DEBUG(0, ("send_mailslot: Cannot write beyond end of packet\n"));
|
|
return false;
|
|
}
|
|
|
|
cli_set_message(ptr,17,strlen(mailslot) + 1 + len,True);
|
|
memcpy(ptr,tmp,4);
|
|
|
|
SCVAL(ptr,smb_com,SMBtrans);
|
|
SSVAL(ptr,smb_vwv1,len);
|
|
SSVAL(ptr,smb_vwv11,len);
|
|
SSVAL(ptr,smb_vwv12,70 + strlen(mailslot));
|
|
SSVAL(ptr,smb_vwv13,3);
|
|
SSVAL(ptr,smb_vwv14,1);
|
|
SSVAL(ptr,smb_vwv15,1);
|
|
SSVAL(ptr,smb_vwv16,2);
|
|
p2 = smb_buf(ptr);
|
|
safe_strcpy_base(p2, mailslot, dgram->data, sizeof(dgram->data));
|
|
p2 = skip_string(ptr,MAX_DGRAM_SIZE,p2);
|
|
|
|
if (((p2+len) > dgram->data+sizeof(dgram->data)) || ((p2+len) < p2)) {
|
|
DEBUG(0, ("send_mailslot: Cannot write beyond end of packet\n"));
|
|
return False;
|
|
} else {
|
|
if (len) {
|
|
memcpy(p2,buf,len);
|
|
}
|
|
p2 += len;
|
|
}
|
|
|
|
dgram->datasize = PTR_DIFF(p2,ptr+4); /* +4 for tcp length. */
|
|
|
|
p.ip = dest_ip;
|
|
p.port = dest_port;
|
|
p.recv_fd = -1;
|
|
p.send_fd = find_subnet_mailslot_fd_for_address( src_ip );
|
|
p.timestamp = time(NULL);
|
|
p.packet_type = DGRAM_PACKET;
|
|
|
|
DEBUG(4,("send_mailslot: Sending to mailslot %s from %s IP %s ", mailslot,
|
|
nmb_namestr(&dgram->source_name), inet_ntoa(src_ip)));
|
|
DEBUG(4,("to %s IP %s\n", nmb_namestr(&dgram->dest_name), inet_ntoa(dest_ip)));
|
|
|
|
debug_browse_data(buf, len);
|
|
|
|
if(loopback_this_packet) {
|
|
struct packet_struct *lo_packet = NULL;
|
|
DEBUG(5,("send_mailslot: sending packet to ourselves.\n"));
|
|
if((lo_packet = copy_packet(&p)) == NULL)
|
|
return False;
|
|
queue_packet(lo_packet);
|
|
return True;
|
|
} else {
|
|
return(send_packet(&p));
|
|
}
|
|
}
|