ftpserver.java

jbuilder应用开发程序

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

public class FtpServer
{
	private Socket socketClient;
	private int counter;
	public static String initDir;
	public static ArrayList users = new ArrayList();
	public static ArrayList usersInfo = new ArrayList();
	
	public FtpServer()
	{
		FtpConsole fc = new FtpConsole();
		fc.start();
		loadUsersInfo();
		int counter = 1;
		int i = 0;
		try
		{

			//监听21号端口
			ServerSocket s = new ServerSocket(21);
			for(;;)
			{
				//接受客户端请求
				Socket incoming = s.accept();
				BufferedReader in = new BufferedReader(new InputStreamReader(incoming.getInputStream()));
			    PrintWriter out = new PrintWriter(incoming.getOutputStream(),true);
				out.println("220 Service ready for new user"+counter);

				//创建服务线程
				FtpHandler h = new FtpHandler(incoming,i);
				h.start();
				users.add(h);
				counter++;
				i++;
			}
		}
		catch(Exception e)
		{
			e.printStackTrace();
		}
		
	} // FtpServer() end
	
	public void loadUsersInfo()
	{
		String s = getClass().getResource("user.cfg").toString();
		s = s.substring(6,s.length());
		int p1 = 0;
		int p2 = 0;
		if(new File(s).exists())
		{
			try
			{
				BufferedReader fin = new BufferedReader(new InputStreamReader(new FileInputStream(s)));
				String line;
				String field;
				 
				
				int i = 0;
				while((line = fin.readLine())!=null)
				{
					UserInfo tempUserInfo = new UserInfo();
					p1 = 0;
					p2 = 0;
					i = 0;
					if(line.startsWith("#"))
						continue;
					while((p2 = line.indexOf("|",p1))!=-1)
					{
						field = line.substring(p1,p2);
						p2 = p2 +1;
						p1 = p2;
						switch(i)
						{
							case 0:
								tempUserInfo.user = field;
								//System.out.println(tempUserInfo.user);
								break;
							case 1:
								tempUserInfo.password = field;
								//System.out.println(tempUserInfo.password);
								break;
							case 2:
								tempUserInfo.workDir = field;
								//System.out.println(tempUserInfo.workDir);
								break;
						}
						i++;
					} //while((p2 = line.indexOf("|",p1))!=-1) end
					usersInfo.add(tempUserInfo);
				}//while((line = fin.readLine())!=null) end
				fin.close();
			}
			catch(Exception e)
			{
				e.printStackTrace();
			}
		}//if(new File(s).exists()) end
	}// loadUsersInfo() end
	
	public static void main(String[] args)
	{
		if(args.length != 0) 
		{
			initDir = args[0];
		}
		else
		{ 
			initDir = "c:/";	
		}
		FtpServer ftpServer = new FtpServer();
		
	} // main end
		
}

class FtpHandler extends Thread
{
	Socket csocket;
	Socket dsocket;
	int id;
	String cmd = "";
	String param = "";
	String user;
	String remoteHost = " ";
	int remotePort = 0;
	String dir = FtpServer.initDir;
	String rootdir = "c:/";
	int state = 0;
	String reply;
	PrintWriter out; 
	int type = 0;
	String requestfile = "";
	boolean isrest = false;
	
	int parseInput(String s)
	{
		int p = 0;
		int i = -1;
		p = s.indexOf(" ");
		if(p == -1)
			cmd = s;
		else 
			cmd = s.substring(0,p);
		
		if(p >= s.length() || p ==-1)
			param = "";
		else
			param = s.substring(p+1,s.length());
		cmd = cmd.toUpperCase();
		
		if(cmd.equals("USER"))
				i = 1;
		if(cmd.equals("PASS"))
				i = 2;
		if(cmd.equals("ACCT"))
				i = 3;
	  	if(cmd.equals("CDUP"))
				i = 4;
		if(cmd.equals("SMNT"))
				i = 5;
		if(cmd.equals("CWD"))
				i = 6;
		if(cmd.equals("QUIT"))
				i = 7;
	  	if(cmd.equals("REIN"))
				i = 8;
		if(cmd.equals("PORT"))
				i = 9;
		if(cmd.equals("PASV"))
				i = 10;
		if(cmd.equals("TYPE"))
				i = 11;
	  	if(cmd.equals("STRU"))
				i = 12;
		if(cmd.equals("MODE"))
				i = 13;
		if(cmd.equals("RETR"))
				i = 14;
		if(cmd.equals("STOR"))
				i = 15;
	  	if(cmd.equals("STOU"))
				i = 16;
		if(cmd.equals("APPE"))
				i = 17;
		if(cmd.equals("ALLO"))
				i = 18;
		if(cmd.equals("REST"))
				i = 19;
	  	if(cmd.equals("RNFR"))
				i = 20;
		if(cmd.equals("RNTO"))
				i = 21;
		if(cmd.equals("ABOR"))
				i = 22;
		if(cmd.equals("DELE"))
				i = 23;
	  	if(cmd.equals("RMD"))
				i = 24;
		if(cmd.equals("XMKD"))
				i = 25;				
		if(cmd.equals("MKD"))
				i = 25;
		if(cmd.equals("PWD"))
				i = 26;
		if(cmd.equals("LIST"))
				i = 27;
	  	if(cmd.equals("NLST"))
				i = 28;
		if(cmd.equals("SITE"))
				i = 29;
		if(cmd.equals("SYST"))
				i = 30;
		if(cmd.equals("HELP"))
				i = 31;
	  	if(cmd.equals("NOOP"))
				i = 32;
		if(cmd.equals("XPWD"))
				i = 33;
	 return i;
	}//parseInput() end
	
