WakeOnLan.java

/**
 * CentralIT - CITSmart
 */
package br.com.centralit.citcorpore.componenteMaquina;

import java.io.IOException;
import java.net.DatagramPacket;
import java.net.DatagramSocket;
import java.net.InetAddress;
import java.util.TimerTask;

import br.com.centralit.citcorpore.negocio.UsuarioService;
import br.com.citframework.excecao.ServiceException;
import br.com.citframework.service.ServiceLocator;

/**
 * Implementa a verifica��o de Status e Liga Computador.
 * 
 * @author valdoilo.damasceno
 * 
 */
public class WakeOnLan extends Thread {	
    public static final int PORT = 8900;
    public static final String NONE = "";
    public static final String OK = "OK";
    public static final String FAIL = "FAIL";
    private String ip;
    private String mascara;
    private String mac;
    private TimerTask tarefaStatus;
    private Boolean status = false;
    private Integer numeroDeTentativas;
    private Integer tempo;
    private String control = WakeOnLan.NONE;

    /**
     * Construtor.
     * 
     * @param ip
     * @param mascara
     * @param mac
     * @param tempo
     * @param numeroDeTentativas
     *  
     * @author valdoilo.damasceno
     */
    public WakeOnLan(String ip, String mascara, String mac, Integer tempo, Integer numeroDeTentativas) {
    	this.ip = ip;
    	this.mascara = mascara;
    	this.mac = mac;
    	this.tempo = tempo * 1000;
    	this.numeroDeTentativas = numeroDeTentativas;
    	setControl(WakeOnLan.NONE);
    }

    @Override
    public void run() {
    	this.ligarComputador();
    	this.verificarStatusComputador();
    }

    /**
     * Verifica Status do Computador.
     * 
     * @return Boolean
     * 
     * @author valdoilo.damasceno
     */
    public Boolean verificarStatusComputador() {
    	try {
    		while (this.getNumeroDeTentativas() > 0) {
    			this.setStatus(this.pingar(this.getIp() ) );
    			if (this.getStatus() ) {
    				this.setControl(WakeOnLan.OK);
    				this.setNumeroDeTentativas(-999);
    				return this.getStatus();
    			}
    			try {
    				Thread.sleep(this.getTempo() );
    			} catch (InterruptedException e) {
    			}
    			this.setNumeroDeTentativas(this.getNumeroDeTentativas() - 1);
    		}
    		if (this.getControl().equalsIgnoreCase(WakeOnLan.NONE) ) {
    			this.setControl(WakeOnLan.FAIL);
    		}
    	} catch (IOException e) {
    		setControl(WakeOnLan.FAIL);
    	}
    	return this.getStatus();
    }

    /**
     * Realiza Ping.
     * 
     * @param ip
     * @return boolean
     * @throws IOException
     * @author valdoilo.damasceno
     */
    public boolean pingar(String ip) throws IOException {
    	InetAddress inetAddress = InetAddress.getByName(this.getIp() );
    	try {
    		System.out.println("######TESTA SE O COMPUTADOR LIGOU######");
    		// Testa se o outro computador na rede � alcan��vel
    		// Timeout (tempo de espera por resposta) definido em 3 segundos
    		return inetAddress.isReachable(3000);
    	} catch (Exception e) {
    		System.out.println(this.getClass().getName() + "-> pingar  --> IP: [" + ip + "] ---> " + e.getMessage() );
    		return false;
    	}
    }

    /**
     * Liga Computador.
     * 
     * @return Boolean
     * @author valdoilo.damasceno
     */
    public Boolean ligarComputador() {
    	try {
    		String ipBroadcast = this.binaryIpToInteger(this.getIpBroadcastAddressBinary(this.getIp(), this.getMascara() ) );
    		byte[] ethernetAddressBytes = this.getEnderecoEthernet(this.getMac() );
    		byte[] wakeupFrame = new byte[17 * ethernetAddressBytes.length];
    		int start = 6;
    		for (int i = 0; i < 6; i++) {
    			wakeupFrame[i] = (byte) 0xff;
    		}
    		for (int i = 0; i < 16; i++) {
    			System.arraycopy(ethernetAddressBytes, 0, wakeupFrame, start, ethernetAddressBytes.length);
    			start += ethernetAddressBytes.length;
    		}
    		InetAddress address = InetAddress.getByName(ipBroadcast);
    		DatagramSocket socket = new DatagramSocket(PORT);
    		DatagramPacket packet = new DatagramPacket(wakeupFrame, wakeupFrame.length, address, PORT);
    		socket.setBroadcast(true);
    		socket.send(packet);
    		socket.close();
    		return this.getStatus();
    	} catch (Exception e) {
    		e.printStackTrace();
    		return false;
    	}
    }

