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

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.20.2.3! misho       6: * $Id: aitrpc.c,v 1.20.2.2 2016/08/02 10:26:58 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.20.2.1  misho      15: Copyright 2004 - 2016
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: 
                     54: #pragma GCC visibility pop
                     55: 
                     56: // rpc_GetErrno() Get error code of last operation
1.10      misho      57: int
1.1       misho      58: rpc_GetErrno()
                     59: {
                     60:        return rpc_Errno;
                     61: }
                     62: 
                     63: // rpc_GetError() Get error text of last operation
1.10      misho      64: const char *
1.1       misho      65: rpc_GetError()
                     66: {
                     67:        return rpc_Error;
                     68: }
                     69: 
                     70: // rpc_SetErr() Set error to variables for internal use!!!
1.10      misho      71: void
1.1       misho      72: rpc_SetErr(int eno, char *estr, ...)
                     73: {
                     74:        va_list lst;
                     75: 
                     76:        rpc_Errno = eno;
1.8       misho      77:        memset(rpc_Error, 0, sizeof rpc_Error);
1.1       misho      78:        va_start(lst, estr);
1.8       misho      79:        vsnprintf(rpc_Error, sizeof rpc_Error, estr, lst);
1.1       misho      80:        va_end(lst);
                     81: }
                     82: 
                     83: 
1.4       misho      84: /*
1.6       misho      85:  * rpc_chkPktSession() - Check RPC session
1.5       misho      86:  *
1.4       misho      87:  * @p = packet session
                     88:  * @s = active session
1.6       misho      89:  * return: -1, 1, 2, 3 are errors or 0 ok
1.4       misho      90:  */
1.10      misho      91: int
1.4       misho      92: rpc_chkPktSession(rpc_sess_t *p, rpc_sess_t *s)
                     93: {
                     94:        if (!p || !s)
                     95:                return -1;
                     96: 
                     97:        if (p->sess_version != s->sess_version)
                     98:                return 1;
1.9       misho      99:        if (p->sess_instance != s->sess_instance)
1.4       misho     100:                return 2;
                    101: 
                    102:        return 0;
                    103: }
                    104: 
                    105: /*
1.6       misho     106:  * rpc_addPktSession() - Prepare session into network format
1.5       misho     107:  *
1.4       misho     108:  * @p = packet session
1.6       misho     109:  * @s = host session
1.4       misho     110:  * return: -1 error or 0 ok
                    111:  */
1.10      misho     112: int
1.4       misho     113: rpc_addPktSession(rpc_sess_t *p, rpc_sess_t *s)
                    114: {
                    115:        if (!p || !s)
                    116:                return -1;
                    117: 
                    118:        p->sess_version = s->sess_version;
1.9       misho     119:        p->sess_instance = s->sess_instance;
1.4       misho     120: 
                    121:        return 0;
                    122: }
1.11      misho     123: 
                    124: /*
                    125:  * rpc_Read() - RPC read operation
                    126:  *
                    127:  * @sock = socket
                    128:  * @type = type of socket
                    129:  * @flags = receive flags
                    130:  * @sa = check client address, if you use udp protocol
1.18      misho     131:  * @pkt = RPC packet
1.14      misho     132:  * return: -1 error, 0 EOF or or >0 readed bytes into buffer
1.11      misho     133:  */
                    134: ssize_t
