Libc-391.2.6.tar.gz

[apple/libc.git] / threads / lu_utils.c

diff --git a/threads/lu_utils.c b/threads/lu_utils.c
index 090f2e9570c3d60656bdf4c0125eff1532b2a7e0..ce686dc1bec7da3bf4d8ff2a07edb37586e38921 100644 (file)
--- a/threads/lu_utils.c
+++ b/threads/lu_utils.c
@@ -1,3 +1,25 @@
+/*
+ * Copyright (c) 2003 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@
+ */
 /*
  * Port and memory management for doing lookups to the lookup server
  * Copyright (C) 1989 by NeXT, Inc.
@@ -9,31 +31,32 @@
  *
  */
 #include <stdio.h>
+#include <stdlib.h>
+#include <sys/types.h>
+#include <unistd.h>
 #include <mach/mach.h>
+#include <mach/error.h>
 #include <servers/bootstrap.h>
 
 mach_port_t _lu_port = MACH_PORT_NULL;
 static name_t LOOKUP_NAME = "lookup daemon v2";
 
 mach_port_t _lookupd_port(mach_port_t port) {
+       kern_return_t ret;
+
        if (port != MACH_PORT_NULL) {
-               kern_return_t ret;
-               mach_port_t bp;
-               ret = task_get_bootstrap_port(mach_task_self(), &bp);
-               if (ret != KERN_SUCCESS) {
-                       mach_error("task_get_bootstrap_port() failed", ret);
-                       abort();
-               }
-               if (bp == MACH_PORT_NULL) {
-                       mach_error("task_get_bootstrap_port() returned MACH_PORT_NULL!", 0);
-                       return MACH_PORT_NULL;
-               }
-               ret = bootstrap_register(bp, LOOKUP_NAME, port);
+               ret = bootstrap_register(bootstrap_port, LOOKUP_NAME, port);
                if (ret != BOOTSTRAP_SUCCESS) {
                        mach_error("bootstrap_register() failed", ret);
                        abort();
                }
                 return port;
+       } else if ((_lu_port == MACH_PORT_NULL) && (getpid() > 1)) {
+               ret = bootstrap_look_up(bootstrap_port, LOOKUP_NAME, &_lu_port);
+               if (ret != BOOTSTRAP_SUCCESS && ret != BOOTSTRAP_UNKNOWN_SERVICE) {
+                       mach_error("bootstrap_look_up() failed", ret);
+                       _lu_port = MACH_PORT_NULL;
+               }
        }
        return _lu_port;
 }
@@ -42,27 +65,7 @@ mach_port_t _lookupd_port(mach_port_t port) {
 void
 _lu_fork_child()
 {
-       mach_port_t bp;
-       kern_return_t ret;
-       int pid = getpid();
-
        _lu_port = MACH_PORT_NULL;
-       if (pid > 2) {
-               ret = task_get_bootstrap_port(mach_task_self(), &bp);
-               if (ret != KERN_SUCCESS) {
-                       mach_error("task_get_bootstrap_port() failed", ret);
-                       abort();
-               }
-               if (bp == MACH_PORT_NULL) {
-                       fprintf(stderr, "task_get_bootstrap_port() returned MACH_PORT_NULL!\n");
-                       return;
-               }
-               ret = bootstrap_look_up(bp, LOOKUP_NAME, &_lu_port);
-               if (ret != BOOTSTRAP_SUCCESS && ret != BOOTSTRAP_UNKNOWN_SERVICE) {
-                       mach_error("bootstrap_look_up() failed", ret);
-                       _lu_port = MACH_PORT_NULL;
-               }
-       }
 }
 
 void
@@ -74,35 +77,9 @@ _lu_setport(mach_port_t desired)
        _lu_port = desired;
 }
 
-static int
-port_valid(mach_port_t port)
-{
-       mach_port_type_t ptype;
-       kern_return_t ret;
-       
-       ret = mach_port_type(mach_task_self(), port, &ptype);
-       if (ret != KERN_SUCCESS) {
-               mach_error("mach_port_type() failed", ret);
-               return 0;
-       }
-       return 1;
-}
-
 int
 _lu_running(void)
 {
-       if (_lu_port == MACH_PORT_NULL) {
-               _lu_fork_child();
-       }
-       if (_lu_port != MACH_PORT_NULL) {
-               if (port_valid(_lu_port)) {
-                       return (1);
-               }
-               _lu_fork_child();
-               if (port_valid(_lu_port)) {
-                       return (1);
-               }
-       }
-       return 0;
+       return ((_lu_port != MACH_PORT_NULL) ||
+               (_lookupd_port(MACH_PORT_NULL) != MACH_PORT_NULL));
 }
-