[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.11.2.3 and 1.12.2.2

version 1.11.2.3, 2012/09/17 14:16:42 version 1.12.2.2, 2012/11/16 08:51:36
Line 46  SUCH DAMAGE. Line 46  SUCH DAMAGE.
 #include "global.h"  #include "global.h"
   
   
   /* SOCK_STREAM */
   static void *acceptClients(sched_task_t *);
   static void *closeClient(sched_task_t *);
   static void *rxPacket(sched_task_t *);
   static void *txPacket(sched_task_t *);
   
   /* SOCK_DGRAM */
   
   /* SOCK_RAW */
   
   static sched_task_func_t cbProto[SOCK_RAW + 1][4] = {
           { acceptClients, closeClient, rxPacket, txPacket },     /* SOCK_STREAM */
           { acceptClients, closeClient, rxPacket, txPacket },     /* SOCK_STREAM */
           { NULL, NULL, NULL, NULL }, 
           { NULL, NULL, NULL, NULL }                              /* SOCK_RAW */
   };
   
   
 static void *  static void *
 closeClient(sched_task_t *task)  closeClient(sched_task_t *task)
 {  {
Line 114  txPacket(sched_task_t *task) Line 132  txPacket(sched_task_t *task)
         ret = send(TASK_FD(task), buf, wlen, MSG_NOSIGNAL);          ret = send(TASK_FD(task), buf, wlen, MSG_NOSIGNAL);
         if (ret == -1 || ret != wlen) {          if (ret == -1 || ret != wlen) {
                 /* close connection */                  /* close connection */
                schedEvent(TASK_ROOT(task), closeClient, c, 42, NULL, 0);                schedEvent(TASK_ROOT(task), cbProto[s->srv_proto][CB_CLOSECLIENT], 
                                 c, 42, NULL, 0);
         }          }
   
         return NULL;          return NULL;
Line 135  execCall(sched_task_t *task) Line 154  execCall(sched_task_t *task)
         if (argc) {          if (argc) {
                 arr = io_buffer2vars(buf + sizeof(struct tagRPCCall),                   arr = io_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, 1);                                argc, 42);
                 if (!arr) {                  if (!arr) {
                         rpc_SetErr(ERPCMISMATCH, "#%d - %s", io_GetErrno(), io_GetError());                          rpc_SetErr(ERPCMISMATCH, "#%d - %s", io_GetErrno(), io_GetError());
                         rpc->call_argc ^= rpc->call_argc;                          rpc->call_argc ^= rpc->call_argc;
Line 185  rxPacket(sched_task_t *task) Line 204  rxPacket(sched_task_t *task)
         u_char *buf = AIT_GET_BUF(&c->cli_buf);          u_char *buf = AIT_GET_BUF(&c->cli_buf);
         struct tagRPCCall *rpc;          struct tagRPCCall *rpc;
   
        memset(buf, 0, AIT_LEN(&c->cli_buf));        if (!off)
                 memset(buf, 0, AIT_LEN(&c->cli_buf));
 //      else
 //              memmove(buf, buf + off, AIT_LEN(&c->cli_buf) - off);
         rlen = recv(TASK_FD(task), buf + off, AIT_LEN(&c->cli_buf) - off, 0);          rlen = recv(TASK_FD(task), buf + off, AIT_LEN(&c->cli_buf) - off, 0);
         if (rlen < 1) {          if (rlen < 1) {
                 /* close connection */                  /* close connection */
                schedEvent(TASK_ROOT(task), closeClient, c, 42, NULL, 0);                schedEvent(TASK_ROOT(task), cbProto[s->srv_proto][CB_CLOSECLIENT], 
                                 c, 42, 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 244  rxPacket(sched_task_t *task) Line 267  rxPacket(sched_task_t *task)
   
                 /* send RPC reply */                  /* send RPC reply */
                 if (!noreply)                  if (!noreply)
                        schedWrite(TASK_ROOT(task), txPacket, TASK_ARG(task), TASK_FD(task), rpc, len);                        schedWrite(TASK_ROOT(task), cbProto[s->srv_proto][CB_TXPACKET], 
                                         TASK_ARG(task), TASK_FD(task), rpc, len);
   
                 off += len;                  off += len;
         } while (rlen > 0);          } while (rlen > 0);
Line 292  acceptClients(sched_task_t *task) Line 316  acceptClients(sched_task_t *task)
         } 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);
   
        schedRead(TASK_ROOT(task), rxPacket, c, c->cli_sock, NULL, 0);        schedRead(TASK_ROOT(task), cbProto[srv->srv_proto][CB_RXPACKET], c, 
                         c->cli_sock, NULL, 0);
 end:  end:
         schedReadSelf(task);          schedReadSelf(task);
         return NULL;          return NULL;
Line 688  rpc_srv_loopBLOBServer(rpc_srv_t * __restrict srv) Line 713  rpc_srv_loopBLOBServer(rpc_srv_t * __restrict srv)
  * @netBuf = Network buffer length (min:512 bytes), if =0 == BUFSIZ (also meaning max RPC packet)   * @netBuf = Network buffer length (min:512 bytes), if =0 == BUFSIZ (also meaning max RPC packet)
  * @csHost = Host name or address for bind server, if NULL any address   * @csHost = Host name or address for bind server, if NULL any address
  * @Port = Port for bind server, if Port == 0 default port is selected   * @Port = Port for bind server, if Port == 0 default port is selected
    * @proto = Protocol, if == 0 choose SOCK_STREAM
  * return: NULL == error or !=NULL bind and created RPC server instance   * return: NULL == error or !=NULL bind and created RPC server instance
  */   */
 rpc_srv_t *  rpc_srv_t *
 rpc_srv_initServer(u_int regProgID, u_char regProcID, int concurentClients,   rpc_srv_initServer(u_int regProgID, u_char regProcID, int concurentClients, 
                int netBuf, const char *csHost, u_short Port)                int netBuf, const char *csHost, u_short Port, int proto)
 {  {
         int n = 1;          int n = 1;
         rpc_srv_t *srv = NULL;          rpc_srv_t *srv = NULL;
         io_sockaddr_t sa = IO_SOCKADDR_INIT;          io_sockaddr_t sa = IO_SOCKADDR_INIT;
   
        if (!concurentClients || !regProgID) {        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)
                   proto = SOCK_STREAM;
         if (!io_gethostbyname(csHost, Port, &sa))          if (!io_gethostbyname(csHost, Port, &sa))
                 return NULL;                  return NULL;
         if (!Port)          if (!Port)
Line 726  rpc_srv_initServer(u_int regProgID, u_char regProcID,  Line 754  rpc_srv_initServer(u_int regProgID, u_char regProcID, 
         } else          } else
                 memset(srv, 0, sizeof(rpc_srv_t));                  memset(srv, 0, sizeof(rpc_srv_t));
   
           srv->srv_proto = proto;
         srv->srv_netbuf = netBuf;          srv->srv_netbuf = netBuf;
         srv->srv_session.sess_version = RPC_VERSION;          srv->srv_session.sess_version = RPC_VERSION;
         srv->srv_session.sess_program = regProgID;          srv->srv_session.sess_program = regProgID;
Line 734  rpc_srv_initServer(u_int regProgID, u_char regProcID,  Line 763  rpc_srv_initServer(u_int regProgID, u_char regProcID, 
         srv->srv_server.cli_parent = srv;          srv->srv_server.cli_parent = srv;
         memcpy(&srv->srv_server.cli_sa, &sa, sizeof srv->srv_server.cli_sa);          memcpy(&srv->srv_server.cli_sa, &sa, sizeof srv->srv_server.cli_sa);
   
        /* init functions list */        /* init functions */
        TAILQ_INIT(&srv->srv_funcs);        pthread_mutex_init(&srv->srv_funcs.mtx, NULL);
         SLIST_INIT(&srv->srv_funcs);
         AVL_INIT(&srv->srv_funcs);
   
         /* init scheduler */          /* init scheduler */
         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());
                   pthread_mutex_destroy(&srv->srv_funcs.mtx);
                 io_free(srv);                  io_free(srv);
                 return NULL;                  return NULL;
         }          }
Line 750  rpc_srv_initServer(u_int regProgID, u_char regProcID,  Line 782  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);
                   pthread_mutex_destroy(&srv->srv_funcs.mtx);
                 io_free(srv);                  io_free(srv);
                 return NULL;                  return NULL;
         }          }
   
         /* create server socket */          /* create server socket */
        srv->srv_server.cli_sock = socket(srv->srv_server.cli_sa.sa.sa_family, SOCK_STREAM, 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);                  io_arrayDestroy(&srv->srv_clients);
                 schedEnd(&srv->srv_root);                  schedEnd(&srv->srv_root);
                   pthread_mutex_destroy(&srv->srv_funcs.mtx);
                 io_free(srv);                  io_free(srv);
                 return NULL;                  return NULL;
         }          }
Line 789  err: /* error condition */ Line 823  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);
           pthread_mutex_destroy(&srv->srv_funcs.mtx);
         io_free(srv);          io_free(srv);
         return NULL;          return NULL;
 }  }
