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