X-Git-Url: https://git.hedgecode.org/?p=chesshog-scanner.git;a=blobdiff_plain;f=src%2Fmain%2Fjava%2Forg%2Fhedgecode%2Fchess%2Fscanner%2Fproxy%2Fclient%2FSocksSocketFactory.java;fp=src%2Fmain%2Fjava%2Forg%2Fhedgecode%2Fchess%2Fscanner%2Fproxy%2Fclient%2FSocksSocketFactory.java;h=c3a679a872a9c70f71d755829503c25316301d1a;hp=0000000000000000000000000000000000000000;hb=70d45ca74e19c98d0bef2a081f77fc1e3543691e;hpb=6e3a8590a26312b6cea579777db885107cae88df

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
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
+        );
+    }
+
+}