Annotation of libaitrpc/src/aitrpc.c, revision 1.15

1.1       misho       1: /*************************************************************************
                      2: * (C) 2010 AITNET ltd - Sofia/Bulgaria - <misho@aitbg.com>
                      3: *  by Michael Pounov <misho@openbsd-bg.org>
                      4: *
                      5: * $Author: misho $
1.15    ! misho       6: * $Id: aitrpc.c,v 1.14.2.1 2014/01/28 14:04:57 misho Exp $
1.1       misho       7: *
1.2       misho       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: 
1.15    ! misho      15: Copyright 2004 - 2014
1.2       misho      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: */
1.1       misho      46: #include "global.h"
                     47: 
                     48: 
                     49: #pragma GCC visibility push(hidden)
                     50: 
                     51: int rpc_Errno;
                     52: char rpc_Error[STRSIZ];
                     53: 
1.7       misho      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: 
1.1       misho      73: #pragma GCC visibility pop
                     74: 
                     75: // rpc_GetErrno() Get error code of last operation
1.10      misho      76: int
1.1       misho      77: rpc_GetErrno()
                     78: {
                     79:        return rpc_Errno;
                     80: }
                     81: 
                     82: // rpc_GetError() Get error text of last operation
1.10      misho      83: const char *
1.1       misho      84: rpc_GetError()
                     85: {
                     86:        return rpc_Error;
                     87: }
                     88: 
                     89: // rpc_SetErr() Set error to variables for internal use!!!
1.10      misho      90: void
1.1       misho      91: rpc_SetErr(int eno, char *estr, ...)
                     92: {
                     93:        va_list lst;
                     94: 
                     95:        rpc_Errno = eno;
1.8       misho      96:        memset(rpc_Error, 0, sizeof rpc_Error);
1.1       misho      97:        va_start(lst, estr);
1.8       misho      98:        vsnprintf(rpc_Error, sizeof rpc_Error, estr, lst);
1.1       misho      99:        va_end(lst);
                    100: }
                    101: 
                    102: 
1.4       misho     103: /*
1.6       misho     104:  * rpc_chkPktSession() - Check RPC session
1.5       misho     105:  *
1.4       misho     106:  * @p = packet session
                    107:  * @s = active session
1.6       misho     108:  * return: -1, 1, 2, 3 are errors or 0 ok
1.4       misho     109:  */
1.10      misho     110: int
1.4       misho     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;
1.9       misho     118:        if (p->sess_instance != s->sess_instance)
1.4       misho     119:                return 2;
                    120: 
                    121:        return 0;
                    122: }
                    123: 
                    124: /*
1.6       misho     125:  * rpc_addPktSession() - Prepare session into network format
1.5       misho     126:  *
1.4       misho     127:  * @p = packet session
1.6       misho     128:  * @s = host session
1.4       misho     129:  * return: -1 error or 0 ok
                    130:  */
1.10      misho     131: int
1.4       misho     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;
1.9       misho     138:        p->sess_instance = s->sess_instance;
1.4       misho     139: 
                    140:        return 0;
                    141: }
1.11      misho     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:  * @buf = buffer
                    151:  * @blen = buffer length
