blob: 3a745e25935d4d9266eaeb33ebea015c866dbafd (
plain)
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
|
/*
* Copyright (c) 2010-2015 Isode Limited.
* All rights reserved.
* See the COPYING file for more information.
*/
/*
* Copyright (c) 2015 Tarun Gupta.
* Licensed under the simplified BSD license.
* See Documentation/Licenses/BSD-simplified.txt for more information.
*/
package com.isode.stroke.filetransfer;
import com.isode.stroke.network.ConnectionServer;
import com.isode.stroke.network.HostAddressPort;
import com.isode.stroke.network.Connection;
import com.isode.stroke.jid.JID;
import com.isode.stroke.signals.SignalConnection;
import com.isode.stroke.signals.Slot1;
import com.isode.stroke.signals.Slot;
import java.util.Vector;
public class SOCKS5BytestreamServer {
private ConnectionServer connectionServer;
private SOCKS5BytestreamRegistry registry;
private Vector<SOCKS5BytestreamServerSession> sessions = new Vector<SOCKS5BytestreamServerSession>();
private SignalConnection onNewConnectionConn;
private SignalConnection onFinishedConnection;
public SOCKS5BytestreamServer(ConnectionServer connectionServer, SOCKS5BytestreamRegistry registry) {
this.connectionServer = connectionServer;
this.registry = registry;
}
public HostAddressPort getAddressPort() {
return connectionServer.getAddressPort();
}
public void start() {
onNewConnectionConn = connectionServer.onNewConnection.connect(new Slot1<Connection>() {
@Override
public void call(Connection c) {
handleNewConnection(c);
}
});
}
public void stop() {
onNewConnectionConn.disconnect();
for (SOCKS5BytestreamServerSession session : sessions) {
session.onFinished.disconnectAll();
session.stop();
}
sessions.clear();
}
public Vector<SOCKS5BytestreamServerSession> getSessions(final String streamID) {
Vector<SOCKS5BytestreamServerSession> result = new Vector<SOCKS5BytestreamServerSession>();
for (SOCKS5BytestreamServerSession session : sessions) {
if (session.getStreamID().equals(streamID)) {
result.add(session);
}
}
return result;
}
private void handleNewConnection(Connection connection) {
final SOCKS5BytestreamServerSession session = new SOCKS5BytestreamServerSession(connection, registry);
onFinishedConnection = session.onFinished.connect(new Slot1<FileTransferError>() {
@Override
public void call(FileTransferError e) {
handleSessionFinished(session);
}
});
sessions.add(session);
session.start();
}
private void handleSessionFinished(SOCKS5BytestreamServerSession session) {
while(sessions.contains(session)) {
sessions.remove(session);
}
onFinishedConnection.disconnect();
}
}
|