【下载文档:  python使用paramiko模块实现ssh远程登陆上传文件并执行.txt 】

---

python使用paramiko模块实现ssh远程登陆上传文件并执行
程序执行时需要读取两个文件command.txt和ipandpass.txt。格式如下：
复制代码 代码如下:command.txt:ThreadNum:1port:22local_dir:hello_mkdirremote_dir:hello_mkdiralter_auth:chmod 755 hello_mkdirexec_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 Queueimport sysimport threadingimport paramikoimport socketfrom threading import Threadimport 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\n" 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('\n') 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\n').split(' ') wm.add_job(uploadAndExecu,username,pwd,IpAdd,commandLine['port'],commandLine)