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.

 */

/** @addtogroup genericipc

 * @{

 */

/** @file

 */

#include <arch.h>

#include <proc/task.h>

#include <proc/thread.h>

#include <errno.h>

#include <memstr.h>

#include <debug.h>

#include <ipc/ipc.h>

#include <ipc/sysipc.h>

#include <ipc/irq.h>

#include <ipc/ipcrsc.h>

#include <arch/interrupt.h>

#include <print.h>

#include <syscall/copy.h>

#include <security/cap.h>

#include <mm/as.h>

#include <print.h>

/** Maximum buffer size allowed for IPC_M_DATA_SEND requests. */

#define DATA_SEND_LIMIT		(64 * 1024)

#define GET_CHECK_PHONE(phone, phoneid, err) \

{ \

	if (phoneid > IPC_MAX_PHONES) { \

		err; \

	} \

	phone = &TASK->phones[phoneid]; \

}

#define STRUCT_TO_USPACE(dst, src)	copy_to_uspace(dst, src, sizeof(*(src)))

/** Decide if the method is a system method.

 *

 * @param method	Method to be decided.

 *

 * @return		Return 1 if the method is a system method.

 *			Otherwise return 0.

 */

static inline int method_is_system(unative_t method)

{

	if (method <= IPC_M_LAST_SYSTEM)

		return 1;

	return 0;

}

/** Decide if the message with this method is forwardable.

 *

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

 *   it is useless

 *

 * @param method	Method to be decided.

 *

 * @return		Return 1 if the method is forwardable.

 *			Otherwise return 0.

 */

static inline int method_is_forwardable(unative_t method)

{

	switch (method) {

	case IPC_M_PHONE_HUNGUP:

		/* This message is meant only for the original recipient. */

		return 0;

	default:

		return 1;

	}

}

/** Decide if the message with this method is immutable on forward.

 *

 * - some system messages may be forwarded but their content cannot be altered

 *

 * @param method	Method to be decided.

 *

 * @return		Return 1 if the method is immutable on forward.

 *			Otherwise return 0.

 */

static inline int method_is_immutable(unative_t method)

{

	switch (method) {

	case IPC_M_AS_AREA_SEND:

	case IPC_M_AS_AREA_RECV:

	case IPC_M_DATA_SEND:

		return 1;

		break;

	default:

		return 0;

	}

}

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

 * Functions that preprocess answer before sending it to the recepient.

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

/** Decide if the caller (e.g. ipc_answer()) should save the old call contents

 * for answer_preprocess().

 *

 * @param call		Call structure to be decided.

 *

 * @return		Return 1 if the old call contents should be saved.

 *			Return 0 otherwise.

 */

static inline int answer_need_old(call_t *call)

{

	switch (IPC_GET_METHOD(call->data)) {

	case IPC_M_CONNECT_TO_ME:

	case IPC_M_CONNECT_ME_TO:

	case IPC_M_AS_AREA_SEND:

	case IPC_M_AS_AREA_RECV:

	case IPC_M_DATA_SEND:

		return 1;

	default:

		return 0;

	}

}

/** Interpret process answer as control information.

 *

 * This function is called directly after sys_ipc_answer().

 *

 * @param answer	Call structure with the answer.

 * @param olddata	Saved data of the request.

 *

 * @return		Return 0 on success or an error code. 

 */

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

