--- libaitrpc/inc/aitrpc.h 2010/06/23 07:47:25 1.1.1.1.2.3 +++ libaitrpc/inc/aitrpc.h 2010/06/28 15:18:43 1.1.1.1.2.12 @@ -3,7 +3,7 @@ * by Michael Pounov * * $Author: misho $ -* $Id: aitrpc.h,v 1.1.1.1.2.3 2010/06/23 07:47:25 misho Exp $ +* $Id: aitrpc.h,v 1.1.1.1.2.12 2010/06/28 15:18:43 misho Exp $ * *************************************************************************/ #ifndef __AITRPC_H @@ -19,14 +19,17 @@ #include -#define STRSIZ 256 - #define RPC_VERSION 1 #define RPC_DEFPORT 2611 /* RPC builtin registed calls */ +#define CALL_BLOBSHUTDOWN "rpcBLOBServerShutdown" +#define CALL_BLOBCLIENTS "rpcBLOBServerClients" +#define CALL_BLOBVARS "rpcBLOBServerVars" +#define CALL_BLOBSTATE "rpcBLOBServerState" + #define CALL_SRVSHUTDOWN "rpcServerShutdown" #define CALL_SRVCLIENTS "rpcServerClients" #define CALL_SRVCALLS "rpcServerCalls" @@ -44,6 +47,12 @@ typedef enum { i8, i16, i32, i64 // integers ... } rpc_type_t; +typedef enum { + disable, enable, // for blob.state + ok, error, // for blob reply + get, set, unset // for blob request +} cmd_type_t; + /* RPC value */ typedef struct { @@ -100,11 +109,8 @@ typedef struct { if (val->val.string) { \ val->val_type = string; val->val_len = strlen(v) + 1; \ } } while (0) -#define RPC_SET_BLOB(vl, v, l) do { rpc_val_t *val = (vl); assert(val); val->val.blob = realloc(val->val.blob, l); \ - if (val->val.blob) { \ - val->val_type = blob; val->val_len = l; \ - memcpy(val->val.blob, v, l); \ - } } while (0) +#define RPC_SET_BLOB(vl, v, l) do { rpc_val_t *val = (vl); assert(val); val->val_type = blob; \ + val->val.blob = (void*) v; val->val_len = l; } while (0) #define RPC_SET_SIZE(vl, v) do { rpc_val_t *val = (vl); assert(val); val->val_type = size; val->val.size = v; \ val->val_len = sizeof(size_t); } while (0) #define RPC_SET_OFF(vl, v) do { rpc_val_t *val = (vl); assert(val); val->val_type = offset; val->val.offset = v; \ @@ -141,10 +147,6 @@ typedef struct { free(val->val.string); \ val->val.string = NULL; \ } \ - if (val->val_type == array && val->val.array) { \ - free(val->val.array); \ - val->val.array = NULL; \ - } \ val->val_type = val->val_len = 0; \ } while (0) @@ -155,8 +157,8 @@ typedef struct { return -1; \ } \ } while (0) -#define RPC_CALLBACK_CHECK_INPUT(s, f) do { \ - if (!s || !f) { \ +#define RPC_CALLBACK_CHECK_INPUT(f) do { \ + if (!f) { \ rpc_SetErr(22, "Error:: invalid callback parameters ...\n"); \ return -1; \ } \ @@ -184,6 +186,7 @@ typedef struct tagRPCFunc { int8_t func_args; rpc_val_t *func_vals; + void *func_parent; struct tagRPCFunc *func_next; } rpc_func_t; @@ -208,6 +211,16 @@ struct tagRPCRet { uint8_t ret_argc; } __packed; +/* Network BLOB packet - Header */ + +struct tagBLOBHdr { + rpc_sess_t hdr_session; + uint8_t hdr_cmd; + uint32_t hdr_var; + uint32_t hdr_seq; + uint32_t hdr_len; +} __packed; + /* Network RPC client & server elements */ typedef struct { @@ -218,19 +231,50 @@ typedef struct { void *cli_parent; // pointer to parent rpc_srv_t for server or to rpc_sess_t for client } rpc_cli_t; + +// BLOB registration element! +typedef struct tagBLOB { + uint32_t blob_var; + + size_t blob_len; // size of allocated BLOB data + void *blob_data; // BLOB data + + struct tagBLOB *blob_next; +} rpc_blob_t; + typedef struct { + rpc_sess_t srv_session; // RPC session registration info int srv_numcli; // maximum concurent client connections - rpc_cli_t srv_server; // server socket - rpc_sess_t srv_session; // RPC session registration info + rpc_cli_t srv_server; // RPC server socket + rpc_cli_t *srv_clients; // connected rpc client sockets - rpc_cli_t *srv_clients; // connected client sockets - rpc_func_t *srv_funcs; // registered functions list + + pthread_mutex_t srv_mtx; + + struct { + int state; // BLOB server state: ==0 disable | !=0 enable + char dir[UCHAR_MAX + 1]; + + rpc_cli_t server; // BLOB server socket + rpc_cli_t *clients; // connected blob client sockets + + rpc_blob_t *blobs; // registered blob variables list + + pthread_mutex_t mtx; + } srv_blob; } rpc_srv_t; -typedef int (*rpc_callback_t)(void * const, rpc_func_t *, int, rpc_val_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, rpc_val_t *); // ----------------------------------------------------------------------- @@ -271,6 +315,34 @@ void rpc_srv_endServer(rpc_srv_t * __restrict srv); int rpc_srv_execServer(rpc_srv_t * __restrict srv); /* + * rpc_srv_initBLOBServer() Init & create BLOB Server + * @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_execBLOBServer() 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_execBLOBServer(rpc_srv_t * __restrict srv); + +/* + * 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_registerCall() Register call to RPC server * @srv = RPC Server instance * @csModule = Module name, if NULL self binary @@ -306,20 +378,29 @@ rpc_func_t *rpc_srv_getFunc(rpc_srv_t * __restrict srv inline rpc_func_t *rpc_srv_getCall(rpc_srv_t * __restrict srv, uint16_t tag, uint32_t hash); /* * rpc_srv_execCall() Execute registered call from RPC server - * @data = RPC const data * @call = Register RPC call * @rpc = IN RPC call structure * @args = IN RPC call array of rpc values * return: -1 error, !=-1 ok */ -int rpc_srv_execCall(void * const data, rpc_func_t * __restrict call, - struct tagRPCCall * __restrict rpc, rpc_val_t * __restrict args); +int rpc_srv_execCall(rpc_func_t * __restrict call, struct tagRPCCall * __restrict rpc, + rpc_val_t * __restrict args); /* - * rpc_srv_declValsCall() Declare return variables for RPC call + * rpc_srv_retValsCall() Declare return variables for RPC call and zeroed values + (for safe handling return values, use this!) * @call = RPC function call * @return_vals = Number of return variables + * return: NULL error, !=NULL array with return values for RPC call with return_vals items + */ +inline rpc_val_t *rpc_srv_retValsCall(rpc_func_t * __restrict call, int return_vals); +/* + * rpc_srv_declValsCall() Declare return variables for RPC call, + if already allocated memory for RPC call return values + function reallocate used space with return_vals count elements + * @call = RPC function call + * @return_vals = Number of return variables * return: -1 error, !=-1 ok */ inline int rpc_srv_declValsCall(rpc_func_t * __restrict call, int return_vals); @@ -337,11 +418,11 @@ inline void rpc_srv_freeValsCall(rpc_func_t * __restri */ inline int rpc_srv_copyValsCall(rpc_func_t * __restrict call, rpc_val_t ** __restrict newvals); /* - * rpc_srv_delValsCall() Clean values from return variables of RPC call + * rpc_srv_zeroValsCall() Clean values from return variables of RPC call * @call = RPC function call * return: -1 error, !=-1 Returned number of cleaned RPC variables */ -inline int rpc_srv_delValsCall(rpc_func_t * __restrict call); +inline int rpc_srv_zeroValsCall(rpc_func_t * __restrict call); /* * rpc_srv_getValsCall() Get return variables for RPC call * @call = RPC function call @@ -349,6 +430,65 @@ inline int rpc_srv_delValsCall(rpc_func_t * __restrict * return: -1 error, !=-1 Number of returned variables */ inline int rpc_srv_getValsCall(rpc_func_t * __restrict call, rpc_val_t ** __restrict vals); + + +/* + * rpc_srv_blobCreate() Create 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: -1 error or 0 register ok + */ +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_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); /* RPC Client side functions */