[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.9.2.8 and 1.21

version 1.9.2.8, 2012/05/16 07:15:45 version 1.21, 2013/11/14 21:44:14
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 46  SUCH DAMAGE. Line 46  SUCH DAMAGE.
 #include "global.h"  #include "global.h"
   
   
static void */* SOCK_STREAM */
closeClient(sched_task_t *task)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 */
 static void *freeClient(sched_task_t *);
 static void *rxUDPPacket(sched_task_t *);
 static void *txUDPPacket(sched_task_t *);
 
 /* SOCK_RAW */
 
 static sched_task_func_t cbProto[SOCK_RAW + 1][4] = {
         { acceptClients, closeClient, rxPacket, txPacket },     /* SOCK_STREAM */
         { acceptClients, closeClient, rxPacket, txPacket },     /* SOCK_STREAM */
         { rxUDPPacket, freeClient, rxUDPPacket, txUDPPacket },  /* SOCK_DGRAM */
         { NULL, NULL, NULL, NULL }                              /* SOCK_RAW */
 };
 
 
 void
 rpc_freeCli(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(s->srv_root, taskMAX, CRITERIA_ARG, TASK_ARG(task), NULL);        schedCancelby(s->srv_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_clients, c->cli_id, 42);        array_Del(s->srv_clients, c->cli_id, 0);
         if (c)
                 e_free(c);
 }
 
 
 static inline int
 _check4freeslot(rpc_srv_t * __restrict srv, sockaddr_t * __restrict sa)
 {
         rpc_cli_t *c = NULL;
         register int i;
 
         /* check free slots for connect */
         for (i = 0; i < array_Size(srv->srv_clients) && 
                         (c = array(srv->srv_clients, i, rpc_cli_t*)); i++)
                 /* check for duplicates */
                 if (sa && !e_addrcmp(&c->cli_sa, sa, 42))
                         break;
         if (i >= array_Size(srv->srv_clients))
                 return -1;      /* no more free slots! */
 
         return i;
 }
 
 static rpc_cli_t *
 _allocClient(rpc_srv_t * __restrict srv, sockaddr_t * __restrict sa)
 {
         rpc_cli_t *c = NULL;
         int n;
 
         n = _check4freeslot(srv, sa);
         if (n == -1)
                 return NULL;
         else
                 c = array(srv->srv_clients, n, rpc_cli_t*);
 
         if (!c) {
                 c = e_malloc(sizeof(rpc_cli_t));
                 if (!c) {
                         LOGERR;
                         srv->srv_kill = 1;
                         return NULL;
                 } else {
                         memset(c, 0, sizeof(rpc_cli_t));
                         array_Set(srv->srv_clients, n, c);
                         c->cli_id = n;
                         c->cli_parent = srv;
                 }
 
                 /* alloc empty buffer */
                 AIT_SET_BUFSIZ(&c->cli_buf, 0, srv->srv_netbuf);
         }
 
         return c;
 }
 
 
 static void *
 freeClient(sched_task_t *task)
 {
         rpc_freeCli(TASK_ARG(task));
 
         return NULL;          return NULL;
 }  }
   
 static void *  static void *
   closeClient(sched_task_t *task)
   {
           int sock = ((rpc_cli_t*) TASK_ARG(task))->cli_sock;
   
           rpc_freeCli(TASK_ARG(task));
   
           /* close client socket */
           shutdown(sock, SHUT_RDWR);
           close(sock);
           return NULL;
   }
   
   static void *
 txPacket(sched_task_t *task)  txPacket(sched_task_t *task)
 {  {
         rpc_cli_t *c = TASK_ARG(task);          rpc_cli_t *c = TASK_ARG(task);
         rpc_srv_t *s = c->cli_parent;          rpc_srv_t *s = c->cli_parent;
         rpc_func_t *f = NULL;          rpc_func_t *f = NULL;
        u_char buf[USHRT_MAX] = { 0 };        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 pollfd pfd;
 #ifdef TCP_SESSION_TIMEOUT
         struct timespec ts = { DEF_RPC_TIMEOUT, 0 };
   
        /* copy RPC header */        schedCancelby(TASK_ROOT(task), taskTIMER, CRITERIA_DATA, TASK_ARG(task), NULL);
        memcpy(buf, TASK_DATA(task), wlen);        schedTimer(TASK_ROOT(task), cbProto[s->srv_proto][CB_CLOSECLIENT], 
                          TASK_ARG(task), ts, TASK_ARG(task), 0);
 #endif
   
         if (rpc->call_argc) {          if (rpc->call_argc) {
                 f = rpc_srv_getCall(s, ntohs(rpc->call_tag));                  f = rpc_srv_getCall(s, ntohs(rpc->call_tag));
                 if (!f) {                  if (!f) {
                           rpc_SetErr(EPROGUNAVAIL, "Function not found at RPC server");
   
                         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);
                 } else {                  } else {
                        rpc->call_argc = htons(io_arraySize(f->func_vars));                        /* 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)));
                         /* Go Encapsulate variables */                          /* Go Encapsulate variables */
                        ret = io_vars2buffer(buf + wlen, sizeof buf - wlen, f->func_vars);                        ret = ait_vars2buffer(buf + wlen, AIT_LEN(&c->cli_buf) - wlen, 
                                         RPC_RETVARS(c));
                         /* Free return values */                          /* Free return values */
                        io_clrVars(f->func_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;
                                 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 101  txPacket(sched_task_t *task) Line 206  txPacket(sched_task_t *task)
                 }                  }
         }          }
   
        rpc->call_len = htons(wlen);        rpc->call_len = htonl(wlen);
   
   #if 0
         /* calculate CRC */          /* calculate CRC */
         rpc->call_crc ^= rpc->call_crc;          rpc->call_crc ^= rpc->call_crc;
         rpc->call_crc = htons(crcFletcher16((u_short*) buf, wlen / 2));          rpc->call_crc = htons(crcFletcher16((u_short*) buf, wlen / 2));
   #endif
   
         /* send reply */          /* send reply */
        ret = send(TASK_FD(task), buf, wlen, MSG_NOSIGNAL);        pfd.fd = TASK_FD(task);
        if (ret == -1 || ret != wlen) {        pfd.events = POLLOUT;
                /* close connection */        for (; wlen > 0; wlen -= ret, buf += ret) {
                schedEvent(TASK_ROOT(task), closeClient, c, 42, NULL, 0);                if ((ret = poll(&pfd, 1, DEF_RPC_TIMEOUT * 1000)) < 1 || 
                                 pfd.revents & (POLLERR | POLLHUP | POLLNVAL)) {
                         if (ret)
                                 LOGERR;
                         else
                                 rpc_SetErr(ETIMEDOUT, "Timeout reached! Server not respond");
                         /* close connection */
                         schedEvent(TASK_ROOT(task), cbProto[s->srv_proto][CB_CLOSECLIENT], 
                                         TASK_ARG(task), 0, NULL, 0);
                         return NULL;
                 }
                 ret = send(TASK_FD(task), buf, MIN(wlen, s->srv_netbuf), MSG_NOSIGNAL);
                 if (ret == -1) {
                         /* close connection */
                         schedEvent(TASK_ROOT(task), cbProto[s->srv_proto][CB_CLOSECLIENT], 
                                         TASK_ARG(task), 0, NULL, 0);
                         return NULL;
                 }
         }          }
   
         return NULL;          return NULL;
Line 124  execCall(sched_task_t *task) Line 248  execCall(sched_task_t *task)
         rpc_srv_t *s = c->cli_parent;          rpc_srv_t *s = c->cli_parent;
         rpc_func_t *f = NULL;          rpc_func_t *f = NULL;
         array_t *arr = NULL;          array_t *arr = NULL;
        u_char *buf = AIT_GET_BUF(&c->cli_buf) + TASK_VAL(task);        u_char *buf = AIT_GET_BUF(&c->cli_buf);
         struct tagRPCCall *rpc = (struct tagRPCCall*) buf;          struct tagRPCCall *rpc = (struct tagRPCCall*) buf;
         int argc = ntohs(rpc->call_argc);          int argc = ntohs(rpc->call_argc);
   
         /* 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) - sizeof(struct tagRPCCall), argc, 42);
                                argc, 1); 
                 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 145  execCall(sched_task_t *task) Line 269  execCall(sched_task_t *task)
   
         if (!(f = rpc_srv_getCall(s, ntohs(rpc->call_tag)))) {          if (!(f = rpc_srv_getCall(s, ntohs(rpc->call_tag)))) {
                 rpc_SetErr(EPROGUNAVAIL, "Function not found at RPC server");                  rpc_SetErr(EPROGUNAVAIL, "Function not found at RPC server");
   
                 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);
         } 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);                        if (!rpc->call_rep.eno) {
                                 LOGERR;
                                 rpc->call_rep.eno = RPC_ERROR(rpc_Errno);
                         }
                         rpc->call_argc ^= rpc->call_argc;                          rpc->call_argc ^= rpc->call_argc;
                 } else {                  } else {
                         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);                                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(f->func_vars));                                rpc->call_argc = htons(array_Size(RPC_RETVARS(c)));
                         }                          }
                 }                  }
         }          }
   
        if (arr)        array_Destroy(&arr);
                io_arrayDestroy(&arr); 
         return NULL;          return NULL;
 }  }
   
Line 178  rxPacket(sched_task_t *task) Line 305  rxPacket(sched_task_t *task)
 {  {
         rpc_cli_t *c = TASK_ARG(task);          rpc_cli_t *c = TASK_ARG(task);
         rpc_srv_t *s = c->cli_parent;          rpc_srv_t *s = c->cli_parent;
           int len, rlen, noreply, estlen;
   #if 0
           u_short crc;
   #endif
         u_char *buf = AIT_GET_BUF(&c->cli_buf);          u_char *buf = AIT_GET_BUF(&c->cli_buf);
        int len, rlen, noreply;        struct tagRPCCall *rpc = (struct tagRPCCall*) buf;
        u_short crc, off = 0;        struct pollfd pfd;
        struct tagRPCCall *rpc;#ifdef TCP_SESSION_TIMEOUT
         struct timespec ts = { DEF_RPC_TIMEOUT, 0 };
   
        memset(buf, 0, AIT_LEN(&c->cli_buf));        schedCancelby(TASK_ROOT(task), taskTIMER, CRITERIA_DATA, TASK_ARG(task), NULL);
        rlen = recv(TASK_FD(task), buf, AIT_LEN(&c->cli_buf), 0);        schedTimer(TASK_ROOT(task), cbProto[s->srv_proto][CB_CLOSECLIENT], 
        if (rlen < 1) {                         TASK_ARG(task), ts, TASK_ARG(task), 0);
 #endif
 
         memset(buf, 0, sizeof(struct tagRPCCall));
         rlen = recv(TASK_FD(task), rpc, sizeof(struct tagRPCCall), MSG_PEEK);
         if (rlen < sizeof(struct tagRPCCall)) {
                 /* close connection */                  /* close connection */
                schedEvent(TASK_ROOT(task), closeClient, c, 0, NULL, 0);                schedEvent(TASK_ROOT(task), cbProto[s->srv_proto][CB_CLOSECLIENT], 
                                 TASK_ARG(task), 0, NULL, 0);
                 return NULL;                  return NULL;
           } else {
                   estlen = ntohl(rpc->call_len);
                   if (estlen > AIT_LEN(&c->cli_buf))
                           AIT_RE_BUF(&c->cli_buf, estlen);
                   rpc = (struct tagRPCCall*) AIT_GET_BUF(&c->cli_buf);
                   buf = AIT_GET_BUF(&c->cli_buf);
                   len = estlen;
         }          }
   
           /* get next part of packet */
           memset(buf, 0, len);
           pfd.fd = TASK_FD(task);
           pfd.events = POLLIN | POLLPRI;
           for (; len > 0; len -= rlen, buf += rlen) {
                   if ((rlen = poll(&pfd, 1, DEF_RPC_TIMEOUT * 1000)) < 1 || 
                                   pfd.revents & (POLLERR | POLLHUP | POLLNVAL)) {
                           if (rlen)
                                   LOGERR;
                           else
                                   rpc_SetErr(ETIMEDOUT, "Timeout reached! Server not respond");
                           schedEvent(TASK_ROOT(task), cbProto[s->srv_proto][CB_CLOSECLIENT], 
                                           TASK_ARG(task), 0, NULL, 0);
                           return NULL;
                   }
                   rlen = recv(TASK_FD(task), buf, len, 0);
                   if (rlen == -1) {
                           /* close connection */
                           schedEvent(TASK_ROOT(task), cbProto[s->srv_proto][CB_CLOSECLIENT], 
                                           TASK_ARG(task), 0, NULL, 0);
                           return NULL;
                   }
           }
   
   #if 0
           /* check integrity of packet */
           crc = ntohs(rpc->call_crc);
           rpc->call_crc ^= rpc->call_crc;
           if (crc != crcFletcher16((u_short*) rpc, len / 2)) {
                   rpc_SetErr(ERPCMISMATCH, "Bad CRC RPC packet");
                   return NULL;
           }
   #endif
   
           noreply = RPC_CHK_NOREPLY(rpc);
   
           /* check RPC packet session info */
           if (rpc_chkPktSession(&rpc->call_session, &s->srv_session)) {
                   rpc_SetErr(ERPCMISMATCH, "Get invalid RPC session");
   
                   rpc->call_argc ^= rpc->call_argc;
                   rpc->call_rep.ret = RPC_ERROR(-1);
                   rpc->call_rep.eno = RPC_ERROR(errno);
           } else {
                   /* execute RPC call */
                   schedEvent(TASK_ROOT(task), execCall, TASK_ARG(task), (int) noreply, rpc, len);
           }
   
           /* send RPC reply */
           if (!noreply)
                   schedWrite(TASK_ROOT(task), cbProto[s->srv_proto][CB_TXPACKET], 
                                   TASK_ARG(task), TASK_FD(task), rpc, len);
   
           /* lets get next packet */
           schedReadSelf(task);
           return NULL;
   }
   
   static void *
   acceptClients(sched_task_t *task)
   {
           rpc_srv_t *srv = TASK_ARG(task);
           rpc_cli_t *c = NULL;
           socklen_t salen = sizeof(sockaddr_t);
           int sock;
   #ifdef TCP_SESSION_TIMEOUT
           struct timespec ts = { DEF_RPC_TIMEOUT, 0 };
   #endif
   
           c = _allocClient(srv, NULL);
           if (!c) {
                   EVERBOSE(1, "RPC client quota exceeded! Connection will be shutdown!\n");
                   if ((sock = accept(TASK_FD(task), NULL, NULL)) != -1) {
                           shutdown(sock, SHUT_RDWR);
                           close(sock);
                   }
                   goto end;
           }
   
           /* accept client */
           c->cli_sock = accept(TASK_FD(task), &c->cli_sa.sa, &salen);
           if (c->cli_sock == -1) {
                   LOGERR;
                   AIT_FREE_VAL(&c->cli_buf);
                   array_Del(srv->srv_clients, c->cli_id, 42);
                   goto end;
           } else
                   fcntl(c->cli_sock, F_SETFL, fcntl(c->cli_sock, F_GETFL) | O_NONBLOCK);
   
   #ifdef TCP_SESSION_TIMEOUT
           /* armed timer for close stateless connection */
           schedCancelby(TASK_ROOT(task), taskTIMER, CRITERIA_DATA, c, NULL);
           schedTimer(TASK_ROOT(task), cbProto[srv->srv_proto][CB_CLOSECLIENT], c, 
                           ts, c, 0);
   #endif
           schedRead(TASK_ROOT(task), cbProto[srv->srv_proto][CB_RXPACKET], c, 
                           c->cli_sock, NULL, 0);
   end:
           schedReadSelf(task);
           return NULL;
   }
   
   
   static void *
   txUDPPacket(sched_task_t *task)
   {
           rpc_cli_t *c = TASK_ARG(task);
           rpc_srv_t *s = c->cli_parent;
           rpc_func_t *f = NULL;
           u_char *buf = AIT_GET_BUF(&c->cli_buf);
           struct tagRPCCall *rpc = (struct tagRPCCall*) buf;
           int ret, wlen = sizeof(struct tagRPCCall);
           struct timespec ts = { DEF_RPC_TIMEOUT, 0 };
   
           schedCancelby(TASK_ROOT(task), taskTIMER, CRITERIA_DATA, TASK_ARG(task), NULL);
           schedTimer(TASK_ROOT(task), cbProto[s->srv_proto][CB_CLOSECLIENT], 
                            TASK_ARG(task), ts, TASK_ARG(task), 0);
   
           if (rpc->call_argc) {
                   f = rpc_srv_getCall(s, ntohs(rpc->call_tag));
                   if (!f) {
                           rpc_SetErr(EPROGUNAVAIL, "Function not found at RPC server");
                           rpc->call_argc ^= rpc->call_argc;
                           rpc->call_rep.ret = RPC_ERROR(-1);
                           rpc->call_rep.eno = RPC_ERROR(rpc_Errno);
                   } else {
                           rpc->call_argc = htons(array_Size(RPC_RETVARS(c)));
                           /* Go Encapsulate variables */
                           ret = ait_vars2buffer(buf + wlen, AIT_LEN(&c->cli_buf) - wlen, 
                                           RPC_RETVARS(c));
                           /* Free return values */
                           ait_freeVars(&c->cli_vars);
                           if (ret == -1) {
                                   rpc_SetErr(EBADRPC, "Prepare RPC packet failed");
                                   rpc->call_argc ^= rpc->call_argc;
                                   rpc->call_rep.ret = RPC_ERROR(-1);
                                   rpc->call_rep.eno = RPC_ERROR(rpc_Errno);
                           } else
                                   wlen += ret;
                   }
           }
   
           rpc->call_len = htonl(wlen);
   
           /* calculate CRC */
           rpc->call_crc ^= rpc->call_crc;
           rpc->call_crc = htons(crcFletcher16((u_short*) buf, wlen / 2));
   
           /* send reply */
           ret = sendto(TASK_FD(task), buf, wlen, MSG_NOSIGNAL, 
                           &c->cli_sa.sa, c->cli_sa.sa.sa_len);
           if (ret == -1 || ret != wlen) {
                   /* close connection */
                   schedEvent(TASK_ROOT(task), cbProto[s->srv_proto][CB_CLOSECLIENT], 
                                    TASK_ARG(task), 0, NULL, 0);
           }
   
           return NULL;
   }
   
   static void *
   rxUDPPacket(sched_task_t *task)
   {
           rpc_srv_t *srv = TASK_ARG(task);
           rpc_cli_t *c = NULL;
           int len, rlen, noreply;
           u_short crc, off = 0;
           u_char buf[USHRT_MAX + 1];
           struct tagRPCCall *rpc = (struct tagRPCCall*) buf;
           sockaddr_t sa;
           socklen_t salen;
           struct timespec ts = { DEF_RPC_TIMEOUT, 0 };
   
           /* receive connect packet */
           salen = sa.ss.ss_len = sizeof(sockaddr_t);
           rlen = recvfrom(TASK_FD(task), buf, sizeof buf, 0, &sa.sa, &salen);
           if (rlen < 1)
                   goto end;
   
           c = _allocClient(srv, &sa);
           if (!c)
                   goto end;
           else {
                   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));
                   /* armed timer for close stateless connection */
                   schedCancelby(TASK_ROOT(task), taskTIMER, CRITERIA_DATA, c, NULL);
                   schedTimer(TASK_ROOT(task), cbProto[srv->srv_proto][CB_CLOSECLIENT], 
                                   c, ts, c, 0);
           }
   
         do {          do {
                 /* check RPC packet */                  /* check RPC packet */
                 if (rlen < sizeof(struct tagRPCCall)) {                  if (rlen < sizeof(struct tagRPCCall)) {
                        rpc_SetErr(ERPCMISMATCH, "Too short RPC packet");                        rpc_SetErr(ERPCMISMATCH, "Short RPC packet");
                        break;
                        schedReadSelf(task); 
                        return NULL; 
                 } else                  } else
                        rpc = (struct tagRPCCall*) (buf + off);                        rpc = (struct tagRPCCall*) (AIT_GET_BUF(&c->cli_buf) + off);
   
                len = ntohs(rpc->call_len);                len = ntohl(rpc->call_len);
                 rlen -= len;                  rlen -= len;
   
                   /* check RPC packet lengths */
                   if (rlen < 0 || len < sizeof(struct tagRPCCall)) {
                           rpc_SetErr(ERPCMISMATCH, "Broken RPC packet length");
                           /* skip entire packet */
                           break;
                   }
   
                 /* check integrity of packet */                  /* check integrity of packet */
                 crc = ntohs(rpc->call_crc);                  crc = ntohs(rpc->call_crc);
                 rpc->call_crc ^= rpc->call_crc;                  rpc->call_crc ^= rpc->call_crc;
                if (crc != crcFletcher16((u_short*) (buf + off), len / 2)) {                if (crc != crcFletcher16((u_short*) rpc, len / 2)) {
                         rpc_SetErr(ERPCMISMATCH, "Bad CRC RPC packet");                          rpc_SetErr(ERPCMISMATCH, "Bad CRC RPC packet");
   
                         off += len;                          off += len;
                        if (rlen < 1)                        /* try next packet remaining into buffer */
                                break;                        continue;
                        else 
                                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, &srv->srv_session)) {
                         rpc_SetErr(ERPCMISMATCH, "Get invalid RPC session");                          rpc_SetErr(ERPCMISMATCH, "Get invalid RPC session");
   
                         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(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, 
                                         (int) noreply, rpc, len);
                 }                  }
   
                 /* 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[srv->srv_proto][CB_TXPACKET], 
                                         c, TASK_FD(task), rpc, len);
   
                 off += len;                  off += len;
         } while (rlen > 0);          } while (rlen > 0);
   
        /* lets get next packet */end:
         schedReadSelf(task);          schedReadSelf(task);
         return NULL;          return NULL;
 }  }
   
