1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
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 first 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 */
|
