使用twisted.web实现代理服务器

简单的实现谷歌的代理：

架构就是下面这么简单。

=================

 my server outside GFW  |    <----------------------> your browser  visit my server at port 8080

=================

代码如下：

#coding=utf-8

from twisted.web import resource, server

from twisted.internet import reactor,endpoints

from twisted.python import log

import urllib2

import urllib

import sys

from twisted.web.util import Redirect

#log.startLogging(sys.stdout)



def get_redirect_url(url, prefix):

    if not url:

        return ''

    index = url.find(prefix)

    if index>-1:

        index2 = url.find('&sa=')

        return url[index:index2]

    return url



class router(resource.Resource):

    def getChild(self, path, request):

        url = r'https://www.google.com'

        if path=="":

            return Index(url)

        else:

            new_url = str(request.uri)

            new_url = get_redirect_url(new_url,'http://')

            new_url = get_redirect_url(new_url,'https://')

            if new_url and (new_url.find('http://')>-1 or new_url.find('https://')>-1):

                return Redirect(new_url)

            return Index(url+ request.uri)



class Index(resource.Resource):

    def __init__(self, url):

        self.url = url



    def render_GET(self, request):

        headers = {'User-Agent':'Mozilla/5.0 (Windows NT 6.1; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/38.0.2125.111 Safari/537.36'}

        req = urllib2.Request(self.url,headers=headers)

        res = urllib2.urlopen(req)

        return res.read()



if __name__=='__main__':

    rsc = router()

    f = server.Site(rsc)

    endpoints.serverFromString(reactor, 'tcp:8080').listen(f)

    reactor.run()

　　

干脆做成启动任务，参考这里：http://www.cnblogs.com/Tommy-Yu/p/4105942.html.

 

发现一个问题，就是，神马链接都给代理了，这样不行，搜索出来的链接得丢出去。改下router的getChild的实现。否则若是你的vps计费，流量得耗光。

解决这个问题，要把链接给redirect出去。方案有两种：

在不同的地方使用不同的方法：

• 在 getChild 中使用 twisted.web.util.Redirect
• 在 render 中使用 twisted.web.util.redirectTo

在 getChild 中：

 

class Index(Resource):

    def getChild(self, path, request):

        from twisted.web.util import Redirect



        return Redirect('/login')

　　

 

在 render render_GET render_POST 中：

class Index(Resource):

    def render(self, request):

        from twisted.web.util import redirectTo

        return redirectTo('/login', request)