[BACK]Return to srv.c CVS log [TXT] [DIR] Up to [ELWIX - Embedded LightWeight unIX -] / libaitrpc / src

Diff for /libaitrpc/src/srv.c between versions 1.12.2.5 and 1.14

version 1.12.2.5, 2012/11/19 10:29:02 version 1.14, 2013/03/07 23:10:50
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, 2011, 2012Copyright 2004, 2005, 2006, 2007, 2008, 2009, 2010, 2011, 2012, 2013
         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 67  static sched_task_func_t cbProto[SOCK_RAW + 1][4] = { Line 67  static sched_task_func_t cbProto[SOCK_RAW + 1][4] = {
 };  };
   
   
   inline void
   rpc_freeCli(rpc_cli_t * __restrict c)
   {
           rpc_srv_t *s = c->cli_parent;
   
           schedCancelby(s->srv_root, taskMAX, CRITERIA_ARG, c, NULL);
   
           /* free buffer */
           AIT_FREE_VAL(&c->cli_buf);
   
           array_Del(s->srv_clients, c->cli_id, 0);
           if (c)
                   e_free(c);
   }
   
   
 static inline int  static inline int
_check4freeslot(rpc_srv_t * __restrict srv, io_sockaddr_t * __restrict sa)_check4freeslot(rpc_srv_t * __restrict srv, sockaddr_t * __restrict sa)
 {  {
         rpc_cli_t *c = NULL;          rpc_cli_t *c = NULL;
         register int i;          register int i;
   
         /* check free slots for connect */          /* check free slots for connect */
        for (i = 0; i < io_arraySize(srv->srv_clients) &&         for (i = 0; i < array_Size(srv->srv_clients) && 
                        (c = io_array(srv->srv_clients, i, rpc_cli_t*)); i++)                        (c = array(srv->srv_clients, i, rpc_cli_t*)); i++)
                 /* check for duplicates */                  /* check for duplicates */
                if (sa && !io_addrcmp(&c->cli_sa, sa, 42))                if (sa && !e_addrcmp(&c->cli_sa, sa, 42))
                         break;                          break;
        if (i >= io_arraySize(srv->srv_clients))        if (i >= array_Size(srv->srv_clients))
                 return -1;      /* no more free slots! */                  return -1;      /* no more free slots! */
   
         return i;          return i;
 }  }
   
 static rpc_cli_t *  static rpc_cli_t *