Line 812  rpc_srv_endServer(rpc_srv_t ** __restrict psrv) Line 847  rpc_srv_endServer(rpc_srv_t ** __restrict psrv)
         (*psrv)->srv_kill = 1;          (*psrv)->srv_kill = 1;
         sleep(RPC_SCHED_POLLING);          sleep(RPC_SCHED_POLLING);
   
           pthread_mutex_destroy(&(*psrv)->srv_funcs.mtx);
         io_free(*psrv);          io_free(*psrv);
         *psrv = NULL;          *psrv = NULL;
 }  }
Line 827  rpc_srv_loopServer(rpc_srv_t * __restrict srv) Line 863  rpc_srv_loopServer(rpc_srv_t * __restrict srv)
 {  {
         rpc_cli_t *c;          rpc_cli_t *c;
         register int i;          register int i;
        rpc_func_t *f, *tmp;        rpc_func_t *f;
         struct timespec ts = { RPC_SCHED_POLLING, 0 };          struct timespec ts = { RPC_SCHED_POLLING, 0 };
   
         if (!srv) {          if (!srv) {
Line 835  rpc_srv_loopServer(rpc_srv_t * __restrict srv) Line 871  rpc_srv_loopServer(rpc_srv_t * __restrict srv)
                 return -1;                  return -1;
         }          }
   
         fcntl(srv->srv_server.cli_sock, F_SETFL,   
                         fcntl(srv->srv_server.cli_sock, F_GETFL) | O_NONBLOCK);  
   
         if (listen(srv->srv_server.cli_sock, io_arraySize(srv->srv_clients)) == -1) {          if (listen(srv->srv_server.cli_sock, io_arraySize(srv->srv_clients)) == -1) {
                 LOGERR;                  LOGERR;
                 return -1;                  return -1;
        }        } else
                 fcntl(srv->srv_server.cli_sock, F_SETFL, 
                                 fcntl(srv->srv_server.cli_sock, F_GETFL) | O_NONBLOCK);
   
        if (!schedRead(srv->srv_root, acceptClients, srv, srv->srv_server.cli_sock, NULL, 0)) {        if (!schedRead(srv->srv_root, cbProto[srv->srv_proto][CB_ACCEPTCLIENT], srv, 
                                 srv->srv_server.cli_sock, NULL, 0)) {
                 rpc_SetErr(sched_GetErrno(), "%s", sched_GetError());                  rpc_SetErr(sched_GetErrno(), "%s", sched_GetError());
                 return -1;                  return -1;
         }          }
Line 870  rpc_srv_loopServer(rpc_srv_t * __restrict srv) Line 906  rpc_srv_loopServer(rpc_srv_t * __restrict srv)
         close(srv->srv_server.cli_sock);          close(srv->srv_server.cli_sock);
   
         /* detach exported calls */          /* detach exported calls */
        TAILQ_FOREACH_SAFE(f, &srv->srv_funcs, func_node, tmp) {        RPC_FUNCS_LOCK(&srv->srv_funcs);
                TAILQ_REMOVE(&srv->srv_funcs, f, func_node);        while ((f = SLIST_FIRST(&srv->srv_funcs))) {
                 SLIST_REMOVE_HEAD(&srv->srv_funcs, func_next);
   
                 AIT_FREE_VAL(&f->func_name);                  AIT_FREE_VAL(&f->func_name);
                 io_free(f);                  io_free(f);
         }          }
           srv->srv_funcs.avlh_root = NULL;
           RPC_FUNCS_UNLOCK(&srv->srv_funcs);
   
         schedEnd(&srv->srv_root);          schedEnd(&srv->srv_root);
         return 0;          return 0;
Removed from v.1.11.2.3  
changed lines
  Added in v.1.12.2.2

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