blob: 9dae088bc550c96e313d3c138ef262919a9b3bd9 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
|
package org.openslx.imagemaster.ftp;
import java.io.File;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.Iterator;
import java.util.List;
import org.apache.ftpserver.FtpServer;
import org.apache.ftpserver.FtpServerFactory;
import org.apache.ftpserver.ftplet.Authority;
import org.apache.ftpserver.ftplet.FtpException;
import org.apache.ftpserver.ftplet.Ftplet;
import org.apache.ftpserver.ftplet.UserManager;
import org.apache.ftpserver.impl.FtpIoSession;
import org.apache.ftpserver.listener.Listener;
import org.apache.ftpserver.listener.ListenerFactory;
import org.apache.ftpserver.ssl.SslConfigurationFactory;
import org.apache.ftpserver.usermanager.PropertiesUserManagerFactory;
import org.apache.ftpserver.usermanager.SaltedPasswordEncryptor;
import org.apache.ftpserver.usermanager.impl.BaseUser;
import org.apache.ftpserver.usermanager.impl.WritePermission;
import org.apache.log4j.Logger;
import org.openslx.imagemaster.Globals;
import org.openslx.imagemaster.Globals.PropString;
import org.openslx.imagemaster.thrift.iface.FtpCredentials;
import org.openslx.imagemaster.util.RandomString;
import org.openslx.imagemaster.util.Util;
public class MasterFtpServer implements Runnable
{
private static Logger log = Logger.getLogger( MasterFtpServer.class );
private FtpServer server;
private UserManager userManager;
private Listener listener;
// key: ftpUsername, value: createTime
public final HashMap<String, Long> users = new HashMap<>();
private boolean ini = false;
public void init( int port )
{
if ( ini )
return;
FtpServerFactory serverFactory = new FtpServerFactory();
ListenerFactory factory = new ListenerFactory();
// config ssl
SslConfigurationFactory sslConfigFactory = new SslConfigurationFactory();
sslConfigFactory.setKeystoreFile( new File( Globals.getPropertyString( PropString.FTPKEYSTOREFILE ) ) );
sslConfigFactory.setKeyAlias( Globals.getPropertyString( PropString.FTPKEYSTOREALIAS ) );
sslConfigFactory.setKeystorePassword( Globals.getPropertyString( PropString.FTPKEYSTOREPASSWORD ) );
// set the port of the listener
factory.setPort( port );
factory.setSslConfiguration( sslConfigFactory.createSslConfiguration() );
factory.setImplicitSsl( true );
// replace the default listener
listener = factory.createListener();
serverFactory.addListener( "default", listener );
// create user manager
PropertiesUserManagerFactory userManagerFactory = new PropertiesUserManagerFactory();
File userFile = new File( "ftp.properties" );
userManagerFactory.setFile( userFile );
userManagerFactory.setPasswordEncryptor( new SaltedPasswordEncryptor() );
userManager = userManagerFactory.createUserManager();
serverFactory.setUserManager( userManager );
// add the Ftplet
HashMap<String, Ftplet> map = new HashMap<String, Ftplet>();
map.put( "Ftplet1", new MasterFtplet() );
serverFactory.setFtplets( map );
// start the server
server = serverFactory.createServer();
ini = true;
}
public FtpCredentials addUser( final String serverSessionId )
{
FtpCredentials ftpCredentials = null;
String generatedUser = RandomString.generate( 10, false );
String generatedPass = RandomString.generate( 16, true );
String dir = Globals.getPropertyString( Globals.PropString.FTPBASEDIR ) + "/"
+ generatedUser + "/";
if ( !new File( dir ).mkdir() ) {
return ftpCredentials;
}
BaseUser user = new BaseUser();
user.setName( generatedUser );
user.setPassword( generatedPass );
user.setHomeDirectory( dir );
List<Authority> authorities = new ArrayList<Authority>();
authorities.add( new WritePermission() );
user.setAuthorities( authorities );
try {
userManager.save( user );
ftpCredentials = new FtpCredentials( generatedUser, generatedPass );
synchronized ( users ) {
users.put( ftpCredentials.username, System.currentTimeMillis());
}
} catch ( FtpException e ) {
}
log.info( "Generated user/pass: " + generatedUser + "\t"
+ generatedPass + "\n with home dir: " + dir );
return ftpCredentials;
}
public boolean removeUser( final String username )
{
if ( !users.containsKey( username ) )
return false;
try {
// first find active session and close it
Iterator<FtpIoSession> iter = listener.getActiveSessions().iterator();
while (iter.hasNext()) {
FtpIoSession session = (FtpIoSession) iter.next();
if (session.getUser() == null) continue;
if (session.getUser().getName() == username) {
session.close();
}
}
// afterwards delete user
userManager.delete( username );
// remove user from map (cache)
synchronized ( users ) {
users.remove( username );
}
// remove his home dir
File dir = new File( Globals.getPropertyString( Globals.PropString.FTPBASEDIR ) + "/" + username );
Util.deleteFolder( dir );
return true;
} catch ( FtpException e ) {
return false;
}
}
@Override
public void run()
{
try {
//if (!ini) throw new Exception("FTP server needs to be initalized.");
log.info( "Starting FTP Sever" );
server.start();
} catch ( FtpException e1 ) {
e1.printStackTrace();
}
}
}
|
