[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.23.6.5 and 1.24.2.4

version 1.23.6.5, 2014/12/18 00:28:47 version 1.24.2.4, 2015/01/18 00:06:55
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 - 2014Copyright 2004 - 2015
         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 63  static void *txUDPPacket(sched_task_t *); Line 63  static void *txUDPPacket(sched_task_t *);
 static void *rxBPFPacket(sched_task_t *);  static void *rxBPFPacket(sched_task_t *);
 static void *txBPFPacket(sched_task_t *);  static void *txBPFPacket(sched_task_t *);
   
static sched_task_func_t cbProto[SOCK_BPF + 1][4] = {/* SOCK_EXT */
 
 static sched_task_func_t cbProto[SOCK_MAX_SUPPORT][4] = {
         { acceptClients, closeClient, rxPacket, txPacket },     /* SOCK_STREAM */          { acceptClients, closeClient, rxPacket, txPacket },     /* SOCK_STREAM */
         { acceptClients, closeClient, rxPacket, txPacket },     /* SOCK_STREAM */          { acceptClients, closeClient, rxPacket, txPacket },     /* SOCK_STREAM */
         { rxUDPPacket, freeClient, rxUDPPacket, txUDPPacket },  /* SOCK_DGRAM */          { rxUDPPacket, freeClient, rxUDPPacket, txUDPPacket },  /* SOCK_DGRAM */
         { NULL, NULL, NULL, NULL },                             /* SOCK_RAW */          { NULL, NULL, NULL, NULL },                             /* SOCK_RAW */
        { rxBPFPacket, freeClient, rxBPFPacket, txBPFPacket }   /* SOCK_BPF */        { rxBPFPacket, freeClient, rxBPFPacket, txBPFPacket },       /* SOCK_BPF */
         { NULL, NULL, NULL, NULL }      /* SOCK_EXT */
 };  };
   
 /* Global Signal Argument when kqueue support disabled */  /* Global Signal Argument when kqueue support disabled */
Line 645  txBPFPacket(sched_task_t *task) Line 648  txBPFPacket(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, estlen, wlen = sizeof(struct tagRPCCall);        int ret, len, wlen = sizeof(struct tagRPCCall);
         struct timespec ts = { DEF_RPC_TIMEOUT, 0 };          struct timespec ts = { DEF_RPC_TIMEOUT, 0 };
        struct pollfd pfd;        struct ether_header *eh;
         ait_val_t b = AIT_VAL_INIT;
   
         schedCancelby(TASK_ROOT(task), taskTIMER, CRITERIA_DATA, TASK_ARG(task), NULL);          schedCancelby(TASK_ROOT(task), taskTIMER, CRITERIA_DATA, TASK_ARG(task), NULL);
         schedTimer(TASK_ROOT(task), cbProto[s->srv_proto][CB_CLOSECLIENT],           schedTimer(TASK_ROOT(task), cbProto[s->srv_proto][CB_CLOSECLIENT], 
Line 662  txBPFPacket(sched_task_t *task) Line 666  txBPFPacket(sched_task_t *task)
                         rpc->call_rep.eno = RPC_ERROR(rpc_Errno);                          rpc->call_rep.eno = RPC_ERROR(rpc_Errno);
                 } else {                  } else {
                         /* calc estimated length */                          /* calc estimated length */
                        estlen = ait_resideVars(RPC_RETVARS(c)) + wlen;                        len = ait_resideVars(RPC_RETVARS(c)) + wlen;
                        if (estlen > AIT_LEN(&c->cli_buf))                        if (len > AIT_LEN(&c->cli_buf))
                                AIT_RE_BUF(&c->cli_buf, estlen);                                AIT_RE_BUF(&c->cli_buf, len);
                         buf = AIT_GET_BUF(&c->cli_buf);                          buf = AIT_GET_BUF(&c->cli_buf);
                         rpc = (struct tagRPCCall*) buf;                          rpc = (struct tagRPCCall*) buf;
   
Line 673  txBPFPacket(sched_task_t *task) Line 677  txBPFPacket(sched_task_t *task)
                         ret = ait_vars2buffer(buf + wlen, AIT_LEN(&c->cli_buf) - wlen,                           ret = ait_vars2buffer(buf + wlen, AIT_LEN(&c->cli_buf) - wlen, 
                                         RPC_RETVARS(c));                                          RPC_RETVARS(c));
                         /* Free return values */                          /* Free return values */
                        ait_freeVars(&c->cli_vars);                        ait_freeVars(&RPC_RETVARS(c));
                         if (ret == -1) {                          if (ret == -1) {
                                 rpc_SetErr(EBADRPC, "Prepare RPC packet failed");                                  rpc_SetErr(EBADRPC, "Prepare RPC packet failed");
                                 rpc->call_argc ^= rpc->call_argc;                                  rpc->call_argc ^= rpc->call_argc;
Line 691  txBPFPacket(sched_task_t *task) Line 695  txBPFPacket(sched_task_t *task)
         rpc->call_crc = htons(crcFletcher16((u_short*) buf, wlen / 2));          rpc->call_crc = htons(crcFletcher16((u_short*) buf, wlen / 2));
   
         /* send reply */          /* send reply */
        pfd.fd = TASK_FD(task);        AIT_SET_BUF(&b, NULL, MIN(wlen, s->srv_netbuf) + ETHER_HDR_LEN);
        pfd.events = POLLOUT;        eh = (struct ether_header*) AIT_GET_BUF(&b);
        for (; wlen > 0; wlen -= ret, buf += ret) {        memcpy(eh->ether_dhost, LLADDR(&c->cli_sa.sdl), ETHER_ADDR_LEN);
                if ((ret = poll(&pfd, 1, DEF_RPC_TIMEOUT * 1000)) < 1 ||         eh->ether_type = htons(RPC_DEFPORT);
                                pfd.revents & (POLLERR | POLLHUP | POLLNVAL)) {        memcpy(eh + 1, buf, MIN(wlen, s->srv_netbuf));
                        if (ret)
                                LOGERR;        ret = write(TASK_FD(task), AIT_GET_BUF(&b), AIT_LEN(&b));
                        else        AIT_FREE_VAL(&b);
                                rpc_SetErr(ETIMEDOUT, "Timeout reached! Client not respond");        if (ret == -1) {
                        /* close connection */                /* close connection */
                        schedEvent(TASK_ROOT(task), cbProto[s->srv_proto][CB_CLOSECLIENT],                 schedEvent(TASK_ROOT(task), cbProto[s->srv_proto][CB_CLOSECLIENT], 
                                        TASK_ARG(task), 0, NULL, 0);                                TASK_ARG(task), 0, NULL, 0);
                        return NULL;                return NULL;
                } 
                ret = write(TASK_FD(task), buf, MIN(wlen, s->srv_netbuf)); 
                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 736  rxBPFPacket(sched_task_t *task) Line 732  rxBPFPacket(sched_task_t *task)
         rlen = read(TASK_FD(task), AIT_GET_BUF(&b), AIT_LEN(&b));          rlen = read(TASK_FD(task), AIT_GET_BUF(&b), AIT_LEN(&b));
         h = (struct bpf_hdr*) AIT_GET_BUF(&b);          h = (struct bpf_hdr*) AIT_GET_BUF(&b);
         rlen -= h->bh_hdrlen;          rlen -= h->bh_hdrlen;
        if (rlen < h->bh_caplen || h->bh_caplen != h->bh_datalen ||         if (rlen < h->bh_datalen || h->bh_caplen != h->bh_datalen || 
                         rlen < ETHER_HDR_LEN + sizeof(struct tagRPCCall)) {                          rlen < ETHER_HDR_LEN + sizeof(struct tagRPCCall)) {
                 rpc_SetErr(ERPCMISMATCH, "Short RPC packet");                  rpc_SetErr(ERPCMISMATCH, "Short RPC packet");
                 goto end;                  goto end;
Line 799  rxBPFPacket(sched_task_t *task) Line 795  rxBPFPacket(sched_task_t *task)
   
         /* send RPC reply */          /* send RPC reply */
         if (!noreply)          if (!noreply)
                schedWrite(TASK_ROOT(task), cbProto[srv->srv_proto][CB_TXPACKET],                 schedEvent(TASK_ROOT(task), cbProto[srv->srv_proto][CB_TXPACKET], 
                                 c, TASK_FD(task), rpc, len);                                  c, TASK_FD(task), rpc, len);
 end:  end:
         AIT_FREE_VAL(&b);          AIT_FREE_VAL(&b);
Line 1249  rpc_srv_initServer(u_char InstID, int concurentClients Line 1245  rpc_srv_initServer(u_char InstID, int concurentClients
         rpc_srv_t *srv = NULL;          rpc_srv_t *srv = NULL;
         sockaddr_t sa = E_SOCKADDR_INIT;          sockaddr_t sa = E_SOCKADDR_INIT;
   
        if (!concurentClients || (proto < 0 || proto > SOCK_DGRAM)) {        if (!concurentClients || (proto < 0 || proto > SOCK_RAW)) {
                 rpc_SetErr(EINVAL, "Invalid parameters can`t init RPC server");                  rpc_SetErr(EINVAL, "Invalid parameters can`t init RPC server");
                 return NULL;                  return NULL;
         }          }
Line 1431  rpc_srv_loopServer(rpc_srv_t * __restrict srv) Line 1427  rpc_srv_loopServer(rpc_srv_t * __restrict srv)
         }          }
         array_Destroy(&srv->srv_clients);          array_Destroy(&srv->srv_clients);
   
        close(srv->srv_server.cli_sock);        if (srv->srv_proto != SOCK_EXT)
                 close(srv->srv_server.cli_sock);
   
         /* detach exported calls */          /* detach exported calls */
         RPC_FUNCS_LOCK(&srv->srv_funcs);          RPC_FUNCS_LOCK(&srv->srv_funcs);
Line 1612  err: /* error condition */ Line 1609  err: /* error condition */
         pthread_mutex_destroy(&srv->srv_funcs.mtx);          pthread_mutex_destroy(&srv->srv_funcs.mtx);
         e_free(srv);          e_free(srv);
         return NULL;          return NULL;
   }
   
   /*
    * rpc_srv_initServerExt() - Init & create pipe RPC Server
    *
    * @InstID = Instance for authentication & recognition
    * @netBuf = Network buffer length (min:512 bytes), if =0 == BUFSIZ (also meaning max RPC packet)
    * @fd = File descriptor
    * return: NULL == error or !=NULL bind and created RPC server instance
    */
   rpc_srv_t *
   rpc_srv_initServerExt(u_char InstID, int netBuf, int fd)
   {
           rpc_srv_t *srv = NULL;
   
   #ifdef HAVE_SRANDOMDEV
           srandomdev();
   #else
           time_t tim;
   
           srandom((time(&tim) ^ getpid()));
   #endif
   
           srv = e_malloc(sizeof(rpc_srv_t));
           if (!srv) {
                   LOGERR;
                   return NULL;
           } else
                   memset(srv, 0, sizeof(rpc_srv_t));
   
           srv->srv_proto = SOCK_EXT;
           srv->srv_netbuf = netBuf;
           srv->srv_session.sess_version = RPC_VERSION;
           srv->srv_session.sess_instance = InstID;
   
           srv->srv_server.cli_parent = srv;
           srv->srv_server.cli_sock = fd;
   
           /* init functions */
           pthread_mutex_init(&srv->srv_funcs.mtx, NULL);
           SLIST_INIT(&srv->srv_funcs);
           AVL_INIT(&srv->srv_funcs);
   
           /* init scheduler */
           srv->srv_root = schedBegin();
           if (!srv->srv_root) {
                   rpc_SetErr(sched_GetErrno(), "%s", sched_GetError());
                   pthread_mutex_destroy(&srv->srv_funcs.mtx);
                   e_free(srv);
                   return NULL;
           }
   
           /* init pool for clients */
           srv->srv_clients = array_Init(1);
           if (!srv->srv_clients) {
                   rpc_SetErr(elwix_GetErrno(), "%s", elwix_GetError());
                   schedEnd(&srv->srv_root);
                   pthread_mutex_destroy(&srv->srv_funcs.mtx);
                   e_free(srv);
                   return NULL;
           }
   
           fcntl(srv->srv_server.cli_sock, F_SETFL, 
                           fcntl(srv->srv_server.cli_sock, F_GETFL) | O_NONBLOCK);
   
           rpc_register_srvPing(srv);
   
           return srv;
 }  }
Removed from v.1.23.6.5  
changed lines
  Added in v.1.24.2.4

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