bootpclient.java

来自「纯java操作系统jnode,安装简单和操作简单的个人使用的Java操作系统」· Java 代码 · 共 131 行

/*
 * $Id: BOOTPClient.java,v 1.2 2004/02/17 20:49:49 markhale Exp $
 */
package org.jnode.net.ipv4.bootp;

import java.io.IOException;
import java.net.DatagramPacket;
import java.net.MulticastSocket;
import java.net.NetworkInterface;

import org.apache.log4j.Logger;
import org.jnode.driver.ApiNotFoundException;
import org.jnode.driver.Device;
import org.jnode.driver.net.NetDeviceAPI;
import org.jnode.driver.net.NetworkException;
import org.jnode.net.ipv4.IPv4Address;
import org.jnode.net.ipv4.util.Ifconfig;
import org.jnode.net.ipv4.util.Route;

/**
 * @author epr
 * @author markhale
 */
public class BOOTPClient {
	
	/** My logger */
	private final Logger log = Logger.getLogger(getClass());

	private static final int RECEIVE_TIMEOUT = 10*1000; // 10 seconds
	public static final int SERVER_PORT = 67;
	public static final int CLIENT_PORT = 68;
	
	protected MulticastSocket socket;

	/**
	 * Configure the given device using BOOTP
	 * @param device
	 */
	public final void configureDevice(Device device) 
	throws IOException {
		
		// Get the API.
		final NetDeviceAPI api;
		try {
			api = (NetDeviceAPI)device.getAPI(NetDeviceAPI.class);
		} catch (ApiNotFoundException ex) {
			throw new NetworkException("Device is not a network device", ex);
		}
		
		// Open a socket
		socket = new MulticastSocket(CLIENT_PORT);
		try {
			// Prepare the socket			
			socket.setBroadcast(true);
			socket.setNetworkInterface(NetworkInterface.getByName(device.getId()));
			socket.setSoTimeout(RECEIVE_TIMEOUT);

			// Create the BOOTP header
			final IPv4Address myIp = IPv4Address.ANY;
			final int transactionID = (int)(System.currentTimeMillis() & 0xFFFFFFFF);
			BOOTPHeader hdr = new BOOTPHeader(BOOTPHeader.BOOTREQUEST, transactionID, myIp, api.getAddress());

			// Send the packet
			final DatagramPacket packet = createRequestPacket(hdr);
			packet.setAddress(IPv4Address.BROADCAST.toInetAddress());
			packet.setPort(SERVER_PORT);
			socket.send(packet);

			boolean configured;
			do {
				// Wait for a response
				socket.receive(packet);

				// Process the response
				configured = processResponse(device, api, transactionID, packet);
			} while (!configured);
						
		} finally {
			socket.close();
		}
				
	}

	/**
	 * Create a BOOTP request packet
	 */
	protected DatagramPacket createRequestPacket(BOOTPHeader hdr) throws IOException {
		return new BOOTPMessage(hdr).toDatagramPacket();
	}

	/**
	 * Process a BOOTP response
	 * @param packet
	 * @return true if the device has been configured, false otherwise
	 */
	protected boolean processResponse(Device device, NetDeviceAPI api, int transactionID, DatagramPacket packet) 
	throws IOException {
		
		final BOOTPHeader hdr = new BOOTPHeader(packet);
		if (hdr.getOpcode() != BOOTPHeader.BOOTREPLY) {
			// Not a response
			return false;
		}
		if (hdr.getTransactionID() != transactionID) {
			// Not for me
			return false;
		}
		
		configureNetwork(device, hdr);

		return true;
	}

	/**
	 * Performs the actual configuration of a network device based on the settings in a BOOTP header.
	 */
	protected void configureNetwork(Device device, BOOTPHeader hdr) throws NetworkException {
		log.info("Got Client IP address  : " + hdr.getClientIPAddress());
		log.info("Got Your IP address    : " + hdr.getYourIPAddress());
		log.info("Got Server IP address  : " + hdr.getServerIPAddress());
		log.info("Got Gateway IP address : " + hdr.getGatewayIPAddress());

		Ifconfig.setDefault(device, hdr.getYourIPAddress(), null);
		if (hdr.getGatewayIPAddress().isAny()) {
			Route.addRoute(hdr.getServerIPAddress(), null, device); 
		} else {
			Route.addRoute(hdr.getServerIPAddress(), hdr.getGatewayIPAddress(), device); 
		}
	}
}