1b1b8bc3fSopenharmony_ci/*
2b1b8bc3fSopenharmony_ci * Copyright (c) 2024 Huawei Device Co., Ltd.
3b1b8bc3fSopenharmony_ci * Licensed under the Apache License, Version 2.0 (the "License");
4b1b8bc3fSopenharmony_ci * you may not use this file except in compliance with the License.
5b1b8bc3fSopenharmony_ci * You may obtain a copy of the License at
6b1b8bc3fSopenharmony_ci *
7b1b8bc3fSopenharmony_ci *     http://www.apache.org/licenses/LICENSE-2.0
8b1b8bc3fSopenharmony_ci *
9b1b8bc3fSopenharmony_ci * Unless required by applicable law or agreed to in writing, software
10b1b8bc3fSopenharmony_ci * distributed under the License is distributed on an "AS IS" BASIS,
11b1b8bc3fSopenharmony_ci * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12b1b8bc3fSopenharmony_ci * See the License for the specific language governing permissions and
13b1b8bc3fSopenharmony_ci * limitations under the License.
14b1b8bc3fSopenharmony_ci */
15b1b8bc3fSopenharmony_ci#ifndef CLAT_MANAGER_H
16b1b8bc3fSopenharmony_ci#define CLAT_MANAGER_H
17b1b8bc3fSopenharmony_ci
18b1b8bc3fSopenharmony_ci#include <linux/bpf.h>
19b1b8bc3fSopenharmony_ci#include <map>
20b1b8bc3fSopenharmony_ci#include <string>
21b1b8bc3fSopenharmony_ci
22b1b8bc3fSopenharmony_ci#include "clat_constants.h"
23b1b8bc3fSopenharmony_ci#include "clat_utils.h"
24b1b8bc3fSopenharmony_ci#include "clatd.h"
25b1b8bc3fSopenharmony_ci#include "inet_addr.h"
26b1b8bc3fSopenharmony_ci#include "net_manager_native.h"
27b1b8bc3fSopenharmony_ci
28b1b8bc3fSopenharmony_cinamespace OHOS {
29b1b8bc3fSopenharmony_cinamespace nmd {
30b1b8bc3fSopenharmony_ciusing namespace OHOS::NetManagerStandard;
31b1b8bc3fSopenharmony_ciclass ClatManager {
32b1b8bc3fSopenharmony_cipublic:
33b1b8bc3fSopenharmony_ci    ClatManager();
34b1b8bc3fSopenharmony_ci
35b1b8bc3fSopenharmony_ci    int32_t ClatStart(const std::string &v6Iface, int32_t netId, const std::string &nat64PrefixStr,
36b1b8bc3fSopenharmony_ci                      NetManagerNative *netsysService);
37b1b8bc3fSopenharmony_ci
38b1b8bc3fSopenharmony_ci    int32_t ClatStop(const std::string &v6Iface);
39b1b8bc3fSopenharmony_ci
40b1b8bc3fSopenharmony_ciprivate:
41b1b8bc3fSopenharmony_ci    uint32_t GetFwmark(int32_t netId);
42b1b8bc3fSopenharmony_ci
43b1b8bc3fSopenharmony_ci    int32_t GenerateClatSrcAddr(const std::string &v6Iface, uint32_t fwmark, const std::string &nat64PrefixStr,
44b1b8bc3fSopenharmony_ci                                INetAddr &v4Addr, INetAddr &v6Addr);
45b1b8bc3fSopenharmony_ci
46b1b8bc3fSopenharmony_ci    int32_t CreateAndConfigureTunIface(const std::string &v6Iface, const std::string &tunIface, const INetAddr &v4Addr,
47b1b8bc3fSopenharmony_ci                                       NetManagerNative *netsysService, int &tunFd);
48b1b8bc3fSopenharmony_ci
49b1b8bc3fSopenharmony_ci    int32_t CreateAndConfigureClatSocket(const std::string &v6Iface, const INetAddr &v6Addr, uint32_t fwmark,
50b1b8bc3fSopenharmony_ci                                         int &readSock6, int &writeSock6);
51b1b8bc3fSopenharmony_ci
52b1b8bc3fSopenharmony_ci    std::map<std::string, Clatd> clatds_;
53b1b8bc3fSopenharmony_ci
54b1b8bc3fSopenharmony_ci    std::map<std::string, ClatdTracker> clatdTrackers_;
55b1b8bc3fSopenharmony_ci};
56b1b8bc3fSopenharmony_ci} // namespace nmd
57b1b8bc3fSopenharmony_ci} // namespace OHOS
58b1b8bc3fSopenharmony_ci#endif