	int validatePath(String s)
	{
		File f = new File(s);
		if(f.exists() && !f.isDirectory())
		{
			String s1 = s.toLowerCase();
			String s2 = rootdir.toLowerCase();
			if(s1.startsWith(s2))
				return 1;
			else
				return 0;
		}
		f = new File(addTail(dir)+s);
		if(f.exists() && !f.isDirectory())
		{
			String s1 = (addTail(dir)+s).toLowerCase();
			String s2 = rootdir.toLowerCase();
			if(s1.startsWith(s2))
				return 2;
			else 
				return 0;
		}
		return 0;
	}// validatePath() end
	
	boolean checkPASS(String s)
	{
		for(int i = 0; i<FtpServer.usersInfo.size();i++)
		{
			if(((UserInfo)FtpServer.usersInfo.get(i)).user.equals(user) && 
				((UserInfo)FtpServer.usersInfo.get(i)).password.equals(s))
			{
				rootdir = ((UserInfo)FtpServer.usersInfo.get(i)).workDir;
				dir = ((UserInfo)FtpServer.usersInfo.get(i)).workDir;
				return true;
			}
		}
		return false;
	}// checkPASS() end

	boolean commandUSER()
	{
		if(cmd.equals("USER"))
		{
			reply = "331 User name okay, need password";
			user = param;
		  	state = FtpState.FS_WAIT_PASS;
			return false;
		}
		else
		{
			reply = "501 Syntax error in parameters or arguments";
			return true;
		}

	}//commandUser() end

	boolean commandPASS()
	{
		if(cmd.equals("PASS"))
		{
			if(checkPASS(param))
			{
				reply = "230 User logged in, proceed";
				state = FtpState.FS_LOGIN;
				System.out.println("Message: user "+user+" Form "+remoteHost+"Login");
				System.out.print("->");
				return false;
			}
			else
			{
				reply = "530 Not logged in";
				return true;
			}
		}
		else
		{
			reply = "501 Syntax error in parameters or arguments";
			return true;
		}

	}//commandPass() end

	void errCMD()
	{
		reply = "500 Syntax error, command unrecognized";
	}	
	
	boolean commandCDUP()
	{
		dir = FtpServer.initDir;
		File f = new File(dir);
		if(f.getParent()!=null &&(!dir.equals(rootdir)))
		{
			dir = f.getParent();
			reply = "200 Command okay";
		}
		else
		{
			reply = "550 Current directory has no parent";
		}
		
		return false;
	}// commandCDUP() end

	boolean commandCWD()
	{
		File f = new File(param);
		String s = "";
		String s1 = "";
		if(dir.endsWith("/"))
			s = dir;
		else
			s = dir + "/";
		File f1 = new File(s+param);
		
		if(f.isDirectory() && f.exists())
		{
			if(param.equals("..") || param.equals("..\\"))
			{
				if(dir.compareToIgnoreCase(rootdir)==0)
				{
					reply = "550 The directory does not exists";
					//return false;
				}
				else
				{
					s1 = new File(dir).getParent();
					if(s1!=null)
					{
						dir = s1;
						reply = "250 Requested file action okay, directory change to "+dir;
					}
					else
						reply = "550 The directory does not exists";
				}
			}
			else if(param.equals(".") || param.equals(".\\"))
			{
			}
			else 
			{
				dir = param;
				reply = "250 Requested file action okay, directory change to "+dir;
			}		
		}
		else if(f1.isDirectory() && f1.exists())
		{
			dir = s+param;
			reply = "250 Requested file action okay, directory change to "+dir;
		}
		else
			reply = "501 Syntax error in parameters or arguments";
		
		return false;
	} // commandCDW() end

