1 files changed, 153 insertions, 0 deletions
diff --git a/src/nSocket/include/nstack_select.h b/src/nSocket/include/nstack_select.h
new file mode 100644
index 0000000..bb23105
--- /dev/null
+++ b/src/nSocket/include/nstack_select.h
@@ -0,0 +1,153 @@
+/*
+*
+* Copyright (c) 2018 Huawei Technologies Co.,Ltd.
+* Licensed under the Apache License, Version 2.0 (the "License");
+* you may not use this file except in compliance with the License.
+* You may obtain a copy of the License at:
+*
+*     http://www.apache.org/licenses/LICENSE-2.0
+*
+* Unless required by applicable law or agreed to in writing, software
+* distributed under the License is distributed on an "AS IS" BASIS,
+* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+* See the License for the specific language governing permissions and
+* limitations under the License.
+*/
+
+/*==============================================*
+ *      include header files                    *
+ *----------------------------------------------*/
+
+/*==============================================*
+ *      constants or macros define              *
+ *----------------------------------------------*/
+
+/*==============================================*
+ *      project-wide global variables           *
+ *----------------------------------------------*/
+
+/*==============================================*
+ *      routines' or functions' implementations *
+ *----------------------------------------------*/
+
+#define NSTACK_SELECT_MODULE
+
+#ifdef NSTACK_SELECT_MODULE
+
+#ifndef __NSTACK_SELECT_H__
+#define __NSTACK_SELECT_H__
+
+#include <sys/types.h>
+#include <sys/time.h>
+#include <stdio.h>
+#include <unistd.h>
+#include "select_adapt.h"
+
+#ifdef __cplusplus
+/* *INDENT-OFF* */
+extern "C" {
+/* *INDENT-ON* */
+#endif
+
+typedef int (*get_select_event) (int nfds, fd_set * readfd,
+                                 fd_set * writefd, fd_set * exceptfd,
+                                 struct timeval * timeout);
+typedef struct
+{
+  unsigned char fds_bits[(NSTACK_SETSIZE + 7) >> 3];
+} __attribute__ ((packed)) nstack_fd_set;
+
+#define NSTACK_FD_SET(n, p)        ((p)->fds_bits[(n)>>3]|=1U<<((n)&0x07))
+#define NSTACK_FD_ISSET(n,p)    (((p)->fds_bits[(n)>>3]&(1U<<((n)&0x07)))?1:0)
+#define NSTACK_FD_CLR(n,p)        ((p)->fds_bits[(n)>>3]&=~(1U<<((n)&0x07)))
+#define NSTACK_FD_ZERO(p)        (MEMSET_S((void *)(p), sizeof(*(p)),0,sizeof(*(p))))
+#define NSTACK_FD_OR(p1 ,p2)     {\
+    int i;\
+    for(i = 0; i < (NSTACK_SETSIZE+7)>>3; i++){\
+        (p1)->fds_bits[i] |= (p2)->fds_bits[i];\
+    }\
+}
+
+struct select_cb_p
+{
+  union
+  {
+    nstack_fd_set nstack_readset;
+    fd_set readset;
+  };
+  union
+  {
+    nstack_fd_set nstack_writeset;
+    fd_set writeset;
+  };
+  union
+  {
+    nstack_fd_set nstack_exceptset;
+    fd_set exceptset;
+  };
+
+  union
+  {
+    i32 count;
+    i32 readyset;
+  };
+
+  i32 inx;
+  i32 select_errno;
+};
+
+struct select_entry_info
+{
+  i32 set_num;                  //how many select_c_p is set
+  i32 index;                    //the frist cb was set
+};
+
+struct select_entry
+{
+  struct select_cb_p cb[NSTACK_MAX_MODULE_NUM];
+  struct select_cb_p ready;
+  struct select_entry *next;
+  struct select_entry *prev;
+  struct select_entry_info info;
+  select_sem_t sem;
+};
+
+struct select_module_info
+{
+
+  struct select_entry *entry_head;
+  struct select_entry *entry_tail;
+  get_select_event get_select_fun_nonblock[NSTACK_MAX_MODULE_NUM];
+  get_select_event get_select_fun_block[NSTACK_MAX_MODULE_NUM];
+  get_select_event default_fun;
+  i32 default_mod;
+  volatile i32 inited;
+  select_spinlock_t lock;
+  select_sem_t sem;
+};
+
+extern i32 select_cb_split_by_mod (i32 nfds,
+                                   fd_set * readfd,
+                                   fd_set * writefd,
+                                   fd_set * exceptfd,
+                                   struct select_entry *entry);
+extern void entry_module_fdset (struct select_entry *entry,
+                                i32 fd_size,
+                                nstack_fd_set * readfd,
+                                nstack_fd_set * writefd,
+                                nstack_fd_set * exceptfd, i32 inx);
+extern i32 select_scan (struct select_entry *entry);
+extern i32 select_add_cb (struct select_entry *entry);
+extern i32 select_rm_cb (struct select_entry *entry);
+extern i32 select_entry_reset (struct select_entry *entry);
+extern i32 select_module_init ();
+extern struct select_module_info *get_select_module (void);
+#ifdef __cplusplus
+/* *INDENT-OFF* */
+}
+/* *INDENT-ON* */
+#endif
+
+#endif /* __NSTACK_SELECT_H__ */
+
+#endif /* NSTACK_SELECT_MODULE */