File:  [ELWIX - Embedded LightWeight unIX -] / libaitrpc / src / aitrpc.c
Revision 1.18.2.1: download - view: text, annotated - select for diffs - revision graph
Sun Jun 28 21:40:45 2015 UTC (9 years, 1 month ago) by misho
Branches: rpc9_1
Diff to: branchpoint 1.18: preferred, unified
rework srv/cli sides

    1: /*************************************************************************
    2: * (C) 2010 AITNET ltd - Sofia/Bulgaria - <misho@aitbg.com>
    3: *  by Michael Pounov <misho@openbsd-bg.org>
    4: *
    5: * $Author: misho $
    6: * $Id: aitrpc.c,v 1.18.2.1 2015/06/28 21:40:45 misho Exp $
    7: *
    8: **************************************************************************
    9: The ELWIX and AITNET software is distributed under the following
   10: terms:
   11: 
   12: All of the documentation and software included in the ELWIX and AITNET
   13: Releases is copyrighted by ELWIX - Sofia/Bulgaria <info@elwix.org>
   14: 
   15: Copyright 2004 - 2015
   16: 	by Michael Pounov <misho@elwix.org>.  All rights reserved.
   17: 
   18: Redistribution and use in source and binary forms, with or without
   19: modification, are permitted provided that the following conditions
   20: are met:
   21: 1. Redistributions of source code must retain the above copyright
   22:    notice, this list of conditions and the following disclaimer.
   23: 2. Redistributions in binary form must reproduce the above copyright
   24:    notice, this list of conditions and the following disclaimer in the
   25:    documentation and/or other materials provided with the distribution.
   26: 3. All advertising materials mentioning features or use of this software
   27:    must display the following acknowledgement:
   28: This product includes software developed by Michael Pounov <misho@elwix.org>
   29: ELWIX - Embedded LightWeight unIX and its contributors.
   30: 4. Neither the name of AITNET nor the names of its contributors
   31:    may be used to endorse or promote products derived from this software
   32:    without specific prior written permission.
   33: 
   34: THIS SOFTWARE IS PROVIDED BY AITNET AND CONTRIBUTORS ``AS IS'' AND
   35: ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
   36: IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
   37: ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
   38: FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
   39: DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
   40: OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
   41: HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
   42: LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
   43: OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
   44: SUCH DAMAGE.
   45: */
   46: #include "global.h"
   47: 
   48: 
   49: #pragma GCC visibility push(hidden)
   50: 
   51: int rpc_Errno;
   52: char rpc_Error[STRSIZ];
   53: 
   54: inline int
   55: rpc_funcs_cmp(struct tagRPCFunc *a, struct tagRPCFunc *b)
   56: {
   57: 	int ret;
   58: 
   59: 	assert(a && b);
   60: 
   61: 	ret = AIT_KEY(&a->func_name) - AIT_KEY(&b->func_name);
   62: 
   63: 	if (ret < 0)
   64: 		return -1;
   65: 	else if (ret > 0)
   66: 		return 1;
   67: 
   68: 	return ret;
   69: }
   70: 
   71: AVL_GENERATE(tagRPCFuncs, tagRPCFunc, func_node, rpc_funcs_cmp);
   72: 
   73: #pragma GCC visibility pop
   74: 
   75: // rpc_GetErrno() Get error code of last operation
   76: int
   77: rpc_GetErrno()
   78: {
   79: 	return rpc_Errno;
   80: }
   81: 
   82: // rpc_GetError() Get error text of last operation
   83: const char *
   84: rpc_GetError()
   85: {
   86: 	return rpc_Error;
   87: }
   88: 
   89: // rpc_SetErr() Set error to variables for internal use!!!
   90: void
   91: rpc_SetErr(int eno, char *estr, ...)
   92: {
   93: 	va_list lst;
   94: 
   95: 	rpc_Errno = eno;
   96: 	memset(rpc_Error, 0, sizeof rpc_Error);
   97: 	va_start(lst, estr);
   98: 	vsnprintf(rpc_Error, sizeof rpc_Error, estr, lst);
   99: 	va_end(lst);
  100: }
  101: 
  102: 
  103: /*
  104:  * rpc_chkPktSession() - Check RPC session
  105:  *
  106:  * @p = packet session
  107:  * @s = active session
  108:  * return: -1, 1, 2, 3 are errors or 0 ok
  109:  */
  110: int
  111: rpc_chkPktSession(rpc_sess_t *p, rpc_sess_t *s)
  112: {
  113: 	if (!p || !s)
  114: 		return -1;
  115: 
  116: 	if (p->sess_version != s->sess_version)
  117: 		return 1;
  118: 	if (p->sess_instance != s->sess_instance)
  119: 		return 2;
  120: 
  121: 	return 0;
  122: }
  123: 
  124: /*
  125:  * rpc_addPktSession() - Prepare session into network format
  126:  *
  127:  * @p = packet session
  128:  * @s = host session
  129:  * return: -1 error or 0 ok
  130:  */
  131: int
  132: rpc_addPktSession(rpc_sess_t *p, rpc_sess_t *s)
  133: {
  134: 	if (!p || !s)
  135: 		return -1;
  136: 
  137: 	p->sess_version = s->sess_version;
  138: 	p->sess_instance = s->sess_instance;
  139: 
  140: 	return 0;
  141: }
  142: 
  143: /*
  144:  * rpc_Read() - RPC read operation
  145:  *
  146:  * @sock = socket
  147:  * @type = type of socket
  148:  * @flags = receive flags
  149:  * @sa = check client address, if you use udp protocol
  150:  * @pkt = RPC packet
  151:  * return: -1 error, 0 EOF or or >0 readed bytes into buffer
  152:  */
  153: ssize_t
  154: rpc_Read(int sock, int type, int flags, sockaddr_t * __restrict sa, ait_val_t * __restrict pkt)
  155: {
  156: 	struct pollfd pfd;
  157: 	sockaddr_t sa2;
  158: 	socklen_t salen;
  159: 	int ret = 0, hlen, cx = 0;
  160: 	struct ether_header *eh;
  161: 	struct bpf_hdr *h;
  162: 	ether_addr_t bcst = {{ [0 ... sizeof bcst - 1] = 0xff }};
  163: 	u_char *buf = AIT_GET_BUF(pkt);
  164: 	size_t blen = AIT_LEN(pkt);
  165: 	struct tagRPCCall *rpc = (struct tagRPCCall *) buf;
  166: 	ait_val_t v = AIT_VAL_INIT;
  167: 
  168: try2read:
  169: 	pfd.fd = sock;
  170: 	pfd.events = POLLIN | POLLPRI;
  171: 	memset(buf, 0, blen);
  172: 	memset(&sa2, 0, sizeof sa2);
  173: 	salen = sa2.ss.ss_len = sizeof(sockaddr_t);
  174: 	if ((ret = poll(&pfd, 1, DEF_RPC_TIMEOUT * 1000)) < 1 || 
  175: 			pfd.revents & (POLLERR | POLLHUP | POLLNVAL)) {
  176: 		if (ret)
  177: 			LOGERR;
  178: 		else
  179: 			rpc_SetErr(ETIMEDOUT, "Timeout reached! Server not respond");
  180: 		return -1;
  181: 	}
  182: 
  183: 	switch (type) {
  184: 		case SOCK_STREAM:
  185: 			ret = recv(sock, buf, blen, flags);
  186: 			break;
  187: 		case SOCK_EXT:
  188: 			ret = read(sock, buf, blen);
  189: 			break;
  190: 		case SOCK_BPF:
  191: 			if (!sa) {
  192: 				rpc_SetErr(EINVAL, "Invalid argument(s)!");
  193: 				return -1;
  194: 			}
  195: 
  196: 			AIT_SET_BUF(&v, NULL, blen);
  197: 			h = (struct bpf_hdr*) AIT_GET_BUF(&v);
  198: 
  199: 			ret = read(sock, AIT_GET_BUF(&v), AIT_LEN(&v));
  200: 			if (ret > 0) {
  201: 				ret -= h->bh_hdrlen;
  202: 				if (ret < h->bh_caplen || h->bh_caplen != h->bh_datalen || 
  203: 						ret < ETHER_HDR_LEN + sizeof(struct tagRPCCall)) {
  204: 					AIT_FREE_VAL(&v);
  205: 					if (cx < 3) {
  206: 						cx++;
  207: 						ret ^= ret;
  208: 						goto try2read;	/* wait for known address */
  209: 					} else
  210: 						return -1;
  211: 				}
  212: 				ret = h->bh_caplen;
  213: 				eh = (struct ether_header*) (AIT_GET_BUF(&v) + h->bh_hdrlen);
  214: 				ret -= ETHER_HDR_LEN;
  215: 				if (eh->ether_type != ntohs(RPC_DEFPORT)) {
  216: 					AIT_FREE_VAL(&v);
  217: 					if (cx < 3) {
  218: 						cx++;
  219: 						ret ^= ret;
  220: 						goto try2read;	/* wait for known address */
  221: 					} else
  222: 						return -1;
  223: 				}
  224: 
  225: 				if (!memcmp(bcst.octet, eh->ether_dhost, sizeof bcst) || 
  226: 						!memcmp(bcst.octet, eh->ether_shost, sizeof bcst)) {
  227: 					AIT_FREE_VAL(&v);
  228: 					if (cx < 3) {
  229: 						cx++;
  230: 						ret ^= ret;
  231: 						goto try2read;	/* wait for known address */
  232: 					} else
  233: 						return -1;
  234: 				}
  235: 				memcpy(buf, (u_char*) (eh + 1), MIN(ret, blen));
  236: 				AIT_FREE_VAL(&v);
  237: 			}
  238: 			break;
  239: 		case SOCK_RAW:
  240: 		case SOCK_DGRAM:
  241: 			if (!sa) {
  242: 				rpc_SetErr(EINVAL, "Invalid argument(s)!");
  243: 				return -1;
  244: 			}
  245: 
  246: 			ret = recvfrom(sock, buf, blen, flags, &sa2.sa, &salen);
  247: 			if (ret > -1 && e_addrcmp(sa, &sa2, 42)) {
  248: 				if (cx < 3) {
  249: 					cx++;
  250: 					ret ^= ret;
  251: 					goto try2read;	/* wait for known address */
  252: 				} else
  253: 					return -1;
  254: 			}
  255: 			break;
  256: 		default:
  257: 			rpc_SetErr(EINVAL, "Invalid argument(s)!");
  258: 			return -1;
  259: 	}
  260: 	if (ret < 0) {
  261: 		LOGERR;
  262: 		return -1;
  263: 	}
  264: 	if (!ret)	/* EOF */
  265: 		return 0;
  266: 
  267: 	/* check RPC packet header */
  268: 	if (type == SOCK_RAW) {
  269: 		hlen = sa->sa.sa_family == AF_INET ? 
  270: 			sizeof(struct ip) : sizeof(struct ip6_hdr);
  271: 		ret -= hlen;
  272: 		if (ret > 0)
  273: 			memmove(buf, buf + hlen, blen - hlen);
  274: 	}
  275: 
  276: 	/* 1st read for RPC header */
  277: 	if (ret < sizeof(struct tagRPCCall) || ret < ntohl(rpc->call_len)) {
  278: 		rpc_SetErr(ERPCMISMATCH, "Short RPC packet %d bytes", ret);
  279: 		return -1;
  280: 	}
  281: 	/* check for loop request */
  282: 	if (!(rpc->call_io & RPC_ACK)) {
  283: 		if (cx < 3) {
  284: 			cx++;
  285: 			ret ^= ret;
  286: 			goto try2read;	/* wait for known address */
  287: 		} else {
  288: 			rpc_SetErr(ERPCMISMATCH, "Loop in RPC communication");
  289: 			return -1;
  290: 		}
  291: 	}
  292: 
  293: 	return ret;
  294: }
  295: 
  296: /*
  297:  * rpc_Write() - RPC write operation
  298:  *
  299:  * @sock = socket
  300:  * @type = type of socket
  301:  * @flags = send flags
  302:  * @sa = send to client address, if you use udp protocol
  303:  * @pkt = RPC packet
  304:  * @blen = buffer length
  305:  * return: -1 error, 0 EOF or >0 written bytes into buffer
  306:  */
  307: ssize_t
  308: rpc_Write(int sock, int type, int flags, sockaddr_t * __restrict sa, 
  309: 		ait_val_t * __restrict pkt, size_t blen)
  310: {
  311: 	struct pollfd pfd;
  312: 	int ret = 0;
  313: 	u_char *buf = AIT_GET_BUF(pkt);
  314: 	ait_val_t v = AIT_VAL_INIT;
  315: 	struct ether_header *eh;
  316: 
  317: 
  318: 	pfd.fd = sock;
  319: 	pfd.events = POLLOUT;
  320: 	if ((ret = poll(&pfd, 1, DEF_RPC_TIMEOUT * 1000)) < 1 || 
  321: 			pfd.revents & (POLLERR | POLLHUP | POLLNVAL)) {
  322: 		if (ret > 0)
  323: 			rpc_SetErr(EPIPE, "Disconnected RPC session\n");
  324: 		else
  325: 			LOGERR;
  326: 		return -1;
  327: 	}
  328: 
  329: 	switch (type) {
  330: 		case SOCK_STREAM:
  331: 			ret = send(sock, buf, blen, flags);
  332: 			break;
  333: 		case SOCK_EXT:
  334: 			ret = write(sock, buf, blen);
  335: 			break;
  336: 		case SOCK_BPF:
  337: 			if (!sa) {
  338: 				rpc_SetErr(EINVAL, "Invalid argument(s)!");
  339: 				return -1;
  340: 			}
  341: 
  342: 			AIT_SET_BUF(&v, NULL, blen + sizeof(struct ether_header));
  343: 			eh = (struct ether_header*) AIT_GET_BUF(&v);
  344: 			memcpy(eh->ether_dhost, LLADDR(&sa->sdl), ETHER_ADDR_LEN);
  345: 			eh->ether_type = htons(RPC_DEFPORT);
  346: 			memcpy(eh + 1, buf, blen);
  347: 			blen += sizeof(struct ether_header);
  348: 
  349: 			ret = write(sock, AIT_GET_BUF(&v), AIT_LEN(&v));
  350: 
  351: 			AIT_FREE_VAL(&v);
  352: 			break;
  353: 		case SOCK_RAW:
  354: 		case SOCK_DGRAM:
  355: 			if (!sa) {
  356: 				rpc_SetErr(EINVAL, "Invalid argument(s)!");
  357: 				return -1;
  358: 			}
  359: 
  360: 			ret = sendto(sock, buf, blen, flags, &sa->sa, sa->sa.sa_len);
  361: 			break;
  362: 		default:
  363: 			rpc_SetErr(EINVAL, "Invalid argument(s)!");
  364: 			return -1;
  365: 	}
  366: 	if (ret < 0) {
  367: 		LOGERR;
  368: 		return -1;
  369: 	}
  370: 	if (!ret)	/* EOF */
  371: 		return 0;
  372: 
  373: 	if (ret != blen) {
  374: 		rpc_SetErr(EPROCUNAVAIL, "RPC request, should be send %d bytes, "
  375: 				"really sended %d bytes", blen, ret);
  376: 		return -1;
  377: 	}
  378: 
  379: 	return ret;
  380: }

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