]> git.saurik.com Git - apple/xnu.git/blobdiff - osfmk/UserNotification/KUNCUserNotifications.c
xnu-517.12.7.tar.gz
[apple/xnu.git] / osfmk / UserNotification / KUNCUserNotifications.c
index 8007e03b1e8421870fd30ab70f71c857963d2825..aec492ca7cc08c3f29b9f1bfa13482ef66431ecb 100644 (file)
@@ -1,24 +1,21 @@
 /*
- * Copyright (c) 2000 Apple Computer, Inc. All rights reserved.
+ * Copyright (c) 2000-2003 Apple Computer, Inc. All rights reserved.
  *
  * @APPLE_LICENSE_HEADER_START@
  * 
- * Copyright (c) 1999-2003 Apple Computer, Inc.  All Rights Reserved.
+ * The contents of this file constitute Original Code as defined in and
+ * are subject to the Apple Public Source License Version 1.1 (the
+ * "License").  You may not use this file except in compliance with the
+ * License.  Please obtain a copy of the License at
+ * http://www.apple.com/publicsource and read it before using this file.
  * 
- * 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
+ * This 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.
+ * FITNESS FOR A PARTICULAR PURPOSE OR NON-INFRINGEMENT.  Please see the
+ * License for the specific language governing rights and limitations
+ * under the License.
  * 
  * @APPLE_LICENSE_HEADER_END@
  */
@@ -47,9 +44,6 @@
  * DEFINES AND STRUCTURES
  */
 
