1 /*
2  * Copyright (C) 2021-2022 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 OHOS_FUNCTION_H
17 #define OHOS_FUNCTION_H
18 
19 #include <iostream>
20 #include <fstream>
21 #include <cstring>
22 #include <iomanip>
23 #include <arpa/inet.h>
24 #include <string>
25 #include <cstdio>
26 
27 #include "dhcp_define.h"
28 
29 namespace OHOS {
30 namespace DHCP {
31 class DhcpFunction {
32 public:
DhcpFunction()33     DhcpFunction() {}
~DhcpFunction()34     ~DhcpFunction() {}
35 
36     static bool Ip4StrConToInt(const std::string& strIp, uint32_t& uIp, bool bHost = true);
37     static std::string Ip4IntConvertToStr(uint32_t uIp, bool bHost = true);
38     static bool Ip6StrConToChar(const std::string& strIp, uint8_t chIp[], size_t uSize);
39     static std::string Ip6CharConToStr(uint8_t chIp[], int size);
40     static bool CheckIpStr(const std::string& strIp);
41     static int GetLocalIp(const std::string strInf, std::string& strIp, std::string& strMask);
42     static int GetLocalMac(const std::string ethInf, std::string& ethMac);
43     static int CheckRangeNetwork(const std::string strInf, const std::string strBegin, const std::string strEnd);
44     static bool CheckSameNetwork(const uint32_t srcIp, const uint32_t dstIp, const uint32_t maskIp);
45 
46     static bool IsExistFile(const std::string& filename);
47     static bool CreateFile(const std::string& filename, const std::string& filedata);
48     static bool RemoveFile(const std::string& filename);
49     static bool AddFileLineData(const std::string& filename, const std::string& prevdata, const std::string& linedata);
50     static bool DelFileLineData(const std::string& filename, const std::string& linedata);
51     static bool ModifyFileLineData(const std::string& filename, const std::string& srcdata, const std::string& dstdata);
52     static int FormatString(struct DhcpPacketResult &result);
53     static int InitPidfile(const std::string& piddir, const std::string& pidfile);
54     static pid_t GetPID(const std::string& pidfile);
55     static int CheckProRunning(const pid_t proPid, const std::string& proName);
56     static int CreateDirs(const std::string dirs, int mode = DIR_DEFAULT_MODE);
57     static bool SplitString(
58         const std::string src, const std::string delim, const int count, std::vector<std::string> &splits);
59     static int WaitProcessExit(const pid_t& serverPid);
60 #ifdef OHOS_ARCH_LITE
61     static int GetDhcpPacketResult(const std::string& filename, struct DhcpPacketResult &result);
62 #else
63 #endif
64 };
65 }  // namespace DHCP
66 }  // namespace OHOS
67 #endif