Subversion Repositories HelenOS

/*

 * Copyright (C) 2006 Ondrej Palkovsky

 * All rights reserved.

 *

 * Redistribution and use in source and binary forms, with or without

 * modification, are permitted provided that the following conditions

 * are met:

 *

 * - Redistributions of source code must retain the above copyright

 *   notice, this list of conditions and the following disclaimer.

 * - Redistributions in binary form must reproduce the above copyright

 *   notice, this list of conditions and the following disclaimer in the

 *   documentation and/or other materials provided with the distribution.

 * - The name of the author may not be used to endorse or promote products

 *   derived from this software without specific prior written permission.

 *

 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR

 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES

 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.

 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,

 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT

 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,

 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY

 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT

 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF

 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

 */

#include <arch.h>

#include <proc/task.h>

#include <errno.h>

#include <mm/page.h>

#include <memstr.h>

#include <debug.h>

#include <ipc/ipc.h>

#include <ipc/sysipc.h>

#include <print.h>

/* TODO: multi-threaded connect-to-me can cause race condition

 * on phone, add counter + thread_kill??

 *

 */

/** Return true if the method is a system method */

static inline int is_system_method(__native method)

{

	if (method <= IPC_M_LAST_SYSTEM)

		return 1;

	return 0;

}

/** Return true if the message with this method is forwardable

 *

 * - some system messages may be forwarded, for some of them

 *   it is useless

 */

static inline int is_forwardable(__native method)

{

	return 1;

}

/** Find call_t * in call table according to callid

 *

 * @return NULL on not found, otherwise pointer to call structure

 */

static inline call_t * get_call(__native callid)

{

	/* TODO: Traverse list of dispatched calls and find one */

	/* TODO: locking of call, ripping it from dispatched calls etc.  */

	return (call_t *) callid;

}

/** Return pointer to phone identified by phoneid or NULL if non-existent */

static phone_t * get_phone(__native phoneid)

{

	phone_t *phone;

	if (phoneid >= IPC_MAX_PHONES)

		return NULL;

	phone = &TASK->phones[phoneid];

	if (!phone->callee)

		return NULL;

	return phone;

}

/** Allocate new phone slot in current TASK structure */

static int phone_alloc(void)

{

	int i;

	spinlock_lock(&TASK->lock);

	for (i=0; i < IPC_MAX_PHONES; i++) {

		if (!TASK->phones[i].busy) {

			TASK->phones[i].busy = 1;

			break;

		}

	}

	spinlock_unlock(&TASK->lock);

	if (i >= IPC_MAX_PHONES)

		return -1;

	return i;

}

/** Disconnect phone */

static void phone_dealloc(int phoneid)

{

	spinlock_lock(&TASK->lock);

	ASSERT(TASK->phones[phoneid].busy);

	if (TASK->phones[phoneid].callee)

		ipc_phone_destroy(&TASK->phones[phoneid]);

	TASK->phones[phoneid].busy = 0;

	spinlock_unlock(&TASK->lock);

}

static void phone_connect(int phoneid, answerbox_t *box)

{

	phone_t *phone = &TASK->phones[phoneid];

	ipc_phone_connect(phone, box);

}

/****************************************************/

/* Functions that preprocess answer before sending 

 * it to the recepient

 */

/** Return true if the caller (ipc_answer) should save

 * the old call contents and call answer_preprocess

 */

static inline int answer_will_preprocess(call_t *call)

{

	if (IPC_GET_METHOD(call->data) == IPC_M_CONNECTTOME)

		return 1;

	return 0;

}

/** Interpret process answer as control information */

static inline void answer_preprocess(call_t *answer, ipc_data_t *olddata)

{

	int phoneid;

	if (IPC_GET_METHOD(*olddata) == IPC_M_CONNECTTOME) {

		phoneid = IPC_GET_ARG3(*olddata);

		if (IPC_GET_RETVAL(answer->data)) {

			/* The connection was not accepted */

			phone_dealloc(phoneid);

		} else {

			/* The connection was accepted */

			phone_connect(phoneid,&answer->sender->answerbox);

		}

	}

}

/****************************************************/

/* Functions called to process received call/answer 

 * before passing to uspace

 */

/** Do basic kernel processing of received call answer */

static int process_answer(answerbox_t *box,call_t *call)

{

	return 0;

}

/** Do basic kernel processing of received call request

 *

 * @return 0 - the call should be passed to userspace, 1 - ignore call

 */

static int process_request(answerbox_t *box,call_t *call)