    /**
     * Retorna Endere�o IP Broadcast.
     * 
     * @param enderecoIp
     * @param mascara
     * @return String
     * @author valdoilo.damasceno
     */
    private String[] getIpBroadcastAddressBinary(String enderecoIp, String mascara) {
    	String[] ipBinario = this.ipToBinary(mascara);
    	String[] mascaraDeSubRedeBin = this.getSubnetAddressBinary(enderecoIp, mascara);
    	String[] invertMask = new String[ipBinario.length];
    	String[] broadCastIp = new String[ipBinario.length];
    	for (int i = 0; i < ipBinario.length; i++) {
    		String ip0 = ipBinario[i];
    		StringBuilder resu = new StringBuilder();
    		for (int j = 0; j < ip0.length(); j++) {
    			if (ip0.charAt(j) == '1')
    				resu.append('0');
    			else {
    				resu.append('1');
    			}
    		}
    		invertMask[i] = resu.toString();
    	}
    	for (int lcI = 0; lcI < invertMask.length; lcI++) {
    		String ip0 = mascaraDeSubRedeBin[lcI];
    		String ip1 = invertMask[lcI];
    		StringBuilder resu = new StringBuilder();
    		for (int lcI2 = 0; lcI2 < ip0.length(); lcI2++) {
    			if ( (ip0.charAt(lcI2) == '1') || (ip1.charAt(lcI2) == '1') )
    				resu.append('1');
    			else {
    				resu.append('0');
    			}
			}
    		broadCastIp[lcI] = resu.toString();
    	}
    	return broadCastIp;
    }

    /**
     * Retorna m�scara de sub-rede.
     * 
     * @param anIp
     * @param aMask
     * @return String
     * @author valdoilo.damasceno
     */
    private String[] getSubnetAddressBinary(String anIp, String aMask) {
    	String[] ipBin = ipToBinary(anIp);
    	String[] masqBin = ipToBinary(aMask);
    	String[] subMasq = new String[ipBin.length];
    	for (int lcI = 0; lcI < masqBin.length; lcI++) {
    		String ip0 = ipBin[lcI];
    		String ip1 = masqBin[lcI];
    		StringBuilder resu = new StringBuilder();
    		for (int lcI2 = 0; lcI2 < ip0.length(); lcI2++) {
    			if ( (ip0.charAt(lcI2) == '1') && (ip1.charAt(lcI2) == '1') )
    				resu.append('1');
    			else {
    				resu.append('0');
    			}
    		}
    		subMasq[lcI] = resu.toString();
    	}
    	return subMasq;
    }

    /**
     * Converte IP em formato texto (String) para o fomato bin�rio.
     * 
     * @param anIp
     * @return String[]
     * @author valdoilo.damasceno
     */
    private String[] ipToBinary(String anIp) {
    	// Recebe o endere�o IP em formato textual e cria um array com as partes entes os pontos (.)
    	String[] ipArray = anIp.split("\\.");
    	String[] ipBin = new String[ipArray.length];
    	for (int lcJ = 0; lcJ < ipBin.length; lcJ++) {
    		// Converte um texto para inteiro e logo em seguida converte o
    		// n�mero para um n�mero bin�rio em formato texto
    		ipBin[lcJ] = Integer.toBinaryString(Integer.parseInt(ipArray[lcJ]) );
    		if (ipBin[lcJ].length() < 8) {
    			int nb0 = 8 - ipBin[lcJ].length();
    			StringBuilder buff = new StringBuilder();
    			while (nb0 != 0) {
    				buff.append('0');
    				nb0--;
    			}
    			ipBin[lcJ] = (buff.toString() + ipBin[lcJ]);
    		}
    	}
    	return ipBin;
    }

