first pass at updating head branch to be to be the same as the SAMBA_2_0 branch
[ira/wip.git] / source / 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 extern int DEBUGLEVEL;
35 extern pstring scope;
36
37 struct sync_record {
38         struct sync_record *next, *prev;
39         fstring workgroup;
40         fstring server;
41         pstring fname;
42         struct in_addr ip;
43         pid_t pid;
44 };
45
46 /* a linked list of current sync connections */
47 static struct sync_record *syncs;
48
49 static FILE *fp;
50
51 /*******************************************************************
52   This is the NetServerEnum callback.
53   ******************************************************************/
54 static void callback(const char *sname, uint32 stype, const char *comment)
55 {
56         fprintf(fp,"\"%s\" %08X \"%s\"\n", sname, stype, comment);
57 }
58
59
60 /*******************************************************************
61   Synchronise browse lists with another browse server.
62   Log in on the remote server's SMB port to their IPC$ service,
63   do a NetServerEnum and record the results in fname
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         extern fstring local_machine;
71         static struct cli_state cli;
72         uint32 local_type = local ? SV_TYPE_LOCAL_LIST_ONLY : 0;
73         struct nmb_name called, calling;
74
75         if (!cli_initialise(&cli) || !cli_connect(&cli, name, &ip)) {
76                 fclose(fp);
77                 return;
78         }
79
80         make_nmb_name(&calling, local_machine, 0x0    , scope);
81         make_nmb_name(&called , name         , nm_type, scope);
82
83         if (!cli_session_request(&cli, &calling, &called))
84         {
85                 cli_shutdown(&cli);
86                 fclose(fp);
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         /* Fetch a workgroup list. */
106         cli_NetServerEnum(&cli, cli.server_domain?cli.server_domain:workgroup, 
107                           local_type|SV_TYPE_DOMAIN_ENUM,
108                           callback);
109         
110         /* Now fetch a server list. */
111         if (servers) {
112                 cli_NetServerEnum(&cli, workgroup, 
113                                   local?SV_TYPE_LOCAL_LIST_ONLY:SV_TYPE_ALL,
114                                   callback);
115         }
116         
117         cli_shutdown(&cli);
118 }
119
120
121 /*******************************************************************
122   initialise a browse sync with another browse server.  Log in on the
123   remote server's SMB port to their IPC$ service, do a NetServerEnum
124   and record the results
125 ******************************************************************/
126 void sync_browse_lists(struct work_record *work,
127                        char *name, int nm_type, 
128                        struct in_addr ip, BOOL local, BOOL servers)
129 {
130         struct sync_record *s;
131         static int counter;
132
133         /* Check we're not trying to sync with ourselves. This can
134            happen if we are a domain *and* a local master browser. */
135         if (ismyip(ip)) {
136                 return;
137         }
138
139         s = (struct sync_record *)malloc(sizeof(*s));
140         if (!s) return;
141
142         ZERO_STRUCTP(s);
143         
144         fstrcpy(s->workgroup, work->work_group);
145         fstrcpy(s->server, name);
146         s->ip = ip;
147
148         slprintf(s->fname, sizeof(pstring)-1,
149                  "%s/sync.%d", lp_lockdir(), counter++);
150         all_string_sub(s->fname,"//", "/", 0);
151         
152         DLIST_ADD(syncs, s);
153
154         /* the parent forks and returns, leaving the child to do the
155            actual sync */
156         CatchChild();
157         if ((s->pid = fork())) return;
158
159         BlockSignals( False, SIGTERM );
160
161         DEBUG(2,("Initiating browse sync for %s to %s(%s)\n",
162                  work->work_group, name, inet_ntoa(ip)));
163
164         fp = sys_fopen(s->fname,"w");
165         if (!fp) _exit(1);      
166
167         sync_child(name, nm_type, work->work_group, ip, local, servers,
168                    s->fname);
169
170         fclose(fp);
171         _exit(0);
172 }
173
174 /**********************************************************************
175 handle one line from a completed sync file
176  **********************************************************************/
177 static void complete_one(struct sync_record *s, 
178                          char *sname, uint32 stype, char *comment)
179 {
180         struct work_record *work;
181         struct server_record *servrec;
182
183         stype &= ~SV_TYPE_LOCAL_LIST_ONLY;
184
185         if (stype & SV_TYPE_DOMAIN_ENUM) {
186                 /* See if we can find the workgroup on this subnet. */
187                 if((work=find_workgroup_on_subnet(unicast_subnet, sname))) {
188                         /* We already know about this workgroup -
189                            update the ttl. */
190                         update_workgroup_ttl(work,lp_max_ttl());
191                 } else {
192                         /* Create the workgroup on the subnet. */
193                         work = create_workgroup_on_subnet(unicast_subnet, 
194                                                           sname, lp_max_ttl());
195                         if (work) {
196                                 /* remember who the master is */
197                                 fstrcpy(work->local_master_browser_name, 
198                                         comment);
199                         }
200                 }
201                 return;
202         } 
203
204         work = find_workgroup_on_subnet(unicast_subnet, s->workgroup);
205         if (!work) {
206                 DEBUG(3,("workgroup %s doesn't exist on unicast subnet?\n",
207                          s->workgroup));
208                 return;
209         }
210
211         if ((servrec = find_server_in_workgroup( work, sname))) {
212                 /* Check that this is not a locally known
213                    server - if so ignore the entry. */
214                 if(!(servrec->serv.type & SV_TYPE_LOCAL_LIST_ONLY)) {
215                         /* We already know about this server - update
216                            the ttl. */
217                         update_server_ttl(servrec, lp_max_ttl());
218                         /* Update the type. */
219                         servrec->serv.type = stype;
220                 }
221                 return;
222         } 
223
224         /* Create the server in the workgroup. */ 
225         create_server_on_workgroup(work, sname,stype, lp_max_ttl(), comment);
226 }
227                 
228
229 /**********************************************************************
230 read the completed sync info
231  **********************************************************************/
232 static void complete_sync(struct sync_record *s)
233 {
234         FILE *f;
235         fstring server, type_str;
236         unsigned type;
237         pstring comment;
238         pstring line;
239         char *ptr;
240         int count=0;
241
242         f = sys_fopen(s->fname,"r");
243
244         if (!f) return;
245         
246         while (!feof(f)) {
247                 
248                 if (!fgets_slash(line,sizeof(pstring),f)) continue;
249                 
250                 ptr = line;
251
252                 if (!next_token(&ptr,server,NULL,sizeof(server)) ||
253                     !next_token(&ptr,type_str,NULL, sizeof(type_str)) ||
254                     !next_token(&ptr,comment,NULL, sizeof(comment))) {
255                         continue;
256                 }
257
258                 sscanf(type_str, "%X", &type);
259
260                 complete_one(s, server, type, comment);
261
262                 count++;
263         }
264
265         fclose(f);
266
267         unlink(s->fname);
268
269         DEBUG(2,("sync with %s(%s) for workgroup %s completed (%d records)\n",
270                  s->server, inet_ntoa(s->ip), s->workgroup, count));
271 }
272
273 /**********************************************************************
274 check for completion of any of the child processes
275  **********************************************************************/
276 void sync_check_completion(void)
277 {
278         struct sync_record *s, *next;
279
280         for (s=syncs;s;s=next) {
281                 next = s->next;
282                 if (!process_exists(s->pid)) {
283                         /* it has completed - grab the info */
284                         complete_sync(s);
285                         DLIST_REMOVE(syncs, s);
286                         ZERO_STRUCTP(s);
287                         free(s);
288                 }
289         }
290 }