Diff for /libaitrpc/src/blob.c between versions 1.1.2.5 and 1.1.2.11

version 1.1.2.5, 2010/06/28 15:54:05 version 1.1.2.11, 2010/07/08 07:32:53
Line 13  rpc_srv_blobCreate(rpc_srv_t * __restrict srv, int len Line 13  rpc_srv_blobCreate(rpc_srv_t * __restrict srv, int len
         rpc_blob_t *blob = NULL;          rpc_blob_t *blob = NULL;
         char szFName[MAXPATHLEN];          char szFName[MAXPATHLEN];
         int f;          int f;
           u_int rnd;
   
           srandomdev();
           srandom(getpid() ^ time(NULL));
   again:
           rnd = random() % UINT_MAX;
   
         memset(szFName, 0, MAXPATHLEN);          memset(szFName, 0, MAXPATHLEN);
        snprintf(szFName, MAXPATHLEN, "%s/BLOB-XXXXXXXX", srv->srv_blob.dir);        snprintf(szFName, MAXPATHLEN, BLOB_FILE, srv->srv_blob.dir, rnd);
        f = mkstemp(szFName);        f = open(szFName, O_CREAT | O_EXCL | O_RDWR, 0600);
         if (f == -1) {          if (f == -1) {
                   if (errno == EEXIST)
                           goto again;
   
                 LOGERR;                  LOGERR;
                 return NULL;                  return NULL;
         }          }
Line 48  rpc_srv_blobCreate(rpc_srv_t * __restrict srv, int len Line 57  rpc_srv_blobCreate(rpc_srv_t * __restrict srv, int len
                 close(f);                  close(f);
   
         blob->blob_len = len;          blob->blob_len = len;
        blob->blob_var = strtol(strrchr(szFName, '-') + 1, NULL, 16);        blob->blob_var = rnd;
         return blob;          return blob;
 }  }
   
Line 72  rpc_srv_blobMap(rpc_srv_t * __restrict srv, rpc_blob_t Line 81  rpc_srv_blobMap(rpc_srv_t * __restrict srv, rpc_blob_t
   
         memset(szFName, 0, MAXPATHLEN);          memset(szFName, 0, MAXPATHLEN);
         snprintf(szFName, MAXPATHLEN, BLOB_FILE, srv->srv_blob.dir, blob->blob_var);          snprintf(szFName, MAXPATHLEN, BLOB_FILE, srv->srv_blob.dir, blob->blob_var);
        f = open(szFName, O_RDONLY);        f = open(szFName, O_RDWR);
         if (f == -1) {          if (f == -1) {
                 LOGERR;                  LOGERR;
                 return -1;                  return -1;
Line 83  rpc_srv_blobMap(rpc_srv_t * __restrict srv, rpc_blob_t Line 92  rpc_srv_blobMap(rpc_srv_t * __restrict srv, rpc_blob_t
                 return -1;                  return -1;
         }          }
   
        blob->blob_data = mmap(NULL, sb.st_size, PROT_READ, MAP_PRIVATE, f, 0);        blob->blob_data = mmap(NULL, sb.st_size, PROT_READ | PROT_WRITE, MAP_SHARED, f, 0);
         if (blob->blob_data == MAP_FAILED) {          if (blob->blob_data == MAP_FAILED) {
                 LOGERR;                  LOGERR;
                 close(f);                  close(f);
Line 217  rpc_srv_recvBLOB(rpc_cli_t * __restrict cli, rpc_blob_ Line 226  rpc_srv_recvBLOB(rpc_cli_t * __restrict cli, rpc_blob_
  * @cli = Client instance   * @cli = Client instance
  * @var = BLOB variable   * @var = BLOB variable
  * @data = BLOB data   * @data = BLOB data
 * return: -1 error, 0 ok * return: -1 error, 0 ok, 1 remote error
  */   */
 int  int
 rpc_cli_sendBLOB(rpc_cli_t * __restrict cli, rpc_val_t * __restrict var, void * __restrict data)  rpc_cli_sendBLOB(rpc_cli_t * __restrict cli, rpc_val_t * __restrict var, void * __restrict data)
 {  {
         int ret, len;          int ret, len;
         uint8_t *pos;          uint8_t *pos;
           struct tagBLOBHdr hdr;
           fd_set fds;
           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, "Error:: invalid arguments ...\n");
                 return -1;                  return -1;
         }          }
   
           memcpy(&hdr.hdr_session, cli->cli_parent, sizeof(rpc_sess_t));
           hdr.hdr_cmd = set;
           hdr.hdr_var = 0;
           hdr.hdr_ret = 0;
           hdr.hdr_len = var->val_len;
           if (send(cli->cli_sock, &hdr, sizeof hdr, 0) == -1) {
                   LOGERR;
                   return -1;
           }
   
         for (ret = var->val_len, pos = data; ret > 0; ret -= len, pos += len)          for (ret = var->val_len, pos = data; ret > 0; ret -= len, pos += len)
                 if ((len = send(cli->cli_sock, pos, ret > BLOBSIZ ? BLOBSIZ : ret, 0)) == -1) {                  if ((len = send(cli->cli_sock, pos, ret > BLOBSIZ ? BLOBSIZ : ret, 0)) == -1) {
                         LOGERR;                          LOGERR;
                         return -1;                          return -1;
                 }                  }
   
        return ret;        FD_ZERO(&fds);
         FD_SET(cli->cli_sock, &fds);
         switch (select(cli->cli_sock + 1, &fds, NULL, NULL, &tv)) {
                 case -1:
                         LOGERR;
                         return -1;
                 case 0:
                         rpc_SetErr(ETIMEDOUT, "Error:: Timeout reached! Server not responde ...\n");
                         return -1;
         }
         if (recv(cli->cli_sock, &hdr, sizeof hdr, 0) == -1) {
                 LOGERR;
                 return -1;
         }
         if (hdr.hdr_cmd != error) {
                 if (hdr.hdr_len != var->val_len) {
                         rpc_SetErr(EBADMSG, "Error:: Bad return length packet ...\n");
                         return -1;
                 }
 
                 var->val.blob = hdr.hdr_var;
         }
 
         return hdr.hdr_cmd == error;
 }  }
   
 /*  /*
Line 244  rpc_cli_sendBLOB(rpc_cli_t * __restrict cli, rpc_val_t Line 289  rpc_cli_sendBLOB(rpc_cli_t * __restrict cli, rpc_val_t
  * @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!
 * return: -1 error, 0 ok, >0 unreceived data from server, may be error? * return: -1 error, 0 ok, 1 remote error
  */   */
 int  int
 rpc_cli_recvBLOB(rpc_cli_t * __restrict cli, rpc_val_t * __restrict var, void ** data)  rpc_cli_recvBLOB(rpc_cli_t * __restrict cli, rpc_val_t * __restrict var, void ** data)
Line 253  rpc_cli_recvBLOB(rpc_cli_t * __restrict cli, rpc_val_t Line 298  rpc_cli_recvBLOB(rpc_cli_t * __restrict cli, rpc_val_t
         uint8_t *pos;          uint8_t *pos;
         fd_set fds;          fd_set fds;
         struct timeval tv = { DEF_RPC_TIMEOUT, 0 };          struct timeval tv = { DEF_RPC_TIMEOUT, 0 };
           struct tagBLOBHdr hdr;
   
         if (!cli || !var || !data) {          if (!cli || !var || !data) {
                 rpc_SetErr(EINVAL, "Error:: invalid arguments ...\n");                  rpc_SetErr(EINVAL, "Error:: invalid arguments ...\n");
Line 266  rpc_cli_recvBLOB(rpc_cli_t * __restrict cli, rpc_val_t Line 312  rpc_cli_recvBLOB(rpc_cli_t * __restrict cli, rpc_val_t
         } else          } else
                 memset(*data, 0, var->val_len);                  memset(*data, 0, var->val_len);
   
           memcpy(&hdr.hdr_session, cli->cli_parent, sizeof(rpc_sess_t));
           hdr.hdr_cmd = get;
           hdr.hdr_var = (uint32_t) RPC_GET_BLOB(var);
           hdr.hdr_ret = 0;
           hdr.hdr_len = 0;
           if (send(cli->cli_sock, &hdr, sizeof hdr, 0) == -1) {
                   LOGERR;
                   free(*data);
                   *data = NULL;
                   return -1;
           }
   
         for (ret = var->val_len, pos = *data; ret > 0; ret -= len, pos += len) {          for (ret = var->val_len, pos = *data; ret > 0; ret -= len, pos += len) {
                 FD_ZERO(&fds);                  FD_ZERO(&fds);
                 FD_SET(cli->cli_sock, &fds);                  FD_SET(cli->cli_sock, &fds);
Line 284  rpc_cli_recvBLOB(rpc_cli_t * __restrict cli, rpc_val_t Line 342  rpc_cli_recvBLOB(rpc_cli_t * __restrict cli, rpc_val_t
                         return -1;                          return -1;
                 }                  }
         }          }
           printf("get all bytes ret=%d *data=%s", ret, *data);
   
        return ret;        FD_ZERO(&fds);
         FD_SET(cli->cli_sock, &fds);
         switch (select(cli->cli_sock + 1, &fds, NULL, NULL, &tv)) {
                 case -1:
                         LOGERR;
                         free(*data);
                         *data = NULL;
                         return -1;
                 case 0:
                         rpc_SetErr(ETIMEDOUT, "Error:: Timeout reached! Server not responde ...\n");
                         free(*data);
                         *data = NULL;
                         return -1;
         }
         if (recv(cli->cli_sock, &hdr, sizeof hdr, 0) == -1) {
                 LOGERR;
                 free(*data);
                 *data = NULL;
                 return -1;
         }
         if (hdr.hdr_cmd != error) {
                 if (hdr.hdr_len != var->val_len) {
                         rpc_SetErr(EBADMSG, "Error:: Bad return length packet ...\n");
                         free(*data);
                         *data = NULL;
                         return -1;
                 }
         }
 
         return hdr.hdr_cmd == error;
 }  }

Removed from v.1.1.2.5  
changed lines
  Added in v.1.1.2.11


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