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