From 88a206400a2e5b84b7c2aa1429b8463f55ddcd5d Mon Sep 17 00:00:00 2001
From: Alexey Shchepin <alexey@process-one.net>
Date: Fri, 25 Sep 2015 19:47:27 +0300
Subject: Add oauth_access option

---
 src/ejabberd_oauth.erl | 27 ++++++++++++++++++++++-----
 1 file changed, 22 insertions(+), 5 deletions(-)

(limited to 'src/ejabberd_oauth.erl')

diff --git a/src/ejabberd_oauth.erl b/src/ejabberd_oauth.erl
index 0295b9d87..c7138981f 100644
--- a/src/ejabberd_oauth.erl
+++ b/src/ejabberd_oauth.erl
@@ -125,10 +125,25 @@ get_client_identity(Client, Ctx) -> {ok, {Ctx, {client, Client}}}.
 verify_redirection_uri(_, _, Ctx) -> {ok, Ctx}.
 
 authenticate_user({User, Server}, {password, Password} = Ctx) ->
-    case ejabberd_auth:check_password(User, Server, Password) of
-        true ->
-            {ok, {Ctx, {user, User, Server}}};
-        false ->
+    case jlib:make_jid(User, Server, <<"">>) of
+        #jid{} = JID ->
+            Access =
+                ejabberd_config:get_option(
+                  {oauth_access, JID#jid.lserver},
+                  fun(A) when is_atom(A) -> A end,
+                  none),
+            case acl:match_rule(JID#jid.lserver, Access, JID) of
+                allow ->
+                    case ejabberd_auth:check_password(User, Server, Password) of
+                        true ->
+                            {ok, {Ctx, {user, User, Server}}};
+                        false ->
+                            {error, badpass}
+                    end;
+                deny ->
+                    {error, badpass}
+            end;
+        error ->
             {error, badpass}
     end.
 
@@ -470,4 +485,6 @@ logo() ->
 
 opt_type(oauth_expire) ->
     fun(I) when is_integer(I), I >= 0 -> I end;
-opt_type(_) -> [oauth_expire].
+opt_type(oauth_access) ->
+    fun(A) when is_atom(A) -> A end;
+opt_type(_) -> [oauth_expire, oauth_access].
-- 
cgit v1.2.3