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 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 if (g_mockTag) {
68 int nLen = MockCustomFunc::GetInstance().GetDhcpRawPacket(getPacket, rawFd);
69 char cliIp[INET_ADDRSTRLEN] = "192.77.2.231";
70 char serIp[INET_ADDRSTRLEN] = "192.77.2.1";
71 uint32_t uCliIp = 0;
72 uint32_t uSerIp = 0;
73 Ip4StrConToInt(cliIp, &uCliIp, false);
74 Ip4StrConToInt(serIp, &uSerIp, false);
75 getPacket->xid = GetDhcpTransID();
76 getPacket->yiaddr = uCliIp;
77 if (nLen == 1) {
78 GetPacketHeaderInfo(getPacket, DHCP_OFFER);
79 AddOptValueToOpts(getPacket->options, SERVER_IDENTIFIER_OPTION, uSerIp);
80 } else if (nLen == NUM_TWO) {
81 GetPacketHeaderInfo(getPacket, DHCP_ACK);
82 } else if (nLen == NUM_THREE) {
83 GetPacketHeaderInfo(getPacket, DHCP_NAK);
84 } else if (nLen == NUM_FOUR) {
85 GetPacketHeaderInfo(getPacket, DHCP_ACK);
86 AddOptValueToOpts(getPacket->options, SERVER_IDENTIFIER_OPTION, uSerIp);
87 AddOptValueToOpts(getPacket->options, SUBNET_MASK_OPTION, 0);
88 AddOptValueToOpts(getPacket->options, IP_ADDRESS_LEASE_TIME_OPTION, htonl(NUM_TEN));
89 AddOptDoubleValueToOpts(getPacket->options, ROUTER_OPTION, uSerIp, uCliIp);
90 AddOptDoubleValueToOpts(getPacket->options, DOMAIN_NAME_SERVER_OPTION, uSerIp, uCliIp);
91 }
92 return nLen;
93 } else {
94 return __real_GetDhcpRawPacket(getPacket, rawFd);
95 }
96 }
97
98 int __real_GetDhcpKernelPacket(struct DhcpPacket *getPacket, int sockFd);
__wrap_GetDhcpKernelPacket(struct DhcpPacket * getPacket,int sockFd)99 int __wrap_GetDhcpKernelPacket(struct DhcpPacket *getPacket, int sockFd)
100 {
101 if (g_mockTag) {
102 int nLen = MockCustomFunc::GetInstance().GetDhcpKernelPacket(getPacket, sockFd);
103 char cliIp[INET_ADDRSTRLEN] = "192.77.3.231";
104 char serIp[INET_ADDRSTRLEN] = "192.77.3.1";
105 uint32_t uCliIp = 0;
106 uint32_t uSerIp = 0;
107 Ip4StrConToInt(cliIp, &uCliIp, false);
108 Ip4StrConToInt(serIp, &uSerIp, false);
109 getPacket->xid = GetDhcpTransID();
110 getPacket->yiaddr = uCliIp;
111 if (nLen == 1) {
112 GetPacketHeaderInfo(getPacket, DHCP_OFFER);
113 AddOptValueToOpts(getPacket->options, SERVER_IDENTIFIER_OPTION, uSerIp);
114 } else if (nLen == NUM_TWO) {
115 GetPacketHeaderInfo(getPacket, DHCP_ACK);
116 AddOptValueToOpts(getPacket->options, SERVER_IDENTIFIER_OPTION, uSerIp);
117 AddOptValueToOpts(getPacket->options, SUBNET_MASK_OPTION, 0);
118 AddOptValueToOpts(getPacket->options, IP_ADDRESS_LEASE_TIME_OPTION, htonl(NUM_FOUR));
119 AddOptDoubleValueToOpts(getPacket->options, ROUTER_OPTION, uSerIp, uCliIp);
120 AddOptDoubleValueToOpts(getPacket->options, DOMAIN_NAME_SERVER_OPTION, uSerIp, uCliIp);
121 } else if (nLen == NUM_THREE) {
122 GetPacketHeaderInfo(getPacket, DHCP_NAK);
123 }
124 return nLen;
125 } else {
126 return __real_GetDhcpKernelPacket(getPacket, sockFd);
127 }
128 }
129 #ifdef __cplusplus
130 }
131 #endif
132