source: branches/guitools-2.0/shared/smbtree.vrs@ 570

Last change on this file since 570 was 570, checked in by Herwig Bauernfeind, 14 years ago

GUI-Tools: EVFSGUI 2.1: Fix Ticket #170
File size: 19.8 KB
Line 
1/* Routines to handle smbtree output */
2
3/*:VRX */
4_RefreshTree:
5 say time()' _RefreshTree() started'
6 FirstRun = 0
7
8 ok = SysFileDelete(samba.!msg)
9
10 if UserCred = 'USERCRED' | UserCred = '' | UserCred = '--user=%' then UserCred = '-N'
11 if ShowHidden = 'SHOWHIDDEN' | ShowHidden = '' then ShowHidden = 0
12
13 if BroadCast = 1 then BroadCast = '-b'; else BroadCast = ''
14
15 say ' detach 'samba.!smbtreeexe' 'BroadCast' -S 'debuglevel' 'UserCred' >'samba.!msg
16 address cmd 'detach 'samba.!smbtreeexe' 'BroadCast' -S 'debuglevel' 'UserCred' >'samba.!msg
17
18 if BroadCast = '-b' then BroadCast = 1; else BroadCast = 0
19 if UserCred = '-N' then UserCred = ''
20
21 ok = VRMethod("CN_smbtree", "RemoveRecord", "ALL")
22
23 RefreshMode = "TREE"
24 ok = VRSet("CN_smbtree","Enabled", 0)
25 ok = VRset("TM_RefreshTreeDisplay","Enabled",1)
26
27 say time()' _RefreshTree() done'
28return
29
30/*:VRX */
31_RefreshTreeDisplay:
32 say time()' _RefreshTreeDisplay() started'
33 say ' samba.!msg = "'samba.!msg'"'
34 stat = stream(samba.!msg,'c','open read')
35 if stat <> "READY:" then return
36
37 if UserCred = 'USERCRED' | UserCred = '' | UserCred = '--user=%' then do
38 UserCred = '-N'
39 end
40
41 if ShowHidden = 'SHOWHIDDEN' | ShowHidden = '' then ShowHidden = 0
42
43 ok = VRset("TM_RefreshTreeDisplay","Enabled",0)
44/* ok = VRSet("Main", "StatusText", NLVGetMessage(50)) */
45 ok = VRSet("CN_smbtree", 'Enabled', 0 )
46/* ok = VRSet("Main", 'Pointer', 'Wait' ) */
47
48 ok = VRSet( "CN_smbtree", "Painting", 0 )
49 ok = VRMethod("CN_smbtree", "RemoveRecord", "ALL")
50
51 drop smbtree.
52 drop smbtreeline.
53
54 sl = 0
55
56 do until lines(samba.!msg) = 0
57 sl = sl + 1
58 smbtreeline.sl = linein(samba.!msg)
59 end
60 smbtreeline.0 = sl
61 ok = stream(samba.!msg,'c','close')
62 ok = SysFileDelete(samba.!msg)
63
64 do sl = 1 to smbtreeline.0
65 Header = c2x(left(smbtreeline.sl,3))
66 select
67 when Header = "09095C" then do /* share - obsolete, we do that differently now see below! */
68 smbtreeline.sl = strip(smbtreeline.sl,,'09'x)
69 parse var smbtreeline.sl '\\'machine'\'share '09'x comment
70 machine = strip(machine)
71 share = strip(share)
72 comment = strip(comment)
73 text = translate(share' 'comment)
74 res = _GuessIcon(text)
75 if VRGet("CN_smbtree","View") = "IconTree" then do
76 parent = smbtree.!machine
77 smbtree.!share = VRMethod( "CN_smbtree", "AddRecord",parent,, share||'0D0A'x||comment, res)
78 ok = VRMethod( "CN_smbtree", "SetRecordAttr", smbtree.!share, "ReadOnly", 1)
79 if pos("$", share) > 0 then ok = VRMethod( "CN_smbtree", "SetRecordAttr", smbtree.!share, "Visible", ShowHidden)
80 end
81 end /* end of obsolete share code */
82 when Header = "095C5C" then do /* Machine */
83 smbtreeline.sl = strip(smbtreeline.sl,,'09'x)
84 parse var smbtreeline.sl '\\'machine '0909'x comment
85 machine = strip(machine)
86 comment = strip(comment)
87 if VRGet("CN_smbtree","View") = "IconTree" then parent = smbtree.!workgroup; else parent = ""
88 smbtree.!machine = VRMethod( "CN_smbtree", "AddRecord",parent,, machine||'0D0A'x||comment)
89 /* We make any machine as sleeping initially */
90 ok = VRMethod( "CN_smbtree", "SetRecordAttr", smbtree.!machine, "Icon","#61:PMWP.DLL")
91 ok = VRMethod( "CN_smbtree", "SetRecordAttr", smbtree.!machine, "ReadOnly", 1)
92 ok = VRMethod( "CN_smbtree", "SetFieldData", smbtree.!machine, NBFH, machine, CommentFH, Comment, WorkGroupFH, CurWG)
93 ok = VRSet( "CN_smbtree", "Painting", 1 )
94 ok = VRSet( "CN_smbtree", "Painting", 0 )
95 if VRGet("CN_smbtree","View") = "Detail" then do
96 address cmd samba.!nmblookupexe' 'machine' 'debuglevel' -N >'samba.!msg
97 ipstr = ""
98 ip = ""
99 do until lines(samba.!msg) = 0
100 nmblookupline = linein(samba.!msg)
101 if pos(strip(machine)'<',nmblookupline) > 0 then do
102 parse var nmblookupline ip .
103 if pos(strip(ip), ipstr) = 0 then do
104 ipstr = ipstr||ip','
105 end
106 end
107 end
108 ok = SysFileDelete(samba.!msg)
109 ipstr = strip(ipstr,,',')
110 ok = VRMethod( "CN_smbtree", "SetFieldData", smbtree.!machine, IPFH, IPStr)
111 ok = stream(samba.!msg,'c','close')
112 if ip <> "" then do
113 address cmd samba.!smbclientexe' -L "'strip(machine)'" -I "'ip'" -N 'debuglevel' 2>'samba.!msg' 1>NUL'
114 smbline = linein(samba.!msg)
115 ok = stream(samba.!msg,'c','close')
116 parse var smbline "Domain=["WorkGroup"] OS=["OS"] Server=["Server"]"Rest
117 ok = VRMethod( "CN_smbtree", "SetFieldData", smbtree.!machine, OSFH, OS)
118 /* WorkGroupFH, Workgroup */
119 address cmd samba.!nmblookupexe' -A 'machine' 'debuglevel' -N >'samba.!msg
120 Master = ""
121 do until lines(samba.!msg) = 0
122 nmblookupline = linein(samba.!msg)
123 if pos('<1b>', nmblookupline) > 0 then Master = Master||"LMB," /* + */
124 if pos('<1d>', nmblookupline) > 0 then Master = Master||"DMB," /* * */
125 if pos('MAC',nmblookupline) > 0 then do
126 parse var nmblookupline . '=' MAC
127 MAC = strip(MAC)
128 end
129 end
130 Master= strip(Master,,',')
131 ok = stream(samba.!msg,'c','close')
132 ok = SysFileDelete(samba.!msg)
133 ok = VRMethod( "CN_smbtree", "SetFieldData", smbtree.!machine, MBFH, Master,MacFH,MAC)
134 end
135 end
136 else do
137 call _RefreshShares
138 end
139 end
140 when smbtreeline.sl = "" then nop
141 otherwise do
142 say ' Workgroup: "'smbtreeline.sl'"'
143 if pos("RECEIVING",translate(smbtreeline.sl)) > 0 | pos("TDB(",translate(smbtreeline.sl)) > 0 then do
144 Msg.Type = "W"
145 Msg.Text = smbtreeline.sl
146 call _ShowMsg
147 end
148 else do
149 if VRGet("CN_smbtree","View") = "IconTree" then do
150 smbtree.!workgroup = VRMethod( "CN_smbtree", "AddRecord",,, smbtreeline.sl,"#62:PMWP.DLL")
151 ok = VRMethod( "CN_smbtree", "SetFieldData", smbtree.!workgroup, WorkGroupFH, smbtreeline.sl)
152 ok = VRMethod( "CN_smbtree", "SetRecordAttr", smbtree.!workgroup, "Collapsed", 0)
153 ok = VRMethod( "CN_smbtree", "SetRecordAttr", smbtree.!workgroup, "ReadOnly", 1)
154 ok = VRMethod( "CN_smbtree", 'SetRecordAttr', smbtree.!workgroup, "UserData", "WORKGROUP|")
155 end
156 CurWG = smbtreeline.sl
157 end
158 end
159 end
160 end
161
162 ok = VRSet( "CN_smbtree", "Painting", 1 )
163
164/* ok = VRSet("Main", 'Pointer', '<default>' ) */
165 ok = VRSet("CN_smbtree","Enabled", 1)
166 ok = VRSet("TM_Throbber","Enabled", 0)
167 ok = VRSet("Pict_Throbber","Visible", 0)
168 say time()' _RefreshTreeDisplay() done'
169return
170
171/*:VRX _RefreshShares */
172_RefreshShares:
173 say time()' _RefreshShares() started'
174 /* RefreshID = RANDOM() */
175 smbmachine = TempDir||"smbmachine."||machine
176 MaxSmbClient = 32 /* Do not run more than MaxSmbClient instances of smbclient.exe at the same time */
177
178 Defer = 1
179 do while Defer = 1
180 SmbCltCount = 0
181 ok = PRProcessList(proc)
182
183 do I = 1 to proc.0
184 CurProc = VRParseFileName(proc.i.name,'NE')
185 if CurProc = "SMBCLIENT.EXE" then SmbCltCount = SmbCltCount + 1
186 end
187 say ' 'SmbCltCount' instance(s) of 'samba.!smbclientexe' is/are running.'
188 if SmbCltCount >= MaxSmbClient then do
189 say " Waiting until at least "SmbCltCount-MaxSmbClient+1" instance(s) of smbclient.exe terminate(s)."
190 ok = SysSleep(1)
191 end
192 else Defer = 0
193 end
194
195 if UserCred = 'USERCRED' | UserCred = '' | UserCred = '--user=%' then UserCred = '-N'
196
197 say ' detach 'samba.!smbclientexe' -L "'strip(machine)'" 'UserCred' 'debuglevel' 1>'smbmachine' 2>NUL'
198 address cmd 'detach 'samba.!smbclientexe' -L "'strip(machine)'" 'UserCred' 'debuglevel' 1>'smbmachine' 2>NUL'
199
200 if UserCred = '-N' then UserCred = ''
201
202 RefreshMode = "SHARE"
203
204 ok = VRSet("CN_smbtree","Enabled", 0)
205 ok = VRset("TM_RefreshTreeDisplay","Enabled",1)
206 say time()' _RefreshShares() done'
207return
208
209/*:VRX _AddSharesDisplay
210*/
211_AddSharesDisplay: /* New get shares code - uses smbclient output and is much faster */
212 say time()' _AddSharesDisplay() started'
213
214 ok = SysFileTree(Tempdir||'smbmachine.*',smbmachine.,'FO')
215 say ' 'smbmachine.0' file(s) to process.'
216
217 if smbmachine.0 = 0 then do /* we are done, no more files around, cleanup, disable Timer and exit */
218 RefreshMode = ""
219 ok = VRSet("CN_smbtree","Enabled", 1)
220 ok = VRset("TM_RefreshTreeDisplay","Enabled",0)
221 ok = VRSet( "CN_smbtree", "Painting", 0 )
222 ok = VRSet( "CN_smbtree", "Painting", 1 )
223 say time()' _AddSharesDisplay() completed'
224 return /* exit here */
225 end
226
227 if UserCred = 'USERCRED' | UserCred = '' | UserCred = '--user=%' then UserCred = '-N'
228 if ShowHidden = 'SHOWHIDDEN' | ShowHidden = '' then ShowHidden = 0
229
230 do I = 1 to smbmachine.0
231 say ' Going for "'smbmachine.I'"'
232 stat = stream(smbmachine.I,'c','open read')
233 if stat = "READY:" then do /* we found a readable output file */
234 OneWorkGroupOnly = 0
235 /* Machine = VRParseFilename(smbmachine.I,'E') */
236 Machine = substr(smbmachine.I,pos('.',smbmachine.I)+1)
237
238 smbtree.!machine = _GetMachinehandle(Machine)
239 say ' Machine (handle) = "'machine'" ('smbtree.!machine')'
240
241 if smbtree.!machine = "" then do /* invalid (old) file */
242 say time()' _AddSharesDisplay() exit with Invalid file found (no corresponding machine)'
243 ok = stream(smbmachine.I,'c','close')
244 ok = SysFileDelete(smbmachine.I)
245 iterate
246 end
247 line = linein(smbmachine.I)
248 say ' Answer "'line'"'
249 ok = VRMethod('CN_smbtree', 'SetRecordAttr', smbtree.!machine, 'UserData', "SERVER|"||strip(line))
250
251 if pos("FAIL", translate(line)) > 0 then do /* we see an error message - the term "FAIL" seems to be common to all */
252 say time()' _AddSharesDisplay() exit with "'line'"'
253 ok = stream(smbmachine.I,'c','close')
254 ok = SysFileDelete(smbmachine.I)
255 iterate
256 end
257
258 retries = 0
259 do while(left(line,1) <> '09'x)
260 line = linein(smbmachine.I)
261 retries = retries + 1
262 say ' Skip 'retries' "'line'"'
263 if retries >=10 then do /* No valid output - error */
264 say time()' _AddSharesDisplay() exit with invalid output error'
265 ok = stream(smbmachine.I,'c','close')
266 ok = SysFileDelete(smbmachine.I)
267 leave
268 end
269 end
270 if retries >=10 then iterate
271
272 /* Skip header */
273 line = linein(smbmachine.I)
274 line = linein(smbmachine.I)
275
276 if translate(left(strip(line),5)) = "ERROR" then ok = VRMethod('CN_smbtree', 'SetRecordAttr', smbtree.!machine, 'UserData', "SERVER|"||strip(line))
277
278 do while(left(line,1) = '09'x) /* Share loop */
279 parse var line '09'x share type comment
280 type = translate(strip(type))
281 comment = strip(comment)
282
283 select
284 when type = "DISK" then res = '#64:PMWP.DLL'
285 when type = "PRINTER" then res = '#65:PMWP.DLL'
286 when type = "IPC" then res = '#59:PMWP.DLL'
287 when type = "DEVICE" then res = '#84:PMWP.DLL' /* There might be better ones around */
288 otherwise res = ''
289 end
290
291 /* Now the machine receives the wakeup icon */
292 ok = VRMethod('CN_smbtree', 'SetRecordAttr', smbtree.!machine, 'Icon', "#35:PMWP.DLL")
293 parent = smbtree.!machine
294 smbtree.!share = VRMethod( "CN_smbtree", "AddRecord",parent,, share||'0D0A'x||comment, res)
295 ok = VRMethod( "CN_smbtree", "SetRecordAttr", smbtree.!share, "ReadOnly", 1, 'UserData', type"|")
296 if pos("$", share) > 0 then ok = VRMethod( "CN_smbtree", "SetRecordAttr", smbtree.!share, "Visible", ShowHidden)
297
298 /* get next share */
299 line = linein(smbmachine.I)
300 end /* Share loop */
301
302 do until left(line,10) = '09'x||'Workgroup'
303 line = linein(smbmachine.I)
304 end
305 line = linein(smbmachine.I)
306 /* Reading FIRST workgroup and master - eventually both empty */
307 line = linein(smbmachine.I)
308
309 /* Multiple workgroups? */
310 if lines(smbmachine.I) = 0 then OneWorkGroupOnly = 1
311 else OneWorkGroupOnly = 0
312 say " OneWorkGroupOnly = "OneWorkGroupOnly
313
314 parse var line '09'x workgroup master
315 master = strip(master)
316
317 /* we use this to set the workgroup for manually added servers - if there is ONLY ONE workgroup */
318 if workgroup <> "" & OneWorkGroupOnly = 1 then do
319 wgh = _GetMachinehandle(workgroup)
320 if wgh = "" then do /* The machine appears to be in a new workgroup - add it as well */
321 /* NOTE: This should be obsolete now because the list of available
322 workgroups should always have been updated before we get here */
323 wgh = VRMethod( "CN_smbtree", "AddRecord",,, workgroup,"#62:PMWP.DLL")
324 ok = VRMethod( "CN_smbtree", "SetFieldData", wgh, WorkGroupFH, workgroup)
325 ok = VRMethod( "CN_smbtree", "SetRecordAttr", wgh, "Collapsed", 0)
326 ok = VRMethod( "CN_smbtree", "SetRecordAttr", wgh, "ReadOnly", 1)
327 ok = VRMethod( "CN_smbtree", 'SetRecordAttr', wgh, "UserData", "WORKGROUP|")
328 end
329 /* we only do this for machines with empty parent (=workgroup) handle */
330 /* IF there is only one workgroup */
331 if wgh <> "" & VRMethod('CN_smbtree', 'GetRecordAttr', smbtree.!machine, 'Parent') = "" then do
332 ok = VRMethod('CN_smbtree', 'SetRecordAttr', smbtree.!machine, 'Parent', wgh)
333 end
334 end
335 else do
336 if workgroup <> "" then do
337 /* There are multiple workgroups, we need additional */
338 /* measures to find out which is our workgroup */
339 say ' 'samba.!smbclientexe' -L "'Machine'" -N 'debuglevel' 2>'samba.!msg' 1>NUL'
340 address cmd samba.!smbclientexe' -L "'Machine'" -N 'debuglevel' 2>'samba.!msg' 1>NUL'
341 infoline = linein(samba.!msg)
342 if word(infoline,1) = "creating" then do /* upcase tables are missing */
343 say "Missing upcase tables detected!"
344 infoline = linein(samba.!msg)
345 infoline = linein(samba.!msg)
346 end
347 IF options.!debug == 1 THEN say ' Response = "'Infoline'"'
348 ok = stream(samba.!msg,'c','close')
349 ok = SysFileDelete(samba.!msg)
350
351 parse var infoline "Domain=["WorkGroup"] "Rest
352 wgh = _GetMachinehandle(workgroup)
353 if wgh <> "" & VRMethod('CN_smbtree', 'GetRecordAttr', smbtree.!machine, 'Parent') = "" then do
354 ok = VRMethod('CN_smbtree', 'SetRecordAttr', smbtree.!machine, 'Parent', wgh)
355 end
356 end
357 else do
358 /* The machine does not allow browsing and does not report a workgroup */
359 say ' "'Machine'" does not allow browsing. Unable to determine workgroup.'
360 end
361 end
362
363 say time()' _AddSharesDisplay() success and cleanup'
364 ok = stream(smbmachine.I,'c','close')
365 ok = SysFileDelete(smbmachine.I)
366 if ok <> 0 then say ' Failure 'ok' deleting "'smbmachine.I'"!'
367 end
368 else say ' Got "'stat'" for "'smbmachine.I'"'
369 end
370
371 if UserCred = '-N' then UserCred = ''
372
373 say time()' _AddSharesDisplay() loop end'
374return
375
376/*:VRX _GetMachinehandle
377*/
378
379_GetMachinehandle: procedure /* get recordhandle by machine name (also works for workgroups) */
380 Machine = translate(arg(1))
381
382 ok = VRMethod("CN_smbtree", "GetRecordList", "All", rh.)
383 match = 0
384
385 do I = 1 to rh.0
386 ResName = translate(VRMethod("CN_smbtree","GetRecordAttr",rh.I,"Caption"))
387
388 parse var ResName ResName '0D0A'x .
389 ResName = strip(ResName)
390
391 if Machine = ResName then do /* we got a matching name */
392 match = 1
393 leave
394 end
395 end
396 if match = 0 then rh.I = "" /* return an empty handle, if there was no match */
397return rh.I
398
399/*:VRX _RefreshWorkgroups
400*/
401_RefreshWorkgroups:
402 say time()' _RefreshWorkgroups() started'
403 if UserCred = 'USERCRED' | UserCred = '' | UserCred = '--user=%' then UserCred = '-N'
404 if BroadCast = 1 then BroadCast = '-b'; else BroadCast = ''
405
406 say samba.!smbtreeexe' 'BroadCast' -D 'debuglevel' 'UserCred' >'samba.!msg
407 address cmd samba.!smbtreeexe' 'BroadCast' -D 'debuglevel' 'UserCred' >'samba.!msg
408
409 if BroadCast = '-b' then BroadCast = 1; else BroadCast = 0
410 if UserCred = '-N' then UserCred = ''
411
412 drop wgline.
413
414 sl = 0
415
416 do until lines(samba.!msg) = 0
417 sl = sl + 1
418 wgline.sl = linein(samba.!msg)
419 end
420 wgline.0 = sl
421 ok = stream(samba.!msg,'c','close')
422 ok = SysFileDelete(samba.!msg)
423
424 do sl = 1 to wgline.0
425 if pos("RECEIVING",translate(wgline.sl)) > 0 | pos("TDB(",translate(wgline.sl)) > 0 then do
426 iterate /* We ignore errors here */
427 end
428 workgroup = wgline.sl
429 wgh = _GetMachinehandle(workgroup)
430 if wgh = "" then do /* A new workgroup was found -- add it */
431 wgh = VRMethod( "CN_smbtree", "AddRecord",,, workgroup,"#62:PMWP.DLL")
432 ok = VRMethod( "CN_smbtree", "SetFieldData", wgh, WorkGroupFH, workgroup)
433 ok = VRMethod( "CN_smbtree", "SetRecordAttr", wgh, "Collapsed", 0)
434 ok = VRMethod( "CN_smbtree", "SetRecordAttr", wgh, "ReadOnly", 1)
435 ok = VRMethod( "CN_smbtree", 'SetRecordAttr', wgh, "UserData", "WORKGROUP|")
436 end
437 end
438 say time()' _RefreshWorkgroups() done'
439return
440
Note: See TracBrowser for help on using the repository browser.