Mono.Zeroconf – Rev 2
?pathlinks?
//
// ServiceBrowser.cs
//
// Authors:
// Aaron Bockover <abockover@novell.com>
//
// Copyright (C) 2006-2007 Novell, Inc (http://www.novell.com)
//
// Permission is hereby granted, free of charge, to any person obtaining
// a copy of this software and associated documentation files (the
// "Software"), to deal in the Software without restriction, including
// without limitation the rights to use, copy, modify, merge, publish,
// distribute, sublicense, and/or sell copies of the Software, and to
// permit persons to whom the Software is furnished to do so, subject to
// the following conditions:
//
// The above copyright notice and this permission notice shall be
// included in all copies or substantial portions of the Software.
//
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
// EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
// MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
// NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
// LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
// OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
// WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
//
using System;
using System.Collections;
using System.Collections.Generic;
using System.Threading;
namespace Mono.Zeroconf.Providers.Bonjour
{
public class ServiceBrowseEventArgs : Mono.Zeroconf.ServiceBrowseEventArgs
{
private bool more_coming;
public ServiceBrowseEventArgs(BrowseService service, bool moreComing) : base(service)
{
this.more_coming = moreComing;
}
public bool MoreComing {
get { return more_coming; }
}
}
public class ServiceBrowser : IServiceBrowser, IDisposable
{
private uint interface_index;
private AddressProtocol address_protocol;
private string regtype;
private string domain;
private ServiceRef sd_ref = ServiceRef.Zero;
private Dictionary<string, IResolvableService> service_table = new Dictionary<string, IResolvableService> ();
private Native.DNSServiceBrowseReply browse_reply_handler;
private Thread thread;
public event ServiceBrowseEventHandler ServiceAdded;
public event ServiceBrowseEventHandler ServiceRemoved;
public ServiceBrowser()
{
browse_reply_handler = new Native.DNSServiceBrowseReply(OnBrowseReply);
}
public void Browse (uint interfaceIndex, AddressProtocol addressProtocol, string regtype, string domain)
{
Configure(interfaceIndex, addressProtocol, regtype, domain);
StartAsync();
}
public void Configure(uint interfaceIndex, AddressProtocol addressProtocol, string regtype, string domain)
{
this.interface_index = interfaceIndex;
this.address_protocol = addressProtocol;
this.regtype = regtype;
this.domain = domain;
if(regtype == null) {
throw new ArgumentNullException("regtype");
}
}
private void Start(bool @async)
{
if(thread != null) {
throw new InvalidOperationException("ServiceBrowser is already started");
}
if(@async) {
thread = new Thread(new ThreadStart(ThreadedStart));
thread.IsBackground = true;
thread.Start();
} else {
ProcessStart();
}
}
public void Start()
{
Start(false);
}
public void StartAsync()
{
Start(true);
}
private void ThreadedStart()
{
try {
ProcessStart();
} catch(ThreadAbortException) {
Thread.ResetAbort();
}
thread = null;
}
private void ProcessStart()
{
ServiceError error = Native.DNSServiceBrowse(out sd_ref, ServiceFlags.Default,
interface_index, regtype, domain, browse_reply_handler, IntPtr.Zero);
if(error != ServiceError.NoError) {
throw new ServiceErrorException(error);
}
sd_ref.Process();
}
public void Stop()
{
if(sd_ref != ServiceRef.Zero) {
sd_ref.Deallocate();
sd_ref = ServiceRef.Zero;
}
if(thread != null) {
thread.Abort();
thread = null;
}
}
public void Dispose()
{
Stop();
}
public IEnumerator<IResolvableService> GetEnumerator ()
{
lock (this) {
foreach (IResolvableService service in service_table.Values) {
yield return service;
}
}
}
System.Collections.IEnumerator System.Collections.IEnumerable.GetEnumerator ()
{
return GetEnumerator ();
}
private void OnBrowseReply(ServiceRef sdRef, ServiceFlags flags, uint interfaceIndex, ServiceError errorCode,
IntPtr serviceName, string regtype, string replyDomain, IntPtr context)
{
string name = Native.Utf8toString(serviceName);
BrowseService service = new BrowseService();
service.Flags = flags;
service.Name = name;
service.RegType = regtype;
service.ReplyDomain = replyDomain;
service.InterfaceIndex = interfaceIndex;
service.AddressProtocol = address_protocol;
ServiceBrowseEventArgs args = new ServiceBrowseEventArgs(
service, (flags & ServiceFlags.MoreComing) != 0);
if((flags & ServiceFlags.Add) != 0) {
lock (service_table) {
if (service_table.ContainsKey (name)) {
service_table[name] = service;
} else {
service_table.Add(name, service);
}
}
ServiceBrowseEventHandler handler = ServiceAdded;
if(handler != null) {
handler(this, args);
}
} else {
lock (service_table) {
if (service_table.ContainsKey (name)) {
service_table.Remove (name);
}
}
ServiceBrowseEventHandler handler = ServiceRemoved;
if(handler != null) {
handler(this, args);
}
}
}
}
}