static void */* ------------------------------------------------------ */
acceptClients(sched_task_t *task)
 void
 rpc_freeBLOBCli(rpc_cli_t * __restrict c)
 {  {
        rpc_srv_t *srv = TASK_ARG(task);        rpc_srv_t *s = c->cli_parent;
        rpc_cli_t *c = NULL; 
        register int i; 
        socklen_t salen = sizeof(io_sockaddr_t); 
   
        /* check free slots for connect */        schedCancelby(s->srv_blob.root, taskMAX, CRITERIA_ARG, c, NULL);
        for (i = 0; i < io_arraySize(srv->srv_clients) &&  
                        (c = io_array(srv->srv_clients, i, rpc_cli_t*)); i++); 
        if (c)  /* no more free slots! */ 
                goto end; 
        c = malloc(sizeof(rpc_cli_t)); 
        if (!c) { 
                LOGERR; 
                srv->srv_kill = kill; 
                return NULL; 
        } else { 
                memset(c, 0, sizeof(rpc_cli_t)); 
                io_arraySet(srv->srv_clients, i, c); 
                c->cli_id = i; 
                c->cli_parent = srv; 
        } 
   
        /* alloc empty buffer */        /* free buffer */
        AIT_SET_BUF2(&c->cli_buf, 0, srv->srv_netbuf);        AIT_FREE_VAL(&c->cli_buf);
   
        /* accept client */        array_Del(s->srv_blob.clients, c->cli_id, 0);
        c->cli_sock = accept(srv->srv_server.cli_sock, &c->cli_sa.sa, &salen);        if (c)
        if (c->cli_sock == -1) {                e_free(c);
                LOGERR;}
                AIT_FREE_VAL(&c->cli_buf); 
                io_arrayDel(srv->srv_clients, i, 42); 
                goto end; 
        } else 
                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);
