Diff for /libaitrpc/src/blob.c between versions 1.4.4.1 and 1.7.2.4

version 1.4.4.1, 2012/03/13 17:21:52 version 1.7.2.4, 2012/05/16 13:17:51
Line 47  SUCH DAMAGE. Line 47  SUCH DAMAGE.
   
   
 /*  /*
 * rpc_srv_blobCreate() Create map blob to memory region and return object * rpc_srv_blobCreate() - Create and map blob to memory region and return object
  *
  * @srv = RPC Server instance   * @srv = RPC Server instance
  * @len = BLOB length object   * @len = BLOB length object
  * return: NULL error or !=NULL allocated BLOB object   * return: NULL error or !=NULL allocated BLOB object
Line 60  rpc_srv_blobCreate(rpc_srv_t * __restrict srv, int len Line 61  rpc_srv_blobCreate(rpc_srv_t * __restrict srv, int len
         int f;          int f;
         u_int rnd;          u_int rnd;
   
 #ifdef HAVE_SRANDOMDEV  
         srandomdev();  
 #else  
         time_t tim;  
   
         srandom((time(&tim) ^ getpid()));  
 #endif  
 again:  again:
         rnd = random() % UINT_MAX;          rnd = random() % UINT_MAX;
   
Line 80  again: Line 74  again:
                 LOGERR;                  LOGERR;
                 return NULL;                  return NULL;
         }          }
        if (lseek(f, len - 1, SEEK_SET) == -1) {        if (ftruncate(f, len) == -1) {
                 LOGERR;                  LOGERR;
                 close(f);                  close(f);
                 unlink(szFName);                  unlink(szFName);
                 return NULL;                  return NULL;
        } else        }
                write(f, "", 1); 
   
         blob = malloc(sizeof(rpc_blob_t));          blob = malloc(sizeof(rpc_blob_t));
         if (!blob) {          if (!blob) {
Line 112  again: Line 105  again:
 }  }
   
 /*  /*
 * rpc_srv_blobMap() Map blob to memory region  * rpc_srv_blobMap() - Map blob to memory region 
  *
  * @srv = RPC Server instance   * @srv = RPC Server instance
  * @blob = Map to this BLOB element   * @blob = Map to this BLOB element
  * return: -1 error or 0 ok   * return: -1 error or 0 ok
Line 124  rpc_srv_blobMap(rpc_srv_t * __restrict srv, rpc_blob_t Line 118  rpc_srv_blobMap(rpc_srv_t * __restrict srv, rpc_blob_t
         char szFName[MAXPATHLEN];          char szFName[MAXPATHLEN];
   
         if (!blob) {          if (!blob) {
                rpc_SetErr(EINVAL, "Error:: invalid arguments ...\n");                rpc_SetErr(EINVAL, "Invalid argument BLOB");
                 return -1;                  return -1;
         }          }
         if (blob->blob_data) {          if (blob->blob_data) {
                rpc_SetErr(EPERM, "Error:: already mmapped object found!\n");                rpc_SetErr(EPERM, "Already mmapped object found!");
                 return -1;                  return -1;
         }          }
   
Line 156  rpc_srv_blobMap(rpc_srv_t * __restrict srv, rpc_blob_t Line 150  rpc_srv_blobMap(rpc_srv_t * __restrict srv, rpc_blob_t
 }  }
   
 /*  /*
 * rpc_srv_blobUnmap() Unmap blob memory region  * rpc_srv_blobUnmap() - Unmap blob memory region 
  *
  * @blob = Mapped BLOB element   * @blob = Mapped BLOB element
  * return: none   * return: none
  */   */
 inline void  inline void
 rpc_srv_blobUnmap(rpc_blob_t * __restrict blob)  rpc_srv_blobUnmap(rpc_blob_t * __restrict blob)
 {  {
        if (!blob || !blob->blob_data)        if (blob && blob->blob_data) {
                rpc_SetErr(EINVAL, "Error:: invalid arguments ...\n"); 
        else { 
                 munmap(blob->blob_data, blob->blob_len);                  munmap(blob->blob_data, blob->blob_len);
                 blob->blob_data = NULL;                  blob->blob_data = NULL;
         }          }
 }  }
   
 /*  /*
 * rpc_srv_blobFree() Free blob from disk & memory * rpc_srv_blobFree() - Free blob from disk & memory
  *
  * @srv = RPC Server instance   * @srv = RPC Server instance
  * @blob = Mapped BLOB element   * @blob = Mapped BLOB element
  * return: -1 error or 0 ok   * return: -1 error or 0 ok
Line 183  rpc_srv_blobFree(rpc_srv_t * __restrict srv, rpc_blob_ Line 177  rpc_srv_blobFree(rpc_srv_t * __restrict srv, rpc_blob_
         char szFName[MAXPATHLEN];          char szFName[MAXPATHLEN];
   
         if (!blob) {          if (!blob) {
                rpc_SetErr(EINVAL, "Error:: invalid arguments ...\n");                rpc_SetErr(EINVAL, "Invalid argument BLOB");
                 return -1;                  return -1;
        }        } else
 
        if (blob->blob_data) 
                 rpc_srv_blobUnmap(blob);                  rpc_srv_blobUnmap(blob);
   
         memset(szFName, 0, sizeof szFName);          memset(szFName, 0, sizeof szFName);
Line 200  rpc_srv_blobFree(rpc_srv_t * __restrict srv, rpc_blob_ Line 192  rpc_srv_blobFree(rpc_srv_t * __restrict srv, rpc_blob_
         return 0;          return 0;
 }  }
   
// ------------------------------------------------------------/* ------------------------------------------------------------ */
   
 /*  /*
 * rpc_srv_sendBLOB() Send mapped BLOB to client * rpc_srv_sendBLOB() - Send mapped BLOB to client
  *
  * @cli = Client instance   * @cli = Client instance
  * @blob = Mapped BLOB element   * @blob = Mapped BLOB element
  * return: -1 error, 0 ok   * return: -1 error, 0 ok
Line 215  rpc_srv_sendBLOB(rpc_cli_t * __restrict cli, rpc_blob_ Line 208  rpc_srv_sendBLOB(rpc_cli_t * __restrict cli, rpc_blob_
         uint8_t *pos;          uint8_t *pos;
   
         if (!cli || !blob || !blob->blob_data) {          if (!cli || !blob || !blob->blob_data) {
                rpc_SetErr(EINVAL, "Error:: invalid arguments ...\n");                rpc_SetErr(EINVAL, "Invalid arguments");
                 return -1;                  return -1;
         }          }
   
         for (ret = blob->blob_len, pos = blob->blob_data; ret > 0; ret -= len, pos += len) {          for (ret = blob->blob_len, pos = blob->blob_data; ret > 0; ret -= len, pos += len) {
                len = send(cli->cli_sock, pos, ret, 0);                len = send(cli->cli_sock, pos, ret, MSG_NOSIGNAL);
                 if (len == -1) {                  if (len == -1) {
                         LOGERR;                          LOGERR;
                         return -1;                          return -1;
Line 231  rpc_srv_sendBLOB(rpc_cli_t * __restrict cli, rpc_blob_ Line 224  rpc_srv_sendBLOB(rpc_cli_t * __restrict cli, rpc_blob_
 }  }
   
 /*  /*
 * rpc_srv_recvBLOB() Receive BLOB from client * rpc_srv_recvBLOB() - Receive BLOB from client
  *
  * @cli = Client instance   * @cli = Client instance
  * @blob = Mapped BLOB element   * @blob = Mapped BLOB element
  * return: -1 error, 0 ok, >0 unreceived data from client, may be error?   * return: -1 error, 0 ok, >0 unreceived data from client, may be error?
Line 241  rpc_srv_recvBLOB(rpc_cli_t * __restrict cli, rpc_blob_ Line 235  rpc_srv_recvBLOB(rpc_cli_t * __restrict cli, rpc_blob_
 {  {
         int ret, len;          int ret, len;
         uint8_t *pos;          uint8_t *pos;
        fd_set fds;        struct pollfd pfd;
        struct timeval tv = { DEF_RPC_TIMEOUT, 0 }; 
   
         if (!cli || !blob || !blob->blob_data) {          if (!cli || !blob || !blob->blob_data) {
                rpc_SetErr(EINVAL, "Error:: invalid arguments ...\n");                rpc_SetErr(EINVAL, "Invalid arguments");
                 return -1;                  return -1;
         }          }
   
           pfd.fd = cli->cli_sock;
           pfd.events = POLLIN | POLLPRI;
         for (ret = blob->blob_len, pos = blob->blob_data; ret > 0; ret -= len, pos += len) {          for (ret = blob->blob_len, pos = blob->blob_data; ret > 0; ret -= len, pos += len) {
                FD_ZERO(&fds);                if ((len = poll(&pfd, 1, DEF_RPC_TIMEOUT * 1000)) < 1 || 
                FD_SET(cli->cli_sock, &fds);                                pfd.revents & (POLLERR | POLLHUP | POLLNVAL)) {
                len = select(cli->cli_sock + 1, &fds, NULL, NULL, &tv); 
                if (len < 1) { 
                         LOGERR;                          LOGERR;
                         return -1;                          return -1;
                 }                  }
Line 268  rpc_srv_recvBLOB(rpc_cli_t * __restrict cli, rpc_blob_ Line 261  rpc_srv_recvBLOB(rpc_cli_t * __restrict cli, rpc_blob_
         return ret;          return ret;
 }  }
   
// ------------------------------------------------------------/* ------------------------------------------------------------ */
   
 /*  /*
 * rpc_cli_sendBLOB() Send BLOB to server * rpc_cli_sendBLOB() - Send BLOB to server
  *
  * @cli = Client instance   * @cli = Client instance
  * @var = BLOB variable   * @var = BLOB variable
  * @data = BLOB data   * @data = BLOB data
Line 283  rpc_cli_sendBLOB(rpc_cli_t * __restrict cli, ait_val_t Line 277  rpc_cli_sendBLOB(rpc_cli_t * __restrict cli, ait_val_t
         int ret, len;          int ret, len;
         uint8_t *pos;          uint8_t *pos;
         struct tagBLOBHdr hdr;          struct tagBLOBHdr hdr;
        fd_set fds;        struct pollfd pfd;
        struct timeval tv = { DEF_RPC_TIMEOUT, 0 }; 
   
         if (!cli || !var || !data) {          if (!cli || !var || !data) {
                rpc_SetErr(EINVAL, "Error:: invalid arguments ...\n");                rpc_SetErr(EINVAL, "Invalid arguments");
                 return -1;                  return -1;
         }          }
   
        memcpy(&hdr.hdr_session, cli->cli_parent, sizeof(rpc_sess_t));        rpc_addPktSession(&hdr.hdr_session, cli->cli_parent);
         hdr.hdr_cmd = set;          hdr.hdr_cmd = set;
         hdr.hdr_var = 0;          hdr.hdr_var = 0;
         hdr.hdr_ret = 0;          hdr.hdr_ret = 0;
        hdr.hdr_len = AIT_LEN(var);        hdr.hdr_len = htonl(AIT_LEN(var));
         /* calculate CRC */
         hdr.hdr_crc ^= hdr.hdr_crc;
         hdr.hdr_crc = htons(crcFletcher16((u_short*) &hdr, sizeof hdr / 2));
 
         /* send SET request */
         if (send(cli->cli_sock, &hdr, sizeof hdr, 0) == -1) {          if (send(cli->cli_sock, &hdr, sizeof hdr, 0) == -1) {
                 LOGERR;                  LOGERR;
                 return -1;                  return -1;
Line 308  rpc_cli_sendBLOB(rpc_cli_t * __restrict cli, ait_val_t Line 306  rpc_cli_sendBLOB(rpc_cli_t * __restrict cli, ait_val_t
                         return -1;                          return -1;
                 }                  }
   
        FD_ZERO(&fds);        /* wait for reply */
        FD_SET(cli->cli_sock, &fds);        pfd.fd = cli->cli_sock;
        switch (select(cli->cli_sock + 1, &fds, NULL, NULL, &tv)) {        pfd.events = POLLIN | POLLPRI;
                case -1:        if ((ret = poll(&pfd, 1, DEF_RPC_TIMEOUT * 1000)) < 1 || 
                         pfd.revents & (POLLERR | POLLHUP | POLLNVAL)) {
                 if (ret)
                         LOGERR;                          LOGERR;
                        return -1;                else
                case 0:                        rpc_SetErr(ETIMEDOUT, "Timeout reached! Server not respond");
                        rpc_SetErr(ETIMEDOUT, "Error:: Timeout reached! Server not responde ...\n");                return -1;
                        return -1; 
         }          }
         if (recv(cli->cli_sock, &hdr, sizeof hdr, 0) == -1) {          if (recv(cli->cli_sock, &hdr, sizeof hdr, 0) == -1) {
                 LOGERR;                  LOGERR;
                return -1;                return 1;
         }          }
           /* check CRC */
           ret = ntohs(hdr.hdr_crc);
           hdr.hdr_crc ^= hdr.hdr_crc;
           if (ret != crcFletcher16((u_short*) &hdr, sizeof hdr / 2)) {
                   rpc_SetErr(ERPCMISMATCH, "Bad CRC BLOB packet");
                   return 1;
           }
   
         if (hdr.hdr_cmd != error) {          if (hdr.hdr_cmd != error) {
                if (hdr.hdr_len != AIT_LEN(var)) {                if (ntohl(hdr.hdr_len) != AIT_LEN(var)) {
                        rpc_SetErr(ECANCELED, "Error:: Bad return length packet ...\n");                        rpc_SetErr(ECANCELED, "Bad return length packet");
                        return -1;                        return 1;
                 }                  }
   
                var->val.blob = hdr.hdr_var;                var->val.blob = ntohl(hdr.hdr_var);
         }          }
   
         return hdr.hdr_cmd == error;          return hdr.hdr_cmd == error;
 }  }
   
 /*  /*
 * rpc_cli_recvBLOB() Receive BLOB from server * rpc_cli_recvBLOB() - Receive BLOB from server
  *
  * @cli = Client instance   * @cli = Client instance
  * @var = BLOB variable   * @var = BLOB variable
  * @data = BLOB data, must be free after use!   * @data = BLOB data, must be free after use!
Line 346  rpc_cli_recvBLOB(rpc_cli_t * __restrict cli, ait_val_t Line 354  rpc_cli_recvBLOB(rpc_cli_t * __restrict cli, ait_val_t
 {  {
         int ret, len;          int ret, len;
         uint8_t *pos;          uint8_t *pos;
        fd_set fds;        struct pollfd pfd;
        struct timeval tv = { DEF_RPC_TIMEOUT, 0 }; 
         struct tagBLOBHdr hdr;          struct tagBLOBHdr hdr;
   
         if (!cli || !var || !data) {          if (!cli || !var || !data) {
                rpc_SetErr(EINVAL, "Error:: invalid arguments ...\n");                rpc_SetErr(EINVAL, "Invalid arguments");
                 return -1;                  return -1;
         }          }
   
Line 362  rpc_cli_recvBLOB(rpc_cli_t * __restrict cli, ait_val_t Line 369  rpc_cli_recvBLOB(rpc_cli_t * __restrict cli, ait_val_t
         } else          } else
                 memset(*data, 0, AIT_LEN(var));                  memset(*data, 0, AIT_LEN(var));
   
        memcpy(&hdr.hdr_session, cli->cli_parent, sizeof(rpc_sess_t));        rpc_addPktSession(&hdr.hdr_session, cli->cli_parent);
         hdr.hdr_cmd = get;          hdr.hdr_cmd = get;
        hdr.hdr_var = (uint32_t) AIT_GET_BLOB(var);        hdr.hdr_var = htonl((uint32_t) AIT_GET_BLOB(var));
         hdr.hdr_ret = 0;          hdr.hdr_ret = 0;
         hdr.hdr_len = 0;          hdr.hdr_len = 0;
           /* calculate CRC */
           hdr.hdr_crc ^= hdr.hdr_crc;
           hdr.hdr_crc = htons(crcFletcher16((u_short*) &hdr, sizeof hdr / 2));
   
           /* send GET request */
         if (send(cli->cli_sock, &hdr, sizeof hdr, 0) == -1) {          if (send(cli->cli_sock, &hdr, sizeof hdr, 0) == -1) {
                 LOGERR;                  LOGERR;
                 free(*data);                  free(*data);
Line 375  rpc_cli_recvBLOB(rpc_cli_t * __restrict cli, ait_val_t Line 387  rpc_cli_recvBLOB(rpc_cli_t * __restrict cli, ait_val_t
         }          }
   
         /* receive BLOB from server */          /* receive BLOB from server */
           pfd.fd = cli->cli_sock;
           pfd.events = POLLIN | POLLPRI;
         for (ret = AIT_LEN(var), pos = *data; ret > 0; ret -= len, pos += len) {          for (ret = AIT_LEN(var), pos = *data; ret > 0; ret -= len, pos += len) {
                FD_ZERO(&fds);                if ((len = poll(&pfd, 1, DEF_RPC_TIMEOUT * 1000)) < 1 || 
                FD_SET(cli->cli_sock, &fds);                                pfd.revents & (POLLERR | POLLHUP | POLLNVAL)) {
                len = select(cli->cli_sock + 1, &fds, NULL, NULL, &tv); 
                if (len < 1) { 
                         LOGERR;                          LOGERR;
                         free(*data);                          free(*data);
                         *data = NULL;                          *data = NULL;
Line 394  rpc_cli_recvBLOB(rpc_cli_t * __restrict cli, ait_val_t Line 406  rpc_cli_recvBLOB(rpc_cli_t * __restrict cli, ait_val_t
                 }                  }
         }          }
   
        FD_ZERO(&fds);        /* wait for reply */
        FD_SET(cli->cli_sock, &fds);        if ((len = poll(&pfd, 1, DEF_RPC_TIMEOUT * 1000)) < 1 || 
        switch (select(cli->cli_sock + 1, &fds, NULL, NULL, &tv)) {                        pfd.revents & (POLLERR | POLLHUP | POLLNVAL)) {
                case -1:                if (len)
                         LOGERR;                          LOGERR;
                        free(*data);                else
                        *data = NULL;                        rpc_SetErr(ETIMEDOUT, "Timeout reached! Server not respond");
                        return -1;                free(*data);
                case 0:                *data = NULL;
                        rpc_SetErr(ETIMEDOUT, "Error:: Timeout reached! Server not responde ...\n");                return 1;
                        free(*data); 
                        *data = NULL; 
                        return -1; 
         }          }
         if (recv(cli->cli_sock, &hdr, sizeof hdr, 0) == -1) {          if (recv(cli->cli_sock, &hdr, sizeof hdr, 0) == -1) {
                 LOGERR;                  LOGERR;
                 free(*data);                  free(*data);
                 *data = NULL;                  *data = NULL;
                return -1;                return 1;
         }          }
           /* check CRC */
           ret = ntohs(hdr.hdr_crc);
           hdr.hdr_crc ^= hdr.hdr_crc;
           if (ret != crcFletcher16((u_short*) &hdr, sizeof hdr / 2)) {
                   rpc_SetErr(ERPCMISMATCH, "Bad CRC BLOB packet");
                   free(*data);
                   *data = NULL;
                   return 1;
           }
         if (hdr.hdr_cmd != error) {          if (hdr.hdr_cmd != error) {
                if (hdr.hdr_len != AIT_LEN(var)) {                if (ntohl(hdr.hdr_len) != AIT_LEN(var)) {
                        rpc_SetErr(ECANCELED, "Error:: Bad return length packet ...\n");                        rpc_SetErr(ECANCELED, "Bad return length packet");
                         free(*data);                          free(*data);
                         *data = NULL;                          *data = NULL;
                        return -1;                        return 1;
                 }                  }
         }          }
   
Line 427  rpc_cli_recvBLOB(rpc_cli_t * __restrict cli, ait_val_t Line 445  rpc_cli_recvBLOB(rpc_cli_t * __restrict cli, ait_val_t
 }  }
   
 /*  /*
 * rpc_cli_delBLOB() Delete BLOB from server * rpc_cli_delBLOB() - Delete BLOB from server
  *
  * @cli = Client instance   * @cli = Client instance
  * @var = BLOB variable   * @var = BLOB variable
  * return: -1 error, 0 ok, 1 remote error   * return: -1 error, 0 ok, 1 remote error
Line 436  int Line 455  int
 rpc_cli_delBLOB(rpc_cli_t * __restrict cli, ait_val_t * __restrict var)  rpc_cli_delBLOB(rpc_cli_t * __restrict cli, ait_val_t * __restrict var)
 {  {
         struct tagBLOBHdr hdr;          struct tagBLOBHdr hdr;
        fd_set fds;        struct pollfd pfd;
        struct timeval tv = { DEF_RPC_TIMEOUT, 0 };        int ret;
   
         if (!cli || !var) {          if (!cli || !var) {
                rpc_SetErr(EINVAL, "Error:: invalid arguments ...\n");                rpc_SetErr(EINVAL, "Invalid arguments");
                 return -1;                  return -1;
         }          }
   
        memcpy(&hdr.hdr_session, cli->cli_parent, sizeof(rpc_sess_t));        rpc_addPktSession(&hdr.hdr_session, cli->cli_parent);
         hdr.hdr_cmd = unset;          hdr.hdr_cmd = unset;
        hdr.hdr_var = (uint32_t) AIT_GET_BLOB(var);        hdr.hdr_var = htonl((uint32_t) AIT_GET_BLOB(var));
         hdr.hdr_ret = 0;          hdr.hdr_ret = 0;
         hdr.hdr_len = 0;          hdr.hdr_len = 0;
           /* calculate CRC */
           hdr.hdr_crc ^= hdr.hdr_crc;
           hdr.hdr_crc = htons(crcFletcher16((u_short*) &hdr, sizeof hdr / 2));
   
           /* send UNSET request */
         if (send(cli->cli_sock, &hdr, sizeof hdr, 0) == -1) {          if (send(cli->cli_sock, &hdr, sizeof hdr, 0) == -1) {
                 LOGERR;                  LOGERR;
                 return -1;                  return -1;
         }          }
   
        FD_ZERO(&fds);        /* wait for reply */
        FD_SET(cli->cli_sock, &fds);        pfd.fd = cli->cli_sock;
        switch (select(cli->cli_sock + 1, &fds, NULL, NULL, &tv)) {        pfd.events = POLLIN | POLLPRI;
                case -1:        if ((ret = poll(&pfd, 1, DEF_RPC_TIMEOUT * 1000)) < 1 || 
                         pfd.revents & (POLLERR | POLLHUP | POLLNVAL)) {
                 if (ret)
                         LOGERR;                          LOGERR;
                        return -1;                else
                case 0:                        rpc_SetErr(ETIMEDOUT, "Timeout reached! Server not respond");
                        rpc_SetErr(ETIMEDOUT, "Error:: Timeout reached! Server not responde ...\n");                return 1;
                        return -1; 
         }          }
         if (recv(cli->cli_sock, &hdr, sizeof hdr, 0) == -1) {          if (recv(cli->cli_sock, &hdr, sizeof hdr, 0) == -1) {
                 LOGERR;                  LOGERR;
                return -1;                return 1;
         }          }
           /* check CRC */
           ret = ntohs(hdr.hdr_crc);
           hdr.hdr_crc ^= hdr.hdr_crc;
           if (ret != crcFletcher16((u_short*) &hdr, sizeof hdr / 2)) {
                   rpc_SetErr(ERPCMISMATCH, "Bad CRC BLOB packet");
                   return 1;
           }
   
         return hdr.hdr_cmd == error;          return hdr.hdr_cmd == error;
 }  }
   
 /*  /*
 * rpc_cli_getBLOB() Receive BLOB from server and Delete after that * rpc_cli_getBLOB() - Receive BLOB from server and Delete after that
  *
  * @cli = Client instance   * @cli = Client instance
  * @var = BLOB variable   * @var = BLOB variable
  * @data = BLOB data, must be free after use!   * @data = BLOB data, must be free after use!

Removed from v.1.4.4.1  
changed lines
  Added in v.1.7.2.4


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