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 auto [resGetFirstArg, fd] = NVal(env, funcArg[NARG_POS::FIRST]).ToInt32();
40 if (!resGetFirstArg || fd < 0) {
41 UniError(EINVAL).ThrowErr(env, "Invalid fd");
42 return nullptr;
43 }
44
45 int64_t len = 0;
46 if (funcArg.GetArgc() == NARG_CNT::TWO) {
47 bool succ = false;
48 tie(succ, len) = NVal(env, funcArg[NARG_POS::SECOND]).ToInt64(len);
49 if (!succ) {
50 UniError(EINVAL).ThrowErr(env, "Invalid len");
51 return nullptr;
52 }
53 }
54 int ret = ftruncate(fd, len);
55 if (ret == -1) {
56 UniError(errno).ThrowErr(env);
57 return nullptr;
58 }
59
60 return NVal::CreateUndefined(env).val_;
61 }
62
Async(napi_env env,napi_callback_info info)63 napi_value Ftruncate::Async(napi_env env, napi_callback_info info)
64 {
65 NFuncArg funcArg(env, info);
66 if (!funcArg.InitArgs(NARG_CNT::ONE, NARG_CNT::THREE)) {
67 UniError(EINVAL).ThrowErr(env, "Number of arguments unmatched");
68 return nullptr;
69 }
70
71 auto [resGetFirstArg, fd] = NVal(env, funcArg[NARG_POS::FIRST]).ToInt32();
72 if (!resGetFirstArg || fd < 0) {
73 UniError(EINVAL).ThrowErr(env, "Invalid fd");
74 return nullptr;
75 }
76
77 int64_t len = 0;
78 if (funcArg.GetArgc() >= NARG_CNT::TWO) {
79 bool resGetSecondArg = false;
80 tie(resGetSecondArg, len) = NVal(env, funcArg[NARG_POS::SECOND]).ToInt64(len);
81 if (!resGetSecondArg) {
82 UniError(EINVAL).ThrowErr(env, "Invalid len");
83 return nullptr;
84 }
85 }
86
87 auto cbExec = [fd = fd, len = len](napi_env env) -> UniError {
88 int ret = ftruncate(fd, len);
89 if (ret == -1) {
90 return UniError(errno);
91 } else {
92 return UniError(ERRNO_NOERR);
93 }
94 };
95
96 auto cbComplete = [](napi_env env, UniError err) -> NVal {
97 if (err) {
98 return { env, err.GetNapiErr(env) };
99 } else {
100 return NVal::CreateUndefined(env);
101 }
102 };
103
104 const string procedureName = "fileIOFtruncate";
105 NVal thisVar(env, funcArg.GetThisVar());
106 if (funcArg.GetArgc() == NARG_CNT::ONE || (funcArg.GetArgc() == NARG_CNT::TWO &&
107 !NVal(env, funcArg[NARG_POS::SECOND]).TypeIs(napi_function))) {
108 return NAsyncWorkPromise(env, thisVar).Schedule(procedureName, cbExec, cbComplete).val_;
109 } else {
110 NVal cb(env, funcArg[((funcArg.GetArgc() == NARG_CNT::TWO) ? NARG_POS::SECOND : NARG_POS::THIRD)]);
111 return NAsyncWorkCallback(env, thisVar, cb).Schedule(procedureName, cbExec, cbComplete).val_;
112 }
113 }
114 } // namespace ModuleFileIO
115 } // namespace DistributedFS
116 } // namespace OHOS