end:static void *
        schedReadSelf(task);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;
 }  }
   
 #if 0  
 static void *  static void *
 txBLOB(sched_task_t *task)  txBLOB(sched_task_t *task)
 {  {
        u_char *buf = TASK_DATA(task);        rpc_cli_t *c = TASK_ARG(task);
        struct tagBLOBHdr *blob = (struct tagBLOBHdr *) buf;        u_char *buf = AIT_GET_BUF(&c->cli_buf);
         int wlen = sizeof(struct tagBLOBHdr);          int wlen = sizeof(struct tagBLOBHdr);
   
         /* calculate CRC */  
         blob->hdr_crc ^= blob->hdr_crc;  
         blob->hdr_crc = htons(crcFletcher16((u_short*) buf, wlen / 2));  
   
         /* send reply */          /* send reply */
        wlen = send(TASK_FD(task), buf, wlen, 0);        wlen = send(TASK_FD(task), buf, wlen, MSG_NOSIGNAL);
        if (wlen == -1)        if (wlen == -1 || wlen != sizeof(struct tagBLOBHdr)) {
                LOGERR;                /* close blob connection */
        else if (wlen != sizeof(struct tagBLOBHdr))                schedEvent(TASK_ROOT(task), closeBLOBClient, c, 42, NULL, 0);
                rpc_SetErr(EPROCUNAVAIL, "RPC reply, should be send %d bytes, "        }
                                "really sended %d bytes", sizeof(struct tagBLOBHdr), wlen); 
   
         return NULL;          return NULL;
 }  }
