Sancaros/egs
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

egs_user_model: Add the key_auth function.

Browse Source
This commit is contained in:
Loïc Hoguin 2010-09-27 19:43:01 +02:00
parent e11dc0be25
commit 559e77de37
2 changed files with 15 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
src/egs_login.erl
View File

@ -69,14 +69,9 @@ event(system_game_server_request, State=#state{socket=Socket}) ->
%% If the user is authenticated, send him the character flags list.
%% @todo Remove the put calls when all the send_xxxx are moved out of psu_game and into psu_proto.
event({system_key_auth_request, AuthGID, AuthKey}, State=#state{socket=Socket}) ->
{ok, User} = egs_user_model:read(AuthGID),
{wait_for_authentication, AuthKey} = User#egs_user_model.state,
egs_user_model:key_auth(AuthGID, AuthKey, Socket),
put(socket, Socket),
put(gid, AuthGID),
LID = 1 + mnesia:dirty_update_counter(counters, lobby, 1) rem 1023,
Time = calendar:datetime_to_gregorian_seconds(calendar:universal_time()),
User2 = User#egs_user_model{id=AuthGID, pid=self(), socket=Socket, state=authenticated, time=Time, lid=LID},
egs_user_model:write(User2),
State2 = State#state{gid=AuthGID},
psu_proto:send_0d05(State2),
{ok, egs_char_select, State2};

15
src/egs_user_model.erl
View File

@ -19,7 +19,7 @@
-module(egs_user_model).
-behavior(gen_server).
-export([start_link/0, stop/0, count/0, read/1, select/1, write/1, delete/1]). %% API.
-export([start_link/0, stop/0, count/0, read/1, select/1, write/1, delete/1, key_auth/3]). %% API.
-export([init/1, handle_call/3, handle_cast/2, handle_info/2, terminate/2, code_change/3]). %% gen_server.
%% Use the module name for the server's name and for the table name.
@ -68,6 +68,9 @@ write(User) ->
delete(ID) ->
gen_server:cast(?SERVER, {delete, ID}).
key_auth(GID, AuthKey, Socket) ->
gen_server:call(?SERVER, {key_auth, GID, AuthKey, Socket}).
%% gen_server
init([]) ->
@ -110,6 +113,16 @@ handle_call({select, {neighbors, User}}, _From, State) ->
])),
{reply, {ok, List}, State};
%% @todo Handle LIDs properly, so not here.
handle_call({key_auth, GID, AuthKey, Socket}, {Pid, _Tag}, State) ->
{atomic, [User]} = mnesia:transaction(fun() -> mnesia:read({?TABLE, GID}) end),
{wait_for_authentication, AuthKey} = User#egs_user_model.state,
LID = 1 + mnesia:dirty_update_counter(counters, lobby, 1) rem 1023,
Time = calendar:datetime_to_gregorian_seconds(calendar:universal_time()),
User2 = User#egs_user_model{pid=Pid, socket=Socket, state=authenticated, time=Time, lid=LID},
mnesia:transaction(fun() -> mnesia:write(User2) end),
{reply, ok, State};
handle_call(stop, _From, State) ->
{stop, normal, stopped, State};