    /**
     * Retorna o endere�o Ethernet.
     * 
     * @param enderecoMAC
     * @return enderecoEthernet
     * @throws IllegalArgumentException
     * @author valdoilo.damasceno
     */
    private byte[] getEnderecoEthernet(String enderecoMAC) throws IllegalArgumentException {
    	String[] hex = enderecoMAC.split("(\\:|\\-)");
    	if (hex.length != 6) {
    		throw new IllegalArgumentException("Endere�o MAC inv�lido.");
    	}
    	byte[] enderecoEthernet = new byte[6];
    	try {
    		for (int i = 0; i < 6; i++) {
    			enderecoEthernet[i] = (byte) Integer.parseInt(hex[i], 16);
    		}
    	} catch (NumberFormatException e) {
    		throw new IllegalArgumentException("Endere�o MAC inv�lido.");
    	}
    	return enderecoEthernet;
    }

    /**
     * Converte IP de bin�rio para Integer.
     * 
     * @param aBinaryIp
     * @return String
     * @author valdoilo.damasceno
     */
    private String binaryIpToInteger(String[] aBinaryIp) {
    	StringBuilder resu = new StringBuilder();
    	for (int lcI = 0; lcI < aBinaryIp.length; lcI++) {
    		resu.append(Integer.parseInt(aBinaryIp[lcI], 2)).append('.');
    	}
    	resu.delete(resu.length() - 1, resu.length());
    	return resu.toString();
    }

    /**
     * Retorna Service de Usuario.
     * 
     * @return UsuarioService
     * @throws ServiceException
     * @throws Exception
     * @author valdoilo.damasceno
     */
    public UsuarioService getUsuarioService() throws ServiceException, Exception {
    	// Obt�m um servi�o de usu�rio atrav�s do localizador de servi�o
    	return (UsuarioService) ServiceLocator.getInstance().getService(UsuarioService.class, null);
    }

    /**
     * @return valor do atributo ip.
     */
    public String getIp() {
    	return ip;
    }

    /**
     * Define valor do atributo ip.
     * 
     * @param ip
     */
    public void setIp(String ip) {
    	this.ip = ip;
    }

    /**
     * @return valor do atributo mascara.
     */
    public String getMascara() {
    	return mascara;
    }

    /**
     * Define valor do atributo mascara.
     * 
     * @param mascara
     */
    public void setMascara(String mascara) {
    	this.mascara = mascara;
    }

    /**
     * @return valor do atributo mac.
     */
    public String getMac() {
    	return mac;
    }

    /**
     * Define valor do atributo mac.
     * 
     * @param mac
     */
    public void setMac(String mac) {
    	this.mac = mac;
    }

    /**
     * @return valor do atributo tarefaStatus.
     */
    public TimerTask getTarefaStatus() {
    	return tarefaStatus;
    }

    /**
     * Define valor do atributo tarefaStatus.
     * 
     * @param tarefaStatus
     */
    public void setTarefaStatus(TimerTask tarefaStatus) {
    	this.tarefaStatus = tarefaStatus;
    }

    /**
     * @return valor do atributo status.
     */
    public Boolean getStatus() {
    	return status;
    }

    /**
     * Define valor do atributo status.
     * 
     * @param status
     */
    public void setStatus(Boolean status) {
    	this.status = status;
    }

    /**
     * @return valor do atributo numeroDeTentativas.
     */
    public Integer getNumeroDeTentativas() {
    	return numeroDeTentativas;
    }

    /**
     * Define valor do atributo numeroDeTentativas.
     * 
     * @param numeroDeTentativas
     */
    public void setNumeroDeTentativas(Integer numeroDeTentativas) {
    	this.numeroDeTentativas = numeroDeTentativas;
    }

    /**
     * @return valor do atributo tempo.
     */
    public Integer getTempo() {
    	return tempo;
    }

    /**
     * Define valor do atributo tempo.
     * 
     * @param tempo
     */
    public void setTempo(Integer tempo) {
    	this.tempo = tempo;
    }

    /**
     * @return valor do atributo control.
     */
    public String getControl() {
    	return control;
    }

    /**
     * Define valor do atributo control.
     * 
     * @param control
     */
    public void setControl(String control) {
    	this.control = control;
    }
}