Line 315  rxBLOB(sched_task_t *task) Line 639  rxBLOB(sched_task_t *task)
         rpc_cli_t *c = TASK_ARG(task);          rpc_cli_t *c = TASK_ARG(task);
         rpc_srv_t *s = c->cli_parent;          rpc_srv_t *s = c->cli_parent;
         rpc_blob_t *b;          rpc_blob_t *b;
        u_char *buf = TASK_DATA(task);        struct tagBLOBHdr blob;
        struct tagBLOBHdr *blob = (struct tagBLOBHdr *) buf; 
         int rlen;          int rlen;
         u_short crc;  
         struct timespec ts;  
   
        /* check for disable service at this moment? */        memset(&blob, 0, sizeof blob);
        if (!s || s->srv_blob.state == disable) {        rlen = recv(TASK_FD(task), &blob, sizeof blob, 0);
                usleep(100000);        if (rlen < 1) {
#ifdef HAVE_PTHREAD_YIELD                /* close blob connection */
                pthread_yield();                schedEvent(TASK_ROOT(task), closeBLOBClient, c, 42, NULL, 0);
#endif 
                schedRead(TASK_ROOT(task), rxBLOB, TASK_ARG(task), TASK_FD(task),  
                                TASK_DATA(task), TASK_DATLEN(task)); 
                 return NULL;                  return NULL;
         }          }
   
        memset(buf, 0, TASK_DATLEN(task));        /* check BLOB packet */
        rlen = recv(TASK_FD(task), buf, TASK_DATLEN(task), 0); 
        if (rlen == -1) { 
                LOGERR; 
                c->cli_kill = kill; 
                return NULL; 
        } else if (!rlen) { /* receive EOF */ 
                c->cli_kill = kill; 
                return NULL; 
        } 
 
         if (rlen < sizeof(struct tagBLOBHdr)) {          if (rlen < sizeof(struct tagBLOBHdr)) {
                rpc_SetErr(ERPCMISMATCH, "Too short BLOB packet");                rpc_SetErr(ERPCMISMATCH, "Short BLOB packet");
                schedRead(TASK_ROOT(task), rxBLOB, TASK_ARG(task), TASK_FD(task),  
                                TASK_DATA(task), TASK_DATLEN(task)); 
                return NULL; 
        } 
   
        /* check integrity of packet */                schedReadSelf(task);
        crc = ntohs(blob->hdr_crc); 
        blob->hdr_crc ^= blob->hdr_crc; 
        if (crc != crcFletcher16((u_short*) buf, rlen / 2)) { 
                rpc_SetErr(ERPCMISMATCH, "Bad CRC BLOB packet"); 
                schedRead(TASK_ROOT(task), rxBLOB, TASK_ARG(task), TASK_FD(task),  
                                TASK_DATA(task), TASK_DATLEN(task)); 
                 return NULL;                  return NULL;
         }          }
   
         /* check RPC packet session info */          /* check RPC packet session info */
        if ((crc = rpc_chkPktSession(&blob->hdr_session, &s->srv_session))) {        if (rpc_chkPktSession(&blob.hdr_session, &s->srv_session)) {
                 rpc_SetErr(ERPCMISMATCH, "Get invalid RPC session");                  rpc_SetErr(ERPCMISMATCH, "Get invalid RPC session");
                blob->hdr_cmd = error;                blob.hdr_cmd = error;
                 goto end;                  goto end;
         } else {  
                 /* change socket timeout from last packet */  
                 ts.tv_sec = blob->hdr_session.sess_timeout;  
                 ts.tv_nsec = 0;  
                 schedPolling(TASK_ROOT(task), &ts, NULL);  
         }          }
   
         /* Go to proceed packet ... */          /* Go to proceed packet ... */
        switch (blob->hdr_cmd) {        switch (blob.hdr_cmd) {
                 case get:                  case get:
                        if (!(b = rpc_srv_getBLOB(s, ntohl(blob->hdr_var)))) {                        if (!(b = rpc_srv_getBLOB(s, ntohl(blob.hdr_var)))) {
                                rpc_SetErr(EINVAL, "Var=%x not found", ntohl(blob->hdr_var));                                rpc_SetErr(EINVAL, "Var=%x not found", ntohl(blob.hdr_var));
                                blob->hdr_cmd = no;                                blob.hdr_cmd = no;
                                blob->hdr_ret = RPC_ERROR(-1);                                blob.hdr_ret = RPC_ERROR(-1);
                                 break;                                  break;
                         } else                          } else
                                blob->hdr_len = htonl(b->blob_len);                                blob.hdr_len = htonl(b->blob_len);
   
                         if (rpc_srv_blobMap(s, b) != -1) {                          if (rpc_srv_blobMap(s, b) != -1) {
                                 /* deliver BLOB variable to client */                                  /* deliver BLOB variable to client */
                                blob->hdr_ret = htonl(rpc_srv_sendBLOB(c, b));                                blob.hdr_ret = htonl(rpc_srv_sendBLOB(c, b));
                                 rpc_srv_blobUnmap(b);                                  rpc_srv_blobUnmap(b);
                         } else {                          } else {
                                blob->hdr_cmd = error;                                blob.hdr_cmd = error;
                                blob->hdr_ret = RPC_ERROR(-1);                                blob.hdr_ret = RPC_ERROR(-1);
                         }                          }
                         break;                          break;
                 case set:                  case set:
                        if ((b = rpc_srv_registerBLOB(s, ntohl(blob->hdr_len)))) {                        if ((b = rpc_srv_registerBLOB(s, ntohl(blob.hdr_len), 
                                                         ntohl(blob.hdr_ret)))) {
                                 /* set new BLOB variable for reply :) */                                  /* set new BLOB variable for reply :) */
                                blob->hdr_var = htonl(b->blob_var);                                blob.hdr_var = htonl(b->blob_var);
   
                                 /* receive BLOB from client */                                  /* receive BLOB from client */
                                blob->hdr_ret = htonl(rpc_srv_recvBLOB(c, b));                                blob.hdr_ret = htonl(rpc_srv_recvBLOB(c, b));
                                 rpc_srv_blobUnmap(b);                                  rpc_srv_blobUnmap(b);
                         } else {                          } else {
                                blob->hdr_cmd = error;                                blob.hdr_cmd = error;
                                blob->hdr_ret = RPC_ERROR(-1);                                blob.hdr_ret = RPC_ERROR(-1);
                         }                          }
                         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);
                         }                          }
                         break;                          break;
                 default:                  default:
                        rpc_SetErr(EPROCUNAVAIL, "Unsupported BLOB command %d", blob->hdr_cmd);                        rpc_SetErr(EPROCUNAVAIL, "Unsupported BLOB command %d", blob.hdr_cmd);
                        blob->hdr_cmd = error;                        blob.hdr_cmd = error;
                        blob->hdr_ret = RPC_ERROR(-1);                        blob.hdr_ret = RPC_ERROR(-1);
         }          }
   
 end:  end:
        schedWrite(TASK_ROOT(task), txBLOB, TASK_ARG(task), TASK_FD(task),         memcpy(AIT_ADDR(&c->cli_buf), &blob, sizeof blob);
                        TASK_DATA(task), TASK_DATLEN(task));        schedWrite(TASK_ROOT(task), txBLOB, TASK_ARG(task), TASK_FD(task), NULL, 0);
        schedRead(TASK_ROOT(task), rxBLOB, TASK_ARG(task), TASK_FD(task),         schedReadSelf(task);
                        TASK_DATA(task), TASK_DATLEN(task)); 
         return NULL;          return NULL;
 }  }
   
 static void *  static void *