{

	int phoneid;

	if (IPC_GET_RETVAL(answer->data) == EHANGUP) {

		/* In case of forward, hangup the forwared phone,

		 * not the originator

		 */

		spinlock_lock(&answer->data.phone->lock);

		spinlock_lock(&TASK->answerbox.lock);

		if (answer->data.phone->state == IPC_PHONE_CONNECTED) {

			list_remove(&answer->data.phone->link);

			answer->data.phone->state = IPC_PHONE_SLAMMED;

		}

		spinlock_unlock(&TASK->answerbox.lock);

		spinlock_unlock(&answer->data.phone->lock);

	}

	if (!olddata)

		return 0;

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

		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);

			/* Set 'phone hash' as arg3 of response */

			IPC_SET_ARG3(answer->data,

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

		}

	} else if (IPC_GET_METHOD(*olddata) == IPC_M_CONNECT_ME_TO) {

		/* If the users accepted call, connect */

		if (!IPC_GET_RETVAL(answer->data)) {

			ipc_phone_connect((phone_t *) IPC_GET_ARG3(*olddata),

			    &TASK->answerbox);

		}

	} else if (IPC_GET_METHOD(*olddata) == IPC_M_AS_AREA_SEND) {

		if (!IPC_GET_RETVAL(answer->data)) {

			/* Accepted, handle as_area receipt */

			ipl_t ipl;

			int rc;

			as_t *as;

			ipl = interrupts_disable();

			spinlock_lock(&answer->sender->lock);

			as = answer->sender->as;

			spinlock_unlock(&answer->sender->lock);

			interrupts_restore(ipl);

			rc = as_area_share(as, IPC_GET_ARG1(*olddata),

			    IPC_GET_ARG2(*olddata), AS,

			    IPC_GET_ARG1(answer->data), IPC_GET_ARG3(*olddata));

			IPC_SET_RETVAL(answer->data, rc);

			return rc;

		}

	} else if (IPC_GET_METHOD(*olddata) == IPC_M_AS_AREA_RECV) {

		if (!IPC_GET_RETVAL(answer->data)) { 

			ipl_t ipl;

			as_t *as;

			int rc;

			ipl = interrupts_disable();

			spinlock_lock(&answer->sender->lock);

			as = answer->sender->as;

			spinlock_unlock(&answer->sender->lock);

			interrupts_restore(ipl);

			rc = as_area_share(AS, IPC_GET_ARG1(answer->data),

			    IPC_GET_ARG2(*olddata), as, IPC_GET_ARG1(*olddata),

			    IPC_GET_ARG2(answer->data));

			IPC_SET_RETVAL(answer->data, rc);

		}

	} else if (IPC_GET_METHOD(*olddata) == IPC_M_DATA_SEND) {

		if (!IPC_GET_RETVAL(answer->data)) {

			int rc;

			uintptr_t dst;

			uintptr_t size;

			ASSERT(answer->buffer);

			dst = IPC_GET_ARG1(answer->data);

			size = IPC_GET_ARG3(answer->data);

			rc = copy_to_uspace((void *) dst, answer->buffer, size);

			if (rc != 0)

				IPC_SET_RETVAL(answer->data, rc);

			free(answer->buffer);

			answer->buffer = NULL;

		}

	}

	return 0;

}

/** Called before the request is sent.

 *

 * @param call		Call structure with the request.

 *

 * @return 		Return 0 on success, ELIMIT or EPERM on error.

 */

static int request_preprocess(call_t *call)

{

	int newphid;

	size_t size;

	uintptr_t src;

	int rc;

	switch (IPC_GET_METHOD(call->data)) {

	case IPC_M_CONNECT_ME_TO:

		newphid = phone_alloc();

		if (newphid < 0)

			return ELIMIT;

		/* Set arg3 for server */

		IPC_SET_ARG3(call->data, (unative_t) &TASK->phones[newphid]);

		call->flags |= IPC_CALL_CONN_ME_TO;

		call->priv = newphid;

		break;

	case IPC_M_AS_AREA_SEND:

		size = as_area_get_size(IPC_GET_ARG1(call->data));

		if (!size)

			return EPERM;

		IPC_SET_ARG2(call->data, size);

		break;

	case IPC_M_DATA_SEND:

		src = IPC_GET_ARG2(call->data);

		size = IPC_GET_ARG3(call->data);

		if ((size <= 0) || (size > DATA_SEND_LIMIT))

			return ELIMIT;

		call->buffer = (uint8_t *) malloc(size, 0);

		rc = copy_from_uspace(call->buffer, (void *) src, size);

		if (rc != 0) {

			free(call->buffer);

			return rc;

		}

		break;

	default:

		break;

	}

	return 0;

}

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

 * Functions called to process received call/answer before passing it to uspace.

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

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

 *

 * @param call		Call structure with the answer.

 */

static void process_answer(call_t *call)

