使用java编写SmartFoxServer自定义安全验证登录扩展
最近接触的东西有点杂,在写SmartFoxServer(以下简称SFS)服务端扩展时发现actionscript竟然只支持1.0,无奈只好用从来没有接触过的java来编写,参考官方cookbook及手册,成功实现了自定义安全验证登录扩展。
系统用的是CentOS5.2(x86_64),数据库是Oracle 11g,SFS版本为1.6.6。
-
配置
数据库DEMO中的USERS表结构:
CREATE TABLE "DEMO"."USERS" ( "USERID" NUMBER(38,0) NOT NULL ENABLE, "USERNAME" VARCHAR2(20) NOT NULL ENABLE, "USERPWD" VARCHAR2(32) NOT NULL ENABLE, "USEREMAIL" VARCHAR2(100), CONSTRAINT "TABLE1_PK" PRIMARY KEY ("USERID") ENABLE ) ;
将连接java连接oracle的驱动(oracle/product/11.1.0/db_1/jdbc/lib/ojdbc6.jar,这里我用的是JDK1.6,所以用ojdbc6.jar,根据自己的需要复制驱动)复制到(SFS_PRO_1.6.6/jre/lib/ext/)下,然后修改SFS根目录下的config.xml,在Zones节点内加入如下Zone配置:
oracle.jdbc.driver.OracleDriver jdbc:oracle:thin:@localhost:1521:DEMO demo demo < ![CDATA[SELECT COUNT(*) FROM USERS]]> 10 10 fail 5000
这里根据自己数据库配置来设置驱动、连接字符串、用户名、密码,如果用的是mysql,那么驱动和连接字符串应该如下:
com.mysql.jdbc.Driver jdbc:mysql://localhost:3306/DEMO
-
编写扩展
在写扩展之前,需要了解一下安全验证的机制,借用一下官方的图示:
第一步:从服务器获取随机码,把用户输入密码用md5加密,然后再把随机码和加密后的md5码再用md5加密,并发送给服务器。
第二步:服务器端从数据库取出经过md5加密的用户密码,同样用md5再加密一次随机码和取出的md5密码,然后和客户端发来的hash过的字符进行对比,来验证登录。理解了这次,就可以开始编写Zone配置里的扩展demo_login了,关于扩展的具体说明,看官方手册就可以了,这里给出源码借参考:
import java.nio.channels.SocketChannel; import java.util.*; import it.gotoandplay.smartfoxserver.crypto.MD5; import it.gotoandplay.smartfoxserver.data.*; import it.gotoandplay.smartfoxserver.db.*; import it.gotoandplay.smartfoxserver.exceptions.*; import it.gotoandplay.smartfoxserver.extensions.*; import it.gotoandplay.smartfoxserver.lib.ActionscriptObject; import it.gotoandplay.smartfoxserver.events.InternalEventObject; public class demo_login extends AbstractExtension { private ExtensionHelper helper; private Zone currentZone; private DbManager db; public void init() { helper = ExtensionHelper.instance(); this.currentZone = helper.getZone(this.getOwnerZone()); } public void destroy() { trace("Extension destroyed"); } public void handleRequest(String cmd, ActionscriptObject ao, User u, int fromRoom) { // Your code here } public void handleRequest(String cmd, String params[], User u, int fromRoom) { // Your code here } /** * Handle Internal Server Events * * @param ieo the event object */ public void handleInternalEvent(InternalEventObject ieo) { if (ieo.getEventName().equals("loginRequest")) { // 根据配置文件里的DatabaseManager配置获得当前Zone的数据库操作对象 db = this.currentZone.dbManager; ActionscriptObject response = new ActionscriptObject(); User loginUser = null; // 获取用户名 String nick = ieo.getParam("nick"); // 获取客户端利用服务器加密字符串和用户密码加密后MD5密码 String ClientPassword = ieo.getParam("pass"); // 获取服务器socket通道 SocketChannel chan = (SocketChannel)ieo.getObject("chan"); // 根据socket通道生成加密字符 String ServerRandom = helper.getSecretKey(chan); // 获取数据库中用户的密码 ArrayList arrList = db.executeQuery("SELECT USERPWD FROM USERS WHERE USERNAME='"+nick+"'"); String ServerPassword = ""; boolean IsLogin = false; if (arrList.size() < = 0) { response.put("_cmd", "loginKO"); response.put("err", "用户名不存在,登录失败。"); } else { // 获取数据库密码 DataRow dr = arrList.get(0); String dbpass = dr.getItem("USERPWD"); // 根据服务器加密字符和数据库用户密码,生成服务器端的混合MD5密码 ServerPassword = MD5.instance().getHash(ServerRandom + dbpass); IsLogin = (ClientPassword.equals(ServerPassword)) ? true : false; if (IsLogin) { try { // 登录成功 loginUser = helper.canLogin(nick, ClientPassword, chan, this.currentZone.getName()); response.put("_cmd", "loginOK"); response.put("id", String.valueOf(loginUser.getUserId())); response.put("name", loginUser.getName()); } catch (LoginException e) { // 登录失败 response.put("_cmd", "loginKO"); response.put("err", e.getMessage()); } } else { response.put("_cmd", "loginKO"); response.put("err", "认证失败."); } } LinkedList linkedlist = new LinkedList(); linkedlist.add(chan); // 返回结果给客户端 this.sendResponse(response, -1, null, linkedlist); if (IsLogin) helper.sendRoomList(chan); } } }
这里有个关键点需要注意:配置文件中Zone标签的属性customLogin要设置成true才能使服务端响应loginRequest事件。
-
客户端处理
服务端的验证结果信息将通过客户端SmartFoxClient对象的onExtensionResponse返回,下面是该事件监听函数:
private function onExtensionResponseHandler(e:SFSEvent):void { // 数据传输类型 var type:String = e.params.type; // 用Object类型接收数据 var data:Object = e.params.dataObj; var cmd:String = data._cmd; switch (cmd) { case "loginKO": this._labWelcome.setText(data.err); break; case "loginOK": _SFSClient.myUserId = data.id; _SFSClient.myUserName = data.name; initResources(); break; default: this._labWelcome.setText(data.err); break; } }