rpc_srv_dispatchVars(void *arg)flushBLOB(sched_task_t *task)
 {  {
        rpc_cli_t *c = arg;        rpc_srv_t *srv = TASK_ARG(task);
        rpc_srv_t *s;        rpc_blob_t *b, *tmp;
        sched_root_task_t *root; 
        u_char *buf; 
        struct timespec ts = { DEF_RPC_TIMEOUT, 0 }; 
   
        if (!arg) {        TAILQ_FOREACH_SAFE(b, &srv->srv_blob.blobs, blob_node, tmp) {
                rpc_SetErr(EINVAL, "Invalid parameter can`t procced BLOB");                TAILQ_REMOVE(&srv->srv_blob.blobs, b, blob_node);
                return NULL; 
        } else 
                s = c->cli_parent; 
   
        /* allocate net buffer */                rpc_srv_blobFree(srv, b);
        buf = malloc(sizeof(struct tagBLOBHdr));                e_free(b);
        if (!buf) { 
                LOGERR; 
                return NULL; 
         }          }
   
        root = schedBegin();        schedSignalSelf(task);
        if (!root) {        return NULL;
                rpc_SetErr(sched_GetErrno(), "%s", sched_GetError());}
                free(buf);
 static void *
 acceptBLOBClients(sched_task_t *task)
 {
         rpc_srv_t *srv = TASK_ARG(task);
         rpc_cli_t *c = NULL;
         register int i;
         socklen_t salen = sizeof(sockaddr_t);
         int sock;
 #ifdef TCP_NOPUSH
         int n = 1;
 #endif
 
         /* check free slots for connect */
         for (i = 0; i < array_Size(srv->srv_blob.clients) && 
                         (c = array(srv->srv_blob.clients, i, rpc_cli_t*)); i++);
         if (c) {        /* no more free slots! */
                 EVERBOSE(1, "BLOB client quota exceeded! Connection will be shutdown!\n");
                 if ((sock = accept(TASK_FD(task), NULL, NULL)) != -1) {
                         shutdown(sock, SHUT_RDWR);
                         close(sock);
                 }
                 goto end;
         }
 
         c = e_malloc(sizeof(rpc_cli_t));
         if (!c) {
                 LOGERR;
                 srv->srv_kill = srv->srv_blob.kill = 1;
                 return NULL;                  return NULL;
         } else {          } else {
                schedTermCondition(root, kill);                memset(c, 0, sizeof(rpc_cli_t));
                schedPolling(root, &ts, NULL);                array_Set(srv->srv_blob.clients, i, c);
                 c->cli_id = i;
                 c->cli_parent = srv;
         }          }
   
        schedRead(root, rxBLOB, c, c->cli_sock, buf, sizeof(struct tagBLOBHdr));        /* alloc empty buffer */
         AIT_SET_BUFSIZ(&c->cli_buf, 0, srv->srv_netbuf);
   
        schedRun(root, (void*) &c->cli_kill);        /* accept client */
        schedEnd(&root);        c->cli_sock = accept(TASK_FD(task), &c->cli_sa.sa, &salen);
         if (c->cli_sock == -1) {
                 LOGERR;
                 AIT_FREE_VAL(&c->cli_buf);
                 array_Del(srv->srv_blob.clients, i, 42);
                 goto end;
         } 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);
         }
   
        shutdown(c->cli_sock, SHUT_RDWR);        schedRead(TASK_ROOT(task), rxBLOB, c, c->cli_sock, NULL, 0);
        close(c->cli_sock);end:
        memset(c, 0, sizeof(rpc_cli_t));        schedReadSelf(task);
        free(buf); 
         return NULL;          return NULL;
 }  }
   
