Fixed addition/removal of devices during discovery (thanks @leolobato).

naxos · Nov 10, 2017 · eb1e3ee · eb1e3ee
1 parent 7b3f2e7
commit eb1e3ee
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 8 deletions.
diff --git a/....xcodeproj/project.xcworkspace/xcuserdata/masc.xcuserdatad/UserInterfaceState.xcuserstate b/....xcodeproj/project.xcworkspace/xcuserdata/masc.xcuserdatad/UserInterfaceState.xcuserstate
diff --git a/SMBClient/SMBDiscovery.m b/SMBClient/SMBDiscovery.m
@@ -114,27 +114,27 @@ - (void)stopDiscovery {
 }
 
 static void _on_entry_added(void *p_opaque, netbios_ns_entry *entry) {
-    if (_addedHandler) {
-        dispatch_async(dispatch_get_main_queue(), ^{
+    dispatch_async(dispatch_get_main_queue(), ^{
+        if (_addedHandler) {
             SMBDevice *device = _device(entry);
 
             if (_typeMask & device.type) {
                 _addedHandler(_device(entry));
             }
-        });
-    }
+        }
+    });
 }
 
 static void _on_entry_removed(void *p_opaque, netbios_ns_entry *entry) {
-    if (_removedHandler) {
-        dispatch_async(dispatch_get_main_queue(), ^{
+    dispatch_async(dispatch_get_main_queue(), ^{
+        if (_removedHandler) {
             SMBDevice *device = _device(entry);
 
             if (_typeMask & device.type) {
                 _removedHandler(_device(entry));
             }
-        });
-    }
+        }
+    });
 }
 
 @end