Chinaunix首页 | 论坛 | 博客

一片旭日天空boyml.blog.chinaunix.net

首页　| 　博文目录　| 　关于我

lovealamo

  • 博客访问： 355079
  • 博文数量： 90
  • 博客积分： 2017
  • 博客等级： 大尉
  • 技术积分： 615
  • 用 户 组： 普通用户
  • 注册时间： 2009-10-19 08:10
文章分类

全部博文（90）

文章存档

2012年（4）

2011年（74）

2010年（11）

2009年（1）

我的朋友
最近访客
推荐博文
CS架构下OAuth认证的实现－附DEMO

分类：

2011-05-22 18:04:51

原文地址：CS架构下OAuth认证的实现－附DEMO 作者：hawkli

研究了一段时间OAuth，在SNS网站之间，OAuth用的很多，手机类应用也不少，但是在桌面程序中很少有使用到OAuth，至于CS架构的程序，至少我是没发现这样的应用了。
新浪的OAuth认证中，有提出一种桌面应用的OAuth认证实现，其原理是使用PIN码，即用户在登录成功之后记录下新浪反馈的一个PIN码，然后在客户端中输入这个PIN值，再用这个进行access token等后续的验证，这个方案相当的搓，试想，一个桌面应用程序，在登录完之后，再要输入一个PIN码，太不方便了，用户体验很差。

我研究了较长时间，最终设计出一个方案，可以实现CS架构模型下使用OAuth认证，这个架构有点一小复杂，也不是一般的程序能用的上的，但这个思路也许能给看本文的人一个思路。
在我的实现中，客户端程序发起登录请求，发起请求时产生一个唯一ID，请注意这个ID必须是绝对唯一的，我的DEMO中使用了UUID代替，我认为对UUID+特定字串做MD5处理，应该能达到要求。然后将这个唯一ID传给server端，同时打开一个IE界面，也将这个唯一ID传递给负责进行OAuth认证的我方网站程序。
server端在收到唯一ID后，保持和客户端的通讯，并可以不做反馈（在我的DEMO中为了简便，设计为客户端每1秒钟发起一次查询请求，server端收到后检查是否通过认证，并立即返回）
我方网站程序是核心，在接收到唯一ID后，构造request Token，向第三方网站发起Oauth认证，引导用户IE浏览器跳转至第三方网站，并最终令用户成功在第三方网站登录完成。
在我方网站最终登录完成之后作access token和唯一ID、网站用户名的绑定关系，并通过socket将唯一ID和网站用户名的对应关系发送给server端，server端在接收到这个信息后更新状态，并通知客户端此用户已经成功登录，即可以走后续业务流程了。

这个思路其实并不复杂，但走的路线比较曲折，附件是我用python写的DEMO，思路就是如上，client.py是模拟了一个客户端请求，其中也实现了模拟发成功登录请求的模块；server.py是模拟接入服务端；p.py是模拟网站程序，在成功OAuth认证完成之后向server.py发送成功登录的socket请求。
请注意p.py需要使用python -m CGIHTTPServer方式运行的，不懂的人自行看python的书吧。

本文中的代码如何运行其实并不重要，重要的是思路，程序我在python2.7,linux/windows平台下都测试通过了。
不熟悉本文的OAuth协议的，请自行google。

