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 <ipc/ipcrsc.h>

#include <print.h>

#include <arch.h>

#include <proc/thread.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;

}

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

/* 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;

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

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

		}

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

		/* If the users accepted call, connect */

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

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

					  &TASK->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;

	if (is_system_method(method))

		return EPERM;

	phone = get_phone_and_lock(phoneid);

	if (!phone)

		return ENOENT;

	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;

	ipc_call_init(&call);

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

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

		return EPERM;

	phone = get_phone_and_lock(phoneid);

	if (!phone)

		return ENOENT;

	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;

	if (is_system_method(method))

		return IPC_CALLRET_FATAL;

	if (check_call_limit())

		return IPC_CALLRET_TEMPORARY;

	phone = get_phone_and_lock(phoneid);

	if (!phone)

		return IPC_CALLRET_FATAL;

	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;

	if (check_call_limit())

		return IPC_CALLRET_TEMPORARY;

	phone = get_phone_and_lock(phoneid);

	if (!phone)

		return IPC_CALLRET_FATAL;

	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

 *

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

 *          arg3 is not rewritten for certain system IPC

 */

__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_and_lock(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;

	ipc_call_init(&call);

	IPC_SET_METHOD(call.data, IPC_M_CONNECTTOME);

	IPC_SET_ARG1(call.data, arg1);

	IPC_SET_ARG2(call.data, arg2);

	phone = get_phone_and_lock(phoneid);

	if (!phone)

		return ENOENT;

	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;

	int newphid;

	phone = get_phone_and_lock(phoneid);

	if (!phone)

		return ENOENT;

	newphid = phone_alloc();

	if (newphid < 0)

		return ELIMIT;

	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_SET_ARG3(call.data, (__native)&TASK->phones[newphid]);

	ipc_call_sync(phone, &call);

	if (IPC_GET_RETVAL(call.data)) { /* Connection failed */

		phone_dealloc(newphid);

		return IPC_GET_RETVAL(call.data);

	}

	return newphid;

}

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

	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;

}