{

	if (IPC_GET_RETVAL(call->data) == EHANGUP &&

	    (call->flags & IPC_CALL_FORWARDED))

		IPC_SET_RETVAL(call->data, EFORWARD);

	if (call->flags & IPC_CALL_CONN_ME_TO) {

		if (IPC_GET_RETVAL(call->data))

			phone_dealloc(call->priv);

		else

			IPC_SET_ARG3(call->data, call->priv);

	}

}

/** Do basic kernel processing of received call request.

 *

 * @param box		Destination answerbox structure.

 * @param call		Call structure with the request.

 *

 * @return 		Return 0 if the call should be passed to userspace.

 *			Return -1 if the call should be ignored.

 */

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

{

	int phoneid;

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

		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;

}

/** Make a fast call over IPC, wait for reply and return to user.

 *

 * This function can handle only three arguments of payload, but is faster than

 * the generic function (i.e. sys_ipc_call_sync_slow()).

 *

 * @param phoneid	Phone handle for the call.

 * @param method	Method of the call.

 * @param arg1		Service-defined payload argument.

 * @param arg2		Service-defined payload argument.

 * @param arg3		Service-defined payload argument.

 * @param data		Address of userspace structure where the reply call will

 *			be stored.

 *

 * @return		Returns 0 on success.

 *			Return ENOENT if there is no such phone handle.

 */

unative_t sys_ipc_call_sync_fast(unative_t phoneid, unative_t method,

    unative_t arg1, unative_t arg2, unative_t arg3, ipc_data_t *data)

{

	call_t call;

	phone_t *phone;

	int res;

	GET_CHECK_PHONE(phone, phoneid, return ENOENT);

	ipc_call_static_init(&call);

	IPC_SET_METHOD(call.data, method);

	IPC_SET_ARG1(call.data, arg1);

	IPC_SET_ARG2(call.data, arg2);

	IPC_SET_ARG3(call.data, arg3);

	if (!(res = request_preprocess(&call))) {

		ipc_call_sync(phone, &call);

		process_answer(&call);

	} else {

		IPC_SET_RETVAL(call.data, res);

	}

	STRUCT_TO_USPACE(&data->args, &call.data.args);

	return 0;

}

/** Make a synchronous IPC call allowing to transmit the entire payload.

 *

 * @param phoneid	Phone handle for the call.

 * @param question	Userspace address of call data with the request.

 * @param reply		Userspace address of call data where to store the

 *			answer.

 *

 * @return		Zero on success or an error code.

 */

unative_t sys_ipc_call_sync_slow(unative_t phoneid, ipc_data_t *question,

    ipc_data_t *reply)

{

	call_t call;

	phone_t *phone;

	int res;

	int rc;

	ipc_call_static_init(&call);

	rc = copy_from_uspace(&call.data.args, &question->args,

	    sizeof(call.data.args));

	if (rc != 0)

		return (unative_t) rc;

	GET_CHECK_PHONE(phone, phoneid, return ENOENT);

	if (!(res = request_preprocess(&call))) {

		ipc_call_sync(phone, &call);

		process_answer(&call);

	} else 

		IPC_SET_RETVAL(call.data, res);

	rc = STRUCT_TO_USPACE(&reply->args, &call.data.args);

	if (rc != 0)

		return rc;

	return 0;

}

/** Check that the task did not exceed the allowed limit of asynchronous calls.

 *

 * @return 		Return 0 if limit not reached or -1 if 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;

}

/** Make a fast asynchronous call over IPC.

 *

 * This function can only handle two arguments of payload, but is faster than

 * the generic function sys_ipc_call_async().

 *

 * @param phoneid	Phone handle for the call.

 * @param method	Method of the call.

 * @param arg1		Service-defined payload argument.

 * @param arg2		Service-defined payload argument.

 *

 * @return 		Return call hash on success.

 *			Return IPC_CALLRET_FATAL in case of a fatal error and 

 *			IPC_CALLRET_TEMPORARY if there are too many pending

 *			asynchronous requests; answers should be handled first.

 */

unative_t sys_ipc_call_async_fast(unative_t phoneid, unative_t method,

    unative_t arg1, unative_t arg2)

