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