File:  [ELWIX - Embedded LightWeight unIX -] / libaitrpc / inc / aitrpc.h
Revision 1.8.2.12: download - view: text, annotated - select for diffs - revision graph
Wed May 16 13:32:47 2012 UTC (12 years, 1 month ago) by misho
Branches: rpc3_3
Diff to: branchpoint 1.8: preferred, colored
...

/*************************************************************************
* (C) 2010 AITNET ltd - Sofia/Bulgaria - <misho@aitbg.com>
*  by Michael Pounov <misho@openbsd-bg.org>
*
* $Author: misho $
* $Id: aitrpc.h,v 1.8.2.12 2012/05/16 13:32:47 misho Exp $
*
**************************************************************************
The ELWIX and AITNET software is distributed under the following
terms:

All of the documentation and software included in the ELWIX and AITNET
Releases is copyrighted by ELWIX - Sofia/Bulgaria <info@elwix.org>

Copyright 2004, 2005, 2006, 2007, 2008, 2009, 2010, 2011, 2012
	by Michael Pounov <misho@elwix.org>.  All rights reserved.

Redistribution and use in source and binary forms, with or without
modification, are permitted provided that the following conditions
are met:
1. Redistributions of source code must retain the above copyright
   notice, this list of conditions and the following disclaimer.
2. 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.
3. All advertising materials mentioning features or use of this software
   must display the following acknowledgement:
This product includes software developed by Michael Pounov <misho@elwix.org>
ELWIX - Embedded LightWeight unIX and its contributors.
4. Neither the name of AITNET nor the names of its contributors
   may be used to endorse or promote products derived from this software
   without specific prior written permission.

THIS SOFTWARE IS PROVIDED BY AITNET AND CONTRIBUTORS ``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 REGENTS OR CONTRIBUTORS 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.
*/
#ifndef __AITRPC_H
#define __AITRPC_H


#include <assert.h>
#include <stdlib.h>
#include <string.h>
#include <errno.h>
#include <sys/types.h>
#include <sys/param.h>
#if !defined(__NetBSD__)
#include <sys/limits.h>
#endif
#include <sys/socket.h>
#include <sys/queue.h>
#include <aitio.h>
#include <aitsched.h>


#define RPC_VERSION		4
#define RPC_DEFPORT		2611

/* RPC call request flags */

#define RPC_REPLY		0x0
#define RPC_NOREPLY		0x1

/* RPC builtin registed calls */

#define CALL_MAX_ID		65535

#define CALL_BLOBSHUTDOWN	"rpcBLOBServerShutdown"
#define CALL_BLOBSHUTDOWN_ID	65530
#define CALL_BLOBCLIENTS	"rpcBLOBServerClients"
#define CALL_BLOBCLIENTS_ID	65529
#define CALL_BLOBVARS		"rpcBLOBServerVars"
#define CALL_BLOBVARS_ID	65528
#define CALL_BLOBSTATE		"rpcBLOBServerState"
#define CALL_BLOBSTATE_ID	65527

#define CALL_SRVSHUTDOWN	"rpcServerShutdown"
#define CALL_SRVSHUTDOWN_ID	65534
#define CALL_SRVCLIENTS		"rpcServerClients"
#define CALL_SRVCLIENTS_ID	65533
#define CALL_SRVCALLS		"rpcServerCalls"
#define CALL_SRVCALLS_ID	65532
#define CALL_SRVSESSIONS	"rpcServerSessions"
#define CALL_SRVSESSIONS_ID	65531


/* RPC types */

typedef enum {
	ok, error, no, 				/* for blob reply */
	get, set, unset				/* for blob request */
} cmd_type_t;


#define RPC_CALLBACK_CHECK_INPUT(_f)	do { \
						assert((_f)); \
						if (!(_f)) { \
							rpc_SetErr(EINVAL, \
									"Invalid callback parameters ..."); \
							return -1; \
						} \
					} while (0)