{

	int phoneid;

	if (IPC_GET_METHOD(call->data) == IPC_M_CONNECTTOME) {

		phoneid = phone_alloc();

		if (phoneid < 0) { /* Failed to allocate phone */

			IPC_SET_RETVAL(call->data, ELIMIT);

			ipc_answer(box,call);

			return -1;

		}

		IPC_SET_ARG3(call->data, phoneid);

	} 

	return 0;

}

/** Send a call over IPC, wait for reply, return to user

 *

 * @return Call identification, returns -1 on fatal error, 

           -2 on 'Too many async request, handle answers first

 */

__native sys_ipc_call_sync_fast(__native phoneid, __native method, 

				__native arg1, __native *data)

{

	call_t call;

	phone_t *phone;

	phone = get_phone(phoneid);

	if (!phone)

		return ENOENT;

	if (is_system_method(method))

		return EPERM;

	ipc_call_init(&call);

	IPC_SET_METHOD(call.data, method);

	IPC_SET_ARG1(call.data, arg1);

	ipc_call_sync(phone, &call);

	copy_to_uspace(data, &call.data, sizeof(call.data));

	return 0;

}

/** Synchronous IPC call allowing to send whole message */

__native sys_ipc_call_sync(__native phoneid, __native *question, 

			   __native *reply)

{

	call_t call;

	phone_t *phone;

	phone = get_phone(phoneid);

	if (!phone)

		return ENOENT;

	ipc_call_init(&call);

	copy_from_uspace(&call.data, question, sizeof(call.data));

	if (is_system_method(IPC_GET_METHOD(call.data)))

		return EPERM;

	ipc_call_sync(phone, &call);

	copy_to_uspace(reply, &call.data, sizeof(call.data));

	return 0;

}

/** Check that the task did not exceed allowed limit

 *

 * @return 0 - Limit OK,   -1 - limit exceeded

 */

static int check_call_limit(void)

{

	if (atomic_preinc(&TASK->active_calls) > IPC_MAX_ASYNC_CALLS) {

		atomic_dec(&TASK->active_calls);

		return -1;

	}

	return 0;

}

/** Send an asynchronous call over ipc

 *

 * @return Call identification, returns -1 on fatal error, 

           -2 on 'Too many async request, handle answers first

 */

__native sys_ipc_call_async_fast(__native phoneid, __native method, 

				 __native arg1, __native arg2)

{

	call_t *call;

	phone_t *phone;

	phone = get_phone(phoneid);

	if (!phone)

		return IPC_CALLRET_FATAL;

	if (is_system_method(method))

		return IPC_CALLRET_FATAL;

	if (check_call_limit())

		return IPC_CALLRET_TEMPORARY;

	call = ipc_call_alloc();

	IPC_SET_METHOD(call->data, method);

	IPC_SET_ARG1(call->data, arg1);

	IPC_SET_ARG2(call->data, arg2);

	ipc_call(phone, call);

	return (__native) call;

}

/** Synchronous IPC call allowing to send whole message

 *

 * @return The same as sys_ipc_call_async

 */

__native sys_ipc_call_async(__native phoneid, __native *data)

{

	call_t *call;

	phone_t *phone;

	phone = get_phone(phoneid);

	if (!phone)

		return IPC_CALLRET_FATAL;

	if (check_call_limit())

		return IPC_CALLRET_TEMPORARY;

	call = ipc_call_alloc();

	copy_from_uspace(&call->data, data, sizeof(call->data));

	if (is_system_method(IPC_GET_METHOD(call->data))) {

		ipc_call_free(call);

		return EPERM;

	}

	ipc_call(phone, call);

	return (__native) call;

}

/** Forward received call to another destination

 *

 * The arg1 and arg2 are changed in the forwarded message

 */

__native sys_ipc_forward_fast(__native callid, __native phoneid,

			      __native method, __native arg1)

{

	call_t *call;

	phone_t *phone;

	call = get_call(callid);

	if (!call)

		return ENOENT;

	phone = get_phone(phoneid);

	if (!phone) {

		IPC_SET_RETVAL(call->data, EFORWARD);

		ipc_answer(&TASK->answerbox, call);

		return ENOENT;

	}

	if (!is_forwardable(IPC_GET_METHOD(call->data))) {

		IPC_SET_RETVAL(call->data, EFORWARD);

		ipc_answer(&TASK->answerbox, call);

		return EPERM;

	}

	/* Userspace is not allowed to change method of system methods

	 * on forward, allow changing ARG1 and ARG2 by means of method and arg1

	 */

	if (is_system_method(IPC_GET_METHOD(call->data))) {

		IPC_SET_ARG1(call->data, method);

		IPC_SET_ARG2(call->data, arg1);

	} else {

		IPC_SET_METHOD(call->data, method);

		IPC_SET_ARG1(call->data, arg1);

	}

	ipc_forward(call, phone->callee, &TASK->answerbox);

	return 0;

}

