MiNode/minode/i2p/controller.py

# -*- coding: utf-8 -*-
import base64
import logging
import os
import socket
import threading
import time

from i2p.util import receive_line, pub_from_priv
import shared


class I2PController(threading.Thread):
    def __init__(self, host='127.0.0.1', port=7656, dest_priv=b''):
        super().__init__(name='I2P Controller')

        self.host = host
        self.port = port
        self.nick = b'MiNode_' + base64.b16encode(os.urandom(4)).lower()

        self.s = socket.create_connection((self.host, self.port))

        self.version_reply = []

        self.init_connection()

        if dest_priv:
            self.dest_priv = dest_priv
            self.dest_pub = pub_from_priv(dest_priv)
        else:
            self.dest_priv = b''
            self.dest_pub = b''
            self.generate_destination()

        self.create_session()

    def _receive_line(self):
        line = receive_line(self.s)
        # logging.debug('I2PController <- ' + str(line))
        return line

    def _send(self, command):
        # logging.debug('I2PController -> ' + str(command))
        self.s.sendall(command)

    def init_connection(self):
        self._send(b'HELLO VERSION MIN=3.0 MAX=3.3\n')
        self.version_reply = self._receive_line().split()
        assert b'RESULT=OK' in self.version_reply

    def generate_destination(self):
        if b'VERSION=3.0' in self.version_reply:
            # We will now receive old DSA_SHA1 destination :(
            self._send(b'DEST GENERATE\n')
        else:
            self._send(b'DEST GENERATE SIGNATURE_TYPE=EdDSA_SHA512_Ed25519\n')

        reply = self._receive_line().split()
        for par in reply:
            if par.startswith(b'PUB='):
                self.dest_pub = par.replace(b'PUB=', b'')
            if par.startswith(b'PRIV='):
                self.dest_priv = par.replace(b'PRIV=', b'')
        assert self.dest_priv

    def create_session(self):
        self._send(b'SESSION CREATE STYLE=STREAM ID=' + self.nick +
                   b' inbound.length=' + str(shared.i2p_tunnel_length).encode() +
                   b' outbound.length=' + str(shared.i2p_tunnel_length).encode() +
                   b' DESTINATION=' + self.dest_priv + b'\n')
        reply = self._receive_line().split()
        if b'RESULT=OK' not in reply:
            logging.warning(reply)
            logging.warning('We could not create I2P session, retrying in 5 seconds.')
            time.sleep(5)
            self.create_session()

    def run(self):
        self.s.settimeout(1)
        while True:
            if not shared.shutting_down:
                try:
                    msg = self._receive_line().split(b' ')
                    if msg[0] == b'PING':
                        self._send(b'PONG ' + msg[1] + b'\n')
                except socket.timeout:
                    pass
            else:
                logging.debug('Shutting down I2P Controller')
                self.s.close()
                break