Windows超级命令行工具箱

#!/usr/bin/env python
# -*- coding: utf-8 -*-

import sys
import socket
import threading
import logging
import optparse


class PipeThread（threading.Thread）:

    def __init__（self source_fd target_fd）:
        super（PipeThread self）.__init__（）
        self.logger = logging.getLogger（‘PipeThread‘）
        self.source_fd = source_fd
        self.target_fd = target_fd
        self.source_addr = self.source_fd.getpeername（）
        self.target_addr = self.target_fd.getpeername（）

    def run（self）:
        while True:
            try:
                data = self.source_fd.recv（4096）
                if len（data） > 0:
                    self.logger.debug（‘read  %04i from %s:%d‘ len（data）
                                      self.source_addr[0] self.source_addr[1]）
                    sent = self.target_fd.send（data）
                    self.logger.debug（‘write %04i to   %s:%d‘ sent
                                      self.target_addr[0] self.target_addr[1]）
                else:
                    break
            except socket.error:
                break
        self.logger.debug（‘connection %s:%d is closed.‘ self.source_addr[0]
                          self.source_addr[1]）
        self.logger.debug（‘connection %s:%d is closed.‘ self.target_addr[0]
                          self.target_addr[1]）
        self.source_fd.close（）
        self.target_fd.close（）


class Forwarder（object）:

    def __init__（self ip port remoteip remoteport backlog=5）:
        self.remoteip = remoteip
        self.remoteport = remoteport
        self.sock = socket.socket（socket.AF_INET socket.SOCK_STREAM）
        self.sock.setsockopt（socket.SOL_SOCKET socket.SO_REUSEADDR 1）
        self.sock.bind（（ip port））
        self.sock.listen（backlog）

    def run（self）:
        while True:
            client_fd client_addr = self.sock.accept（）
            target_fd = socket.socket（socket.AF_INET socket.SOCK_STREAM）
            target_fd.connect（（self.remoteip self.remoteport））

            threads = [
                PipeThread（client_fd target_fd）
                PipeThread（target_fd client_fd）
            ]

            for t in threads:
                t.setDaemon（True）
                t.start（）

    def __del__（self）:
        self.sock.close（）


if __name__ == ‘__main__‘:
    parser = optparse.OptionParser（）

    parser.add_option（
        ‘-l‘ ‘--local-ip‘ dest=‘local_ip‘
        help=‘Local IP address to bind to‘）
    parser.add_option（
        ‘-p‘ ‘--local-port‘
        type=‘int‘ dest=‘local_port‘
        help=‘Local port to bind to‘）
    parser.add_option（
        ‘-r‘ ‘--remote-ip‘ dest=‘remote_ip‘
        help=‘Local IP address to bind to‘）
    parser.add_option（
        ‘-P‘ ‘--remote-port‘
        type=‘int‘ dest=‘remote_port‘
        help=‘Remote port to bind to‘）
    parser.add_option（
        ‘-v‘ ‘--verbose‘
        action=‘stor