tecnovert
7 months ago
parent
ee2f462ee9
commit
dc4f0ac2d3
8 changed files with 23 additions and 844 deletions
@ -1,258 +0,0 @@ |
||||
# -*- coding: utf-8 -*- |
||||
|
||||
import os |
||||
import json |
||||
import socks |
||||
import time |
||||
import urllib |
||||
import hashlib |
||||
from xmlrpc.client import ( |
||||
Fault, |
||||
Transport, |
||||
SafeTransport, |
||||
) |
||||
from sockshandler import SocksiPyConnection |
||||
from .util import jsonDecimal |
||||
|
||||
|
||||
class SocksTransport(Transport): |
||||
|
||||
def set_proxy(self, proxy_host, proxy_port): |
||||
self.proxy_host = proxy_host |
||||
self.proxy_port = proxy_port |
||||
|
||||
self.proxy_type = socks.PROXY_TYPE_SOCKS5 |
||||
self.proxy_rdns = True |
||||
self.proxy_username = None |
||||
self.proxy_password = None |
||||
|
||||
def make_connection(self, host): |
||||
# return an existing connection if possible. This allows |
||||
# HTTP/1.1 keep-alive. |
||||
if self._connection and host == self._connection[0]: |
||||
return self._connection[1] |
||||
# create a HTTP connection object from a host descriptor |
||||
chost, self._extra_headers, x509 = self.get_host_info(host) |
||||
self._connection = host, SocksiPyConnection(self.proxy_type, self.proxy_host, self.proxy_port, self.proxy_rdns, self.proxy_username, self.proxy_password, chost) |
||||
return self._connection[1] |
||||
|
||||
|
||||
class JsonrpcDigest(): |
||||
# __getattr__ complicates extending ServerProxy |
||||
def __init__(self, uri, transport=None, encoding=None, verbose=False, |
||||
allow_none=False, use_datetime=False, use_builtin_types=False, |
||||
*, context=None): |
||||
|
||||
parsed = urllib.parse.urlparse(uri) |
||||
if parsed.scheme not in ('http', 'https'): |
||||
raise OSError('unsupported XML-RPC protocol') |
||||
self.__host = parsed.netloc |
||||
self.__handler = parsed.path |
||||
|
||||
if transport is None: |
||||
handler = SafeTransport if parsed.scheme == 'https' else Transport |
||||
extra_kwargs = {} |
||||
transport = handler(use_datetime=use_datetime, |
||||
use_builtin_types=use_builtin_types, |
||||
**extra_kwargs) |
||||
self.__transport = transport |
||||
|
||||
self.__encoding = encoding or 'utf-8' |
||||
self.__verbose = verbose |
||||
self.__allow_none = allow_none |
||||
|
||||
self.__request_id = 0 |
||||
|
||||
def close(self): |
||||
if self.__transport is not None: |
||||
self.__transport.close() |
||||
|
||||
def request_id(self): |
||||
return self.__request_id |
||||
|
||||
def post_request(self, method, params, timeout=None): |
||||
try: |
||||
connection = self.__transport.make_connection(self.__host) |
||||
if timeout: |
||||
connection.timeout = timeout |
||||
headers = self.__transport._extra_headers[:] |
||||
|
||||
connection.putrequest('POST', self.__handler) |
||||
headers.append(('Content-Type', 'application/json')) |
||||
headers.append(('User-Agent', 'jsonrpc')) |
||||
self.__transport.send_headers(connection, headers) |
||||
self.__transport.send_content(connection, '' if params is None else json.dumps(params, default=jsonDecimal).encode('utf-8')) |
||||
self.__request_id += 1 |
||||
|
||||
resp = connection.getresponse() |
||||
return resp.read() |
||||
|
||||
except Fault: |
||||
raise |
||||
except Exception: |
||||
self.__transport.close() |
||||
raise |
||||
|
||||
def json_request(self, request_body, username='', password='', timeout=None): |
||||
try: |
||||
connection = self.__transport.make_connection(self.__host) |
||||
if timeout: |
||||
connection.timeout = timeout |
||||
|
||||
headers = self.__transport._extra_headers[:] |
||||
|
||||
connection.putrequest('POST', self.__handler) |
||||
headers.append(('Content-Type', 'application/json')) |
||||
headers.append(('Connection', 'keep-alive')) |
||||
self.__transport.send_headers(connection, headers) |
||||
self.__transport.send_content(connection, json.dumps(request_body, default=jsonDecimal).encode('utf-8') if request_body else '') |
||||
resp = connection.getresponse() |
||||
|
||||
if resp.status == 401: |
||||
resp_headers = resp.getheaders() |
||||
v = resp.read() |
||||
|
||||
algorithm = '' |
||||
realm = '' |
||||
nonce = '' |
||||
for h in resp_headers: |
||||
if h[0] != 'WWW-authenticate': |
||||
continue |
||||
fields = h[1].split(',') |
||||
for f in fields: |
||||
key, value = f.split('=', 1) |
||||
if key == 'algorithm' and value != 'MD5': |
||||
break |
||||
if key == 'realm': |
||||
realm = value.strip('"') |
||||
if key == 'nonce': |
||||
nonce = value.strip('"') |
||||
if realm != '' and nonce != '': |
||||
break |
||||
|
||||
if realm == '' or nonce == '': |
||||
raise ValueError('Authenticate header not found.') |
||||
|
||||
path = self.__handler |
||||
HA1 = hashlib.md5(f'{username}:{realm}:{password}'.encode('utf-8')).hexdigest() |
||||
|
||||
http_method = 'POST' |
||||
HA2 = hashlib.md5(f'{http_method}:{path}'.encode('utf-8')).hexdigest() |
||||
|
||||
ncvalue = '{:08x}'.format(1) |
||||
s = ncvalue.encode('utf-8') |
||||
s += nonce.encode('utf-8') |
||||
s += time.ctime().encode('utf-8') |
||||
s += os.urandom(8) |
||||
cnonce = (hashlib.sha1(s).hexdigest()[:16]) |
||||
|
||||
# MD5-SESS |
||||
HA1 = hashlib.md5(f'{HA1}:{nonce}:{cnonce}'.encode('utf-8')).hexdigest() |
||||
|
||||
respdig = hashlib.md5(f'{HA1}:{nonce}:{ncvalue}:{cnonce}:auth:{HA2}'.encode('utf-8')).hexdigest() |
||||
|
||||
header_value = f'Digest username="{username}", realm="{realm}", nonce="{nonce}", uri="{path}", response="{respdig}", algorithm="MD5-sess", qop="auth", nc={ncvalue}, cnonce="{cnonce}"' |
||||
headers = self.__transport._extra_headers[:] |
||||
headers.append(('Authorization', header_value)) |
||||
|
||||
connection.putrequest('POST', self.__handler) |
||||
headers.append(('Content-Type', 'application/json')) |
||||
headers.append(('Connection', 'keep-alive')) |
||||
self.__transport.send_headers(connection, headers) |
||||
self.__transport.send_content(connection, json.dumps(request_body, default=jsonDecimal).encode('utf-8') if request_body else '') |
||||
resp = connection.getresponse() |
||||
|
||||
self.__request_id += 1 |
||||
return resp.read() |
||||
|
||||
except Fault: |
||||
raise |
||||
except Exception: |
||||
self.__transport.close() |
||||
raise |
||||
|
||||
|
||||
def callrpc_wow(rpc_port, method, params=[], rpc_host='127.0.0.1', path='json_rpc', auth=None, timeout=120, transport=None, tag=''): |
||||
# auth is a tuple: (username, password) |
||||
try: |
||||
if rpc_host.count('://') > 0: |
||||
url = '{}:{}/{}'.format(rpc_host, rpc_port, path) |
||||
else: |
||||
url = 'http://{}:{}/{}'.format(rpc_host, rpc_port, path) |
||||
|
||||
x = JsonrpcDigest(url, transport=transport) |
||||
request_body = { |
||||
'method': method, |
||||
'params': params, |
||||
'jsonrpc': '2.0', |
||||
'id': x.request_id() |
||||
} |
||||
if auth: |
||||
v = x.json_request(request_body, username=auth[0], password=auth[1], timeout=timeout) |
||||
else: |
||||
v = x.json_request(request_body, timeout=timeout) |
||||
x.close() |
||||
r = json.loads(v.decode('utf-8')) |
||||
except Exception as ex: |
||||
raise ValueError('{}RPC Server Error: {}'.format(tag, str(ex))) |
||||
|
||||
if 'error' in r and r['error'] is not None: |
||||
raise ValueError(tag + 'RPC error ' + str(r['error'])) |
||||
|
||||
return r['result'] |
||||
|
||||
|
||||
def callrpc_wow2(rpc_port: int, method: str, params=None, auth=None, rpc_host='127.0.0.1', timeout=120, transport=None, tag=''): |
||||
try: |
||||
if rpc_host.count('://') > 0: |
||||
url = '{}:{}/{}'.format(rpc_host, rpc_port, method) |
||||
else: |
||||
url = 'http://{}:{}/{}'.format(rpc_host, rpc_port, method) |
||||
|
||||
x = JsonrpcDigest(url, transport=transport) |
||||
if auth: |
||||
v = x.json_request(params, username=auth[0], password=auth[1], timeout=timeout) |
||||
else: |
||||
v = x.json_request(params, timeout=timeout) |
||||
x.close() |
||||
r = json.loads(v.decode('utf-8')) |
||||
except Exception as ex: |
||||
raise ValueError('{}RPC Server Error: {}'.format(tag, str(ex))) |
||||
|
||||
return r |
||||
|
||||
|
||||
def make_wow_rpc2_func(port, auth, host='127.0.0.1', proxy_host=None, proxy_port=None, default_timeout=120, tag=''): |
||||
port = port |
||||
auth = auth |
||||
host = host |
||||
transport = None |
||||
default_timeout = default_timeout |
||||
tag = tag |
||||
|
||||
if proxy_host: |
||||
transport = SocksTransport() |
||||
transport.set_proxy(proxy_host, proxy_port) |
||||
|
||||
def rpc_func(method, params=None, wallet=None, timeout=default_timeout): |
||||
nonlocal port, auth, host, transport, tag |
||||
return callrpc_wow2(port, method, params, auth=auth, rpc_host=host, timeout=timeout, transport=transport, tag=tag) |
||||
return rpc_func |
||||
|
||||
|
||||
def make_wow_rpc_func(port, auth, host='127.0.0.1', proxy_host=None, proxy_port=None, default_timeout=120, tag=''): |
||||
port = port |
||||
auth = auth |
||||
host = host |
||||
transport = None |
||||
default_timeout = default_timeout |
||||
tag = tag |
||||
|
||||
if proxy_host: |
||||
transport = SocksTransport() |
||||
transport.set_proxy(proxy_host, proxy_port) |
||||
|
||||
def rpc_func(method, params=None, wallet=None, timeout=default_timeout): |
||||
nonlocal port, auth, host, transport, tag |
||||
return callrpc_wow(port, method, params, rpc_host=host, auth=auth, timeout=timeout, transport=transport, tag=tag) |
||||
return rpc_func |
||||
Loading…
Reference in new issue