[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.2 and 1.3.2.1

version 1.2, 2011/05/02 23:12:08 version 1.3.2.1, 2011/08/18 15:38:18
Line 52  rpc_srv_dispatchCall(void *arg) Line 52  rpc_srv_dispatchCall(void *arg)
         rpc_cli_t *c = arg;          rpc_cli_t *c = arg;
         rpc_srv_t *s;          rpc_srv_t *s;
         rpc_val_t *vals = NULL, *v = NULL;          rpc_val_t *vals = NULL, *v = NULL;
        rpc_func_t *f;        rpc_func_t *f = NULL;
         struct tagRPCCall *rpc;          struct tagRPCCall *rpc;
        struct tagRPCRet rrpc;        struct tagRPCRet *rrpc;
         rpc_sess_t ses = { 0 };
         fd_set fds;          fd_set fds;
         u_char buf[BUFSIZ], *data;          u_char buf[BUFSIZ], *data;
         int ret, argc = 0, Limit = 0;          int ret, argc = 0, Limit = 0;
Line 67  rpc_srv_dispatchCall(void *arg) Line 68  rpc_srv_dispatchCall(void *arg)
                 s = c->cli_parent;                  s = c->cli_parent;
   
         do {          do {
                   v = NULL;
                 FD_ZERO(&fds);                  FD_ZERO(&fds);
                 FD_SET(c->cli_sock, &fds);                  FD_SET(c->cli_sock, &fds);
                 ret = select(c->cli_sock + 1, &fds, NULL, NULL, NULL);                  ret = select(c->cli_sock + 1, &fds, NULL, NULL, NULL);
                 if (ret == -1) {                  if (ret == -1) {
                         ret = -2;                          ret = -2;
                 }                  }
                 memset(&rrpc, 0, sizeof rrpc);  
                 memset(buf, 0, BUFSIZ);                  memset(buf, 0, BUFSIZ);
                 if ((ret = recv(c->cli_sock, buf, BUFSIZ, 0)) == -1) {                  if ((ret = recv(c->cli_sock, buf, BUFSIZ, 0)) == -1) {
                         LOGERR;                          LOGERR;
Line 123  rpc_srv_dispatchCall(void *arg) Line 124  rpc_srv_dispatchCall(void *arg)
                                                 data += v[i].val_len;                                                  data += v[i].val_len;
                                                 break;                                                  break;
                                         case string:                                          case string:
                                                if (v[i].val_len + 1 > BUFSIZ - Limit) {                                                if (v[i].val_len > BUFSIZ - Limit) {
                                                         rpc_SetErr(EMSGSIZE, "Error:: Too big RPC packet ...\n");                                                          rpc_SetErr(EMSGSIZE, "Error:: Too big RPC packet ...\n");
                                                         ret = -5;                                                          ret = -5;
                                                         goto makeReply;                                                          goto makeReply;
Line 131  rpc_srv_dispatchCall(void *arg) Line 132  rpc_srv_dispatchCall(void *arg)
                                                         Limit += v[i].val_len;                                                          Limit += v[i].val_len;
   
                                                 v[i].val.string = (int8_t*) data;                                                  v[i].val.string = (int8_t*) data;
                                                data += v[i].val_len + 1;                                                data += v[i].val_len;
                                                 break;                                                  break;
                                         case blob:                                          case blob:
                                                 if (s->srv_blob.state == disable) {                                                  if (s->srv_blob.state == disable) {
Line 147  rpc_srv_dispatchCall(void *arg) Line 148  rpc_srv_dispatchCall(void *arg)
   
                 argc = 0;                  argc = 0;
                 vals = NULL;                  vals = NULL;
                   memcpy(&ses, &rpc->call_session, sizeof ses);
                 if (!(f = rpc_srv_getCall(s, rpc->call_tag, rpc->call_hash))) {                  if (!(f = rpc_srv_getCall(s, rpc->call_tag, rpc->call_hash))) {
                         rpc_SetErr(EINVAL, "Error:: call not found into RPC server ...\n");                          rpc_SetErr(EINVAL, "Error:: call not found into RPC server ...\n");
                         ret = -6;                          ret = -6;
Line 155  rpc_srv_dispatchCall(void *arg) Line 157  rpc_srv_dispatchCall(void *arg)
                                 ret = -9;                                  ret = -9;
                         else                          else
                                 argc = rpc_srv_getValsCall(f, &vals);                                  argc = rpc_srv_getValsCall(f, &vals);
   
 makeReply:  makeReply:
                 memcpy(&rrpc.ret_session, &rpc->call_session, sizeof rrpc.ret_session);  
                 rrpc.ret_tag = rpc->call_tag;  
                 rrpc.ret_hash = rpc->call_hash;  
                 rrpc.ret_errno = rpc_Errno;  
                 rrpc.ret_retcode = ret;  
                 rrpc.ret_argc = argc;  
   
                 memset(buf, 0, BUFSIZ);                  memset(buf, 0, BUFSIZ);
                memcpy(buf, &rrpc, (Limit = sizeof rrpc));                rrpc = (struct tagRPCRet*) buf;
                 Limit = sizeof(struct tagRPCRet);
 
                 memcpy(&rrpc->ret_session, &ses, sizeof(rpc_sess_t));
                 rrpc->ret_tag = rpc->call_tag;
                 rrpc->ret_hash = rpc->call_hash;
                 rrpc->ret_errno = rpc_Errno;
                 rrpc->ret_retcode = ret;
                 rrpc->ret_argc = argc;
 
                 if (argc && vals) {                  if (argc && vals) {
                        v = (rpc_val_t*) (buf + sizeof rrpc);                        v = (rpc_val_t*) (buf + Limit);
                         if (argc * sizeof(rpc_val_t) > BUFSIZ - Limit) {
                                 for (i = 0; i < argc; i++)
                                         RPC_FREE_VAL(&vals[i]);
                                 rpc_srv_freeValsCall(f);
                                 vals = NULL;
                                 argc = 0;
                                 ret = -7;
                                 rpc_SetErr(EMSGSIZE, "Error:: in prepare RPC packet values (-7) ...\n");
                                 goto makeReply;
                         } else
                                 Limit += argc * sizeof(rpc_val_t);
                         memcpy(v, vals, argc * sizeof(rpc_val_t));                          memcpy(v, vals, argc * sizeof(rpc_val_t));
                         Limit += argc * sizeof(rpc_val_t);  
                         data = (u_char*) v + argc * sizeof(rpc_val_t);                          data = (u_char*) v + argc * sizeof(rpc_val_t);
                         for (ret = i = 0; i < argc; i++) {                          for (ret = i = 0; i < argc; i++) {
                                 switch (vals[i].val_type) {                                  switch (vals[i].val_type) {
                                         case buffer:                                          case buffer:
                                                 if (ret || Limit + vals[i].val_len > BUFSIZ) {                                                  if (ret || Limit + vals[i].val_len > BUFSIZ) {
                                                         rpc_SetErr(EMSGSIZE, "Error:: in prepare RPC packet (-7) ...\n");                                                          rpc_SetErr(EMSGSIZE, "Error:: in prepare RPC packet (-7) ...\n");
                                                        rrpc.ret_retcode = ret = -7;                                                        rrpc->ret_retcode = ret = -7;
                                                        rrpc.ret_argc = 0;                                                        rrpc->ret_argc = 0;
                                                         break;                                                          break;
                                                 }                                                  }
   
Line 186  makeReply: Line 199  makeReply:
                                                 Limit += vals[i].val_len;                                                  Limit += vals[i].val_len;
                                                 break;                                                  break;
                                         case string:                                          case string:
                                                if (ret || Limit + vals[i].val_len + 1 > BUFSIZ) {                                                if (ret || Limit + vals[i].val_len > BUFSIZ) {
                                                         rpc_SetErr(EMSGSIZE, "Error:: in prepare RPC packet (-7) ...\n");                                                          rpc_SetErr(EMSGSIZE, "Error:: in prepare RPC packet (-7) ...\n");
                                                        rrpc.ret_retcode = ret = -7;                                                        rrpc->ret_retcode = ret = -7;
                                                        rrpc.ret_argc = 0;                                                        rrpc->ret_argc = 0;
                                                         break;                                                          break;
                                                 }                                                  }
   
                                                memcpy(data, vals[i].val.string, vals[i].val_len + 1);                                                memcpy(data, vals[i].val.string, vals[i].val_len);
                                                data += vals[i].val_len + 1;                                                data += vals[i].val_len;
                                                Limit += vals[i].val_len + 1;                                                Limit += vals[i].val_len;
                                                 break;                                                  break;
                                         case blob:                                          case blob:
                                                 if (s->srv_blob.state == disable) {                                                  if (s->srv_blob.state == disable) {
                                                         rpc_SetErr(ENOTSUP, "Error:: BLOB server is disabled\n");                                                          rpc_SetErr(ENOTSUP, "Error:: BLOB server is disabled\n");
                                                        rrpc.ret_retcode = ret = -5;                                                        rrpc->ret_retcode = ret = -5;
                                                        rrpc.ret_argc = 0;                                                        rrpc->ret_argc = 0;
                                                         break;                                                          break;
                                                 }                                                  }
                                         default:                                          default:
Line 210  makeReply: Line 223  makeReply:
   
                                 RPC_FREE_VAL(&vals[i]);                                  RPC_FREE_VAL(&vals[i]);
                         }                          }
                           rpc_srv_freeValsCall(f);
                           vals = NULL;
                           argc = 0;
                 }                  }
   
                 if ((ret = send(c->cli_sock, buf, Limit, 0)) == -1) {                  if ((ret = send(c->cli_sock, buf, Limit, 0)) == -1) {
Line 543  rpc_srv_execBLOBServer(rpc_srv_t * __restrict srv) Line 559  rpc_srv_execBLOBServer(rpc_srv_t * __restrict srv)
  * @regProgID = ProgramID for authentication & recognition   * @regProgID = ProgramID for authentication & recognition
  * @regProcID = ProcessID 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
 * @family = Family socket type, AF_INET or AF_INET6 * @sockType = Socket type, like SOCK_STREAM, SOCK_DGRAM or SOCK_RAW
 * @csHost = Host name or IP address for bind server, if NULL any address * @family = Family type, AF_INET, AF_INET6 or AF_LOCAL
  * @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
  * 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_int regProcID, int concurentClients,   rpc_srv_initServer(u_int regProgID, u_int regProcID, int concurentClients, 
                u_short family, const char *csHost, u_short Port)                int sockType, u_short family, const char *csHost, u_short Port)
 {  {
         rpc_srv_t *srv = NULL;          rpc_srv_t *srv = NULL;
         int n = 1;          int n = 1;
         struct hostent *host = NULL;          struct hostent *host = NULL;
        struct sockaddr_in sin;        struct sockaddr sa;
        struct sockaddr_in6 sin6;        struct sockaddr_in *sin = (struct sockaddr_in*) &sa;
         struct sockaddr_in6 *sin6 = (struct sockaddr_in6*) &sa;
         struct sockaddr_un *sun = (struct sockaddr_un*) &sa;
   
        if (!concurentClients || !regProgID || (family != AF_INET && family != AF_INET6)) {        if (!concurentClients || !regProgID || 
                         (family != AF_INET && family != AF_INET6 && family != AF_LOCAL)) {
                 rpc_SetErr(EINVAL, "Error:: Invalid parameters can`t init RPC server ...\n");                  rpc_SetErr(EINVAL, "Error:: Invalid parameters can`t init RPC server ...\n");
                 return NULL;                  return NULL;
         }          }
         if (!Port)          if (!Port)
                 Port = RPC_DEFPORT;                  Port = RPC_DEFPORT;
        if (csHost) {        if (csHost && family != AF_LOCAL) {
                 host = gethostbyname2(csHost, family);                  host = gethostbyname2(csHost, family);
                 if (!host) {                  if (!host) {
                         rpc_SetErr(h_errno, "Error:: %s\n", hstrerror(h_errno));                          rpc_SetErr(h_errno, "Error:: %s\n", hstrerror(h_errno));
                         return NULL;                          return NULL;
                 }                  }
         }          }
           memset(&sa, 0, sizeof sa);
           sa.sa_family = family;
         switch (family) {          switch (family) {
                 case AF_INET:                  case AF_INET:
                        memset(&sin, 0, sizeof sin);                        sin->sin_len = sizeof(struct sockaddr_in);
                        sin.sin_len = sizeof sin;                        sin->sin_port = htons(Port);
                        sin.sin_family = family; 
                        sin.sin_port = htons(Port); 
                         if (csHost)                          if (csHost)
                                memcpy(&sin.sin_addr, host->h_addr, host->h_length);                                memcpy(&sin->sin_addr, host->h_addr, host->h_length);
                         break;                          break;
                 case AF_INET6:                  case AF_INET6:
                        memset(&sin6, 0, sizeof sin6);                        sin6->sin6_len = sizeof(struct sockaddr_in6);
                        sin6.sin6_len = sizeof sin6;                        sin6->sin6_port = htons(Port);
                        sin6.sin6_family = family; 
                        sin6.sin6_port = htons(Port); 
                         if (csHost)                          if (csHost)
                                memcpy(&sin6.sin6_addr, host->h_addr, host->h_length);                                memcpy(&sin6->sin6_addr, host->h_addr, host->h_length);
                         break;                          break;
                   case AF_LOCAL:
                           sun->sun_len = sizeof(struct sockaddr_un);
                           if (csHost)
                                   strlcpy(sun->sun_path, csHost, sizeof sun->sun_path);
                           break;
                 default:                  default:
                         rpc_SetErr(EINVAL, "Error:: Invalid parameters can`t start RPC server ...\n");                          rpc_SetErr(EINVAL, "Error:: Invalid parameters can`t start RPC server ...\n");
                         return NULL;                          return NULL;
Line 607  rpc_srv_initServer(u_int regProgID, u_int regProcID, i Line 630  rpc_srv_initServer(u_int regProgID, u_int regProcID, i
   
         srv->srv_server.cli_tid = pthread_self();          srv->srv_server.cli_tid = pthread_self();
         srv->srv_server.cli_parent = srv;          srv->srv_server.cli_parent = srv;
        if (family == AF_INET)        switch (family) {
                memcpy(&srv->srv_server.cli_sa, &sin, sizeof srv->srv_server.cli_sa);                case AF_INET:
        else                        memcpy(&srv->srv_server.cli_sa, &sin, sizeof srv->srv_server.cli_sa);
                memcpy(&srv->srv_server.cli_sa, &sin6, sizeof srv->srv_server.cli_sa);                        break;
        srv->srv_server.cli_sock = socket(family, SOCK_STREAM, 0);                case AF_INET6:
                         memcpy(&srv->srv_server.cli_sa, &sin6, sizeof srv->srv_server.cli_sa);
                         break;
                 case AF_LOCAL:
                         memcpy(&srv->srv_server.cli_sa, &sun, sizeof srv->srv_server.cli_sa);
                         unlink(sun->sun_path);
                         break;
         }
 
         /* create server socket */
         srv->srv_server.cli_sock = socket(family, sockType ? sockType : SOCK_STREAM, 0);
         if (srv->srv_server.cli_sock == -1) {          if (srv->srv_server.cli_sock == -1) {
                 LOGERR;                  LOGERR;
                 free(srv);                  free(srv);
Line 630  rpc_srv_initServer(u_int regProgID, u_int regProcID, i Line 663  rpc_srv_initServer(u_int regProgID, u_int regProcID, i
                 return NULL;                  return NULL;
         }          }
   
           /* allocate pool for concurent clients */
         srv->srv_clients = calloc(srv->srv_numcli, sizeof(rpc_cli_t));          srv->srv_clients = calloc(srv->srv_numcli, sizeof(rpc_cli_t));
         if (!srv->srv_clients) {          if (!srv->srv_clients) {
                 LOGERR;                  LOGERR;
Line 667  rpc_srv_endServer(rpc_srv_t * __restrict srv) Line 701  rpc_srv_endServer(rpc_srv_t * __restrict srv)
   
         rpc_srv_endBLOBServer(srv);          rpc_srv_endBLOBServer(srv);
   
           /* close all clients connections & server socket */
         for (i = 0, c = srv->srv_clients; i < srv->srv_numcli && c; i++, c++)          for (i = 0, c = srv->srv_clients; i < srv->srv_numcli && c; i++, c++)
                 if (c->cli_sa.sa_family) {                  if (c->cli_sa.sa_family) {
                         shutdown(c->cli_sock, SHUT_RDWR);                          shutdown(c->cli_sock, SHUT_RDWR);
Line 680  rpc_srv_endServer(rpc_srv_t * __restrict srv) Line 715  rpc_srv_endServer(rpc_srv_t * __restrict srv)
                 srv->srv_numcli = 0;                  srv->srv_numcli = 0;
         }          }
   
           /* detach exported calls */
         pthread_mutex_lock(&srv->srv_mtx);          pthread_mutex_lock(&srv->srv_mtx);
         while ((f = srv->srv_funcs)) {          while ((f = srv->srv_funcs)) {
                 srv->srv_funcs = f->func_next;                  srv->srv_funcs = f->func_next;
Removed from v.1.2  
changed lines
  Added in v.1.3.2.1

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