android.py

from bluetooth import *
import subprocess
import settings


class Android:
    def __init__(self, bluetooth_channel=3):
        self.server_socket = None
        self.client_socket = None
        self._is_connected = False
        self.bluetooth_channel = bluetooth_channel
        try:
            print("[@] Turning on bluetooth programmatically..")
            subprocess.check_output(["sudo", "hciconfig", "hci0", "piscan"])
            time.sleep(3)
        except Exception as e:
            print(
                "[!] Error turning on bluetooth. Please `sudo hciconfig hci0 piscan` manually")

    def close(self):
        if self.client_socket != None:
            print("[@] Android: closing bluetooth socket")
            self.client_socket.close()
        if self.server_socket != None:
            print("[@] Android: closing server socket")
            self.server_socket.close()
        self._is_connected = False

    def is_connected(self):
        return self._is_connected

    def connect(self):
        try:
            self.server_socket = BluetoothSocket(RFCOMM)
            self.server_socket.bind(("", self.bluetooth_channel))
            self.server_socket.listen(1)
            self.port = self.server_socket.getsockname()[1]
            uuid = "00001101-0000-1000-8000-00805F9B34FB"
            advertise_service(
                self.server_socket,
                "MDP22",
                service_id=uuid,
                service_classes=[uuid, SERIAL_PORT_CLASS],
                profiles=[SERIAL_PORT_PROFILE],)
            print("Waiting for connection on RFCOMM channel on port {}".format(self.port))
            self.client_socket, client_address = self.server_socket.accept()
            print("[@] Accepted connection from {}".format(client_address))
            self._is_connected = True
        except Exception as e:
            print(e)

    def write(self, msg=None):
        if msg is None:
            raise Exception(
                'Error: Attempted to write None object to android!')

        if settings.comms:
            print("[@] Writing msg <{}> to Android".format(msg))

        try:
            msg = bytes(msg, "UTF-8")
            self.client_socket.send(msg)
        except Exception as e:
            print("[!] Error writing message to BT device.")
            print(e)

    def read(self):
        try:
            msg = self.client_socket.recv(2048)
            if msg is None:
                print("[!] Failed to receive any message from Android.")
                return None

            msg = msg.decode("UTF-8")
            if settings.comms:
                print("[@] Received msg <{}> from android".format(
                    msg))
            return msg
        except Exception as e:
            print("[!] Error reading message from BT device. Reconnecting..")
            self.connect()