Diff for /libaitrpc/src/cli.c between versions 1.4.2.3 and 1.13.4.4

version 1.4.2.3, 2011/08/30 12:59:02 version 1.13.4.4, 2013/03/07 23:03:03
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, 2013
         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;  
         struct sockaddr_in *sin = (struct sockaddr_in*) &sa;  
         struct sockaddr_in6 *sin6 = (struct sockaddr_in6*) &sa;  
         struct sockaddr_un *sun = (struct sockaddr_un*) &sa;  
         int n;          int n;
   
        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 = e_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(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);                e_free(cli);
                 return NULL;                  return NULL;
         }          }
         n = cli->cli_netbuf;  
         if (setsockopt(cli->cli_sock, SOL_SOCKET, SO_SNDBUF, &n, sizeof n) == -1) {          if (setsockopt(cli->cli_sock, SOL_SOCKET, SO_SNDBUF, &n, sizeof n) == -1) {
                 LOGERR;                  LOGERR;
                 close(cli->cli_sock);                  close(cli->cli_sock);
                free(cli->cli_parent);                e_free(cli);
                free(cli); 
                 return NULL;                  return NULL;
         }          }
         if (setsockopt(cli->cli_sock, SOL_SOCKET, SO_RCVBUF, &n, sizeof n) == -1) {          if (setsockopt(cli->cli_sock, SOL_SOCKET, SO_RCVBUF, &n, sizeof n) == -1) {
                 LOGERR;                  LOGERR;
                 close(cli->cli_sock);                  close(cli->cli_sock);
                free(cli->cli_parent);                e_free(cli);
                free(cli); 
                 return NULL;                  return NULL;
         }          }
        if (connect(cli->cli_sock, &cli->cli_sa, sizeof cli->cli_sa) == -1) {        if (connect(cli->cli_sock, &cli->cli_sa.sa, cli->cli_sa.sa.sa_len) == -1) {
                 LOGERR;                  LOGERR;
                free(cli);                close(cli->cli_sock);
                 e_free(cli);
                 return NULL;                  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;
         e_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
 * @netBuf = Network buffer length, if =0 == BUFSIZ (also meaning max RPC packet) * @netBuf = Network buffer length (min:512 bytes), if =0 == BUFSIZ (also meaning max RPC packet)
 * @family = Family socket type, AF_INET or AF_INET6 
  * @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, int netBuf, 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;        sockaddr_t sa = E_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; 
        int n; 
   
        if (!csHost || (family != AF_INET && family != AF_INET6 && family != AF_LOCAL)) {        if (!e_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 (!netBuf)        if (!proto)
                 proto = SOCK_STREAM;
         if (netBuf < RPC_MIN_BUFSIZ)
                 netBuf = BUFSIZ;                  netBuf = BUFSIZ;
        if (csHost && family != AF_LOCAL) {        else
                host = gethostbyname2(csHost, family);                netBuf = E_ALIGN(netBuf, 2);    /* align netBuf length */
                if (!host) { 
                        rpc_SetErr(h_errno, "Error:: %s\n", hstrerror(h_errno)); 
                        return NULL; 
                } 
        } 
        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 = e_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_netbuf = netBuf;        /* build session */
        cli->cli_parent = malloc(sizeof(rpc_sess_t));        cli->cli_parent = e_malloc(sizeof(rpc_sess_t));
         if (!cli->cli_parent) {          if (!cli->cli_parent) {
                 LOGERR;                  LOGERR;
                free(cli);                e_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 227  rpc_cli_openClient(u_int ProgID, u_int ProcID, int net Line 201  rpc_cli_openClient(u_int ProgID, u_int ProcID, int net
                 ((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_BUFSIZ(&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; 
         }          }
        n = cli->cli_netbuf;        if (setsockopt(cli->cli_sock, SOL_SOCKET, SO_SNDBUF, 
        if (setsockopt(cli->cli_sock, SOL_SOCKET, SO_SNDBUF, &n, sizeof n) == -1) {                                &netBuf, sizeof netBuf) == -1) {
                 LOGERR;                  LOGERR;
                close(cli->cli_sock);                goto err;
                free(cli->cli_parent); 
                free(cli); 
                return NULL; 
         }          }
        if (setsockopt(cli->cli_sock, SOL_SOCKET, SO_RCVBUF, &n, sizeof n) == -1) {        if (setsockopt(cli->cli_sock, SOL_SOCKET, SO_RCVBUF, 
                                 &netBuf, sizeof netBuf) == -1) {
                 LOGERR;                  LOGERR;
                close(cli->cli_sock);                goto err;
                free(cli->cli_parent); 
                free(cli); 
                return NULL; 
         }          }
        if (connect(cli->cli_sock, &cli->cli_sa, sizeof cli->cli_sa) == -1) {        if (cli->cli_id == SOCK_STREAM)
                LOGERR;                if (connect(cli->cli_sock, &cli->cli_sa.sa, cli->cli_sa.sa.sa_len) == -1) {
                close(cli->cli_sock);                        LOGERR;
                free(cli->cli_parent);                        goto err;
                free(cli);                }
                return NULL; 
        } 
   
           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);
           e_free(cli->cli_parent);
           e_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); 
        cli = NULL; 
} 
   
           if ((*cli)->cli_parent)
                   e_free((*cli)->cli_parent);
   
           e_free(*cli);
           *cli = NULL;
   }
   
 /*  /*
 * rpc_cli_execCall() Execute RPC call * rpc_pkt_Send() - Send RPC packet
 * @cli = RPC Client session *
 * @csModule = Module name, if NULL self binary * @sock = Socket
 * @csFunc = Function name for execute * @type = Type of socket
 * @in_argc = IN count of arguments * @sa = Server address
 * @in_vars = IN RPC call array of rpc values * @pkt = RPC packet
 * @out_argc = OUT returned count of arguments * @len = Length of packet
 * @out_vars = OUT returned array of rpc values, must be free after use (see rpc_cli_freeVals()) * return: -1 error or !=-1 sended bytes
 * 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_pkt_Send(int sock, int type, sockaddr_t * __restrict sa, ait_val_t * __restrict pkt, int len)
                ait_val_t * __restrict in_vars, int *out_argc, ait_val_t ** __restrict out_vars) 
 {  {
        fd_set fds;        struct pollfd pfd;
        u_char *buf, str[MAXPATHLEN + UCHAR_MAX + 1], *data;        int ret;
        struct tagRPCCall *rpc;        u_char *buf;
        struct tagRPCRet *rrpc = NULL; 
        int ret = 0, Limit = 0; 
        register int i; 
        ait_val_t *v; 
        struct timeval tv = { DEF_RPC_TIMEOUT, 0 }; 
   
        if (!cli || !csFunc || (in_argc && !in_vars)) {        if (!pkt) {
                rpc_SetErr(EINVAL, "Error:: Can`t execute call because parameter is null or invalid!\n");                rpc_SetErr(EINVAL, "Invalid argument(s)!");
                 return -1;                  return -1;
           } else
                   buf = AIT_GET_BUF(pkt);
   
           pfd.fd = sock;
           pfd.events = POLLOUT;
           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 (type == SOCK_STREAM)
                           ret = send(sock, buf, len, MSG_NOSIGNAL);
                   else if (sa)
                           ret = sendto(sock, buf, len, MSG_NOSIGNAL, &sa->sa, sa->sa.sa_len);
                   else {
                           rpc_SetErr(EINVAL, "Invalid argument(s)!");
                           return -1;
                   }
                   if (ret == -1) {
                           if (errno == EAGAIN)
                                   continue;
                           LOGERR;
                           return -1;
                   } else if (ret != len) {
                           rpc_SetErr(EPROCUNAVAIL, "RPC request, should be send %d bytes, "
                                           "really sended %d bytes", len, ret);
                           return -1;
                   }
           } while (0);
   
        buf = malloc(cli->cli_netbuf);        return ret;
        if (!buf) {}
                LOGERR;
 /*
  * rpc_pkt_Receive() - Receive RPC packet
  *
  * @sock = Socket
  * @type = Type of socket
  * @sa = Server address
  * @pkt = RPC packet
  * return: -1 error or !=-1 sended bytes
  */
 int
 rpc_pkt_Receive(int sock, int type, sockaddr_t * __restrict sa, ait_val_t * __restrict pkt)
 {
         struct pollfd pfd;
         int ret, len = 0;
         u_char *buf;
         sockaddr_t sa2;
         socklen_t salen;
 
         if (!pkt) {
                 rpc_SetErr(EINVAL, "Invalid argument(s)!");
                 return -1;                  return -1;
         } else          } else
                memset(buf, 0, cli->cli_netbuf);                buf = AIT_GET_BUF(pkt);
   
        /* build RPC call string for hash */        /* reply from RPC server */
        memset(str, 0, sizeof str);        pfd.fd = sock;
        if (csModule)        pfd.events = POLLIN | POLLPRI;
                strlcpy((char*) str, csModule, sizeof str);        do {
        strlcat((char*) str, "__", sizeof str);                if ((ret = poll(&pfd, 1, DEF_RPC_TIMEOUT * 1000)) < 1 || 
        strlcat((char*) str, csFunc, sizeof str);                                pfd.revents & (POLLERR | POLLHUP | POLLNVAL)) {
                         if (ret) {
                                 LOGERR;
                         } else {
                                 if (len++ < 7)
                                         continue;
                                 else
                                         rpc_SetErr(ETIMEDOUT, 
                                                         "Timeout, no answer from RPC server");
                         }
   
                           return -1;
                   }
   
                   memset(buf, 0, AIT_LEN(pkt));
                   if (type == SOCK_STREAM)
                           ret = recv(sock, buf, AIT_LEN(pkt), 0);
                   else {
                           memset(&sa2, 0, sizeof sa2);
                           salen = sa2.ss.ss_len = sizeof(sockaddr_t);
                           ret = recvfrom(sock, buf, AIT_LEN(pkt), 0, &sa2.sa, &salen);
                   }
                   if (ret < 1) {
                           if (ret) {
                                   if (errno == EAGAIN)
                                           continue;
                                   else
                                           LOGERR;
                           }
                           return -1;
                   }
   
                   /* check for response from known address */
                   if (type == SOCK_DGRAM)
                           if (e_addrcmp(sa, &sa2, 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 ret;
   }
   
   /*
    * rpc_pkt_Request() - Build RPC Request packet
    *
    * @pkt = Packet buffer
    * @sess = RPC session info
    * @tag = Function tag for execution
    * @vars = Function argument array of values, may be NULL
    * @noreply = We not want RPC reply
    * return: -1 error or != -1 prepared bytes into packet
    */
   int
   rpc_pkt_Request(ait_val_t * __restrict pkt, rpc_sess_t * __restrict sess, u_short tag, 
                   array_t * __restrict vars, int noreply)
   {
           struct tagRPCCall *rpc;
           int ret = 0, len = sizeof(struct tagRPCCall);
           u_char *buf;
   
           if (!pkt || !sess) {
                   rpc_SetErr(EINVAL, "Invalid argument(s)!");
                   return -1;
           } else
                   buf = AIT_GET_BUF(pkt);
   
         /* prepare RPC call */          /* prepare RPC call */
         rpc = (struct tagRPCCall*) buf;          rpc = (struct tagRPCCall*) buf;
        memcpy(&rpc->call_session, cli->cli_parent, sizeof rpc->call_session);        rpc_addPktSession(&rpc->call_session, sess);
        rpc->call_argc = in_argc;        rpc->call_tag = htons(tag);
        rpc->call_tag = crcFletcher16((u_short*) str, sizeof str / 2);        rpc->call_seq = htons(random() % USHRT_MAX);
        rpc->call_hash = hash_fnv((char*) str, sizeof str);        if (!vars)
        Limit = sizeof(struct tagRPCCall);                rpc->call_argc = 0;
         else
                 rpc->call_argc = htons(array_Size(vars));
   
        if (in_argc) {        /* set reply */
         rpc->call_req.flags = noreply ? RPC_NOREPLY : RPC_REPLY;
 
         if (array_Size(vars)) {
                 /* marshaling variables */                  /* marshaling variables */
                v = (ait_val_t*) (buf + sizeof(struct tagRPCCall));                ret = ait_vars2buffer(buf + len, AIT_LEN(pkt) - len, vars);
                if (in_argc * sizeof(ait_val_t) > cli->cli_netbuf - Limit) {                if (ret == -1) {
                        rpc_SetErr(EMSGSIZE, "Error:: in prepare RPC packet values (-7) ...\n");                        rpc_SetErr(EBADRPC, "Failed to prepare RPC packet values");
                        free(buf);                        return -1;
                        return -7; 
                 } else                  } else
                        Limit += in_argc * sizeof(ait_val_t);                        len += ret;
                memcpy(v, in_vars, in_argc * sizeof(ait_val_t));        }
   
                /* adding additional data */        /* total packet length */
                data = (u_char*) v + in_argc * sizeof(ait_val_t);        rpc->call_len = htons(len);
                for (i = 0; i < in_argc && !ret; i++) { 
                        switch (AIT_TYPE(&in_vars[i])) { 
                                case buffer: 
                                        if (Limit + AIT_LEN(&in_vars[i]) > cli->cli_netbuf) { 
                                                ret = -7; 
                                                break; 
                                        } 
   
                                        memcpy(data, in_vars[i].val.buffer, AIT_LEN(&in_vars[i]));        /* calculate CRC */
                                        v[i].val.buffer = (uint8_t*) ((void*) data - (void*) v);        rpc->call_crc ^= rpc->call_crc;
                                        data += AIT_LEN(&in_vars[i]);        rpc->call_crc = htons(crcFletcher16((u_short*) buf, len / 2));
                                        Limit += AIT_LEN(&in_vars[i]); 
                                        break; 
                                case string: 
                                        if (Limit + AIT_LEN(&in_vars[i]) > cli->cli_netbuf) { 
                                                ret = -7; 
                                                break; 
                                        } 
   
                                        memcpy(data, in_vars[i].val.string, AIT_LEN(&in_vars[i]));        return len;
                                        v[i].val.string = (int8_t*) ((void*) data - (void*) v);}
                                        data += AIT_LEN(&in_vars[i]);
                                        Limit += AIT_LEN(&in_vars[i]);/*
                                        break; * rpc_pkt_Replay() - Decode RPC Replay packet
                                default: *
                                        break; * @pkt = Packet buffer
                        } * @sess = RPC session info
                } * @tag = Function tag
                if (ret < 0) { * @vars = Function argument array of values, may be NULL
                        rpc_SetErr(EMSGSIZE, "Error:: in prepare RPC packet (-7) ...\n"); * return: -1 error or != -1 return value from function
                        free(buf); */
                        return ret;int
                }rpc_pkt_Replay(ait_val_t * __restrict pkt, rpc_sess_t * __restrict sess, u_short tag, 
                 array_t ** __restrict vars)
 {
         struct tagRPCCall *rpc;
         int len;
         u_char *buf;
         uint16_t crc;
 
         if (!pkt || !sess) {
                 rpc_SetErr(EINVAL, "Invalid argument(s)!");
                 return -1;
         } else
                 buf = AIT_GET_BUF(pkt);
 
         rpc = (struct tagRPCCall*) buf;
         /* calculate CRC */
         crc = ntohs(rpc->call_crc);
         rpc->call_crc ^= rpc->call_crc;
         if (crc != crcFletcher16((u_short*) buf, ntohs(rpc->call_len) / 2)) {
                 rpc_SetErr(ERPCMISMATCH, "Bad CRC RPC packet");
                 return -1;
         }          }
   
        if ((ret = send(cli->cli_sock, buf, Limit, 0)) == -1) {        /* check RPC packet session info */
                LOGERR;        if (rpc_chkPktSession(&rpc->call_session, sess)) {
                free(buf);                rpc_SetErr(ERPCMISMATCH, "Get invalid RPC session");
                 return -1;                  return -1;
         }          }
        if (ret != Limit) {        if (ntohs(rpc->call_tag) != tag) {
                rpc_SetErr(ECANCELED, "Error:: in send RPC request, should be send %d bytes, really is %d\n",                 rpc_SetErr(ERPCMISMATCH, "Get wrong RPC reply");
                                Limit, ret);                return -1;
                free(buf); 
                return -9; 
         }          }
        if (ntohl(rpc->call_rep.eno) && ntohl(rpc->call_rep.ret) == -1) {
        /* reply from RPC server */                rpc_SetErr(ntohl(rpc->call_rep.eno), "Server side: retcode=%d #%d %s", 
        FD_ZERO(&fds);                                ntohl(rpc->call_rep.ret), ntohl(rpc->call_rep.eno), 
        FD_SET(cli->cli_sock, &fds);                                strerror(ntohl(rpc->call_rep.eno)));
        if ((ret = select(cli->cli_sock + 1, &fds, NULL, NULL, &tv)) < 1) { 
                if (ret) 
                        LOGERR; 
                else 
                        rpc_SetErr(ETIMEDOUT, "Error:: timeout, no return from RPC server?\n"); 
 
                free(buf); 
                 return -1;                  return -1;
         }          }
        memset(buf, 0, cli->cli_netbuf);        len = ntohs(rpc->call_argc) * sizeof(ait_val_t);
        if ((ret = recv(cli->cli_sock, buf, cli->cli_netbuf, 0)) == -1) {        if (len > AIT_LEN(pkt) - sizeof(struct tagRPCCall)) {
                LOGERR;                rpc_SetErr(EMSGSIZE, "Reply RPC packet not enough buffer space ...");
                free(buf);                return -1;
                return -3; 
         }          }
        if (!ret) { /* receive EOF! */        if (len > ntohs(rpc->call_len) - sizeof(struct tagRPCCall)) {
                free(buf);                rpc_SetErr(EMSGSIZE, "Reply RPC packet is too short ...");
                return 0;                return -1;
         }          }
        if (ret < sizeof(struct tagRPCRet)) {
                rpc_SetErr(EMSGSIZE, "Error:: too short RPC packet ...\n");        /* RPC is OK! Go de-marshaling variables ... */
                free(buf);        if (vars && ntohs(rpc->call_argc)) {
                return -4;#ifdef CLI_RES_ZCOPY
        } else                *vars = ait_buffer2vars(buf + sizeof(struct tagRPCCall), len, 
                rrpc = (struct tagRPCRet*) buf;                                ntohs(rpc->call_argc), 42);
        /* check RPC packet session info */#else
        if (memcmp(&rrpc->ret_session, cli->cli_parent, sizeof rrpc->ret_session)) {                *vars = ait_buffer2vars(buf + sizeof(struct tagRPCCall), len, 
                rpc_SetErr(EINVAL, "Error:: get invalid RPC session ...\n");                                ntohs(rpc->call_argc), 0);
                free(buf);#endif
                return -5;                if (!*vars) {
        } else                        rpc_SetErr(elwix_GetErrno(), "%s", elwix_GetError());
                Limit = sizeof(struct tagRPCRet);                        return -1;
        if (rrpc->ret_retcode < 0 && rrpc->ret_errno) {                }
                rpc_SetErr(rrpc->ret_errno, "Error::Server side: %d %s\n",  
                                rrpc->ret_retcode, strerror(rrpc->ret_errno)); 
                free(buf); 
                return -6; 
         }          }
        if (rrpc->ret_argc * sizeof(ait_val_t) > cli->cli_netbuf - Limit) {
                rpc_SetErr(EMSGSIZE, "Error:: reply RPC packet is too long ...\n");        return ntohl(rpc->call_rep.ret);
                free(buf);}
                return -7;
 /*
  * rpc_cli_execCall() - Execute RPC call
  *
  * @cli = RPC Client session
  * @noreply = We not want RPC reply
  * @tag = Function tag for execution
  * @in_vars = IN function argument array of values, may be NULL
  * @out_vars = OUT returned array of rpc values, if !=NULL must be free after use with ait_freeVars()
  * return: -1 error or != -1 ok result
  */
 int
 rpc_cli_execCall(rpc_cli_t *cli, int noreply, u_short tag, 
                 array_t * __restrict in_vars, array_t ** __restrict out_vars)
 {
         int ret = 0, wlen;
         u_char *buf;
 
         if (!cli) {
                 rpc_SetErr(EINVAL, "Can`t execute call because parameter is null or invalid!");
                 return -1;
         } else          } else
                Limit += rrpc->ret_argc * sizeof(ait_val_t);                buf = AIT_GET_BUF(&cli->cli_buf);
        /* RPC is OK! Go decapsulate variables ... */        if (out_vars)
        if (rrpc->ret_argc) {                *out_vars = NULL;
                *out_argc = rrpc->ret_argc; 
                *out_vars = calloc(rrpc->ret_argc, sizeof(ait_val_t)); 
                if (!*out_vars) { 
                        LOGERR; 
                        *out_argc = 0; 
                        free(buf); 
                        return -1; 
                } else 
                        memcpy(*out_vars, buf + sizeof(struct tagRPCRet), Limit - sizeof(struct tagRPCRet)); 
   
                /* RPC received variables types OK! */        if ((wlen = rpc_pkt_Request(&cli->cli_buf, cli->cli_parent, tag, in_vars, noreply)) == -1)
                data = (u_char*) buf + Limit;                return -1;
                for (i = 0; i < rrpc->ret_argc; i++) 
                        switch (AIT_TYPE(&(*out_vars)[i])) { 
                                case buffer: 
                                        if (AIT_LEN(&(*out_vars)[i]) > cli->cli_netbuf - Limit) { 
                                                rpc_SetErr(EMSGSIZE, "Error:: Too long RPC packet ...\n"); 
                                                free(*out_vars); 
                                                *out_vars = NULL; 
                                                *out_argc = 0; 
                                                free(buf); 
                                                return -7; 
                                        } else 
                                                Limit += AIT_LEN(&(*out_vars)[i]); 
   
                                        (*out_vars)[i].val.buffer = malloc(AIT_LEN(&(*out_vars)[i]));        if (rpc_pkt_Send(cli->cli_sock, cli->cli_id, &cli->cli_sa, &cli->cli_buf, wlen) == -1)
                                        if (!(*out_vars)[i].val.buffer) {                return -1;
                                                rpc_SetErr(errno, "Error:: in prepare RPC reply ...\n"); 
                                                free(*out_vars); 
                                                *out_vars = NULL; 
                                                *out_argc = 0; 
                                                free(buf); 
                                                return -1; 
                                        } else 
                                                memcpy((*out_vars)[i].val.buffer, data, AIT_LEN(&(*out_vars)[i])); 
                                        data += AIT_LEN(&(*out_vars)[i]); 
                                        break; 
                                case string: 
                                        if (AIT_LEN(&(*out_vars)[i]) > cli->cli_netbuf - Limit) { 
                                                rpc_SetErr(EMSGSIZE, "Error:: Too long RPC packet ...\n"); 
                                                free(*out_vars); 
                                                *out_vars = NULL; 
                                                *out_argc = 0; 
                                                free(buf); 
                                                return -7; 
                                        } else 
                                                Limit += AIT_LEN(&(*out_vars)[i]); 
   
                                        (*out_vars)[i].val.string = malloc(AIT_LEN(&(*out_vars)[i]));        if (noreply)        /* we not want reply */
                                        if (!(*out_vars)[i].val.string) {                return 0;
                                                rpc_SetErr(errno, "Error:: in prepare RPC reply ...\n"); 
                                                free(*out_vars); 
                                                *out_vars = NULL; 
                                                *out_argc = 0; 
                                                free(buf); 
                                                return -1; 
                                        } else 
                                                memcpy((*out_vars)[i].val.string, data, AIT_LEN(&(*out_vars)[i])); 
                                        data += AIT_LEN(&(*out_vars)[i]); 
                                        break; 
                                default: 
                                        break; 
                        } 
        } else { 
                if (out_argc) 
                        *out_argc = 0; 
                if (out_vars) 
                        *out_vars = NULL; 
        } 
   
        ret = rrpc->ret_retcode;        if (rpc_pkt_Receive(cli->cli_sock, cli->cli_id, &cli->cli_sa, &cli->cli_buf) == -1)
        free(buf);                return -1;
 
         if ((wlen = rpc_pkt_Replay(&cli->cli_buf, cli->cli_parent, tag, out_vars)) == -1)
                 return -1;
 
         return ret;          return ret;
 }  }
   
 /*  /*
 * rpc_cli_freeVals() Free ait_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, ait_val_t * __restrict vars)rpc_cli_freeCall(array_t ** __restrict out_vars)
 {  {
        register int i;#ifdef CLI_RES_ZCOPY
         array_Destroy(out_vars);
 #else
         ait_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)
                AIT_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(array(arr, 0, ait_val_t*));
         rpc_cli_freeCall(&arr);
 
         return ret;
 }  }

Removed from v.1.4.2.3  
changed lines
  Added in v.1.13.4.4


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