1 /*
2 * Copyright (c) 2021 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 "ftruncate.h"
17
18 #include <cstring>
19 #include <tuple>
20 #include <unistd.h>
21
22 #include "../../common/napi/n_async/n_async_work_callback.h"
23 #include "../../common/napi/n_async/n_async_work_promise.h"
24 #include "../../common/napi/n_func_arg.h"
25
26 namespace OHOS {
27 namespace DistributedFS {
28 namespace ModuleFileIO {
29 using namespace std;
30
Sync(napi_env env,napi_callback_info info)31 napi_value Ftruncate::Sync(napi_env env, napi_callback_info info)
32 {
33 NFuncArg funcArg(env, info);
34 if (!funcArg.InitArgs(NARG_CNT::ONE, NARG_CNT::TWO)) {
35 UniError(EINVAL).ThrowErr(env, "Number of arguments unmatched");
36 return nullptr;
37 }
38
39 bool succ = false;
40
41 int fd;
42 tie(succ, fd) = NVal(env, funcArg[NARG_POS::FIRST]).ToInt32();
43 if (!succ) {
44 UniError(EINVAL).ThrowErr(env, "Invalid fd");
45 return nullptr;
46 }
47
48 size_t argc = funcArg.GetArgc();
49 int ret = -1;
50 if (argc == NARG_CNT::ONE) {
51 ret = ftruncate(fd, 0);
52 } else {
53 int len;
54 tie(succ, len) = NVal(env, funcArg[NARG_POS::SECOND]).ToInt32();
55 if (!succ) {
56 UniError(EINVAL).ThrowErr(env, "Invalid len");
57 return nullptr;
58 }
59 ret = ftruncate(fd, len);
60 }
61
62 if (ret == -1) {
63 UniError(errno).ThrowErr(env);
64 return nullptr;
65 }
66
67 return NVal::CreateUndefined(env).val_;
68 }
Async(napi_env env,napi_callback_info info)69 napi_value Ftruncate::Async(napi_env env, napi_callback_info info)
70 {
71 NFuncArg funcArg(env, info);
72 if (!funcArg.InitArgs(NARG_CNT::ONE, NARG_CNT::THREE)) {
73 UniError(EINVAL).ThrowErr(env, "Number of arguments unmatched");
74 return nullptr;
75 }
76 bool succ = false;
77 int fd;
78 int len = 0;
79 tie(succ, fd) = NVal(env, funcArg[NARG_POS::FIRST]).ToInt32();
80 if (!succ) {
81 UniError(EINVAL).ThrowErr(env, "Invalid fd");
82 return nullptr;
83 }
84 size_t argc = funcArg.GetArgc();
85 if (argc > NARG_CNT::ONE) {
86 tie(succ, len) = NVal(env, funcArg[NARG_POS::SECOND]).ToInt32();
87 if (!succ) {
88 UniError(EINVAL).ThrowErr(env, "Invalid len");
89 return nullptr;
90 }
91 }
92
93 auto cbExec = [fd, len](napi_env env) -> UniError {
94 int ret = ftruncate(fd, len);
95 if (ret == -1) {
96 return UniError(errno);
97 } else {
98 return UniError(ERRNO_NOERR);
99 }
100 };
101 auto cbComplete = [](napi_env env, UniError err) -> NVal {
102 if (err) {
103 return { env, err.GetNapiErr(env) };
104 } else {
105 return NVal::CreateUndefined(env);
106 }
107 };
108 string procedureName = "fileIOFtruncate";
109 NVal thisVar(env, funcArg.GetThisVar());
110 if (argc == NARG_CNT::ONE || (argc == NARG_CNT::TWO && NVal(env, funcArg[NARG_POS::SECOND]).TypeIs(napi_number))) {
111 return NAsyncWorkPromise(env, thisVar).Schedule(procedureName, cbExec, cbComplete).val_;
112 } else {
113 NVal cb(env, funcArg[NARG_POS::THIRD]);
114 return NAsyncWorkCallback(env, thisVar, cb).Schedule(procedureName, cbExec, cbComplete).val_;
115 }
116 }
117 } // namespace ModuleFileIO
118 } // namespace DistributedFS
119 } // namespace OHOS