python实现用socket传输文件

python传输文件最重要的有两步:

1).将要传输的文件的信息发送过去,包括文件包,大小以及其它信息;

2).发送端读取文件内容并发送过去,接受端将缓存里面的内容写入文件.

发送端:

# -*- coding: cp936 -*-

from socket import *

import os

import struct

ADDR = ('192.168.0.178',8000)

BUFSIZE = 1024

filename = 'wubi.exe'

FILEINFO_SIZE=struct.calcsize('128s32sI8s')

sendSock = socket(AF_INET,SOCK_STREAM)

sendSock.connect(ADDR)

fhead=struct.pack('128s11I',filename,0,0,0,0,0,0,0,0,os.stat(filename).st_size,0,0)

sendSock.send(fhead)

fp = open(filename,'rb')

while 1:

    filedata = fp.read(BUFSIZE)

    if not filedata: break

    sendSock.send(filedata)

print "文件传送完毕，正在断开连接..."

fp.close()

sendSock.close()

print "连接已关闭..."

 

接收端:

# -*- coding: cp936 -*-

from socket import *

import struct

ADDR = ('192.168.0.178',8000)

BUFSIZE = 1024

FILEINFO_SIZE=struct.calcsize('128s32sI8s')

recvSock = socket(AF_INET,SOCK_STREAM)

recvSock.bind(ADDR)

recvSock.listen(True)

print "等待连接..."

conn,addr = recvSock.accept()

print "客户端已连接—> ",addr

fhead = conn.recv(FILEINFO_SIZE)

filename,temp1,filesize,temp2=struct.unpack('128s32sI8s',fhead)

#print filename,temp1,filesize,temp2

print filename,len(filename),type(filename)

print filesize

filename = 'new_'+filename.strip('\00') #...

fp = open(filename,'wb')

restsize = filesize

print "正在接收文件... ",

while 1:

    if restsize > BUFSIZE:

        filedata = conn.recv(BUFSIZE)

    else:

        filedata = conn.recv(restsize)

    if not filedata: break

    fp.write(filedata)

    restsize = restsize-len(filedata)

    if restsize == 0:

     break

print "接收文件完毕，正在断开连接..."

fp.close()

conn.close()

recvSock.close()

print "连接已关闭..."

 

该程序经过实际测试可用，大家经过修改相应代码即可在本地进行测试使用。

友情提示：

用于传送比较大的文件时会比较慢，还可能照成命令行假死的情况...

#############################################################

用一种编程语言实现一个简单的Server/Client程序；

该程序的主要功能是利用Client从端下载一个文件；

在下载之前，需要有一定的用户身份验证机制（说白了就是先输入以下用户名和密码）；

Server应该是多线程机制的，即为每个Client请求Server都要有一个线程去处理，而不是所有的Client都是一个Server线程处理。

 

处理流程：

1. server启动，监听client请求；
2. client启动；
3. server监听到client，发送“Hi”；
4. client收到“Hi”
5. client要求用户输入用户名；
6. 用户输入用户名（如yangsq），发送到服务器（形式为“username:yangsq”）；
7. 服务器验证是否存在这样一个用户，如果有到step 8，否则转到5；
8. client用求用户输入密码（如123456），发送到服务器（形式为“password:123456”）；
9. 服务器验证密码是否正确，不正确转到step 8，正确开始传输文件（为了简单，文件预先指定好）；
10. client收到文件，结束后发送“bye”；同时server端收到“bye”后结束线程。

import SocketServer, time

class MyServer(SocketServer.BaseRequestHandler):
    userInfo = {
        'yangsq'    : 'yangsq',
        'hudeyong'  : 'hudeyong',
        'mudan'     : 'mudan' }

    def handle(self):
        print 'Connected from', self.client_address

        while True:
            receivedData = self.request.recv(8192)
            if not receivedData:
                continue

            elif receivedData == 'Hi, server':
                self.request.sendall('hi, client')

            elif receivedData.startswith('name'):
                self.clientName = receivedData.split(':')[-1]
                if MyServer.userInfo.has_key(self.clientName):
                    self.request.sendall('valid')
                else:
                    self.request.sendall('invalid')

            elif receivedData.startswith('pwd'):
                self.clientPwd = receivedData.split(':')[-1]
                if self.clientPwd == MyServer.userInfo[self.clientName]:
                    self.request.sendall('valid')
                    time.sleep(5)

                    sfile = open('PyNet.pdf', 'rb')
                    while True:
                        data = sfile.read(1024)
                        if not data:
                            break
                        while len(data) > 0:
                            intSent = self.request.send(data)
                            data = data[intSent:]

                    time.sleep(3)
                    self.request.sendall('EOF')
                else:
                    self.request.sendall('invalid')

            elif receivedData == 'bye':
                break

        self.request.close()

        print 'Disconnected from', self.client_address
        print

if __name__ == '__main__':
    print 'Server is started\nwaiting for connection...\n'
    srv = SocketServer.ThreadingTCPServer(('localhost', 50000), MyServer)
    srv.serve_forever()

说明：

• line-55到line-58的作用就相当于中某个类里面的main函数，即一个类的入口。
• python中SocketServer module里提供了好多实用的现成的类，BaseRequestHandler就是一个，它的作用是为每一个请求fork一个线程，只要继承它，就有这个能力了，哈哈，真是美事。
• 当然，我们继承了BaseRequestHandler，就是override它的handle方法，就像java中继承了Thread后要实现run方法一样。实际上这个handle方法的内容和我们的java版本的run函数实现的完全一样。
• line-30到line-43就是处理文件下载的主要内容了。看着都挺眼熟的呵：）
• 这里在文件发送完后发了一个“EOF”，告诉client文件传完了。

客户端：

python 代码

line-34到line-41处理文件下载，client收到server的“EOF”信号后，就知道文件传完了。

最后需要说明一下python的文件，由于是内置类型，所以不想java那样有那么多的reader，writer，input，ouput啊。python中，在打开或建立一个文件时，主要是通过模式（mode）来区别的。

python的网络编程确实简单，因为它提供了各种功能的已经写好的类，直接继承就Ok了。