Onionr/src/httpapi/miscpublicapi/upload.py

72 lines
2.4 KiB
Python
Raw Normal View History

"""Onionr - Private P2P Communication.
Accept block uploads to the public API server
"""
import sys
2019-05-07 17:56:20 +00:00
from gevent import spawn
2019-12-22 19:42:10 +00:00
from gevent import threading
from flask import Response
from flask import abort
from flask import g
2019-12-22 19:42:10 +00:00
from onionrutils import localcommand
from onionrblocks import blockimporter
import onionrexceptions
import logger
"""
2019-05-07 17:56:20 +00:00
This program is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with this program. If not, see <https://www.gnu.org/licenses/>.
"""
2019-08-02 23:00:04 +00:00
def accept_upload(request):
"""Accept uploaded blocks to our public Onionr protocol API server"""
2019-05-07 17:56:20 +00:00
resp = 'failure'
2019-09-12 15:54:36 +00:00
data = request.get_data()
2019-12-22 19:42:10 +00:00
b_hash = ''
2019-09-12 15:54:36 +00:00
if sys.getsizeof(data) < 100000000:
try:
2019-12-22 19:42:10 +00:00
b_hash = blockimporter.import_block_from_data(data)
if b_hash:
if g.too_many.get_by_string("OnionrCommunicatorDaemon").onlinePeers:
spawn(
localcommand.local_command,
f'/daemon-event/upload_event',
post=True,
is_json=True,
postData={'block': b_hash}
).get(timeout=10)
2019-09-12 15:54:36 +00:00
resp = 'success'
else:
resp = 'failure'
logger.warn(
f'Error encountered importing uploaded block {b_hash}')
2019-09-12 15:54:36 +00:00
except onionrexceptions.BlacklistedBlock:
logger.debug('uploaded block is blacklisted')
resp = 'failure'
except onionrexceptions.InvalidProof:
resp = 'proof'
except onionrexceptions.DataExists:
resp = 'exists'
2019-05-07 17:56:20 +00:00
if resp == 'failure':
abort(400)
2019-09-12 15:54:36 +00:00
elif resp == 'proof':
resp = Response(resp, 400)
logger.warn(
f'Error importing uploaded block, invalid proof {b_hash}')
2019-09-12 15:54:36 +00:00
else:
resp = Response(resp)
2019-11-16 04:18:38 +00:00
return resp