Diff for /libaitrpc/src/srv.c between versions 1.9.2.28 and 1.11.2.3

version 1.9.2.28, 2012/05/17 16:47:35 version 1.11.2.3, 2012/09/17 14:16:42
Line 64  closeClient(sched_task_t *task) Line 64  closeClient(sched_task_t *task)
   
         io_arrayDel(s->srv_clients, c->cli_id, 0);          io_arrayDel(s->srv_clients, c->cli_id, 0);
         if (c)          if (c)
                free(c);                io_free(c);
         return NULL;          return NULL;
 }  }
   
Line 89  txPacket(sched_task_t *task) Line 89  txPacket(sched_task_t *task)
                         rpc->call_rep.ret = RPC_ERROR(-1);                          rpc->call_rep.ret = RPC_ERROR(-1);
                         rpc->call_rep.eno = RPC_ERROR(rpc_Errno);                          rpc->call_rep.eno = RPC_ERROR(rpc_Errno);
                 } else {                  } else {
                        rpc->call_argc = htons(io_arraySize(f->func_vars));                        rpc->call_argc = htons(io_arraySize(RPC_RETVARS(c)));
                         /* Go Encapsulate variables */                          /* Go Encapsulate variables */
                        ret = io_vars2buffer(buf + wlen, sizeof buf - wlen, f->func_vars);                        ret = io_vars2buffer(buf + wlen, sizeof buf - wlen, RPC_RETVARS(c));
                         /* Free return values */                          /* Free return values */
                        io_clrVars(f->func_vars);                        io_freeVars(&c->cli_vars);
                         if (ret == -1) {                          if (ret == -1) {
                                 rpc_SetErr(EBADRPC, "Prepare RPC packet failed");                                  rpc_SetErr(EBADRPC, "Prepare RPC packet failed");
                                 rpc->call_argc ^= rpc->call_argc;                                  rpc->call_argc ^= rpc->call_argc;
Line 153  execCall(sched_task_t *task) Line 153  execCall(sched_task_t *task)
                 rpc->call_rep.eno = RPC_ERROR(rpc_Errno);                  rpc->call_rep.eno = RPC_ERROR(rpc_Errno);
         } else {          } else {
                 /* if client doesn't want reply */                  /* if client doesn't want reply */
                argc = rpc->call_req.flags & RPC_NOREPLY;                argc = RPC_CHK_NOREPLY(rpc);
                rpc->call_rep.ret = RPC_ERROR(rpc_srv_execCall(f, rpc, arr));                rpc->call_rep.ret = RPC_ERROR(rpc_srv_execCall(c, rpc, f->func_name, arr));
                 if (rpc->call_rep.ret == htonl(-1)) {                  if (rpc->call_rep.ret == htonl(-1)) {
                         rpc->call_rep.eno = RPC_ERROR(errno);                          rpc->call_rep.eno = RPC_ERROR(errno);
                         rpc->call_argc ^= rpc->call_argc;                          rpc->call_argc ^= rpc->call_argc;
Line 162  execCall(sched_task_t *task) Line 162  execCall(sched_task_t *task)
                         rpc->call_rep.eno ^= rpc->call_rep.eno;                          rpc->call_rep.eno ^= rpc->call_rep.eno;
                         if (argc) {                          if (argc) {
                                 /* without reply */                                  /* without reply */
                                io_clrVars(f->func_vars);                                io_freeVars(&c->cli_vars);
                                 rpc->call_argc ^= rpc->call_argc;                                  rpc->call_argc ^= rpc->call_argc;
                         } else {                          } else {
                                 /* reply */                                  /* reply */
                                rpc->call_argc = htons(io_arraySize(f->func_vars));                                rpc->call_argc = htons(io_arraySize(RPC_RETVARS(c)));
                         }                          }
                 }                  }
         }          }
Line 229  rxPacket(sched_task_t *task) Line 229  rxPacket(sched_task_t *task)
                         continue;                          continue;
                 }                  }
   
                noreply = rpc->call_req.flags & RPC_NOREPLY;                noreply = RPC_CHK_NOREPLY(rpc);
   
                 /* check RPC packet session info */                  /* check RPC packet session info */
                 if (rpc_chkPktSession(&rpc->call_session, &s->srv_session)) {                  if (rpc_chkPktSession(&rpc->call_session, &s->srv_session)) {
Line 267  acceptClients(sched_task_t *task) Line 267  acceptClients(sched_task_t *task)
                         (c = io_array(srv->srv_clients, i, rpc_cli_t*)); i++);                          (c = io_array(srv->srv_clients, i, rpc_cli_t*)); i++);
         if (c)  /* no more free slots! */          if (c)  /* no more free slots! */
                 goto end;                  goto end;
        c = malloc(sizeof(rpc_cli_t));        c = io_malloc(sizeof(rpc_cli_t));
         if (!c) {          if (!c) {
                 LOGERR;                  LOGERR;
                 srv->srv_kill = 1;                  srv->srv_kill = 1;
Line 318  closeBLOBClient(sched_task_t *task) Line 318  closeBLOBClient(sched_task_t *task)
   
         io_arrayDel(s->srv_blob.clients, c->cli_id, 0);          io_arrayDel(s->srv_blob.clients, c->cli_id, 0);
         if (c)          if (c)
                free(c);                io_free(c);
         return NULL;          return NULL;
 }  }
   