-UNDServerRef gUNDServer = UND_SERVER_NULL;
-
-
 struct UNDReply {
        decl_mutex_data(,lock)                          /* UNDReply lock */
        int                             userLandNotificationKey;
@@ -67,8 +61,10 @@ UNDReply_deallocate(
        UNDReplyRef             reply)
 {
        ipc_port_t port;
+
        UNDReply_lock(reply);
        port = reply->self_port;
+       assert(IP_VALID(port));
        ipc_kobject_set(port, IKO_NULL, IKOT_NONE);
        reply->self_port = IP_NULL;
        UNDReply_unlock(reply);
@@ -78,6 +74,25 @@ UNDReply_deallocate(
        return;
 }
 
+static UNDServerRef
+UNDServer_reference(void)
+{
+       UNDServerRef UNDServer;
+       kern_return_t kr;
+
+       kr = host_get_user_notification_port(host_priv_self(), &UNDServer);
+       assert(kr == KERN_SUCCESS);
+       return UNDServer;
+}
+
+static void
+UNDServer_deallocate(
+       UNDServerRef    UNDServer)
+{
+       if (IP_VALID(UNDServer))
+               ipc_port_release_send(UNDServer);
+}
+
 /* 
  * UND Mig Callbacks
 */
@@ -184,7 +199,17 @@ KUNCGetNotificationID()
 
 kern_return_t KUNCExecute(char executionPath[1024], int uid, int gid)
 {
-    return UNDExecute_rpc(gUNDServer, executionPath, uid, gid);
+
+       UNDServerRef UNDServer;
+
+       UNDServer = UNDServer_reference();
+       if (IP_VALID(UNDServer)) {
+               kern_return_t kr;
+               kr = UNDExecute_rpc(UNDServer, executionPath, uid, gid);
+               UNDServer_deallocate(UNDServer);
+               return kr;
+       }
+       return MACH_SEND_INVALID_DEST;
 }
 
 kern_return_t KUNCUserNotificationCancel(
@@ -205,9 +230,17 @@ kern_return_t KUNCUserNotificationCancel(
 
        reply->inprogress = FALSE;
        if (ulkey = reply->userLandNotificationKey) {
+               UNDServerRef UNDServer;
+
                reply->userLandNotificationKey = 0;
                UNDReply_unlock(reply);
-               kr = UNDCancelNotification_rpc(gUNDServer,ulkey);
+
+               UNDServer = UNDServer_reference();
+               if (IP_VALID(UNDServer)) {
+                       kr = UNDCancelNotification_rpc(UNDServer,ulkey);
+                       UNDServer_deallocate(UNDServer);
+               } else
+                       kr = MACH_SEND_INVALID_DEST;
        } else {
                UNDReply_unlock(reply);
                kr = KERN_SUCCESS;
@@ -227,8 +260,12 @@ KUNCUserNotificationDisplayNotice(
        char            *alertMessage,
        char            *defaultButtonTitle)
 {
-       kern_return_t kr;
-       kr = UNDDisplayNoticeSimple_rpc(gUNDServer,
+       UNDServerRef UNDServer;
+
+       UNDServer = UNDServer_reference();
+       if (IP_VALID(UNDServer)) {
+               kern_return_t kr;
+               kr = UNDDisplayNoticeSimple_rpc(UNDServer,
                                        timeout,
                                        flags,
                                        iconPath,
@@ -237,7 +274,10 @@ KUNCUserNotificationDisplayNotice(
                                        alertHeader,
                                        alertMessage,
                                        defaultButtonTitle);
-       return kr;
+               UNDServer_deallocate(UNDServer);
+               return kr;
+       }
+       return MACH_SEND_INVALID_DEST;
 }
 
 kern_return_t
@@ -254,9 +294,12 @@ KUNCUserNotificationDisplayAlert(
        char            *otherButtonTitle,
        unsigned        *responseFlags)
 {
-       kern_return_t   kr;
+       UNDServerRef    UNDServer;
        
-       kr = UNDDisplayAlertSimple_rpc(gUNDServer,
+       UNDServer = UNDServer_reference();
+       if (IP_VALID(UNDServer)) {
+               kern_return_t   kr;
+               kr = UNDDisplayAlertSimple_rpc(UNDServer,
                                       timeout,
                                       flags,
                                       iconPath,
@@ -268,7 +311,10 @@ KUNCUserNotificationDisplayAlert(
                                       alternateButtonTitle,
                                       otherButtonTitle,
                                       responseFlags);
-       return kr;
+               UNDServer_deallocate(UNDServer);
+               return kr;
+       }
+       return MACH_SEND_INVALID_DEST;
 }
 
 kern_return_t
@@ -283,8 +329,8 @@ KUNCUserNotificationDisplayFromBundle(
        int                          contextKey)
 {
        UNDReplyRef reply = (UNDReplyRef)id;
+       UNDServerRef UNDServer;
        ipc_port_t reply_port;
-       kern_return_t kr;
 
        if (reply == UND_REPLY_NULL)
                return KERN_INVALID_ARGUMENT;
@@ -298,14 +344,21 @@ KUNCUserNotificationDisplayFromBundle(
        reply_port = ipc_port_make_send(reply->self_port);
        UNDReply_unlock(reply);
 
-       kr = UNDDisplayCustomFromBundle_rpc(gUNDServer,
+       UNDServer = UNDServer_reference();
+       if (IP_VALID(UNDServer)) {
+               kern_return_t kr;
+
+               kr = UNDDisplayCustomFromBundle_rpc(UNDServer,
                                            reply_port,
                                            bundlePath,
                                            fileName,
                                            fileExtension,
                                            messageKey,
                                            tokenString);
-       return kr;
+               UNDServer_deallocate(UNDServer);
+               return kr;
+       }
+       return MACH_SEND_INVALID_DEST;
 }
 
 /*
@@ -332,6 +385,7 @@ convert_port_to_UNDReply(
                reply = (UNDReplyRef) port->ip_kobject;
                assert(reply != UND_REPLY_NULL);
                ip_unlock(port);
+               return reply;
        }
        return UND_REPLY_NULL;
 }
@@ -345,13 +399,7 @@ host_set_UNDServer(
         host_priv_t     host_priv,
         UNDServerRef    server)
 {
-
-       if (host_priv == HOST_PRIV_NULL || server == UND_SERVER_NULL)
-               return KERN_INVALID_ARGUMENT;
-       if (gUNDServer != UND_SERVER_NULL)
-               ipc_port_dealloc_kernel(gUNDServer);
-       gUNDServer = server;
-       return KERN_SUCCESS;
+       return (host_set_user_notification_port(host_priv, server));
 }
 
 /*
@@ -361,11 +409,7 @@ host_set_UNDServer(
 kern_return_t
 host_get_UNDServer(
        host_priv_t     host_priv,
-       UNDServerRef    *server)
+       UNDServerRef    *serverp)
 {
-       if (host_priv == HOST_PRIV_NULL)
-               return KERN_INVALID_ARGUMENT;
-        *server = gUNDServer;
-        return KERN_SUCCESS;
+       return (host_get_user_notification_port(host_priv, serverp));
 }
-