004---基于TCP的套接字

基于TCP的套接字

tcp是基于链接的，必须先启动服务端，然后再启动客户端去连接服务端。

之前实现的简单套接字就是基于TCP的，但是只能实现收发消息一次、服务器与客户端都断开了。不够过瘾。

通信循环版本

服务端

#! /usr/bin/env python

# -*- coding: utf-8 -*-

# __author__ = "ziya"

# Date: 2019-02-01

import socket

ip_port = ('127.0.0.1', 8000)

buffer_size = 1024

service = socket.socket(family=socket.AF_INET, type=socket.SOCK_STREAM)

service.bind(ip_port)

service.listen(5)

conn, addr = service.accept()

# 通信循环(c/s之间不只收发一次消息)

while 1:

    msg = conn.recv(buffer_size)

    print('客户端发来的消息：', msg.decode('utf-8'))

    data = input('给客户端发的消息：')

    conn.send(data.encode('utf-8'))

    print('发生给客户端的消息成功：', data)

conn.close()

sk.close()

客户端

#! /usr/bin/env python

# -*- coding: utf-8 -*-

# __author__ = "ziya"

# Date: 2019-02-01

import socket

ip_port = ('127.0.0.1', 8000)

buffer_size = 1024

client = socket.socket(family=socket.AF_INET, type=socket.SOCK_STREAM)

client.connect(ip_port)

while 1:

    msg = input('输入发给服务端的消息：').strip()

    client.send(msg.encode('utf-8'))

    data = client.recv(buffer_size)

    print('服务端回复的消息：', data.decode('utf-8'))

client.close()

从图中可以看出有bug，当客户端强制端口连接，服务器那边会抛出异常。

还有就是客户端发送不了空消息(空格)。

小Bug修复

服务端

#! /usr/bin/env python

# -*- coding: utf-8 -*-

# __author__ = "ziya"

# Date: 2019-02-01

import socket

ip_port = ('127.0.0.1', 8000)

buffer_size = 1024

service = socket.socket(family=socket.AF_INET, type=socket.SOCK_STREAM)

# 有时候会出现端口占用的情况，加上下面代码

service.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)

service.bind(ip_port)

service.listen(5)

conn, addr = service.accept()

# 通信循环(c/s之间不只收发一次消息)

while 1:

    try:

        msg = conn.recv(buffer_size)

        if not msg:

            print('客户端主动退出')

            break

        print('客户端发来的消息：', msg.decode('utf-8'))

        data = input('给客户端发的消息：')

        conn.send(data.encode('utf-8'))

        print('发生给客户端的消息成功：', data)

    except ConnectionResetError as con:

        print('客户端强制退出')

        break

conn.close()

service.close()

客户端

#! /usr/bin/env python

# -*- coding: utf-8 -*-

# __author__ = "ziya"

# Date: 2019-02-01

import socket

ip_port = ('127.0.0.1', 8000)

buffer_size = 1024

client = socket.socket(family=socket.AF_INET, type=socket.SOCK_STREAM)

client.connect(ip_port)

while 1:

    msg = input('输入发给服务端的消息(q/Q退出)：').strip()

    if not msg:

        print('不能发送空消息给服务器')

        continue

    if msg.upper() == 'Q':

        print('正在主动退出')

        break

    client.send(msg.encode('utf-8'))

    data = client.recv(buffer_size)

    print('服务端回复的消息：', data.decode('utf-8'))

client.close()

print('exit ok')

连接循环版本

虽然没啥大问题。但是还有一个大问题，就是只能有一个客户端连接。客服端正常断开或者强制断开，服务器也会跟着断，并不能保证其他客户端来连接。

服务器

#! /usr/bin/env python

# -*- coding: utf-8 -*-

# __author__ = "ziya"

# Date: 2019-02-01

import socket

ip_port = ('127.0.0.1', 8000)

buffer_size = 1024

service = socket.socket(family=socket.AF_INET, type=socket.SOCK_STREAM)

# 有时候会出现端口占用的情况，加上下面代码

service.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)

service.bind(ip_port)

service.listen(5)

while 1:

    # 连接循环(客户端断开之后其他客户端还能连接。

    conn, addr = service.accept()

    while 1:

        # 通信循环(c/s之间不只收发一次消息)

        try:

            msg = conn.recv(buffer_size)

            if not msg:

                print('客户端主动退出')

                break

            print('客户端发来的消息：', msg.decode('utf-8'))

            data = input('给客户端发的消息：')

            conn.send(data.encode('utf-8'))

            print('发生给客户端的消息成功：', data)

        except ConnectionResetError as con:

            print('客户端强制退出')

            break

    conn.close()

service.close()

客户端

#! /usr/bin/env python

# -*- coding: utf-8 -*-

# __author__ = "ziya"

# Date: 2019-02-01

import socket

ip_port = ('127.0.0.1', 8000)

buffer_size = 1024

client = socket.socket(family=socket.AF_INET, type=socket.SOCK_STREAM)

client.connect(ip_port)

while 1:

    msg = input('输入发给服务端的消息(q/Q退出)：').strip()

    if not msg:

        print('不能发送空消息给服务器')

        continue

    if msg.upper() == 'Q':

        print('正在主动退出')

        break

    client.send(msg.encode('utf-8'))

    data = client.recv(buffer_size)

    print('服务端回复的消息：', data.decode('utf-8'))

client.close()

print('exit ok')