这是个不错的练习,使用python开发P2P程序,或许通过这个我们可以自己搞出来一个P2P下载工具,类似于迅雷。说到迅雷,关于其原理不知道大家是否了解,如果你不了解,我想看完这篇文章,你一定会了解的。啥,你已经了解了?那就过来指点一番。
以前在java中也接触过类似的概念。一个是RMI( Remote Method Invocation)的概念,另外一个就是XML-RPC的概念。
那么什么是XML-RPC呢?它和P2P有什么关系?下面谈谈我的个人理解。
XML-RPC是一个远程过程调用(remote procedure call,RPC)的分布式计算协议,通过XML将调用函数封装,并使用HTTP协议作为传送机制[摘自维基百科]。所以这个XML-RPC可以帮助我们完成远程调用的工作,即调用相邻电脑中的方法,当然前提是在相邻电脑中已经有我们编写的供远程调用的程序在运行(不管是在前台还是后台,就像迅雷一样,总是悄悄运行)。
这里还要提出来一个概念:Node,即节点。每一个电脑被为一个节点,这个只是针对每个电脑只运行一个我们通过XML-RPC编写的程序,如果电脑中同时运行了多个程序,其实每一个程序都是一个节点。有了节点这样的一个概念之后,我想大家可以想象的出来了,不同节点之间相连,形成各种复杂的网状结构。
这时每个节点可以和其他多个节点进行相连,但是我们没必要让一个节点通其他所有的节点都相连,链接太多会很乱,就像人际关系一样。那什么时候连哪些节点呢?这时就要说到P2P了,所谓P2P即指peer to peer,也就是点到点。说是点到点,你可千万别认为只是从一点到一点,因为他可能是从多点到一点,或者一点到多点。而没有固定的从哪个点到哪个点,所有的点都可以相连。
因此在下载东西方面,这样的协议就比传统的只是从某一点下载数据要快很多,资源也会多很多。
其运作流程是这样的,我打一个比方:比如小A在迅雷里下载B片,迅雷上面可以没有这个资源,但是他可以帮你从节点中找,看谁又这个资源,刚好小C电脑里有,并且在迅雷共享目录下,然后迅雷就会把小C电脑中把资源通过自己的节点传回到小A的电脑上,当然更可能的情况是直接让小A和小C相连。
大家在使用迅雷下载东西的时候肯定注意过里面有一项资源:x/xx这样的东西,我觉得,前面的那个x的意思表示当为你提供资源的节点数量,后面的那个xx表示,所有拥有该资源的节点数目,这些节点可能并不在线。
理解了基本的概念之后,再来看python中如何来实现。
可以先做一个小小的尝试: 首先进入命令行,输入python,然后输入一下代码:
::
from simpleXMLRPCServerr import SimpleXMLRPCServerr
s = SimpleXMLRPCServer(("",4242))
def twice(x):
return x*x
s.register_function(twice) #向服务器添加功能
s.serve_forever() #启动服务器
然后在启动一个命令行,进入pyhon。 输入:
from xmlrpclib import ServerProxy s = ServerProxy('http://localhost:4242') s.twice(2) #通过ServerProxy调用远程的方法,
然后你就会看到通过远程方法的计算完成。
是不是很轻松,这个还是比较简陋,不过足以让你理解python的远程调用,再来看看完整的吧。
先上代码,然后再详解。 首先是Server.py:
.. code:: python
#coding=utf-8
from xmlrpclib import ServerProxy,Fault
from os.path import join, abspath,isfile
from SimpleXMLRPCServer import SimpleXMLRPCServer
from urlparse import urlparse
import sys
SimpleXMLRPCServer.allow_reuse_address = 1
MAX_HISTORY_LENGTH = 6
UNHANDLED = 100
ACCESS_DENIED = 200
class UnhandledQuery(Fault):
'''
that's show can't handle the query exception
'''
def __init__(self,message="Couldn't handle the query"):
Fault.__init__(self, UNHANDLED, message)
class AccessDenied(Fault):
'''
when user try to access the forbiden resources raise exception
'''
def __init__(self, message="Access denied"):
Fault.__init__(self, ACCESS_DENIED, message)
def inside(dir,name):
'''
check the dir that user defined is contain the filename the user given
'''
dir = abspath(dir)
name = abspath(name)
return name.startswith(join(dir,''))
def getPort(url):
'''
get the port num from the url
'''
name = urlparse(url)[1]
parts = name.split(':')
return int(parts[-1])
class Node:
def __init__(self, url, dirname, secret):
self.url = url
self.dirname = dirname
self.secret = secret
self.known = set()
def query(self, query, history = []):
try:
return self._handle(query)
except UnhandledQuery:
history = history + [self.url]
if len(history) > MAX_HISTORY_LENGTH: raise
return self._broadcast(query,history)
def hello(self,other):
self.known.add(other)
return 0
def fetch(self, query, secret):
if secret != self.secret: raise
result = self.query(query)
f = open(join(self.dirname, query),'w')
f.write(result)
f.close()
return 0
def _start(self):
s = SimpleXMLRPCServer(("",getPort(self.url)),logRequests=False)
s.register_instance(self)
s.serve_forever()
def _handle(self, query):
dir = self.dirname
name = join(dir, query)
if not isfile(name):raise UnhandledQuery
if not inside(dir,name):raise AccessDenied
return open(name).read()
def _broadcast(self, query, history):
for other in self.known.copy():
if other in history: continue
try:
s = ServerProxy(other)
return s.query(query, history)
except Fault, f:
if f.faultCode == UNHANDLED:pass
else: self.known.remove(other)
except:
self.known.remove(other)
raise UnhandledQuery
def main():
url, directory, secret = sys.argv[1:]
n = Node(url,directory,secret)
n._start()
if __name__ == '__main__': main()
首先来看上面的几个常量设置: SimpleXMLRPCServer.allow_reuse_address表示,其所占用的端口可以重用,即如果你强制关闭node server之后再次重启,不会出现端口被占用的情况。
MAX_HISTORY_LENGTH = 6 这个是设置最大的节点长度,因为不能让让节点无休止的搜索下去。
UNHANDLED = 100 ACCESS_DENIED = 200 这俩就是返回码。
然后再来看个node节点的具体流程。 这个段代码的流程这这样的,首先,启动供远程调用的服务器,调用的接口就是Node类。在Node类中有三个方法供远程调用的,一个是hello,一个是fetch还有一个query。hello 这个方法就是添加邻节点信息到当前节点中。而fetch则是用来获取数据的方法,query是节点之间用来交互的。
在fetch方法中,首先判断密码是否正确,然后通过调用自己的query方法查找数据。我们来看query方法,这个方法中,先是调用私有方法_handle本地查找,如果没找到,那么在通过_broadcast接口在所有已知节点中发送广播,这里要注意histroy,每次广播都会传递history这个参数,这个参数的作用有二:一是、防止往重复的节点中发送广播;二是、限制当前所有链接节点的长度。
理解了一个node server的基础功能之后,再来看对server进行管理的控制类代码。
client.py:
.. code:: python
#coding=utf-8
from xmlrpclib import ServerProxy, Fault
from cmd import Cmd
from random import choice
from string import lowercase
from server import Node,UNHANDLED #引入前面的server
from threading import Thread
from time import sleep
import sys
HEAD_START = 0.1
SECRET_LENGTH = 100
def randomString(length):
chars = []
letters = lowercase[:26]
while length > 0:
length -= 1
chars.append(choice(letters))
return ''.join(chars)
class Client(Cmd):
prompt = '> '
def __init__(self, url, dirname, urlfile):
Cmd.__init__(self)
self.secret = randomString(SECRET_LENGTH)
n = Node(url, dirname, self.secret)
t = Thread(target = n._start)
t.setDaemon(1)
t.start()
sleep(HEAD_START)
self.server = ServerProxy(url)
for line in open(urlfile):
line = line.strip()
self.server.hello(line)
def do_fetch(self, arg):
try:
self.server.fetch(arg,self.secret)
except Fault,f:
if f.faultCode != UNHANDLED: raise
print "Couldn't find the file",arg
def do_exit(self,arg):
print
sys.exit()
do_EOR = do_exit
def main():
urlfile, directory, url = sys.argv[1:]
client = Client(url, directory, urlfile)
client.cmdloop()
if __name__ == '__main__':main()
来分析一下这段代码,前面的参数就不看了,很好理解,一开始有一个随机生成密码的函数,做什么用的呢?主要是用来防止别人非法调用该控制所控制的node server的。这密码 我们也不用记,因为我们有client的合法使用权。呵呵。
这段代码的总体作用就是为你提供一个可视的命令行的界面,通过继承cmd这个类,来解析你输入的命令,比如程序运行之后,出现命令提示符,你输入fetch,那么它会调用到do_fetch这个方法中来,并把参数传递进来。
do_fetch这个方法的所用就是调用node server中的fetch方法,获取资源。
另外的一个do_exit很好理解,就是接受exit命令退出程序。
在程序初始化的时候,还有一点需要注意,就是它会读取你urlfile参数传递的文件中的数据,这个里面放的是节点的url地址。读取之后程序会把这些地址加到相邻节点中,供以后访问。不过这个程序还有些不完善的地方就是在程序运行时,如果你修改了url配置的文件,他不会读取你新添加的节点url。不过这个修改很简单,把获取url的代码放到do_fetch中就行了。
在运行程序之前还有一些工作要做。 首先需要建立两个文件夹,A和C,C文件夹里面创建一个文件,B.txt,在A和C所在文件夹中建立urlsA.txt和urlsC.txt文件。里面在urlsA.txt中写入:http://localhost:4243,然后开启两个命令行,
第一个输入:python client.py urlsA.txt A http://localhost:4242 回车,是不是出来提示符了。输入fetch B.txt回车,看到提示Couldn't find the file B.txt。
然后在第二个命令行中输入python client.py urlsC.txt C http://localhost:4243回车。同样输入fetch B.txt回车,是不是没反应。说明该文件存在。接在在第一个命令行中再次输入fetch B.txt看,是否还是提示没找到文件,如果你对代码根据我上面的建议进行了修改的话,就不会出现错误了,如果没有修改,此时你需要把输入exit退出程序,再次重启,然后在fetch B.txt,然后到A文件夹下查看一下,看是不是把B.txt下载到你的文件夹中了。
PS:上面的程序只能传输文本文件,大文件或者其他格式的文件无法传输,刚才研究了一下,使用xmlrpclib这个库中的Binary函数即可,具体使用访问为: 先引入xmlrpclib,import xmlrpclib 在server类的的_handle方法中最后返回的那句代码return open(name).read() 修改为 return xmlrpclib.Binary(open(name,'rb').read()) 再把fetch方法中的f.write(result)修改为f.write(result.data) 另外这句话前面的那个写文件的方式要改为wb。
- from the5fire.com微信公众号:Python程序员杂谈