1/*
2   Unix SMB/CIFS implementation.
3   NBT netbios routines and daemon - version 2
4   Copyright (C) Andrew Tridgell 1994-1998
5   Copyright (C) Luke Kenneth Casson Leighton 1994-1998
6   Copyright (C) Jeremy Allison 1994-2003
7
8   This program is free software; you can redistribute it and/or modify
9   it under the terms of the GNU General Public License as published by
10   the Free Software Foundation; either version 2 of the License, or
11   (at your option) any later version.
12
13   This program is distributed in the hope that it will be useful,
14   but WITHOUT ANY WARRANTY; without even the implied warranty of
15   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
16   GNU General Public License for more details.
17
18   You should have received a copy of the GNU General Public License
19   along with this program; if not, write to the Free Software
20   Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
21
22*/
23
24#include "includes.h"
25
26extern uint16 samba_nb_type; /* Samba's NetBIOS type. */
27
28/****************************************************************************
29 Fail funtion when registering my netbios names.
30**************************************************************************/
31
32static void my_name_register_failed(struct subnet_record *subrec,
33                              struct response_record *rrec, struct nmb_name *nmbname)
34{
35	DEBUG(0,("my_name_register_failed: Failed to register my name %s on subnet %s.\n",
36		nmb_namestr(nmbname), subrec->subnet_name));
37}
38
39
40/****************************************************************************
41  Add my workgroup and my given names to one subnet
42  Also add the magic Samba names.
43**************************************************************************/
44
45void register_my_workgroup_one_subnet(struct subnet_record *subrec)
46{
47	int i;
48
49	struct work_record *work;
50
51	/* Create the workgroup on the subnet. */
52	if((work = create_workgroup_on_subnet(subrec, lp_workgroup(),
53					      PERMANENT_TTL)) == NULL) {
54		DEBUG(0,("register_my_workgroup_and_names: Failed to create my workgroup %s on subnet %s. \
55Exiting.\n", lp_workgroup(), subrec->subnet_name));
56		return;
57	}
58
59	/* Each subnet entry, except for the wins_server_subnet has
60           the magic Samba names. */
61	add_samba_names_to_subnet(subrec);
62
63	/* Register all our names including aliases. */
64	for (i=0; my_netbios_names(i); i++) {
65		register_name(subrec, my_netbios_names(i),0x20,samba_nb_type,
66			      NULL,
67			      my_name_register_failed, NULL);
68		register_name(subrec, my_netbios_names(i),0x03,samba_nb_type,
69			      NULL,
70			      my_name_register_failed, NULL);
71		register_name(subrec, my_netbios_names(i),0x00,samba_nb_type,
72			      NULL,
73			      my_name_register_failed, NULL);
74	}
75
76	/* Initiate election processing, register the workgroup names etc. */
77	initiate_myworkgroup_startup(subrec, work);
78}
79
80/*******************************************************************
81 Utility function to add a name to the unicast subnet, or add in
82 our IP address if it already exists.
83******************************************************************/
84
85static void insert_refresh_name_into_unicast( struct subnet_record *subrec,
86                                                struct nmb_name *nmbname, uint16 nb_type )
87{
88	struct name_record *namerec;
89
90	if (!we_are_a_wins_client()) {
91		insert_permanent_name_into_unicast(subrec, nmbname, nb_type);
92		return;
93	}
94
95	if((namerec = find_name_on_subnet(unicast_subnet, nmbname, FIND_SELF_NAME)) == NULL) {
96		unstring name;
97		pull_ascii_nstring(name, sizeof(name), nmbname->name);
98		/* The name needs to be created on the unicast subnet. */
99		(void)add_name_to_subnet( unicast_subnet, name,
100				nmbname->name_type, nb_type,
101				MIN(lp_max_ttl(), MAX_REFRESH_TIME), SELF_NAME, 1, &subrec->myip);
102	} else {
103		/* The name already exists on the unicast subnet. Add our local
104			IP for the given broadcast subnet to the name. */
105		add_ip_to_name_record( namerec, subrec->myip);
106	}
107}
108
109/****************************************************************************
110  Add my workgroup and my given names to the subnet lists.
111  Also add the magic Samba names.
112**************************************************************************/
113
114BOOL register_my_workgroup_and_names(void)
115{
116	struct subnet_record *subrec;
117	int i;
118
119	for(subrec = FIRST_SUBNET; subrec; subrec = NEXT_SUBNET_INCLUDING_UNICAST(subrec)) {
120		register_my_workgroup_one_subnet(subrec);
121	}
122
123	/* We still need to add the magic Samba
124		names and the netbios names to the unicast subnet directly. This is
125		to allow unicast node status requests and queries to still work
126		in a broadcast only environment. */
127
128	add_samba_names_to_subnet(unicast_subnet);
129
130	for (i=0; my_netbios_names(i); i++) {
131		for(subrec = FIRST_SUBNET; subrec; subrec = NEXT_SUBNET_EXCLUDING_UNICAST(subrec)) {
132			/*
133			 * Ensure all the IP addresses are added if we are multihomed.
134			 */
135			struct nmb_name nmbname;
136
137			make_nmb_name(&nmbname, my_netbios_names(i),0x20);
138			insert_refresh_name_into_unicast(subrec, &nmbname, samba_nb_type);
139
140			make_nmb_name(&nmbname, my_netbios_names(i),0x3);
141			insert_refresh_name_into_unicast(subrec, &nmbname, samba_nb_type);
142
143			make_nmb_name(&nmbname, my_netbios_names(i),0x0);
144			insert_refresh_name_into_unicast(subrec, &nmbname, samba_nb_type);
145		}
146	}
147
148	/*
149	 * Add the WORKGROUP<0> and WORKGROUP<1e> group names to the unicast subnet
150	 * also for the same reasons.
151	 */
152
153	for(subrec = FIRST_SUBNET; subrec; subrec = NEXT_SUBNET_EXCLUDING_UNICAST(subrec)) {
154		/*
155		 * Ensure all the IP addresses are added if we are multihomed.
156		 */
157		struct nmb_name nmbname;
158
159		make_nmb_name(&nmbname, lp_workgroup(), 0x0);
160		insert_refresh_name_into_unicast(subrec, &nmbname, samba_nb_type|NB_GROUP);
161
162		make_nmb_name(&nmbname, lp_workgroup(), 0x1e);
163		insert_refresh_name_into_unicast(subrec, &nmbname, samba_nb_type|NB_GROUP);
164	}
165
166	/*
167	 * We need to add the Samba names to the remote broadcast subnet,
168	 * as NT 4.x does directed broadcast requests to the *<0x0> name.
169	 */
170
171	add_samba_names_to_subnet(remote_broadcast_subnet);
172
173	return True;
174}
175
176/****************************************************************************
177  Remove all the names we registered.
178**************************************************************************/
179
180void release_wins_names(void)
181{
182	struct subnet_record *subrec = unicast_subnet;
183	struct name_record *namerec, *nextnamerec;
184
185	for (namerec = (struct name_record *)ubi_trFirst( subrec->namelist ); namerec; namerec = nextnamerec) {
186		nextnamerec = (struct name_record *)ubi_trNext( namerec );
187		if( (namerec->data.source == SELF_NAME)
188		    && !NAME_IS_DEREGISTERING(namerec) )
189			release_name( subrec, namerec, standard_success_release,
190				      NULL, NULL);
191	}
192}
193
194/*******************************************************************
195  Refresh our registered names with WINS
196******************************************************************/
197
198void refresh_my_names(time_t t)
199{
200	struct name_record *namerec;
201
202	if (wins_srv_count() < 1)
203		return;
204
205	for (namerec = (struct name_record *)ubi_trFirst(unicast_subnet->namelist);
206	     namerec;
207	     namerec = (struct name_record *)ubi_trNext(namerec)) {
208		/* Each SELF name has an individual time to be refreshed. */
209		if ((namerec->data.source == SELF_NAME) &&
210		    (namerec->data.refresh_time < t) &&
211		    (namerec->data.death_time != PERMANENT_TTL)) {
212			/* We cheat here and pretend the refresh is going to be
213			   successful & update the refresh times. This stops
214			   multiple refresh calls being done. We actually
215			   deal with refresh failure in the fail_fn.
216			*/
217			if (!is_refresh_already_queued(unicast_subnet, namerec)) {
218				wins_refresh_name(namerec);
219			}
220			namerec->data.death_time = t + lp_max_ttl();
221			namerec->data.refresh_time = t + MIN(lp_max_ttl()/2, MAX_REFRESH_TIME);
222		}
223	}
224}
225