#define RPC_CALLBACK_CHK_ARGS(_f, _n)	do { \
						RPC_CALLBACK_CHECK_INPUT((_f)); \
						if ((_f) && io_arraySize((_f)->func_vars) < _n) { \
							rpc_SetErr(EINVAL, \
									"Not enough number of arguments!"); \
							return -1; \
						} \
					} while (0)


/* RPC session identification */

typedef struct {
	uint8_t		sess_version;
	uint32_t	sess_program;
	uint8_t		sess_process;
} __packed rpc_sess_t;		/* size == 6 bytes */


/* Server managment RPC functions ... */

/* Network RPC packet - Client request */

struct tagRPCCall {
	rpc_sess_t	call_session;

	uint16_t	call_seq;
	uint16_t	call_len;
	uint16_t	call_crc;

	union {
		struct {
			uint64_t	flags;
		}	call_req;
		struct {
			int32_t		ret;
			int32_t		eno;
		}	call_rep;
	};

	uint16_t	call_tag;
	uint16_t	call_argc;
	ait_val_t	call_argv[0];
} __packed;			/* size == 24 bytes */

/* Network BLOB packet - Header */

struct tagBLOBHdr {
	rpc_sess_t	hdr_session;
	uint8_t		hdr_cmd;
	uint32_t	hdr_var;
	uint32_t	hdr_len;
	uint32_t	hdr_ret;
	uint16_t	hdr_crc;
	uint8_t		hdr_pad;
} __packed;

/* Network RPC client & server elements */

/* RPC function registration element! */
typedef struct tagRPCFunc {
	ait_val_t		func_name;

	array_t			*func_vars;

	void			*func_parent;
	TAILQ_ENTRY(tagRPCFunc)	func_node;
} rpc_func_t;


/* BLOB register element */
typedef struct tagBLOB {
	uint32_t		blob_var;	/* BLOB id */

	size_t			blob_len;	/* size of allocated BLOB data */
	void			*blob_data;	/* mapped BLOB data */

	TAILQ_ENTRY(tagBLOB)	blob_node;
} rpc_blob_t;


typedef struct {
	int		cli_id;		/* slot id */
	int		cli_sock;	/* socket fd */
	io_sockaddr_t	cli_sa;		/* host address */
	ait_val_t	cli_buf;	/* network buffer */

	void		*cli_parent;	/* pointer to parent rpc_srv_t for server or to rpc_sess_t for client */
} rpc_cli_t;

typedef struct {
	rpc_sess_t			srv_session;	/* RPC session registration info */
	int				srv_netbuf;	/* size of network buffer */

	sched_root_task_t		*srv_root;	/* RPC server scheduler */
	intptr_t			srv_kill;	/* Scheduler condition variable */

	rpc_cli_t			srv_server;	/* RPC server socket */
	array_t				*srv_clients;	/* connected rpc client sockets */

	TAILQ_HEAD(, tagRPCFunc)	srv_funcs;	/* RPC functions list */

	struct {
		sched_root_task_t		*root;		/* BLOB server scheduler */
		intptr_t			kill;		/* BLOB server state: ==0 disable | !=0 enable */

		ait_val_t			dir;		/* BLOB states directory */

		rpc_cli_t			server;		/* BLOB server socket */
		array_t				*clients;	/* connected blob client sockets */

		TAILQ_HEAD(, tagBLOB)		blobs;		/* registered blob variables list */
	} 				srv_blob;
} rpc_srv_t;


/* 
 * (*rpc_callback_t)() - Callback type definition for RPC call in server process
 *
 * @arg1 = current execution RPC call function
 * @arg2 = number of items in input array from call request
 * @arg3 = input array with values from RPC call execution request
 * return: -1 error or >-1 success execution
 */
typedef int (*rpc_callback_t)(rpc_func_t *, int, array_t *);


// -----------------------------------------------------------------------

/* Error support functions */

// rpc_GetErrno() Get error code of last operation
inline int rpc_GetErrno();
// rpc_GetError() Get error text of last operation
inline const char *rpc_GetError();


