filetransfersettingsplugin.java

来自「开源项目openfire的完整源程序」· Java 代码 · 共 139 行

package org.jivesoftware.spark.plugins.transfersettings;

import org.jivesoftware.smack.packet.Message;
import org.jivesoftware.smackx.filetransfer.FileTransferRequest;
import org.jivesoftware.spark.SparkManager;
import org.jivesoftware.spark.plugins.transfersettings.FileTransferSettings;
import org.jivesoftware.spark.filetransfer.FileTransferListener;
import org.jivesoftware.spark.filetransfer.SparkTransferManager;
import org.jivesoftware.spark.plugin.Plugin;
import org.jivesoftware.spark.preference.PreferenceManager;

/**
 * Spark plugin which allows configuration of allowed file sizes, types, and senders for file transfer.
 * Transfer requests which don't meet the configured preferences are automatically rejecte.
 */
public class FileTransferSettingsPlugin implements Plugin {

    private PreferenceManager prefManager;

    /**
     * Called after Spark is loaded to initialize the new plugin.
     */
    public void initialize() {
        addTransferListener();
        prefManager = SparkManager.getPreferenceManager();
        prefManager.addPreference(new TransferSettingsPreference());
    }

    /**
     * Called when Spark is shutting down to allow for persistence of information
     * or releasing of resources.
     */
    public void shutdown() {

    }

    /**
     * Called when the plugin is uninstalled with the Spark plugin manager.
     */
    public void uninstall() {
    }

    /**
     * Return true if the Spark can shutdown on users request.
     *
     * @return true if Spark can shutdown on users request.
     */
    public boolean canShutDown() {
        return true;
    }

    /**
     * Tests the supplied {@link FileTransferRequest} against the supplied {@link FileTransferSettings}. Returns true if  the request
     * fails to match the configuration for allowed files.
     *
     * @param request  the transfer request to test.
     * @param settings the transfer settings to use in testing the request.
     * @return true if the request fails to match the configuration for allowed files.
     */
    private boolean requestContainsBannedFile(FileTransferRequest request, FileTransferSettings settings) {
        if (settings.getCheckFileSize() && request.getFileSize() > settings.getMaxFileSize()) {
            return true;
        }
        if (settings.getBlockedJIDs().contains(trimJID(request.getRequestor()))) {
            return true;
        }
        if (settings.getBlockedExtensions().contains(getFileExtensionFromName(request.getFileName()))) {
            return true;
        }
        return false;

    }

    /**
     * Strips the extension off the supplied filename and prepends an asterisk. For example 'bad.doc' would return
     * '*.doc'.
     *
     * @param filename to return the extension for.
     * @return the extension.
     */
    private String getFileExtensionFromName(String filename) {
        int dotIdx = filename.lastIndexOf(".");
        if (dotIdx > 0 && dotIdx < (filename.length() - 1)) {
            StringBuffer buffer = new StringBuffer("*");
            buffer.append(filename.substring(dotIdx));
            return buffer.toString();
        }

        return null;
    }

    /**
     * Trims the resource off the end of the supplied JID. For example, '[e-mail]/spark' would return
     * '[e-mail]'
     *
     * @param completeJID the JID which possibly includes a resource.
     * @return the JID without the resource.
     */
    private String trimJID(String completeJID) {
        int slashIDX = completeJID.indexOf('/');
        if (slashIDX > 0) {
            return completeJID.substring(0, slashIDX);
        }
        else {
            return completeJID;
        }
    }


    /**
     * Adds a {@link FileTransferListener to allow this plugin to intercept {@link FileTransferRequest}s.
     */
    private void addTransferListener() {

        SparkTransferManager transferManager = SparkManager.getTransferManager();

        transferManager.addTransferListener(new FileTransferListener() {
            public boolean handleTransfer(FileTransferRequest request) {
                FileTransferSettings settings = (FileTransferSettings)prefManager.getPreferenceData("transferSettings");

                if (requestContainsBannedFile(request, settings)) {
                    request.reject();

                    String responseMessage = settings.getCannedRejectionMessage();
                    if (responseMessage != null && responseMessage.length() > 0) {
                        Message message = new Message();
                        message.setTo(request.getRequestor());
                        message.setBody(responseMessage);
                        SparkManager.getConnection().sendPacket(message);
                    }
                    return true;
                }
                else {
                    return false;
                }
            }
        });
    }
}