--- libaitrpc/src/cli.c 2013/11/14 22:53:12 1.19.6.1 +++ libaitrpc/src/cli.c 2013/11/15 09:55:53 1.20 @@ -3,7 +3,7 @@ * by Michael Pounov * * $Author: misho $ -* $Id: cli.c,v 1.19.6.1 2013/11/14 22:53:12 misho Exp $ +* $Id: cli.c,v 1.20 2013/11/15 09:55:53 misho Exp $ * ************************************************************************** The ELWIX and AITNET software is distributed under the following @@ -295,8 +295,10 @@ rpc_pkt_Receive(int sock, int type, sockaddr_t * __res if (!pkt) { rpc_SetErr(EINVAL, "Invalid argument(s)!"); return -1; - } else + } else { buf = AIT_GET_BUF(pkt); + rpc = (struct tagRPCCall*) buf; + } /* reply from RPC server */ do { @@ -304,10 +306,8 @@ rpc_pkt_Receive(int sock, int type, sockaddr_t * __res ret = rpc_Read(sock, type, !estlen ? MSG_PEEK : 0, NULL, buf, blen); else ret = rpc_Read(sock, type, !estlen ? MSG_PEEK : 0, sa, buf, blen); - if (ret < 1) { - LOGERR; + if (ret < 1) return -1; - } /* check for response from known address */ if (!estlen) { @@ -318,13 +318,12 @@ rpc_pkt_Receive(int sock, int type, sockaddr_t * __res } /* calc estimated length */ - rpc = (struct tagRPCCall*) buf; estlen = ntohl(rpc->call_len); if (estlen > AIT_LEN(pkt)) AIT_RE_BUF(pkt, estlen); buf = AIT_GET_BUF(pkt); - blen = estlen; rpc = (struct tagRPCCall*) buf; + blen = estlen; continue; } @@ -332,8 +331,9 @@ rpc_pkt_Receive(int sock, int type, sockaddr_t * __res break; } while (42); - if (ret < sizeof(struct tagRPCCall)) { - rpc_SetErr(ERPCMISMATCH, "Short RPC packet %d bytes", ret); + if (ret < sizeof(struct tagRPCCall) || estlen != ret) { + rpc_SetErr(ERPCMISMATCH, "RPC packet mismatch estimate %d bytes, but received %d\n", + estlen, ret); return -1; }