Diff for /libaitrpc/src/cli.c between versions 1.4 and 1.13

version 1.4, 2011/08/29 22:37:06 version 1.13, 2012/11/19 21:50:26
Line 12  terms: Line 12  terms:
 All of the documentation and software included in the ELWIX and AITNET  All of the documentation and software included in the ELWIX and AITNET
 Releases is copyrighted by ELWIX - Sofia/Bulgaria <info@elwix.org>  Releases is copyrighted by ELWIX - Sofia/Bulgaria <info@elwix.org>
   
Copyright 2004, 2005, 2006, 2007, 2008, 2009, 2010, 2011Copyright 2004, 2005, 2006, 2007, 2008, 2009, 2010, 2011, 2012
         by Michael Pounov <misho@elwix.org>.  All rights reserved.          by Michael Pounov <misho@elwix.org>.  All rights reserved.
   
 Redistribution and use in source and binary forms, with or without  Redistribution and use in source and binary forms, with or without
Line 47  SUCH DAMAGE. Line 47  SUCH DAMAGE.
   
   
 /*  /*
 * rpc_cli_openBLOBClient() Connect to BLOB Server * rpc_cli_openBLOBClient() - Connect to BLOB Server
  *
  * @rpccli = RPC Client session   * @rpccli = RPC Client session
  * @Port = Port for bind server, if Port == 0 default port is selected   * @Port = Port for bind server, if Port == 0 default port is selected
  * return: NULL == error or !=NULL connection to BLOB server established   * return: NULL == error or !=NULL connection to BLOB server established
Line 56  rpc_cli_t * Line 57  rpc_cli_t *
 rpc_cli_openBLOBClient(rpc_cli_t * __restrict rpccli, u_short Port)  rpc_cli_openBLOBClient(rpc_cli_t * __restrict rpccli, u_short Port)
 {  {
         rpc_cli_t *cli = NULL;          rpc_cli_t *cli = NULL;
        struct sockaddr sa;        int n;
        struct sockaddr_in *sin = (struct sockaddr_in*) &sa; 
        struct sockaddr_in6 *sin6 = (struct sockaddr_in6*) &sa; 
        struct sockaddr_un *sun = (struct sockaddr_un*) &sa; 
   
        if (!rpccli ||         if (!rpccli) {
                        (rpccli->cli_sa.sa_family != AF_INET && rpccli->cli_sa.sa_family != AF_INET6 &&                 rpc_SetErr(EINVAL, "Invalid parameters can`t connect to BLOB server");
                         rpccli->cli_sa.sa_family != AF_LOCAL)) { 
                rpc_SetErr(EINVAL, "Error:: Invalid parameters can`t connect to BLOB server ...\n"); 
                 return NULL;                  return NULL;
         }          }
         if (!Port)  
                 Port = RPC_DEFPORT + 1;  
   
        cli = malloc(sizeof(rpc_cli_t));        cli = io_malloc(sizeof(rpc_cli_t));
         if (!cli) {          if (!cli) {
                 LOGERR;                  LOGERR;
                 return NULL;                  return NULL;
         } else          } else
                 memcpy(cli, rpccli, sizeof(rpc_cli_t));                  memcpy(cli, rpccli, sizeof(rpc_cli_t));
   
        memcpy(&sa, &rpccli->cli_sa, sizeof sa);        memcpy(&cli->cli_sa, &rpccli->cli_sa, sizeof(io_sockaddr_t));
        switch (rpccli->cli_sa.sa_family) {        switch (cli->cli_sa.sa.sa_family) {
                 case AF_INET:                  case AF_INET:
                        sin->sin_port = htons(Port);                        cli->cli_sa.sin.sin_port = 
                        memcpy(&cli->cli_sa, sin, sizeof(struct sockaddr));                                htons(Port ? Port : ntohs(cli->cli_sa.sin.sin_port) + 1);
                         break;                          break;
                 case AF_INET6:                  case AF_INET6:
                        sin6->sin6_port = htons(Port);                        cli->cli_sa.sin6.sin6_port = 
                        memcpy(&cli->cli_sa, sin6, sizeof(struct sockaddr));                                htons(Port ? Port : ntohs(cli->cli_sa.sin6.sin6_port) + 1);
                         break;                          break;
                 case AF_LOCAL:                  case AF_LOCAL:
                        strlcat(sun->sun_path, ".blob", sizeof sun->sun_path);                        strlcat(cli->cli_sa.sun.sun_path, ".blob", sizeof cli->cli_sa.sun.sun_path);
                        memcpy(&cli->cli_sa, sun, sizeof(struct sockaddr)); 
                         break;                          break;
                   default:
                           rpc_SetErr(EINVAL, "Invalid socket type %d", cli->cli_sa.sa.sa_family);
                           return NULL;
         }          }
   
           AIT_COPY_VAL(&cli->cli_buf, &rpccli->cli_buf);
           n = AIT_LEN(&cli->cli_buf);
   
         /* connect to BLOB server */          /* connect to BLOB server */
        cli->cli_sock = socket(cli->cli_sa.sa_family, SOCK_STREAM, 0);        cli->cli_sock = socket(cli->cli_sa.sa.sa_family, SOCK_STREAM, 0);
         if (cli->cli_sock == -1) {          if (cli->cli_sock == -1) {
                 LOGERR;                  LOGERR;
                free(cli);                io_free(cli);
                 return NULL;                  return NULL;
         }          }
        if (connect(cli->cli_sock, &cli->cli_sa, sizeof cli->cli_sa) == -1) {        if (setsockopt(cli->cli_sock, SOL_SOCKET, SO_SNDBUF, &n, sizeof n) == -1) {
                 LOGERR;                  LOGERR;
                free(cli);                close(cli->cli_sock);
                 io_free(cli);
                 return NULL;                  return NULL;
         }          }
           if (setsockopt(cli->cli_sock, SOL_SOCKET, SO_RCVBUF, &n, sizeof n) == -1) {
                   LOGERR;
                   close(cli->cli_sock);
                   io_free(cli);
                   return NULL;
           }
           if (connect(cli->cli_sock, &cli->cli_sa.sa, cli->cli_sa.sa.sa_len) == -1) {
                   LOGERR;
                   close(cli->cli_sock);
                   io_free(cli);
                   return NULL;
           } else
                   fcntl(cli->cli_sock, F_SETFL, fcntl(cli->cli_sock, F_GETFL) | O_NONBLOCK);
   
         return cli;          return cli;
 }  }
   
 /*  /*
 * rpc_cli_closeBLOBClient() Close connection to BLOB server and free resources * rpc_cli_closeBLOBClient() - Close connection to BLOB server and free resources
  *
  * @cli = BLOB Client session   * @cli = BLOB Client session
  * return: none   * return: none
  */   */
 void  void
