import socket


# 　处理浏览器的http请求
def handle(connfd):
    print("Request from", connfd.getpeername())
    request = connfd.recv(4096)  # 接收http请求
    # 防止request为空造成下方数组越界报错
    if not request:
        return
    # 将请求按行分割
    request_line = request.splitlines()[0].decode()
    # 获取请求内容，浏览器默认请求/
    info = request_line.split(' ')[1]

    if info == '/':
        # 打开当前目录下的index.html
        f = open('index.html', encoding='UTF-8')
        response = "HTTP/1.1 200 OK\r\n"
        response += "Content-Type: text/html\r\n"
        response += '\r\n'
        response += f.read()
    else:
        response = "HTTP/1.1 404 Not Found\r\n"
        response += "Content-Type: text/html\r\n"
        response += '\r\n'
        response += "<h1>Sorry....</h1>"
    # 向浏览器发送内容
    connfd.send(response.encode())


# 　搭建TCP网络
def TCP():
    server = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    server.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
    server.bind(('0.0.0.0', 8000))
    server.listen(3)
    print("Listen the port 8000...")
    while True:
        conn, addr = server.accept()
        handle(conn)
        conn.close()


if __name__ == "__main__":
    TCP()

# 服务端
import os
import signal
import struct
import sys
from socket import *
from threading import Thread


def handle(c):
    print("连接的客户端", c.getpeername())
    while True:
        data = c.recv(1024)
        if not data:
            break
        print(data.decode())
        fmt = struct.Struct('32s')
        c.send(fmt.pack('客户端接收成功'.encode('utf-8')))
    c.close()


def tcp_fork(sock, c):
    signal.signal(signal.SIGCHLD, signal.SIG_IGN)
    pid = os.fork()
    if pid == 0:
        sock.close()  # 子进程不需要sock
        handle(c)  # 具体处理客户端请求
        os._exit(0)  # handle()结束意味着客户端断开连接，此时结束子进程
    else:
        c.close()  # 父进程不需要c


def tcp_thread(c):
    t = Thread(target=handle, args=(c,))
    t.setDaemon(True)  # 分支线程随主线程退出
    t.start()


if __name__ == '__main__':
    s = socket()
    # 设置端口立即重用
    s.setsockopt(SOL_SOCKET, SO_REUSEADDR, 1)
    s.bind(("0.0.0.0", 8888))
    s.listen(3)

    print('正在监听', s.getsockname())

    while True:
        try:
            # 阻塞等待其它客户端连接
            conn, addr = s.accept()
        except KeyboardInterrupt:
            sys.exit("服务器退出")
        except Exception as e:
            print(e)
            continue

        tcp_fork(s, conn)
        # tcp_thread(conn)

# 客户端
import socket
import struct
import sys


def socket_client():
    try:
        client = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        client.connect(('127.0.0.1', 8888))
    except socket.error as msg:
        print(msg)
        sys.exit(1)
    while 1:
        data = input('please input:')
        if client:
            client.send(data.encode())
            fmt = struct.Struct('32s')
            receive = fmt.unpack(client.recv(1024))
            print(receive[0].strip(b'\x00').decode('utf-8'))
        if data == 'exit':
            break
    client.close()


if __name__ == '__main__':
    socket_client()