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-1998
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/* this file handles asynchronous browse synchronisation requests. The
25   requests are done by forking and putting the result in a file in the
26   locks directory. We do it this way because we don't want nmbd to be
27   blocked waiting for some server to respond on a TCP connection. This
28   also allows us to have more than 1 sync going at once (tridge) */
29
30#include "includes.h"
31
32struct sync_record {
33	struct sync_record *next, *prev;
34	nstring workgroup;
35	nstring server;
36	pstring fname;
37	struct in_addr ip;
38	pid_t pid;
39};
40
41/* a linked list of current sync connections */
42static struct sync_record *syncs;
43
44static XFILE *fp;
45
46/*******************************************************************
47  This is the NetServerEnum callback.
48  Note sname and comment are in UNIX codepage format.
49  ******************************************************************/
50
51static void callback(const char *sname, uint32 stype,
52                     const char *comment, void *state)
53{
54	x_fprintf(fp,"\"%s\" %08X \"%s\"\n", sname, stype, comment);
55}
56
57/*******************************************************************
58  Synchronise browse lists with another browse server.
59  Log in on the remote server's SMB port to their IPC$ service,
60  do a NetServerEnum and record the results in fname
61******************************************************************/
62
63static void sync_child(char *name, int nm_type,
64		       char *workgroup,
65		       struct in_addr ip, BOOL local, BOOL servers,
66		       char *fname)
67{
68	extern fstring local_machine;
69	fstring unix_workgroup;
70	static struct cli_state cli;
71	uint32 local_type = local ? SV_TYPE_LOCAL_LIST_ONLY : 0;
72	struct nmb_name called, calling;
73
74	/* W2K DMB's return empty browse lists on port 445. Use 139.
75	 * Patch from Andy Levine andyl@epicrealm.com.
76	 */
77
78	if (!cli_initialise(&cli) || !cli_set_port(&cli, 139) || !cli_connect(&cli, name, &ip)) {
79		return;
80	}
81
82	make_nmb_name(&calling, local_machine, 0x0);
83	make_nmb_name(&called , name, nm_type);
84
85	if (!cli_session_request(&cli, &calling, &called)) {
86		cli_shutdown(&cli);
87		return;
88	}
89
90	if (!cli_negprot(&cli)) {
91		cli_shutdown(&cli);
92		return;
93	}
94
95	if (!cli_session_setup(&cli, "", "", 1, "", 0, workgroup)) {
96		cli_shutdown(&cli);
97		return;
98	}
99
100	if (!cli_send_tconX(&cli, "IPC$", "IPC", "", 1)) {
101		cli_shutdown(&cli);
102		return;
103	}
104
105	/* All the cli_XX functions take UNIX character set. */
106	fstrcpy(unix_workgroup, cli.server_domain?cli.server_domain:workgroup);
107
108	/* Fetch a workgroup list. */
109	cli_NetServerEnum(&cli, unix_workgroup,
110			  local_type|SV_TYPE_DOMAIN_ENUM,
111			  callback, NULL);
112
113	/* Now fetch a server list. */
114	if (servers) {
115		fstrcpy(unix_workgroup, workgroup);
116		cli_NetServerEnum(&cli, unix_workgroup,
117				  local?SV_TYPE_LOCAL_LIST_ONLY:SV_TYPE_ALL,
118				  callback, NULL);
119	}
120
121	cli_shutdown(&cli);
122}
123
124/*******************************************************************
125  initialise a browse sync with another browse server.  Log in on the
126  remote server's SMB port to their IPC$ service, do a NetServerEnum
127  and record the results
128******************************************************************/
129
130void sync_browse_lists(struct work_record *work,
131		       char *name, int nm_type,
132		       struct in_addr ip, BOOL local, BOOL servers)
133{
134	struct sync_record *s;
135	static int counter;
136
137	START_PROFILE(sync_browse_lists);
138	/* Check we're not trying to sync with ourselves. This can
139	   happen if we are a domain *and* a local master browser. */
140	if (ismyip(ip)) {
141done:
142		END_PROFILE(sync_browse_lists);
143		return;
144	}
145
146	s = (struct sync_record *)malloc(sizeof(*s));
147	if (!s) goto done;
148
149	ZERO_STRUCTP(s);
150
151	nstrcpy(s->workgroup, work->work_group);
152	nstrcpy(s->server, name);
153	s->ip = ip;
154
155	slprintf(s->fname, sizeof(pstring)-1,
156		 "%s/sync.%d", lp_lockdir(), counter++);
157	all_string_sub(s->fname,"//", "/", 0);
158
159	DLIST_ADD(syncs, s);
160
161	/* the parent forks and returns, leaving the child to do the
162	   actual sync and call END_PROFILE*/
163	CatchChild();
164	if ((s->pid = sys_fork())) return;
165
166	BlockSignals( False, SIGTERM );
167
168	DEBUG(2,("Initiating browse sync for %s to %s(%s)\n",
169		 work->work_group, name, inet_ntoa(ip)));
170
171	fp = x_fopen(s->fname,O_WRONLY|O_CREAT|O_TRUNC, 0644);
172	if (!fp) {
173		END_PROFILE(sync_browse_lists);
174		_exit(1);
175	}
176
177	sync_child(name, nm_type, work->work_group, ip, local, servers,
178		   s->fname);
179
180	x_fclose(fp);
181	END_PROFILE(sync_browse_lists);
182	_exit(0);
183}
184
185/**********************************************************************
186 Handle one line from a completed sync file.
187 **********************************************************************/
188
189static void complete_one(struct sync_record *s,
190			 char *sname, uint32 stype, char *comment)
191{
192	struct work_record *work;
193	struct server_record *servrec;
194
195	stype &= ~SV_TYPE_LOCAL_LIST_ONLY;
196
197	if (stype & SV_TYPE_DOMAIN_ENUM) {
198		/* See if we can find the workgroup on this subnet. */
199		if((work=find_workgroup_on_subnet(unicast_subnet, sname))) {
200			/* We already know about this workgroup -
201                           update the ttl. */
202			update_workgroup_ttl(work,lp_max_ttl());
203		} else {
204			/* Create the workgroup on the subnet. */
205			work = create_workgroup_on_subnet(unicast_subnet,
206							  sname, lp_max_ttl());
207			if (work) {
208				/* remember who the master is */
209				nstrcpy(work->local_master_browser_name, comment);
210			}
211		}
212		return;
213	}
214
215	work = find_workgroup_on_subnet(unicast_subnet, s->workgroup);
216	if (!work) {
217		DEBUG(3,("workgroup %s doesn't exist on unicast subnet?\n",
218			 s->workgroup));
219		return;
220	}
221
222	if ((servrec = find_server_in_workgroup( work, sname))) {
223		/* Check that this is not a locally known
224		   server - if so ignore the entry. */
225		if(!(servrec->serv.type & SV_TYPE_LOCAL_LIST_ONLY)) {
226			/* We already know about this server - update
227                           the ttl. */
228			update_server_ttl(servrec, lp_max_ttl());
229			/* Update the type. */
230			servrec->serv.type = stype;
231		}
232		return;
233	}
234
235	/* Create the server in the workgroup. */
236	create_server_on_workgroup(work, sname,stype, lp_max_ttl(), comment);
237}
238
239/**********************************************************************
240 Read the completed sync info.
241**********************************************************************/
242
243static void complete_sync(struct sync_record *s)
244{
245	XFILE *f;
246	fstring server, type_str;
247	unsigned type;
248	pstring comment;
249	pstring line;
250	const char *ptr;
251	int count=0;
252
253	f = x_fopen(s->fname,O_RDONLY, 0);
254
255	if (!f)
256		return;
257
258	while (!x_feof(f)) {
259
260		if (!fgets_slash(line,sizeof(pstring),f))
261			continue;
262
263		ptr = line;
264
265		if (!next_token(&ptr,server,NULL,sizeof(server)) ||
266		    !next_token(&ptr,type_str,NULL, sizeof(type_str)) ||
267		    !next_token(&ptr,comment,NULL, sizeof(comment))) {
268			continue;
269		}
270
271		sscanf(type_str, "%X", &type);
272
273		complete_one(s, server, type, comment);
274
275		count++;
276	}
277
278	x_fclose(f);
279
280	unlink(s->fname);
281
282	DEBUG(2,("sync with %s(%s) for workgroup %s completed (%d records)\n",
283		 s->server, inet_ntoa(s->ip), s->workgroup, count));
284}
285
286/**********************************************************************
287 Check for completion of any of the child processes.
288**********************************************************************/
289
290void sync_check_completion(void)
291{
292	struct sync_record *s, *next;
293
294	for (s=syncs;s;s=next) {
295		next = s->next;
296		if (!process_exists(s->pid)) {
297			/* it has completed - grab the info */
298			complete_sync(s);
299			DLIST_REMOVE(syncs, s);
300			ZERO_STRUCTP(s);
301			SAFE_FREE(s);
302		}
303	}
304}
305