start hw4

1 files changed, 91 insertions, 0 deletions

diff --git a/csci4131/hw4/strap012/strap012.py b/csci4131/hw4/strap012/strap012.py
new file mode 100644
index 0000000..7b676c1
--- /dev/null
+++ b/csci4131/hw4/strap012/strap012.py
@@ -0,0 +1,91 @@
+#!/usr/bin/env python3
+# See https://docs.python.org/3.2/library/socket.html
+# for a decscription of python socket and its parameters
+import socket
+
+from threading import Thread
+from argparse import ArgumentParser
+
+BUFSIZE = 4096
+CRLF = '\r\n'
+METHOD_NOT_ALLOWED = 'HTTP/1.1 405  METHOD NOT ALLOWED{}Allow: GET, HEAD, POST {}Connection: close{}{}'.format(CRLF, CRLF, CRLF, CRLF)
+OK = 'HTTP/1.1 200 OK{}{}{}'.format(CRLF, CRLF, CRLF) # head request only
+NOT_FOUND = 'HTTP/1.1 404 NOT FOUND{}Connection: close{}{}'.format(CRLF, CRLF, CRLF)
+FORBIDDEN = 'HTTP/1.1 403 FORBIDDEN{}Connection: close{}{}'.format(CRLF, CRLF, CRLF)
+
+def getContents(type, file):
+  fileExist = False
+  returnValue = ""
+  try:
+    content = open(file)
+  except FileNotFoundError:
+    returnValue = NOT_FOUND
+  except PermissionError:
+    returnValue = FORBIDDEN
+  else:
+    if type == "HEAD":
+      returnValue = OK
+    elif type == "GET":
+      returnValue = OK + content.read() + "{}{}".format(CRLF, CRLF)
+    elif type == "POST":
+      print("B")
+    else:
+      returnValue = METHOD_NOT_ALLOWED
+    content.close()
+  return returnValue
+
+def client_recv(client_sock, client_addr):
+    print('talking to {}'.format(client_addr))
+    data = client_sock.recv(BUFSIZE)
+    data = data.decode('utf-8').replace("\r", "")
+    data = data.split("\n")
+    request = data[0].split(" ")
+    want = getContents(request[0], request[1][1:])
+    print(want)
+    client_sock.send(bytes(want, 'utf-8'))
+
+    client_sock.shutdown(1)
+    client_sock.close()
+   
+    print('connection closed.')
+
+
+class EchoServer:
+  def __init__(self, host, port):
+    print("Server")
+    print('listening on port {}'.format(port))
+    self.host = host
+    self.port = port
+
+    self.setup_socket()
+
+    self.accept()
+
+    self.sock.shutdown()
+    self.sock.close()
+
+  def setup_socket(self):
+    self.sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
+    self.sock.bind((self.host, self.port))
+    self.sock.listen(128)
+
+  def accept(self):
+    while True:
+      (client, address) = self.sock.accept()
+      th = Thread(target=client_recv, args=(client, address))
+      th.start()
+
+def parse_args():
+  parser = ArgumentParser()
+  parser.add_argument('--host', type=str, default='localhost',
+                      help='specify a host to operate on (default: localhost)')
+  parser.add_argument('-p', '--port', type=int, default=9001,
+                      help='specify a port to operate on (default: 900)')
+  args = parser.parse_args()
+  return (args.host, args.port)
+
+
+if __name__ == '__main__':
+  (host, port) = parse_args()
+  EchoServer(host, port)
+