1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20 package org.apache.mina.example.proxy;
21
22 import java.net.InetSocketAddress;
23 import java.net.URL;
24 import java.security.Security;
25 import java.util.ArrayList;
26 import java.util.HashMap;
27 import java.util.List;
28
29 import org.apache.mina.core.RuntimeIoException;
30 import org.apache.mina.core.future.ConnectFuture;
31 import org.apache.mina.core.session.IdleStatus;
32 import org.apache.mina.core.session.IoSession;
33 import org.apache.mina.filter.logging.LoggingFilter;
34 import org.apache.mina.proxy.ProxyConnector;
35 import org.apache.mina.proxy.handlers.ProxyRequest;
36 import org.apache.mina.proxy.handlers.http.HttpAuthenticationMethods;
37 import org.apache.mina.proxy.handlers.http.HttpProxyConstants;
38 import org.apache.mina.proxy.handlers.http.HttpProxyRequest;
39 import org.apache.mina.proxy.handlers.socks.SocksProxyConstants;
40 import org.apache.mina.proxy.handlers.socks.SocksProxyRequest;
41 import org.apache.mina.proxy.session.ProxyIoSession;
42 import org.apache.mina.proxy.utils.MD4Provider;
43 import org.apache.mina.transport.socket.nio.NioSocketConnector;
44
45
46
47
48
49
50
51
52 public class ProxyTestClient {
53
54
55
56
57
58
59
60
61 public final static String USER = "TED_KODS";
62
63
64
65
66 public final static String PWD = "EDOUARD";
67
68
69
70
71 public final static String DOMAIN = "MYDOMAIN";
72
73
74
75
76 public final static String WORKSTATION = "MYWORKSTATION";
77
78
79
80
81 private final static boolean USE_HTTP_1_1 = false;
82
83
84
85
86 static {
87 if (Security.getProvider("MINA") == null) {
88 Security.addProvider(new MD4Provider());
89 }
90 }
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110 public ProxyTestClient(String[] args) throws Exception {
111 if (args.length < 3) {
112 System.out
113 .println(ProxyTestClient.class.getName()
114 + " <proxy-hostname> <proxy-port> <url> <proxy-method> (<proxy-method> is OPTIONNAL)");
115 return;
116 }
117
118
119 NioSocketConnector socketConnector = new NioSocketConnector(Runtime
120 .getRuntime().availableProcessors() + 1);
121
122 ProxyConnector connector = new ProxyConnector(socketConnector);
123
124
125 connector.setConnectTimeoutMillis(5000);
126
127 URL url = new URL(args[2]);
128 int port = url.getPort() == -1 ? url.getDefaultPort() : url.getPort();
129
130 ProxyRequest req = null;
131
132 if (args.length == 4) {
133 if ("SOCKS4".equals(args[3])) {
134 req = new SocksProxyRequest(
135 SocksProxyConstants.SOCKS_VERSION_4,
136 SocksProxyConstants.ESTABLISH_TCPIP_STREAM,
137 new InetSocketAddress(url.getHost(), port), USER);
138 } else if ("SOCKS4a".equals(args[3])) {
139 req = new SocksProxyRequest(
140 SocksProxyConstants.ESTABLISH_TCPIP_STREAM, url
141 .getHost(), port, USER);
142 } else if ("SOCKS5".equals(args[3])) {
143 req = new SocksProxyRequest(
144 SocksProxyConstants.SOCKS_VERSION_5,
145 SocksProxyConstants.ESTABLISH_TCPIP_STREAM,
146 new InetSocketAddress(url.getHost(), port), USER);
147 ((SocksProxyRequest) req).setPassword(PWD);
148 ((SocksProxyRequest) req)
149 .setServiceKerberosName(Socks5GSSAPITestServer.SERVICE_NAME);
150 } else {
151 req = createHttpProxyRequest(args[2]);
152 }
153 } else {
154 req = createHttpProxyRequest(args[2]);
155 }
156
157 ProxyIoSession proxyIoSession = new ProxyIoSession(
158 new InetSocketAddress(args[0], Integer.parseInt(args[1])), req);
159
160
161
162 List<HttpAuthenticationMethods> l = new ArrayList<HttpAuthenticationMethods>();
163 l.add(HttpAuthenticationMethods.DIGEST);
164 l.add(HttpAuthenticationMethods.BASIC);
165 proxyIoSession.setPreferedOrder(l);
166
167 connector.setProxyIoSession(proxyIoSession);
168
169 socketConnector.getSessionConfig().setIdleTime(IdleStatus.BOTH_IDLE, 5);
170
171 connector.getFilterChain().addLast("logger", new LoggingFilter());
172
173
174 String cmd = "GET " + url.toExternalForm() + " HTTP/1.0"
175 + HttpProxyConstants.CRLF + HttpProxyConstants.CRLF;
176
177 connector.setHandler(new ClientSessionHandler(cmd));
178
179 IoSession session;
180 for (;;) {
181 try {
182 ConnectFuture future = connector.connect();
183 future.awaitUninterruptibly();
184 session = future.getSession();
185 break;
186 } catch (RuntimeIoException e) {
187 System.err.println("Failed to connect. Retrying in 5 secs ...");
188 Thread.sleep(5000);
189 }
190 }
191
192
193 if (session != null) {
194 session.getCloseFuture().awaitUninterruptibly();
195 }
196 connector.dispose();
197 System.exit(0);
198 }
199
200
201
202
203
204
205
206
207
208
209 private HttpProxyRequest createHttpProxyRequest(String uri) {
210 HttpProxyRequest req = new HttpProxyRequest(uri);
211 HashMap<String, String> props = new HashMap<String, String>();
212 props.put(HttpProxyConstants.USER_PROPERTY, USER);
213 props.put(HttpProxyConstants.PWD_PROPERTY, PWD);
214 props.put(HttpProxyConstants.DOMAIN_PROPERTY, DOMAIN);
215 props.put(HttpProxyConstants.WORKSTATION_PROPERTY, WORKSTATION);
216
217 req.setProperties(props);
218 if (USE_HTTP_1_1) {
219 req.setHttpVersion(HttpProxyConstants.HTTP_1_1);
220 }
221
222 return req;
223 }
224
225
226
227
228 public static void main(String[] args) throws Exception {
229 new ProxyTestClient(args);
230 }
231 }