1 /*
2  * Copyright (c) 2024  Huawei Device Co., Ltd.
3  * Licensed under the Apache License, Version 2.0 (the "License");
4  * you may not use this file except in compliance with the License.
5  * You may obtain a copy of the License at
6  *
7  *     http://www.apache.org/licenses/LICENSE-2.0
8  *
9  * Unless required by applicable law or agreed to in writing, software
10  * distributed under the License is distributed on an "AS IS" BASIS,
11  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12  * See the License for the specific language governing permissions and
13  * limitations under the License.
14  */
15 
16 #ifndef NET_TRANS_COMMON_H
17 #define NET_TRANS_COMMON_H
18 
19 #include <pthread.h>
20 #include <securec.h>
21 #include <sys/time.h>
22 #include <unistd.h>
23 #include <inttypes.h>
24 #include "session.h"
25 #include "socket.h"
26 #include "softbus_bus_center.h"
27 #include "softbus_common.h"
28 #include "softbus_errcode.h"
29 #include "nativetoken_kit.h"
30 #include "token_setproc.h"
31 
32 #ifdef __cplusplus
33 extern "C" {
34 #endif
35 
36 #define DEF_GROUP_ID "DEF_GROUP_ID"
37 #define DEF_SESSION_KEY "DEF_SESSION_KEY"
38 #define DEF_PKG_NAME "com.communication.demo1"
39 #define PKG_NAME_TWO "com.communication.demo2"
40 #define SESSIONNAME_MAX "com.communication.demo1.max.len.12345678901234567\
41     0123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789\
42     01234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567"
43 #define SESSIONNAME_MAX_OUT "com.communication.demo1.max.len.12345678901234567\
44     0123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789\
45     012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678"
46 #define PKGNAME_MAX "com.communication.demo1.max.len.1234567890123456789012345678901"
47 #define PKGNAME_MAX_OUT "com.communication.demo1.max.len.123456789012345678901234567890123"
48 
49 #define SOCKET_NAME_BYTE "com.communication.demo1.socket.byte"
50 #define SOCKET_NAME_MESSAGE "com.communication.demo1.socket.message"
51 #define SOCKET_NAME_FILE "com.communication.demo1.socket.file"
52 #define SOCKET_NAME_STREAM "com.communication.demo1.socket.stream"
53 
54 #define DSOFT_PKG_NAME "com.plrdtest.dsoftbus"
55 #define SESSION_NAME_DSOFT_DATA "com.plrdtest.dsoftbus.data"
56 #define RECV_FILE_PATH "/data/"
57 
58 #define CTRL_MSG_CLOSE_WIFI_THREE_SEC "close wifi 3sec"
59 #define CTRL_MSG_CLOSE_WIFI_TEN_MIN "close wifi 10min"
60 #define CTRL_MSG_CLOSE_WIFI_FIVE_MIN "close wifi 5min"
61 #define CTRL_MSG_CLOSE_WIFI_TEN_SEC "close wifi 10sec"
62 #define CTRL_MSG_CHANGE_WIFI_TEN_SEC "change wifi 10sec"
63 #define CTRL_MSG_CHANGE_WIFI_SIXTY_SEC "change wifi 60sec"
64 #define CTRL_MSG_CLOSE_WIFI_LOOP20 "close wifi loop 20"
65 #define CTRL_MSG_CHANGE_WIFI_LOOP20 "change wifi loop 20"
66 #define CTRL_MSG_CLOSE_WIFI_LOOP100 "close wifi loop 100"
67 #define CTRL_MSG_CHANGE_WIFI_LOOP100 "change wifi loop 100"
68 #define CTRL_MSG_OPEN_SESSION "open session"
69 #define CTRL_MSG_OPEN_SESSION_MSG "open session msg"
70 #define CTRL_MSG_OPEN_SESSION_NOT_EXIST "open session not exist"
71 #define CTRL_MSG_OPEN_SESSION_FAIL "open session fail"
72 #define CTRL_MSG_CLOSE_SESSION "close session"
73 #define CTRL_MSG_CLOSE_BR "close br"
74 #define CTRL_MSG_CLOSE_OPEN_BR "close-open br"
75 #define CTRL_MSG_CLOSE_AIR "close air"
76 #define CTRL_MSG_CLOSE_OPEN_AIR "close-open air"
77 #define CTRL_MSG_SEND_BIG_DATA "send big data"
78 
79 #define WAIT_DEF_VALUE (-1)
80 #define WAIT_SUCCESS_VALUE 1
81 #define WAIT_FAIL_VALUE 0
82 #define SESSION_ID_MIN 1
83 #define MAX_SESSION_NUM 16
84 #define BOOL_TRUE 1
85 #define BOOL_FALSE 0
86 #define OPEN_SESSION_TIMEOUT 19
87 #define MULTI_REMOTE_DEV_COUNT 3
88 #define DEF_REMOTE_DEV_COUNT 1
89 #define SIZE_1K 1024
90 #define CMD_PIDOF_SOFTBUS "pidof softbus_server"
91 #define MAX_SESSION_SERVER_NUM_CLIENT 8
92 #define MAX_SESSION_SERVER_NUM_SERVER 32
93 #define TRANS_BYTES_LENGTH_MAX (4 * 1024 * 1024)
94 #define TRANS_MESSAGE_LENGTH_MAX (4 * 1024)
95 #define TRANS_PROXY_BYTES_LENGTH_MAX (4 * 1024 * 1024)
96 #define TRANS_PROXY_MESSAGE_LENGTH_MAX (4 * 1024)
97 #define DEF_SEND_DATA_SID_COUNT 4
98 #define GROUP_ID_LEN 4
99 #define ONE_SECOND 1
100 
101 #define CASE_0 0
102 #define CASE_1 1
103 #define CASE_2 2
104 #define CASE_3 3
105 #define CASE_4 4
106 #define CASE_5 5
107 #define CASE_6 6
108 #define CASE_7 7
109 #define CASE_8 8
110 #define CASE_9 9
111 #define CASE_10 10
112 
113 #define INDEX_0 0
114 #define INDEX_1 1
115 #define INDEX_2 2
116 #define INDEX_3 3
117 #define INDEX_4 4
118 
119 #define TEN_PLACE 10
120 #define HUNDRED_PLACE  100
121 #define THOUSAND_PLACE  1000
122 #define FRAME_10 10
123 #define FRAME_100 100
124 #define FRAME_1000 1000
125 /*stream*/
126 #define I_FRAME_TYPE (1)
127 #define P_FRAME_TYPE (2)
128 /*common*/
129 #define I_FRAME_SIZE0 (0 * 1024)
130 #define P_FRAME_SIZE0 (0 * 1024)
131 /*1080P10 I_FRAME 400k P_FRAME 27k*/
132 #define I_FRAME_SIZE1 (400 * 1024)
133 #define P_FRAME_SIZE1 (27 * 1024)
134 /*1080P30 I_FRAME 200k P_FRAME 46k*/
135 #define I_FRAME_SIZE2 (200 * 1024)
136 #define P_FRAME_SIZE2 (46 * 1024)
137 /*1080P60 I_FRAME 150k P_FRAME 30k*/
138 #define I_FRAME_SIZE3 (150 * 1024)
139 #define P_FRAME_SIZE3 (30 * 1024)
140 /*2kP60 I_FRAME 180k P_FRAME 62k*/
141 #define I_FRAME_SIZE4 (180 * 1024)
142 #define P_FRAME_SIZE4 (62 * 1024)
143 /*2kP60 I_FRAME 270k P_FRAME 60k*/
144 #define I_FRAME_SIZE5 (270 * 1024)
145 #define P_FRAME_SIZE5 (60 * 1024)
146 /*4KP30 I_FRAME 400k P_FRAME 119k*/
147 #define I_FRAME_SIZE6 (400 * 1024)
148 #define P_FRAME_SIZE6 (119 * 1024)
149 /*4KP30 I_FRAME 600k P_FRAME 112k*/
150 #define I_FRAME_SIZE7 (600 * 1024)
151 #define P_FRAME_SIZE7 (112 * 1024)
152 /*4KP60 I_FRAME 400K P_FRAME 123K*/
153 #define I_FRAME_SIZE8 (400 * 1024)
154 #define P_FRAME_SIZE8 (123 * 1024)
155 /*4KP60 I_FRAME 600K P_FRAME 120K*/
156 #define I_FRAME_SIZE9 (600 * 1024)
157 #define P_FRAME_SIZE9 (120 * 1024)
158 /*1080P60 I_FRAME 867K P_FRAME 7K*/
159 #define I_FRAME_SIZE10 (867 * 1024)
160 #define P_FRAME_SIZE10 (7 * 1024)
161 /*1080P60 I_FRAME 1330K P_FRAME 10K*/
162 #define I_FRAME_SIZE11 (1330 * 1024)
163 #define P_FRAME_SIZE11 (10 * 1024)
164 
165 typedef enum {
166     SOCKET_BYTES = 1,
167     SOCKET_MSG,
168     SOCKET_FILE,
169     SOCKET_STREAM,
170     SOCKET_SHUTDOWN,
171 }WaitSocketType;
172 
173 typedef enum {
174     DATA_TYPE_MSG = 1,
175     DATA_TYPE_BYTE,
176 } DataType;
177 
178 typedef enum {
179     STATE_ONLINE = 1,
180     STATE_OFFLINE,
181 } WaitNodeStateType;
182 
183 typedef enum {
184     CONC_CLOSE_SESSION = 1,
185     CONC_LEAVE_NET,
186     CONC_DIS_NET_RECOVERY,
187     CONC_DIS_NET_NOT_RECOVERY,
188     CONC_DOWN_NET,
189     CONC_SEND_DATA_ONLY,
190     CONC_CLOSE_SESSION_REMOTE,
191 } ConcurrentType;
192 
193 typedef enum {
194     CTRL_CODE_CLOSE_WIFI_TEN_SEC = 1,
195     CTRL_CODE_CLOSE_WIFI_TEN_MIN,
196     CTRL_CODE_CLOSE_WIFI_FIVE_MIN,
197     CTRL_CODE_CLOSE_WIFI_THREE_SEC,
198     CTRL_CODE_CHANGE_WIFI_TEN_SEC,
199     CTRL_CODE_CHANGE_WIFI_SIXTY_SEC,
200     CTRL_CODE_CHANGE_WIFI_LOOP20,
201     CTRL_CODE_CLOSE_WIFI_LOOP20,
202     CTRL_CODE_CHANGE_WIFI_LOOP100,
203     CTRL_CODE_CLOSE_WIFI_LOOP100,
204     CTRL_CODE_OPEN_SESSION,
205     CTRL_CODE_OPEN_SESSION_MSG,
206     CTRL_CODE_OPEN_SESSION_FAIL,
207     CTRL_CODE_OPEN_SESSION_NOT_EXIST,
208     CTRL_CODE_CLOSE_SESSION,
209     CTRL_CODE_CLOSE_BR,
210     CTRL_CODE_CLOSE_OPEN_BR,
211     CTRL_CODE_CLOSE_AIR,
212     CTRL_CODE_CLOSE_OPEN_AIR,
213     CTRL_CODE_SEND_BIG_DATA,
214 } CtrlCodeType;
215 
216 #define LOG(format, ...)                                                           \
217     do {                                                                           \
218         time_t timeSec;                                                            \
219         time(&timeSec);                                                            \
220         struct tm tmRst;                                                           \
221         localtime_r(&timeSec, &tmRst);                                             \
222         char strTime[10];                                                          \
223         strftime(strTime, sizeof(strTime), "%H:%M:%S", &tmRst);                    \
224         fprintf(stdout, "[Test-softbus] %s " format "\n", strTime, ##__VA_ARGS__); \
225     } while (0)
226 
227 int Wait4Socket(int timeout, WaitSocketType type);
228 int WaitNodeCount(int timeout, WaitNodeStateType state, int expectCount);
229 
230 void Sleepn(int n);
231 void TestSetUp(void);
232 void TestTearDown(void);
233 
234 int RegisterDeviceStateDefCallback(void);
235 int UnRegisterDeviceStateDefCallback(void);
236 int CheckRemoteDeviceIsNull(int isSetNetId);
237 
238 void ResetWaitFlag(void);
239 void ResetWaitFlag4Shutdown(void);
240 void ResetWaitFlag4Byte(void);
241 void ResetWaitFlag4Message(void);
242 void ResetWaitFlag4Stream(void);
243 void ResetWaitFlag4File(void);
244 void ResetwaitCount4Online(void);
245 void ResetwaitCount4Offline(void);
246 void AddPermission(void);
247 
248 int SendSocket4Data(DataType type, int size);
249 
250 char* GetSoftbusPid(void);
251 char* GetNetworkId(void);
252 
253 void SetCurrentSocket4Data(int socket);
254 
255 int GetCurrentSocket4Data(void);
256 
257 ISocketListener* GetSocketListnerData(void);
258 
259 #ifdef __cplusplus
260 }
261 #endif
262 #endif