17da02ad3b7a8a0b13a2ccc84e666baa8e6370482Stan Chesnutt/*
27da02ad3b7a8a0b13a2ccc84e666baa8e6370482Stan Chesnutt * Copyright (C) 2010 The Android Open Source Project
37da02ad3b7a8a0b13a2ccc84e666baa8e6370482Stan Chesnutt *
47da02ad3b7a8a0b13a2ccc84e666baa8e6370482Stan Chesnutt * Licensed under the Apache License, Version 2.0 (the "License");
57da02ad3b7a8a0b13a2ccc84e666baa8e6370482Stan Chesnutt * you may not use this file except in compliance with the License.
67da02ad3b7a8a0b13a2ccc84e666baa8e6370482Stan Chesnutt * You may obtain a copy of the License at
77da02ad3b7a8a0b13a2ccc84e666baa8e6370482Stan Chesnutt *
87da02ad3b7a8a0b13a2ccc84e666baa8e6370482Stan Chesnutt *      http://www.apache.org/licenses/LICENSE-2.0
97da02ad3b7a8a0b13a2ccc84e666baa8e6370482Stan Chesnutt *
107da02ad3b7a8a0b13a2ccc84e666baa8e6370482Stan Chesnutt * Unless required by applicable law or agreed to in writing, software
117da02ad3b7a8a0b13a2ccc84e666baa8e6370482Stan Chesnutt * distributed under the License is distributed on an "AS IS" BASIS,
127da02ad3b7a8a0b13a2ccc84e666baa8e6370482Stan Chesnutt * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
137da02ad3b7a8a0b13a2ccc84e666baa8e6370482Stan Chesnutt * See the License for the specific language governing permissions and
147da02ad3b7a8a0b13a2ccc84e666baa8e6370482Stan Chesnutt * limitations under the License.
157da02ad3b7a8a0b13a2ccc84e666baa8e6370482Stan Chesnutt */
167da02ad3b7a8a0b13a2ccc84e666baa8e6370482Stan Chesnutt
177da02ad3b7a8a0b13a2ccc84e666baa8e6370482Stan Chesnuttpackage android.net.dhcp;
187da02ad3b7a8a0b13a2ccc84e666baa8e6370482Stan Chesnutt
197da02ad3b7a8a0b13a2ccc84e666baa8e6370482Stan Chesnuttimport java.net.InetAddress;
207da02ad3b7a8a0b13a2ccc84e666baa8e6370482Stan Chesnuttimport java.net.Inet4Address;
217da02ad3b7a8a0b13a2ccc84e666baa8e6370482Stan Chesnuttimport java.nio.ByteBuffer;
227da02ad3b7a8a0b13a2ccc84e666baa8e6370482Stan Chesnutt
237da02ad3b7a8a0b13a2ccc84e666baa8e6370482Stan Chesnutt/**
247da02ad3b7a8a0b13a2ccc84e666baa8e6370482Stan Chesnutt * This class implements the DHCP-ACK packet.
257da02ad3b7a8a0b13a2ccc84e666baa8e6370482Stan Chesnutt */
267da02ad3b7a8a0b13a2ccc84e666baa8e6370482Stan Chesnuttclass DhcpAckPacket extends DhcpPacket {
277da02ad3b7a8a0b13a2ccc84e666baa8e6370482Stan Chesnutt
287da02ad3b7a8a0b13a2ccc84e666baa8e6370482Stan Chesnutt    /**
297da02ad3b7a8a0b13a2ccc84e666baa8e6370482Stan Chesnutt     * The address of the server which sent this packet.
307da02ad3b7a8a0b13a2ccc84e666baa8e6370482Stan Chesnutt     */
317da02ad3b7a8a0b13a2ccc84e666baa8e6370482Stan Chesnutt    private final InetAddress mSrcIp;
327da02ad3b7a8a0b13a2ccc84e666baa8e6370482Stan Chesnutt
337da02ad3b7a8a0b13a2ccc84e666baa8e6370482Stan Chesnutt    DhcpAckPacket(int transId, boolean broadcast, InetAddress serverAddress,
347da02ad3b7a8a0b13a2ccc84e666baa8e6370482Stan Chesnutt                  InetAddress clientIp, byte[] clientMac) {
35eafccc0859c58ad853fbe26bfa928cc9a91653f3Stan Chesnutt        super(transId, Inet4Address.ANY, clientIp, serverAddress,
367da02ad3b7a8a0b13a2ccc84e666baa8e6370482Stan Chesnutt            Inet4Address.ANY, clientMac, broadcast);
377da02ad3b7a8a0b13a2ccc84e666baa8e6370482Stan Chesnutt        mBroadcast = broadcast;
387da02ad3b7a8a0b13a2ccc84e666baa8e6370482Stan Chesnutt        mSrcIp = serverAddress;
397da02ad3b7a8a0b13a2ccc84e666baa8e6370482Stan Chesnutt    }
407da02ad3b7a8a0b13a2ccc84e666baa8e6370482Stan Chesnutt
417da02ad3b7a8a0b13a2ccc84e666baa8e6370482Stan Chesnutt    public String toString() {
427da02ad3b7a8a0b13a2ccc84e666baa8e6370482Stan Chesnutt        String s = super.toString();
437da02ad3b7a8a0b13a2ccc84e666baa8e6370482Stan Chesnutt        String dnsServers = " DNS servers: ";
447da02ad3b7a8a0b13a2ccc84e666baa8e6370482Stan Chesnutt
457da02ad3b7a8a0b13a2ccc84e666baa8e6370482Stan Chesnutt        for (InetAddress dnsServer: mDnsServers) {
467da02ad3b7a8a0b13a2ccc84e666baa8e6370482Stan Chesnutt            dnsServers += dnsServer.toString() + " ";
477da02ad3b7a8a0b13a2ccc84e666baa8e6370482Stan Chesnutt        }
487da02ad3b7a8a0b13a2ccc84e666baa8e6370482Stan Chesnutt
497da02ad3b7a8a0b13a2ccc84e666baa8e6370482Stan Chesnutt        return s + " ACK: your new IP " + mYourIp +
507da02ad3b7a8a0b13a2ccc84e666baa8e6370482Stan Chesnutt                ", netmask " + mSubnetMask +
517da02ad3b7a8a0b13a2ccc84e666baa8e6370482Stan Chesnutt                ", gateway " + mGateway + dnsServers +
527da02ad3b7a8a0b13a2ccc84e666baa8e6370482Stan Chesnutt                ", lease time " + mLeaseTime;
537da02ad3b7a8a0b13a2ccc84e666baa8e6370482Stan Chesnutt    }
547da02ad3b7a8a0b13a2ccc84e666baa8e6370482Stan Chesnutt
557da02ad3b7a8a0b13a2ccc84e666baa8e6370482Stan Chesnutt    /**
567da02ad3b7a8a0b13a2ccc84e666baa8e6370482Stan Chesnutt     * Fills in a packet with the requested ACK parameters.
577da02ad3b7a8a0b13a2ccc84e666baa8e6370482Stan Chesnutt     */
587da02ad3b7a8a0b13a2ccc84e666baa8e6370482Stan Chesnutt    public ByteBuffer buildPacket(int encap, short destUdp, short srcUdp) {
597da02ad3b7a8a0b13a2ccc84e666baa8e6370482Stan Chesnutt        ByteBuffer result = ByteBuffer.allocate(MAX_LENGTH);
607da02ad3b7a8a0b13a2ccc84e666baa8e6370482Stan Chesnutt        InetAddress destIp = mBroadcast ? Inet4Address.ALL : mYourIp;
617da02ad3b7a8a0b13a2ccc84e666baa8e6370482Stan Chesnutt        InetAddress srcIp = mBroadcast ? Inet4Address.ANY : mSrcIp;
627da02ad3b7a8a0b13a2ccc84e666baa8e6370482Stan Chesnutt
637da02ad3b7a8a0b13a2ccc84e666baa8e6370482Stan Chesnutt        fillInPacket(encap, destIp, srcIp, destUdp, srcUdp, result,
647da02ad3b7a8a0b13a2ccc84e666baa8e6370482Stan Chesnutt            DHCP_BOOTREPLY, mBroadcast);
657da02ad3b7a8a0b13a2ccc84e666baa8e6370482Stan Chesnutt        result.flip();
667da02ad3b7a8a0b13a2ccc84e666baa8e6370482Stan Chesnutt        return result;
677da02ad3b7a8a0b13a2ccc84e666baa8e6370482Stan Chesnutt    }
687da02ad3b7a8a0b13a2ccc84e666baa8e6370482Stan Chesnutt
697da02ad3b7a8a0b13a2ccc84e666baa8e6370482Stan Chesnutt    /**
707da02ad3b7a8a0b13a2ccc84e666baa8e6370482Stan Chesnutt     * Adds the optional parameters to the client-generated ACK packet.
717da02ad3b7a8a0b13a2ccc84e666baa8e6370482Stan Chesnutt     */
727da02ad3b7a8a0b13a2ccc84e666baa8e6370482Stan Chesnutt    void finishPacket(ByteBuffer buffer) {
737da02ad3b7a8a0b13a2ccc84e666baa8e6370482Stan Chesnutt        addTlv(buffer, DHCP_MESSAGE_TYPE, DHCP_MESSAGE_TYPE_ACK);
747da02ad3b7a8a0b13a2ccc84e666baa8e6370482Stan Chesnutt        addTlv(buffer, DHCP_SERVER_IDENTIFIER, mServerIdentifier);
757da02ad3b7a8a0b13a2ccc84e666baa8e6370482Stan Chesnutt        addTlv(buffer, DHCP_LEASE_TIME, mLeaseTime);
767da02ad3b7a8a0b13a2ccc84e666baa8e6370482Stan Chesnutt
777da02ad3b7a8a0b13a2ccc84e666baa8e6370482Stan Chesnutt        // the client should renew at 1/2 the lease-expiry interval
787da02ad3b7a8a0b13a2ccc84e666baa8e6370482Stan Chesnutt        if (mLeaseTime != null) {
797da02ad3b7a8a0b13a2ccc84e666baa8e6370482Stan Chesnutt            addTlv(buffer, DHCP_RENEWAL_TIME,
807da02ad3b7a8a0b13a2ccc84e666baa8e6370482Stan Chesnutt                Integer.valueOf(mLeaseTime.intValue() / 2));
817da02ad3b7a8a0b13a2ccc84e666baa8e6370482Stan Chesnutt        }
827da02ad3b7a8a0b13a2ccc84e666baa8e6370482Stan Chesnutt
837da02ad3b7a8a0b13a2ccc84e666baa8e6370482Stan Chesnutt        addTlv(buffer, DHCP_SUBNET_MASK, mSubnetMask);
847da02ad3b7a8a0b13a2ccc84e666baa8e6370482Stan Chesnutt        addTlv(buffer, DHCP_ROUTER, mGateway);
857da02ad3b7a8a0b13a2ccc84e666baa8e6370482Stan Chesnutt        addTlv(buffer, DHCP_DOMAIN_NAME, mDomainName);
867da02ad3b7a8a0b13a2ccc84e666baa8e6370482Stan Chesnutt        addTlv(buffer, DHCP_BROADCAST_ADDRESS, mBroadcastAddress);
877da02ad3b7a8a0b13a2ccc84e666baa8e6370482Stan Chesnutt        addTlv(buffer, DHCP_DNS_SERVER, mDnsServers);
887da02ad3b7a8a0b13a2ccc84e666baa8e6370482Stan Chesnutt        addTlvEnd(buffer);
897da02ad3b7a8a0b13a2ccc84e666baa8e6370482Stan Chesnutt    }
907da02ad3b7a8a0b13a2ccc84e666baa8e6370482Stan Chesnutt
917da02ad3b7a8a0b13a2ccc84e666baa8e6370482Stan Chesnutt    /**
927da02ad3b7a8a0b13a2ccc84e666baa8e6370482Stan Chesnutt     * Un-boxes an Integer, returning 0 if a null reference is supplied.
937da02ad3b7a8a0b13a2ccc84e666baa8e6370482Stan Chesnutt     */
947da02ad3b7a8a0b13a2ccc84e666baa8e6370482Stan Chesnutt    private static final int getInt(Integer v) {
957da02ad3b7a8a0b13a2ccc84e666baa8e6370482Stan Chesnutt        if (v == null) {
967da02ad3b7a8a0b13a2ccc84e666baa8e6370482Stan Chesnutt            return 0;
977da02ad3b7a8a0b13a2ccc84e666baa8e6370482Stan Chesnutt        } else {
987da02ad3b7a8a0b13a2ccc84e666baa8e6370482Stan Chesnutt            return v.intValue();
997da02ad3b7a8a0b13a2ccc84e666baa8e6370482Stan Chesnutt        }
1007da02ad3b7a8a0b13a2ccc84e666baa8e6370482Stan Chesnutt    }
1017da02ad3b7a8a0b13a2ccc84e666baa8e6370482Stan Chesnutt
1027da02ad3b7a8a0b13a2ccc84e666baa8e6370482Stan Chesnutt    /**
1037da02ad3b7a8a0b13a2ccc84e666baa8e6370482Stan Chesnutt     * Notifies the specified state machine of the ACK packet parameters.
1047da02ad3b7a8a0b13a2ccc84e666baa8e6370482Stan Chesnutt     */
1057da02ad3b7a8a0b13a2ccc84e666baa8e6370482Stan Chesnutt    public void doNextOp(DhcpStateMachine machine) {
1067da02ad3b7a8a0b13a2ccc84e666baa8e6370482Stan Chesnutt        machine.onAckReceived(mYourIp, mSubnetMask, mGateway, mDnsServers,
1077da02ad3b7a8a0b13a2ccc84e666baa8e6370482Stan Chesnutt            mServerIdentifier, getInt(mLeaseTime));
1087da02ad3b7a8a0b13a2ccc84e666baa8e6370482Stan Chesnutt    }
1097da02ad3b7a8a0b13a2ccc84e666baa8e6370482Stan Chesnutt}
110