ssignature.java

windows系统下用java开发的网络入侵检测程序

import java.sql.*;
import java.io.*;
/**
 * Transact connected user's look for or modify his/her signature.<p>
 * 2005.8.25<p>
 * @version 0.1.2
 * @author Daxin Tian
 *
 */
public class Ssignature {

	/**
	 * Who want to look for or modify signature.
	 */
	String user_name;
	/**
	 * Server's DataInputStream
	 */
	DataInputStream s_dis=null;
	/**
	 * Server's DataOutputStream
	 */
	DataOutputStream s_dos=null;
	/**
	 * Some information about the transact.
	 */
	String info=null;
	/**
	 * Contains the founded signature.
	 */
	String sig=null;
	/**
	 * Contains the new signature.
	 */
	String newsig=null;
	
	Connection con;
	Statement sql;
	ResultSet rs;
	
	/**
	 * 
	 * @param name Who sends the request.
	 * @param dis Server's DataInputStream
	 * @param dos Server's DataOutputStream
	 */
	Ssignature(String name,DataInputStream dis,DataOutputStream dos)
	{
		user_name=name;
	    s_dis=dis;
	    s_dos=dos;
	}
	/**
	 * Look for signature.
	 *
	 */
	public void req()
	{
		StringBuffer s_sb=null;
		
		try
		{
			Class.forName("com.mysql.jdbc.Driver");
			//Class.forName("sun.jdbc.odbc.JdbcOdbcDriver");
		}
		catch(ClassNotFoundException e)
		{
			info="db driver wrong";
			
		}
		try
		{
			con=DriverManager.getConnection("jdbc:mysql://localhost/jidx");
			//con=DriverManager.getConnection("jdbc:odbc:jidxdb","","");
			sql=con.createStatement();
			
			if(user_name!=null)
			{
				rs=sql.executeQuery("select info from register where username='"+user_name+"'");
				if(rs.next())
				{
					sig=rs.getString(1);
				}
				
				
				s_sb=new StringBuffer();
				s_sb.append("<msg><ack_signature><sig>"+sig+"</sig></ack_signature></msg>");
				try
				{
					//System.out.println("send ack sig: "+s_sb.toString());
					s_dos.writeUTF(s_sb.toString());
				}
				catch(IOException e)
				{
					
				}
				con.close();
			}
		}
		catch(SQLException e)
		{
			//System.out.println(e);
		}
	}
	/**
	 * Modify signature.
	 * @param sig New signature.
	 */
	public void mod(String sig)
	{
		newsig=sig;
		
		
		try
		{
			Class.forName("com.mysql.jdbc.Driver");
			//Class.forName("sun.jdbc.odbc.JdbcOdbcDriver");
		}
		catch(ClassNotFoundException e)
		{
			info="db driver wrong";
			
		}
		try
		{
			con=DriverManager.getConnection("jdbc:mysql://localhost/jidx");
			//con=DriverManager.getConnection("jdbc:odbc:jidxdb","","");
			sql=con.createStatement();
			
			if(user_name!=null)
			{
				sql.execute("update register set info='"+newsig+"' where username='"+user_name+"'");
			}
			con.close();
		}
		catch(SQLException e)
		{
			//System.out.println("moc sig"+e);
		}
	}
}