Browse Source

add SimpleFtpServer

Looly 5 years ago
parent
commit
cd1e9fbaad

+ 1 - 0
CHANGELOG.md

@@ -17,6 +17,7 @@
 * 【core   】     增加enum转数字支持(issue#I24QZY@Gitee)
 * 【core   】     NumberUtil.toBigDecimal空白符转换为0(issue#I24MRP@Gitee)
 * 【core   】     CollUtil和IterUtil增加size方法(pr#208@Gitee)
+* 【extra  】     新增SimpleFtpServer
 
 ### Bug修复
 * 【core   】     修复DateUtil.current使用System.nanoTime的问题(issue#1198@Github)

+ 13 - 0
hutool-extra/pom.xml

@@ -146,6 +146,19 @@
 			<scope>compile</scope>
 			<optional>true</optional>
 		</dependency>
+		<dependency>
+			<groupId>org.apache.ftpserver</groupId>
+			<artifactId>ftpserver-core</artifactId>
+			<version>1.1.1</version>
+			<scope>compile</scope>
+			<exclusions>
+				<exclusion>
+					<artifactId>slf4j-api</artifactId>
+					<groupId>org.slf4j</groupId>
+				</exclusion>
+			</exclusions>
+			<optional>true</optional>
+		</dependency>
 		<!-- Emoji工具依赖 -->
 		<dependency>
 			<groupId>com.vdurmont</groupId>

+ 221 - 0
hutool-extra/src/main/java/cn/hutool/extra/ftp/SimpleFtpServer.java

@@ -0,0 +1,221 @@
+package cn.hutool.extra.ftp;
+
+import cn.hutool.core.lang.Assert;
+import cn.hutool.core.net.NetUtil;
+import org.apache.ftpserver.ConnectionConfig;
+import org.apache.ftpserver.FtpServerFactory;
+import org.apache.ftpserver.ftplet.Authority;
+import org.apache.ftpserver.ftplet.Ftplet;
+import org.apache.ftpserver.ftplet.User;
+import org.apache.ftpserver.ftplet.UserManager;
+import org.apache.ftpserver.listener.ListenerFactory;
+import org.apache.ftpserver.ssl.SslConfiguration;
+import org.apache.ftpserver.ssl.SslConfigurationFactory;
+import org.apache.ftpserver.usermanager.PropertiesUserManagerFactory;
+import org.apache.ftpserver.usermanager.impl.BaseUser;
+import org.apache.ftpserver.usermanager.impl.WritePermission;
+
+import java.io.File;
+import java.util.ArrayList;
+import java.util.List;
+
+/**
+ * 基于 Apache FtpServer(http://mina.apache.org/ftpserver-project/)的FTP服务端简单封装。
+ *
+ * @author looly
+ * @since 5.5.0
+ */
+public class SimpleFtpServer {
+
+	FtpServerFactory serverFactory;
+	ListenerFactory listenerFactory;
+
+	/**
+	 * 构造
+	 */
+	public SimpleFtpServer() {
+		serverFactory = new FtpServerFactory();
+		listenerFactory = new ListenerFactory();
+	}
+
+	/**
+	 * 创建FTP服务器,调用{@link SimpleFtpServer#start()}启动即可
+	 *
+	 * @return SimpleFtpServer
+	 */
+	public static SimpleFtpServer create() {
+		return new SimpleFtpServer();
+	}
+
+	/**
+	 * 获取 {@link FtpServerFactory},用于设置FTP服务器相关信息
+	 *
+	 * @return {@link FtpServerFactory}
+	 */
+	public FtpServerFactory getServerFactory() {
+		return this.serverFactory;
+	}
+
+	/**
+	 * 设置连接相关配置,使用ConnectionConfigFactory创建{@link ConnectionConfig}对象
+	 *
+	 * @param connectionConfig 连接配置
+	 * @return this
+	 */
+	public SimpleFtpServer setConnectionConfig(ConnectionConfig connectionConfig) {
+		this.serverFactory.setConnectionConfig(connectionConfig);
+		return this;
+	}
+
+	/**
+	 * 获取{@link ListenerFactory},用于设置端口、用户、SSL等信息
+	 *
+	 * @return {@link ListenerFactory}
+	 */
+	public ListenerFactory getListenerFactory() {
+		return this.listenerFactory;
+	}
+
+	/**
+	 * 自定义默认端口,如果不设置,使用默认端口:21
+	 *
+	 * @param port 端口
+	 * @return this
+	 */
+	public SimpleFtpServer setPort(int port) {
+		Assert.isTrue(NetUtil.isValidPort(port), "Invalid port!");
+		this.listenerFactory.setPort(port);
+		return this;
+	}
+
+	/**
+	 * 获取用户管理器,用于新增、查找和删除用户信息
+	 *
+	 * @return 用户管理器
+	 */
+	public UserManager getUserManager() {
+		return this.serverFactory.getUserManager();
+	}
+
+	/**
+	 * 增加FTP用户
+	 *
+	 * @param user FTP用户信息
+	 * @return this
+	 */
+	public SimpleFtpServer addUser(User user) {
+		try {
+			getUserManager().save(user);
+		} catch (org.apache.ftpserver.ftplet.FtpException e) {
+			throw new FtpException(e);
+		}
+		return this;
+	}
+
+	/**
+	 * 添加匿名用户
+	 *
+	 * @param homePath 用户路径,匿名用户对此路径有读写权限
+	 * @return this
+	 */
+	public SimpleFtpServer addAnonymous(String homePath) {
+		BaseUser user = new BaseUser();
+		user.setName("anonymous");
+		user.setHomeDirectory(homePath);
+		List<Authority> authorities = new ArrayList<>();
+		// 添加用户读写权限
+		authorities.add(new WritePermission());
+		user.setAuthorities(authorities);
+		return addUser(user);
+	}
+
+	/**
+	 * 删除用户
+	 *
+	 * @param userName 用户名
+	 * @return this
+	 */
+	public SimpleFtpServer delUser(String userName) {
+		try {
+			getUserManager().delete(userName);
+		} catch (org.apache.ftpserver.ftplet.FtpException e) {
+			throw new FtpException(e);
+		}
+		return this;
+	}
+
+	/**
+	 * 使用SSL安全连接,可以使用SslConfigurationFactory创建{@link SslConfiguration}
+	 *
+	 * @param ssl {@link SslConfiguration}
+	 * @return this
+	 */
+	public SimpleFtpServer setSsl(SslConfiguration ssl) {
+		this.listenerFactory.setSslConfiguration(ssl);
+		listenerFactory.setImplicitSsl(true);
+		return this;
+	}
+
+	/**
+	 * 使用SSL安全连接
+	 *
+	 * @param keystoreFile 密钥文件
+	 * @param password     密钥文件密码
+	 * @return this
+	 */
+	public SimpleFtpServer setSsl(File keystoreFile, String password) {
+		SslConfigurationFactory sslFactory = new SslConfigurationFactory();
+		sslFactory.setKeystoreFile(keystoreFile);
+		sslFactory.setKeystorePassword(password);
+		return setSsl(sslFactory.createSslConfiguration());
+	}
+
+	/**
+	 * 自定义用户管理器,一般用于使用配置文件配置用户信息
+	 *
+	 * @param userManager {@link UserManager}
+	 * @return this
+	 */
+	public SimpleFtpServer setUserManager(UserManager userManager) {
+		this.serverFactory.setUserManager(userManager);
+		return this;
+	}
+
+	/**
+	 * 自定义用户信息配置文件,此方法会重置用户管理器
+	 *
+	 * @param propertiesFile 配置文件
+	 * @return this
+	 */
+	public SimpleFtpServer setUsersConfig(File propertiesFile) {
+		final PropertiesUserManagerFactory userManagerFactory = new PropertiesUserManagerFactory();
+		userManagerFactory.setFile(propertiesFile);
+		return this.setUserManager(userManagerFactory.createUserManager());
+	}
+
+	/**
+	 * 增加FTP动作行为监听处理器,通过实现{@link Ftplet},可以对用户的行为监听并执行相应动作
+	 *
+	 * @param name   名称
+	 * @param ftplet {@link Ftplet},用户自定义监听规则
+	 * @return this
+	 */
+	public SimpleFtpServer addFtplet(String name, Ftplet ftplet) {
+		this.serverFactory.getFtplets().put(name, ftplet);
+		return this;
+	}
+
+	/**
+	 * 启动FTP服务,阻塞当前线程
+	 */
+	public void start() {
+		// 一个Listener对应一个监听端口
+		// 可以创建多个监听,此处默认只监听一个
+		serverFactory.addListener("default", listenerFactory.createListener());
+		try {
+			serverFactory.createServer().start();
+		} catch (org.apache.ftpserver.ftplet.FtpException e) {
+			throw new cn.hutool.extra.ftp.FtpException(e);
+		}
+	}
+}

+ 11 - 0
hutool-extra/src/test/java/cn/hutool/extra/ftp/SimpleFtpServerTest.java

@@ -0,0 +1,11 @@
+package cn.hutool.extra.ftp;
+
+public class SimpleFtpServerTest {
+
+	public static void main(String[] args) {
+		SimpleFtpServer
+				.create()
+				.addAnonymous("d:/test/ftp/")
+				.start();
+	}
+}