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

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.18.2.2! misho       6: * $Id: aitrpc.c,v 1.18.2.1 2015/06/28 21:40:45 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.17      misho      15: Copyright 2004 - 2015
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
1.18      misho     150:  * @pkt = RPC packet
1.14      misho     151:  * return: -1 error, 0 EOF or or >0 readed bytes into buffer
1.11      misho     152:  */
                    153: ssize_t
1.18      misho     154: rpc_Read(int sock, int type, int flags, sockaddr_t * __restrict sa, ait_val_t * __restrict pkt)
1.11      misho     155: {
                    156:        struct pollfd pfd;
                    157:        sockaddr_t sa2;
                    158:        socklen_t salen;
1.18.2.1  misho     159:        int ret = 0, hlen, cx = 0;
1.16      misho     160:        struct ether_header *eh;
                    161:        struct bpf_hdr *h;
                    162:        ether_addr_t bcst = {{ [0 ... sizeof bcst - 1] = 0xff }};
1.18      misho     163:        u_char *buf = AIT_GET_BUF(pkt);
                    164:        size_t blen = AIT_LEN(pkt);
                    165:        struct tagRPCCall *rpc = (struct tagRPCCall *) buf;
1.18.2.1  misho     166:        ait_val_t v = AIT_VAL_INIT;
1.16      misho     167: 
1.18.2.1  misho     168: try2read:
1.11      misho     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);
1.18.2.1  misho     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);
1.11      misho     198: 
1.16      misho     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)) {
1.18.2.1  misho     204:                                        AIT_FREE_VAL(&v);
1.16      misho     205:                                        if (cx < 3) {
                    206:                                                cx++;
                    207:                                                ret ^= ret;
1.18.2.1  misho     208:                                                goto try2read;  /* wait for known address */
                    209:                                        } else
1.16      misho     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)) {
1.18.2.1  misho     216:                                        AIT_FREE_VAL(&v);
1.16      misho     217:                                        if (cx < 3) {
                    218:                                                cx++;
                    219:                                                ret ^= ret;
1.18.2.1  misho     220:                                                goto try2read;  /* wait for known address */
                    221:                                        } else
1.16      misho     222:                                                return -1;
                    223:                                }
1.18      misho     224: 
1.16      misho     225:                                if (!memcmp(bcst.octet, eh->ether_dhost, sizeof bcst) || 
                    226:                                                !memcmp(bcst.octet, eh->ether_shost, sizeof bcst)) {
1.18.2.1  misho     227:                                        AIT_FREE_VAL(&v);
1.16      misho     228:                                        if (cx < 3) {
                    229:                                                cx++;
                    230:                                                ret ^= ret;
1.18.2.1  misho     231:                                                goto try2read;  /* wait for known address */
                    232:                                        } else
1.16      misho     233:                                                return -1;
                    234:                                }
1.18.2.1  misho     235:                                memcpy(buf, (u_char*) (eh + 1), MIN(ret, blen));
1.16      misho     236:                                AIT_FREE_VAL(&v);
                    237:                        }
1.18.2.1  misho     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)) {
1.18      misho     248:                                if (cx < 3) {
                    249:                                        cx++;
                    250:                                        ret ^= ret;
1.18.2.1  misho     251:                                        goto try2read;  /* wait for known address */
1.18      misho     252:                                } else
                    253:                                        return -1;
1.11      misho     254:                        }
1.18.2.1  misho     255:                        break;
                    256:                default:
                    257:                        rpc_SetErr(EINVAL, "Invalid argument(s)!");
1.13      misho     258:                        return -1;
1.18.2.1  misho     259:        }
                    260:        if (ret < 0) {
                    261:                LOGERR;
                    262:                return -1;
                    263:        }
                    264:        if (!ret)       /* EOF */
                    265:                return 0;
1.18      misho     266: 
1.18.2.1  misho     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:        }
1.18      misho     275: 
1.18.2.1  misho     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;
1.18      misho     290:                }
1.11      misho     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
1.18      misho     303:  * @pkt = RPC packet
1.11      misho     304:  * @blen = buffer length
1.14      misho     305:  * return: -1 error, 0 EOF or >0 written bytes into buffer
1.11      misho     306:  */
                    307: ssize_t
                    308: rpc_Write(int sock, int type, int flags, sockaddr_t * __restrict sa, 
1.18      misho     309:                ait_val_t * __restrict pkt, size_t blen)
1.11      misho     310: {
                    311:        struct pollfd pfd;
1.18.2.1  misho     312:        int ret = 0;
                    313:        u_char *buf = AIT_GET_BUF(pkt);
1.16      misho     314:        ait_val_t v = AIT_VAL_INIT;
                    315:        struct ether_header *eh;
                    316: 
1.11      misho     317: 
                    318:        pfd.fd = sock;
                    319:        pfd.events = POLLOUT;
1.18.2.1  misho     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: 
1.16      misho     349:                        ret = write(sock, AIT_GET_BUF(&v), AIT_LEN(&v));
1.18.2.1  misho     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:
1.13      misho     363:                        rpc_SetErr(EINVAL, "Invalid argument(s)!");
                    364:                        return -1;
1.11      misho     365:        }
1.18.2.1  misho     366:        if (ret < 0) {
                    367:                LOGERR;
                    368:                return -1;
                    369:        }
                    370:        if (!ret)       /* EOF */
                    371:                return 0;
                    372: 
1.13      misho     373:        if (ret != blen) {
1.11      misho     374:                rpc_SetErr(EPROCUNAVAIL, "RPC request, should be send %d bytes, "
1.16      misho     375:                                "really sended %d bytes", blen, ret);
1.11      misho     376:                return -1;
                    377:        }
                    378: 
                    379:        return ret;
                    380: }
1.18.2.2! misho     381: 
        !           382: /*
        !           383:  * rpc_pktFreeSpace() - Get free space for payload into RPC packet
        !           384:  *
        !           385:  * @c = RPC client 
        !           386:  * return: remains free bytes from packet
        !           387:  */
        !           388: size_t
        !           389: rpc_pktFreeSpace(rpc_cli_t * __restrict c)
        !           390: {
        !           391:        return (sizeof(struct tagRPCCall) + ait_resideVars(RPC_RETVARS(c)));
        !           392: }

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