// -------------------------------------------------/* ------------------------------------------------------ */
   
 /*  /*
  * rpc_srv_initBLOBServer() - Init & create BLOB Server   * rpc_srv_initBLOBServer() - Init & create BLOB Server
Line 483  int Line 808  int
 rpc_srv_initBLOBServer(rpc_srv_t * __restrict srv, u_short Port, const char *diskDir)  rpc_srv_initBLOBServer(rpc_srv_t * __restrict srv, u_short Port, const char *diskDir)
 {  {
         int n = 1;          int n = 1;
         io_sockaddr_t sa;  
   
        if (!srv) {        if (!srv || srv->srv_kill) {
                 rpc_SetErr(EINVAL, "Invalid parameters can`t init BLOB server");                  rpc_SetErr(EINVAL, "Invalid parameters can`t init BLOB server");
                 return -1;                  return -1;
         }          }
         if (srv->srv_blob.state) {  
                 rpc_SetErr(EPERM, "Already started BLOB server!");  
                 return 0;  
         }  
   
         memset(&srv->srv_blob, 0, sizeof srv->srv_blob);          memset(&srv->srv_blob, 0, sizeof srv->srv_blob);
         if (access(diskDir, R_OK | W_OK) == -1) {          if (access(diskDir, R_OK | W_OK) == -1) {
Line 501  rpc_srv_initBLOBServer(rpc_srv_t * __restrict srv, u_s Line 821  rpc_srv_initBLOBServer(rpc_srv_t * __restrict srv, u_s
         } else          } else
                 AIT_SET_STR(&srv->srv_blob.dir, diskDir);                  AIT_SET_STR(&srv->srv_blob.dir, diskDir);
   
        srv->srv_blob.server.cli_tid = pthread_self();        /* init blob list */
         TAILQ_INIT(&srv->srv_blob.blobs);
 
         srv->srv_blob.server.cli_parent = srv;          srv->srv_blob.server.cli_parent = srv;
   
        memcpy(&sa, &srv->srv_server.cli_sa, sizeof sa);        memcpy(&srv->srv_blob.server.cli_sa, &srv->srv_server.cli_sa, sizeof(sockaddr_t));
        switch (sa.sa.sa_family) {        switch (srv->srv_blob.server.cli_sa.sa.sa_family) {
                 case AF_INET:                  case AF_INET:
                        sa.sin.sin_port = htons(Port ? Port : ntohs(sa.sin.sin_port) + 1);                        srv->srv_blob.server.cli_sa.sin.sin_port = 
                                 htons(Port ? Port : ntohs(srv->srv_blob.server.cli_sa.sin.sin_port) + 1);
                         break;                          break;
                 case AF_INET6:                  case AF_INET6:
                        sa.sin6.sin6_port = htons(Port ? Port : ntohs(sa.sin6.sin6_port) + 1);                        srv->srv_blob.server.cli_sa.sin6.sin6_port = 
                                 htons(Port ? Port : ntohs(srv->srv_blob.server.cli_sa.sin6.sin6_port) + 1);
                         break;                          break;
                 case AF_LOCAL:                  case AF_LOCAL:
                        strlcat(sa.sun.sun_path, ".blob", sizeof sa.sun.sun_path);                        strlcat(srv->srv_blob.server.cli_sa.sun.sun_path, ".blob", 
                                         sizeof srv->srv_blob.server.cli_sa.sun.sun_path);
                         break;                          break;
                 default:                  default:
                         AIT_FREE_VAL(&srv->srv_blob.dir);                          AIT_FREE_VAL(&srv->srv_blob.dir);
                         return -1;                          return -1;
         }          }
         memcpy(&srv->srv_blob.server.cli_sa, &sa, sizeof sa);  
   
         /* create BLOB server socket */          /* create BLOB server socket */
         srv->srv_blob.server.cli_sock = socket(srv->srv_server.cli_sa.sa.sa_family, SOCK_STREAM, 0);          srv->srv_blob.server.cli_sock = socket(srv->srv_server.cli_sa.sa.sa_family, SOCK_STREAM, 0);
Line 553  rpc_srv_initBLOBServer(rpc_srv_t * __restrict srv, u_s Line 877  rpc_srv_initBLOBServer(rpc_srv_t * __restrict srv, u_s
                 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;
        }        } else
                 fcntl(srv->srv_blob.server.cli_sock, F_SETFL, 
                                 fcntl(srv->srv_blob.server.cli_sock, F_GETFL) | O_NONBLOCK);
   
        /* allocate pool for concurent clients */
        srv->srv_blob.clients = calloc(srv->srv_numcli, sizeof(rpc_cli_t));        /* allocate pool for concurent blob clients */
         srv->srv_blob.clients = array_Init(array_Size(srv->srv_clients));
         if (!srv->srv_blob.clients) {          if (!srv->srv_blob.clients) {
                LOGERR;                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;
        } else        }
                memset(srv->srv_blob.clients, 0, srv->srv_numcli * sizeof(rpc_cli_t)); 
   
        rpc_srv_registerCall(srv, NULL, CALL_BLOBSHUTDOWN, 0);        /* init blob scheduler */
        rpc_srv_registerCall(srv, NULL, CALL_BLOBCLIENTS, 1);        srv->srv_blob.root = schedBegin();
        rpc_srv_registerCall(srv, NULL, CALL_BLOBVARS, 1);        if (!srv->srv_blob.root) {
        rpc_srv_registerCall(srv, NULL, CALL_BLOBSTATE, 0);                rpc_SetErr(sched_GetErrno(), "%s", sched_GetError());
                 array_Destroy(&srv->srv_blob.clients);
                 close(srv->srv_blob.server.cli_sock);
                 AIT_FREE_VAL(&srv->srv_blob.dir);
                 return -1;
         }
   
         srv->srv_blob.state = enable;   /* enable BLOB */  
         return 0;          return 0;
 }  }
   