_allocClient(rpc_srv_t * __restrict srv, io_sockaddr_t * __restrict sa)_allocClient(rpc_srv_t * __restrict srv, sockaddr_t * __restrict sa)
 {  {
         rpc_cli_t *c = NULL;          rpc_cli_t *c = NULL;
         int n;          int n;
Line 95  _allocClient(rpc_srv_t * __restrict srv, io_sockaddr_t Line 111  _allocClient(rpc_srv_t * __restrict srv, io_sockaddr_t
         if (n == -1)          if (n == -1)
                 return NULL;                  return NULL;
         else          else
                c = io_array(srv->srv_clients, n, rpc_cli_t*);                c = array(srv->srv_clients, n, rpc_cli_t*);
   
         if (!c) {          if (!c) {
                c = io_malloc(sizeof(rpc_cli_t));                c = e_malloc(sizeof(rpc_cli_t));
                 if (!c) {                  if (!c) {
                         LOGERR;                          LOGERR;
                         srv->srv_kill = 1;                          srv->srv_kill = 1;
                         return NULL;                          return NULL;
                 } else {                  } else {
                         memset(c, 0, sizeof(rpc_cli_t));                          memset(c, 0, sizeof(rpc_cli_t));
                        io_arraySet(srv->srv_clients, n, c);                        array_Set(srv->srv_clients, n, c);
                         c->cli_id = n;                          c->cli_id = n;
                         c->cli_parent = srv;                          c->cli_parent = srv;
                 }                  }
   
                 /* alloc empty buffer */                  /* alloc empty buffer */
                AIT_SET_BUF2(&c->cli_buf, 0, srv->srv_netbuf);                AIT_SET_BUFSIZ(&c->cli_buf, 0, srv->srv_netbuf);
         }          }
   
         return c;          return c;
Line 121  _allocClient(rpc_srv_t * __restrict srv, io_sockaddr_t Line 137  _allocClient(rpc_srv_t * __restrict srv, io_sockaddr_t
 static void *  static void *
 freeClient(sched_task_t *task)  freeClient(sched_task_t *task)
 {  {
        rpc_cli_t *c = TASK_ARG(task);        rpc_freeCli(TASK_ARG(task));
        rpc_srv_t *s = c->cli_parent; 
   
         schedCancelby(TASK_ROOT(task), taskMAX, CRITERIA_ARG, TASK_ARG(task), NULL);  
   
         /* free buffer */  
         AIT_FREE_VAL(&c->cli_buf);  
   
         io_arrayDel(s->srv_clients, c->cli_id, 0);  
         if (c)  
                 io_free(c);  
         return NULL;          return NULL;
 }  }
   
 static void *  static void *
 closeClient(sched_task_t *task)  closeClient(sched_task_t *task)
 {  {
        rpc_cli_t *c = TASK_ARG(task);        int sock = ((rpc_cli_t*) TASK_ARG(task))->cli_sock;
        rpc_srv_t *s = c->cli_parent; 
   
        schedCancelby(TASK_ROOT(task), taskMAX, CRITERIA_ARG, TASK_ARG(task), NULL);        rpc_freeCli(TASK_ARG(task));
   
         /* close client socket */          /* close client socket */
        if (TASK_VAL(task))        shutdown(sock, SHUT_RDWR);
                shutdown(c->cli_sock, SHUT_RDWR);        close(sock);
        close(c->cli_sock); 
 
        /* free buffer */ 
        AIT_FREE_VAL(&c->cli_buf); 
 
        io_arrayDel(s->srv_clients, c->cli_id, 0); 
        if (c) 
                io_free(c); 
         return NULL;          return NULL;
 }  }
   
Line 178  txPacket(sched_task_t *task) Line 176  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(RPC_RETVARS(c)));                        rpc->call_argc = htons(array_Size(RPC_RETVARS(c)));
                         /* Go Encapsulate variables */                          /* Go Encapsulate variables */
                        ret = io_vars2buffer(buf + wlen, sizeof buf - wlen, RPC_RETVARS(c));                        ret = ait_vars2buffer(buf + wlen, sizeof buf - wlen, RPC_RETVARS(c));
                         /* Free return values */                          /* Free return values */
                        io_freeVars(&c->cli_vars);                        ait_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 204  txPacket(sched_task_t *task) Line 202  txPacket(sched_task_t *task)
         if (ret == -1 || ret != wlen) {          if (ret == -1 || ret != wlen) {
                 /* close connection */                  /* close connection */
                 schedEvent(TASK_ROOT(task), cbProto[s->srv_proto][CB_CLOSECLIENT],                   schedEvent(TASK_ROOT(task), cbProto[s->srv_proto][CB_CLOSECLIENT], 
                                c, 42, NULL, 0);                                TASK_ARG(task), 0, NULL, 0);
         }          }
   
         return NULL;          return NULL;
Line 223  execCall(sched_task_t *task) Line 221  execCall(sched_task_t *task)
   
         /* Go decapsulate variables ... */          /* Go decapsulate variables ... */
         if (argc) {          if (argc) {
                arr = io_buffer2vars(buf + sizeof(struct tagRPCCall),                 arr = ait_buffer2vars(buf + sizeof(struct tagRPCCall), 
                                 AIT_LEN(&c->cli_buf) - TASK_VAL(task) - sizeof(struct tagRPCCall),                                   AIT_LEN(&c->cli_buf) - TASK_VAL(task) - sizeof(struct tagRPCCall), 
                                 argc, 42);                                  argc, 42);
                 if (!arr) {                  if (!arr) {
                        rpc_SetErr(ERPCMISMATCH, "#%d - %s", io_GetErrno(), io_GetError());                        rpc_SetErr(ERPCMISMATCH, "#%d - %s", elwix_GetErrno(), elwix_GetError());
                         rpc->call_argc ^= rpc->call_argc;                          rpc->call_argc ^= rpc->call_argc;
                         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);
Line 252  execCall(sched_task_t *task) Line 250  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_freeVars(&c->cli_vars);                                ait_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(RPC_RETVARS(c)));                                rpc->call_argc = htons(array_Size(RPC_RETVARS(c)));
                         }                          }
                 }                  }
         }          }
   
        io_arrayDestroy(&arr);        array_Destroy(&arr);
         return NULL;          return NULL;
 }  }
   
