发布一个erlang服务端,flash客户端的聊天服务(带安全沙箱解决)[更新]

bs

erlang服务端，主要实现与flash客户端的连接、消息传递服务
自带843端口服务解决flash9以上暴汗的安全沙箱问题

chat.erl

1. 
   
2. -module(chat).
   
3. -export([start/0,manage_clients/1,stop/0,find/1]).
   
4. -record(player, {id,socket,time=none}).
   
5. 
   
6. start() ->
   
7.         chat_server:start(?MODULE,manage_clients).
   
8. 
   
9. stop()->
   
10.         chat_server:stop(?MODULE),
    
11.         timer:sleep(2000),                                                %%等待所有连接断开
    
12.         manage_clients!{exit}.
    
13. 
    
14. manage_clients(Players) ->
    
15.         receive
    
16.             {connect, Socket} ->
    
17.                 Player = #player{
    
18.                         id="guest_"++integer_to_list(random:uniform(100000)),
    
19.                         socket=Socket,
    
20.                         time=time()
    
21.                         },        %%新增记录
    
22.                 io:fwrite("Socket connected: ~w~n", [Socket]),
    
23.                 NewPlayers =  [Player|Players];                                %%累加到列表
    
24. 
    
25.             {disconnect, Socket} ->
    
26.                 io:fwrite("Socket disconnected: ~w~n", [Socket]),
    
27.                 Player = find_Socket(Socket, Players),                        %%查找socket对应的记录
    
28.                 NewPlayers = lists:delete(Player, Players);                %%删除记录
    
29. 
    
30.             {chat,Data} ->
    
31.                 send_data(Players, Data),
    
32.                 NewPlayers=Players;
    
33. 
    
34.             {list,Uname}->
    
35.                 find_player(Uname,Players),
    
36.                 NewPlayers = Players;
    
37. 
    
38.             {exit}->
    
39.                 NewPlayers="",
    
40.                 exit({stop, exit})                                       
    
41.         end,
    
42.         manage_clients(NewPlayers).
    
43. 
    
44. 
    
45. 
    
46. 
    
47. 
    
48. %%对客户端进行广播
    
49. send_data(Players, Data) ->
    
50.   SendData = fun(Player) ->
    
