[LIB-13] Several options for working through a proxy server

[chesshog-scanner.git] / src / main / java / org / hedgecode / chess / scanner / proxy / client / SocksSocketFactory.java

diff --git a/src/main/java/org/hedgecode/chess/scanner/proxy/client/SocksSocketFactory.java b/src/main/java/org/hedgecode/chess/scanner/proxy/client/SocksSocketFactory.java
new file mode 100644 (file)
index 0000000..c3a679a
--- /dev/null
+++ b/src/main/java/org/hedgecode/chess/scanner/proxy/client/SocksSocketFactory.java
@@ -0,0 +1,60 @@
+/*
+ * Copyright (c) 2019-2020. Developed by Hedgecode.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *   http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package org.hedgecode.chess.scanner.proxy.client;
+
+import java.io.IOException;
+import java.net.InetSocketAddress;
+import java.net.Proxy;
+import java.net.Socket;
+
+import org.apache.http.HttpHost;
+import org.apache.http.conn.socket.PlainConnectionSocketFactory;
+import org.apache.http.protocol.HttpContext;
+
+import org.hedgecode.chess.scanner.ScannerConstants;
+
+/**
+ * SocksSocketFactory
+ *
+ * @author Dmitry Samoshin aka gotty
+ */
+public class SocksSocketFactory extends PlainConnectionSocketFactory {
+
+    @Override
+    public Socket createSocket(final HttpContext context) throws IOException {
+        Proxy proxy = new Proxy(
+                Proxy.Type.SOCKS,
+                (InetSocketAddress) context.getAttribute(ScannerConstants.PROXY_SOCKS_ADDRESS)
+        );
+        return new Socket(proxy);
+    }
+
+    @Override
+    public Socket connectSocket(
+            int connectTimeout, Socket socket, HttpHost host,
+            InetSocketAddress remoteAddress, InetSocketAddress localAddress,
+            HttpContext context
+    ) throws IOException {
+        InetSocketAddress unresolvedAddress = InetSocketAddress.createUnresolved(
+                host.getHostName(), remoteAddress.getPort()
+        );
+        return super.connectSocket(
+                connectTimeout, socket, host, unresolvedAddress, localAddress, context
+        );
+    }
+
+}