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