mirror of https://gitlab.com/bashrc2/epicyon
95 lines
3.4 KiB
Python
95 lines
3.4 KiB
Python
__filename__ = "session.py"
|
|
__author__ = "Bob Mottram"
|
|
__license__ = "AGPL3+"
|
|
__version__ = "0.0.1"
|
|
__maintainer__ = "Bob Mottram"
|
|
__email__ = "bob@freedombone.net"
|
|
__status__ = "Production"
|
|
|
|
import os
|
|
import requests
|
|
from utils import urlPermitted
|
|
import json
|
|
|
|
baseDirectory=None
|
|
|
|
def createSession(domain: str, port: int, onionRoute: bool):
|
|
session = requests.session()
|
|
#if domain.startswith('127.') or domain.startswith('192.') or domain.startswith('10.'):
|
|
# session.mount('http://', SourceAddressAdapter(domain))
|
|
#session.mount('http://', SourceAddressAdapter((domain, port)))
|
|
if onionRoute:
|
|
session.proxies = {}
|
|
session.proxies['http'] = 'socks5h://localhost:9050'
|
|
session.proxies['https'] = 'socks5h://localhost:9050'
|
|
return session
|
|
|
|
def getJson(session,url: str,headers: {},params: {}) -> {}:
|
|
sessionParams={}
|
|
sessionHeaders={}
|
|
if headers:
|
|
sessionHeaders=headers
|
|
if params:
|
|
sessionParams=params
|
|
sessionHeaders['User-agent'] = "Mozilla/5.0 (Linux; Android 5.1.1; Nexus 5)"
|
|
if not session:
|
|
print('WARN: no session specified for getJson')
|
|
session.cookies.clear()
|
|
try:
|
|
result=session.get(url, headers=sessionHeaders, params=sessionParams)
|
|
return result.json()
|
|
except:
|
|
print('ERROR: getJson failed')
|
|
print('url: '+url)
|
|
print('headers: '+str(sessionHeaders))
|
|
print('params: '+str(sessionParams))
|
|
return None
|
|
|
|
def postJson(session,postJsonObject: {},federationList: [],inboxUrl: str,headers: {},capability: str) -> str:
|
|
"""Post a json message to the inbox of another person
|
|
Supplying a capability, such as "inbox:write"
|
|
"""
|
|
|
|
# always allow capability requests
|
|
if not capability.startswith('cap'):
|
|
# check that we are posting to a permitted domain
|
|
if not urlPermitted(inboxUrl,federationList,capability):
|
|
print('postJson: '+inboxUrl+' not permitted')
|
|
return None
|
|
|
|
postResult = session.post(url = inboxUrl, data = json.dumps(postJsonObject), headers=headers)
|
|
return postResult.text
|
|
|
|
def postImage(session,attachImageFilename: str,federationList: [],inboxUrl: str,headers: {},capability: str) -> str:
|
|
"""Post an image to the inbox of another person or outbox via c2s
|
|
Supplying a capability, such as "inbox:write"
|
|
"""
|
|
# always allow capability requests
|
|
if not capability.startswith('cap'):
|
|
# check that we are posting to a permitted domain
|
|
if not urlPermitted(inboxUrl,federationList,capability):
|
|
print('postJson: '+inboxUrl+' not permitted')
|
|
return None
|
|
|
|
if not (attachImageFilename.endswith('.jpg') or \
|
|
attachImageFilename.endswith('.jpeg') or \
|
|
attachImageFilename.endswith('.png') or \
|
|
attachImageFilename.endswith('.gif')):
|
|
print('Image must be png, jpg, or gif')
|
|
return None
|
|
if not os.path.isfile(attachImageFilename):
|
|
print('Image not found: '+attachImageFilename)
|
|
return None
|
|
contentType='image/jpeg'
|
|
if attachImageFilename.endswith('.png'):
|
|
contentType='image/png'
|
|
if attachImageFilename.endswith('.gif'):
|
|
contentType='image/gif'
|
|
headers['Content-type']=contentType
|
|
|
|
with open(attachImageFilename, 'rb') as avFile:
|
|
mediaBinary = avFile.read()
|
|
postResult = session.post(url=inboxUrl, data=mediaBinary, headers=headers)
|
|
return postResult.text
|
|
return None
|