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
|
package org.openslx.imagemaster.server;
import java.io.File;
import java.util.ArrayList;
import java.util.Date;
import java.util.HashMap;
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.listener.ListenerFactory;
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.thrift.iface.FtpCredentials;
import org.openslx.imagemaster.util.RandomString;
public class MasterFtpServer implements Runnable
{
private static Logger log = Logger.getLogger( MasterFtpServer.class );
private FtpServer server;
private UserManager userManager;
// key: serverSessionId, value: FtpCredentials
public final HashMap<String, FtpCredentials> users = new HashMap<>();
// key: ftpUsername, value: createTime
public final HashMap<String, Date> timeouts = new HashMap<>();
private boolean ini = false;
public void init(int port)
{
if (ini) return;
FtpServerFactory serverFactory = new FtpServerFactory();
ListenerFactory factory = new ListenerFactory();
// set the port of the listener
factory.setPort( port );
// replace the default listener
serverFactory.addListener( "default", factory.createListener() );
// create user manager
PropertiesUserManagerFactory userManagerFactory = new PropertiesUserManagerFactory();
userManagerFactory.setFile( new File( "src/main/properties/ftp.properties" ) );
userManagerFactory.setPasswordEncryptor( new SaltedPasswordEncryptor() );
userManager = userManagerFactory.createUserManager();
// create new admin user
//addUser(adminUsername, adminPassword, ftproot, true);
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();
}
public FtpCredentials addUser( final String serverSessionId )
{
// TODO: enable SSL
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 );
users.put( serverSessionId, ftpCredentials );
timeouts.put( ftpCredentials.username, new Date() );
} catch ( FtpException e ) {
}
log.info( "Generated user/pass: " + generatedUser + "\t"
+ generatedPass + "\n with home dir: " + dir );
return ftpCredentials;
}
public boolean removeUser( final String serverSessionId )
{
if ( !users.containsKey( serverSessionId ) )
return false;
try {
userManager.delete( users.get( serverSessionId ).username );
// remove user from both maps
timeouts.remove( users.remove( serverSessionId ).username );
return true;
} catch ( FtpException e ) {
return false;
}
}
public FtpCredentials getCredentialsFromSessionId( String serverSessionId )
{
return users.get( serverSessionId );
}
@Override
public void run()
{
try {
log.info( "Starting FTP Sever" );
server.start();
} catch ( FtpException e1 ) {
e1.printStackTrace();
}
}
}
|
