Posted inPython

Lập trình mạng với Python Socket: Từ ‘bắt tay’ TCP đến Server đa luồng thực chiến

Python tutorial - IT technology blog
Python tutorial - IT technology blog

Tại sao SysAdmin nên tự viết công cụ bằng Python Socket?

Dân vận hành hệ thống thường dựa vào netcat, telnet hay curl để check nhanh kết nối. Tuy nhiên, thực tế luôn nảy sinh những ca khó mà công cụ có sẵn không xử lý được. Chẳng hạn, bạn cần một script siêu nhẹ (chỉ khoảng 50-100 dòng code) để đẩy log từ 500 nodes về một máy chủ tập trung mà không muốn cài thêm agent nặng nề.

Dùng Flask hay FastAPI lúc này giống như dùng dao mổ trâu để giết gà. Chúng quá nặng và chiếm dụng tài nguyên không cần thiết. Thư viện socket có sẵn trong Python cho phép bạn can thiệp sâu vào tầng transport. Bạn sẽ kiểm soát được từng byte dữ liệu, tiết kiệm RAM tối đa và tùy biến protocol theo đúng ý đồ riêng của mình.

Chọn TCP hay UDP: Quyết định sống còn cho hiệu suất

Trước khi gõ code, hãy dừng lại 30 giây để chọn giao thức. Chọn sai không chỉ làm hệ thống chậm mà còn có thể gây mất dữ liệu nghiêm trọng.

TCP (Transmission Control Protocol): Chắc chắn nhưng tốn sức

TCP giống như một cuộc gọi điện thoại, phải có người nhấc máy (bắt tay 3 bước) thì mới nói chuyện được. Nó đảm bảo dữ liệu không mất mát và đúng thứ tự.

  • Ưu điểm: Tin cậy tuyệt đối, tự động gửi lại gói tin lỗi.
  • Nhược điểm: Độ trễ cao hơn do phải duy trì trạng thái kết nối.
  • Ứng dụng: Truyền file cấu hình, gửi lệnh điều khiển (SSH), các giao dịch cần độ chính xác 100%.

UDP (User Datagram Protocol): Nhanh như chớp nhưng hên xui

UDP hoạt động theo kiểu “bắn và quên”. Nó không quan tâm bên nhận có online hay không. Điều này giúp giảm thiểu tối đa độ trễ (overhead).

  • Ưu điểm: Tốc độ cực nhanh, không tốn tài nguyên duy trì kết nối.
  • Nhược điểm: Dễ mất gói tin trên đường truyền chập chờn.
  • Ứng dụng: Streaming, gửi log Syslog hàng loạt, hoặc các hệ thống giám sát nhiệt độ server mỗi giây.

Kinh nghiệm xương máu: Nếu anh em viết tool đẩy cấu hình cho router, hãy dùng TCP. Nếu làm hệ thống thu thập số liệu traffic thời gian thực, UDP là lựa chọn thông minh hơn.

Triển khai TCP Server và Client thực chiến

Hãy bắt đầu với mô hình TCP. Server sẽ đóng vai trò “tổng đài” chờ cuộc gọi từ Client.

1. Xây dựng TCP Server

Server cần được gán (bind) vào một IP và Port cố định để Client biết đường mà tìm.

import socket

def start_tcp_server(host='127.0.0.1', port=65432):
    # AF_INET: Dùng IPv4, SOCK_STREAM: Dùng TCP
    with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
        s.bind((host, port))
        s.listen()
        print(f"[*] Server dang cho ket noi tai {host}:{port}...")
        
        conn, addr = s.accept()
        with conn:
            print(f"[+] Co khach ghe tham: {addr}")
            while True:
                data = conn.recv(1024)
                if not data: break
                print(f"-> Nhan: {data.decode('utf-8')}")
                conn.sendall(b"Da nhan tin nhan!")

if __name__ == "__main__":
    start_tcp_server()

2. Xây dựng TCP Client

import socket

def start_tcp_client(host='127.0.0.1', port=65432):
    with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
        s.connect((host, port))
        s.sendall(b"Hello Server, day la Client!")
        data = s.recv(1024)
        print(f"<- Server phan hoi: {data.decode('utf-8')}")

if __name__ == "__main__":
    start_tcp_client()

Xây dựng UDP cho các tác vụ đẩy log thần tốc

Với UDP, code gọn hơn hẳn vì không cần bước thiết lập kết nối phức tạp.

1. UDP Server

import socket

def start_udp_server(host='127.0.0.1', port=65433):
    with socket.socket(socket.AF_INET, socket.SOCK_DGRAM) as s:
        s.bind((host, port))
        print(f"[*] UDP Server dang truc tai {host}:{port}...")
        while True:
            data, addr = s.recvfrom(1024)
            print(f"-> Tu {addr}: {data.decode('utf-8')}")
            s.sendto(b"OK!", addr)

if __name__ == "__main__":
    start_udp_server()

Mẹo xử lý dữ liệu: Đừng để script bị treo

Dữ liệu qua socket là dạng byte thô. Bạn cần decode sang UTF-8 hoặc parse JSON để sử dụng. Trong các dự án giám sát, mình thường dùng Regex để lọc nhanh các thông báo lỗi từ đống log hỗn độn.

Để test nhanh các mẫu Regex trước khi đưa vào code, anh em có thể dùng Regex Tester online. Nó giúp tiết kiệm hàng giờ đồng hồ ngồi debug thủ công trên terminal.

Nâng cấp Server đa luồng (Multi-threading)

TCP Server cơ bản chỉ phục vụ được duy nhất 1 khách tại 1 thời điểm. Nếu Client 1 chưa ngắt máy, Client 2 sẽ bị treo. Để xử lý hàng chục kết nối cùng lúc, chúng ta cần dùng threading.

import socket
import threading

def handle_client(conn, addr):
    print(f"[!] Dang xu ly cho: {addr}")
    try:
        while True:
            data = conn.recv(1024)
            if not data: break
            conn.sendall(b"ACK")
    finally:
        conn.close()

def multi_threaded_server(host='0.0.0.0', port=65432):
    server = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    server.bind((host, port))
    server.listen(5)
    print("[*] Server san sang nhan nhieu ket noi...")
    
    while True:
        conn, addr = server.accept()
        t = threading.Thread(target=handle_client, args=(conn, addr))
        t.start() # Moi client mot thread rieng

4 lưu ý để code chạy ổn định trên Production

  1. Set Timeout: Đừng bao giờ để socket đợi mãi mãi. Dùng s.settimeout(10.0) để script tự giải phóng nếu mạng gặp sự cố.
  2. Buffer Size: Con số 1024 là kích thước vùng đệm. Nếu gửi file lớn, hãy tăng lên 4096 hoặc 8192 để tối ưu tốc độ đọc.
  3. Firewall: Nhớ mở port trên iptables hoặc Security Group của AWS/GCP. Rất nhiều anh em tốn cả ngày debug code chỉ vì quên mở port.
  4. Bảo mật: Socket mặc định truyền tin dạng clear-text. Nếu truyền dữ liệu nhạy cảm, hãy dùng thư viện ssl để bọc kết nối lại.

Lập trình socket là kỹ năng giúp bạn tách biệt khỏi nhóm chỉ biết dùng tool có sẵn. Hy vọng vài dòng code mẫu này sẽ giúp anh em tự tin hơn khi xử lý các bài toán mạng phức tạp.

Share:
Tags: