Loading...
threads/lu_utils.c Libc-583 Libc-262.2.12
--- Libc/Libc-583/threads/lu_utils.c
+++ Libc/Libc-262.2.12/threads/lu_utils.c
@@ -1,25 +1,3 @@
-/*
- * 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.
@@ -31,42 +9,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;
-mach_port_t _ds_port = MACH_PORT_NULL;
-mach_port_t _mbr_port = MACH_PORT_NULL;
-
 static name_t LOOKUP_NAME = "lookup daemon v2";
 
-#ifndef kDSStdMachDSLookupPortName
-#define kDSStdMachDSLookupPortName "com.apple.system.DirectoryService.libinfo_v1"
-#define kDSStdMachDSMembershipPortName "com.apple.system.DirectoryService.membership_v1"
-#endif
-
-mach_port_t
-_lookupd_port(mach_port_t port)
-{
-	kern_return_t status;
-
-	if (port != MACH_PORT_NULL)
-	{
-		status = bootstrap_register(bootstrap_port, LOOKUP_NAME, port);
-		if (status != BOOTSTRAP_SUCCESS) abort();
-
-		return port;
+mach_port_t _lookupd_port(mach_port_t port) {
+	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);
+		if (ret != BOOTSTRAP_SUCCESS) {
+			mach_error("bootstrap_register() failed", ret);
+			abort();
+		}
+                return port;
 	}
-	else if ((_lu_port == MACH_PORT_NULL) && (getpid() > 1))
-	{
-		status = bootstrap_look_up(bootstrap_port, LOOKUP_NAME, &_lu_port);
-		if ((status != BOOTSTRAP_SUCCESS) && (status != BOOTSTRAP_UNKNOWN_SERVICE)) _lu_port = MACH_PORT_NULL;
-	}
-
 	return _lu_port;
 }
 
@@ -74,36 +42,67 @@
 void
 _lu_fork_child()
 {
+	mach_port_t bp;
+	kern_return_t ret;
+	int pid = getpid();
+
 	_lu_port = MACH_PORT_NULL;
-	_ds_port = MACH_PORT_NULL;
-	_mbr_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
 _lu_setport(mach_port_t desired)
 {
-	if (_lu_port != MACH_PORT_NULL) mach_port_deallocate(mach_task_self(), _lu_port);
+	if (_lu_port != MACH_PORT_NULL) {
+		mach_port_deallocate(mach_task_self(), _lu_port);
+	}
 	_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)
 {
-	return ((_lu_port != MACH_PORT_NULL) || (_lookupd_port(MACH_PORT_NULL) != MACH_PORT_NULL));
+	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;
 }
 
-int
-_ds_running(void)
-{
-	kern_return_t status;
-
-	if (_ds_port != MACH_PORT_NULL) return 1;
-
-	status = bootstrap_look_up(bootstrap_port, kDSStdMachDSLookupPortName, &_ds_port);
-	if ((status != BOOTSTRAP_SUCCESS) && (status != BOOTSTRAP_UNKNOWN_SERVICE)) _ds_port = MACH_PORT_NULL;
-
-	status = bootstrap_look_up(bootstrap_port, kDSStdMachDSMembershipPortName, &_mbr_port);
-	if ((status != BOOTSTRAP_SUCCESS) && (status != BOOTSTRAP_UNKNOWN_SERVICE)) _mbr_port = MACH_PORT_NULL;
-
-	return (_ds_port != MACH_PORT_NULL);
-}