| /* |
| * Copyright (C) 2015 The Android Open Source Project |
| * |
| * Licensed under the Apache License, Version 2.0 (the "License"); |
| * you may not use this file except in compliance with the License. |
| * You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| |
| package android.net.dhcp; |
| |
| import android.net.NetworkUtils; |
| import android.system.OsConstants; |
| import android.test.suitebuilder.annotation.SmallTest; |
| import junit.framework.TestCase; |
| |
| import java.net.Inet4Address; |
| import java.nio.ByteBuffer; |
| |
| import static android.net.dhcp.DhcpPacket.*; |
| |
| |
| public class DhcpPacketTest extends TestCase { |
| |
| private static Inet4Address SERVER_ADDR = |
| (Inet4Address) NetworkUtils.numericToInetAddress("192.0.2.1"); |
| private static Inet4Address CLIENT_ADDR = |
| (Inet4Address) NetworkUtils.numericToInetAddress("192.0.2.234"); |
| private static byte[] CLIENT_MAC = new byte[] { 0x00, 0x01, 0x02, 0x03, 0x04, 0x05 }; |
| |
| class TestDhcpPacket extends DhcpPacket { |
| private byte mType; |
| // TODO: Make this a map of option numbers to bytes instead. |
| private byte[] mDomainBytes, mVendorInfoBytes; |
| |
| public TestDhcpPacket(byte type, byte[] domainBytes, byte[] vendorInfoBytes) { |
| super(0xdeadbeef, (short) 0, INADDR_ANY, CLIENT_ADDR, INADDR_ANY, INADDR_ANY, |
| CLIENT_MAC, true); |
| mType = type; |
| mDomainBytes = domainBytes; |
| mVendorInfoBytes = vendorInfoBytes; |
| } |
| |
| public ByteBuffer buildPacket(int encap, short unusedDestUdp, short unusedSrcUdp) { |
| ByteBuffer result = ByteBuffer.allocate(MAX_LENGTH); |
| fillInPacket(encap, CLIENT_ADDR, SERVER_ADDR, |
| DHCP_CLIENT, DHCP_SERVER, result, DHCP_BOOTREPLY, false); |
| return result; |
| } |
| |
| public void finishPacket(ByteBuffer buffer) { |
| addTlv(buffer, DHCP_MESSAGE_TYPE, mType); |
| if (mDomainBytes != null) { |
| addTlv(buffer, DHCP_DOMAIN_NAME, mDomainBytes); |
| } |
| if (mVendorInfoBytes != null) { |
| addTlv(buffer, DHCP_VENDOR_CLASS_ID, mVendorInfoBytes); |
| } |
| addTlvEnd(buffer); |
| } |
| |
| // Convenience method. |
| public ByteBuffer build() { |
| // ENCAP_BOOTP packets don't contain ports, so just pass in 0. |
| ByteBuffer pkt = buildPacket(ENCAP_BOOTP, (short) 0, (short) 0); |
| pkt.flip(); |
| return pkt; |
| } |
| } |
| |
| private void assertDomainAndVendorInfoParses( |
| String expectedDomain, byte[] domainBytes, |
| String expectedVendorInfo, byte[] vendorInfoBytes) { |
| ByteBuffer packet = new TestDhcpPacket(DHCP_MESSAGE_TYPE_OFFER, |
| domainBytes, vendorInfoBytes).build(); |
| DhcpPacket offerPacket = DhcpPacket.decodeFullPacket(packet, ENCAP_BOOTP); |
| assertEquals(expectedDomain, offerPacket.mDomainName); |
| assertEquals(expectedVendorInfo, offerPacket.mVendorId); |
| } |
| |
| @SmallTest |
| public void testDomainName() throws Exception { |
| byte[] nullByte = new byte[] { 0x00 }; |
| byte[] twoNullBytes = new byte[] { 0x00, 0x00 }; |
| byte[] nonNullDomain = new byte[] { |
| (byte) 'g', (byte) 'o', (byte) 'o', (byte) '.', (byte) 'g', (byte) 'l' |
| }; |
| byte[] trailingNullDomain = new byte[] { |
| (byte) 'g', (byte) 'o', (byte) 'o', (byte) '.', (byte) 'g', (byte) 'l', 0x00 |
| }; |
| byte[] embeddedNullsDomain = new byte[] { |
| (byte) 'g', (byte) 'o', (byte) 'o', 0x00, 0x00, (byte) 'g', (byte) 'l' |
| }; |
| byte[] metered = "ANDROID_METERED".getBytes("US-ASCII"); |
| |
| byte[] meteredEmbeddedNull = metered.clone(); |
| meteredEmbeddedNull[7] = (char) 0; |
| |
| byte[] meteredTrailingNull = metered.clone(); |
| meteredTrailingNull[meteredTrailingNull.length - 1] = (char) 0; |
| |
| assertDomainAndVendorInfoParses("", nullByte, "\u0000", nullByte); |
| assertDomainAndVendorInfoParses("", twoNullBytes, "\u0000\u0000", twoNullBytes); |
| assertDomainAndVendorInfoParses("goo.gl", nonNullDomain, "ANDROID_METERED", metered); |
| assertDomainAndVendorInfoParses("goo", embeddedNullsDomain, |
| "ANDROID\u0000METERED", meteredEmbeddedNull); |
| assertDomainAndVendorInfoParses("goo.gl", trailingNullDomain, |
| "ANDROID_METERE\u0000", meteredTrailingNull); |
| } |
| } |