f6d1165e576d27cc7118844be7c23eca0ea2d820
[ira/wip.git] / source3 / nmbd / nmbd_synclists.c
1 /* 
2    Unix SMB/Netbios implementation.
3    Version 1.9.
4    NBT netbios routines and daemon - version 2
5    Copyright (C) Andrew Tridgell 1994-1998
6    Copyright (C) Luke Kenneth Casson Leighton 1994-1998
7    Copyright (C) Jeremy Allison 1994-1998
8    
9    This program is free software; you can redistribute it and/or modify
10    it under the terms of the GNU General Public License as published by
11    the Free Software Foundation; either version 2 of the License, or
12    (at your option) any later version.
13    
14    This program is distributed in the hope that it will be useful,
15    but WITHOUT ANY WARRANTY; without even the implied warranty of
16    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
17    GNU General Public License for more details.
18    
19    You should have received a copy of the GNU General Public License
20    along with this program; if not, write to the Free Software
21    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
22    
23 */
24
25 /* this file handles asynchronous browse synchronisation requests. The
26    requests are done by forking and putting the result in a file in the
27    locks directory. We do it this way because we don't want nmbd to be
28    blocked waiting for some server to respond on a TCP connection. This
29    also allows us to have more than 1 sync going at once (tridge) */
30
31 #include "includes.h"
32 #include "smb.h"
33
34 struct sync_record {
35         struct sync_record *next, *prev;
36         fstring workgroup;
37         fstring server;
38         pstring fname;
39         struct in_addr ip;
40         pid_t pid;
41 };
42
43 /* a linked list of current sync connections */
44 static struct sync_record *syncs;
45
46 static XFILE *fp;
47
48 /*******************************************************************
49   This is the NetServerEnum callback.
50   Note sname and comment are in UNIX codepage format.
51   ******************************************************************/
52 static void callback(const char *sname, uint32 stype, 
53                      const char *comment, void *state)
54 {
55         x_fprintf(fp,"\"%s\" %08X \"%s\"\n", sname, stype, comment);
56 }
57
58 /*******************************************************************
59   Synchronise browse lists with another browse server.
60   Log in on the remote server's SMB port to their IPC$ service,
61   do a NetServerEnum and record the results in fname
62 ******************************************************************/
63 static 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         if (!cli_initialise(&cli) || !cli_connect(&cli, name, &ip)) {
75                 return;
76         }
77
78         make_nmb_name(&calling, local_machine, 0x0);
79         make_nmb_name(&called , name         , nm_type);
80
81         if (!cli_session_request(&cli, &calling, &called))
82         {
83                 cli_shutdown(&cli);
84                 return;
85         }
86
87         if (!cli_negprot(&cli)) {
88                 cli_shutdown(&cli);
89                 return;
90         }
91
92         if (!cli_session_setup(&cli, "", "", 1, "", 0, workgroup)) {
93                 cli_shutdown(&cli);
94                 return;
95         }
96
97         if (!cli_send_tconX(&cli, "IPC$", "IPC", "", 1)) {
98                 cli_shutdown(&cli);
99                 return;
100         }
101
102         /* All the cli_XX functions take UNIX character set. */
103         fstrcpy(unix_workgroup, cli.server_domain?cli.server_domain:workgroup);
104
105         /* Fetch a workgroup list. */
106         cli_NetServerEnum(&cli, unix_workgroup,
107                           local_type|SV_TYPE_DOMAIN_ENUM, 
108                           callback, NULL);
109         
110         /* Now fetch a server list. */
111         if (servers) {
112                 fstrcpy(unix_workgroup, workgroup);
113                 cli_NetServerEnum(&cli, unix_workgroup, 
114                                   local?SV_TYPE_LOCAL_LIST_ONLY:SV_TYPE_ALL,
115                                   callback, NULL);
116         }
117         
118         cli_shutdown(&cli);
119 }
120
121
122 /*******************************************************************
123   initialise a browse sync with another browse server.  Log in on the
124   remote server's SMB port to their IPC$ service, do a NetServerEnum
125   and record the results
126 ******************************************************************/
127 void sync_browse_lists(struct work_record *work,
128                        char *name, int nm_type, 
129                        struct in_addr ip, BOOL local, BOOL servers)
130 {
131         struct sync_record *s;
132         static int counter;
133
134         START_PROFILE(sync_browse_lists);
135         /* Check we're not trying to sync with ourselves. This can
136            happen if we are a domain *and* a local master browser. */
137         if (ismyip(ip)) {
138 done:
139                 END_PROFILE(sync_browse_lists);
140                 return;
141         }
142
143         s = (struct sync_record *)malloc(sizeof(*s));
144         if (!s) goto done;
145
146         ZERO_STRUCTP(s);
147         
148         fstrcpy(s->workgroup, work->work_group);
149         fstrcpy(s->server, name);
150         s->ip = ip;
151
152         slprintf(s->fname, sizeof(pstring)-1,
153                  "%s/sync.%d", lp_lockdir(), counter++);
154         all_string_sub(s->fname,"//", "/", 0);
155         
156         DLIST_ADD(syncs, s);
157
158         /* the parent forks and returns, leaving the child to do the
159            actual sync and call END_PROFILE*/
160         CatchChild();
161         if ((s->pid = sys_fork())) return;
162
163         BlockSignals( False, SIGTERM );
164
165         DEBUG(2,("Initiating browse sync for %s to %s(%s)\n",
166                  work->work_group, name, inet_ntoa(ip)));
167
168         fp = x_fopen(s->fname,O_WRONLY|O_CREAT|O_TRUNC, 0644);
169         if (!fp) {
170                 END_PROFILE(sync_browse_lists);
171                 _exit(1);       
172         }
173
174         sync_child(name, nm_type, work->work_group, ip, local, servers,
175                    s->fname);
176
177         x_fclose(fp);
178         END_PROFILE(sync_browse_lists);
179         _exit(0);
180 }
181
182 /**********************************************************************
183 handle one line from a completed sync file
184  **********************************************************************/
185 static void complete_one(struct sync_record *s, 
186                          char *sname, uint32 stype, char *comment)
187 {
188         struct work_record *work;
189         struct server_record *servrec;
190
191         stype &= ~SV_TYPE_LOCAL_LIST_ONLY;
192
193         if (stype & SV_TYPE_DOMAIN_ENUM) {
194                 /* See if we can find the workgroup on this subnet. */
195                 if((work=find_workgroup_on_subnet(unicast_subnet, sname))) {
196                         /* We already know about this workgroup -
197                            update the ttl. */
198                         update_workgroup_ttl(work,lp_max_ttl());
199                 } else {
200                         /* Create the workgroup on the subnet. */
201                         work = create_workgroup_on_subnet(unicast_subnet, 
202                                                           sname, lp_max_ttl());
203                         if (work) {
204                                 /* remember who the master is */
205                                 fstrcpy(work->local_master_browser_name, 
206                                         comment);
207                         }
208                 }
209                 return;
210         } 
211
212         work = find_workgroup_on_subnet(unicast_subnet, s->workgroup);
213         if (!work) {
214                 DEBUG(3,("workgroup %s doesn't exist on unicast subnet?\n",
215                          s->workgroup));
216                 return;
217         }
218
219         if ((servrec = find_server_in_workgroup( work, sname))) {
220                 /* Check that this is not a locally known
221                    server - if so ignore the entry. */
222                 if(!(servrec->serv.type & SV_TYPE_LOCAL_LIST_ONLY)) {
223                         /* We already know about this server - update
224                            the ttl. */
225                         update_server_ttl(servrec, lp_max_ttl());
226                         /* Update the type. */
227                         servrec->serv.type = stype;
228                 }
229                 return;
230         } 
231
232         /* Create the server in the workgroup. */ 
233         create_server_on_workgroup(work, sname,stype, lp_max_ttl(), comment);
234 }
235                 
236
237 /**********************************************************************
238 read the completed sync info
239  **********************************************************************/
240 static void complete_sync(struct sync_record *s)
241 {
242         XFILE *f;
243         fstring server, type_str;
244         unsigned type;
245         pstring comment;
246         pstring line;
247         char *ptr;
248         int count=0;
249
250         f = x_fopen(s->fname,O_RDONLY, 0);
251
252         if (!f) return;
253         
254         while (!x_feof(f)) {
255                 
256                 if (!fgets_slash(line,sizeof(pstring),f)) continue;
257                 
258                 ptr = line;
259
260                 if (!next_token(&ptr,server,NULL,sizeof(server)) ||
261                     !next_token(&ptr,type_str,NULL, sizeof(type_str)) ||
262                     !next_token(&ptr,comment,NULL, sizeof(comment))) {
263                         continue;
264                 }
265
266                 sscanf(type_str, "%X", &type);
267
268                 complete_one(s, server, type, comment);
269
270                 count++;
271         }
272
273         x_fclose(f);
274
275         unlink(s->fname);
276
277         DEBUG(2,("sync with %s(%s) for workgroup %s completed (%d records)\n",
278                  s->server, inet_ntoa(s->ip), s->workgroup, count));
279 }
280
281 /**********************************************************************
282 check for completion of any of the child processes
283  **********************************************************************/
284 void sync_check_completion(void)
285 {
286         struct sync_record *s, *next;
287
288         for (s=syncs;s;s=next) {
289                 next = s->next;
290                 if (!process_exists(s->pid)) {
291                         /* it has completed - grab the info */
292                         complete_sync(s);
293                         DLIST_REMOVE(syncs, s);
294                         ZERO_STRUCTP(s);
295                         SAFE_FREE(s);
296                 }
297         }
298 }