/*
 * rpc_chkPktSession() - Check RPC session
 *
 * @p = packet session
 * @s = active session
 * return: -1, 1, 2, 3 are errors or 0 ok
 */
inline int rpc_chkPktSession(rpc_sess_t *p, rpc_sess_t *s);
/*
 * rpc_addPktSession() - Prepare session into network format
 *
 * @p = packet session
 * @s = host session
 * return: -1 error or 0 ok
 */
inline int rpc_addPktSession(rpc_sess_t *p, rpc_sess_t *s);
/*
 * rpc_register_srvServices() - Register internal service functions
 *
 * @srv = RPC server instance
 * return: -1 error or 0 ok
 */
int rpc_register_srvServices(rpc_srv_t * __restrict srv);
/*
 * rpc_register_blobServices() - Register internal service functions
 *
 * @srv = RPC server instance
 * return: -1 error or 0 ok
 */
int rpc_register_blobServices(rpc_srv_t * __restrict srv);


/* RPC Server side functions */

/*
 * rpc_srv_initServer() - Init & create RPC Server
 *
 * @regProgID = ProgramID for authentication & recognition
 * @regProcID = ProcessID for authentication & recognition
 * @concurentClients = Concurent clients at same time to this server
 * @netBuf = Network buffer length, if =0 == BUFSIZ (also meaning max RPC packet)
 * @csHost = Host name or address for bind server, if NULL any address
 * @Port = Port for bind server, if Port == 0 default port is selected
 * return: NULL == error or !=NULL bind and created RPC server instance
 */
rpc_srv_t *rpc_srv_initServer(unsigned int regProgID, unsigned char regProcID, 
		int concurentClients, int netBuf, 
		const char *csHost, unsigned short Port);
/*
 * rpc_srv_endServer() - Destroy RPC server, close all opened sockets and free resources
 *
 * @psrv = RPC Server instance
 * return: none
 */
void rpc_srv_endServer(rpc_srv_t ** __restrict psrv);
/*
 * rpc_srv_loopServer() - Execute Main server loop and wait for clients requests
 *
 * @srv = RPC Server instance
 * return: -1 error or 0 ok, infinite loop ...
 */
int rpc_srv_loopServer(rpc_srv_t * __restrict srv);
#define rpc_srv_execServer(_srv, _sync)	do { assert((_srv)); pthread_t __tid; \
						if (!(_srv)->srv_kill) { \
							pthread_create(&__tid, NULL, (void*(*)(void*)) \
									rpc_srv_loopServer, (_srv)); \
							if ((_sync)) \
								pthread_join(__tid, (void**) (_sync)); \
							else \
								pthread_detach(__tid); \
						} } while (0)

/*
 * rpc_srv_initBLOBServer() - Init & create BLOB Server
 *
 * @srv = RPC server instance
 * @Port = Port for bind server, if Port == 0 default port is selected
 * @diskDir = Disk place for BLOB file objects
 * return: -1 == error or 0 bind and created BLOB server instance
 */
int rpc_srv_initBLOBServer(rpc_srv_t * __restrict srv, u_short Port, const char *diskDir);
/*
 * rpc_srv_endBLOBServer() - Destroy BLOB server, close all opened sockets and free resources
 *
 * @srv = RPC Server instance
 * return: none
 */
void rpc_srv_endBLOBServer(rpc_srv_t * __restrict srv);
/*
 * rpc_srv_loopBLOB() - Execute Main BLOB server loop and wait for clients requests
 *
 * @srv = RPC Server instance
 * return: -1 error or 0 ok, infinite loop ...
 */
int rpc_srv_loopBLOBServer(rpc_srv_t * __restrict srv);
#define rpc_srv_execBLOBServer(_srv)	do { assert((_srv)); pthread_t __tid; \
						if (!(_srv)->srv_kill && !(_srv)->srv_blob.kill) { \
							pthread_create(&__tid, NULL, (void*(*)(void*)) \
									rpc_srv_loopBLOBServer, (_srv)); \
							pthread_detach(__tid); \
						} \
					} while (0)