1.18      misho     135: rpc_Read(int sock, int type, int flags, sockaddr_t * __restrict sa, ait_val_t * __restrict pkt)
1.11      misho     136: {
                    137:        struct pollfd pfd;
                    138:        sockaddr_t sa2;
                    139:        socklen_t salen;
1.19      misho     140:        int ret = 0, hlen, cx = 0;
1.16      misho     141:        ether_addr_t bcst = {{ [0 ... sizeof bcst - 1] = 0xff }};
1.18      misho     142:        u_char *buf = AIT_GET_BUF(pkt);
                    143:        size_t blen = AIT_LEN(pkt);
                    144:        struct tagRPCCall *rpc = (struct tagRPCCall *) buf;
1.20.2.1  misho     145: #ifndef __linux__
                    146:        struct ether_header *eh;
                    147:        struct bpf_hdr *h;
1.19      misho     148:        ait_val_t v = AIT_VAL_INIT;
1.20.2.1  misho     149: #endif
1.16      misho     150: 
1.19      misho     151: try2read:
1.11      misho     152:        pfd.fd = sock;
                    153:        pfd.events = POLLIN | POLLPRI;
                    154:        memset(buf, 0, blen);
                    155:        memset(&sa2, 0, sizeof sa2);
1.20.2.3! misho     156:        salen = E_SOCKADDR_MAX;
1.20.2.1  misho     157: #ifndef __linux__
1.20.2.2  misho     158:        sa2.ss.ss_len = salen;
1.20.2.1  misho     159: #endif
1.19      misho     160:        if ((ret = poll(&pfd, 1, DEF_RPC_TIMEOUT * 1000)) < 1 || 
                    161:                        pfd.revents & (POLLERR | POLLHUP | POLLNVAL)) {
                    162:                if (ret)
                    163:                        LOGERR;
                    164:                else
                    165:                        rpc_SetErr(ETIMEDOUT, "Timeout reached! Server not respond");
                    166:                return -1;
                    167:        }
                    168: 
                    169:        switch (type) {
                    170:                case SOCK_STREAM:
                    171:                        ret = recv(sock, buf, blen, flags);
                    172:                        break;
                    173:                case SOCK_EXT:
                    174:                        ret = read(sock, buf, blen);
                    175:                        break;
1.20.2.1  misho     176: #ifndef __linux__
1.19      misho     177:                case SOCK_BPF:
                    178:                        if (!sa) {
                    179:                                rpc_SetErr(EINVAL, "Invalid argument(s)!");
                    180:                                return -1;
                    181:                        }
                    182: 
                    183:                        AIT_SET_BUF(&v, NULL, blen);
                    184:                        h = (struct bpf_hdr*) AIT_GET_BUF(&v);
1.11      misho     185: 
1.16      misho     186:                        ret = read(sock, AIT_GET_BUF(&v), AIT_LEN(&v));
                    187:                        if (ret > 0) {
                    188:                                ret -= h->bh_hdrlen;
                    189:                                if (ret < h->bh_caplen || h->bh_caplen != h->bh_datalen || 
                    190:                                                ret < ETHER_HDR_LEN + sizeof(struct tagRPCCall)) {
1.19      misho     191:                                        AIT_FREE_VAL(&v);
1.16      misho     192:                                        if (cx < 3) {
                    193:                                                cx++;
                    194:                                                ret ^= ret;
1.19      misho     195:                                                goto try2read;  /* wait for known address */
                    196:                                        } else
1.16      misho     197:                                                return -1;
                    198:                                }
                    199:                                ret = h->bh_caplen;
                    200:                                eh = (struct ether_header*) (AIT_GET_BUF(&v) + h->bh_hdrlen);
                    201:                                ret -= ETHER_HDR_LEN;
                    202:                                if (eh->ether_type != ntohs(RPC_DEFPORT)) {
1.19      misho     203:                                        AIT_FREE_VAL(&v);
1.16      misho     204:                                        if (cx < 3) {
                    205:                                                cx++;
                    206:                                                ret ^= ret;
1.19      misho     207:                                                goto try2read;  /* wait for known address */
                    208:                                        } else
1.16      misho     209:                                                return -1;
                    210:                                }
1.18      misho     211: 
1.16      misho     212:                                if (!memcmp(bcst.octet, eh->ether_dhost, sizeof bcst) || 
                    213:                                                !memcmp(bcst.octet, eh->ether_shost, sizeof bcst)) {
1.19      misho     214:                                        AIT_FREE_VAL(&v);
1.16      misho     215:                                        if (cx < 3) {
                    216:                                                cx++;
                    217:                                                ret ^= ret;
1.19      misho     218:                                                goto try2read;  /* wait for known address */
                    219:                                        } else
1.16      misho     220:                                                return -1;
                    221:                                }
1.19      misho     222:                                memcpy(buf, (u_char*) (eh + 1), MIN(ret, blen));
1.16      misho     223:                                AIT_FREE_VAL(&v);
                    224:                        }
1.19      misho     225:                        break;
1.20.2.1  misho     226: #endif
1.19      misho     227:                case SOCK_RAW:
                    228:                case SOCK_DGRAM:
                    229:                        if (!sa) {
                    230:                                rpc_SetErr(EINVAL, "Invalid argument(s)!");
                    231:                                return -1;
                    232:                        }
                    233: 
                    234:                        ret = recvfrom(sock, buf, blen, flags, &sa2.sa, &salen);
                    235:                        if (ret > -1 && e_addrcmp(sa, &sa2, 42)) {
1.18      misho     236:                                if (cx < 3) {
                    237:                                        cx++;
                    238:                                        ret ^= ret;
1.19      misho     239:                                        goto try2read;  /* wait for known address */
1.18      misho     240:                                } else
                    241:                                        return -1;
1.11      misho     242:                        }
1.19      misho     243:                        break;
                    244:                default:
                    245:                        rpc_SetErr(EINVAL, "Invalid argument(s)!");
1.13      misho     246:                        return -1;
1.19      misho     247:        }
                    248:        if (ret < 0) {
                    249:                LOGERR;
                    250:                return -1;
                    251:        }
                    252:        if (!ret)       /* EOF */
                    253:                return 0;
1.18      misho     254: 
1.19      misho     255:        /* check RPC packet header */
                    256:        if (type == SOCK_RAW) {
                    257:                hlen = sa->sa.sa_family == AF_INET ? 
                    258:                        sizeof(struct ip) : sizeof(struct ip6_hdr);
                    259:                ret -= hlen;
                    260:                if (ret > 0)
                    261:                        memmove(buf, buf + hlen, blen - hlen);
                    262:        }
1.18      misho     263: 
1.19      misho     264:        /* 1st read for RPC header */
                    265:        if (ret < sizeof(struct tagRPCCall) || ret < ntohl(rpc->call_len)) {
                    266:                rpc_SetErr(ERPCMISMATCH, "Short RPC packet %d bytes", ret);
                    267:                return -1;
                    268:        }
                    269:        /* check for loop request */
                    270:        if (!(rpc->call_io & RPC_ACK)) {
                    271:                if (cx < 3) {
                    272:                        cx++;
                    273:                        ret ^= ret;
                    274:                        goto try2read;  /* wait for known address */
                    275:                } else {
                    276:                        rpc_SetErr(ERPCMISMATCH, "Loop in RPC communication");
                    277:                        return -1;
1.18      misho     278:                }
1.11      misho     279:        }
                    280: 
                    281:        return ret;
                    282: }
                    283: 
                    284: /*
                    285:  * rpc_Write() - RPC write operation
                    286:  *
                    287:  * @sock = socket
                    288:  * @type = type of socket
                    289:  * @flags = send flags
                    290:  * @sa = send to client address, if you use udp protocol
1.18      misho     291:  * @pkt = RPC packet
1.11      misho     292:  * @blen = buffer length
1.14      misho     293:  * return: -1 error, 0 EOF or >0 written bytes into buffer
1.11      misho     294:  */
                    295: ssize_t
                    296: rpc_Write(int sock, int type, int flags, sockaddr_t * __restrict sa, 
1.18      misho     297:                ait_val_t * __restrict pkt, size_t blen)
1.11      misho     298: {
                    299:        struct pollfd pfd;
1.19      misho     300:        int ret = 0;
                    301:        u_char *buf = AIT_GET_BUF(pkt);
1.20.2.1  misho     302: #ifndef __linux__
1.16      misho     303:        ait_val_t v = AIT_VAL_INIT;
                    304:        struct ether_header *eh;
1.20.2.1  misho     305: #endif
1.16      misho     306: 
1.11      misho     307: 
                    308:        pfd.fd = sock;
                    309:        pfd.events = POLLOUT;
1.19      misho     310:        if ((ret = poll(&pfd, 1, DEF_RPC_TIMEOUT * 1000)) < 1 || 
                    311:                        pfd.revents & (POLLERR | POLLHUP | POLLNVAL)) {
                    312:                if (ret > 0)
                    313:                        rpc_SetErr(EPIPE, "Disconnected RPC session\n");
                    314:                else
                    315:                        LOGERR;
                    316:                return -1;
                    317:        }
                    318: 
                    319:        switch (type) {
                    320:                case SOCK_STREAM:
                    321:                        ret = send(sock, buf, blen, flags);
                    322:                        break;
                    323:                case SOCK_EXT:
                    324:                        ret = write(sock, buf, blen);
                    325:                        break;
1.20.2.1  misho     326: #ifndef __linux__
1.19      misho     327:                case SOCK_BPF:
                    328:                        if (!sa) {
                    329:                                rpc_SetErr(EINVAL, "Invalid argument(s)!");
                    330:                                return -1;
                    331:                        }
                    332: 
                    333:                        AIT_SET_BUF(&v, NULL, blen + sizeof(struct ether_header));
                    334:                        eh = (struct ether_header*) AIT_GET_BUF(&v);
                    335:                        memcpy(eh->ether_dhost, LLADDR(&sa->sdl), ETHER_ADDR_LEN);
                    336:                        eh->ether_type = htons(RPC_DEFPORT);
                    337:                        memcpy(eh + 1, buf, blen);
                    338:                        blen += sizeof(struct ether_header);
                    339: 
1.16      misho     340:                        ret = write(sock, AIT_GET_BUF(&v), AIT_LEN(&v));
1.19      misho     341: 
                    342:                        AIT_FREE_VAL(&v);
                    343:                        break;
1.20.2.1  misho     344: #endif
1.19      misho     345:                case SOCK_RAW:
                    346:                case SOCK_DGRAM:
                    347:                        if (!sa) {
                    348:                                rpc_SetErr(EINVAL, "Invalid argument(s)!");
                    349:                                return -1;
                    350:                        }
                    351: 
1.20.2.1  misho     352:                        ret = sendto(sock, buf, blen, flags, &sa->sa, sizeof sa->sa);
1.19      misho     353:                        break;
                    354:                default:
1.13      misho     355:                        rpc_SetErr(EINVAL, "Invalid argument(s)!");
                    356:                        return -1;
1.11      misho     357:        }
1.19      misho     358:        if (ret < 0) {
                    359:                LOGERR;
                    360:                return -1;
                    361:        }
                    362:        if (!ret)       /* EOF */
                    363:                return 0;
                    364: 
1.13      misho     365:        if (ret != blen) {
1.11      misho     366:                rpc_SetErr(EPROCUNAVAIL, "RPC request, should be send %d bytes, "
1.16      misho     367:                                "really sended %d bytes", blen, ret);
1.11      misho     368:                return -1;
                    369:        }
                    370: 
                    371:        return ret;
                    372: }
1.19      misho     373: 
                    374: /*
                    375:  * rpc_pktFreeSpace() - Get free space for payload into RPC packet
                    376:  *
                    377:  * @c = RPC client 
                    378:  * return: remains free bytes from packet
                    379:  */
                    380: size_t
                    381: rpc_pktFreeSpace(rpc_cli_t * __restrict c)
                    382: {
                    383:        return (sizeof(struct tagRPCCall) + ait_resideVars(RPC_RETVARS(c)));
                    384: }

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