Diff for /libaitrpc/src/cli.c between versions 1.6.2.1 and 1.9.2.3

version 1.6.2.1, 2012/03/13 17:10:13 version 1.9.2.3, 2012/05/15 16:06:13
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 59  rpc_cli_openBLOBClient(rpc_cli_t * __restrict rpccli,  Line 60  rpc_cli_openBLOBClient(rpc_cli_t * __restrict rpccli, 
         io_sockaddr_t sa;          io_sockaddr_t sa;
         int n;          int n;
   
        if (!rpccli ||         if (!rpccli) {
                        (rpccli->cli_sa.sa.sa_family != AF_INET && rpccli->cli_sa.sa.sa_family != AF_INET6 &&                 rpc_SetErr(EINVAL, "Invalid parameters can`t connect to BLOB server");
                         rpccli->cli_sa.sa.sa_family != AF_LOCAL)) { 
                rpc_SetErr(EINVAL, "Error:: Invalid parameters can`t connect to BLOB server ...\n"); 
                 return NULL;                  return NULL;
         }          }
   
Line 84  rpc_cli_openBLOBClient(rpc_cli_t * __restrict rpccli,  Line 83  rpc_cli_openBLOBClient(rpc_cli_t * __restrict rpccli, 
                 case AF_LOCAL:                  case AF_LOCAL:
                         strlcat(sa.sun.sun_path, ".blob", sizeof sa.sun.sun_path);                          strlcat(sa.sun.sun_path, ".blob", sizeof sa.sun.sun_path);
                         break;                          break;
                   default:
                           rpc_SetErr(EINVAL, "Invalid socket type %d", sa.sa.sa_family);
                           return NULL;
         }          }
         memcpy(&cli->cli_sa, &sa, sizeof sa);          memcpy(&cli->cli_sa, &sa, sizeof sa);
   
Line 118  rpc_cli_openBLOBClient(rpc_cli_t * __restrict rpccli,  Line 120  rpc_cli_openBLOBClient(rpc_cli_t * __restrict rpccli, 
 }  }
   
 /*  /*
 * 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
  */   */
Line 126  void Line 129  void
 rpc_cli_closeBLOBClient(rpc_cli_t * __restrict cli)  rpc_cli_closeBLOBClient(rpc_cli_t * __restrict cli)
 {  {
         if (!cli) {          if (!cli) {
                rpc_SetErr(EINVAL, "Error:: Can`t close connection because parameter is null!\n");                rpc_SetErr(EINVAL, "Can`t close connection because parameter is null!");
                 return;                  return;
         }          }
   
Line 140  rpc_cli_closeBLOBClient(rpc_cli_t * __restrict cli) Line 143  rpc_cli_closeBLOBClient(rpc_cli_t * __restrict cli)
 // --------------------------------------------------------------  // --------------------------------------------------------------
   
 /*  /*
 * 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, if =0 == BUFSIZ (also meaning max RPC packet)
    * @Timeout = RPC timeout in seconds, if =0 set default RPC timeout
  * @family = Family socket type, AF_INET or AF_INET6   * @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_int ProcID, int netBuf, u_char Timeout, 
                 u_short family, const char *csHost, u_short Port)
 {  {
         rpc_cli_t *cli = NULL;          rpc_cli_t *cli = NULL;
         struct hostent *host = NULL;  
         io_sockaddr_t sa;          io_sockaddr_t sa;
         int n;          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)
                 netBuf = BUFSIZ;                  netBuf = BUFSIZ;
        if (csHost && family != AF_LOCAL) {        if (!Timeout)
                host = gethostbyname2(csHost, family);                Timeout = DEF_RPC_TIMEOUT;
                if (!host) { 
                        rpc_SetErr(h_errno, "Error:: %s\n", hstrerror(h_errno)); 
                        return NULL; 
                } 
        } 
        memset(&sa, 0, sizeof sa); 
        sa.sa.sa_family = family; 
        switch (family) { 
                case AF_INET: 
                        sa.sin.sin_len = sizeof(struct sockaddr_in); 
                        sa.sin.sin_port = htons(Port); 
                        if (csHost) 
                                memcpy(&sa.sin.sin_addr, host->h_addr, host->h_length); 
                        break; 
                case AF_INET6: 
                        sa.sin6.sin6_len = sizeof(struct sockaddr_in6); 
                        sa.sin6.sin6_port = htons(Port); 
                        if (csHost) 
                                memcpy(&sa.sin6.sin6_addr, host->h_addr, host->h_length); 
                        break; 
                case AF_LOCAL: 
                        sa.sun.sun_len = sizeof(struct sockaddr_un); 
                        if (csHost) 
                                strlcpy(sa.sun.sun_path, csHost, sizeof sa.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));          cli = malloc(sizeof(rpc_cli_t));
         if (!cli) {          if (!cli) {
Line 212  rpc_cli_openClient(u_int ProgID, u_int ProcID, int net Line 186  rpc_cli_openClient(u_int ProgID, u_int ProcID, int net
                 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;
                   ((rpc_sess_t*) cli->cli_parent)->sess_timeout = Timeout;
                 ((rpc_sess_t*) cli->cli_parent)->sess_program = ProgID;                  ((rpc_sess_t*) cli->cli_parent)->sess_program = ProgID;
                 ((rpc_sess_t*) cli->cli_parent)->sess_process = ProcID;                  ((rpc_sess_t*) cli->cli_parent)->sess_process = ProcID;
         }          }
Line 219  rpc_cli_openClient(u_int ProgID, u_int ProcID, int net Line 194  rpc_cli_openClient(u_int ProgID, u_int ProcID, int net
         memcpy(&cli->cli_sa, &sa, sizeof cli->cli_sa);          memcpy(&cli->cli_sa, &sa, sizeof cli->cli_sa);
   
         /* 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;
                 free(cli->cli_parent);                  free(cli->cli_parent);
Line 253  rpc_cli_openClient(u_int ProgID, u_int ProcID, int net Line 228  rpc_cli_openClient(u_int ProgID, u_int ProcID, int net
 }  }
   
 /*  /*
 * 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
  */   */
Line 261  void Line 237  void
 rpc_cli_closeClient(rpc_cli_t * __restrict cli)  rpc_cli_closeClient(rpc_cli_t * __restrict cli)
 {  {
         if (!cli) {          if (!cli) {
                rpc_SetErr(EINVAL, "Error:: Can`t close connection because parameter is null!\n");                rpc_SetErr(EINVAL, "Can`t close connection because parameter is null!");
                 return;                  return;
         }          }
   
Line 276  rpc_cli_closeClient(rpc_cli_t * __restrict cli) Line 252  rpc_cli_closeClient(rpc_cli_t * __restrict cli)
   
   
 /*  /*
 * rpc_cli_execCall() Execute RPC call * rpc_cli_execCall() - Execute RPC call
  *
  * @cli = RPC Client session   * @cli = RPC Client session
    * @noreply = We not want RPC reply
  * @csModule = Module name, if NULL self binary   * @csModule = Module name, if NULL self binary
  * @csFunc = Function name for execute   * @csFunc = Function name for execute
  * @in_vars = IN RPC call array of rpc values   * @in_vars = IN RPC call array of rpc values
Line 285  rpc_cli_closeClient(rpc_cli_t * __restrict cli) Line 263  rpc_cli_closeClient(rpc_cli_t * __restrict cli)
  * 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, rpc_cli_execCall(rpc_cli_t *cli, int noreply, const char *csModule, const char *csFunc, 
                 array_t * __restrict in_vars, array_t ** __restrict out_vars)                  array_t * __restrict in_vars, array_t ** __restrict out_vars)
 {  {
        fd_set fds;        u_char *buf;
        u_char *buf, str[MAXPATHLEN + UCHAR_MAX + 1];        rpc_func_t func;
         struct tagRPCCall *rpc;          struct tagRPCCall *rpc;
        struct tagRPCCall *rrpc = NULL;        int ret = 0, wlen = sizeof(struct tagRPCCall);
        int ret = 0, Limit = 0;        uint16_t tag, crc;
        struct timeval tv = { DEF_RPC_TIMEOUT, 0 }; 
        uint16_t tag; 
         uint32_t hash;          uint32_t hash;
           struct pollfd pfd;
   
         if (!cli || !csFunc) {          if (!cli || !csFunc) {
                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;
         }          }
         if (out_vars)          if (out_vars)
Line 311  rpc_cli_execCall(rpc_cli_t *cli, const char *csModule, Line 288  rpc_cli_execCall(rpc_cli_t *cli, const char *csModule,
         } else          } else
                 memset(buf, 0, cli->cli_netbuf);                  memset(buf, 0, cli->cli_netbuf);
   
        /* build RPC call string for hash */        /* calculate hashes */
        memset(str, 0, sizeof str);        if (rpc_calcHashes(&func, csModule, csFunc) == -1)
        if (csModule)                return -1;
                strlcpy((char*) str, csModule, sizeof str);        else {
        strlcat((char*) str, "__", sizeof str);                tag = htons(func.func_tag);
        strlcat((char*) str, csFunc, sizeof str);                hash = htonl(func.func_hash);
                 AIT_FREE_VAL(&func.func_name);
                 AIT_FREE_VAL(&func.func_file);
         }
   
         /* prepare RPC call */          /* prepare RPC call */
         rpc = (struct tagRPCCall*) buf;          rpc = (struct tagRPCCall*) buf;
         rpc_addPktSession(&rpc->call_session, cli->cli_parent);          rpc_addPktSession(&rpc->call_session, cli->cli_parent);
         rpc->call_argc = htons(io_arraySize(in_vars));          rpc->call_argc = htons(io_arraySize(in_vars));
        rpc->call_tag = tag = htons(crcFletcher16((u_short*) str, sizeof str / 2));        rpc->call_tag = tag;
        rpc->call_hash = hash = htonl(hash_fnv((char*) str, sizeof str));        rpc->call_hash = hash;
        Limit = sizeof(struct tagRPCCall); 
   
           /* set reply */
           rpc->call_req.flags = noreply ? RPC_NOREPLY : RPC_REPLY;
   
         if (io_arraySize(in_vars)) {          if (io_arraySize(in_vars)) {
                 /* marshaling variables */                  /* marshaling variables */
                ret = io_vars2buffer(buf + Limit, cli->cli_netbuf - Limit, in_vars);                ret = io_vars2buffer(buf + wlen, cli->cli_netbuf - wlen, in_vars);
                 if (ret == -1) {                  if (ret == -1) {
                        rpc_SetErr(EBADRPC, "Error:: in prepare RPC packet values (-7) ...\n");                        rpc_SetErr(EBADRPC, "Failed to prepare RPC packet values");
                         free(buf);                          free(buf);
                        return -7;                        return -1;
                 } else                  } else
                        Limit += ret;                        wlen += ret;
         }          }
   
        if ((ret = send(cli->cli_sock, buf, Limit, 0)) == -1) {        rpc->call_len = htons(wlen);
 
         /* calculate CRC */
         rpc->call_crc ^= rpc->call_crc;
         rpc->call_crc = htons(crcFletcher16((u_short*) buf, wlen / 2));
 
         if ((ret = send(cli->cli_sock, buf, wlen, 0)) == -1) {
                 LOGERR;                  LOGERR;
                 free(buf);                  free(buf);
                 return -1;                  return -1;
           } else if (ret != wlen) {
                   rpc_SetErr(EPROCUNAVAIL, "RPC request, should be send %d bytes, "
                                   "really sended %d bytes", wlen, ret);
                   free(buf);
                   return -1;
         }          }
        if (ret != Limit) {
                rpc_SetErr(EPROCUNAVAIL, "Error:: in send RPC request, should be send %d bytes, really is %d\n",         if (noreply) {
                                Limit, ret);                /* we not want reply */
                 free(buf);                  free(buf);
                return -9;                return 0;
         }          }
   
         /* reply from RPC server */          /* reply from RPC server */
        FD_ZERO(&fds);        pfd.fd = cli->cli_sock;
        FD_SET(cli->cli_sock, &fds);        pfd.events = POLLIN | POLLPRI;
        if ((ret = select(cli->cli_sock + 1, &fds, NULL, NULL, &tv)) < 1) {        if ((ret = poll(&pfd, 1, DEF_RPC_TIMEOUT * 1000)) == -1 || 
                         pfd.revents & (POLLERR | POLLHUP | POLLNVAL)) {
                 if (ret)                  if (ret)
                         LOGERR;                          LOGERR;
                 else                  else
                        rpc_SetErr(ETIMEDOUT, "Error:: timeout, no return from RPC server?\n");                        rpc_SetErr(ETIMEDOUT, "Timeout, no return from RPC server?");
   
                 free(buf);                  free(buf);
                 return -1;                  return -1;
Line 365  rpc_cli_execCall(rpc_cli_t *cli, const char *csModule, Line 359  rpc_cli_execCall(rpc_cli_t *cli, const char *csModule,
         if ((ret = recv(cli->cli_sock, buf, cli->cli_netbuf, 0)) == -1) {          if ((ret = recv(cli->cli_sock, buf, cli->cli_netbuf, 0)) == -1) {
                 LOGERR;                  LOGERR;
                 free(buf);                  free(buf);
                return -3;                return -1;
         }          }
         if (!ret) {     /* receive EOF! */          if (!ret) {     /* receive EOF! */
                 free(buf);                  free(buf);
                 return 0;                  return 0;
         }          }
         if (ret < sizeof(struct tagRPCCall)) {          if (ret < sizeof(struct tagRPCCall)) {
                rpc_SetErr(ERPCMISMATCH, "Error:: too short RPC packet ...\n");                rpc_SetErr(ERPCMISMATCH, "Too short RPC packet ...");
                 free(buf);                  free(buf);
                return -4;                return -1;
        } else        }
                rrpc = (struct tagRPCCall*) buf;
         /* calculate CRC */
         crc = ntohs(rpc->call_crc);
         rpc->call_crc ^= rpc->call_crc;
         if (crc != crcFletcher16((u_short*) buf, ret / 2)) {
                 rpc_SetErr(ERPCMISMATCH, "Bad CRC RPC packet");
                 free(buf);
                 return -1;
         }
 
         /* check RPC packet session info */          /* check RPC packet session info */
        if (rpc_chkPktSession(&rrpc->call_session, cli->cli_parent)) {        if (rpc_chkPktSession(&rpc->call_session, cli->cli_parent)) {
                rpc_SetErr(ERPCMISMATCH, "Error:: get invalid RPC session ...\n");                rpc_SetErr(ERPCMISMATCH, "Get invalid RPC session ...");
                 free(buf);                  free(buf);
                return -5;                return -1;
         } else          } else
                Limit = sizeof(struct tagRPCCall);                wlen = sizeof(struct tagRPCCall);
        if (rrpc->call_tag != tag || rrpc->call_hash != hash) {        if (rpc->call_tag != tag || rpc->call_hash != hash) {
                rpc_SetErr(ERPCMISMATCH, "Error:: get wrong RPC reply ...\n");                rpc_SetErr(ERPCMISMATCH, "Get wrong RPC reply ...");
                 free(buf);                  free(buf);
                return -5;                return -1;
         }          }
        if (ntohl(rrpc->call_rep.ret) < 0 && ntohl(rrpc->call_rep.eno)) {        if (ntohl(rpc->call_rep.ret) < 0 && ntohl(rpc->call_rep.eno)) {
                rpc_SetErr(ntohl(rrpc->call_rep.eno), "Error::Server side: retcode=%d #%d %s\n",                 rpc_SetErr(ntohl(rpc->call_rep.eno), "Server side: retcode=%d #%d %s", 
                                ntohl(rrpc->call_rep.ret), ntohl(rrpc->call_rep.eno),                                 ntohl(rpc->call_rep.ret), ntohl(rpc->call_rep.eno), 
                                strerror(ntohl(rrpc->call_rep.eno)));                                strerror(ntohl(rpc->call_rep.eno)));
                 free(buf);                  free(buf);
                return -6;                return -1;
         }          }
        if (ntohs(rrpc->call_argc) * sizeof(ait_val_t) > cli->cli_netbuf - Limit) {        if (ntohs(rpc->call_argc) * sizeof(ait_val_t) > cli->cli_netbuf - wlen) {
                rpc_SetErr(EMSGSIZE, "Error:: reply RPC packet is too long ...\n");                rpc_SetErr(EMSGSIZE, "Reply RPC packet is too long ...");
                 free(buf);                  free(buf);
                return -7;                return -1;
         }          }
   
         /* RPC is OK! Go de-marshaling variables ... */          /* RPC is OK! Go de-marshaling variables ... */
        if (ntohs(rrpc->call_argc)) {        if (ntohs(rpc->call_argc)) {
                *out_vars = io_buffer2vars(buf + Limit, cli->cli_netbuf - Limit                *out_vars = io_buffer2vars(buf + wlen, cli->cli_netbuf - wlen
                                ntohs(rrpc->call_argc), 0);                                ntohs(rpc->call_argc), 0);
                 if (!*out_vars) {                  if (!*out_vars) {
                         free(buf);                          free(buf);
                         return -1;                          return -1;
                 }                  }
         }          }
   
        ret = ntohl(rrpc->call_rep.ret);        ret = ntohl(rpc->call_rep.ret);
         free(buf);          free(buf);
         return ret;          return ret;
 }  }

Removed from v.1.6.2.1  
changed lines
  Added in v.1.9.2.3


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