localizedftpreply.java

来自「JAVA FTP 上传下载 的源文件」· Java 代码 · 共 507 行 · 第 1/2 页

/*
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you under the Apache License, Version 2.0 (the
 * "License"); you may not use this file except in compliance
 * with the License.  You may obtain a copy of the License at
 *
 *  http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing,
 * software distributed under the License is distributed on an
 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
 * KIND, either express or implied.  See the License for the
 * specific language governing permissions and limitations
 * under the License.
 */

package org.apache.ftpserver.impl;

import java.net.InetAddress;
import java.net.InetSocketAddress;
import java.net.SocketAddress;

import org.apache.ftpserver.ftplet.DefaultFtpReply;
import org.apache.ftpserver.ftplet.FileSystemView;
import org.apache.ftpserver.ftplet.FtpRequest;
import org.apache.ftpserver.ftplet.FtpStatistics;
import org.apache.ftpserver.message.MessageResource;
import org.apache.ftpserver.util.DateUtils;

/**
 * <strong>Internal class, do not use directly.</strong>
 * 
 * FTP reply translator.
 *
 * @author The Apache MINA Project ([e-mail])
 * @version $Rev$, $Date$
 */
public class LocalizedFtpReply extends DefaultFtpReply {


    public static final String CLIENT_ACCESS_TIME = "client.access.time";

    public static final String CLIENT_CON_TIME = "client.con.time";

    public static final String CLIENT_DIR = "client.dir";

    public static final String CLIENT_HOME = "client.home";

    public static final String CLIENT_IP = "client.ip";

    public static final String CLIENT_LOGIN_NAME = "client.login.name";

    public static final String CLIENT_LOGIN_TIME = "client.login.time";

    public static final String OUTPUT_CODE = "output.code";

    public static final String OUTPUT_MSG = "output.msg";

    public static final String REQUEST_ARG = "request.arg";

    public static final String REQUEST_CMD = "request.cmd";

    public static final String REQUEST_LINE = "request.line";

    // /////////////////////// All Server Vatiables /////////////////////////
    public static final String SERVER_IP = "server.ip";

    public static final String SERVER_PORT = "server.port";

    public static final String STAT_CON_CURR = "stat.con.curr";

    public static final String STAT_CON_TOTAL = "stat.con.total";

    public static final String STAT_DIR_CREATE_COUNT = "stat.dir.create.count";

    public static final String STAT_DIR_DELETE_COUNT = "stat.dir.delete.count";

    public static final String STAT_FILE_DELETE_COUNT = "stat.file.delete.count";

    public static final String STAT_FILE_DOWNLOAD_BYTES = "stat.file.download.bytes";

    public static final String STAT_FILE_DOWNLOAD_COUNT = "stat.file.download.count";

    public static final String STAT_FILE_UPLOAD_BYTES = "stat.file.upload.bytes";

    public static final String STAT_FILE_UPLOAD_COUNT = "stat.file.upload.count";

    public static final String STAT_LOGIN_ANON_CURR = "stat.login.anon.curr";

    public static final String STAT_LOGIN_ANON_TOTAL = "stat.login.anon.total";

    public static final String STAT_LOGIN_CURR = "stat.login.curr";

    public static final String STAT_LOGIN_TOTAL = "stat.login.total";

    public static final String STAT_START_TIME = "stat.start.time";

    public static LocalizedFtpReply translate(FtpIoSession session, FtpRequest request,
            FtpServerContext context, int code, String subId, String basicMsg) {
        String msg = translateMessage(session, request, context, code, subId,
                basicMsg);

        return new LocalizedFtpReply(code, msg);
    }

    private static String translateMessage(FtpIoSession session,
            FtpRequest request, FtpServerContext context, int code,
            String subId, String basicMsg) {
        MessageResource resource = context.getMessageResource();
        String lang = session.getLanguage();

        String msg = null;
        if (resource != null) {
            msg = resource.getMessage(code, subId, lang);
        }
        if (msg == null) {
            msg = "";
        }
        msg = replaceVariables(session, request, context, code, basicMsg, msg);

        return msg;
    }

    /**
     * Replace server variables.
     */
    private static String replaceVariables(FtpIoSession session,
            FtpRequest request, FtpServerContext context, int code,
            String basicMsg, String str) {

        int startIndex = 0;
        int openIndex = str.indexOf('{', startIndex);
        if (openIndex == -1) {
            return str;
        }

        int closeIndex = str.indexOf('}', startIndex);
        if ((closeIndex == -1) || (openIndex > closeIndex)) {
            return str;
        }

        StringBuffer sb = new StringBuffer(128);
        sb.append(str.substring(startIndex, openIndex));
        while (true) {
            String varName = str.substring(openIndex + 1, closeIndex);
            sb.append(getVariableValue(session, request, context, code,
                    basicMsg, varName));

            startIndex = closeIndex + 1;
            openIndex = str.indexOf('{', startIndex);
            if (openIndex == -1) {
                sb.append(str.substring(startIndex));
                break;
            }

            closeIndex = str.indexOf('}', startIndex);
            if ((closeIndex == -1) || (openIndex > closeIndex)) {
                sb.append(str.substring(startIndex));
                break;
            }
            sb.append(str.substring(startIndex, openIndex));
        }
        return sb.toString();
    }

    /**
     * Get the variable value.
     */
    private static String getVariableValue(FtpIoSession session,
            FtpRequest request, FtpServerContext context, int code,
            String basicMsg, String varName) {

        String varVal = null;

        // all output variables
        if (varName.startsWith("output.")) {
            varVal = getOutputVariableValue(session, code, basicMsg, varName);
        }

        // all server variables
        else if (varName.startsWith("server.")) {
            varVal = getServerVariableValue(session, varName);
        }

        // all request variables
        else if (varName.startsWith("request.")) {
            varVal = getRequestVariableValue(session, request, varName);
        }

        // all statistical variables
        else if (varName.startsWith("stat.")) {
            varVal = getStatisticalVariableValue(session, context, varName);
        }

        // all client variables
        else if (varName.startsWith("client.")) {
            varVal = getClientVariableValue(session, varName);
        }

        if (varVal == null) {
            varVal = "";
        }
        return varVal;
    }

    /**
     * Get client variable value.
     */
    private static String getClientVariableValue(FtpIoSession session,
            String varName) {

        String varVal = null;

        // client ip
        if (varName.equals(CLIENT_IP)) {
            if (session.getRemoteAddress() instanceof InetSocketAddress) {
                InetSocketAddress remoteSocketAddress = (InetSocketAddress) session
                        .getRemoteAddress();
                varVal = remoteSocketAddress.getAddress().getHostAddress();
            }

        }

        // client connection time
        else if (varName.equals(CLIENT_CON_TIME)) {
            varVal = DateUtils.getISO8601Date(session.getCreationTime());
        }

        // client login name
        else if (varName.equals(CLIENT_LOGIN_NAME)) {
            if (session.getUser() != null) {
                varVal = session.getUser().getName();
            }
        }

        // client login time
        else if (varName.equals(CLIENT_LOGIN_TIME)) {
            varVal = DateUtils.getISO8601Date(session.getLoginTime().getTime());
        }

        // client last access time
        else if (varName.equals(CLIENT_ACCESS_TIME)) {
            varVal = DateUtils.getISO8601Date(session.getLastAccessTime()
                    .getTime());
        }

        // client home
        else if (varName.equals(CLIENT_HOME)) {
            varVal = session.getUser().getHomeDirectory();
        }