Open Menu / threads / lu_utils.c
Loading...
  1
  2
  3
  4
  5
  6
  7
  8
  9
 10
 11
 12
 13
 14
 15
 16
 17
 18
 19
 20
 21
 22
 23
 24
 25
 26
 27
 28
 29
 30
 31
 32
 33
 34
 35
 36
 37
 38
 39
 40
 41
 42
 43
 44
 45
 46
 47
 48
 49
 50
 51
 52
 53
 54
 55
 56
 57
 58
 59
 60
 61
 62
 63
 64
 65
 66
 67
 68
 69
 70
 71
 72
 73
 74
 75
 76
 77
 78
 79
 80
 81
 82
 83
 84
 85
 86
 87
 88
 89
 90
 91
 92
 93
 94
 95
 96
 97
 98
 99
100
101
102
103
104
105
106
107
108
/*
 * Port and memory management for doing lookups to the lookup server
 * Copyright (C) 1989 by NeXT, Inc.
 */
/*
 * HISTORY
 * 27-Mar-90  Gregg Kellogg (gk) at NeXT
 *	Changed to use bootstrap port instead of service port.
 *
 */
#include <stdio.h>
#include <mach/mach.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) {
	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;
	}
	return _lu_port;
}

/* called as child starts up.  mach ports aren't inherited: trash cache */
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
_lu_setport(mach_port_t desired)
{
	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)
{
	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;
}