Diff for /libaitrpc/src/srv.c between versions 1.21.2.2 and 1.22.6.2

version 1.21.2.2, 2013/11/14 23:38:41 version 1.22.6.2, 2014/01/28 14:04:57
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, 2012, 2013Copyright 2004 - 2014
         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 66  static sched_task_func_t cbProto[SOCK_RAW + 1][4] = { Line 66  static sched_task_func_t cbProto[SOCK_RAW + 1][4] = {
         { NULL, NULL, NULL, NULL }                              /* SOCK_RAW */          { NULL, NULL, NULL, NULL }                              /* SOCK_RAW */
 };  };
   
   /* Global Signal Argument when kqueue support disabled */
   
   static volatile uintptr_t _glSigArg = 0;
   
   
 void  void
 rpc_freeCli(rpc_cli_t * __restrict c)  rpc_freeCli(rpc_cli_t * __restrict c)
 {  {
Line 223  txPacket(sched_task_t *task) Line 227  txPacket(sched_task_t *task)
                         if (ret)                          if (ret)
                                 LOGERR;                                  LOGERR;
                         else                          else
                                rpc_SetErr(ETIMEDOUT, "Timeout reached! Server not respond");                                rpc_SetErr(ETIMEDOUT, "Timeout reached! Client not respond");
                         /* close connection */                          /* close connection */
                         schedEvent(TASK_ROOT(task), cbProto[s->srv_proto][CB_CLOSECLIENT],                           schedEvent(TASK_ROOT(task), cbProto[s->srv_proto][CB_CLOSECLIENT], 
                                         TASK_ARG(task), 0, NULL, 0);                                          TASK_ARG(task), 0, NULL, 0);
Line 346  rxPacket(sched_task_t *task) Line 350  rxPacket(sched_task_t *task)
                         if (rlen)                          if (rlen)
                                 LOGERR;                                  LOGERR;
                         else                          else
                                rpc_SetErr(ETIMEDOUT, "Timeout reached! Server not respond");                                rpc_SetErr(ETIMEDOUT, "Timeout reached! Client not respond");
                         schedEvent(TASK_ROOT(task), cbProto[s->srv_proto][CB_CLOSECLIENT],                           schedEvent(TASK_ROOT(task), cbProto[s->srv_proto][CB_CLOSECLIENT], 
                                         TASK_ARG(task), 0, NULL, 0);                                          TASK_ARG(task), 0, NULL, 0);
                         return NULL;                          return NULL;
Line 448  txUDPPacket(sched_task_t *task) Line 452  txUDPPacket(sched_task_t *task)
         rpc_func_t *f = NULL;          rpc_func_t *f = NULL;
         u_char *buf = AIT_GET_BUF(&c->cli_buf);          u_char *buf = AIT_GET_BUF(&c->cli_buf);
         struct tagRPCCall *rpc = (struct tagRPCCall*) buf;          struct tagRPCCall *rpc = (struct tagRPCCall*) buf;
        int ret, wlen = sizeof(struct tagRPCCall);        int ret, estlen, wlen = sizeof(struct tagRPCCall);
         struct timespec ts = { DEF_RPC_TIMEOUT, 0 };          struct timespec ts = { DEF_RPC_TIMEOUT, 0 };
         struct pollfd pfd;          struct pollfd pfd;
   
Line 464  txUDPPacket(sched_task_t *task) Line 468  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 {
                           /* calc estimated length */
                           estlen = ait_resideVars(RPC_RETVARS(c)) + wlen;
                           if (estlen > AIT_LEN(&c->cli_buf))
                                   AIT_RE_BUF(&c->cli_buf, estlen);
                           buf = AIT_GET_BUF(&c->cli_buf);
                           rpc = (struct tagRPCCall*) buf;
   
                         rpc->call_argc = htons(array_Size(RPC_RETVARS(c)));                          rpc->call_argc = htons(array_Size(RPC_RETVARS(c)));
                         /* Go Encapsulate variables */                          /* Go Encapsulate variables */
                         ret = ait_vars2buffer(buf + wlen, AIT_LEN(&c->cli_buf) - wlen,                           ret = ait_vars2buffer(buf + wlen, AIT_LEN(&c->cli_buf) - wlen, 
Line 495  txUDPPacket(sched_task_t *task) Line 506  txUDPPacket(sched_task_t *task)
                         if (ret)                          if (ret)
                                 LOGERR;                                  LOGERR;
                         else                          else
                                rpc_SetErr(ETIMEDOUT, "Timeout reached! Server not respond");                                rpc_SetErr(ETIMEDOUT, "Timeout reached! Client not respond");
                         /* close connection */                          /* close connection */
                         schedEvent(TASK_ROOT(task), cbProto[s->srv_proto][CB_CLOSECLIENT],                           schedEvent(TASK_ROOT(task), cbProto[s->srv_proto][CB_CLOSECLIENT], 
                                         TASK_ARG(task), 0, NULL, 0);                                          TASK_ARG(task), 0, NULL, 0);
Line 537  rxUDPPacket(sched_task_t *task) Line 548  rxUDPPacket(sched_task_t *task)
         }          }
   
         c = _allocClient(srv, &sa);          c = _allocClient(srv, &sa);
        if (!c)        if (!c) {
                 EVERBOSE(1, "RPC client quota exceeded! Connection will be shutdown!\n");
                 usleep(2000);   /* blocked client delay */
                 goto end;                  goto end;
        else {        } else {
                 estlen = ntohl(rpc->call_len);                  estlen = ntohl(rpc->call_len);
                 if (estlen > AIT_LEN(&c->cli_buf))                  if (estlen > AIT_LEN(&c->cli_buf))
                         AIT_RE_BUF(&c->cli_buf, estlen);                          AIT_RE_BUF(&c->cli_buf, estlen);
Line 566  rxUDPPacket(sched_task_t *task) Line 579  rxUDPPacket(sched_task_t *task)
                         if (rlen)                          if (rlen)
                                 LOGERR;                                  LOGERR;
                         else                          else
                                rpc_SetErr(ETIMEDOUT, "Timeout reached! Server not respond");                                rpc_SetErr(ETIMEDOUT, "Timeout reached! Client not respond");
                         schedEvent(TASK_ROOT(task), cbProto[srv->srv_proto][CB_CLOSECLIENT],                           schedEvent(TASK_ROOT(task), cbProto[srv->srv_proto][CB_CLOSECLIENT], 
                                         c, 0, NULL, 0);                                          c, 0, NULL, 0);
                         return NULL;                          return NULL;
                 }                  }
                   salen = sa.ss.ss_len = sizeof(sockaddr_t);
                 rlen = recvfrom(TASK_FD(task), buf, len, 0, &sa.sa, &salen);                  rlen = recvfrom(TASK_FD(task), buf, len, 0, &sa.sa, &salen);
                 if (rlen == -1) {                  if (rlen == -1) {
                         /* close connection */                          /* close connection */
Line 578  rxUDPPacket(sched_task_t *task) Line 592  rxUDPPacket(sched_task_t *task)
                                         c, 0, NULL, 0);                                          c, 0, NULL, 0);
                         return NULL;                          return NULL;
                 }                  }
                   if (e_addrcmp(&c->cli_sa, &sa, 42))
                           rlen ^= rlen;   /* skip if arrive from different address */
         }          }
         len = estlen;          len = estlen;
   