Line 583  rpc_srv_initBLOBServer(rpc_srv_t * __restrict srv, u_s Line 913  rpc_srv_initBLOBServer(rpc_srv_t * __restrict srv, u_s
 void  void
 rpc_srv_endBLOBServer(rpc_srv_t * __restrict srv)  rpc_srv_endBLOBServer(rpc_srv_t * __restrict srv)
 {  {
        rpc_cli_t *c;        if (!srv)
        register int i; 
        rpc_blob_t *f; 
 
        if (!srv || srv->srv_blob.state == disable) 
                 return;                  return;
         else  
                 srv->srv_blob.state = kill;  
   
        rpc_srv_unregisterCall(srv, NULL, CALL_BLOBSHUTDOWN);        srv->srv_blob.kill = 1;
        rpc_srv_unregisterCall(srv, NULL, CALL_BLOBCLIENTS); 
        rpc_srv_unregisterCall(srv, NULL, CALL_BLOBVARS); 
        rpc_srv_unregisterCall(srv, NULL, CALL_BLOBSTATE); 
   
        /* close all clients connections & server socket */        schedEnd(&srv->srv_blob.root);
        for (i = 0, c = srv->srv_blob.clients; i < srv->srv_numcli && c; i++, c++) 
                if (c->cli_sa.sa.sa_family) 
                        shutdown(c->cli_sock, SHUT_RDWR); 
        close(srv->srv_blob.server.cli_sock); 
 
        if (srv->srv_blob.clients) { 
                free(srv->srv_blob.clients); 
                srv->srv_blob.clients = NULL; 
        } 
 
        /* detach blobs */ 
        while ((f = srv->srv_blob.blobs)) { 
                srv->srv_blob.blobs = f->blob_next; 
                rpc_srv_blobFree(srv, f); 
                free(f); 
        } 
 
        AIT_FREE_VAL(&srv->srv_blob.dir); 
 
        /* at final, save disable BLOB service state */ 
        srv->srv_blob.state = disable; 
 }  }
   
 /*  /*
 * rpc_srv_loopBLOB() - Execute Main BLOB server loop and wait for clients requests * rpc_srv_loopBLOBServer() - Execute Main BLOB server loop and wait for clients requests
  *   *
  * @srv = RPC Server instance   * @srv = RPC Server instance
  * return: -1 error or 0 ok, infinite loop ...   * return: -1 error or 0 ok, infinite loop ...
  */   */
 int  int
rpc_srv_loopBLOB(rpc_srv_t * __restrict srv)rpc_srv_loopBLOBServer(rpc_srv_t * __restrict srv)
 {  {
         socklen_t salen = sizeof(io_sockaddr_t);  
         register int i;  
         rpc_cli_t *c;          rpc_cli_t *c;
        int ret;        register int i;
        pthread_attr_t attr;        rpc_blob_t *b, *tmp;
        struct pollfd pfd;        struct timespec ts = { RPC_SCHED_POLLING, 0 };
   
        if (!srv || srv->srv_blob.state == kill) {        if (!srv || srv->srv_kill) {
                 rpc_SetErr(EINVAL, "Invalid parameter can`t start BLOB server");                  rpc_SetErr(EINVAL, "Invalid parameter can`t start BLOB server");
                 return -1;                  return -1;
         }          }
   
        if (listen(srv->srv_blob.server.cli_sock, SOMAXCONN) == -1) {        if (listen(srv->srv_blob.server.cli_sock, array_Size(srv->srv_blob.clients)) == -1) {
                 LOGERR;                  LOGERR;
                 return -1;                  return -1;
        } else        }
                fcntl(srv->srv_blob.server.cli_sock, F_SETFL,  
                                fcntl(srv->srv_blob.server.cli_sock, F_GETFL) | O_NONBLOCK); 
   
        pthread_attr_init(&attr);        schedSignal(srv->srv_blob.root, flushBLOB, srv, SIGFBLOB, NULL, 0);
        pthread_attr_setdetachstate(&attr, PTHREAD_CREATE_DETACHED);        if (!schedRead(srv->srv_blob.root, acceptBLOBClients, srv, 
                                 srv->srv_blob.server.cli_sock, NULL, 0)) {
                 rpc_SetErr(sched_GetErrno(), "%s", sched_GetError());
                 return -1;
         }
   
        pfd.fd = srv->srv_blob.server.cli_sock;        schedPolling(srv->srv_blob.root, &ts, NULL);
        pfd.events = POLLIN | POLLPRI;        /* main rpc loop */
        /* main BLOB loop */        schedRun(srv->srv_blob.root, &srv->srv_blob.kill);
        while (srv->srv_blob.state != kill && srv->srv_kill != kill) { 
                for (c = srv->srv_blob.clients, i = 0; i < srv->srv_numcli && c; i++, c++) 
                        if (!c->cli_sa.sa.sa_family) 
                                break; 
                if (i >= srv->srv_numcli) { 
#ifdef HAVE_PTHREAD_YIELD 
                        pthread_yield(); 
#endif 
                        usleep(1000000); 
                        continue; 
                } 
   
                if ((ret = poll(&pfd, 1, srv->srv_session.sess_timeout * 1000)) == -1 ||         /* detach blobs */
                                pfd.revents & (POLLERR | POLLHUP | POLLNVAL)) {        TAILQ_FOREACH_SAFE(b, &srv->srv_blob.blobs, blob_node, tmp) {
                        LOGERR;                TAILQ_REMOVE(&srv->srv_blob.blobs, b, blob_node);
                        ret = 1; 
                        break; 
                } else if (!ret) 
                        continue; 
   
                c->cli_sock = accept(srv->srv_blob.server.cli_sock, &c->cli_sa.sa, &salen);                rpc_srv_blobFree(srv, b);
                if (c->cli_sock == -1) {                e_free(b);
                        LOGERR;        }
                        continue; 
                } else 
                        c->cli_parent = srv; 
   
                /* spawn dispatch thread for BLOB client */        /* close all clients connections & server socket */
                c->cli_kill = enable;        for (i = 0; i < array_Size(srv->srv_blob.clients); i++) {
                if (pthread_create(&c->cli_tid, &attr, rpc_srv_dispatchVars, c)) {                c = array(srv->srv_blob.clients, i, rpc_cli_t*);
                        LOGERR;                if (c) {
                        continue;                        shutdown(c->cli_sock, SHUT_RDWR);
                         close(c->cli_sock);
 
                         schedCancelby(srv->srv_blob.root, taskMAX, CRITERIA_ARG, c, NULL);
                         AIT_FREE_VAL(&c->cli_buf);
                 }                  }
                   array_Del(srv->srv_blob.clients, i, 42);
         }          }
           array_Destroy(&srv->srv_blob.clients);
   
        srv->srv_blob.state = kill;        close(srv->srv_blob.server.cli_sock);
   
        pthread_attr_destroy(&attr);        AIT_FREE_VAL(&srv->srv_blob.dir);
         return 0;          return 0;
 }  }
 #endif  
   
   
 /*  /*
  * rpc_srv_initServer() - Init & create RPC Server   * rpc_srv_initServer() - Init & create RPC Server
  *   *
 * @regProgID = ProgramID for authentication & recognition * @InstID = Instance for authentication & recognition
 * @regProcID = ProcessID for authentication & recognition 
  * @concurentClients = Concurent clients at same time to this server   * @concurentClients = Concurent clients at same time to this server
 * @netBuf = Network buffer length, 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_char InstID, int concurentClients, int netBuf, 
                int netBuf, const char *csHost, u_short Port)                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;        sockaddr_t sa = E_SOCKADDR_INIT;
   
        if (!concurentClients || !regProgID) {        if (!concurentClients || (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 (!io_gethostbyname(csHost, Port, &sa))        if (!e_gethostbyname(csHost, Port, &sa))
                 return NULL;                  return NULL;
         if (!Port)          if (!Port)
                 Port = RPC_DEFPORT;                  Port = RPC_DEFPORT;
        if (!netBuf)        if (!proto)
                 proto = SOCK_STREAM;
         if (netBuf < RPC_MIN_BUFSIZ)
                 netBuf = BUFSIZ;                  netBuf = BUFSIZ;
         else          else
                netBuf = io_align(netBuf, 1);      /* align netBuf length */                netBuf = E_ALIGN(netBuf, 2);      /* align netBuf length */
   
        srv = malloc(sizeof(rpc_srv_t));#ifdef HAVE_SRANDOMDEV
         srandomdev();
 #else
         time_t tim;
 
         srandom((time(&tim) ^ getpid()));
 #endif
 
         srv = e_malloc(sizeof(rpc_srv_t));
         if (!srv) {          if (!srv) {
                 LOGERR;                  LOGERR;
                 return NULL;                  return NULL;
         } 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_instance = InstID;
        srv->srv_session.sess_process = 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());
                free(srv);                pthread_mutex_destroy(&srv->srv_funcs.mtx);
                 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);
                free(srv);                pthread_mutex_destroy(&srv->srv_funcs.mtx);
                 e_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);                array_Destroy(&srv->srv_clients);
                 schedEnd(&srv->srv_root);                  schedEnd(&srv->srv_root);
                free(srv);                pthread_mutex_destroy(&srv->srv_funcs.mtx);
                 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 791  rpc_srv_initServer(u_int regProgID, u_char regProcID,  Line 1093  rpc_srv_initServer(u_int regProgID, u_char regProcID, 
                                 srv->srv_server.cli_sa.sa.sa_len) == -1) {                                  srv->srv_server.cli_sa.sa.sa_len) == -1) {
                 LOGERR;                  LOGERR;
                 goto err;                  goto err;
        }        } else
                 fcntl(srv->srv_server.cli_sock, F_SETFL, 
                                 fcntl(srv->srv_server.cli_sock, F_GETFL) | O_NONBLOCK);
   
           rpc_register_srvPing(srv);
   
         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);
        free(srv);        pthread_mutex_destroy(&srv->srv_funcs.mtx);
         e_free(srv);
         return NULL;          return NULL;
 }  }
   
