mirror of
https://github.com/onionshare/onionshare.git
synced 2025-02-05 01:15:29 -05:00
Merge branch '685_use_dist_socks' of https://github.com/mig5/onionshare into mig5-685_use_dist_socks
This commit is contained in:
commit
2605967011
2
BUILD.md
2
BUILD.md
@ -11,7 +11,7 @@ cd onionshare
|
|||||||
|
|
||||||
Install the needed dependencies:
|
Install the needed dependencies:
|
||||||
|
|
||||||
For Debian-like distros: `apt install -y build-essential fakeroot python3-all python3-stdeb dh-python python3-flask python3-stem python3-pyqt5 python-nautilus python3-pytest tor obfs4proxy`
|
For Debian-like distros: `apt install -y build-essential fakeroot python3-all python3-stdeb dh-python python3-socks python3-flask python3-stem python3-pyqt5 python-nautilus python3-pytest tor obfs4proxy`
|
||||||
|
|
||||||
For Fedora-like distros: `dnf install -y rpm-build python3-flask python3-stem python3-qt5 python3-pytest nautilus-python tor obfs4 rpm-build`
|
For Fedora-like distros: `dnf install -y rpm-build python3-flask python3-stem python3-qt5 python3-pytest nautilus-python tor obfs4 rpm-build`
|
||||||
|
|
||||||
|
@ -23,7 +23,6 @@ from stem import ProtocolError, SocketClosed
|
|||||||
from stem.connection import MissingPassword, UnreadableCookieFile, AuthenticationFailure
|
from stem.connection import MissingPassword, UnreadableCookieFile, AuthenticationFailure
|
||||||
import os, sys, tempfile, shutil, urllib, platform, subprocess, time, shlex
|
import os, sys, tempfile, shutil, urllib, platform, subprocess, time, shlex
|
||||||
|
|
||||||
from . import socks
|
|
||||||
from . import common, strings
|
from . import common, strings
|
||||||
from .settings import Settings
|
from .settings import Settings
|
||||||
|
|
||||||
|
@ -1,530 +0,0 @@
|
|||||||
"""
|
|
||||||
SocksiPy - Python SOCKS module.
|
|
||||||
Version 1.5.0
|
|
||||||
|
|
||||||
Copyright 2006 Dan-Haim. All rights reserved.
|
|
||||||
|
|
||||||
Redistribution and use in source and binary forms, with or without modification,
|
|
||||||
are permitted provided that the following conditions are met:
|
|
||||||
1. Redistributions of source code must retain the above copyright notice, this
|
|
||||||
list of conditions and the following disclaimer.
|
|
||||||
2. Redistributions in binary form must reproduce the above copyright notice,
|
|
||||||
this list of conditions and the following disclaimer in the documentation
|
|
||||||
and/or other materials provided with the distribution.
|
|
||||||
3. Neither the name of Dan Haim nor the names of his contributors may be used
|
|
||||||
to endorse or promote products derived from this software without specific
|
|
||||||
prior written permission.
|
|
||||||
|
|
||||||
THIS SOFTWARE IS PROVIDED BY DAN HAIM "AS IS" AND ANY EXPRESS OR IMPLIED
|
|
||||||
WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
|
|
||||||
MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO
|
|
||||||
EVENT SHALL DAN HAIM OR HIS CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
|
|
||||||
INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
|
|
||||||
LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA
|
|
||||||
OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
|
|
||||||
LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT
|
|
||||||
OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
|
||||||
|
|
||||||
|
|
||||||
This module provides a standard socket-like interface for Python
|
|
||||||
for tunneling connections through SOCKS proxies.
|
|
||||||
|
|
||||||
===============================================================================
|
|
||||||
|
|
||||||
Minor modifications made by Christopher Gilbert (http://motomastyle.com/)
|
|
||||||
for use in PyLoris (http://pyloris.sourceforge.net/)
|
|
||||||
|
|
||||||
Minor modifications made by Mario Vilas (http://breakingcode.wordpress.com/)
|
|
||||||
mainly to merge bug fixes found in Sourceforge
|
|
||||||
|
|
||||||
Modifications made by Anorov (https://github.com/Anorov)
|
|
||||||
-Forked and renamed to PySocks
|
|
||||||
-Fixed issue with HTTP proxy failure checking (same bug that was in the old ___recvall() method)
|
|
||||||
-Included SocksiPyHandler (sockshandler.py), to be used as a urllib2 handler,
|
|
||||||
courtesy of e000 (https://github.com/e000): https://gist.github.com/869791#file_socksipyhandler.py
|
|
||||||
-Re-styled code to make it readable
|
|
||||||
-Aliased PROXY_TYPE_SOCKS5 -> SOCKS5 etc.
|
|
||||||
-Improved exception handling and output
|
|
||||||
-Removed irritating use of sequence indexes, replaced with tuple unpacked variables
|
|
||||||
-Fixed up Python 3 bytestring handling - chr(0x03).encode() -> b"\x03"
|
|
||||||
-Other general fixes
|
|
||||||
-Added clarification that the HTTP proxy connection method only supports CONNECT-style tunneling HTTP proxies
|
|
||||||
-Various small bug fixes
|
|
||||||
"""
|
|
||||||
|
|
||||||
__version__ = "1.5.0"
|
|
||||||
|
|
||||||
import socket
|
|
||||||
import struct
|
|
||||||
|
|
||||||
PROXY_TYPE_SOCKS4 = SOCKS4 = 1
|
|
||||||
PROXY_TYPE_SOCKS5 = SOCKS5 = 2
|
|
||||||
PROXY_TYPE_HTTP = HTTP = 3
|
|
||||||
|
|
||||||
PRINTABLE_PROXY_TYPES = {SOCKS4: "SOCKS4", SOCKS5: "SOCKS5", HTTP: "HTTP"}
|
|
||||||
|
|
||||||
_orgsocket = _orig_socket = socket.socket
|
|
||||||
|
|
||||||
|
|
||||||
class ProxyError(IOError):
|
|
||||||
"""
|
|
||||||
socket_err contains original socket.error exception.
|
|
||||||
"""
|
|
||||||
def __init__(self, msg, socket_err=None):
|
|
||||||
self.msg = msg
|
|
||||||
self.socket_err = socket_err
|
|
||||||
|
|
||||||
if socket_err:
|
|
||||||
self.msg = msg + ": {}".format(socket_err)
|
|
||||||
|
|
||||||
def __str__(self):
|
|
||||||
return self.msg
|
|
||||||
|
|
||||||
|
|
||||||
class GeneralProxyError(ProxyError):
|
|
||||||
pass
|
|
||||||
|
|
||||||
|
|
||||||
class ProxyConnectionError(ProxyError):
|
|
||||||
pass
|
|
||||||
|
|
||||||
|
|
||||||
class SOCKS5AuthError(ProxyError):
|
|
||||||
pass
|
|
||||||
|
|
||||||
|
|
||||||
class SOCKS5Error(ProxyError):
|
|
||||||
pass
|
|
||||||
|
|
||||||
|
|
||||||
class SOCKS4Error(ProxyError):
|
|
||||||
pass
|
|
||||||
|
|
||||||
|
|
||||||
class HTTPError(ProxyError):
|
|
||||||
pass
|
|
||||||
|
|
||||||
|
|
||||||
SOCKS4_ERRORS = {
|
|
||||||
0x5B: "Request rejected or failed",
|
|
||||||
0x5C: "Request rejected because SOCKS server cannot connect to identd on the client",
|
|
||||||
0x5D: "Request rejected because the client program and identd report different user-ids",
|
|
||||||
}
|
|
||||||
|
|
||||||
SOCKS5_ERRORS = {
|
|
||||||
0x01: "General SOCKS server failure",
|
|
||||||
0x02: "Connection not allowed by ruleset",
|
|
||||||
0x03: "Network unreachable",
|
|
||||||
0x04: "Host unreachable",
|
|
||||||
0x05: "Connection refused",
|
|
||||||
0x06: "TTL expired",
|
|
||||||
0x07: "Command not supported, or protocol error",
|
|
||||||
0x08: "Address type not supported",
|
|
||||||
}
|
|
||||||
|
|
||||||
DEFAULT_PORTS = {
|
|
||||||
SOCKS4: 1080,
|
|
||||||
SOCKS5: 1080,
|
|
||||||
HTTP: 8080,
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
def set_default_proxy(proxy_type=None, addr=None, port=None, rdns=True, username=None, password=None):
|
|
||||||
"""
|
|
||||||
set_default_proxy(proxy_type, addr[, port[, rdns[, username, password]]])
|
|
||||||
|
|
||||||
Sets a default proxy which all further socksocket objects will use,
|
|
||||||
unless explicitly changed.
|
|
||||||
"""
|
|
||||||
socksocket.default_proxy = (proxy_type, addr.encode(), port, rdns,
|
|
||||||
username.encode() if username else None,
|
|
||||||
password.encode() if password else None)
|
|
||||||
|
|
||||||
setdefaultproxy = set_default_proxy
|
|
||||||
|
|
||||||
|
|
||||||
def get_default_proxy():
|
|
||||||
"""
|
|
||||||
Returns the default proxy, set by set_default_proxy.
|
|
||||||
"""
|
|
||||||
return socksocket.default_proxy
|
|
||||||
|
|
||||||
getdefaultproxy = get_default_proxy
|
|
||||||
|
|
||||||
|
|
||||||
def wrap_module(module):
|
|
||||||
"""
|
|
||||||
Attempts to replace a module's socket library with a SOCKS socket. Must set
|
|
||||||
a default proxy using set_default_proxy(...) first.
|
|
||||||
This will only work on modules that import socket directly into the namespace;
|
|
||||||
most of the Python Standard Library falls into this category.
|
|
||||||
"""
|
|
||||||
if socksocket.default_proxy:
|
|
||||||
module.socket.socket = socksocket
|
|
||||||
else:
|
|
||||||
raise GeneralProxyError("No default proxy specified")
|
|
||||||
|
|
||||||
wrapmodule = wrap_module
|
|
||||||
|
|
||||||
|
|
||||||
def create_connection(dest_pair, proxy_type=None, proxy_addr=None,
|
|
||||||
proxy_port=None, proxy_username=None,
|
|
||||||
proxy_password=None, timeout=None):
|
|
||||||
"""create_connection(dest_pair, **proxy_args) -> socket object
|
|
||||||
|
|
||||||
Like socket.create_connection(), but connects to proxy
|
|
||||||
before returning the socket object.
|
|
||||||
|
|
||||||
dest_pair - 2-tuple of (IP/hostname, port).
|
|
||||||
**proxy_args - Same args passed to socksocket.set_proxy().
|
|
||||||
timeout - Optional socket timeout value, in seconds.
|
|
||||||
"""
|
|
||||||
sock = socksocket()
|
|
||||||
if isinstance(timeout, (int, float)):
|
|
||||||
sock.settimeout(timeout)
|
|
||||||
sock.set_proxy(proxy_type, proxy_addr, proxy_port,
|
|
||||||
proxy_username, proxy_password)
|
|
||||||
sock.connect(dest_pair)
|
|
||||||
return sock
|
|
||||||
|
|
||||||
|
|
||||||
class socksocket(socket.socket):
|
|
||||||
"""socksocket([family[, type[, proto]]]) -> socket object
|
|
||||||
|
|
||||||
Open a SOCKS enabled socket. The parameters are the same as
|
|
||||||
those of the standard socket init. In order for SOCKS to work,
|
|
||||||
you must specify family=AF_INET, type=SOCK_STREAM and proto=0.
|
|
||||||
"""
|
|
||||||
|
|
||||||
default_proxy = None
|
|
||||||
|
|
||||||
def __init__(self, family=socket.AF_INET, type=socket.SOCK_STREAM, proto=0, _sock=None):
|
|
||||||
_orig_socket.__init__(self, family, type, proto, _sock)
|
|
||||||
|
|
||||||
if self.default_proxy:
|
|
||||||
self.proxy = self.default_proxy
|
|
||||||
else:
|
|
||||||
self.proxy = (None, None, None, None, None, None)
|
|
||||||
self.proxy_sockname = None
|
|
||||||
self.proxy_peername = None
|
|
||||||
|
|
||||||
self.proxy_negotiators = {
|
|
||||||
SOCKS4: self._negotiate_SOCKS4,
|
|
||||||
SOCKS5: self._negotiate_SOCKS5,
|
|
||||||
HTTP: self._negotiate_HTTP,
|
|
||||||
}
|
|
||||||
|
|
||||||
def _recvall(self, count):
|
|
||||||
"""
|
|
||||||
Receive EXACTLY the number of bytes requested from the socket.
|
|
||||||
Blocks until the required number of bytes have been received.
|
|
||||||
"""
|
|
||||||
data = b""
|
|
||||||
while len(data) < count:
|
|
||||||
d = self.recv(count - len(data))
|
|
||||||
if not d:
|
|
||||||
raise GeneralProxyError("Connection closed unexpectedly")
|
|
||||||
data += d
|
|
||||||
return data
|
|
||||||
|
|
||||||
def set_proxy(self, proxy_type=None, addr=None, port=None, rdns=True, username=None, password=None):
|
|
||||||
"""set_proxy(proxy_type, addr[, port[, rdns[, username[, password]]]])
|
|
||||||
Sets the proxy to be used.
|
|
||||||
|
|
||||||
proxy_type - The type of the proxy to be used. Three types
|
|
||||||
are supported: PROXY_TYPE_SOCKS4 (including socks4a),
|
|
||||||
PROXY_TYPE_SOCKS5 and PROXY_TYPE_HTTP
|
|
||||||
addr - The address of the server (IP or DNS).
|
|
||||||
port - The port of the server. Defaults to 1080 for SOCKS
|
|
||||||
servers and 8080 for HTTP proxy servers.
|
|
||||||
rdns - Should DNS queries be performed on the remote side
|
|
||||||
(rather than the local side). The default is True.
|
|
||||||
Note: This has no effect with SOCKS4 servers.
|
|
||||||
username - Username to authenticate with to the server.
|
|
||||||
The default is no authentication.
|
|
||||||
password - Password to authenticate with to the server.
|
|
||||||
Only relevant when username is also provided.
|
|
||||||
"""
|
|
||||||
self.proxy = (proxy_type, addr.encode(), port, rdns,
|
|
||||||
username.encode() if username else None,
|
|
||||||
password.encode() if password else None)
|
|
||||||
|
|
||||||
setproxy = set_proxy
|
|
||||||
|
|
||||||
def get_proxy_sockname(self):
|
|
||||||
"""
|
|
||||||
Returns the bound IP address and port number at the proxy.
|
|
||||||
"""
|
|
||||||
return self.proxy_sockname
|
|
||||||
|
|
||||||
getproxysockname = get_proxy_sockname
|
|
||||||
|
|
||||||
def get_proxy_peername(self):
|
|
||||||
"""
|
|
||||||
Returns the IP and port number of the proxy.
|
|
||||||
"""
|
|
||||||
return _orig_socket.getpeername(self)
|
|
||||||
|
|
||||||
getproxypeername = get_proxy_peername
|
|
||||||
|
|
||||||
def get_peername(self):
|
|
||||||
"""
|
|
||||||
Returns the IP address and port number of the destination
|
|
||||||
machine (note: get_proxy_peername returns the proxy)
|
|
||||||
"""
|
|
||||||
return self.proxy_peername
|
|
||||||
|
|
||||||
getpeername = get_peername
|
|
||||||
|
|
||||||
def _negotiate_SOCKS5(self, dest_addr, dest_port):
|
|
||||||
"""
|
|
||||||
Negotiates a connection through a SOCKS5 server.
|
|
||||||
"""
|
|
||||||
proxy_type, addr, port, rdns, username, password = self.proxy
|
|
||||||
|
|
||||||
# First we'll send the authentication packages we support.
|
|
||||||
if username and password:
|
|
||||||
# The username/password details were supplied to the
|
|
||||||
# set_proxy method so we support the USERNAME/PASSWORD
|
|
||||||
# authentication (in addition to the standard none).
|
|
||||||
self.sendall(b"\x05\x02\x00\x02")
|
|
||||||
else:
|
|
||||||
# No username/password were entered, therefore we
|
|
||||||
# only support connections with no authentication.
|
|
||||||
self.sendall(b"\x05\x01\x00")
|
|
||||||
|
|
||||||
# We'll receive the server's response to determine which
|
|
||||||
# method was selected
|
|
||||||
chosen_auth = self._recvall(2)
|
|
||||||
|
|
||||||
if chosen_auth[0:1] != b"\x05":
|
|
||||||
# Note: string[i:i+1] is used because indexing of a bytestring
|
|
||||||
# via bytestring[i] yields an integer in Python 3
|
|
||||||
raise GeneralProxyError("SOCKS5 proxy server sent invalid data")
|
|
||||||
|
|
||||||
# Check the chosen authentication method
|
|
||||||
|
|
||||||
if chosen_auth[1:2] == b"\x02":
|
|
||||||
# Okay, we need to perform a basic username/password
|
|
||||||
# authentication.
|
|
||||||
self.sendall(b"\x01" + chr(len(username)).encode()
|
|
||||||
+ username
|
|
||||||
+ chr(len(password)).encode()
|
|
||||||
+ password)
|
|
||||||
auth_status = self._recvall(2)
|
|
||||||
if auth_status[0:1] != b"\x01":
|
|
||||||
# Bad response
|
|
||||||
raise GeneralProxyError("SOCKS5 proxy server sent invalid data")
|
|
||||||
if auth_status[1:2] != b"\x00":
|
|
||||||
# Authentication failed
|
|
||||||
raise SOCKS5AuthError("SOCKS5 authentication failed")
|
|
||||||
|
|
||||||
# Otherwise, authentication succeeded
|
|
||||||
|
|
||||||
# No authentication is required if 0x00
|
|
||||||
elif chosen_auth[1:2] != b"\x00":
|
|
||||||
# Reaching here is always bad
|
|
||||||
if chosen_auth[1:2] == b"\xFF":
|
|
||||||
raise SOCKS5AuthError("All offered SOCKS5 authentication methods were rejected")
|
|
||||||
else:
|
|
||||||
raise GeneralProxyError("SOCKS5 proxy server sent invalid data")
|
|
||||||
|
|
||||||
# Now we can request the actual connection
|
|
||||||
req = b"\x05\x01\x00"
|
|
||||||
# If the given destination address is an IP address, we'll
|
|
||||||
# use the IPv4 address request even if remote resolving was specified.
|
|
||||||
try:
|
|
||||||
addr_bytes = socket.inet_aton(dest_addr)
|
|
||||||
req += b"\x01" + addr_bytes
|
|
||||||
except socket.error:
|
|
||||||
# Well it's not an IP number, so it's probably a DNS name.
|
|
||||||
if rdns:
|
|
||||||
# Resolve remotely
|
|
||||||
addr_bytes = None
|
|
||||||
req += b"\x03" + chr(len(dest_addr)).encode() + dest_addr.encode()
|
|
||||||
else:
|
|
||||||
# Resolve locally
|
|
||||||
addr_bytes = socket.inet_aton(socket.gethostbyname(dest_addr))
|
|
||||||
req += b"\x01" + addr_bytes
|
|
||||||
|
|
||||||
req += struct.pack(">H", dest_port)
|
|
||||||
self.sendall(req)
|
|
||||||
|
|
||||||
# Get the response
|
|
||||||
resp = self._recvall(4)
|
|
||||||
if resp[0:1] != b"\x05":
|
|
||||||
raise GeneralProxyError("SOCKS5 proxy server sent invalid data")
|
|
||||||
|
|
||||||
status = ord(resp[1:2])
|
|
||||||
if status != 0x00:
|
|
||||||
# Connection failed: server returned an error
|
|
||||||
error = SOCKS5_ERRORS.get(status, "Unknown error")
|
|
||||||
raise SOCKS5Error("{:#04x}: {}".format(status, error))
|
|
||||||
|
|
||||||
# Get the bound address/port
|
|
||||||
if resp[3:4] == b"\x01":
|
|
||||||
bound_addr = self._recvall(4)
|
|
||||||
elif resp[3:4] == b"\x03":
|
|
||||||
resp += self.recv(1)
|
|
||||||
bound_addr = self._recvall(ord(resp[4:5]))
|
|
||||||
else:
|
|
||||||
raise GeneralProxyError("SOCKS5 proxy server sent invalid data")
|
|
||||||
|
|
||||||
bound_port = struct.unpack(">H", self._recvall(2))[0]
|
|
||||||
self.proxy_sockname = bound_addr, bound_port
|
|
||||||
if addr_bytes:
|
|
||||||
self.proxy_peername = socket.inet_ntoa(addr_bytes), dest_port
|
|
||||||
else:
|
|
||||||
self.proxy_peername = dest_addr, dest_port
|
|
||||||
|
|
||||||
def _negotiate_SOCKS4(self, dest_addr, dest_port):
|
|
||||||
"""
|
|
||||||
Negotiates a connection through a SOCKS4 server.
|
|
||||||
"""
|
|
||||||
proxy_type, addr, port, rdns, username, password = self.proxy
|
|
||||||
|
|
||||||
# Check if the destination address provided is an IP address
|
|
||||||
remote_resolve = False
|
|
||||||
try:
|
|
||||||
addr_bytes = socket.inet_aton(dest_addr)
|
|
||||||
except socket.error:
|
|
||||||
# It's a DNS name. Check where it should be resolved.
|
|
||||||
if rdns:
|
|
||||||
addr_bytes = b"\x00\x00\x00\x01"
|
|
||||||
remote_resolve = True
|
|
||||||
else:
|
|
||||||
addr_bytes = socket.inet_aton(socket.gethostbyname(dest_addr))
|
|
||||||
|
|
||||||
# Construct the request packet
|
|
||||||
req = struct.pack(">BBH", 0x04, 0x01, dest_port) + addr_bytes
|
|
||||||
|
|
||||||
# The username parameter is considered userid for SOCKS4
|
|
||||||
if username:
|
|
||||||
req += username
|
|
||||||
req += b"\x00"
|
|
||||||
|
|
||||||
# DNS name if remote resolving is required
|
|
||||||
# NOTE: This is actually an extension to the SOCKS4 protocol
|
|
||||||
# called SOCKS4A and may not be supported in all cases.
|
|
||||||
if remote_resolve:
|
|
||||||
req += dest_addr.encode() + b"\x00"
|
|
||||||
self.sendall(req)
|
|
||||||
|
|
||||||
# Get the response from the server
|
|
||||||
resp = self._recvall(8)
|
|
||||||
if resp[0:1] != b"\x00":
|
|
||||||
# Bad data
|
|
||||||
raise GeneralProxyError("SOCKS4 proxy server sent invalid data")
|
|
||||||
|
|
||||||
status = ord(resp[1:2])
|
|
||||||
if status != 0x5A:
|
|
||||||
# Connection failed: server returned an error
|
|
||||||
error = SOCKS4_ERRORS.get(status, "Unknown error")
|
|
||||||
raise SOCKS4Error("{:#04x}: {}".format(status, error))
|
|
||||||
|
|
||||||
# Get the bound address/port
|
|
||||||
self.proxy_sockname = (socket.inet_ntoa(resp[4:]), struct.unpack(">H", resp[2:4])[0])
|
|
||||||
if remote_resolve:
|
|
||||||
self.proxy_peername = socket.inet_ntoa(addr_bytes), dest_port
|
|
||||||
else:
|
|
||||||
self.proxy_peername = dest_addr, dest_port
|
|
||||||
|
|
||||||
def _negotiate_HTTP(self, dest_addr, dest_port):
|
|
||||||
"""
|
|
||||||
Negotiates a connection through an HTTP server.
|
|
||||||
NOTE: This currently only supports HTTP CONNECT-style proxies.
|
|
||||||
"""
|
|
||||||
proxy_type, addr, port, rdns, username, password = self.proxy
|
|
||||||
|
|
||||||
# If we need to resolve locally, we do this now
|
|
||||||
addr = dest_addr if rdns else socket.gethostbyname(dest_addr)
|
|
||||||
|
|
||||||
self.sendall(b"CONNECT " + addr.encode() + b":" + str(dest_port).encode() +
|
|
||||||
b" HTTP/1.1\r\n" + b"Host: " + dest_addr.encode() + b"\r\n\r\n")
|
|
||||||
|
|
||||||
# We just need the first line to check if the connection was successful
|
|
||||||
fobj = self.makefile()
|
|
||||||
status_line = fobj.readline()
|
|
||||||
fobj.close()
|
|
||||||
|
|
||||||
if not status_line:
|
|
||||||
raise GeneralProxyError("Connection closed unexpectedly")
|
|
||||||
|
|
||||||
try:
|
|
||||||
proto, status_code, status_msg = status_line.split(" ", 2)
|
|
||||||
except ValueError:
|
|
||||||
raise GeneralProxyError("HTTP proxy server sent invalid response")
|
|
||||||
|
|
||||||
if not proto.startswith("HTTP/"):
|
|
||||||
raise GeneralProxyError("Proxy server does not appear to be an HTTP proxy")
|
|
||||||
|
|
||||||
try:
|
|
||||||
status_code = int(status_code)
|
|
||||||
except ValueError:
|
|
||||||
raise HTTPError("HTTP proxy server did not return a valid HTTP status")
|
|
||||||
|
|
||||||
if status_code != 200:
|
|
||||||
error = "{}: {}".format(status_code, status_msg)
|
|
||||||
if status_code in (400, 403, 405):
|
|
||||||
# It's likely that the HTTP proxy server does not support the CONNECT tunneling method
|
|
||||||
error += ("\n[*] Note: The HTTP proxy server may not be supported by PySocks"
|
|
||||||
" (must be a CONNECT tunnel proxy)")
|
|
||||||
raise HTTPError(error)
|
|
||||||
|
|
||||||
self.proxy_sockname = (b"0.0.0.0", 0)
|
|
||||||
self.proxy_peername = addr, dest_port
|
|
||||||
|
|
||||||
def connect(self, dest_pair):
|
|
||||||
"""
|
|
||||||
Connects to the specified destination through a proxy.
|
|
||||||
Uses the same API as socket's connect().
|
|
||||||
To select the proxy server, use set_proxy().
|
|
||||||
|
|
||||||
dest_pair - 2-tuple of (IP/hostname, port).
|
|
||||||
"""
|
|
||||||
proxy_type, proxy_addr, proxy_port, rdns, username, password = self.proxy
|
|
||||||
dest_addr, dest_port = dest_pair
|
|
||||||
|
|
||||||
# Do a minimal input check first
|
|
||||||
if (not isinstance(dest_pair, (list, tuple))
|
|
||||||
or len(dest_pair) != 2
|
|
||||||
or not isinstance(dest_addr, type(""))
|
|
||||||
or not isinstance(dest_port, int)):
|
|
||||||
raise GeneralProxyError("Invalid destination-connection (host, port) pair")
|
|
||||||
|
|
||||||
if proxy_type is None:
|
|
||||||
# Treat like regular socket object
|
|
||||||
_orig_socket.connect(self, (dest_addr, dest_port))
|
|
||||||
return
|
|
||||||
|
|
||||||
proxy_port = proxy_port or DEFAULT_PORTS.get(proxy_type)
|
|
||||||
if not proxy_port:
|
|
||||||
raise GeneralProxyError("Invalid proxy type")
|
|
||||||
|
|
||||||
try:
|
|
||||||
# Initial connection to proxy server
|
|
||||||
_orig_socket.connect(self, (proxy_addr, proxy_port))
|
|
||||||
|
|
||||||
except socket.error as error:
|
|
||||||
# Error while connecting to proxy
|
|
||||||
self.close()
|
|
||||||
proxy_server = "{}:{}".format(proxy_addr.decode(), proxy_port)
|
|
||||||
printable_type = PRINTABLE_PROXY_TYPES[proxy_type]
|
|
||||||
|
|
||||||
msg = "Error connecting to {} proxy {}".format(printable_type,
|
|
||||||
proxy_server)
|
|
||||||
raise ProxyConnectionError(msg, error)
|
|
||||||
|
|
||||||
else:
|
|
||||||
# Connected to proxy server, now negotiate
|
|
||||||
try:
|
|
||||||
# Calls negotiate_{SOCKS4, SOCKS5, HTTP}
|
|
||||||
self.proxy_negotiators[proxy_type](dest_addr, dest_port)
|
|
||||||
except socket.error as error:
|
|
||||||
# Wrap socket errors
|
|
||||||
self.close()
|
|
||||||
raise GeneralProxyError("Socket error", error)
|
|
||||||
except ProxyError:
|
|
||||||
# Protocol error while negotiating with proxy
|
|
||||||
self.close()
|
|
||||||
raise
|
|
@ -19,9 +19,9 @@ along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|||||||
"""
|
"""
|
||||||
from PyQt5 import QtCore
|
from PyQt5 import QtCore
|
||||||
import datetime, time, socket, re, platform
|
import datetime, time, socket, re, platform
|
||||||
|
import socks
|
||||||
from distutils.version import LooseVersion as Version
|
from distutils.version import LooseVersion as Version
|
||||||
|
|
||||||
from onionshare import socks
|
|
||||||
from onionshare.settings import Settings
|
from onionshare.settings import Settings
|
||||||
from onionshare.onion import Onion
|
from onionshare.onion import Onion
|
||||||
|
|
||||||
|
Loading…
x
Reference in New Issue
Block a user