From be77eeb6c1f93c2eeb3d3c77dd9423567e29f18c Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=D0=A0=D0=BE=D0=BC=D0=B0=D0=BD=20=D0=91=D0=BE=D1=80=D0=BE?=
 =?UTF-8?q?=D0=B4=D0=B8=D0=BD?= <roman@ukamnya.ru>
Date: Wed, 3 Apr 2019 00:27:08 +0300
Subject: [PATCH] =?UTF-8?q?=D0=9D=D0=B0=D1=87=D0=B0=D0=BB=D1=8C=D0=BD?=
 =?UTF-8?q?=D1=8B=D0=B9=20=D0=BA=D0=BE=D0=BC=D0=BC=D0=B8=D1=82?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 mods/udp_srv.py | 78 +++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 78 insertions(+)
 create mode 100644 mods/udp_srv.py

diff --git a/mods/udp_srv.py b/mods/udp_srv.py
new file mode 100644
index 0000000..edf7db4
--- /dev/null
+++ b/mods/udp_srv.py
@@ -0,0 +1,78 @@
+import socketserver
+import pickle
+import simplecrypto
+from uuid import uuid4
+
+BUFSIZE = 8192
+BLOCKSIZE = 4096
+
+PACKET_TYPE_RECV_RESULT = 'recv_result'
+PACKET_TYPE_DATA_FRAGMENT = 'data_fragment'
+PACKET_TYPE_SVC_MESSAGE = 'svc_message'
+PACKET_TYPE_NEW_CONNECTION = 'new_connection'
+
+DATA_TYPE_FILE_CHUNK = 'file_chunk'
+DATA_TYPE_CMD = 'cmd'
+DATA_TYPE_SEARCH_QUERY = 'search_query'
+DATA_TYPE_SEARCH_RESULT = 'search_result'
+DATA_TYPE_PEER_PUBKEY = 'peer_pubkey'
+
+SVC_MESSAGE_BAD_PACKET = 'bad_packet'
+SVC_MESSAGE_DECRYPT_ERROR = 'decrypt_error'
+SVC_MESSAGE_YOU_ARE_STRANGER = 'you_are_stranger'
+
+RECV_OK_CODE = 0
+RECV_ERROR_CODE = 1
+
+HEADER_SVC_YOU_ARE_STRANGER = {'type': PACKET_TYPE_SVC_MESSAGE, 'msg': SVC_MESSAGE_YOU_ARE_STRANGER}
+
+HEADER_RECV_OK = {'type': PACKET_TYPE_RECV_RESULT, 'code': RECV_OK_CODE}
+HEADER_RECV_ERROR = {'type': PACKET_TYPE_RECV_RESULT, 'code': RECV_ERROR_CODE}
+
+key = None
+
+peers = {}
+
+class Peer:
+    def __init__(self, peer_addr, pubkey):
+        self.addr = peer_addr
+        self.pubkey = pubkey
+
+def get_key():
+    global key
+    if not key:
+        key = simplecrypto.RsaKeypair()
+    return key
+
+def pickle_data(data):
+    return pickle.dumps(data, protocol=4)
+
+def write_svc_msg(fobj, svc_msg, extra_data={}):
+    MSG = {'type': PACKET_TYPE_SVC_MESSAGE, 'msg': svc_msg}
+    if extra_data:
+        MSG.update(extra_data)
+    fobj.write(pickle_data(MSG))
+
+class UDPRequestHandler(socketserver.DatagramRequestHandler):
+    def handle(self):
+        datagram = self.rfile.read(BUFSIZE)
+        if self.client_address not in peers:
+            write_svc_msg(self.wfile, SVC_MESSAGE_YOU_ARE_STRANGER)
+            return
+        try:
+            unpickled_datagram = pickle.dumps(datagram)
+        except pickle.UnpicklingError:
+            write_svc_msg(self.wfile, SVC_MESSAGE_BAD_PACKET)
+            return
+        pk = get_key()
+        try:
+            data = pickle.loads(pk.decrypt_raw(unpickled_datagram['packet']))
+        except:
+            write_svc_msg(self.wfile, SVC_MESSAGE_DECRYPT_ERROR, {'packet_id': unpickled_datagram['packet_id']})
+            return
+        block_id = data['block_id']
+        fragment_num = data['num']
+        cmd = data['']
+
+
+