sitecommandhandler.java

一个利用Java语言实现的ftp程序

        return response;
    } 
      
    /**
     * Describe user.
     */
    public String doDESCUSER(String[] args, FtpRequest cmd) {
        if (args.length != 2) {
            return mConfig.getStatus().getResponse(501, cmd, mUser, null);
        }
        
        StringBuffer sb = new StringBuffer();
        sb.append('\n');
        User user = mConfig.getUserManager().getUserByName(args[1]);
        if (user != null) {
            sb.append("login : ").append(user.getName()).append('\n');
            sb.append("password : ").append("******").append('\n');
            sb.append("home : ").append(user.getVirtualDirectory().getRootDirectory()).append('\n');
            sb.append("writepermission : ").append(user.getVirtualDirectory().getWritePermission()).append('\n');
            sb.append("enable : ").append(user.getEnabled()).append('\n');
            sb.append("maxidletime : ").append(user.getMaxIdleTime()).append('\n');
            sb.append("maxuploadrate : ").append(user.getMaxUploadRate()).append('\n');
            sb.append("maxdownloadrate : ").append(user.getMaxDownloadRate()).append('\n');
        }
        sb.append('\n');
        return mConfig.getStatus().processNewLine(sb.toString(), 200);
    }  
            
    
    /**
     * Display site help.
     */
    public String doHELP(String[] args, FtpRequest cmd) {
        StringBuffer sb = new StringBuffer();
        sb.append('\n');
        sb.append("ADDIP <IP> : add banned IP entry").append('\n');
        sb.append("ADDUSER <userName> : add user").append('\n');
        sb.append("DELIP <IP> : delete banned IP entry").append('\n');
        sb.append("DELUSER <userName> : delete user").append('\n');
        sb.append("DESCUSER <userName> : describe user").append('\n');
        sb.append("HELP : display this message").append('\n');
        sb.append("KICK <userName> : close the connection").append('\n');
        sb.append("LISTIP : display all banned IPs").append('\n');
        sb.append("LISTUSER : display all user names").append('\n');
        sb.append("SETATTR <userName> <attrName> <attrValue> : set user attributes").append('\n');
        sb.append("STAT : show statistics").append('\n');
        sb.append("WHO : display all connected users").append('\n');
        sb.append('\n');
        return mConfig.getStatus().processNewLine(sb.toString(), 200);
    } 
   
    /**
     * Disconnect ftp connections
     */
    public String doKICK(String[] args, FtpRequest cmd) {
        if (args.length != 2) {
            return mConfig.getStatus().getResponse(501, cmd, mUser, null);
        }
        String userName = args[1];
        List allUsers = mConfig.getConnectionService().getAllUsers();
        for(Iterator userIt=allUsers.iterator(); userIt.hasNext(); ) {
            FtpUser user = (FtpUser)userIt.next();
            if ( userName.equals(user.getName()) ) {
                mConfig.getConnectionService().closeConnection(user.getSessionId());
            }
        }
        return mConfig.getStatus().getResponse(200, cmd, mUser, null);
    }
   
   
    /**
     * List all banned IPs.
     */
    public String doLISTIP(String[] args, FtpRequest cmd) {
        StringBuffer sb = new StringBuffer();
        sb.append('\n');
        for(Iterator ipIt=mConfig.getIpRestrictor().iterator(); ipIt.hasNext(); ) {
            sb.append(ipIt.next()).append('\n');
        }
        sb.append('\n');
        return mConfig.getStatus().processNewLine(sb.toString(), 200);
    }
   
   
    /**
     * List all the users.
     */
    public String doLISTUSER(String[] args, FtpRequest cmd) {
        Collection userNames = mConfig.getUserManager().getAllUserNames();
        StringBuffer sb = new StringBuffer();
        sb.append('\n');
        for(Iterator userIt=userNames.iterator(); userIt.hasNext(); ) {
            sb.append(userIt.next()).append('\n');
        }
        sb.append('\n');
        return mConfig.getStatus().processNewLine(sb.toString(), 200);
    }
   
   
    /**
     * Delete user from repository.
     */
    public String doSETATTR(String[] args, FtpRequest cmd) {
        if (args.length != 4) {
            return mConfig.getStatus().getResponse(501, cmd, mUser, null);
        }
        
        boolean bSuccess = true;
        try {
            User user = mConfig.getUserManager().getUserByName(args[1]);
            if (user != null) {
            
                if ( "password".equals(args[2]) ) {
                    user.setPassword(args[3]);
                }
                else if ( "home".equals(args[2]) ) {
                    user.getVirtualDirectory().setRootDirectory(args[3]);
                }
                else if ( "writepermission".equals(args[2]) ) {
                    user.getVirtualDirectory().setWritePermission("true".equals(args[3]));
                }
                else if ( "enable".equals(args[2]) ) {
                    user.setEnabled("true".equals(args[3]));
                }
                else if ( "maxidletime".equals(args[2]) ) {
                    user.setMaxIdleTime(Integer.parseInt(args[3]));
                }
                else if ( "maxuploadrate".equals(args[2]) ) {
                    user.setMaxUploadRate(Integer.parseInt(args[3]));
                }
                else if ( "maxdownloadrate".equals(args[2]) ) {
                    user.setMaxDownloadRate(Integer.parseInt(args[3]));
                }
                else {
                    bSuccess = false;
                }
                
                if (bSuccess) {
                    mConfig.getUserManager().save(user);
                }
            }
            else {
                bSuccess = false;
            }
        }
        catch(Exception ex) {
            mConfig.getLogger().warn(ex);
            bSuccess = false;
        }
        
        String response = "";
        if (bSuccess) {
            response = mConfig.getStatus().getResponse(200, cmd, mUser, null);
        }
        else {
            response = mConfig.getStatus().getResponse(451, cmd, mUser, null);
        }
        
        return response;
    }
    
   
    /**
     * Delete user from repository.
     */
    public String doSTAT(String[] args, FtpRequest cmd) {
        FtpStatistics stat = mConfig.getStatistics();
        StringBuffer sb = new StringBuffer();
        sb.append('\n');
        sb.append("Start Time               : ").append( ((SimpleDateFormat)DATE_FMT.get()).format(stat.getStartTime()) ).append('\n');
        sb.append("Upload Number            : ").append(stat.getFileUploadNbr()).append('\n');
        sb.append("Download Number          : ").append(stat.getFileDownloadNbr()).append('\n');
        sb.append("Delete Number            : ").append(stat.getFileDeleteNbr()).append('\n');
        sb.append("Uploade Bytes            : ").append(stat.getFileUploadSize()).append('\n');
        sb.append("Downloaded Bytes         : ").append(stat.getFileDownloadSize()).append('\n');
        sb.append("Current Logins           : ").append(stat.getLoginNbr()).append('\n');
        sb.append("Total Logins             : ").append(stat.getTotalLoginNbr()).append('\n');
        sb.append("Current Anonymous Logins : ").append(stat.getAnonLoginNbr()).append('\n');
        sb.append("Total Anonymous Logins   : ").append(stat.getTotalAnonLoginNbr()).append('\n');
        sb.append("Current Connections      : ").append(stat.getConnectionNbr()).append('\n');
        sb.append("Total Connections        : ").append(stat.getTotalConnectionNbr()).append('\n');
        sb.append('\n');
        return mConfig.getStatus().processNewLine(sb.toString(), 200);
    }
   
   
    /**
     * Display all connected users.
     */
    public String doWHO(String[] args, FtpRequest cmd) {
        StringBuffer sb = new StringBuffer();
        List allUsers = mConfig.getConnectionService().getAllUsers();
        
        sb.append('\n');
        for(Iterator userIt = allUsers.iterator(); userIt.hasNext(); ) {
            FtpUser user = (FtpUser)userIt.next();
            if (!user.hasLoggedIn()) {
                continue;
            }
            
            SimpleDateFormat fmt = (SimpleDateFormat)DATE_FMT.get();
            sb.append( StringUtils.pad(user.getName(), ' ', true, 16) );
            sb.append( StringUtils.pad(user.getClientAddress().getHostAddress(), ' ', true, 16) );
            sb.append( StringUtils.pad(fmt.format(new Date(user.getLoginTime())), ' ', true, 16) );
            sb.append( StringUtils.pad(fmt.format(new Date(user.getLastAccessTime())), ' ', true, 16) );
            sb.append('\n');
        }
        sb.append('\n');
        return mConfig.getStatus().processNewLine(sb.toString(), 200);
    }
      
}