{

	call_t *call;

	phone_t *phone;

	int res;

	if (check_call_limit())

		return IPC_CALLRET_TEMPORARY;

	GET_CHECK_PHONE(phone, phoneid, return IPC_CALLRET_FATAL);

	call = ipc_call_alloc(0);

	IPC_SET_METHOD(call->data, method);

	IPC_SET_ARG1(call->data, arg1);

	IPC_SET_ARG2(call->data, arg2);

	IPC_SET_ARG3(call->data, 0);

	if (!(res = request_preprocess(call)))

		ipc_call(phone, call);

	else

		ipc_backsend_err(phone, call, res);

	return (unative_t) call;

}

/** Make an asynchronous IPC call allowing to transmit the entire payload.

 *

 * @param phoneid	Phone handle for the call.

 * @param data		Userspace address of call data with the request.

 *

 * @return		See sys_ipc_call_async_fast(). 

 */

unative_t sys_ipc_call_async(unative_t phoneid, ipc_data_t *data)

{

	call_t *call;

	phone_t *phone;

	int res;

	int rc;

	if (check_call_limit())

		return IPC_CALLRET_TEMPORARY;

	GET_CHECK_PHONE(phone, phoneid, return IPC_CALLRET_FATAL);

	call = ipc_call_alloc(0);

	rc = copy_from_uspace(&call->data.args, &data->args,

	    sizeof(call->data.args));

	if (rc != 0) {

		ipc_call_free(call);

		return (unative_t) rc;

	}

	if (!(res = request_preprocess(call)))

		ipc_call(phone, call);

	else

		ipc_backsend_err(phone, call, res);

	return (unative_t) call;

}

/** Forward a received call to another destination.

 *

 * @param callid	Hash of the call to forward.

 * @param phoneid	Phone handle to use for forwarding.

 * @param method	New method to use for the forwarded call.

 * @param arg1		New value of the first argument for the forwarded call.

 *

 * @return		Return 0 on succes, otherwise return an error code.

 *

 * In case the original method is a system method, ARG1 and ARG2 are overwritten

 * in the forwarded message with the new method and the new arg1, respectively.

 * Otherwise the METHOD and ARG1 are rewritten with the new method and arg1,

 * respectively. Also note there is a set of immutable methods, for which the

 * new method and argument is not set and these values are ignored.

 *

 * Warning: If implementing non-fast version, make sure that

 *          ARG3 is not rewritten for certain system IPC

 */

unative_t sys_ipc_forward_fast(unative_t callid, unative_t phoneid,

    unative_t method, unative_t arg1)

{

	call_t *call;

	phone_t *phone;

	call = get_call(callid);

	if (!call)

		return ENOENT;

	call->flags |= IPC_CALL_FORWARDED;

	GET_CHECK_PHONE(phone, phoneid, { 

		IPC_SET_RETVAL(call->data, EFORWARD);

		ipc_answer(&TASK->answerbox, call);

		return ENOENT;

	});		

	if (!method_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 the method is immutable, don't change anything.

	 */

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

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

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

				phone_dealloc(IPC_GET_ARG3(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);

		}

	}

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

}

/** Answer an IPC call - fast version.

 *

 * This function can handle only two return arguments of payload, but is faster

 * than the generic sys_ipc_answer().

 *

 * @param callid	Hash of the call to be answered.

 * @param retval	Return value of the answer.

 * @param arg1		Service-defined return value.

 * @param arg2		Service-defined return value.

 *

 * @return		Return 0 on success, otherwise return an error code.	

 */

unative_t sys_ipc_answer_fast(unative_t callid, unative_t retval,

    unative_t arg1, unative_t arg2)

{

	call_t *call;

	ipc_data_t saved_data;

	int saveddata = 0;

	int rc;

	/* Do not answer notification callids */

	if (callid & IPC_CALLID_NOTIFICATION)

		return 0;

	call = get_call(callid);

	if (!call)

		return ENOENT;

	if (answer_need_old(call)) {

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

		saveddata = 1;

	}

	IPC_SET_RETVAL(call->data, retval);

	IPC_SET_ARG1(call->data, arg1);

	IPC_SET_ARG2(call->data, arg2);

	rc = answer_preprocess(call, saveddata ? &saved_data : NULL);

	ipc_answer(&TASK->answerbox, call);

	return rc;

}

/** Answer an IPC call.

 *

 * @param callid	Hash of the call to be answered.

 * @param data		Userspace address of call data with the answer.

 *

 * @return		Return 0 on success, otherwise return an error code.

 */

