Loading...
--- Libc/Libc-391/threads/lu_utils.c
+++ Libc/Libc-262/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,32 +9,31 @@
*
*/
#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) {
- ret = bootstrap_register(bootstrap_port, LOOKUP_NAME, port);
+ 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)) {
- 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;
}
@@ -65,7 +42,27 @@
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
@@ -77,9 +74,35 @@
_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;
}
+