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

version 1.4.2.3, 2011/08/30 12:59:02 version 1.9.2.18, 2012/05/18 23:03:00
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;  
         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 (!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 = malloc(sizeof(rpc_cli_t));
         if (!cli) {          if (!cli) {
Line 78  rpc_cli_openBLOBClient(rpc_cli_t * __restrict rpccli,  Line 70  rpc_cli_openBLOBClient(rpc_cli_t * __restrict rpccli, 
         } 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);
   
         /* 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);                  free(cli);
                 return NULL;                  return NULL;
         }          }
        n = cli->cli_netbuf;        if (connect(cli->cli_sock, &cli->cli_sa.sa, cli->cli_sa.sa.sa_len) == -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);  
                 free(cli);                  free(cli);
                 return NULL;                  return NULL;
        }        } else
        if (setsockopt(cli->cli_sock, SOL_SOCKET, SO_RCVBUF, &n, sizeof n) == -1) {                fcntl(cli->cli_sock, F_SETFL, fcntl(cli->cli_sock, F_GETFL) | O_NONBLOCK);
                LOGERR; 
                close(cli->cli_sock); 
                free(cli->cli_parent); 
                free(cli); 
                return NULL; 
        } 
        if (connect(cli->cli_sock, &cli->cli_sa, sizeof cli->cli_sa) == -1) { 
                LOGERR; 
                free(cli); 
                return NULL; 
        } 
   
         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;
         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
  * 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)
 {  {
         rpc_cli_t *cli = NULL;          rpc_cli_t *cli = NULL;
        struct hostent *host = NULL;        io_sockaddr_t sa;
        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 (!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 (!netBuf)        if (netBuf < RPC_MIN_BUFSIZ)
                 netBuf = BUFSIZ;                  netBuf = BUFSIZ;
        if (csHost && family != AF_LOCAL) {        else
                host = gethostbyname2(csHost, family);                netBuf = io_align(netBuf, 1);   /* 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; 
        } 
   
   #ifdef HAVE_SRANDOMDEV
           srandomdev();
   #else
           time_t tim;
   
           srandom((time(&tim) ^ getpid()));
   #endif
   
         cli = malloc(sizeof(rpc_cli_t));          cli = malloc(sizeof(rpc_cli_t));
         if (!cli) {          if (!cli) {
                 LOGERR;                  LOGERR;
Line 215  rpc_cli_openClient(u_int ProgID, u_int ProcID, int net Line 171  rpc_cli_openClient(u_int ProgID, u_int ProcID, int net
         } 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 = malloc(sizeof(rpc_sess_t));
         if (!cli->cli_parent) {          if (!cli->cli_parent) {
                 LOGERR;                  LOGERR;
Line 227  rpc_cli_openClient(u_int ProgID, u_int ProcID, int net Line 183  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) {        memcpy(&cli->cli_sa, &sa, sizeof cli->cli_sa);
                case AF_INET:        AIT_SET_BUF2(&cli->cli_buf, 0, netBuf);
                        memcpy(&cli->cli_sa, sin, sizeof cli->cli_sa); 
                        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, SOCK_STREAM, 0);
         if (cli->cli_sock == -1) {          if (cli->cli_sock == -1) {
                 LOGERR;                  LOGERR;
                   AIT_FREE_VAL(&cli->cli_buf);
                 free(cli->cli_parent);                  free(cli->cli_parent);
                 free(cli);                  free(cli);
                 return NULL;                  return NULL;
         }          }
        n = cli->cli_netbuf;        if (connect(cli->cli_sock, &cli->cli_sa.sa, cli->cli_sa.sa.sa_len) == -1) {
        if (setsockopt(cli->cli_sock, SOL_SOCKET, SO_SNDBUF, &n, sizeof n) == -1) { 
                 LOGERR;                  LOGERR;
                   AIT_FREE_VAL(&cli->cli_buf);
                 close(cli->cli_sock);                  close(cli->cli_sock);
                 free(cli->cli_parent);                  free(cli->cli_parent);
                 free(cli);                  free(cli);
                 return NULL;                  return NULL;
        }        } else
        if (setsockopt(cli->cli_sock, SOL_SOCKET, SO_RCVBUF, &n, sizeof n) == -1) {                fcntl(cli->cli_sock, F_SETFL, fcntl(cli->cli_sock, F_GETFL) | O_NONBLOCK);
                LOGERR; 
                close(cli->cli_sock); 
                free(cli->cli_parent); 
                free(cli); 
                return NULL; 
        } 
        if (connect(cli->cli_sock, &cli->cli_sa, sizeof cli->cli_sa) == -1) { 
                LOGERR; 
                close(cli->cli_sock); 
                free(cli->cli_parent); 
                free(cli); 
                return NULL; 
        } 
   
         return cli;          return cli;
 }  }
   
 /*  /*
 * 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);        shutdown((*cli)->cli_sock, SHUT_RDWR);
        close(cli->cli_sock);        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;                free((*cli)->cli_parent);
 
         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, 
                ait_val_t * __restrict in_vars, int *out_argc, ait_val_t ** __restrict out_vars)                array_t * __restrict in_vars, array_t ** __restrict out_vars)
 {  {
         fd_set fds;  
         u_char *buf, str[MAXPATHLEN + UCHAR_MAX + 1], *data;  
         struct tagRPCCall *rpc;          struct tagRPCCall *rpc;
        struct tagRPCRet *rrpc = NULL;        int ret = 0, wlen = sizeof(struct tagRPCCall);
        int ret = 0, Limit = 0;        uint16_t crc;
        register int i;        u_char *buf;
        ait_val_t *v;        struct pollfd pfd;
        struct timeval tv = { DEF_RPC_TIMEOUT, 0 }; 
   
        if (!cli || !csFunc || (in_argc && !in_vars)) {        if (!cli) {
                rpc_SetErr(EINVAL, "Error:: Can`t execute call because parameter is null or invalid!\n");                rpc_SetErr(EINVAL, "Can`t execute call because parameter is null or invalid!");
                 return -1;                  return -1;
         }  
   
         buf = malloc(cli->cli_netbuf);  
         if (!buf) {  
                 LOGERR;  
                 return -1;  
         } else          } else
                memset(buf, 0, cli->cli_netbuf);                buf = AIT_GET_BUF(&cli->cli_buf);
         if (out_vars)
                 *out_vars = NULL;
   
         /* build RPC call string for hash */  
         memset(str, 0, sizeof str);  
         if (csModule)  
                 strlcpy((char*) str, csModule, sizeof str);  
         strlcat((char*) str, "__", sizeof str);  
         strlcat((char*) str, csFunc, sizeof str);  
   
         /* 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, cli->cli_parent);
        rpc->call_argc = in_argc;        rpc->call_argc = htons(io_arraySize(in_vars));
        rpc->call_tag = crcFletcher16((u_short*) str, sizeof str / 2);        rpc->call_tag = htons(tag);
        rpc->call_hash = hash_fnv((char*) str, sizeof str);        rpc->call_seq = htons(random() % USHRT_MAX);
        Limit = sizeof(struct tagRPCCall); 
   
        if (in_argc) {        /* set reply */
         rpc->call_req.flags = noreply ? RPC_NOREPLY : RPC_REPLY;
 
         if (io_arraySize(in_vars)) {
                 /* marshaling variables */                  /* marshaling variables */
                v = (ait_val_t*) (buf + sizeof(struct tagRPCCall));                ret = io_vars2buffer(buf + wlen, AIT_LEN(&cli->cli_buf) - wlen, in_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);                        wlen += 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(wlen);
                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, wlen / 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]));        pfd.fd = cli->cli_sock;
                                        v[i].val.string = (int8_t*) ((void*) data - (void*) v);        pfd.events = POLLOUT;
                                        data += AIT_LEN(&in_vars[i]);        if ((ret = poll(&pfd, 1, DEF_RPC_TIMEOUT * 1000)) < 1 || 
                                        Limit += AIT_LEN(&in_vars[i]);                        pfd.revents & (POLLERR | POLLHUP | POLLNVAL)) {
                                        break;                if (ret)
                                default:                        LOGERR;
                                        break;                else
                         rpc_SetErr(ETIMEDOUT, "Timeout, can't send to RPC server");
                 return -1;
         }
         do {
                 if ((ret = send(cli->cli_sock, buf, wlen, MSG_NOSIGNAL)) == -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) {        } while (0);
                        rpc_SetErr(EMSGSIZE, "Error:: in prepare RPC packet (-7) ...\n");        do {
                        free(buf);                memset(buf, 0, AIT_LEN(&cli->cli_buf));
                        return ret;                if ((ret = recv(cli->cli_sock, buf, AIT_LEN(&cli->cli_buf), 0)) < 1) {
                         if (ret) {
                                 if (errno == EAGAIN)
                                         continue;
                                 else
                                         LOGERR;
                         }
                         return -1;
                 }                  }
           } while (0);
           if (ret < sizeof(struct tagRPCCall)) {
                   rpc_SetErr(ERPCMISMATCH, "Short RPC packet %d bytes", ret);
                   return -1;
         }          }
   
        if ((ret = send(cli->cli_sock, buf, Limit, 0)) == -1) {        /* calculate CRC */
                LOGERR;        crc = ntohs(rpc->call_crc);
                free(buf);        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;                  return -1;
         }          }
         if (ret != Limit) {  
                 rpc_SetErr(ECANCELED, "Error:: in send RPC request, should be send %d bytes, really is %d\n",   
                                 Limit, ret);  
                 free(buf);  
                 return -9;  
         }  
   
        /* reply from RPC server */        /* check RPC packet session info */
        FD_ZERO(&fds);        if (rpc_chkPktSession(&rpc->call_session, cli->cli_parent)) {
        FD_SET(cli->cli_sock, &fds);                rpc_SetErr(ERPCMISMATCH, "Get invalid RPC session");
        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;
           } else
                   wlen = sizeof(struct tagRPCCall);
           if (ntohs(rpc->call_tag) != tag) {
                   rpc_SetErr(ERPCMISMATCH, "Get wrong RPC reply");
                   return -1;
         }          }
        memset(buf, 0, cli->cli_netbuf);        if (ntohl(rpc->call_rep.eno) && ntohl(rpc->call_rep.ret) == -1) {
        if ((ret = recv(cli->cli_sock, buf, cli->cli_netbuf, 0)) == -1) {                rpc_SetErr(ntohl(rpc->call_rep.eno), "Server side: retcode=%d #%d %s", 
                LOGERR;                                ntohl(rpc->call_rep.ret), ntohl(rpc->call_rep.eno), 
                free(buf);                                strerror(ntohl(rpc->call_rep.eno)));
                return -3;                return -1;
         }          }
        if (!ret) { /* receive EOF! */        if (ntohs(rpc->call_argc) * sizeof(ait_val_t) > AIT_LEN(&cli->cli_buf) - wlen) {
                free(buf);                rpc_SetErr(EMSGSIZE, "Reply RPC packet is too long ...");
                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 (out_vars && ntohs(rpc->call_argc)) {
                return -4;                *out_vars = io_buffer2vars(buf + wlen, AIT_LEN(&cli->cli_buf) - wlen, 
        } else                                ntohs(rpc->call_argc), 0);
                rrpc = (struct tagRPCRet*) buf; 
        /* check RPC packet session info */ 
        if (memcmp(&rrpc->ret_session, cli->cli_parent, sizeof rrpc->ret_session)) { 
                rpc_SetErr(EINVAL, "Error:: get invalid RPC session ...\n"); 
                free(buf); 
                return -5; 
        } else 
                Limit = sizeof(struct tagRPCRet); 
        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"); 
                free(buf); 
                return -7; 
        } else 
                Limit += rrpc->ret_argc * sizeof(ait_val_t); 
        /* RPC is OK! Go decapsulate variables ... */ 
        if (rrpc->ret_argc) { 
                *out_argc = rrpc->ret_argc; 
                *out_vars = calloc(rrpc->ret_argc, sizeof(ait_val_t)); 
                 if (!*out_vars) {                  if (!*out_vars) {
                        LOGERR;                        rpc_SetErr(io_GetErrno(), "%s", io_GetError());
                        *out_argc = 0; 
                        free(buf); 
                         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 (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 (!(*out_vars)[i].val.buffer) { 
                                                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 (!(*out_vars)[i].val.string) { 
                                                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;        ret = ntohl(rpc->call_rep.ret);
        free(buf); 
         return ret;          return ret;
 }  }
   
 /*  /*
 * rpc_cli_freeVals() Free ait_val_t array returned from RPC call * rpc_cli_ping() - Ping RPC server
 * @args = Number of arguments in array *
 * @vars = Value elements * @cli = connected client
 * return: none * return: -1 error or !=-1 ping seq id
  */   */
inline voidinline int
rpc_cli_freeVals(int args, ait_val_t * __restrict vars)rpc_cli_ping(rpc_cli_t *cli)
 {  {
        register int i;        int ret = 0;
         array_t *arr = NULL;
   
        if (!vars)        if (!cli)
                return;                return -1;
   
        for (i = 0; i < args; i++)        if (rpc_cli_execCall(cli, RPC_REPLY, CALL_SRVPING, NULL, &arr))
                AIT_FREE_VAL(&vars[i]);                return -1;
         else
                 ret = AIT_GET_U16(io_array(arr, 0, ait_val_t*));
         io_freeVars(&arr);
   
        free(vars);        return ret;
        vars = NULL; 
 }  }

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


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