/*
 * rpc_srv_registerCall() - Register call to RPC server
 *
 * @srv = RPC Server instance
 * @tag = Function tag
 * @funcaddr = Function address
 * @args = Number of return function arguments
 * return: -1 error, 0 already registered tag or 1 register ok
 */
int rpc_srv_registerCall(rpc_srv_t * __restrict srv, unsigned short tag, 
		void *funcaddr, unsigned short args);
/*
 * rpc_srv_unregisterCall() - Unregister call from RPC server
 *
 * @srv = RPC Server instance
 * @tag = Function tag
 * return: -1 error, 0 not found call, 1 unregister ok
 */
int rpc_srv_unregisterCall(rpc_srv_t * __restrict srv, unsigned short tag);
/*
 * rpc_srv_getCall()  - Get registered call from RPC server
 *
 * @srv = RPC Server instance
 * @tag = tag for function
 * return: NULL not found call, !=NULL return call
 */
inline rpc_func_t *rpc_srv_getCall(rpc_srv_t * __restrict srv, uint16_t tag);
/*
 * rpc_srv_execCall() - Execute registered call from RPC server
 *
 * @call = Register RPC call
 * @rpc = IN RPC call structure
 * @args = IN RPC calling arguments from RPC client
 * return: -1 error, !=-1 ok
 */
int rpc_srv_execCall(rpc_func_t * __restrict call, struct tagRPCCall * __restrict rpc, 
		array_t * __restrict args);


/*
 * rpc_srv_blobCreate() - Create and map blob to memory region and return object
 *
 * @srv = RPC Server instance
 * @len = BLOB length object
 * return: NULL error or !=NULL allocated BLOB object
 */
inline rpc_blob_t *rpc_srv_blobCreate(rpc_srv_t * __restrict srv, int len);
/*
 * rpc_srv_blobMap() - Map blob to memory region 
 *
 * @srv = RPC Server instance
 * @blob = Map to this BLOB element
 * return: -1 error or 0 ok
 */
inline int rpc_srv_blobMap(rpc_srv_t * __restrict srv, rpc_blob_t * __restrict blob);
/*
 * rpc_srv_blobUnmap() - Unmap blob memory region 
 *
 * @blob = Mapped BLOB element
 * return: none
 */
inline void rpc_srv_blobUnmap(rpc_blob_t * __restrict blob);
/*
 * rpc_srv_blobFree() - Free blob from disk & memory
 *
 * @srv = RPC Server instance
 * @blob = Mapped BLOB element
 * return: -1 error or 0 ok
 */
inline int rpc_srv_blobFree(rpc_srv_t * __restrict srv, rpc_blob_t * __restrict blob);

/*
 * rpc_srv_registerBLOB() - Register new BLOB to server
 *
 * @srv = RPC Server instance
 * @len = BLOB length
 * return: NULL error or new registered BLOB
 */
rpc_blob_t *rpc_srv_registerBLOB(rpc_srv_t * __restrict srv, size_t len);
/*
 * rpc_srv_unregisterBLOB() - Unregister BLOB from server
 *
 * @srv = RPC Server instance
 * @var = BLOB Variable for unregister
 * return: -1 error, 0 not found call, 1 unregister ok
 */
int rpc_srv_unregisterBLOB(rpc_srv_t * __restrict srv, uint32_t var);
/*
 * rpc_srv_getBLOB() - Get registered BLOB 
 *
 * @srv = RPC Server instance
 * @var = hash for variable
 * return: NULL not found, !=NULL return blob var
 */
inline rpc_blob_t *rpc_srv_getBLOB(rpc_srv_t * __restrict srv, uint32_t var);

/*
 * rpc_srv_sendBLOB() - Send mapped BLOB to client
 *
 * @cli = Client instance
 * @blob = Mapped BLOB element
 * return: -1 error, 0 ok
 */
