mbenjamin.java.bu

httptunnel.jar httptunnel java 源码

package net.jumperz.app.MBenjamin;

import java.net.*;
import java.io.*;
import net.jumperz.util.*;

public class MBenjamin
{
private static final int CONNECT_TIMEOUT	= 8000;
private static final String ADMIN_ADDR		= "[e-mail]";
private static final int ECHO_PORT		= 7; 
private static final int HTTP_PORT		= 80; 
private static final int FTP_PORT		= 21; 
private static final int SMTP_PORT		= 25; 
private static final int PROCESS_TIMEOUT	= 60000; //60昩偱廔椆

private String host;
private MThreadPool threadPool;
//----------------------------------------------------------------------------------
public static void main( String[] args )
{
if( args.length != 1 )
	{
	System.out.println( "Usage: Benjamin HOST" );
	return;
	}
new MBenjamin( args[ 0 ] );
}
//----------------------------------------------------------------------------------
public MBenjamin( String IN_host )
{
threadPool = new MThreadPool( 1 );
MTimer timer = new MTimer( PROCESS_TIMEOUT );
MTimerObserver observer = new MTimerObserver( this );
timer.regist( observer );
threadPool.addCommand( timer );

host = IN_host;
start();
}
//----------------------------------------------------------------------------------
private void start()
{
String alert	= "To: [e-mail]\r\nCc: [e-mail]\r\nSubject: benjamin alert\r\n\r\n" + host + " may be down.\r\n";

boolean error	= false;

if( isEchoAlive() == true )
	{
	return;
	}

if( isHTTPAlive() == false
 && isHTTPAlive() == false
  )
	{
	alert += "HTTP\r\n";
	error = true;
	}

if( isSMTPAlive() == false
 && isSMTPAlive() == false
  )
	{
	alert += "SMTP\r\n";
	error = true;
	}

if( isFTPAlive() == false
 && isFTPAlive() == false
  )
	{
	alert += "FTP\r\n";
	error = true;
	}

if( error )
	{
	SendMail( alert );
	}
}
//----------------------------------------------------------------------------------
private boolean isEchoAlive()
{
try
	{
	Socket socket = new Socket();
	InetSocketAddress addr = new InetSocketAddress( host, ECHO_PORT );
	socket.connect( addr, CONNECT_TIMEOUT );

	PrintWriter out = new PrintWriter( socket.getOutputStream() );
	BufferedReader in = new BufferedReader( new InputStreamReader( socket.getInputStream() ) );

	String greetz = "Are you alive?";
	out.println( greetz );
	out.flush();
	if( in.readLine().equals( greetz ) )
		{
		socket.close();
		return true;
		}
	socket.close();
	}
catch( IOException e )
	{
	}
return false;
}
//----------------------------------------------------------------------------------
private boolean isHTTPAlive()
{
try
	{
	Socket socket = new Socket();
	InetSocketAddress addr = new InetSocketAddress( host, HTTP_PORT );
	socket.connect( addr, CONNECT_TIMEOUT );

	PrintWriter out = new PrintWriter( socket.getOutputStream() );
	BufferedReader in = new BufferedReader( new InputStreamReader( socket.getInputStream() ) );

	String greetz = "HEAD / HTTP/1.0\r\n\r\n";
	out.print( greetz );
	out.flush();

	if( in.readLine().indexOf( "200 OK" ) > 0 )
		{
		while( in.readLine() != null )
			{
			}
		socket.close();
		return true;
		}
	socket.close();
	}
catch( IOException e )
	{
	//e.printStackTrace();
	}
return false;
}
//----------------------------------------------------------------------------------
private boolean isSMTPAlive()
{
try
	{
	Socket socket = new Socket();
	InetSocketAddress addr = new InetSocketAddress( host, SMTP_PORT );
	socket.connect( addr, CONNECT_TIMEOUT );

	BufferedReader in = new BufferedReader( new InputStreamReader( socket.getInputStream() ) );

	if( in.readLine().indexOf( "220" ) == 0 )
		{
		socket.close();
		return true;
		}
	socket.close();
	}
catch( IOException e )
	{
	//e.printStackTrace();
	}
return false;
}
//----------------------------------------------------------------------------------
private boolean isFTPAlive()
{
try
	{
	Socket socket = new Socket();
	InetSocketAddress addr = new InetSocketAddress( host, FTP_PORT );
	socket.connect( addr, CONNECT_TIMEOUT );

	BufferedReader in = new BufferedReader( new InputStreamReader( socket.getInputStream() ) );

	if( in.readLine().indexOf( "220" ) == 0 )
		{
		socket.close();
		return true;
		}
	socket.close();
	}
catch( IOException e )
	{
	//e.printStackTrace();
	}
return false;
}
//----------------------------------------------------------------------------------
private void SendMail( String message )
{
try
	{
	String command	= "/usr/sbin/sendmail -t";// + ADMIN_ADDR;
	Process process	= Runtime.getRuntime().exec( command );
	OutputStream os	= process.getOutputStream();

	byte[] buffer	= message.getBytes();
	os.write( buffer, 0, buffer.length );
	os.close();
	}
catch( Exception e )
	{
	e.printStackTrace();
	}
}
//----------------------------------------------------------------------------------
public void exit()
{
System.exit( 0 );
}
//----------------------------------------------------------------------------------
}