new

diff --git a/measure_energy_iot/client_tcp4.py b/measure_energy_iot/client_tcp4.py
new file mode 100644 (file)
index 0000000..b96dbd3
--- /dev/null
+++ b/measure_energy_iot/client_tcp4.py
@@ -0,0 +1,32 @@
+import socket
+import threading
+import os
+
+import buffer
+
+HOST = 'bilbo'
+PORT = 2345
+
+s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
+s.connect((HOST, PORT))
+
+with s:
+    sbuf = buffer.Buffer(s)
+
+
+#    files = input('Enter file(s) to send: ')
+ #   files_to_send = files.split()
+    file_name="lena.jpg"
+    for i in range(10):
+#    for file_name in files_to_send:
+        print(file_name)
+        sbuf.put_utf8(file_name)
+
+        file_size = os.path.getsize(file_name)
+        sbuf.put_utf8(str(file_size))
+
+        with open(file_name, 'rb') as f:
+            sbuf.put_bytes(f.read())
+        print('File Sent')
+
+

diff --git a/measure_energy_iot/server_tcp4.py b/measure_energy_iot/server_tcp4.py
new file mode 100644 (file)
index 0000000..52c3afd
--- /dev/null
+++ b/measure_energy_iot/server_tcp4.py
@@ -0,0 +1,52 @@
+import socket
+import os
+
+import buffer
+
+HOST = 'bilbo'
+PORT = 2345
+
+# If server and client run in same local directory,
+# need a separate place to store the uploads.
+try:
+    os.mkdir('uploads')
+except FileExistsError:
+    pass
+
+s = socket.socket()
+s.bind((HOST, PORT))
+s.listen(10)
+print("Waiting for a connection.....")
+
+while True:
+    conn, addr = s.accept()
+    print("Got a connection from ", addr)
+    connbuf = buffer.Buffer(conn)
+
+    while True:
+
+
+        file_name = connbuf.get_utf8()
+        if not file_name:
+            break
+#        file_name = os.path.join('uploads',file_name)
+#        print('file name: ', file_name)
+
+        file_size = int(connbuf.get_utf8())
+        print('file size: ', file_size )
+
+        with open("/tmp/"+file_name, 'wb') as f:
+            remaining = file_size
+            while remaining:
+                chunk_size = 4096 if remaining >= 4096 else remaining
+                chunk = connbuf.get_bytes(chunk_size)
+                if not chunk: break
+                f.write(chunk)
+                remaining -= len(chunk)
+            if remaining:
+                print('File incomplete.  Missing',remaining,'bytes.')
+            else:
+                print('File received successfully.')
+    print('Connection closed.')
+    conn.close()
+