rpc_cli_closeBLOBClient(rpc_cli_t * __restrict cli)rpc_cli_closeBLOBClient(rpc_cli_t ** __restrict cli)
 {  {
        if (!cli) {        if (!cli || !*cli)
                rpc_SetErr(EINVAL, "Error:: Can`t close connection because parameter is null!\n"); 
                 return;                  return;
         }  
   
        shutdown(cli->cli_sock, SHUT_RDWR);        shutdown((*cli)->cli_sock, SHUT_RDWR);
        close(cli->cli_sock);        close((*cli)->cli_sock);
   
        free(cli);        AIT_FREE_VAL(&(*cli)->cli_buf);
        cli = NULL;
         io_free(*cli);
         *cli = NULL;
 }  }
   
// --------------------------------------------------------------/* -------------------------------------------------------------- */
   
 /*  /*
 * rpc_cli_openClient() Connect to RPC Server * rpc_cli_openClient() - Connect to RPC Server
  *
  * @ProgID = ProgramID for RPC session request   * @ProgID = ProgramID for RPC session request
  * @ProcID = ProcessID for RPC session request   * @ProcID = ProcessID for RPC session request
 * @family = Family socket type, AF_INET or AF_INET6 * @netBuf = Network buffer length (min:512 bytes), if =0 == BUFSIZ (also meaning max RPC packet)
  * @csHost = Host name or IP address for bind server   * @csHost = Host name or IP address for bind server
  * @Port = Port for bind server, if Port == 0 default port is selected   * @Port = Port for bind server, if Port == 0 default port is selected
    * @proto = Protocol, if == 0 choose SOCK_STREAM
  * return: NULL == error or !=NULL connection to RPC server established   * return: NULL == error or !=NULL connection to RPC server established
  */   */
 rpc_cli_t *  rpc_cli_t *