	boolean commandQUIT()
	{
		reply = "221 Service closing control connection";
		return true;
	}// commandQuit() end

	boolean commandPORT()
	{
		int p1 = 0;
		int p2 = 0;
		int[] a = new int[6];
		int i = 0;
		try
		{
			while((p2 = param.indexOf(",",p1))!=-1)
			{
				 a[i] = Integer.parseInt(param.substring(p1,p2));
				 p2 = p2+1;
				 p1 = p2;
				 i++;
			}
			a[i] = Integer.parseInt(param.substring(p1,param.length()));
		}
		catch(NumberFormatException e)
		{
			reply = "501 Syntax error in parameters or arguments";
			return false;
		}
		
		remoteHost = a[0]+"."+a[1]+"."+a[2]+"."+a[3];
		remotePort = a[4] * 256+a[5];
		reply = "200 Command okay";
		return false;
	}//commandPort() end	
	
	boolean commandLIST()
	{
		try
		{
			dsocket = new Socket(remoteHost,remotePort,InetAddress.getLocalHost(),20);
			PrintWriter dout = new PrintWriter(dsocket.getOutputStream(),true);
			if(param.equals("") || param.equals("LIST"))
			{
				out.println("150 Opening ASCII mode data connection for /bin/ls. ");
				File f = new File(dir);
				String[] dirStructure = f.list();
				String fileType;
				for(int i =0; i<dirStructure.length;i++)
				{
					if(dirStructure[i].indexOf(".")!=-1)
					{
						fileType = "- ";
					}
					else
					{
						fileType = "d ";
					}
					dout.println(fileType+dirStructure[i]);
				}
			} 
			dout.close();
			dsocket.close();
			reply = "226 Transfer complete !";
		}
		catch(Exception e)
		{
			e.printStackTrace();
			reply = "451 Requested action aborted: local error in processing";
			return false;
		}
		
		return false;
	}// commandLIST() end

	boolean commandTYPE()
	{
		if(param.equals("a"))
		{
			type = FtpState.FTYPE_ASCII;
			reply = "200 Command okay Change to ASCII mode";
		}
		else if(param.equals("i"))
		{
			type = FtpState.FTYPE_IMAGE;
			reply = "200 Command okay Change to BINARY mode";
		}
		else
			reply = "504 Command not implemented for that parameter";
			
		return false;
	}//commandTYPE() end

	boolean commandRETR()
	{
		requestfile = param;
		File f =  new File(requestfile);
  		if(!f.exists())
		{
	  		f = new File(addTail(dir)+param);
			if(!f.exists())
			{
	   			reply = "550 File not found";
	   			return  false;
			}
			requestfile = addTail(dir)+param;
		}
  
  		if(isrest)
		{
     
		}
		else
		{
	 		if(type==FtpState.FTYPE_IMAGE)
			{
				try
				{
					out.println("150 Opening Binary mode data connection for "+ requestfile);
					dsocket = new Socket(remoteHost,remotePort,InetAddress.getLocalHost(),20);
    				BufferedInputStream  fin = new BufferedInputStream(new FileInputStream(requestfile));
	  				PrintStream dout = new PrintStream(dsocket.getOutputStream(),true);
					byte[] buf = new byte[1024];
					int l = 0;
					while((l=fin.read(buf,0,1024))!=-1)
					{
			  			dout.write(buf,0,l);
					}
		 			fin.close();
     				dout.close();
		 			dsocket.close();
		 			reply ="226 Transfer complete !";

				}
				catch(Exception e)
				{
					e.printStackTrace();
					reply = "451 Requested action aborted: local error in processing";
					return false;
				}

			}
			if(type==FtpState.FTYPE_ASCII)
			{
	  			try
				{
					out.println("150 Opening ASCII mode data connection for "+ requestfile);
					dsocket = new Socket(remoteHost,remotePort,InetAddress.getLocalHost(),20);
    				BufferedReader  fin = new BufferedReader(new FileReader(requestfile));
	  				PrintWriter dout = new PrintWriter(dsocket.getOutputStream(),true);
					String s;
					while((s=fin.readLine())!=null)
					{
		   				dout.println(s);
					}
		 			fin.close();
     				dout.close();
		 			dsocket.close();
		 			reply ="226 Transfer complete !";
				}
				catch(Exception e)
				{
					e.printStackTrace();
					reply = "451 Requested action aborted: local error in processing";
					return false;
				}
			}
		}