client.py
  1. #-*- coding:utf-8-*-
  2. '''
  3. Created on 2011-5-15

  5. @author: HAWK.Li
  6. '''
  7. import uuid
  8. import sys,os,time,subprocess
  9. import socket

  11. class Client():
  12.     """封装客户端核心方法
  13.     """
  14.     def login(self,tempid):
  15.         """登录"""
  16.         
  17.      
  18.         try:
  19.             self.sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
  20.             self.sock.connect((host, port))
  21.             msg = 'login '+tempid
  22.             self.sock.sendall(msg)
  23.             self.response = self.sock.recv(sockSize)
  24.             #分割字串，result[0]为标识，result[1]为返回结果
  25.             result = self.response.split(' ')
  26.             if result[0] == 'success':
  27.                 print "Login success! tempid(%s) is user(%s)!"%(tempid,result[1])
  28.                 print "connect close."
  29.                 global FLAG
  30.                 FLAG = False
  31.             elif result[0] == 'again':
  32.                 print "sleep 1 second and retry..."
  33.             elif result[0] == 'first':
  34.                 print "First Login..."
  35.             else:
  36.                 print "failed..."
  37.             #print self.response
  38.             return self.response
  39.         except:
  40.             print "登录出错！".decode('utf-8').encode("gbk")
  41.             print "Client.login Error:"+str(sys.exc_info()[0])+str(sys.exc_info()[1])
  42.         
  43.     def finished(self,tempid):
  44.         """模拟成功的socket通知"""

  46.         try:
  47.             self.sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
  48.             self.sock.connect((host, port))
  49.             username = "Andy"
  50.             msg = 'finished '+tempid+" "+username
  51.             self.sock.sendall(msg)
  52.             self.response = self.sock.recv(sockSize)
  53.             #分割字串，result[0]为标识，result[1]为返回结果
  54.             result = self.response.split(' ')
  55.             print result
  56.         except:
  57.             print "模拟通知出错！".decode('utf-8').encode("gbk")
  58.             print "finished Error:"+str(sys.exc_info()[0])+str(sys.exc_info()[1])
  59.         
  60.     def listid(self):
  61.         """列出服务器上的uuid和用户名的对应关系表"""
  62.         
  63.      
  64.         try:
  65.             self.sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
  66.             self.sock.connect((host, port))
  67.             msg = 'list '
  68.             self.sock.sendall(msg)
  69.             self.response = self.sock.recv(sockSize)
  70.             result = self.response
  71.             print result
  72.         except:
  73.             print "列表出错！".decode('utf-8').encode("gbk")
  74.             print "listid Error:"+str(sys.exc_info()[0])+str(sys.exc_info()[1])

  76.     def closeSock(self):
  77.         self.sock.close()
  78.         

  80. if __name__ == "__main__":
  81.     global host,port,client,sockSize
  82.     sockSize = 64000
  83.     uid = str(uuid.uuid1())
  84.     host = '127.0.0.1'
  85.     port = int('1234')
  86.     
  87.     client = Client()
  88.     #这里的requestToken.py文件并不重要，我也没写
  89.     p = subprocess.Popen('"C:\Program Files\Internet Explorer\iexplore.exe" '+uid)

  91.     #退出标记
  92.     FLAG = True

  94.     if len(sys.argv) >= 2:
  95.         #登录成功的标记
  96.         if sys.argv[1] == 'finished':
  97.             existid = sys.argv[2]
  98.             r = client.finished(existid)
  99.         elif sys.argv[1] == 'list':
  100.             r = client.listid()
  101.     else:
  102.         print 'Send tempid('+uid+') to LoginServer...'
  103.         for i in range(120):
  104.             if FLAG:
  105.                 r = client.login(uid)
  106.                 time.sleep(1)
  107.             else:
  108.                 break

server.py
  1. #-*- coding:utf-8-*-
  2. '''
  3. Created on 2011-5-15

  5. @author: HAWK.Li
  6. '''
  7. import uuid
  8. import sys,os,time,subprocess
  9. import SocketServer,socket
  10. import time,datetime
  11. import threading


  14. class MyTCPHandler(SocketServer.StreamRequestHandler):
  15.     def handle(self):
  16.         uid = None

  18.         while True:
  19.             self.data = self.request.recv(sockSize)
  20.             cur_thread = threading.currentThread()
  21.             cmd = self.data
  22.             if cmd == None or len(cmd) == 0:
  23.                 break;
  24.             recvStr = cmd.split(' ')
  25.             #print recvStr
  26. # try:
  27.             if cmd.startswith('login'):
  28.                 firstflag = True
  29.                 uid = recvStr[1]
  30.                 print "RECV from ", self.client_address[0],uid
  31.                 for key in ids.keys():
  32.                     #检查登录的ID是否已经在列表中
  33.                     if key == uid:
  34.                         firstflag = False
  35.                         #检查登录的ID是否已经通过用户名验证，是None则告诉客户端还没验证通过
  36.                         if ids[key] == None:
  37.                             result = "again %s"%(uid)
  38.                         else:
  39.                             result = "success %s"%(ids[key])
  40.                         continue
  41.                 if firstflag:
  42.                     #说明这个客户端第一次登录
  43.                     result = "first %s"%(uid)
  44.                     username=None
  45.                     ids[uid] = username

  47.                 print result
  48.                 print ids
  49.                 self.request.sendall(result)
  50.             #收到用户名成功验证的通知
  51.             elif cmd.startswith('finished'):
  52.                 uid = recvStr[1]
  53.                 username = recvStr[2]
  54.                 ids[uid] = username
  55.                 self.request.sendall('OK')
  56.             elif cmd.startswith('list'):
  57.                 self.request.sendall(str(ids))
  58.             else:
  59.                 result = 'error'
  60. # self.wfile.write(result)
  61. # self.wfile.write('\n')
  62. # except:
  63. # print 'error'
  64.                 break
  65.         try:
  66.             if uid != None:
  67.                 ids.remove(uid)
  68.         except:
  69.             pass
  70.         #print uid, ' closed.'

  72. class ThreadedTCPServer(SocketServer.ThreadingMixIn, SocketServer.TCPServer):
  73.     #socket.setdefaulttimeout(5)
  74.     pass

  76. if __name__ == "__main__":
  77.     HOST = ''
  78.     PORT = int('1234')
  79.     sockSize = 64000
  80.     ids = {'id':None}
  81.     print 'login server start...'
  82.     server = ThreadedTCPServer((HOST, PORT), MyTCPHandler)
  83.     server_thread = threading.Thread(target=server.serve_forever)
  84.     server_thread.setDaemon(True)
  85.     server_thread.start()
  86.     server.serve_forever()