51.     gen_tcp:send(Player#player.socket, Data)                         
    
52.   end,
    
53.   lists:foreach(SendData, Players).
    
54. 
    
55. %%查找指定记录
    
56. find_Socket(Socket, Players) ->
    
57.     {value, Player} = lists:keysearch(Socket, #player.socket, Players),
    
58.     Player.
    
59. 
    
60. find_player(Uname, Players) ->
    
61.     case lists:keysearch(Uname, #player.id, Players) of
    
62.     {value, Player}->
    
63.         io:fwrite("Find a player: ~w~n", [Player]);
    
64.     _->io:fwrite("No found.~n")
    
65.     end.
    
66. 
    
67. 
    
68. %%查找用户
    
69. find(Uname)->
    
70.         client_manager ! {list,Uname}.
    

复制代码

chat_server.erl

1. 
   
2. -module(chat_server).
   
3. -behaviour(gen_server).
   
4. -export([start/2,stop/1]).
   
5. 
   
6. %% gen_server回调
   
7. -export([init/1,handle_call/3,handle_cast/2,handle_info/2,
   
8.         terminate/2,code_change/3]).
   
9. -compile(export_all).
   
10. 
    
11. -define(TCP_OPTIONS, [list, {packet, 0}, {active, false}, {reuseaddr, true},{nodelay, false},{delay_send, true}]).  
    
12. %%状态表
    
13. -record(state, {name,loop,socket}).
    
14. 
    
15. %%回调gen_server创建服务
    
16. start(Name,Loop) ->
    
17.         State=#state{name=Name,loop=Loop},                                %%定为记录
    
18.         gen_server:start_link({local,Name},?MODULE,State,[]).                       
    
19. 
    
20. 
    
21. %%start() ->
    
22. %%        gen_server:start_link({local,?MODULE},?MODULE,[],[]).               
    
23. 
    
24. 
    
25. %%停止服务
    
26. stop(Name)  ->
    
27.         gen_server:call(Name,stop).
    
28. 
    
29. 
    
30. init(State) ->
    
31.         %%维护队列 chat.rar (17.12 KB, 下载次数: 425)
    2009-12-15 20:04 上传

    点击文件名下载附件
    
    
32.         register(State#state.loop, spawn(fun() -> (State#state.name):(State#state.loop)([]) end)),
    
33.         {Tag, LSocket}=gen_tcp:listen(8080, ?TCP_OPTIONS),                        %%默认8080
    
34.         case Tag of                                                                %%创建监听
    
35.                 ok        ->        spawn(fun() -> do_accept(State#state{socket=LSocket}) end);        %%统一接收
    
36.                 error        ->        exit({stop, exit})                                %%出错，退出连接
    
37.         end,
    
38.                                
    
39.         %%与flash通信的安全策略文件传送监听接口
    
40.         {Req, AsSocket}=gen_tcp:listen(843, ?TCP_OPTIONS),
    
41.         case Req of                                                               
    
42.                 ok        ->        spawn(fun() -> do_accept2(AsSocket) end);       
    
43.                 error        ->        exit({stop, exit})                               
    
44.         end,
    
45.         {ok,LSocket}.
    
46. 
    
47. handle_call(stop,_From,Tab)        ->
    
48.         {stop,normal,stopped,Tab}.
    
49. 
    
50. 
    
51. handle_cast(stop,State)                ->{stop, normal, State};
    
52. handle_cast(_Msg,State)                ->{noreply,State}.
    
53. handle_info(_Info,State)        ->{noreply,State}.
    
54. terminate(_Reason,_State)        ->ok.
    
55. code_change(_OldVsn,State,_)        ->{ok,State}.
    
56. 
    
57. 
    
58. 
    
59. 
    
60. 
    
61. 
    
62. 
    
63. %% -----------------------------------
    
64. %% -------------私有函数--------------
    
65. %% -----------------------------------
    
66. 
    
67. 
    
68. 
    
69. %%新建连接,只有新连接时才调用
    
70. do_accept(State) ->
    
71.         case gen_tcp:accept(State#state.socket) of
    
72.                 {ok, Socket}->
    
73.                         %%创建进程处理响应
    
74.                         spawn(fun() -> handle_client(State,Socket) end),
    
75.                         %%创建新连接,持久化
    
76.                         State#state.loop ! {connect, Socket},
    
77.                         do_accept(State);
    
78.                 _->
    
79.                         ok
    
80.         end.       
    
81. 
    
82. 
    
83. 
    
84. %%处理flash安全策略文件的发送服务
    
85. do_accept2(LSocket) ->
    
86.         case gen_tcp:accept(LSocket) of
    
87.                 {ok, Socket}->
    
88.                         spawn(fun() -> put_file(Socket) end),
    
89.                         do_accept2(LSocket);
    
90.                 _->ok
    
91.         end.
    
92. 
    
93. %%短连接策略，连接完关闭
    
94. put_file(Socket)->
    
95.           case gen_tcp:recv(Socket, 0) of
    
96.                 {ok, _} ->
    
97.                         %%判断数据返回策略文件
    
98.                         gen_tcp:send(Socket, "<?xml version=\"1.0\"?>
    
99.                 <cross-domain-policy>
    
100.                 <allow-access-from domain=\"*\" to-ports=\"*\"/>  
     
101.                 </cross-domain-policy>\0");
     
102.                 {error, closed} ->ok
     
103.           end,
     
104.           gen_tcp:close(Socket).                                        %%关闭连接
     
105. 
     
106. 
     
107. 
     
108. 
     
109. handle_client(State,Socket) ->
     
110.           case gen_tcp:recv(Socket, 0) of
     
111.                 {ok, Data} ->
     
112.                         State#state.loop ! {chat, Data},
     
113.                         handle_client(State,Socket);
     
114.                 {error, closed} ->
     
115.                         State#state.loop ! {disconnect, Socket}               
     
116.           end.
     
117. 
     
118. 
     
119. 
     

复制代码

简单的flash客户端，为as3代码：

1. 
   
2. package  
   
3. {
   
4.         import flash.net.Socket;
   
5.         import flash.display.Sprite;
   
6.         import flash.events.*;
   
7.         import flash.text.TextField;
   
8. 
   
9.         [SWF(width=600, height=300)]
   
10.         public class chat extends Sprite
    
11.         {
    
12.                 private var textArea:TextField;
    
13.                 private var input:TextField;
    
14.                 private var label:TextField;
    
15.                 private var socket:Socket=new Socket();       
    
16. 
    
17.                 public function chat()
    
18.                 {
    
19.                         socket.addEventListener(ProgressEvent.SOCKET_DATA, onSocketData);
    
20.                         socket.connect("192.168.10.112", 8080);                                                                //连接到服务端，这里改成服务端的IP和端口
    
21.                         socket.flush();
    
22. 
    
23.                         initUI();
    
24.                         input.addEventListener(KeyboardEvent.KEY_DOWN, keyDownHandler);                //按回车响应
    
25.                         textArea.addEventListener(Event.RENDER, renderHandler);                       
    
26.                 }
    
27.                
    
28.                 private function sendMsg(Msg:String):void
    
29.                 {
    
30.                         socket.writeUTFBytes(Msg);
    
31.                         socket.flush();               
    
32.                 }
    
33.                
    
34.                 private function keyDownHandler(e:KeyboardEvent):void
    
35.                 {
    
36.                         if (13 == e.charCode)
    
37.                         {
    
38.                                 var s:String = input.text ? input.text : "null";
    
39.                                 input.text = "";
    
40.                                 sendMsg(s);
    
41.                         }
    
42.                 }
    
43.                
    
44.                 private function renderHandler(e:Event):void
    
45.                 {
    
46.                         textArea.scrollV = textArea.maxScrollV;
    
47.                 }
    
48.                
    
49.                
    
50.                 //初始化
    
51.                 private function initUI():void
    
52.                 {
    
53.                         textArea = new TextField();
    
54.                         textArea.width = 400;
    
55.                         textArea.height = 150;
    
56.                         textArea.border = true;
    
57.                         textArea.x = 100;
    
58.                         textArea.y = 30;
    
59.                         addChild(textArea);
    
60.                        
    
61.                         input = new TextField();
    
62.                         input.height = 20;
    
63.                         input.width = 350;
    
64.                         input.type = "input";
    
65.                         input.border = true;
    
66.                         input.multiline = false;
    
67.                         input.x = 100;
    
68.                         input.y = 200;
    
69.                         addChild(input);
    
70.                        
    
71.                         label = new TextField();
    
72.                         label.text = "回车发送";
    
73.                         label.height = label.textHeight + 4;
    
74.                         label.width = label.textWidth + 4;
    
75.                         label.mouseEnabled = false;
    
76.                         label.x = 460;
    
77.                         label.y = 200;
    
78.                         addChild(label);
    
79.                 }
    
80. 
    
81.                
    
82.                 // 事件处理回调函数,显示到屏幕上
    
83.                 private function onSocketData(event:ProgressEvent):void
    
84.                 {
    
85.                         while (socket.bytesAvailable)
    
86.                         {
    
87.                                 var s:String = socket.readUTFBytes(socket.bytesAvailable);
    
88.                                 if (textArea.text) s = "\n" + s;
    
89.                                 textArea.appendText(s);                       
    
90.                                 stage.invalidate();
    
91.                         }
    
92.                 }
    
93.         }
    
94. }
    

复制代码

运行：

1> c(chat).
{ok,chat}
2> c(chat_server).
{ok,chat_server}
3> chat:start().  
{ok,<0.435.0>}

.....
n> chat:stop().   
{exit}



更多部署见附件：

[ 本帖最后由 bs 于 2009-12-15 20:04 编辑 ]

dogkyan

多谢BS版主共享好东西啊。

thematice

多谢版主.......

needspeedboy

没有AMF和RPC实用价值也不是太大

bs

为防止粘包

erlang 下设置
define(TCP_OPTIONS, [list, {packet, 2}, {active, false}, {reuseaddr, true},{nodelay, false},{delay_send, true}]).  

as下，使用
readUTF读取
writeUTF发送


不过，与as3安全策略通信时的843端口默认要设置成  {packet, 0} 否则很可能无法通信

[ 本帖最后由 bs 于 2010-1-25 15:04 编辑 ]