1.14      misho     152:  * return: -1 error, 0 EOF or or >0 readed bytes into buffer
1.11      misho     153:  */
                    154: ssize_t
                    155: rpc_Read(int sock, int type, int flags, sockaddr_t * __restrict sa, 
                    156:                u_char * __restrict buf, size_t blen)
                    157: {
                    158:        struct pollfd pfd;
                    159:        sockaddr_t sa2;
                    160:        socklen_t salen;
                    161:        int ret = 0, len;
                    162: 
                    163:        pfd.fd = sock;
                    164:        pfd.events = POLLIN | POLLPRI;
                    165:        memset(buf, 0, blen);
                    166:        memset(&sa2, 0, sizeof sa2);
                    167:        salen = sa2.ss.ss_len = sizeof(sockaddr_t);
                    168: 
1.13      misho     169:        for (len = blen; len > 0; len -= ret, buf += ret) {
                    170:                if ((ret = poll(&pfd, 1, DEF_RPC_TIMEOUT * 1000)) < 1 || 
                    171:                                pfd.revents & (POLLERR | POLLHUP | POLLNVAL)) {
                    172:                        if (ret)
1.11      misho     173:                                LOGERR;
1.13      misho     174:                        else
                    175:                                rpc_SetErr(ETIMEDOUT, "Timeout reached! Server not respond");
                    176:                        return -1;
                    177:                }
                    178:                if (type == SOCK_STREAM)
                    179:                        ret = recv(sock, buf, len, flags);
                    180:                else {
                    181:                        ret = recvfrom(sock, buf, len, flags, &sa2.sa, &salen);
                    182:                        if (ret > -1 && sa && e_addrcmp(sa, &sa2, 42)) {
                    183:                                rpc_SetErr(ERPCMISMATCH, 
                    184:                                                "Received RPC response from unknown address");
1.11      misho     185:                                return -1;
                    186:                        }
                    187:                }
1.13      misho     188:                if (ret == -1) {
                    189:                        LOGERR;
                    190:                        return -1;
                    191:                }
1.14      misho     192:                if (!ret)       /* EOF */
                    193:                        return 0;
1.11      misho     194:        }
1.13      misho     195:        ret = blen - len;
1.11      misho     196: 
                    197:        return ret;
                    198: }
                    199: 
                    200: /*
                    201:  * rpc_Write() - RPC write operation
                    202:  *
                    203:  * @sock = socket
                    204:  * @type = type of socket
                    205:  * @flags = send flags
                    206:  * @sa = send to client address, if you use udp protocol
                    207:  * @buf = buffer
                    208:  * @blen = buffer length
1.14      misho     209:  * return: -1 error, 0 EOF or >0 written bytes into buffer
1.11      misho     210:  */
                    211: ssize_t
                    212: rpc_Write(int sock, int type, int flags, sockaddr_t * __restrict sa, 
                    213:                u_char * __restrict buf, size_t blen)
                    214: {
                    215:        struct pollfd pfd;
                    216:        int ret = 0, len = 0;
                    217: 
                    218:        pfd.fd = sock;
                    219:        pfd.events = POLLOUT;
1.13      misho     220:        for (len = blen; len > 0; len -= ret, buf += ret) {
                    221:                if ((ret = poll(&pfd, 1, DEF_RPC_TIMEOUT * 1000)) < 1 || 
                    222:                                pfd.revents & (POLLERR | POLLHUP | POLLNVAL)) {
1.14      misho     223: #if 0
1.13      misho     224:                        if (!ret)
                    225:                                continue;
1.14      misho     226: #endif
1.13      misho     227:                        if (ret > 0)
                    228:                                rpc_SetErr(EPIPE, "Disconnected RPC session\n");
                    229:                        else
1.11      misho     230:                                LOGERR;
1.13      misho     231:                        return -1;
                    232:                }
                    233:                if (type == SOCK_STREAM)
                    234:                        ret = send(sock, buf, len, flags);
                    235:                else if (sa)
                    236:                        ret = sendto(sock, buf, len, flags, &sa->sa, sa->sa.sa_len);
                    237:                else {
                    238:                        rpc_SetErr(EINVAL, "Invalid argument(s)!");
                    239:                        return -1;
                    240:                }
                    241:                if (ret == -1) {
                    242:                        LOGERR;
                    243:                        return -1;
1.11      misho     244:                }
1.14      misho     245:                if (!ret)       /* EOF */
                    246:                        return 0;
1.11      misho     247:        }
1.13      misho     248:        ret = blen - len;
                    249:        if (ret != blen) {
1.11      misho     250:                rpc_SetErr(EPROCUNAVAIL, "RPC request, should be send %d bytes, "
                    251:                                "really sended %d bytes", len, ret);
                    252:                return -1;
                    253:        }
                    254: 
                    255:        return ret;
                    256: }

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