X-Git-Url: https://git.saurik.com/apple/security.git/blobdiff_plain/72a12576750f52947eb043106ba5c12c0d07decf..b1ab9ed8d0e0f1c3b66d7daa8fd5564444c56195:/libsecurity_utilities/lib/machrunloopserver.cpp?ds=inline

diff --git a/libsecurity_utilities/lib/machrunloopserver.cpp b/libsecurity_utilities/lib/machrunloopserver.cpp
new file mode 100644
index 00000000..cc0ed5c1
--- /dev/null
+++ b/libsecurity_utilities/lib/machrunloopserver.cpp
@@ -0,0 +1,111 @@
+/*
+ * Copyright (c) 2000-2004 Apple Computer, Inc. All Rights Reserved.
+ * 
+ * @APPLE_LICENSE_HEADER_START@
+ * 
+ * This file contains Original Code and/or Modifications of Original Code
+ * as defined in and that are subject to the Apple Public Source License
+ * Version 2.0 (the 'License'). You may not use this file except in
+ * compliance with the License. Please obtain a copy of the License at
+ * http://www.opensource.apple.com/apsl/ and read it before using this
+ * file.
+ * 
+ * The Original Code and all software distributed under the License are
+ * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
+ * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
+ * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY,
+ * FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT.
+ * Please see the License for the specific language governing rights and
+ * limitations under the License.
+ * 
+ * @APPLE_LICENSE_HEADER_END@
+ */
+
+
+//
+// machrunloopserver - C++ shell for writing Mach 3 servers called by CFRunLoop
+//
+#include "machrunloopserver.h"
+#include <security_utilities/cfutilities.h>
+#include <mach/mach_error.h>
+#include <security_utilities/debugging.h>
+
+
+namespace Security {
+namespace MachPlusPlus {
+
+
+//
+// Generic Mach server
+//
+MachRunLoopServer::MachRunLoopServer(const char *name)
+	: MachServer(name), CFAutoPort(primaryServicePort())
+{
+}
+
+MachRunLoopServer::MachRunLoopServer(const char *name, const Bootstrap &boot)
+	: MachServer(name, boot), CFAutoPort(primaryServicePort())
+{
+}
+
+void MachRunLoopServer::run(size_t bufferSize, mach_msg_options_t options)
+{
+	// allocate reply buffer
+	mReplyMessage.setBuffer(bufferSize);
+	
+	// enable reception
+	CFAutoPort::enable();
+
+	// we are it!
+	perThread().server = this;
+}
+
+
+MachRunLoopServer::~MachRunLoopServer()
+{
+    // no longer active on this thread
+    perThread().server = NULL;
+}
+
+
+//
+// Handle dead-port notifications.
+// Since we don't actually run our own runloop here, we can't well use standard
+// notifications to our own server port. So we use a CFMachPort facility instead.
+//
+void MachRunLoopServer::notifyIfDead(Port port) const
+{
+	if (CFMachPortRef cfPort = CFMachPortCreateWithPort(NULL, port, NULL, NULL, NULL))
+		CFMachPortSetInvalidationCallBack(cfPort, cfInvalidate);
+}
+
+void MachRunLoopServer::cfInvalidate(CFMachPortRef cfPort, void *context)
+{
+	reinterpret_cast<MachRunLoopServer *>(context)->notifyDeadName(CFMachPortGetPort(cfPort));
+	//@@@ should we CFRelease cfPort here?
+}
+
+
+//
+// Reception callback
+//
+void MachRunLoopServer::receive(Message &request)
+{
+	active().oneRequest(request);
+}
+
+void MachRunLoopServer::oneRequest(Message &request)
+{
+	if (!handle(request, mReplyMessage)) {	// MIG dispatch failed
+		secdebug("machrls", "MachRunLoopServer dispatch failed");
+	} else {
+		// MIG dispatch handled the call. Send reply back to caller.
+		mReplyMessage.send((MACH_MSGH_BITS_REMOTE(mReplyMessage.bits()) == MACH_MSG_TYPE_MOVE_SEND_ONCE) ?
+			MACH_SEND_MSG :	MACH_SEND_MSG|MACH_SEND_TIMEOUT);
+	}
+	active().releaseDeferredAllocations();
+}
+
+
+} // end namespace MachPlusPlus
+} // end namespace Security