rpc_cli_openClient(u_int ProgID, u_int ProcID, u_short family, const char *csHost, u_short Port)rpc_cli_openClient(u_int ProgID, u_char ProcID, int netBuf, const char *csHost, 
                 u_short Port, int proto)
 {  {
         rpc_cli_t *cli = NULL;          rpc_cli_t *cli = NULL;
        struct hostent *host = NULL;        io_sockaddr_t sa = IO_SOCKADDR_INIT;
        struct sockaddr sa; 
        struct sockaddr_in *sin = (struct sockaddr_in*) &sa; 
        struct sockaddr_in6 *sin6 = (struct sockaddr_in6*) &sa; 
        struct sockaddr_un *sun = (struct sockaddr_un*) &sa; 
   
        if (!csHost || (family != AF_INET && family != AF_INET6 && family != AF_LOCAL)) {        if (!io_gethostbyname(csHost, Port, &sa))
                rpc_SetErr(EINVAL, "Error:: Invalid parameters can`t connect to RPC server ...\n"); 
                 return NULL;                  return NULL;
         }  
         if (!Port)          if (!Port)
                 Port = RPC_DEFPORT;                  Port = RPC_DEFPORT;
        if (csHost && family != AF_LOCAL) {        if (!proto)
                host = gethostbyname2(csHost, family);                proto = SOCK_STREAM;
                if (!host) {        if (netBuf < RPC_MIN_BUFSIZ)
                        rpc_SetErr(h_errno, "Error:: %s\n", hstrerror(h_errno));                netBuf = BUFSIZ;
                        return NULL;        else
                }                netBuf = io_align(netBuf, 2);   /* align netBuf length */
        } 
        memset(&sa, 0, sizeof sa); 
        sa.sa_family = family; 
        switch (family) { 
                case AF_INET: 
                        sin->sin_len = sizeof(struct sockaddr_in); 
                        sin->sin_port = htons(Port); 
                        if (csHost) 
                                memcpy(&sin->sin_addr, host->h_addr, host->h_length); 
                        break; 
                case AF_INET6: 
                        sin6->sin6_len = sizeof(struct sockaddr_in6); 
                        sin6->sin6_port = htons(Port); 
                        if (csHost) 
                                memcpy(&sin6->sin6_addr, host->h_addr, host->h_length); 
                        break; 
                case AF_LOCAL: 
                        sun->sun_len = sizeof(struct sockaddr_un); 
                        if (csHost) 
                                strlcpy(sun->sun_path, csHost, sizeof sun->sun_path); 
                        break; 
                default: 
                        rpc_SetErr(EINVAL, "Error:: Invalid parameters can`t connect to RPC server ...\n"); 
                        return NULL; 
        } 
   
        cli = malloc(sizeof(rpc_cli_t));#ifdef HAVE_SRANDOMDEV
         srandomdev();
 #else
         time_t tim;
 
         srandom((time(&tim) ^ getpid()));
 #endif
 
         cli = io_malloc(sizeof(rpc_cli_t));
         if (!cli) {          if (!cli) {
                 LOGERR;                  LOGERR;
                 return NULL;                  return NULL;
         } else          } else
                 memset(cli, 0, sizeof(rpc_cli_t));                  memset(cli, 0, sizeof(rpc_cli_t));
        cli->cli_parent = malloc(sizeof(rpc_sess_t));
         /* build session */
         cli->cli_parent = io_malloc(sizeof(rpc_sess_t));
         if (!cli->cli_parent) {          if (!cli->cli_parent) {
                 LOGERR;                  LOGERR;
                free(cli);                io_free(cli);
                 return NULL;                  return NULL;
         } else {          } else {
                 ((rpc_sess_t*) cli->cli_parent)->sess_version = RPC_VERSION;                  ((rpc_sess_t*) cli->cli_parent)->sess_version = RPC_VERSION;
Line 205  rpc_cli_openClient(u_int ProgID, u_int ProcID, u_short Line 201  rpc_cli_openClient(u_int ProgID, u_int ProcID, u_short
                 ((rpc_sess_t*) cli->cli_parent)->sess_process = ProcID;                  ((rpc_sess_t*) cli->cli_parent)->sess_process = ProcID;
         }          }
   
        switch (family) {        cli->cli_id = proto;
                case AF_INET:        memcpy(&cli->cli_sa, &sa, sizeof cli->cli_sa);
                        memcpy(&cli->cli_sa, sin, sizeof cli->cli_sa);        AIT_SET_BUF2(&cli->cli_buf, 0, netBuf);
                        break; 
                case AF_INET6: 
                        memcpy(&cli->cli_sa, sin6, sizeof cli->cli_sa); 
                        break; 
                case AF_LOCAL: 
                        memcpy(&cli->cli_sa, sun, sizeof cli->cli_sa); 
                        break; 
        } 
   
         /* connect to RPC server */          /* connect to RPC server */
        cli->cli_sock = socket(family, SOCK_STREAM, 0);        cli->cli_sock = socket(cli->cli_sa.sa.sa_family, cli->cli_id, 0);
         if (cli->cli_sock == -1) {          if (cli->cli_sock == -1) {
                 LOGERR;                  LOGERR;
                free(cli->cli_parent);                goto err;
                free(cli); 
                return NULL; 
         }          }
        if (connect(cli->cli_sock, &cli->cli_sa, sizeof cli->cli_sa) == -1) {        if (setsockopt(cli->cli_sock, SOL_SOCKET, SO_SNDBUF, 
                                 &netBuf, sizeof netBuf) == -1) {
                 LOGERR;                  LOGERR;
                free(cli->cli_parent);                goto err;
                free(cli); 
                return NULL; 
         }          }
           if (setsockopt(cli->cli_sock, SOL_SOCKET, SO_RCVBUF, 
                                   &netBuf, sizeof netBuf) == -1) {
                   LOGERR;
                   goto err;
           }
           if (cli->cli_id == SOCK_STREAM)
                   if (connect(cli->cli_sock, &cli->cli_sa.sa, cli->cli_sa.sa.sa_len) == -1) {
                           LOGERR;
                           goto err;
                   }
   
           fcntl(cli->cli_sock, F_SETFL, fcntl(cli->cli_sock, F_GETFL) | O_NONBLOCK);
         return cli;          return cli;
   err:
           AIT_FREE_VAL(&cli->cli_buf);
           if (cli->cli_sock > 2)
                   close(cli->cli_sock);
           io_free(cli->cli_parent);
           io_free(cli);
           return NULL;
 }  }
   
 /*  /*
 * rpc_cli_closeClient() Close connection to RPC server and free resources * rpc_cli_closeClient() - Close connection to RPC server and free resources
  *
  * @cli = RPC Client session   * @cli = RPC Client session
  * return: none   * return: none
  */   */
 void  void
rpc_cli_closeClient(rpc_cli_t * __restrict cli)rpc_cli_closeClient(rpc_cli_t ** __restrict cli)
 {  {
        if (!cli) {        if (!cli || !*cli)
                rpc_SetErr(EINVAL, "Error:: Can`t close connection because parameter is null!\n"); 
                 return;                  return;
         }  
   
        shutdown(cli->cli_sock, SHUT_RDWR);        if ((*cli)->cli_id == SOCK_STREAM)
        close(cli->cli_sock);                shutdown((*cli)->cli_sock, SHUT_RDWR);
         close((*cli)->cli_sock);
   
        if (cli->cli_parent)        AIT_FREE_VAL(&(*cli)->cli_buf);
                free(cli->cli_parent);
        free(cli);        if ((*cli)->cli_parent)
        cli = NULL;                io_free((*cli)->cli_parent);
 
         io_free(*cli);
         *cli = NULL;
 }  }
   
   
 /*  /*
 * rpc_cli_execCall() Execute RPC call * rpc_cli_execCall() - Execute RPC call
  *
  * @cli = RPC Client session   * @cli = RPC Client session
 * @csModule = Module name, if NULL self binary * @noreply = We not want RPC reply
 * @csFunc = Function name for execute * @tag = Function tag for execution
 * @in_argc = IN count of arguments * @in_vars = IN RPC call array of rpc values, may be NULL
 * @in_vars = IN RPC call array of rpc values * @out_vars = OUT returned array of rpc values, if !=NULL must be free after use with io_freeVars()
 * @out_argc = OUT returned count of arguments 
 * @out_vars = OUT returned array of rpc values, must be free after use (see rpc_cli_freeVals()) 
  * return: -1 error or != -1 ok result   * return: -1 error or != -1 ok result
  */   */
 int  int
rpc_cli_execCall(rpc_cli_t *cli, const char *csModule, const char *csFunc, int in_argc, rpc_cli_execCall(rpc_cli_t *cli, int noreply, u_short tag, 
                rpc_val_t * __restrict in_vars, int *out_argc, rpc_val_t ** __restrict out_vars)                array_t * __restrict in_vars, array_t ** __restrict out_vars)
 {  {
        fd_set fds;        struct tagRPCCall *rpc;
        u_char buf[BUFSIZ], str[MAXPATHLEN + UCHAR_MAX + 1], *data;        int ret = 0, wlen = sizeof(struct tagRPCCall);
        struct tagRPCCall *rpc = (struct tagRPCCall*) buf;        uint16_t crc;
        struct tagRPCRet *rrpc = NULL;        u_char *buf;
        int ret = 0, Limit = 0;        struct pollfd pfd;
        register int i;        io_sockaddr_t sa;
        rpc_val_t *v;        socklen_t salen;
        struct timeval tv = { DEF_RPC_TIMEOUT, 0 }; 
   
        FD_ZERO(&fds);        if (!cli) {
        memset(buf, 0, sizeof buf);                rpc_SetErr(EINVAL, "Can`t execute call because parameter is null or invalid!");
        memset(str, 0, sizeof str); 
        if (!cli || !csFunc || (in_argc && !in_vars)) { 
                rpc_SetErr(EINVAL, "Error:: Can`t execute call because parameter is null or invalid!\n"); 
                 return -1;                  return -1;
        }        } else
        /* build RPC call string for hash */                buf = AIT_GET_BUF(&cli->cli_buf);
        if (csModule)        if (out_vars)
                strlcpy((char*) str, csModule, sizeof str);                *out_vars = NULL;
        strlcat((char*) str, "__", sizeof str); 
        strlcat((char*) str, csFunc, sizeof str); 
   
           memset(&sa, 0, sizeof sa);
   
         /* prepare RPC call */          /* prepare RPC call */
        memcpy(&rpc->call_session, cli->cli_parent, sizeof rpc->call_session);        rpc = (struct tagRPCCall*) buf;
        rpc->call_argc = in_argc;        rpc_addPktSession(&rpc->call_session, cli->cli_parent);
        rpc->call_tag = crcFletcher16((u_short*) str, sizeof str / 2);        rpc->call_argc = htons(io_arraySize(in_vars));
        rpc->call_hash = hash_fnv((char*) str, sizeof str);        rpc->call_tag = htons(tag);
        Limit = sizeof(struct tagRPCCall);        rpc->call_seq = htons(random() % USHRT_MAX);
        if (in_argc) {
                v = (rpc_val_t*) (buf + sizeof(struct tagRPCCall));        /* set reply */
                if (in_argc * sizeof(rpc_val_t) > sizeof buf - Limit) {        rpc->call_req.flags = noreply ? RPC_NOREPLY : RPC_REPLY;
                        rpc_SetErr(EMSGSIZE, "Error:: in prepare RPC packet values (-7) ...\n");
                        return -7;        if (io_arraySize(in_vars)) {
                 /* marshaling variables */
                 ret = io_vars2buffer(buf + wlen, AIT_LEN(&cli->cli_buf) - wlen, in_vars);
                 if (ret == -1) {
                         rpc_SetErr(EBADRPC, "Failed to prepare RPC packet values");
                         return -1;
                 } else                  } else
                        Limit += in_argc * sizeof(rpc_val_t);                        wlen += ret;
                memcpy(v, in_vars, in_argc * sizeof(rpc_val_t));        }
                data = (u_char*) v + in_argc * sizeof(rpc_val_t); 
                for (i = 0; i < in_argc && !ret; i++) { 
                        switch (in_vars[i].val_type) { 
                                case buffer: 
                                        if (Limit + in_vars[i].val_len > sizeof buf) { 
                                                ret = -7; 
                                                break; 
                                        } 
   
                                        memcpy(data, in_vars[i].val.buffer, in_vars[i].val_len);        /* total packet length */
                                        v[i].val.buffer = (uint8_t*) ((void*) data - (void*) v);        rpc->call_len = htons(wlen);
                                        data += in_vars[i].val_len; 
                                        Limit += in_vars[i].val_len; 
                                        break; 
                                case string: 
                                        if (Limit + in_vars[i].val_len > sizeof buf) { 
                                                ret = -7; 
                                                break; 
                                        } 
   
                                        memcpy(data, in_vars[i].val.string, in_vars[i].val_len);        /* calculate CRC */
                                        v[i].val.string = (int8_t*) ((void*) data - (void*) v);        rpc->call_crc ^= rpc->call_crc;
                                        data += in_vars[i].val_len;        rpc->call_crc = htons(crcFletcher16((u_short*) buf, wlen / 2));
                                        Limit += in_vars[i].val_len;
                                        break;        pfd.fd = cli->cli_sock;
                                default:        pfd.events = POLLOUT;
                                        break;        if ((ret = poll(&pfd, 1, DEF_RPC_TIMEOUT * 1000)) < 1 || 
                         pfd.revents & (POLLERR | POLLHUP | POLLNVAL)) {
                 if (ret)
                         LOGERR;
                 else
                         rpc_SetErr(ETIMEDOUT, "Timeout, can't send to RPC server");
                 return -1;
         }
         do {
                 if (cli->cli_id == SOCK_STREAM)
                         ret = send(cli->cli_sock, buf, wlen, MSG_NOSIGNAL);
                 else
                         ret = sendto(cli->cli_sock, buf, wlen, MSG_NOSIGNAL, 
                                         &cli->cli_sa.sa, cli->cli_sa.sa.sa_len);
                 if (ret == -1) {
                         if (errno == EAGAIN)
                                 continue;
                         LOGERR;
                         return -1;
                 } else if (ret != wlen) {
                         rpc_SetErr(EPROCUNAVAIL, "RPC request, should be send %d bytes, "
                                         "really sended %d bytes", wlen, ret);
                         return -1;
                 }
         } while (0);
 
         if (noreply)    /* we not want reply */
                 return 0;
 
         wlen = 0;
         /* reply from RPC server */
         pfd.events = POLLIN | POLLPRI;
         do {
                 if ((ret = poll(&pfd, 1, DEF_RPC_TIMEOUT * 1000)) < 1 || 
                                 pfd.revents & (POLLERR | POLLHUP | POLLNVAL)) {
                         if (ret) {
                                 LOGERR;
                         } else {
                                 if (wlen++ < 7)
                                         continue;
                                 else
                                         rpc_SetErr(ETIMEDOUT, "Timeout, no answer from RPC server");
                         }                          }
   
                           return -1;
                 }                  }
                if (ret < 0) {
                        rpc_SetErr(EMSGSIZE, "Error:: in prepare RPC packet (-7) ...\n");                memset(buf, 0, AIT_LEN(&cli->cli_buf));
                        return ret;                if (cli->cli_id == SOCK_STREAM)
                         ret = recv(cli->cli_sock, buf, AIT_LEN(&cli->cli_buf), 0);
                 else
                         ret = recvfrom(cli->cli_sock, buf, AIT_LEN(&cli->cli_buf), 0, 
                                         &sa.sa, &salen);
                 if (ret < 1) {
                         if (ret) {
                                 if (errno == EAGAIN)
                                         continue;
                                 else
                                         LOGERR;
                         }
                         return -1;
                 }                  }
         }  
   
        if ((ret = send(cli->cli_sock, buf, Limit, 0)) == -1) {                if (cli->cli_id == SOCK_DGRAM) {
                LOGERR;                        /* check for response from known address */
                         if (io_addrcmp(&cli->cli_sa, &sa, 42)) {
                                 rpc_SetErr(ERPCMISMATCH, "Received RPC response from unknown address");
                                 continue;
                         }
                 }
         } while (0);
         if (ret < sizeof(struct tagRPCCall)) {
                 rpc_SetErr(ERPCMISMATCH, "Short RPC packet %d bytes", ret);
                 return -1;                  return -1;
         }          }
         if (ret != Limit) {  
                 rpc_SetErr(ECANCELED, "Error:: in send RPC request, should be send %d bytes, really is %d\n",   
                                 Limit, ret);  
                 return -9;  
         }  
   
        FD_SET(cli->cli_sock, &fds);        /* calculate CRC */
        if ((ret = select(cli->cli_sock + 1, &fds, NULL, NULL, &tv)) < 1) {        crc = ntohs(rpc->call_crc);
                if (ret)        rpc->call_crc ^= rpc->call_crc;
                        LOGERR;        if (crc != crcFletcher16((u_short*) buf, ntohs(rpc->call_len) / 2)) {
                else                rpc_SetErr(ERPCMISMATCH, "Bad CRC RPC packet");
                        rpc_SetErr(ETIMEDOUT, "Error:: timeout, no return from RPC server?\n"); 
 
                 return -1;                  return -1;
         }          }
        memset(buf, 0, sizeof buf);
        if ((ret = recv(cli->cli_sock, buf, sizeof buf, 0)) == -1) { 
                LOGERR; 
                return -3; 
        } 
        if (!ret)       /* receive EOF! */ 
                return 0; 
        if (ret < sizeof(struct tagRPCRet)) { 
                rpc_SetErr(EMSGSIZE, "Error:: too short RPC packet ...\n"); 
                return -4; 
        } else 
                rrpc = (struct tagRPCRet*) buf; 
         /* check RPC packet session info */          /* check RPC packet session info */
        if (memcmp(&rrpc->ret_session, cli->cli_parent, sizeof rrpc->ret_session)) {        if (rpc_chkPktSession(&rpc->call_session, cli->cli_parent)) {
                rpc_SetErr(EINVAL, "Error:: get invalid RPC session ...\n");                rpc_SetErr(ERPCMISMATCH, "Get invalid RPC session");
                return -5;                return -1;
         } else          } else
                Limit = sizeof(struct tagRPCRet);                wlen = sizeof(struct tagRPCCall);
        if (rrpc->ret_retcode < 0 && rrpc->ret_errno) {        if (ntohs(rpc->call_tag) != tag) {
                rpc_SetErr(rrpc->ret_errno, "Error::Server side: %d %s\n",                 rpc_SetErr(ERPCMISMATCH, "Get wrong RPC reply");
                                rrpc->ret_retcode, strerror(rrpc->ret_errno));                return -1;
                return -6; 
         }          }
        if (rrpc->ret_argc * sizeof(rpc_val_t) > sizeof buf - Limit) {        if (ntohl(rpc->call_rep.eno) && ntohl(rpc->call_rep.ret) == -1) {
                rpc_SetErr(EMSGSIZE, "Error:: reply RPC packet is too long ...\n");                rpc_SetErr(ntohl(rpc->call_rep.eno), "Server side: retcode=%d #%d %s", 
                return -7;                                ntohl(rpc->call_rep.ret), ntohl(rpc->call_rep.eno), 
        } else                                strerror(ntohl(rpc->call_rep.eno)));
                Limit += rrpc->ret_argc * sizeof(rpc_val_t);                return -1;
        /* RPC is OK! Go decapsulate variables ... */        }
        if (rrpc->ret_argc) {        if (ntohs(rpc->call_argc) * sizeof(ait_val_t) > AIT_LEN(&cli->cli_buf) - wlen) {
                *out_argc = rrpc->ret_argc;                rpc_SetErr(EMSGSIZE, "Reply RPC packet is too long ...");
                *out_vars = calloc(rrpc->ret_argc, sizeof(rpc_val_t));                return -1;
         }
 
         /* RPC is OK! Go de-marshaling variables ... */
         if (out_vars && ntohs(rpc->call_argc)) {
 #ifdef CLI_RES_ZCOPY
                 *out_vars = io_buffer2vars(buf + wlen, AIT_LEN(&cli->cli_buf) - wlen, 
                                 ntohs(rpc->call_argc), 42);
 #else
                 *out_vars = io_buffer2vars(buf + wlen, AIT_LEN(&cli->cli_buf) - wlen, 
                                 ntohs(rpc->call_argc), 0);
 #endif
                 if (!*out_vars) {                  if (!*out_vars) {
                        LOGERR;                        rpc_SetErr(io_GetErrno(), "%s", io_GetError());
                        *out_argc = 0; 
                         return -1;                          return -1;
                } else                }
                        memcpy(*out_vars, buf + sizeof(struct tagRPCRet), Limit - sizeof(struct tagRPCRet)); 
                /* RPC received variables types OK! */ 
                data = (u_char*) buf + Limit; 
                for (i = 0; i < rrpc->ret_argc; i++) 
                        switch ((*out_vars)[i].val_type) { 
                                case buffer: 
                                        if ((*out_vars)[i].val_len > sizeof buf - Limit) { 
                                                rpc_SetErr(EMSGSIZE, "Error:: Too long RPC packet ...\n"); 
                                                free(*out_vars); 
                                                *out_vars = NULL; 
                                                *out_argc = 0; 
                                                return -7; 
                                        } else 
                                                Limit += (*out_vars)[i].val_len; 
 
                                        (*out_vars)[i].val.buffer = malloc((*out_vars)[i].val_len); 
                                        if (!(*out_vars)[i].val.buffer) { 
                                                rpc_SetErr(errno, "Error:: in prepare RPC reply ...\n"); 
                                                free(*out_vars); 
                                                *out_vars = NULL; 
                                                *out_argc = 0; 
                                                return -1; 
                                        } else 
                                                memcpy((*out_vars)[i].val.buffer, data, (*out_vars)[i].val_len); 
                                        data += (*out_vars)[i].val_len; 
                                        break; 
                                case string: 
                                        if ((*out_vars)[i].val_len > sizeof buf - Limit) { 
                                                rpc_SetErr(EMSGSIZE, "Error:: Too long RPC packet ...\n"); 
                                                free(*out_vars); 
                                                *out_vars = NULL; 
                                                *out_argc = 0; 
                                                return -7; 
                                        } else 
                                                Limit += (*out_vars)[i].val_len; 
 
                                        (*out_vars)[i].val.string = malloc((*out_vars)[i].val_len); 
                                        if (!(*out_vars)[i].val.string) { 
                                                rpc_SetErr(errno, "Error:: in prepare RPC reply ...\n"); 
                                                free(*out_vars); 
                                                *out_vars = NULL; 
                                                *out_argc = 0; 
                                                return -1; 
                                        } else 
                                                memcpy((*out_vars)[i].val.string, data, (*out_vars)[i].val_len); 
                                        data += (*out_vars)[i].val_len; 
                                        break; 
                                default: 
                                        break; 
                        } 
        } else { 
                if (out_argc) 
                        *out_argc = 0; 
                if (out_vars) 
                        *out_vars = NULL; 
         }          }
   
        return rrpc->ret_retcode;        ret = ntohl(rpc->call_rep.ret);
         return ret;
 }  }
   
 /*  /*
 * rpc_cli_freeVals() Free rpc_val_t array returned from RPC call * rpc_cli_freeCall() - Free resouce allocated by RPC call
 * @args = Number of arguments in array *
 * @vars = Value elements * @out_vars = Returned array with variables from RPC call
  * return: none   * return: none
  */   */
 inline void  inline void
rpc_cli_freeVals(int args, rpc_val_t *vars)rpc_cli_freeCall(array_t ** __restrict out_vars)
 {  {
        register int i;#ifdef CLI_RES_ZCOPY
         io_arrayDestroy(out_vars);
 #else
         io_freeVars(out_vars);
 #endif
 }
   
        if (!vars)/*
                return; * rpc_cli_ping() - Ping RPC server
  *
  * @cli = connected client
  * return: -1 error or !=-1 ping seq id
  */
 inline int
 rpc_cli_ping(rpc_cli_t *cli)
 {
         int ret = 0;
         array_t *arr = NULL;
   
        for (i = 0; i < args; i++)        if (!cli)
                RPC_FREE_VAL(&vars[i]);                return -1;
   
        free(vars);        if (rpc_cli_execCall(cli, RPC_REPLY, CALL_SRVPING, NULL, &arr))
        vars = NULL;                return -1;
         else
                 ret = AIT_GET_U16(io_array(arr, 0, ait_val_t*));
         rpc_cli_freeCall(&arr);
 
         return ret;
 }  }

Removed from v.1.4  
changed lines
  Added in v.1.13


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