python은paramiko 모듈을 사용하여 ssh 원격 로그인 업로드 파일을 실현하고 실행

프로그램이 실행될 때 두 개의 파일command를 읽어야 합니다.txt 및 ipandpass.txt.형식은 다음과 같습니다.

 
  
command.txt: 
  
ThreadNum:1 
  
port:22 
  
local_dir:hello_mkdir 
  
remote_dir:hello_mkdir 
  
alter_auth:chmod 755 hello_mkdir 
  
exec_program:./hello_mkdir 
   
  
ipandpass.txt:
ip username password
 
 

프로그램의 대기열 조작은 수정된 다른 프로그램인데, 확실히 잘 썼다.이 프로그램도 옳고 그르다. 만약 가져가서 나쁜 짓을 한다면, 나는 먼저 나와 무관하다고 말할 것이다. 나는 단지 나의 코드를 공유할 뿐이다.관심 있는 동지들이 기술 응용에 대해 토론하기를 바란다.이 중에서paramiko, 대열, 다선정을 사용했고 후속으로도 이 세 가지 측면의 것을 썼다.비평과 시정을 환영합니다.사실 이 프로그램은 일부 부분이 아직 최적화되어야 한다.

 
  
#function:upload files through ssh protocal and excute the files 
  
#lib:paramiko 
  
#MyThread:init a thread to run the function 
  
#ThreadPol:init a thread pool 
  
#uploadAndExecu:upload file and excute 
  
#readConf:read config file 
  
#-*- coding = utf-8 -*- 
   
  

import Queue
import sys
import threading
import paramiko
import socket
from threading import Thread
import time
 
  
class MyThread(Thread):
    def __init__(self, workQueue, timeout=1):
        Thread.__init__(self)
        self.timeout = timeout
        self.setDaemon(False)
        self.workQueue = workQueue
        self.start()
        #print 'i am runnning ...'
 
  
    def run(self):
        emptyQueue = 0
        while True:
            try:
                callable, username, password, ipAddress, port,comms = self.workQueue.get(timeout = self.timeout)
                #print 'attacking :',ipAddress,username,password,threading.currentThread().getName(),' time : '
                callable(username,password, ipAddress, port,comms)

            except Queue.Empty:
                print threading.currentThread().getName(),":queue is empty ; sleep 5 seconds
"
                emptyQueue += 1
                #judge the queue,if it is empty or not.
                time.sleep(5)
                if emptyQueue == 5:
                    print threading.currentThread().getName(),'i  quit,the queue is empty'
                    break
            except Exception, error:
                print error
 
  
class ThreadPool:
    def __init__(self, num_of_threads=10):
        self.workQueue = Queue.Queue()
        self.threads = []
        self.__createThreadPool(num_of_threads)

    #create the threads pool
    def __createThreadPool(self, num_of_threads):
        for i in range(num_of_threads):
            thread = MyThread(self.workQueue)
            self.threads.append(thread)

    def wait_for_complete(self):
        #print len(self.threads)
        while len(self.threads):
            thread = self.threads.pop()
            if thread.isAlive():

                thread.join()
 
  
    def add_job(self, callable, username, password, ipAddress, Port,comms):
        self.workQueue.put((callable, username, password, ipAddress, Port,comms))

def uploadAndExecu(usernam,password,hostname,port,comm):
    print usernam,password,hostname,port,comm
    try:

        t = paramiko.Transport((hostname,int(port)))
        t.connect(username=username,password=password)
        sftp=paramiko.SFTPClient.from_transport(t)
        sftp.put(comm['local_dir'],comm['remote_dir'])
    except Exception,e:
        print 'upload files failed:',e
        t.close()
    finally:
        t.close()

    
    try:
        ssh = paramiko.SSHClient()
        ssh.load_system_host_keys()
        ssh.set_missing_host_key_policy(paramiko.MissingHostKeyPolicy())
        ssh.connect(hostname, port=int(port), username=username, password=password)
        ssh.exec_command(comm['alter_auth'])
        ssh.exec_command(comm['exec_program'])
    except Exception,e:
        print 'chang file auth or execute the file failed:',e
    ssh.close()

    
def readConf():
    comm={}
    try:
        f = file('command.txt','r')
        for l in f:
            sp = l.split(':')
            comm[sp[0]]=sp[1].strip('
')
    except Exception,e:
        print 'open file command.txt failed:',e
    f.close()

    return comm
 
  
if __name__ == "__main__":

    commandLine = readConf()
    print commandLine 
    #prepare the ips

    wm = ThreadPool(int(commandLine['ThreadNum']))

    try:
        ipFile = file('ipandpass.txt','r')
    except:
        print "[-] ip.txt Open file Failed!"
        sys.exit(1)

    for line in ipFile:
        IpAdd,username,pwd = line.strip('\r
').split(' ')
        wm.add_job(uploadAndExecu,username,pwd,IpAdd,commandLine['port'],commandLine)