irPICode.py

import time
import serial
import sys
import socket
import threading

import pypacker.pypacker as pypacker
from pypacker import psocket
from pypacker.layer12 import ethernet
from pypacker.layer3 import ip, icmp

Interface = "wlan0"


class Etherneter:
    def __init__(self, serial_instance, socket):
        self.serial = serial_instance
        self.socket = socket
        self._write_lock = threading.Lock()

    def run(self):
        self.alive = True
        self.thread_read = threading.Thread(target=self.reader)
        self.thread_read.daemon = True
        self.thread_read.name = 'serial->socket'
        self.thread_read.start()
        self.thread_write = threading.Thread(target=self.writer)
        self.thread_write.daemon = True
        self.thread_write.name = 'socket->serial'
        self.thread_write.start()

    def reader(self):
        """loop forever and copy serial->socket"""
        while self.alive:
            try:
                data = self.serial.read_until(b'~')[:-1]
                packet = ethernet.Ethernet(data)
                if packet[icmp.ICMP]:
                    packet[ethernet.Ethernet].dst_s = "dc:a6:32:00:a7:8b"
                    packet[ip.IP].dst_s = "192.168.1.35"
                    packet[icmp.ICMP].sum = b'0x1783'
                    print("\n\n__________________RESPONSE FROM VISIBLE PI__________________")
                    print(packet)
                if data:
                    self.write(packet.bin())
            except socket.error as msg:
                break
        self.alive = False

    def write(self, data):
        """thread safe socket write with no data escaping. used to send telnet stuff"""
        with self._write_lock:
            self.socket.send(data)

    def writer(self):
        """loop forever and copy socket->serial"""
        #while self.alive:
        try:
            icmpreq = ethernet.Ethernet(src_s="dc:a6:32:00:a7:8b", dst_s="ec:84:b4:3e:c8:20", type=ethernet.ETH_TYPE_IP) +\
                ip.IP(p=ip.IP_PROTO_ICMP, src_s="192.168.1.35", dst_s="172.217.166.110") +\
                icmp.ICMP(type=8) +\
                icmp.ICMP.Echo(id=1, ts=123456789, body_bytes=b"12345678901234567890")
            self.serial.write(icmpreq.bin()+b'~')
        except socket.error as msg:
            print(msg)
        self.stop()

    def stop(self):
        """Stop copying"""
        if self.alive:
            self.thread_read.join()
            self.thread_write.join()
            self.alive = False


if __name__ == '__main__':

    ser = serial.Serial()
    ser.baudrate = 9600
    ser.port = '/dev/serial0'

    psock = None

    try:
        ser.open()
    except serial.SerialException as e:
        sys.exit(1)

    try:
        psock = psocket.SocketHndl(iface_name=Interface, timeout=10)
    except socket.error as msg:
        sys.exit(1)

    try:
        e = Etherneter(ser, psock)
        try:
            e.run()
        finally:
            e.stop()
            psock.close()
    except KeyboardInterrupt:
        sys.stdout.write('\n')