unative_t sys_ipc_answer(unative_t callid, ipc_data_t *data)

{

	call_t *call;

	ipc_data_t saved_data;

	int saveddata = 0;

	int rc;

	/* Do not answer notification callids */

	if (callid & IPC_CALLID_NOTIFICATION)

		return 0;

	call = get_call(callid);

	if (!call)

		return ENOENT;

	if (answer_need_old(call)) {

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

		saveddata = 1;

	}

	rc = copy_from_uspace(&call->data.args, &data->args, 

	    sizeof(call->data.args));

	if (rc != 0)

		return rc;

	rc = answer_preprocess(call, saveddata ? &saved_data : NULL);

	ipc_answer(&TASK->answerbox, call);

	return rc;

}

/** Hang up a phone.

 *

 * @param		Phone handle of the phone to be hung up.

 *

 * @return		Return 0 on success or an error code.

 */

unative_t sys_ipc_hangup(int phoneid)

{

	phone_t *phone;

	GET_CHECK_PHONE(phone, phoneid, return ENOENT);

	if (ipc_phone_hangup(phone))

		return -1;

	return 0;

}

/** Wait for an incoming IPC call or an answer.

 *

 * @param calldata	Pointer to buffer where the call/answer data is stored.

 * @param usec 		Timeout. See waitq_sleep_timeout() for explanation.

 * @param flags		Select mode of sleep operation. See waitq_sleep_timeout()

 *			for explanation.

 *

 * @return 		Hash of the call.

 *			If IPC_CALLID_NOTIFICATION bit is set in the hash, the

 *			call is a notification. IPC_CALLID_ANSWERED denotes an

 *			answer.

 */

unative_t sys_ipc_wait_for_call(ipc_data_t *calldata, uint32_t usec, int flags)

{

	call_t *call;

restart:	

	call = ipc_wait_for_call(&TASK->answerbox, usec,

	    flags | SYNCH_FLAGS_INTERRUPTIBLE);

	if (!call)

		return 0;

	if (call->flags & IPC_CALL_NOTIF) {

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

		/* Set in_phone_hash to the interrupt counter */

		call->data.phone = (void *) call->priv;

		STRUCT_TO_USPACE(calldata, &call->data);

		ipc_call_free(call);

		return ((unative_t) call) | IPC_CALLID_NOTIFICATION;

	}

	if (call->flags & IPC_CALL_ANSWERED) {

		process_answer(call);

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

		atomic_dec(&TASK->active_calls);

		if (call->flags & IPC_CALL_DISCARD_ANSWER) {

			ipc_call_free(call);

			goto restart;

		}

		STRUCT_TO_USPACE(&calldata->args, &call->data.args);

		ipc_call_free(call);

		return ((unative_t) call) | IPC_CALLID_ANSWERED;

	}

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

		goto restart;

	/* Include phone address('id') of the caller in the request,

	 * copy whole call->data, not only call->data.args */

	if (STRUCT_TO_USPACE(calldata, &call->data)) {

		return 0;

	}

	return (unative_t)call;

}

/** Connect an IRQ handler to a task.

 *

 * @param inr		IRQ number.

 * @param devno		Device number.

 * @param method	Method to be associated with the notification.

 * @param ucode		Uspace pointer to the top-half pseudocode.

 *

 * @return		EPERM or a return code returned by ipc_irq_register().

 */

unative_t sys_ipc_register_irq(inr_t inr, devno_t devno, unative_t method,

    irq_code_t *ucode)

{

	if (!(cap_get(TASK) & CAP_IRQ_REG))

		return EPERM;

	return ipc_irq_register(&TASK->answerbox, inr, devno, method, ucode);

}

/** Disconnect an IRQ handler from a task.

 *

 * @param inr		IRQ number.

 * @param devno		Device number.

 *

 * @return		Zero on success or EPERM on error..

 */

unative_t sys_ipc_unregister_irq(inr_t inr, devno_t devno)

{

	if (!(cap_get(TASK) & CAP_IRQ_REG))

		return EPERM;

	ipc_irq_unregister(&TASK->answerbox, inr, devno);

	return 0;

}

/** @}

 */