Line 421  rxBLOB(sched_task_t *task) Line 421  rxBLOB(sched_task_t *task)
                         }                          }
                         break;                          break;
                 case unset:                  case unset:
                        if (rpc_srv_unregisterBLOB(s, blob.hdr_var) == -1) {                        if (rpc_srv_unregisterBLOB(s, ntohl(blob.hdr_var)) == -1) {
                                 blob.hdr_cmd = error;                                  blob.hdr_cmd = error;
                                 blob.hdr_ret = RPC_ERROR(-1);                                  blob.hdr_ret = RPC_ERROR(-1);
                         }                          }
Line 446  acceptBLOBClients(sched_task_t *task) Line 446  acceptBLOBClients(sched_task_t *task)
         rpc_cli_t *c = NULL;          rpc_cli_t *c = NULL;
         register int i;          register int i;
         socklen_t salen = sizeof(io_sockaddr_t);          socklen_t salen = sizeof(io_sockaddr_t);
   #ifdef TCP_NOPUSH
           int n = 1;
   #endif
   
         /* check free slots for connect */          /* check free slots for connect */
         for (i = 0; i < io_arraySize(srv->srv_blob.clients) &&           for (i = 0; i < io_arraySize(srv->srv_blob.clients) && 
                         (c = io_array(srv->srv_blob.clients, i, rpc_cli_t*)); i++);                          (c = io_array(srv->srv_blob.clients, i, rpc_cli_t*)); i++);
         if (c)  /* no more free slots! */          if (c)  /* no more free slots! */
                 goto end;                  goto end;
        c = malloc(sizeof(rpc_cli_t));        c = io_malloc(sizeof(rpc_cli_t));
         if (!c) {          if (!c) {
                 LOGERR;                  LOGERR;
                 srv->srv_kill = srv->srv_blob.kill = 1;                  srv->srv_kill = srv->srv_blob.kill = 1;
Line 474  acceptBLOBClients(sched_task_t *task) Line 477  acceptBLOBClients(sched_task_t *task)
                 AIT_FREE_VAL(&c->cli_buf);                  AIT_FREE_VAL(&c->cli_buf);
                 io_arrayDel(srv->srv_blob.clients, i, 42);                  io_arrayDel(srv->srv_blob.clients, i, 42);
                 goto end;                  goto end;
        } else        } else {
 #ifdef TCP_NOPUSH
                 setsockopt(c->cli_sock, IPPROTO_TCP, TCP_NOPUSH, &n, sizeof n);
 #endif
                 fcntl(c->cli_sock, F_SETFL, fcntl(c->cli_sock, F_GETFL) | O_NONBLOCK);                  fcntl(c->cli_sock, F_SETFL, fcntl(c->cli_sock, F_GETFL) | O_NONBLOCK);
           }
   
         schedRead(TASK_ROOT(task), rxBLOB, c, c->cli_sock, NULL, 0);          schedRead(TASK_ROOT(task), rxBLOB, c, c->cli_sock, NULL, 0);
 end:  end:
Line 663  rpc_srv_loopBLOBServer(rpc_srv_t * __restrict srv) Line 670  rpc_srv_loopBLOBServer(rpc_srv_t * __restrict srv)
                 TAILQ_REMOVE(&srv->srv_blob.blobs, b, blob_node);                  TAILQ_REMOVE(&srv->srv_blob.blobs, b, blob_node);
   
                 rpc_srv_blobFree(srv, b);                  rpc_srv_blobFree(srv, b);
                free(b);                io_free(b);
         }          }
   
         schedEnd(&srv->srv_blob.root);          schedEnd(&srv->srv_blob.root);
Line 689  rpc_srv_initServer(u_int regProgID, u_char regProcID,  Line 696  rpc_srv_initServer(u_int regProgID, u_char regProcID, 
 {  {
         int n = 1;          int n = 1;
         rpc_srv_t *srv = NULL;          rpc_srv_t *srv = NULL;
        io_sockaddr_t sa;        io_sockaddr_t sa = IO_SOCKADDR_INIT;
   
         if (!concurentClients || !regProgID) {          if (!concurentClients || !regProgID) {
                 rpc_SetErr(EINVAL, "Invalid parameters can`t init RPC server");                  rpc_SetErr(EINVAL, "Invalid parameters can`t init RPC server");
Line 702  rpc_srv_initServer(u_int regProgID, u_char regProcID,  Line 709  rpc_srv_initServer(u_int regProgID, u_char regProcID, 
         if (netBuf < RPC_MIN_BUFSIZ)          if (netBuf < RPC_MIN_BUFSIZ)
                 netBuf = BUFSIZ;                  netBuf = BUFSIZ;
         else          else
                netBuf = io_align(netBuf, 1);      /* align netBuf length */                netBuf = io_align(netBuf, 2);      /* align netBuf length */
   
 #ifdef HAVE_SRANDOMDEV  #ifdef HAVE_SRANDOMDEV
         srandomdev();          srandomdev();
Line 712  rpc_srv_initServer(u_int regProgID, u_char regProcID,  Line 719  rpc_srv_initServer(u_int regProgID, u_char regProcID, 
         srandom((time(&tim) ^ getpid()));          srandom((time(&tim) ^ getpid()));
 #endif  #endif
   
        srv = malloc(sizeof(rpc_srv_t));        srv = io_malloc(sizeof(rpc_srv_t));
         if (!srv) {          if (!srv) {
                 LOGERR;                  LOGERR;
                 return NULL;                  return NULL;
Line 734  rpc_srv_initServer(u_int regProgID, u_char regProcID,  Line 741  rpc_srv_initServer(u_int regProgID, u_char regProcID, 
         srv->srv_root = schedBegin();          srv->srv_root = schedBegin();
         if (!srv->srv_root) {          if (!srv->srv_root) {
                 rpc_SetErr(sched_GetErrno(), "%s", sched_GetError());                  rpc_SetErr(sched_GetErrno(), "%s", sched_GetError());
                free(srv);                io_free(srv);
                 return NULL;                  return NULL;
         }          }
   
Line 743  rpc_srv_initServer(u_int regProgID, u_char regProcID,  Line 750  rpc_srv_initServer(u_int regProgID, u_char regProcID, 
         if (!srv->srv_clients) {          if (!srv->srv_clients) {
                 rpc_SetErr(io_GetErrno(), "%s", io_GetError());                  rpc_SetErr(io_GetErrno(), "%s", io_GetError());
                 schedEnd(&srv->srv_root);                  schedEnd(&srv->srv_root);
                free(srv);                io_free(srv);
                 return NULL;                  return NULL;
         }          }
   
Line 753  rpc_srv_initServer(u_int regProgID, u_char regProcID,  Line 760  rpc_srv_initServer(u_int regProgID, u_char regProcID, 
                 LOGERR;                  LOGERR;
                 io_arrayDestroy(&srv->srv_clients);                  io_arrayDestroy(&srv->srv_clients);
                 schedEnd(&srv->srv_root);                  schedEnd(&srv->srv_root);
                free(srv);                io_free(srv);
                 return NULL;                  return NULL;
         }          }
         if (setsockopt(srv->srv_server.cli_sock, SOL_SOCKET, SO_REUSEADDR, &n, sizeof n) == -1) {          if (setsockopt(srv->srv_server.cli_sock, SOL_SOCKET, SO_REUSEADDR, &n, sizeof n) == -1) {
Line 782  err: /* error condition */ Line 789  err: /* error condition */
         close(srv->srv_server.cli_sock);          close(srv->srv_server.cli_sock);
         io_arrayDestroy(&srv->srv_clients);          io_arrayDestroy(&srv->srv_clients);
         schedEnd(&srv->srv_root);          schedEnd(&srv->srv_root);
        free(srv);        io_free(srv);
         return NULL;          return NULL;
 }  }
   
Line 805  rpc_srv_endServer(rpc_srv_t ** __restrict psrv) Line 812  rpc_srv_endServer(rpc_srv_t ** __restrict psrv)
         (*psrv)->srv_kill = 1;          (*psrv)->srv_kill = 1;
         sleep(RPC_SCHED_POLLING);          sleep(RPC_SCHED_POLLING);
   
        free(*psrv);        io_free(*psrv);
         *psrv = NULL;          *psrv = NULL;
 }  }
   
Line 853  rpc_srv_loopServer(rpc_srv_t * __restrict srv) Line 860  rpc_srv_loopServer(rpc_srv_t * __restrict srv)
                         close(c->cli_sock);                          close(c->cli_sock);
   
                         schedCancelby(srv->srv_root, taskMAX, CRITERIA_ARG, c, NULL);                          schedCancelby(srv->srv_root, taskMAX, CRITERIA_ARG, c, NULL);
                           io_freeVars(&RPC_RETVARS(c));
                         AIT_FREE_VAL(&c->cli_buf);                          AIT_FREE_VAL(&c->cli_buf);
                 }                  }
                 io_arrayDel(srv->srv_clients, i, 42);                  io_arrayDel(srv->srv_clients, i, 42);
Line 865  rpc_srv_loopServer(rpc_srv_t * __restrict srv) Line 873  rpc_srv_loopServer(rpc_srv_t * __restrict srv)
         TAILQ_FOREACH_SAFE(f, &srv->srv_funcs, func_node, tmp) {          TAILQ_FOREACH_SAFE(f, &srv->srv_funcs, func_node, tmp) {
                 TAILQ_REMOVE(&srv->srv_funcs, f, func_node);                  TAILQ_REMOVE(&srv->srv_funcs, f, func_node);
   
                 io_freeVars(&f->func_vars);  
                 AIT_FREE_VAL(&f->func_name);                  AIT_FREE_VAL(&f->func_name);
                free(f);                io_free(f);
         }          }
   
         schedEnd(&srv->srv_root);          schedEnd(&srv->srv_root);
Line 878  rpc_srv_loopServer(rpc_srv_t * __restrict srv) Line 885  rpc_srv_loopServer(rpc_srv_t * __restrict srv)
 /*  /*
  * rpc_srv_execCall() Execute registered call from RPC server   * rpc_srv_execCall() Execute registered call from RPC server
  *   *
 * @call = Register RPC call * @cli = RPC client
  * @rpc = IN RPC call structure   * @rpc = IN RPC call structure
    * @funcname = Execute RPC function
  * @args = IN RPC calling arguments from RPC client   * @args = IN RPC calling arguments from RPC client
  * return: -1 error, !=-1 ok   * return: -1 error, !=-1 ok
  */   */
 int  int
rpc_srv_execCall(rpc_func_t * __restrict call, struct tagRPCCall * __restrict rpc, rpc_srv_execCall(rpc_cli_t * __restrict cli, struct tagRPCCall * __restrict rpc, 
                array_t * __restrict args)                ait_val_t funcname, array_t * __restrict args)
 {  {
         rpc_callback_t func;          rpc_callback_t func;
   
        if (!call || !rpc || !call->func_parent || !AIT_ADDR(&call->func_name)) {        if (!cli || !rpc || !AIT_ADDR(&funcname)) {
                 rpc_SetErr(EINVAL, "Invalid parameter can`t exec function");                  rpc_SetErr(EINVAL, "Invalid parameter can`t exec function");
                 return -1;                  return -1;
         }          }
   
        func = AIT_GET_LIKE(&call->func_name, rpc_callback_t);        func = AIT_GET_LIKE(&funcname, rpc_callback_t);
        return func(call, rpc, args);        return func(cli, rpc, args);
 }  }

Removed from v.1.9.2.28  
changed lines
  Added in v.1.11.2.3


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