103 lines
3.2 KiB
Python
103 lines
3.2 KiB
Python
import datetime
|
|
from uuid import uuid4
|
|
from xmlrpc.server import SimpleXMLRPCServer
|
|
from xmlrpc.client import ServerProxy
|
|
import logging
|
|
import os
|
|
import os.path
|
|
from requests.graphql import get_catalog
|
|
from pygost import gost34112012256
|
|
import xml.etree.ElementTree as ET
|
|
from requests.request_xml_service import RequestXmlService
|
|
import zipfile
|
|
|
|
PASSWORD = 'gost_2012$a742ec53198ec2a5027086fba8814a89982a57112d1a72d02260161108f39b50'
|
|
|
|
|
|
# Expose a function
|
|
def list_contents(dir_name):
|
|
logging.debug('list_contents(%s)', dir_name)
|
|
return os.listdir(dir_name)
|
|
|
|
|
|
def aud_add(message):
|
|
logging.debug(message)
|
|
return 'OK'
|
|
|
|
|
|
def auth_response(challenge, server_id, is_server):
|
|
# logging.debug(f'Challenge: {challenge}, Server: {server_id}, IsServer: {is_server}')
|
|
msg = '%s%s%s' % (challenge, server_id, PASSWORD)
|
|
response = gost34112012256.new(msg.encode('utf-8')[::-1]).digest()[::-1].hex()
|
|
return {'error': False, 'response': response}
|
|
|
|
|
|
def auth_challenge():
|
|
# logging.debug('get challenge')
|
|
return uuid4().hex
|
|
|
|
|
|
def accept(params, files, url):
|
|
print(params, files, url)
|
|
print('Accept')
|
|
date = datetime.datetime.now()
|
|
if params['query_type'] == 4:
|
|
rxmls = RequestXmlService()
|
|
req = ET.fromstring(params['query_data'])
|
|
req_id = rxmls.get_request_uuid(req)
|
|
res_id = uuid4().hex
|
|
res = rxmls.get_request_document(res_id, req_id)
|
|
rxmls.set_result(res, 0, '')
|
|
response_params = {
|
|
'from': params['to'],
|
|
'to': params['from'],
|
|
'ts_added': date.timestamp(),
|
|
'user_id': '1',
|
|
'user_id_to': '3302',
|
|
'query_type': 1004,
|
|
'query_data': ET.tostring(res, encoding='unicode', xml_declaration=True)
|
|
}
|
|
catalog = get_catalog()
|
|
filename = uuid4().hex
|
|
filepath = '/tmp/' + filename
|
|
zipf = zipfile.ZipFile(filepath, "w")
|
|
zipf.writestr('WF.CLL', catalog)
|
|
zipf.close()
|
|
response_files = [{'name': filename, 'url': filepath, 'size': os.path.getsize(filepath)}]
|
|
proxy = ServerProxy(url)
|
|
proxy.send(response_params, response_files, 'http://10.10.8.27:9000/')
|
|
return True
|
|
|
|
|
|
def onSent(params, files, callback_url):
|
|
print('onSent')
|
|
|
|
|
|
def main():
|
|
print('Use Control-C to exit')
|
|
logging.basicConfig(level=logging.DEBUG)
|
|
server = SimpleXMLRPCServer(('0.0.0.0', 9000), logRequests=True, allow_none=True)
|
|
server.register_function(list_contents)
|
|
server.register_function(aud_add)
|
|
server.register_function(auth_response)
|
|
server.register_function(auth_challenge)
|
|
server.register_function(accept)
|
|
server.register_function(onSent)
|
|
|
|
try:
|
|
server.serve_forever()
|
|
except KeyboardInterrupt:
|
|
print('Exiting')
|
|
|
|
|
|
def test():
|
|
params = {'from': 'tcp://kptsp_vb', 'query_data': '<?xml version="1.0" encoding="utf-8"?><request><header parcel_id="1665c872f24e46589f36d264a75a72dd"/><getCatalog from="1675776233.21" withRef="NO"/></request>', 'query_type': 4, 'to': 'tcp://bnd127', 'user_id': '1', 'ts_added': 1679213998.282467}
|
|
files = []
|
|
url = 'http://127.0.0.1:7000/xmlrpc'
|
|
accept(params, files, url)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|
|
# test()
|