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