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