r17571: Change the return code of cli_session_setup from BOOL to NTSTATUS
[samba.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 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
32 extern fstring local_machine;
33
34 struct 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 */
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
53 static 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
65 static 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
137 void 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)) {
148 done:
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
196 static 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
250 static 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
297 void 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 }