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 #include "mock_custom_func.h"
16 #include "dhcp_client_state_machine.h"
17 #include "dhcp_options.h"
18 #include "dhcp_function.h"
19 #include "securec.h"
20 
21 using namespace OHOS;
22 
23 static bool g_mockTag = false;
24 static int NUM_TWO = 2;
25 static int NUM_THREE = 3;
26 static int NUM_FOUR = 4;
27 static int NUM_TEN = 10;
GetInstance()28 MockCustomFunc &MockCustomFunc::GetInstance()
29 {
30     static MockCustomFunc gMockCustomFunc;
31     return gMockCustomFunc;
32 };
33 
MockCustomFunc()34 MockCustomFunc::MockCustomFunc()
35 {}
36 
SetMockFlag(bool flag)37 void MockCustomFunc::SetMockFlag(bool flag)
38 {
39     g_mockTag = flag;
40 }
41 
GetMockFlag(void)42 bool MockCustomFunc::GetMockFlag(void)
43 {
44     return g_mockTag;
45 }
46 
47 #ifdef __cplusplus
48 extern "C" {
49 #endif
AddOptDoubleValueToOpts(uint8_t * pOpts,uint8_t code,uint32_t value1,uint32_t value2)50 static int AddOptDoubleValueToOpts(uint8_t *pOpts, uint8_t code, uint32_t value1, uint32_t value2)
51 {
52     uint8_t uOption[DHCP_OPT_CODE_BYTES + DHCP_OPT_LEN_BYTES + DHCP_UINT32_DOUBLE_BYTES] = {0};
53     uOption[DHCP_OPT_CODE_INDEX] = code;
54     uOption[DHCP_OPT_LEN_INDEX] = DHCP_UINT32_BYTES * NUM_TWO;
55     if ((memcpy_s(uOption + DHCP_OPT_DATA_INDEX, DHCP_UINT32_BYTES, &value1, DHCP_UINT32_BYTES) != EOK) ||
56         (memcpy_s(uOption + DHCP_OPT_DATA_INDEX + DHCP_UINT32_BYTES,
57             DHCP_UINT32_BYTES, &value2, DHCP_UINT32_BYTES) != EOK)) {
58         return 0;
59     }
60     int nLen = DHCP_OPT_CODE_BYTES + DHCP_OPT_LEN_BYTES + uOption[DHCP_OPT_LEN_INDEX];
61     return AddOptStrToOpts(pOpts, uOption, nLen);
62 }
63 
64 int __real_GetDhcpRawPacket(struct DhcpPacket *getPacket, int rawFd);
__wrap_GetDhcpRawPacket(struct DhcpPacket * getPacket,int rawFd)65 int __wrap_GetDhcpRawPacket(struct DhcpPacket *getPacket, int rawFd)
66 {
67     std::unique_ptr<OHOS::DHCP::DhcpClientStateMachine> testMachine;
68     if (g_mockTag) {
69         int nLen = MockCustomFunc::GetInstance().GetDhcpRawPacket(getPacket, rawFd);
70         char cliIp[INET_ADDRSTRLEN] = "192.77.2.231";
71         char serIp[INET_ADDRSTRLEN] = "192.77.2.1";
72         uint32_t uCliIp = 0;
73         uint32_t uSerIp = 0;
74         Ip4StrConToInt(cliIp, &uCliIp, false);
75         Ip4StrConToInt(serIp, &uSerIp, false);
76         getPacket->xid = testMachine->GetDhcpTransID();
77         getPacket->yiaddr = uCliIp;
78         if (nLen == 1) {
79             testMachine->GetPacketHeaderInfo(getPacket, DHCP_OFFER);
80             AddOptValueToOpts(getPacket->options, SERVER_IDENTIFIER_OPTION, uSerIp);
81         } else if (nLen == NUM_TWO) {
82             testMachine->GetPacketHeaderInfo(getPacket, DHCP_ACK);
83         } else if (nLen == NUM_THREE) {
84             testMachine->GetPacketHeaderInfo(getPacket, DHCP_NAK);
85         } else if (nLen == NUM_FOUR) {
86             testMachine->GetPacketHeaderInfo(getPacket, DHCP_ACK);
87             AddOptValueToOpts(getPacket->options, SERVER_IDENTIFIER_OPTION, uSerIp);
88             AddOptValueToOpts(getPacket->options, SUBNET_MASK_OPTION, 0);
89             AddOptValueToOpts(getPacket->options, IP_ADDRESS_LEASE_TIME_OPTION, htonl(NUM_TEN));
90             AddOptDoubleValueToOpts(getPacket->options, ROUTER_OPTION, uSerIp, uCliIp);
91             AddOptDoubleValueToOpts(getPacket->options, DOMAIN_NAME_SERVER_OPTION, uSerIp, uCliIp);
92         }
93         return nLen;
94     } else {
95         return __real_GetDhcpRawPacket(getPacket, rawFd);
96     }
97 }
98 
99 int __real_GetDhcpKernelPacket(struct DhcpPacket *getPacket, int sockFd);
__wrap_GetDhcpKernelPacket(struct DhcpPacket * getPacket,int sockFd)100 int __wrap_GetDhcpKernelPacket(struct DhcpPacket *getPacket, int sockFd)
101 {
102     std::unique_ptr<OHOS::DHCP::DhcpClientStateMachine> testMachine;
103     if (g_mockTag) {
104         int nLen = MockCustomFunc::GetInstance().GetDhcpKernelPacket(getPacket, sockFd);
105         char cliIp[INET_ADDRSTRLEN] = "192.77.3.231";
106         char serIp[INET_ADDRSTRLEN] = "192.77.3.1";
107         uint32_t uCliIp = 0;
108         uint32_t uSerIp = 0;
109         Ip4StrConToInt(cliIp, &uCliIp, false);
110         Ip4StrConToInt(serIp, &uSerIp, false);
111         getPacket->xid = 1;
112         getPacket->yiaddr = uCliIp;
113         if (nLen == 1) {
114             testMachine->GetPacketHeaderInfo(getPacket, DHCP_OFFER);
115             AddOptValueToOpts(getPacket->options, SERVER_IDENTIFIER_OPTION, uSerIp);
116         } else if (nLen == NUM_TWO) {
117             testMachine->GetPacketHeaderInfo(getPacket, DHCP_ACK);
118             AddOptValueToOpts(getPacket->options, SERVER_IDENTIFIER_OPTION, uSerIp);
119             AddOptValueToOpts(getPacket->options, SUBNET_MASK_OPTION, 0);
120             AddOptValueToOpts(getPacket->options, IP_ADDRESS_LEASE_TIME_OPTION, htonl(NUM_FOUR));
121             AddOptDoubleValueToOpts(getPacket->options, ROUTER_OPTION, uSerIp, uCliIp);
122             AddOptDoubleValueToOpts(getPacket->options, DOMAIN_NAME_SERVER_OPTION, uSerIp, uCliIp);
123         } else if (nLen == NUM_THREE) {
124             testMachine->GetPacketHeaderInfo(getPacket, DHCP_NAK);
125         }
126         return nLen;
127     } else {
128         return __real_GetDhcpKernelPacket(getPacket, sockFd);
129     }
130 }
131 #ifdef __cplusplus
132 }
133 #endif
134