Security-57031.1.35.tar.gz

[apple/security.git] / Security / libsecurity_utilities / lib / cfmach++.cpp

diff --git a/Security/libsecurity_utilities/lib/cfmach++.cpp b/Security/libsecurity_utilities/lib/cfmach++.cpp
new file mode 100644 (file)
index 0000000..db6676d
--- /dev/null
+++ b/Security/libsecurity_utilities/lib/cfmach++.cpp
@@ -0,0 +1,130 @@
+/*
+ * Copyright (c) 2000-2004,2011-2012,2014 Apple 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@
+ */
+
+
+//
+// cfmach++ - a marriage of CoreFoundation with Mach/C++
+//
+#include <security_utilities/cfmach++.h>
+
+
+namespace Security {
+namespace MachPlusPlus {
+
+
+//
+// Construct CFAutoPorts
+//
+CFAutoPort::CFAutoPort()
+       : mEnabled(false)
+{ }
+
+CFAutoPort::CFAutoPort(mach_port_t p)
+       : Port(p), mEnabled(false)
+{ }
+
+
+//
+// On destruction, make sure we're disengaged from the CFRunLoop
+//
+CFAutoPort::~CFAutoPort()
+{
+       disable();
+       
+       // invalidate everything
+       if (mPort)
+       {
+               CFMachPortInvalidate(mPort);
+               CFRunLoopSourceInvalidate(mSource);
+       }
+}
+
+
+//
+// enable() will lazily allocate needed resources, then click into the runloop
+//
+void CFAutoPort::enable()
+{
+       if (!mEnabled) {
+               if (!*this)
+                       allocate();
+               if (!mPort) {
+                       // first-time creation of CF resources
+                       CFMachPortContext ctx = { 1, this, NULL, NULL, NULL };
+                       CFMachPortRef machPort = CFMachPortCreateWithPort(NULL, port(), cfCallback, &ctx, NULL);
+                       if (machPort != NULL)
+                       {
+                               // using take here because "assignment" causes an extra retain, which will make the
+                               // CF objects leak when this data structure goes away.
+                               mPort.take(machPort);
+                               
+                               CFRunLoopSourceRef sr = CFMachPortCreateRunLoopSource(NULL, mPort, 10);
+                               mSource.take(sr);
+                       }
+                       if (!mPort || !mSource)
+                               CFError::throwMe();             // CF won't tell us why...
+               }
+               CFRunLoopAddSource(CFRunLoopGetCurrent(), mSource, kCFRunLoopCommonModes);
+               mEnabled = true;
+               secdebug("autoport", "%p enabled", this);
+       }
+}
+
+
+//
+// Disable() just removes us from the runloop. All the other resources stay
+// around, ready to be re-enable()d.
+//
+void CFAutoPort::disable()
+{
+       if (mEnabled) {
+               CFRunLoopRemoveSource(CFRunLoopGetCurrent(), mSource, kCFRunLoopCommonModes);
+               mEnabled = false;
+               secdebug("autoport", "%p disabled", this);
+       }
+}
+
+
+//
+// The CF-sponsored port callback.
+// We pass this to our receive() virtual and eat all exceptions.
+//
+static int gNumTimesCalled = 0;
+
+void CFAutoPort::cfCallback(CFMachPortRef cfPort, void *msg, CFIndex size, void *context)
+{
+       ++gNumTimesCalled;
+       secdebug("adhoc", "Callback was called %d times.", gNumTimesCalled);
+
+#warning Cast to mach_msg_size_t may loose precision
+       Message message(msg, (mach_msg_size_t)size);
+       try {
+               reinterpret_cast<CFAutoPort *>(context)->receive(message);
+       } catch (...) {
+               secdebug("autoport", "%p receive handler failed with exception", context);
+       }
+}
+
+
+} // end namespace MachPlusPlus
+} // end namespace Security