Line 748  end: Line 764  end:
 static void *  static void *
 flushBLOB(sched_task_t *task)  flushBLOB(sched_task_t *task)
 {  {
        rpc_srv_t *srv = TASK_ARG(task);        uintptr_t sigArg = atomic_load_acq_ptr(&_glSigArg);
         rpc_srv_t *srv = sigArg ? (void*) sigArg : TASK_ARG(task);
         rpc_blob_t *b, *tmp;          rpc_blob_t *b, *tmp;
   
         TAILQ_FOREACH_SAFE(b, &srv->srv_blob.blobs, blob_node, tmp) {          TAILQ_FOREACH_SAFE(b, &srv->srv_blob.blobs, blob_node, tmp) {
Line 758  flushBLOB(sched_task_t *task) Line 775  flushBLOB(sched_task_t *task)
                 e_free(b);                  e_free(b);
         }          }
   
        schedSignalSelf(task);        if (!schedSignalSelf(task)) {
                 /* disabled kqueue support in libaitsched */
                 struct sigaction sa;
 
                 memset(&sa, 0, sizeof sa);
                 sigemptyset(&sa.sa_mask);
                 sa.sa_handler = (void (*)(int)) flushBLOB;
                 sa.sa_flags = SA_RESTART | SA_RESETHAND;
                 sigaction(SIGFBLOB, &sa, NULL);
         }
 
         return NULL;          return NULL;
 }  }
   
Line 972  rpc_srv_loopBLOBServer(rpc_srv_t * __restrict srv) Line 999  rpc_srv_loopBLOBServer(rpc_srv_t * __restrict srv)
                 return -1;                  return -1;
         }          }
   
        schedSignal(srv->srv_blob.root, flushBLOB, srv, SIGFBLOB, NULL, 0);        if (!schedSignal(srv->srv_blob.root, flushBLOB, srv, SIGFBLOB, NULL, 0)) {
                 /* disabled kqueue support in libaitsched */
                 struct sigaction sa;
 
                 atomic_store_rel_ptr(&_glSigArg, (uintptr_t) srv);
 
                 memset(&sa, 0, sizeof sa);
                 sigemptyset(&sa.sa_mask);
                 sa.sa_handler = (void (*)(int)) flushBLOB;
                 sa.sa_flags = SA_RESTART | SA_RESETHAND;
                 sigaction(SIGFBLOB, &sa, NULL);
         }
 
         if (!schedRead(srv->srv_blob.root, acceptBLOBClients, srv,           if (!schedRead(srv->srv_blob.root, acceptBLOBClients, srv, 
                                 srv->srv_blob.server.cli_sock, NULL, 0)) {                                  srv->srv_blob.server.cli_sock, NULL, 0)) {
                 rpc_SetErr(sched_GetErrno(), "%s", sched_GetError());                  rpc_SetErr(sched_GetErrno(), "%s", sched_GetError());

Removed from v.1.21.2.2  
changed lines
  Added in v.1.22.6.2


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