Line 811  err: /* error condition */ Line 1118  err: /* error condition */
 void  void
 rpc_srv_endServer(rpc_srv_t ** __restrict psrv)  rpc_srv_endServer(rpc_srv_t ** __restrict psrv)
 {  {
        rpc_cli_t *c;        if (!psrv || !*psrv)
        register int i; 
        rpc_func_t *f, *tmp; 
 
        if (!psrv || !*psrv) { 
                rpc_SetErr(EINVAL, "Can`t destroy server because parameter is null!"); 
                 return;                  return;
         }  
   
        /* close all clients connections & server socket */        /* if send kill to blob server */
        for (i = 0; i < io_arraySize((*psrv)->srv_clients); i++) {        rpc_srv_endBLOBServer(*psrv);
                c = io_array((*psrv)->srv_clients, i, rpc_cli_t*); 
                if (c) { 
                        shutdown(c->cli_sock, SHUT_RDWR); 
                        close(c->cli_sock); 
   
                        free(c);        (*psrv)->srv_kill = 1;
                        io_arrayDel((*psrv)->srv_clients, i, 0);        sleep(RPC_SCHED_POLLING);
                } 
        } 
        io_arrayDestroy(&(*psrv)->srv_clients); 
   
         close((*psrv)->srv_server.cli_sock);  
   
         /* detach exported calls */  
         TAILQ_FOREACH_SAFE(f, &(*psrv)->srv_funcs, func_node, tmp) {  
                 io_freeVars(&f->func_vars);  
                 AIT_FREE_VAL(&f->func_name);  
                 free(f);  
         }  
   
         schedEnd(&(*psrv)->srv_root);          schedEnd(&(*psrv)->srv_root);
        free(*psrv);
         pthread_mutex_destroy(&(*psrv)->srv_funcs.mtx);
         e_free(*psrv);
         *psrv = NULL;          *psrv = NULL;
 }  }
   
Line 856  rpc_srv_endServer(rpc_srv_t ** __restrict psrv) Line 1143  rpc_srv_endServer(rpc_srv_t ** __restrict psrv)
 int  int
 rpc_srv_loopServer(rpc_srv_t * __restrict srv)  rpc_srv_loopServer(rpc_srv_t * __restrict srv)
 {  {
           rpc_cli_t *c;
           register int i;
           rpc_func_t *f;
           struct timespec ts = { RPC_SCHED_POLLING, 0 };
   
         if (!srv) {          if (!srv) {
                 rpc_SetErr(EINVAL, "Invalid parameter can`t start RPC server");                  rpc_SetErr(EINVAL, "Invalid parameter can`t start RPC server");
                 return -1;                  return -1;
         }          }
   
        fcntl(srv->srv_server.cli_sock, F_SETFL,         if (srv->srv_proto == SOCK_STREAM)
                        fcntl(srv->srv_server.cli_sock, F_GETFL) | O_NONBLOCK);                if (listen(srv->srv_server.cli_sock, array_Size(srv->srv_clients)) == -1) {
                         LOGERR;
                         return -1;
                 }
   
        if (listen(srv->srv_server.cli_sock, io_arraySize(srv->srv_clients)) == -1) {        if (!schedRead(srv->srv_root, cbProto[srv->srv_proto][CB_ACCEPTCLIENT], srv, 
                LOGERR;                                srv->srv_server.cli_sock, NULL, 0)) {
                return -1; 
        } 
 
        if (!schedRead(srv->srv_root, acceptClients, 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;
         }          }
   
           schedPolling(srv->srv_root, &ts, NULL);
         /* main rpc loop */          /* main rpc loop */
         schedRun(srv->srv_root, &srv->srv_kill);          schedRun(srv->srv_root, &srv->srv_kill);
   
           /* close all clients connections & server socket */
           for (i = 0; i < array_Size(srv->srv_clients); i++) {
                   c = array(srv->srv_clients, i, rpc_cli_t*);
                   if (c) {
                           shutdown(c->cli_sock, SHUT_RDWR);
                           close(c->cli_sock);
   
                           schedCancelby(srv->srv_root, taskMAX, CRITERIA_ARG, c, NULL);
                           ait_freeVars(&RPC_RETVARS(c));
                           AIT_FREE_VAL(&c->cli_buf);
                   }
                   array_Del(srv->srv_clients, i, 42);
           }
           array_Destroy(&srv->srv_clients);
   
           close(srv->srv_server.cli_sock);
   
           /* detach exported calls */
           RPC_FUNCS_LOCK(&srv->srv_funcs);
           while ((f = SLIST_FIRST(&srv->srv_funcs))) {
                   SLIST_REMOVE_HEAD(&srv->srv_funcs, func_next);
   
                   AIT_FREE_VAL(&f->func_name);
                   e_free(f);
           }
           srv->srv_funcs.avlh_root = NULL;
           RPC_FUNCS_UNLOCK(&srv->srv_funcs);
   
         return 0;          return 0;
 }  }
   
Line 883  rpc_srv_loopServer(rpc_srv_t * __restrict srv) Line 1204  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, ntohs(rpc->call_argc), args);        return func(cli, rpc, args);
 }  }
Removed from v.1.9.2.8  
changed lines
  Added in v.1.21

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