1 /*
2  * Copyright (c) 2023 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 #include "napi/native_api.h"
17 #include "common/native_common.h"
18 #include <cstdio>
19 #include <cstring>
20 #include <fcntl.h>
21 #include <js_native_api_types.h>
22 #include <malloc.h>
23 #include <unistd.h>
24 
25 #define SUCCESS 1
26 #define STRLENGTH 64
27 #define DEFAULT_VALUE 0
28 #define FAIL (-1)
29 #define ONE 1
30 #define TWO 2
31 #define THREE 3
32 #define BUFF_SIZE (20)
33 #define FILEFLAG (0640)
34 #define PARAM_0 0
35 #define PARAM_1 1
36 #define PARAM_2 2
37 #define PARAM_UNNORMAL (-1)
38 #define RETURN_0 0
39 #define FAILD (-1)
40 #define ERRON_0 0
41 #define SIZE_6 6
42 #define SIZE_8 8
43 #define SIZE_10 10
44 #define SIZE_100 100
45 #define SIZE_123 123
46 #define SIZE_256 256
47 #define SIZE_1024 1024
48 #define SIZE_4096 4096
49 #define SIZE_8192 8192
50 #define SIZE_32768 32768
51 #define PARAM_0777 0777
52 extern "C" size_t __fwrite_chk(const void *buf, size_t size, size_t count, FILE *stream, size_t buf_size);
53 extern "C" int __open_chk(const char *, int);
54 extern "C" int __openat_chk(int, const char *, int);
55 extern "C" int __open64_chk(const char *, int);
56 extern "C" int __openat64_chk(int, const char *, int);
57 
Splice(napi_env env, napi_callback_info info)58 static napi_value Splice(napi_env env, napi_callback_info info)
59 {
60     size_t argc = PARAM_1;
61     napi_value args[1] = {nullptr};
62     napi_get_cb_info(env, info, &argc, args, nullptr, nullptr);
63     int value = napi_get_value_int32(env, args[0], &value);
64     int bytes = PARAM_0;
65     if (value == ONE) {
66         off_t off = PARAM_0;
67         size_t len = PARAM_0;
68         bytes = splice(PARAM_0, &off, PARAM_2, &off, len, PARAM_0);
69     } else if (value == TWO) {
70         int pipe1[PARAM_1 + PARAM_1];
71         size_t len = SIZE_256;
72         int flags = SPLICE_F_MORE | SPLICE_F_MOVE;
73         bytes = splice(PARAM_0, nullptr, pipe1[PARAM_1], nullptr, len, flags);
74     } else if (value == THREE) {
75         bytes = splice(PARAM_UNNORMAL, nullptr, PARAM_UNNORMAL, nullptr, PARAM_UNNORMAL, PARAM_UNNORMAL);
76     }
77     napi_value result = nullptr;
78     napi_create_int32(env, bytes, &result);
79     return result;
80 }
Tee(napi_env env, napi_callback_info info)81 static napi_value Tee(napi_env env, napi_callback_info info)
82 {
83     char buf[BUFF_SIZE];
84     char text[] = "Hello";
85     int result;
86     memset(buf, PARAM_0, sizeof(buf));
87     int pipefd1[PARAM_2];
88     pipe(pipefd1);
89     write(pipefd1[PARAM_1], text, strlen(text));
90     int pipefd2[PARAM_2];
91     pipe(pipefd2);
92     result = tee(pipefd1[PARAM_0], pipefd2[PARAM_1], SIZE_32768, SPLICE_F_NONBLOCK);
93     close(pipefd1[PARAM_0]);
94     close(pipefd1[PARAM_1]);
95     close(pipefd2[PARAM_0]);
96     close(pipefd2[PARAM_1]);
97     napi_value resultS = nullptr;
98     napi_create_int32(env, result, &resultS);
99     return resultS;
100 }
Vmsplice(napi_env env, napi_callback_info info)101 static napi_value Vmsplice(napi_env env, napi_callback_info info)
102 {
103     int pipeSize = PARAM_2;
104     int iovLength = SIZE_6;
105     int pipe_fds[pipeSize];
106     pipe(pipe_fds);
107     struct iovec v[pipeSize];
108     char strHello[] = "hello ";
109     char str_world[] = "world\n";
110     v[0].iov_base = strHello;
111     v[0].iov_len = iovLength;
112     v[1].iov_base = str_world;
113     v[1].iov_len = iovLength;
114     size_t result = vmsplice(pipe_fds[PARAM_1], v, sizeof(v) / sizeof(struct iovec), PARAM_0);
115     close(pipe_fds[1]);
116     char buf[BUFSIZ];
117     FILE *fp = fdopen(pipe_fds[PARAM_0], "r");
118     fgets(buf, sizeof(buf), fp);
119     fclose(fp);
120     napi_value resultS = nullptr;
121     napi_create_int32(env, result, &resultS);
122     return resultS;
123 }
124 
Readahead(napi_env env, napi_callback_info info)125 static napi_value Readahead(napi_env env, napi_callback_info info)
126 {
127     int resultValue = readahead(PARAM_UNNORMAL, PARAM_0, SIZE_123);
128     napi_value result = nullptr;
129 
130     napi_create_int32(env, resultValue, &result);
131     return result;
132 }
133 
PosixFallocate(napi_env env, napi_callback_info info)134 static napi_value PosixFallocate(napi_env env, napi_callback_info info)
135 {
136     napi_value result = nullptr;
137     int fd = open("/data/storage/el2/base/files/fff.txt", O_CREAT | O_RDWR, PARAM_0777);
138     int ret = posix_fallocate(fd, SIZE_8, SIZE_1024);
139     napi_create_int32(env, ret, &result);
140     close(fd);
141     return result;
142 }
143 
PosixFallocate64(napi_env env, napi_callback_info info)144 static napi_value PosixFallocate64(napi_env env, napi_callback_info info)
145 {
146     napi_value result = nullptr;
147     int fd = open("/data/storage/el2/base/files/fff.txt", O_CREAT | O_RDWR, PARAM_0777);
148     int ret = posix_fallocate64(fd, SIZE_8, SIZE_1024);
149     napi_create_int32(env, ret, &result);
150     close(fd);
151     return result;
152 }
153 
NameToHandleAt(napi_env env, napi_callback_info info)154 static napi_value NameToHandleAt(napi_env env, napi_callback_info info)
155 {
156     napi_value result = nullptr;
157     napi_create_int32(env, RETURN_0, &result);
158     return result;
159 }
160 
Open(napi_env env, napi_callback_info info)161 static napi_value Open(napi_env env, napi_callback_info info)
162 {
163     int fd = open("/data/storage/el2/base/files/test.txt", O_CREAT, PARAM_0777);
164     int ret = FAIL;
165     if (fd != FAIL) {
166         ret = SUCCESS;
167     }
168     close(fd);
169     napi_value result = nullptr;
170     napi_create_int32(env, ret, &result);
171     return result;
172 }
173 
Open64(napi_env env, napi_callback_info info)174 static napi_value Open64(napi_env env, napi_callback_info info)
175 {
176     int fd = open64("/data/storage/el2/base/files/test.txt", O_CREAT);
177     int ret = FAIL;
178     if (fd != FAIL) {
179         ret = SUCCESS;
180     }
181     close(fd);
182     napi_value result = nullptr;
183     napi_create_int32(env, ret, &result);
184     return result;
185 }
186 
OpenByHandleAt(napi_env env, napi_callback_info info)187 static napi_value OpenByHandleAt(napi_env env, napi_callback_info info)
188 {
189     napi_value result = nullptr;
190     napi_create_int32(env, PARAM_1, &result);
191     return result;
192 }
193 
Openat(napi_env env, napi_callback_info info)194 static napi_value Openat(napi_env env, napi_callback_info info)
195 {
196     int fd;
197     int dirFd = open("/data/storage/el2/base/files/test.txt", O_CREAT, PARAM_0777);
198     fd = openat(dirFd, "/data/storage/el2/base/files/test1.txt", O_CREAT | O_RDWR | O_TRUNC, FILEFLAG);
199     int ret = FAIL;
200     if (fd != FAIL) {
201         ret = SUCCESS;
202     }
203     close(fd);
204     close(dirFd);
205     napi_value result = nullptr;
206     napi_create_int32(env, ret, &result);
207     return result;
208 }
209 
Openat64(napi_env env, napi_callback_info info)210 static napi_value Openat64(napi_env env, napi_callback_info info)
211 {
212     int fd;
213     int dirFd = open("/data/storage/el2/base/files/test.txt", O_CREAT, PARAM_0777);
214     fd = openat64(dirFd, "/data/storage/el2/base/files/test2.txt", O_CREAT | O_RDWR | O_TRUNC, FILEFLAG);
215     int ret = FAIL;
216     if (fd != FAIL) {
217         ret = SUCCESS;
218     }
219     close(fd);
220     close(dirFd);
221     napi_value result = nullptr;
222     napi_create_int32(env, ret, &result);
223     return result;
224 }
225 
PosixFadvise(napi_env env, napi_callback_info info)226 static napi_value PosixFadvise(napi_env env, napi_callback_info info)
227 {
228     napi_value result = nullptr;
229     int fileDescribe = open("/data/storage/el2/base/files/fzl.txt", O_CREAT | O_RDWR, PARAM_0777);
230     int returnValue = posix_fadvise(fileDescribe, SIZE_8, SIZE_1024, POSIX_FADV_NORMAL);
231     napi_create_int32(env, returnValue, &result);
232     close(fileDescribe);
233     return result;
234 }
235 
PosixFadvise64(napi_env env, napi_callback_info info)236 static napi_value PosixFadvise64(napi_env env, napi_callback_info info)
237 {
238     napi_value result = nullptr;
239     int fileDescribe = open("/data/storage/el2/base/files/fzl.txt", O_CREAT | O_RDWR, PARAM_0777);
240     int returnValue = posix_fadvise64(fileDescribe, SIZE_8, SIZE_1024, POSIX_FADV_NORMAL);
241     napi_create_int32(env, returnValue, &result);
242     close(fileDescribe);
243     return result;
244 }
245 
FalLocate(napi_env env, napi_callback_info info)246 static napi_value FalLocate(napi_env env, napi_callback_info info)
247 {
248     int backParam, mode = PARAM_0;
249     off_t offset = PARAM_0, len = PARAM_1;
250     int fileDescribe = open("/data/storage/el2/base/files/test.txt", O_CREAT | O_RDWR, PARAM_0777);
251     backParam = fallocate(fileDescribe, mode, offset, len);
252     napi_value result = nullptr;
253     napi_create_int32(env, backParam, &result);
254     close(fileDescribe);
255     return result;
256 }
257 
FalLocate64(napi_env env, napi_callback_info info)258 static napi_value FalLocate64(napi_env env, napi_callback_info info)
259 {
260     int backParam, mode = PARAM_0;
261     off_t offset = PARAM_0, len = PARAM_1;
262     int fileDescribe = open("/data/storage/el2/base/files/test.txt", O_CREAT | O_RDWR, PARAM_0777);
263     backParam = fallocate64(fileDescribe, mode, offset, len);
264     napi_value result = nullptr;
265     napi_create_int32(env, backParam, &result);
266     close(fileDescribe);
267     return result;
268 }
269 
FcnTl(napi_env env, napi_callback_info info)270 static napi_value FcnTl(napi_env env, napi_callback_info info)
271 {
272     int backParam;
273     int fileDescribe = open("/data/storage/el2/base/files/Fzl.txt", O_CREAT, PARAM_0777);
274     backParam = fcntl(fileDescribe, STDIN_FILENO, F_GETFL, PARAM_0);
275     napi_value result = nullptr;
276     if (backParam != fileDescribe && backParam != -1) {
277         napi_create_int32(env, PARAM_0, &result);
278     } else {
279         napi_create_int32(env, PARAM_UNNORMAL, &result);
280     }
281     close(fileDescribe);
282     return result;
283 }
284 
FwriteChk(napi_env env, napi_callback_info info)285 static napi_value FwriteChk(napi_env env, napi_callback_info info)
286 {
287     const char *msg = "This is a c";
288     int len = SIZE_10;
289     FILE *files = fopen("/data/storage/el2/base/files/test.txt", "w");
290     NAPI_ASSERT(env, files != nullptr, "FwriteChk fopen Error");
291     int result = __fwrite_chk(msg, len, PARAM_0, files, strlen(msg));
292     fclose(files);
293     napi_value resultS = nullptr;
294     napi_create_int32(env, result, &resultS);
295     return resultS;
296 }
OpenChk(napi_env env, napi_callback_info info)297 static napi_value OpenChk(napi_env env, napi_callback_info info)
298 {
299     int dirFileDescribe = open("/data/storage/el2/base/files/test.txt", O_CREAT, PARAM_0777);
300     int fileDescribe = __open_chk("/data/storage/el2/base/files/test.txt", O_RDWR);
301     int ret = FAIL;
302     if (fileDescribe != FAIL) {
303         ret = RETURN_0;
304     }
305     close(dirFileDescribe);
306     close(fileDescribe);
307     napi_value result = nullptr;
308     napi_create_int32(env, ret, &result);
309     return result;
310 }
311 
Open64Chk(napi_env env, napi_callback_info info)312 static napi_value Open64Chk(napi_env env, napi_callback_info info)
313 {
314     int dirFileDescribe = open("/data/storage/el2/base/files/test.txt", O_CREAT, PARAM_0777);
315     int fileDescribe = __open64_chk("/data/storage/el2/base/files/test.txt", O_RDWR);
316     int ret = FAIL;
317     if (fileDescribe != FAIL) {
318         ret = RETURN_0;
319     }
320     close(dirFileDescribe);
321     close(fileDescribe);
322     napi_value result = nullptr;
323     napi_create_int32(env, ret, &result);
324     return result;
325 }
OpenatChk(napi_env env, napi_callback_info info)326 static napi_value OpenatChk(napi_env env, napi_callback_info info)
327 {
328     int dirFileDescribe = open("/data/storage/el2/base/files/test.txt", O_CREAT, PARAM_0777);
329     int fileDescribe = __openat_chk(dirFileDescribe, "/data/storage/el2/base/files/test.txt", O_RDWR | O_TRUNC);
330     int ret = FAIL;
331     if (fileDescribe != FAIL) {
332         ret = RETURN_0;
333     }
334     close(dirFileDescribe);
335     close(fileDescribe);
336     napi_value result = nullptr;
337     napi_create_int32(env, ret, &result);
338     return result;
339 }
Openat64Chk(napi_env env, napi_callback_info info)340 static napi_value Openat64Chk(napi_env env, napi_callback_info info)
341 {
342     int dirFileDescribe = open("/data/storage/el2/base/files/test.txt", O_CREAT, PARAM_0777);
343     int fileDescribe = __openat64_chk(dirFileDescribe, "/data/storage/el2/base/files/test.txt", O_RDWR);
344     int ret = FAIL;
345     if (fileDescribe != FAIL) {
346         ret = RETURN_0;
347     }
348     close(dirFileDescribe);
349     close(fileDescribe);
350     napi_value result = nullptr;
351     napi_create_int32(env, ret, &result);
352     return result;
353 }
354 
355 EXTERN_C_START
Init(napi_env env, napi_value exports)356 static napi_value Init(napi_env env, napi_value exports)
357 {
358     napi_property_descriptor desc[] = {
359         {"nameToHandleAt", nullptr, NameToHandleAt, nullptr, nullptr, nullptr, napi_default, nullptr},
360         {"open", nullptr, Open, nullptr, nullptr, nullptr, napi_default, nullptr},
361         {"openByHandleAt", nullptr, OpenByHandleAt, nullptr, nullptr, nullptr, napi_default, nullptr},
362         {"openat", nullptr, Openat, nullptr, nullptr, nullptr, napi_default, nullptr},
363         {"open64", nullptr, Open64, nullptr, nullptr, nullptr, napi_default, nullptr},
364         {"openat64", nullptr, Openat64, nullptr, nullptr, nullptr, napi_default, nullptr},
365         {"splice", nullptr, Splice, nullptr, nullptr, nullptr, napi_default, nullptr},
366         {"tee", nullptr, Tee, nullptr, nullptr, nullptr, napi_default, nullptr},
367         {"vmsplice", nullptr, Vmsplice, nullptr, nullptr, nullptr, napi_default, nullptr},
368         {"readahead", nullptr, Readahead, nullptr, nullptr, nullptr, napi_default, nullptr},
369         {"posixfallocate", nullptr, PosixFallocate, nullptr, nullptr, nullptr, napi_default, nullptr},
370         {"posixfadvise", nullptr, PosixFadvise, nullptr, nullptr, nullptr, napi_default, nullptr},
371         {"posixfadvise64", nullptr, PosixFadvise64, nullptr, nullptr, nullptr, napi_default, nullptr},
372         {"posixfallocate64", nullptr, PosixFallocate64, nullptr, nullptr, nullptr, napi_default, nullptr},
373         {"fcntl", nullptr, FcnTl, nullptr, nullptr, nullptr, napi_default, nullptr},
374         {"fallocate64", nullptr, FalLocate64, nullptr, nullptr, nullptr, napi_default, nullptr},
375         {"fallocate", nullptr, FalLocate, nullptr, nullptr, nullptr, napi_default, nullptr},
376         {"fwritechk", nullptr, FwriteChk, nullptr, nullptr, nullptr, napi_default, nullptr},
377         {"openChk", nullptr, OpenChk, nullptr, nullptr, nullptr, napi_default, nullptr},
378         {"open64Chk", nullptr, Open64Chk, nullptr, nullptr, nullptr, napi_default, nullptr},
379         {"openatChk", nullptr, OpenatChk, nullptr, nullptr, nullptr, napi_default, nullptr},
380         {"openat64Chk", nullptr, Openat64Chk, nullptr, nullptr, nullptr, napi_default, nullptr},
381 
382     };
383     napi_define_properties(env, exports, sizeof(desc) / sizeof(desc[0]), desc);
384     return exports;
385 }
386 EXTERN_C_END
387 
388 static napi_module demoModule = {
389     .nm_version = 1,
390     .nm_flags = 0,
391     .nm_filename = nullptr,
392     .nm_register_func = Init,
393     .nm_modname = "fcntl",
394     .nm_priv = ((void *)0),
395     .reserved = {0},
396 };
397 
RegisterModule(void)398 extern "C" __attribute__((constructor)) void RegisterModule(void) { napi_module_register(&demoModule); }
399