int rpc_srv_sendBLOB(rpc_cli_t * __restrict cli, rpc_blob_t * __restrict blob);
/*
 * rpc_srv_recvBLOB() - Receive BLOB from client
 *
 * @cli = Client instance
 * @blob = Mapped BLOB element
 * return: -1 error, 0 ok, >0 unreceived data from client, may be error?
 */
int rpc_srv_recvBLOB(rpc_cli_t * __restrict cli, rpc_blob_t * __restrict blob);

/* CLIENT part of functions */

/*
 * rpc_cli_sendBLOB() - Send BLOB to server
 *
 * @cli = Client instance
 * @var = BLOB variable
 * @data = BLOB data
 * return: -1 error, 0 ok, 1 remote error
 */
int rpc_cli_sendBLOB(rpc_cli_t * __restrict cli, ait_val_t * __restrict var, void * __restrict data);
/*
 * rpc_cli_recvBLOB() - Receive BLOB from server
 *
 * @cli = Client instance
 * @var = BLOB variable
 * @data = BLOB data, must be free after use!
 * return: -1 error, 0 ok, 1 remote error
 */
int rpc_cli_recvBLOB(rpc_cli_t * __restrict cli, ait_val_t * __restrict var, void ** __restrict data);
/*
 * rpc_cli_delBLOB() - Delete BLOB from server
 *
 * @cli = Client instance
 * @var = BLOB variable
 * return: -1 error, 0 ok, 1 remote error
 */
int rpc_cli_delBLOB(rpc_cli_t * __restrict cli, ait_val_t * __restrict var);
/*
 * rpc_cli_getBLOB() - Receive BLOB from server and Delete after that.
 *
 * @cli = Client instance
 * @var = BLOB variable
 * @data = BLOB data, must be free after use!
 * return: -1 error, 0 ok, >0 remote error
 */
inline int rpc_cli_getBLOB(rpc_cli_t * __restrict cli, ait_val_t * __restrict var, 
		void ** __restrict data);


/* RPC Client side functions */

/*
 * rpc_cli_openClient() - Connect to RPC Server
 *
 * @ProgID = ProgramID for RPC session request
 * @ProcID = ProcessID for RPC session request
 * @netBuf = Network buffer length, if =0 == BUFSIZ (also meaning max RPC packet)
 * @family = Family socket type, AF_INET or AF_INET6
 * @csHost = Host name or IP address for bind server
 * @Port = Port for bind server, if Port == 0 default port is selected
 * return: NULL == error or !=NULL connection to RPC server established
 */
rpc_cli_t *rpc_cli_openClient(unsigned int ProgID, unsigned char ProcID, int netBuf, 
		unsigned short family, const char *csHost, unsigned short Port);
/*
 * rpc_cli_closeClient() - Close connection to RPC server and free resources
 *
 * @cli = RPC Client session
 * return: none
 */
void rpc_cli_closeClient(rpc_cli_t ** __restrict cli);
/*
 * rpc_cli_execCall() - Execute RPC call
 *
 * @cli = RPC Client session
 * @noreply = We not want RPC reply
 * @tag = Function tag for execution
 * @in_vars = IN RPC call array of rpc values
 * @out_vars = OUT returned array of rpc values, must be free after use with rpc_cli_freeVals()
 * return: -1 error or != -1 ok result
 */
int rpc_cli_execCall(rpc_cli_t *cli, int noreply, u_short tag, 
		array_t * __restrict in_vars, array_t ** __restrict out_vars);


/*
 * rpc_cli_openBLOBClient() - Connect to BLOB Server
 *
 * @rpccli = RPC Client session
 * @Port = Port for bind server, if Port == 0 default port is selected
 * return: NULL == error or !=NULL connection to BLOB server established
 */
rpc_cli_t *rpc_cli_openBLOBClient(rpc_cli_t * __restrict rpccli, u_short Port);
/*
 * rpc_cli_closeBLOBClient() - Close connection to BLOB server and free resources
 *
 * @cli = BLOB Client session
 * return: none
 */
void rpc_cli_closeBLOBClient(rpc_cli_t ** __restrict cli);


#endif

FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>