Ashita-WebServer/ashita_ws.py at main · unknown989/Ashita-WebServer · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
# -*- coding: utf-8 -*-

import socket
import sys
from configparser import ConfigParser as cp
from _thread import start_new_thread as snt
import mimetypes

class Server():
    def __init__(self):
        try:
            co = cp()

            co.read("config.ini")

            sc = co["SERVERINFO"]

            host = sc["HOST"]
            port = sc["PORT"]
            self.mainFile = sc["INDEXFILE"]
            self.defDir = sc["DEFAULTDIR"]
            self.redirect = sc["REDIRECT"]
            self.shownotfounderrorinpage = sc["404_ERROR_PAGE_TOGGLE"]
            self.shownotfounderrorinpage = True if self.shownotfounderrorinpage == "true" else False
            self.redirect = True if self.redirect == "true" else False
            self.redirectloc = sc["REDIRECT_LOCATION"]
            self.error404 = sc["404PAGE"]
            self.error404 = self.defDir+self.error404 if self.defDir.endswith("/") else self.defDir+"/"+self.error404
            self.ip,self.cl = [],[]
            self.s = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
            self.s.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
            self.isFileBin = False
            try:
                print("Binding...")
                self.s.bind((host,int(port)))
                print("Listening...")
                self.s.listen()
                print("Starting The Thread For Accepting Connections...")
                snt(self.acceptClients())
                print("Done")
                self.s.close()
            except KeyboardInterrupt:
                self.s.close()
        except Exception as e:
            print(str(e))
            sys.exit()

    def acceptClients(self):
        while True:
            i,c = self.s.accept()
            self.ip.append(i)
            self.cl.append(c)
            filename = i.recv(1024)
            print(filename.decode())
            filename = filename.split()[1]
            filename = filename.decode()
            #print("FILENAME REQUESTES : "+str(filename))

            if filename == "/":
                filename = self.defDir+"/"+self.mainFile if not self.defDir.endswith("/") else self.defDir+"/"+self.mainFile
            else:
                if str(filename).startswith("/"):
                    filename = self.defDir+"/"+"".join(list(filename)[1:]) if not self.defDir.endswith("/") else self.defDir+"/"+"".join(list(filename)[1:])
                else:
                    filename = self.defDir+"/"+filename if not self.defDir.endswith("/") else self.defDir+"/"+self.mainFile
            print("FILENAME REQUESTES : "+str(filename))
            try:
                if "text" in mimetypes.guess_type(filename)[0]:
                    with open(filename,"r") as f:
                        content = f.read()
                        self.isFileBin = False
                else:
                    with open(filename,"rb") as f:
                        content = f.read()
                        self.isFileBin = True
                self.sendMsg("HTTP/1.1 200 OK\r\n".encode(encoding="utf8"),i)
                if not self.isFileBin:
                    self.sendMsg("Content-Type: {}\r\nContent-Length: {}\r\nConnection: Close\r\n\r\n".format(mimetypes.guess_type(filename)[0],str(len(content))).encode(encoding="utf8"),i)
                    for cc in content:
                        self.sendMsg(str(cc).encode(encoding="utf8"),i)
                else:
                    self.sendMsg("Connection: Close\r\nContent-Type: {}\r\nContent-Length: {}\r\n\r\n".format(mimetypes.guess_type(filename)[0],str(len(content))).encode()+content,i)

            except Exception as e:
                print(str(e))
                if self.redirect:
                    self.sendMsg("HTTP/1.1 303 See Other\r\nLocation: {}\r\n".format(self.redirectloc).encode(encoding="utf8"),i)
                else:
                    self.sendMsg("HTTP/1.1 404 Not Found\r\n\r\n".encode(encoding="utf8"),i)
                    if self.shownotfounderrorinpage:
                        with open(self.error404,"r") as f:
                            for ce in f.read():
                                self.sendMsg(ce.encode(encoding="utf8"),i)
            i.close()

    def sendMsg(self,msg,ip):
        ip.send(msg)
        print("Sent : "+str(msg))


# Running The Server
try:
    Server()
except KeyboardInterrupt:
    sys.exit()