From 0bc9c35f76e77f7f4e9b52a4fc557bfe2715839d Mon Sep 17 00:00:00 2001
From: RhysMJ <rhysmjones11@gmail.com>
Date: Thu, 28 Jun 2018 13:35:13 +0100
Subject: [PATCH] TCP Module v0.9

This is the TCP module which can be used with or without the example program. This module makes it easier to setup TCP and uses the socket module which is built into python.
---
 tcp.py | 108 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 108 insertions(+)
 create mode 100644 tcp.py

diff --git a/tcp.py b/tcp.py
new file mode 100644
index 0000000..e4c2c91
--- /dev/null
+++ b/tcp.py
@@ -0,0 +1,108 @@
+## TCP Easy Setup Module v0.9##
+## Component Connect to other Machine ##
+## R. M. Jones ##
+
+import socket
+import sys
+import re
+
+def validate(Type, data):
+    ip = re.compile("^[0-9]{1,3}[.]{1}[0-9]{1,3}[.]{1}[0-9]{1,3}[.]{1}$")
+
+    if Type == 'ip':
+        if ip.match(data):
+            return True
+        else:
+            return False
+    
+    else:
+        return "No Type: "+Type
+    
+#if domain name is sent then it is resolved here
+def domain(host):
+    try:
+        remote_ip = socket.gethostbyname(host)
+        return remote_ip
+    
+    except socket.gaierror:
+        print("Host name could not be resolved")
+        return False
+
+def getmyip():
+    s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
+    s.connect(("1.1.1.1", 80))
+    TCP_IP = s.getsockname()[0]
+    s.close()
+    return TCP_IP
+
+#The default ip is your local one and port is 5050
+def bind(TCP_IP=getmyip(), TCP_PORT=5050):
+    s = create()
+    s.bind((TCP_IP, TCP_PORT))
+    print("Ready to connect on: "+TCP_IP+" on port:", TCP_PORT)
+    s.listen(1)
+    conn, addr = s.accept()
+    print ('Connection address:', addr)
+    return conn
+
+def create():
+    try:
+        s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
+        return s
+    except socket.error:
+        print("Failed to connect")
+        return False
+    
+    
+def connect(remote_ip, port):
+    
+    if not validate('ip', remote_ip):
+        remote_ip = domain(remote_ip)
+        if remote_ip == False:
+            print("Error Resolving Host name")
+            return False
+        
+    print("Connecting to: " + remote_ip+" Port:",port)
+    try:
+        s.connect((remote_ip, port))
+        
+    
+    except NameError:
+        try:
+            s = create()
+            s.connect((remote_ip, port))
+            
+        except ConnectionRefusedError:
+            print("Connection Refused Error (likely because other machine's port isn't open)")
+            return False
+        
+    except ConnectionRefusedError:
+        print("Connection Refused Error (likely because other machine's port isn't open)")
+        return False
+    
+    except:
+        print("Unknown Error")
+        return False
+    print("Socket connected to IP: " + remote_ip)
+    return s
+
+def send(data, s):
+    try:
+        s.sendall(data.encode())
+        return True
+    except:
+        print("Couldn't send data")
+        return False
+
+def receive(s, buff=4069):
+    try:
+        return s.recv(buff).decode()
+    
+    except ConnectionResetError:
+        print("Error ConnectionResetError")
+        return False
+    
+    except ConnectionAbortedError:
+        print("Error ConnectionAbortedError")
+        return False
+#connect(socket.gethostbyname(socket.gethostname()), 5005)