--- libaitrpc/src/srv.c	2012/11/13 09:22:10	1.12
+++ libaitrpc/src/srv.c	2012/11/16 08:33:06	1.12.2.1
@@ -3,7 +3,7 @@
 *  by Michael Pounov <misho@openbsd-bg.org>
 *
 * $Author: misho $
-* $Id: srv.c,v 1.12 2012/11/13 09:22:10 misho Exp $
+* $Id: srv.c,v 1.12.2.1 2012/11/16 08:33:06 misho Exp $
 *
 **************************************************************************
 The ELWIX and AITNET software is distributed under the following
@@ -46,6 +46,15 @@ SUCH DAMAGE.
 #include "global.h"
 
 
+static void *acceptClients(sched_task_t *);
+static void *closeClient(sched_task_t *);
+static sched_task_func_t cbProto[SOCK_RAW][2] = {
+	{ acceptClients, closeClient }, 
+	{ acceptClients, closeClient }, 
+	{ NULL, NULL }
+};
+
+
 static void *
 closeClient(sched_task_t *task)
 {
@@ -114,7 +123,8 @@ txPacket(sched_task_t *task)
 	ret = send(TASK_FD(task), buf, wlen, MSG_NOSIGNAL);
 	if (ret == -1 || ret != wlen) {
 		/* close connection */
-		schedEvent(TASK_ROOT(task), closeClient, c, 42, NULL, 0);
+		schedEvent(TASK_ROOT(task), cbProto[s->srv_proto][CB_CLOSECLIENT], 
+				c, 42, NULL, 0);
 	}
 
 	return NULL;
@@ -192,7 +202,8 @@ rxPacket(sched_task_t *task)
 	rlen = recv(TASK_FD(task), buf + off, AIT_LEN(&c->cli_buf) - off, 0);
 	if (rlen < 1) {
 		/* close connection */
-		schedEvent(TASK_ROOT(task), closeClient, c, 42, NULL, 0);
+		schedEvent(TASK_ROOT(task), cbProto[s->srv_proto][CB_CLOSECLIENT], 
+				c, 42, NULL, 0);
 		return NULL;
 	} else {
 		rlen += off;	/* add reminded bytes from previous rxPacket, if exists! */
@@ -691,20 +702,23 @@ rpc_srv_loopBLOBServer(rpc_srv_t * __restrict srv)
  * @netBuf = Network buffer length (min:512 bytes), 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
+ * @proto = Protocol, if == 0 choose SOCK_STREAM
  * return: NULL == error or !=NULL bind and created RPC server instance
  */
 rpc_srv_t *
 rpc_srv_initServer(u_int regProgID, u_char regProcID, int concurentClients, 
-		int netBuf, const char *csHost, u_short Port)
+		int netBuf, const char *csHost, u_short Port, int proto)
 {
 	int n = 1;
 	rpc_srv_t *srv = NULL;
 	io_sockaddr_t sa = IO_SOCKADDR_INIT;
 
-	if (!concurentClients || !regProgID) {
+	if (!concurentClients || !regProgID || (proto < 0 || proto > SOCK_DGRAM)) {
 		rpc_SetErr(EINVAL, "Invalid parameters can`t init RPC server");
 		return NULL;
 	}
+	if (!proto)
+		proto = SOCK_STREAM;
 	if (!io_gethostbyname(csHost, Port, &sa))
 		return NULL;
 	if (!Port)
@@ -729,6 +743,7 @@ rpc_srv_initServer(u_int regProgID, u_char regProcID, 
 	} else
 		memset(srv, 0, sizeof(rpc_srv_t));
 
+	srv->srv_proto = proto;
 	srv->srv_netbuf = netBuf;
 	srv->srv_session.sess_version = RPC_VERSION;
 	srv->srv_session.sess_program = regProgID;
@@ -762,7 +777,7 @@ rpc_srv_initServer(u_int regProgID, u_char regProcID, 
 	}
 
 	/* create server socket */
-	srv->srv_server.cli_sock = socket(srv->srv_server.cli_sa.sa.sa_family, SOCK_STREAM, 0);
+	srv->srv_server.cli_sock = socket(srv->srv_server.cli_sa.sa.sa_family, srv->srv_proto, 0);
 	if (srv->srv_server.cli_sock == -1) {
 		LOGERR;
 		io_arrayDestroy(&srv->srv_clients);
@@ -852,7 +867,8 @@ rpc_srv_loopServer(rpc_srv_t * __restrict srv)
 		fcntl(srv->srv_server.cli_sock, F_SETFL, 
 				fcntl(srv->srv_server.cli_sock, F_GETFL) | O_NONBLOCK);
 
-	if (!schedRead(srv->srv_root, acceptClients, srv, srv->srv_server.cli_sock, NULL, 0)) {
+	if (!schedRead(srv->srv_root, cbProto[srv->srv_proto][CB_ACCEPTCLIENT], srv, 
+				srv->srv_server.cli_sock, NULL, 0)) {
 		rpc_SetErr(sched_GetErrno(), "%s", sched_GetError());
 		return -1;
 	}