f.py
这个是一个CGI程序，需要使用python -m CGIHTTPServer方式运行，并放在cgi-bin目录下，给他执行权限
  1. #!/usr/bin/env python

  3. import cgi
  4. import uuid
  5. import sys,os,time,subprocess
  6. import socket

  8. #CGIAuth DEMO
  9. header = 'Content-Type: text/html\n\n'

  11. formhtml = '''<br></span></li><li>CGIAuth Demo

  12. USER



  14. Enter your uuid:


  17. USERNAME:

  18. %s

  19. '''


  21. fradio = ' %s\n'

  23. def showForm():
  24.     friends = ''
  25.     for i in [0, 10, 25, 50, 100]:
  26.         checked = ''
  27.         if i == 0:
  28.             checked = 'CHECKED'
  29.         friends = friends + fradio % \
  30.             (str(i), checked, str(i))

  32.     print header + formhtml % (friends)

  34. reshtml = '''<br></span></li><li>CGIAuth Demo

  35. Users : %s


  36. Your uuid is: %s


  37. Your username is: %s .
  38. '''

  40. def doResults(who, howmany):
  41.     print header + reshtml % (who, who, howmany)
  42.     global host,port,client,sockSize
  43.     sockSize = 64000
  44.     #uid = str(uuid.uuid1())
  45.     uid = who
  46.     host = '127.0.0.1'
  47.     port = int('1234')
  48.     client = Client()
  49.     users = howmany
  50.     r = client.finished(uid,users)
  51.     print r

  53. def process():
  54.     form = cgi.FieldStorage()
  55.     if form.has_key('person'):
  56.         who = form['person'].value
  57.     else:
  58.         who = 'NEW USER'

  60.     if form.has_key('howmany'):
  61.         howmany = form['howmany'].value
  62.     else:
  63.         howmany = 0

  65.     if form.has_key('action'):
  66.         doResults(who, howmany)
  67.     else:
  68.         showForm()

  70. class Client():
  71.         
  72.     def finished(self,tempid,username):
  73.         try:
  74.             self.sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
  75.             self.sock.connect((host, port))
  76.             msg = 'finished '+tempid+" "+username
  77.             self.sock.sendall(msg)
  78.             self.response = self.sock.recv(sockSize)
  79.             result = self.response.split(' ')
  80.             return result
  81.         except:
  82.             print "finished Error:"+str(sys.exc_info()[0])+str(sys.exc_info()[1])
  83.         

  85.     def closeSock(self):
  86.         self.sock.close()
  87.         

  89. if __name__ == '__main__':
  90.     process()






阅读(820) | 评论(0) | 转发(0) |
0

上一篇：Linux的启动SD卡的格式化方法

下一篇：KVM在嵌入式μCLinux操作系统上的移植

给主人留下些什么吧！~~
关于我们 | 关于IT168 | 联系方式 | 广告合作 | 法律声明 | 免费注册

Copyright 2001-2010 ChinaUnix.net All Rights Reserved 北京皓辰网域网络信息技术有限公司. 版权所有

感谢所有关心和支持过ChinaUnix的朋友们

16024965号-6