WebKit2/Platform/mac/WorkQueueMac.cpp
changeset 0 4f2f89ce4247
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/WebKit2/Platform/mac/WorkQueueMac.cpp	Fri Sep 17 09:02:29 2010 +0300
@@ -0,0 +1,193 @@
+/*
+ * Copyright (C) 2010 Apple Inc. All rights reserved.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions
+ * are met:
+ * 1. Redistributions of source code must retain the above copyright
+ *    notice, this list of conditions and the following disclaimer.
+ * 2. Redistributions in binary form must reproduce the above copyright
+ *    notice, this list of conditions and the following disclaimer in the
+ *    documentation and/or other materials provided with the distribution.
+ *
+ * THIS SOFTWARE IS PROVIDED BY APPLE INC. AND ITS CONTRIBUTORS ``AS IS''
+ * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
+ * THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
+ * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR ITS CONTRIBUTORS
+ * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
+ * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
+ * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
+ * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
+ * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
+ * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
+ * THE POSSIBILITY OF SUCH DAMAGE.
+ */
+
+#include "WorkQueue.h"
+
+#include <mach/mach_port.h>
+#include <wtf/PassOwnPtr.h>
+
+#if HAVE(DISPATCH_H)
+
+void WorkQueue::executeWorkItem(void* item)
+{
+    WorkQueue* queue = static_cast<WorkQueue*>(dispatch_get_context(dispatch_get_current_queue()));
+    OwnPtr<WorkItem> workItem(static_cast<WorkItem*>(item));
+    
+    {
+        MutexLocker locker(queue->m_isValidMutex);
+        if (!queue->m_isValid)
+            return;
+    }
+    
+    workItem->execute();
+}
+
+void WorkQueue::scheduleWork(PassOwnPtr<WorkItem> item)
+{
+    dispatch_async_f(m_dispatchQueue, item.leakPtr(), executeWorkItem);
+}
+
+class WorkQueue::EventSource {
+public:
+    EventSource(MachPortEventType eventType, dispatch_source_t dispatchSource, PassOwnPtr<WorkItem> workItem)
+        : m_eventType(eventType)
+        , m_dispatchSource(dispatchSource)
+        , m_workItem(workItem)
+    {
+    }
+    
+    dispatch_source_t dispatchSource() const { return m_dispatchSource; }
+    
+    static void eventHandler(void* source) 
+    {
+        EventSource* eventSource = static_cast<EventSource*>(source);
+        
+        eventSource->m_workItem->execute();
+    }
+    
+    static void cancelHandler(void* source)
+    {
+        EventSource* eventSource = static_cast<EventSource*>(source);
+        
+        mach_port_t machPort = dispatch_source_get_handle(eventSource->m_dispatchSource);
+        
+        switch (eventSource->m_eventType) {
+        case MachPortDataAvailable:
+            // Release our receive right.
+            mach_port_mod_refs(mach_task_self(), machPort, MACH_PORT_RIGHT_RECEIVE, -1);
+            break;
+        case MachPortDeadNameNotification:
+            // Release our send right.
+            mach_port_deallocate(mach_task_self(), machPort);
+            break;
+        }
+    }
+    
+    static void finalizeHandler(void* source)
+    {
+        EventSource* eventSource = static_cast<EventSource*>(source);
+        
+        delete eventSource;
+    }
+    
+private:
+    MachPortEventType m_eventType;
+    
+    // This is a weak reference, since m_dispatchSource references the event source.
+    dispatch_source_t m_dispatchSource;
+    
+    OwnPtr<WorkItem> m_workItem;
+};
+
+void WorkQueue::registerMachPortEventHandler(mach_port_t machPort, MachPortEventType eventType, PassOwnPtr<WorkItem> workItem)
+{
+    dispatch_source_type_t sourceType = 0;
+    switch (eventType) {
+    case MachPortDataAvailable:
+        sourceType = DISPATCH_SOURCE_TYPE_MACH_RECV;
+        break;
+    case MachPortDeadNameNotification:
+        sourceType = DISPATCH_SOURCE_TYPE_MACH_SEND;
+        break;
+    }
+    
+    dispatch_source_t dispatchSource = dispatch_source_create(sourceType, machPort, 0, m_dispatchQueue);
+    
+    EventSource* eventSource = new EventSource(eventType, dispatchSource, workItem);
+    dispatch_set_context(dispatchSource, eventSource);
+    
+    dispatch_source_set_event_handler_f(dispatchSource, &EventSource::eventHandler);
+    dispatch_source_set_cancel_handler_f(dispatchSource, &EventSource::cancelHandler);
+    dispatch_set_finalizer_f(dispatchSource, &EventSource::finalizeHandler);
+    
+    // Add the source to our set of sources.
+    {
+        MutexLocker locker(m_eventSourcesMutex);
+        
+        ASSERT(!m_eventSources.contains(machPort));
+        
+        m_eventSources.set(machPort, eventSource);
+        
+        // And start it!
+        dispatch_resume(dispatchSource);
+    }
+}
+
+void WorkQueue::unregisterMachPortEventHandler(mach_port_t machPort)
+{
+    ASSERT(machPort);
+    
+    MutexLocker locker(m_eventSourcesMutex);
+    
+    HashMap<mach_port_t, EventSource*>::iterator it = m_eventSources.find(machPort);
+    ASSERT(it != m_eventSources.end());
+    
+    ASSERT(m_eventSources.contains(machPort));
+
+    EventSource* eventSource = it->second;
+    // Cancel and release the source. It will be deleted in its finalize handler.
+    dispatch_source_cancel(eventSource->dispatchSource());
+    dispatch_release(eventSource->dispatchSource());
+
+    m_eventSources.remove(it);    
+}
+
+void WorkQueue::platformInitialize(const char* name)
+{
+    m_dispatchQueue = dispatch_queue_create(name, 0);
+    dispatch_set_context(m_dispatchQueue, this);
+}
+
+void WorkQueue::platformInvalidate()
+{
+#if !ASSERT_DISABLED
+    MutexLocker locker(m_eventSourcesMutex);
+    ASSERT(m_eventSources.isEmpty());
+#endif
+}
+
+#else /* !HAVE(DISPATCH_H) */
+
+void WorkQueue::scheduleWork(PassOwnPtr<WorkItem> item)
+{
+}
+
+void WorkQueue::registerMachPortEventHandler(mach_port_t, MachPortEventType, PassOwnPtr<WorkItem>)
+{
+}
+
+void WorkQueue::unregisterMachPortEventHandler(mach_port_t)
+{
+}
+
+void WorkQueue::platformInitialize(const char*)
+{
+}
+
+void WorkQueue::platformInvalidate()
+{
+}
+
+#endif