Line 281  rxPacket(sched_task_t *task) Line 279  rxPacket(sched_task_t *task)
         if (rlen < 1) {          if (rlen < 1) {
                 /* close connection */                  /* close connection */
                 schedEvent(TASK_ROOT(task), cbProto[s->srv_proto][CB_CLOSECLIENT],                   schedEvent(TASK_ROOT(task), cbProto[s->srv_proto][CB_CLOSECLIENT], 
                                c, 42, NULL, 0);                                TASK_ARG(task), 0, NULL, 0);
                 return NULL;                  return NULL;
         } else {          } else {
                 rlen += off;    /* add reminded bytes from previous rxPacket, if exists! */                  rlen += off;    /* add reminded bytes from previous rxPacket, if exists! */
Line 352  acceptClients(sched_task_t *task) Line 350  acceptClients(sched_task_t *task)
 {  {
         rpc_srv_t *srv = TASK_ARG(task);          rpc_srv_t *srv = TASK_ARG(task);
         rpc_cli_t *c = NULL;          rpc_cli_t *c = NULL;
        socklen_t salen = sizeof(io_sockaddr_t);        socklen_t salen = sizeof(sockaddr_t);
   
         c = _allocClient(srv, NULL);          c = _allocClient(srv, NULL);
         if (!c)          if (!c)
Line 363  acceptClients(sched_task_t *task) Line 361  acceptClients(sched_task_t *task)
         if (c->cli_sock == -1) {          if (c->cli_sock == -1) {
                 LOGERR;                  LOGERR;
                 AIT_FREE_VAL(&c->cli_buf);                  AIT_FREE_VAL(&c->cli_buf);
                io_arrayDel(srv->srv_clients, c->cli_id, 42);                array_Del(srv->srv_clients, c->cli_id, 42);
                 goto end;                  goto end;
         } else          } else
                 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);
Line 387  txUDPPacket(sched_task_t *task) Line 385  txUDPPacket(sched_task_t *task)
         int ret, wlen = sizeof(struct tagRPCCall);          int ret, wlen = sizeof(struct tagRPCCall);
         struct timespec ts = { DEF_RPC_TIMEOUT, 0 };          struct timespec ts = { DEF_RPC_TIMEOUT, 0 };
   
        schedCancelby(TASK_ROOT(task), taskTIMER, CRITERIA_DATA, c, NULL);        schedCancelby(TASK_ROOT(task), taskTIMER, CRITERIA_DATA, TASK_ARG(task), NULL);
         schedTimer(TASK_ROOT(task), cbProto[s->srv_proto][CB_CLOSECLIENT],           schedTimer(TASK_ROOT(task), cbProto[s->srv_proto][CB_CLOSECLIENT], 
                        c, ts, c, 0);                         TASK_ARG(task), ts, TASK_ARG(task), 0);
   
         /* copy RPC header */          /* copy RPC header */
         memcpy(buf, TASK_DATA(task), wlen);          memcpy(buf, TASK_DATA(task), wlen);
Line 402  txUDPPacket(sched_task_t *task) Line 400  txUDPPacket(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(RPC_RETVARS(c)));                        rpc->call_argc = htons(array_Size(RPC_RETVARS(c)));
                         /* Go Encapsulate variables */                          /* Go Encapsulate variables */
                        ret = io_vars2buffer(buf + wlen, sizeof buf - wlen, RPC_RETVARS(c));                        ret = ait_vars2buffer(buf + wlen, sizeof buf - wlen, RPC_RETVARS(c));
                         /* Free return values */                          /* Free return values */
                        io_freeVars(&c->cli_vars);                        ait_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 429  txUDPPacket(sched_task_t *task) Line 427  txUDPPacket(sched_task_t *task)
         if (ret == -1 || ret != wlen) {          if (ret == -1 || ret != wlen) {
                 /* close connection */                  /* close connection */
                 schedEvent(TASK_ROOT(task), cbProto[s->srv_proto][CB_CLOSECLIENT],                   schedEvent(TASK_ROOT(task), cbProto[s->srv_proto][CB_CLOSECLIENT], 
                                c, 42, NULL, 0);                                 TASK_ARG(task), 0, NULL, 0);
         }          }
   
         return NULL;          return NULL;
Line 440  rxUDPPacket(sched_task_t *task) Line 438  rxUDPPacket(sched_task_t *task)
 {  {
         rpc_srv_t *srv = TASK_ARG(task);          rpc_srv_t *srv = TASK_ARG(task);
         rpc_cli_t *c = NULL;          rpc_cli_t *c = NULL;
         socklen_t salen = sizeof(io_sockaddr_t);  
         int len, rlen, noreply;          int len, rlen, noreply;
         u_short crc, off = 0;          u_short crc, off = 0;
         u_char buf[USHRT_MAX + 1];          u_char buf[USHRT_MAX + 1];
         struct tagRPCCall *rpc = (struct tagRPCCall*) buf;          struct tagRPCCall *rpc = (struct tagRPCCall*) buf;
        io_sockaddr_t sa;        sockaddr_t sa;
         socklen_t salen;
         struct timespec ts = { DEF_RPC_TIMEOUT, 0 };          struct timespec ts = { DEF_RPC_TIMEOUT, 0 };
   
         /* receive connect packet */          /* receive connect packet */
           salen = sa.ss.ss_len = sizeof(sockaddr_t);
         rlen = recvfrom(TASK_FD(task), buf, sizeof buf, 0, &sa.sa, &salen);          rlen = recvfrom(TASK_FD(task), buf, sizeof buf, 0, &sa.sa, &salen);
         if (rlen < 1)          if (rlen < 1)
                 goto end;                  goto end;
Line 457  rxUDPPacket(sched_task_t *task) Line 456  rxUDPPacket(sched_task_t *task)
         if (!c)          if (!c)
                 goto end;                  goto end;
         else {          else {
                /* armed timer for close stateless connection */                c->cli_sock = TASK_FD(task);
                 memcpy(&c->cli_sa, &sa, sizeof c->cli_sa);
                 memcpy(AIT_GET_BUF(&c->cli_buf), buf, AIT_LEN(&c->cli_buf));                  memcpy(AIT_GET_BUF(&c->cli_buf), buf, AIT_LEN(&c->cli_buf));
                   /* armed timer for close stateless connection */
                 schedCancelby(TASK_ROOT(task), taskTIMER, CRITERIA_DATA, c, NULL);                  schedCancelby(TASK_ROOT(task), taskTIMER, CRITERIA_DATA, c, NULL);
                 schedTimer(TASK_ROOT(task), cbProto[srv->srv_proto][CB_CLOSECLIENT],                   schedTimer(TASK_ROOT(task), cbProto[srv->srv_proto][CB_CLOSECLIENT], 
                                 c, ts, c, 0);                                  c, ts, c, 0);
Line 503  rxUDPPacket(sched_task_t *task) Line 504  rxUDPPacket(sched_task_t *task)
                         rpc->call_rep.eno = RPC_ERROR(errno);                          rpc->call_rep.eno = RPC_ERROR(errno);
                 } else {                  } else {
                         /* execute RPC call */                          /* execute RPC call */
                        schedEvent(TASK_ROOT(task), execCall, TASK_ARG(task), off, NULL, 0);                        schedEvent(TASK_ROOT(task), execCall, c, off, NULL, 0);
                 }                  }
   
                 /* send RPC reply */                  /* send RPC reply */
                 if (!noreply)                  if (!noreply)
                         schedWrite(TASK_ROOT(task), cbProto[srv->srv_proto][CB_TXPACKET],                           schedWrite(TASK_ROOT(task), cbProto[srv->srv_proto][CB_TXPACKET], 
                                        TASK_ARG(task), TASK_FD(task), rpc, len);                                        c, TASK_FD(task), rpc, len);
   
                 off += len;                  off += len;
         } while (rlen > 0);          } while (rlen > 0);
Line 521  end: Line 522  end:
   
 /* ------------------------------------------------------ */  /* ------------------------------------------------------ */
   
static void *inline void
closeBLOBClient(sched_task_t *task)rpc_freeBLOBCli(rpc_cli_t * __restrict c)
 {  {
         rpc_cli_t *c = TASK_ARG(task);  
         rpc_srv_t *s = c->cli_parent;          rpc_srv_t *s = c->cli_parent;
   
        schedCancelby(TASK_ROOT(task), taskMAX, CRITERIA_ARG, TASK_ARG(task), NULL);        schedCancelby(s->srv_blob.root, taskMAX, CRITERIA_ARG, c, NULL);
   
         /* close client socket */  
         if (TASK_VAL(task))  
                 shutdown(c->cli_sock, SHUT_RDWR);  
         close(c->cli_sock);  
   
         /* free buffer */          /* free buffer */
         AIT_FREE_VAL(&c->cli_buf);          AIT_FREE_VAL(&c->cli_buf);
   
        io_arrayDel(s->srv_blob.clients, c->cli_id, 0);        array_Del(s->srv_blob.clients, c->cli_id, 0);
         if (c)          if (c)
                io_free(c);                e_free(c);
 }
 
 
 static void *
 closeBLOBClient(sched_task_t *task)
 {
         int sock = ((rpc_cli_t*) TASK_ARG(task))->cli_sock;
 
         rpc_freeBLOBCli(TASK_ARG(task));
 
         /* close client socket */
         shutdown(sock, SHUT_RDWR);
         close(sock);
         return NULL;          return NULL;
 }  }
   
Line 666  acceptBLOBClients(sched_task_t *task) Line 674  acceptBLOBClients(sched_task_t *task)
         rpc_srv_t *srv = TASK_ARG(task);          rpc_srv_t *srv = TASK_ARG(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(sockaddr_t);
 #ifdef TCP_NOPUSH  #ifdef TCP_NOPUSH
         int n = 1;          int n = 1;
 #endif  #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 < array_Size(srv->srv_blob.clients) && 
                        (c = io_array(srv->srv_blob.clients, i, rpc_cli_t*)); i++);                        (c = 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 = io_malloc(sizeof(rpc_cli_t));        c = e_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;
                 return NULL;                  return NULL;
         } else {          } else {
                 memset(c, 0, sizeof(rpc_cli_t));                  memset(c, 0, sizeof(rpc_cli_t));
                io_arraySet(srv->srv_blob.clients, i, c);                array_Set(srv->srv_blob.clients, i, c);
                 c->cli_id = i;                  c->cli_id = i;
                 c->cli_parent = srv;                  c->cli_parent = srv;
         }          }
   
         /* alloc empty buffer */          /* alloc empty buffer */
        AIT_SET_BUF2(&c->cli_buf, 0, srv->srv_netbuf);        AIT_SET_BUFSIZ(&c->cli_buf, 0, srv->srv_netbuf);
   
         /* accept client */          /* accept client */
         c->cli_sock = accept(TASK_FD(task), &c->cli_sa.sa, &salen);          c->cli_sock = accept(TASK_FD(task), &c->cli_sa.sa, &salen);
         if (c->cli_sock == -1) {          if (c->cli_sock == -1) {
                 LOGERR;                  LOGERR;
                 AIT_FREE_VAL(&c->cli_buf);                  AIT_FREE_VAL(&c->cli_buf);
                io_arrayDel(srv->srv_blob.clients, i, 42);                array_Del(srv->srv_blob.clients, i, 42);
                 goto end;                  goto end;
         } else {          } else {
 #ifdef TCP_NOPUSH  #ifdef TCP_NOPUSH
Line 743  rpc_srv_initBLOBServer(rpc_srv_t * __restrict srv, u_s Line 751  rpc_srv_initBLOBServer(rpc_srv_t * __restrict srv, u_s
   
         srv->srv_blob.server.cli_parent = srv;          srv->srv_blob.server.cli_parent = srv;
   
        memcpy(&srv->srv_blob.server.cli_sa, &srv->srv_server.cli_sa, sizeof(io_sockaddr_t));        memcpy(&srv->srv_blob.server.cli_sa, &srv->srv_server.cli_sa, sizeof(sockaddr_t));
         switch (srv->srv_blob.server.cli_sa.sa.sa_family) {          switch (srv->srv_blob.server.cli_sa.sa.sa_family) {
                 case AF_INET:                  case AF_INET:
                         srv->srv_blob.server.cli_sa.sin.sin_port =                           srv->srv_blob.server.cli_sa.sin.sin_port = 
Line 800  rpc_srv_initBLOBServer(rpc_srv_t * __restrict srv, u_s Line 808  rpc_srv_initBLOBServer(rpc_srv_t * __restrict srv, u_s
   
   
         /* allocate pool for concurent blob clients */          /* allocate pool for concurent blob clients */
        srv->srv_blob.clients = io_arrayInit(io_arraySize(srv->srv_clients));        srv->srv_blob.clients = array_Init(array_Size(srv->srv_clients));
         if (!srv->srv_blob.clients) {          if (!srv->srv_blob.clients) {
                rpc_SetErr(io_GetErrno(), "%s", io_GetError());                rpc_SetErr(elwix_GetErrno(), "%s", elwix_GetError());
                 close(srv->srv_blob.server.cli_sock);                  close(srv->srv_blob.server.cli_sock);
                 AIT_FREE_VAL(&srv->srv_blob.dir);                  AIT_FREE_VAL(&srv->srv_blob.dir);
                 return -1;                  return -1;
Line 812  rpc_srv_initBLOBServer(rpc_srv_t * __restrict srv, u_s Line 820  rpc_srv_initBLOBServer(rpc_srv_t * __restrict srv, u_s
         srv->srv_blob.root = schedBegin();          srv->srv_blob.root = schedBegin();
         if (!srv->srv_blob.root) {          if (!srv->srv_blob.root) {
                 rpc_SetErr(sched_GetErrno(), "%s", sched_GetError());                  rpc_SetErr(sched_GetErrno(), "%s", sched_GetError());
                io_arrayDestroy(&srv->srv_blob.clients);                array_Destroy(&srv->srv_blob.clients);
                 close(srv->srv_blob.server.cli_sock);                  close(srv->srv_blob.server.cli_sock);
                 AIT_FREE_VAL(&srv->srv_blob.dir);                  AIT_FREE_VAL(&srv->srv_blob.dir);
                 return -1;                  return -1;
Line 855  rpc_srv_loopBLOBServer(rpc_srv_t * __restrict srv) Line 863  rpc_srv_loopBLOBServer(rpc_srv_t * __restrict srv)
                 return -1;                  return -1;
         }          }
   
        if (listen(srv->srv_blob.server.cli_sock, io_arraySize(srv->srv_blob.clients)) == -1) {        if (listen(srv->srv_blob.server.cli_sock, array_Size(srv->srv_blob.clients)) == -1) {
                 LOGERR;                  LOGERR;
                 return -1;                  return -1;
         }          }
Line 871  rpc_srv_loopBLOBServer(rpc_srv_t * __restrict srv) Line 879  rpc_srv_loopBLOBServer(rpc_srv_t * __restrict srv)
         schedRun(srv->srv_blob.root, &srv->srv_blob.kill);          schedRun(srv->srv_blob.root, &srv->srv_blob.kill);
   
         /* close all clients connections & server socket */          /* close all clients connections & server socket */
        for (i = 0; i < io_arraySize(srv->srv_blob.clients); i++) {        for (i = 0; i < array_Size(srv->srv_blob.clients); i++) {
                c = io_array(srv->srv_blob.clients, i, rpc_cli_t*);                c = array(srv->srv_blob.clients, i, rpc_cli_t*);
                 if (c) {                  if (c) {
                         shutdown(c->cli_sock, SHUT_RDWR);                          shutdown(c->cli_sock, SHUT_RDWR);
                         close(c->cli_sock);                          close(c->cli_sock);
Line 880  rpc_srv_loopBLOBServer(rpc_srv_t * __restrict srv) Line 888  rpc_srv_loopBLOBServer(rpc_srv_t * __restrict srv)
                         schedCancelby(srv->srv_blob.root, taskMAX, CRITERIA_ARG, c, NULL);                          schedCancelby(srv->srv_blob.root, taskMAX, CRITERIA_ARG, c, NULL);
                         AIT_FREE_VAL(&c->cli_buf);                          AIT_FREE_VAL(&c->cli_buf);
                 }                  }
                io_arrayDel(srv->srv_blob.clients, i, 42);                array_Del(srv->srv_blob.clients, i, 42);
         }          }
        io_arrayDestroy(&srv->srv_blob.clients);        array_Destroy(&srv->srv_blob.clients);
   
         close(srv->srv_blob.server.cli_sock);          close(srv->srv_blob.server.cli_sock);
   
Line 891  rpc_srv_loopBLOBServer(rpc_srv_t * __restrict srv) Line 899  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);
                io_free(b);                e_free(b);
         }          }
   
         schedEnd(&srv->srv_blob.root);          schedEnd(&srv->srv_blob.root);
Line 918  rpc_srv_initServer(u_int regProgID, u_char regProcID,  Line 926  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_INIT;        sockaddr_t sa = E_SOCKADDR_INIT;
   
         if (!concurentClients || !regProgID || (proto < 0 || proto > SOCK_DGRAM)) {          if (!concurentClients || !regProgID || (proto < 0 || proto > SOCK_DGRAM)) {
                 rpc_SetErr(EINVAL, "Invalid parameters can`t init RPC server");                  rpc_SetErr(EINVAL, "Invalid parameters can`t init RPC server");
                 return NULL;                  return NULL;
         }          }
        if (!proto)        if (!e_gethostbyname(csHost, Port, &sa))
                proto = SOCK_STREAM; 
        if (!io_gethostbyname(csHost, Port, &sa)) 
                 return NULL;                  return NULL;
         if (!Port)          if (!Port)
                 Port = RPC_DEFPORT;                  Port = RPC_DEFPORT;
           if (!proto)
                   proto = SOCK_STREAM;
         if (netBuf < RPC_MIN_BUFSIZ)          if (netBuf < RPC_MIN_BUFSIZ)
                 netBuf = BUFSIZ;                  netBuf = BUFSIZ;
         else          else
                netBuf = io_align(netBuf, 2);    /* align netBuf length */                netBuf = E_ALIGN(netBuf, 2);    /* align netBuf length */
   
 #ifdef HAVE_SRANDOMDEV  #ifdef HAVE_SRANDOMDEV
         srandomdev();          srandomdev();
Line 943  rpc_srv_initServer(u_int regProgID, u_char regProcID,  Line 951  rpc_srv_initServer(u_int regProgID, u_char regProcID, 
         srandom((time(&tim) ^ getpid()));          srandom((time(&tim) ^ getpid()));
 #endif  #endif
   
        srv = io_malloc(sizeof(rpc_srv_t));        srv = e_malloc(sizeof(rpc_srv_t));
         if (!srv) {          if (!srv) {
                 LOGERR;                  LOGERR;
                 return NULL;                  return NULL;
Line 969  rpc_srv_initServer(u_int regProgID, u_char regProcID,  Line 977  rpc_srv_initServer(u_int regProgID, u_char regProcID, 
         if (!srv->srv_root) {          if (!srv->srv_root) {
                 rpc_SetErr(sched_GetErrno(), "%s", sched_GetError());                  rpc_SetErr(sched_GetErrno(), "%s", sched_GetError());
                 pthread_mutex_destroy(&srv->srv_funcs.mtx);                  pthread_mutex_destroy(&srv->srv_funcs.mtx);
                io_free(srv);                e_free(srv);
                 return NULL;                  return NULL;
         }          }
   
         /* init pool for clients */          /* init pool for clients */
        srv->srv_clients = io_arrayInit(concurentClients);        srv->srv_clients = array_Init(concurentClients);
         if (!srv->srv_clients) {          if (!srv->srv_clients) {
                rpc_SetErr(io_GetErrno(), "%s", io_GetError());                rpc_SetErr(elwix_GetErrno(), "%s", elwix_GetError());
                 schedEnd(&srv->srv_root);                  schedEnd(&srv->srv_root);
                 pthread_mutex_destroy(&srv->srv_funcs.mtx);                  pthread_mutex_destroy(&srv->srv_funcs.mtx);
                io_free(srv);                e_free(srv);
                 return NULL;                  return NULL;
         }          }
   
Line 987  rpc_srv_initServer(u_int regProgID, u_char regProcID,  Line 995  rpc_srv_initServer(u_int regProgID, u_char regProcID, 
         srv->srv_server.cli_sock = socket(srv->srv_server.cli_sa.sa.sa_family, srv->srv_proto, 0);          srv->srv_server.cli_sock = socket(srv->srv_server.cli_sa.sa.sa_family, srv->srv_proto, 0);
         if (srv->srv_server.cli_sock == -1) {          if (srv->srv_server.cli_sock == -1) {
                 LOGERR;                  LOGERR;
                io_arrayDestroy(&srv->srv_clients);                array_Destroy(&srv->srv_clients);
                 schedEnd(&srv->srv_root);                  schedEnd(&srv->srv_root);
                 pthread_mutex_destroy(&srv->srv_funcs.mtx);                  pthread_mutex_destroy(&srv->srv_funcs.mtx);
                io_free(srv);                e_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 1019  rpc_srv_initServer(u_int regProgID, u_char regProcID,  Line 1027  rpc_srv_initServer(u_int regProgID, u_char regProcID, 
         return srv;          return srv;
 err:    /* error condition */  err:    /* error condition */
         close(srv->srv_server.cli_sock);          close(srv->srv_server.cli_sock);
        io_arrayDestroy(&srv->srv_clients);        array_Destroy(&srv->srv_clients);
         schedEnd(&srv->srv_root);          schedEnd(&srv->srv_root);
         pthread_mutex_destroy(&srv->srv_funcs.mtx);          pthread_mutex_destroy(&srv->srv_funcs.mtx);
        io_free(srv);        e_free(srv);
         return NULL;          return NULL;
 }  }
   
Line 1046  rpc_srv_endServer(rpc_srv_t ** __restrict psrv) Line 1054  rpc_srv_endServer(rpc_srv_t ** __restrict psrv)
         sleep(RPC_SCHED_POLLING);          sleep(RPC_SCHED_POLLING);
   
         pthread_mutex_destroy(&(*psrv)->srv_funcs.mtx);          pthread_mutex_destroy(&(*psrv)->srv_funcs.mtx);
        io_free(*psrv);        e_free(*psrv);
         *psrv = NULL;          *psrv = NULL;
 }  }
   
Line 1069  rpc_srv_loopServer(rpc_srv_t * __restrict srv) Line 1077  rpc_srv_loopServer(rpc_srv_t * __restrict srv)
                 return -1;                  return -1;
         }          }
   
        if (listen(srv->srv_server.cli_sock, io_arraySize(srv->srv_clients)) == -1) {        if (srv->srv_proto == SOCK_STREAM)
                LOGERR;                if (listen(srv->srv_server.cli_sock, array_Size(srv->srv_clients)) == -1) {
                return -1;                        LOGERR;
        }                        return -1;
                 }
   
         if (!schedRead(srv->srv_root, cbProto[srv->srv_proto][CB_ACCEPTCLIENT], srv,           if (!schedRead(srv->srv_root, cbProto[srv->srv_proto][CB_ACCEPTCLIENT], srv, 
                                 srv->srv_server.cli_sock, NULL, 0)) {                                  srv->srv_server.cli_sock, NULL, 0)) {
Line 1085  rpc_srv_loopServer(rpc_srv_t * __restrict srv) Line 1094  rpc_srv_loopServer(rpc_srv_t * __restrict srv)
         schedRun(srv->srv_root, &srv->srv_kill);          schedRun(srv->srv_root, &srv->srv_kill);
   
         /* close all clients connections & server socket */          /* close all clients connections & server socket */
        for (i = 0; i < io_arraySize(srv->srv_clients); i++) {        for (i = 0; i < array_Size(srv->srv_clients); i++) {
                c = io_array(srv->srv_clients, i, rpc_cli_t*);                c = array(srv->srv_clients, i, rpc_cli_t*);
                 if (c) {                  if (c) {
                         shutdown(c->cli_sock, SHUT_RDWR);                          shutdown(c->cli_sock, SHUT_RDWR);
                         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_freeVars(&RPC_RETVARS(c));
                         AIT_FREE_VAL(&c->cli_buf);                          AIT_FREE_VAL(&c->cli_buf);
                 }                  }
                io_arrayDel(srv->srv_clients, i, 42);                array_Del(srv->srv_clients, i, 42);
         }          }
        io_arrayDestroy(&srv->srv_clients);        array_Destroy(&srv->srv_clients);
   
         close(srv->srv_server.cli_sock);          close(srv->srv_server.cli_sock);
   
Line 1107  rpc_srv_loopServer(rpc_srv_t * __restrict srv) Line 1116  rpc_srv_loopServer(rpc_srv_t * __restrict srv)
                 SLIST_REMOVE_HEAD(&srv->srv_funcs, func_next);                  SLIST_REMOVE_HEAD(&srv->srv_funcs, func_next);
   
                 AIT_FREE_VAL(&f->func_name);                  AIT_FREE_VAL(&f->func_name);
                io_free(f);                e_free(f);
         }          }
         srv->srv_funcs.avlh_root = NULL;          srv->srv_funcs.avlh_root = NULL;
         RPC_FUNCS_UNLOCK(&srv->srv_funcs);          RPC_FUNCS_UNLOCK(&srv->srv_funcs);
Removed from v.1.12.2.5  
changed lines
  Added in v.1.14

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