/** Send IPC answer */

__native sys_ipc_answer_fast(__native callid, __native retval, 

			     __native arg1, __native arg2)

{

	call_t *call;

	ipc_data_t saved_data;

	int preprocess = 0;

	call = get_call(callid);

	if (!call)

		return ENOENT;

	if (answer_will_preprocess(call)) {

		memcpy(&saved_data, &call->data, sizeof(call->data));

		preprocess = 1;

	}

	IPC_SET_RETVAL(call->data, retval);

	IPC_SET_ARG1(call->data, arg1);

	IPC_SET_ARG2(call->data, arg2);

	if (preprocess)

		answer_preprocess(call, &saved_data);

	ipc_answer(&TASK->answerbox, call);

	return 0;

}

/** Send IPC answer */

inline __native sys_ipc_answer(__native callid, __native *data)

{

	call_t *call;

	ipc_data_t saved_data;

	int preprocess = 0;

	call = get_call(callid);

	if (!call)

		return ENOENT;

	if (answer_will_preprocess(call)) {

		memcpy(&saved_data, &call->data, sizeof(call->data));

		preprocess = 1;

	}

	copy_from_uspace(&call->data, data, sizeof(call->data));

	if (preprocess)

		answer_preprocess(call, &saved_data);

	ipc_answer(&TASK->answerbox, call);

	return 0;

}

/** Ask the other side of connection to do 'callback' connection

 *

 * @return 0 if no error, error otherwise 

 */

__native sys_ipc_connect_to_me(__native phoneid, __native arg1,

			       __native arg2, task_id_t *taskid)

{

	call_t call;

	phone_t *phone;

	phone = get_phone(phoneid);

	if (!phone)

		return ENOENT;

	ipc_call_init(&call);

	IPC_SET_METHOD(call.data, IPC_M_CONNECTTOME);

	IPC_SET_ARG1(call.data, arg1);

	IPC_SET_ARG2(call.data, arg2);

	ipc_call_sync(phone, &call);

	if (!IPC_GET_RETVAL(call.data) && taskid)

		copy_to_uspace(taskid, 

			       &phone->callee->task->taskid,

			       sizeof(TASK->taskid));

	return IPC_GET_RETVAL(call.data);

}

/** Ask target process to connect me somewhere

 *

 * @return phoneid - on success, error otherwise

 */

__native sys_ipc_connect_me_to(__native phoneid, __native arg1,

			       __native arg2)

{

	call_t call;

	phone_t *phone;

	phone = get_phone(phoneid);

	if (!phone)

		return ENOENT;

	ipc_call_init(&call);

	IPC_SET_METHOD(call.data, IPC_M_CONNECTMETO);

	IPC_SET_ARG1(call.data, arg1);

	IPC_SET_ARG2(call.data, arg2);

	ipc_call_sync(phone, &call);

	if (!IPC_GET_RETVAL(call.data)) {

		/* Everybody accepted, we should be connected by now */

	}

	return 0;

}

/** Wait for incoming ipc call or answer

 *

 * Generic function - can serve either as inkernel or userspace call

 * - inside kernel does probably unnecessary copying of data (TODO)

 *

 * @param result 

 * @param taskid 

 * @param flags

 * @return Callid, if callid & 1, then the call is answer

 */

inline __native sys_ipc_wait_for_call(ipc_data_t *calldata, 

				      task_id_t *taskid,

				      __native flags)

{

	call_t *call;

restart:	

	call = ipc_wait_for_call(&TASK->answerbox, flags);

	printf("Received call %P from sender: %P\n", call, call->sender);

	if (call->flags & IPC_CALL_ANSWERED) {

		if (process_answer(&TASK->answerbox, call))

			goto restart;

		copy_to_uspace(calldata, &call->data, sizeof(call->data));

		atomic_dec(&TASK->active_calls);

		ASSERT(! (call->flags & IPC_CALL_STATIC_ALLOC));

		ipc_call_free(call);

		return ((__native)call) | IPC_CALLID_ANSWERED;

	}

	if (process_request(&TASK->answerbox, call))

		goto restart;

	copy_to_uspace(calldata, &call->data, sizeof(call->data));

	copy_to_uspace(taskid, (void *)&TASK->taskid, sizeof(TASK->taskid));

	return (__native)call;

}