Improve compatibility with CockroachDB (#3074)

2 files changed, 27 insertions, 22 deletions

diff --git a/src/ejabberd_sql.erl b/src/ejabberd_sql.erl
index c1d9383a..22151b69 100644
--- a/src/ejabberd_sql.erl
+++ b/src/ejabberd_sql.erl
@@ -44,6 +44,7 @@
 	 escape_like/1,
 	 escape_like_arg/1,
 	 escape_like_arg_circumflex/1,
+         to_string_literal/2,
 	 to_bool/1,
 	 sqlite_db/1,
 	 sqlite_file/1,
@@ -251,6 +252,17 @@ to_array(EscapeFun, Val) ->
     Escaped = lists:join(<<",">>, lists:map(EscapeFun, Val)),
     [<<"{">>, Escaped, <<"}">>].
 
+to_string_literal(odbc, S) ->
+    <<"'", (escape(S))/binary, "'">>;
+to_string_literal(mysql, S) ->
+    <<"'", (escape(S))/binary, "'">>;
+to_string_literal(mssql, S) ->
+    <<"'", (standard_escape(S))/binary, "'">>;
+to_string_literal(sqlite, S) ->
+    <<"'", (standard_escape(S))/binary, "'">>;
+to_string_literal(pgsql, S) ->
+    <<"E'", (escape(S))/binary, "'">>.
+
 encode_term(Term) ->
     escape(list_to_binary(
              erl_prettypr:format(erl_syntax:abstract(Term),
diff --git a/src/mod_mam_sql.erl b/src/mod_mam_sql.erl
index 9e5df26a..654e1f42 100644
--- a/src/mod_mam_sql.erl
+++ b/src/mod_mam_sql.erl
@@ -354,12 +354,7 @@ make_sql_query(User, LServer, MAMQuery, RSM, ExtraUsernames) ->
     WithText = proplists:get_value(withtext, MAMQuery),
     {Max, Direction, ID} = get_max_direction_id(RSM),
     ODBCType = ejabberd_option:sql_type(LServer),
-    Escape =
-        case ODBCType of
-            mssql -> fun ejabberd_sql:standard_escape/1;
-            sqlite -> fun ejabberd_sql:standard_escape/1;
-            _ -> fun ejabberd_sql:escape/1
-        end,
+    ToString = fun(S) -> ejabberd_sql:to_string_literal(ODBCType, S) end,
     LimitClause = if is_integer(Max), Max >= 0, ODBCType /= mssql ->
 			  [<<" limit ">>, integer_to_binary(Max+1)];
 		     true ->
@@ -371,20 +366,18 @@ make_sql_query(User, LServer, MAMQuery, RSM, ExtraUsernames) ->
 			  []
 		  end,
     WithTextClause = if is_binary(WithText), WithText /= <<>> ->
-			     [<<" and match (txt) against ('">>,
-			      Escape(WithText), <<"')">>];
+			     [<<" and match (txt) against (">>,
+			      ToString(WithText), <<")">>];
 			true ->
 			     []
 		     end,
     WithClause = case catch jid:tolower(With) of
 		     {_, _, <<>>} ->
-			 [<<" and bare_peer='">>,
-			  Escape(jid:encode(With)),
-			  <<"'">>];
+			 [<<" and bare_peer=">>,
+			  ToString(jid:encode(With))];
 		     {_, _, _} ->
-			 [<<" and peer='">>,
-			  Escape(jid:encode(With)),
-			  <<"'">>];
+			 [<<" and peer=">>,
+			  ToString(jid:encode(With))];
 		     _ ->
 			 []
 		 end,
@@ -415,23 +408,23 @@ make_sql_query(User, LServer, MAMQuery, RSM, ExtraUsernames) ->
 		    _ ->
 			[]
 		end,
-    SUser = Escape(User),
-    SServer = Escape(LServer),
+    SUser = ToString(User),
+    SServer = ToString(LServer),
 
     HostMatch = case ejabberd_sql:use_new_schema() of
 		    true ->
-			[<<" and server_host='", SServer/binary, "'">>];
+			[<<" and server_host=", SServer/binary>>];
 		    _ ->
 			<<"">>
 		end,
 
     {UserSel, UserWhere} = case ExtraUsernames of
 			       Users when is_list(Users) ->
-				   EscUsers = [<<"'", (Escape(U))/binary, "'">> || U <- [User | Users]],
+				   EscUsers = [ToString(U) || U <- [User | Users]],
 				   {<<" username,">>,
 				    [<<" username in (">>, str:join(EscUsers, <<",">>), <<")">>]};
 			       subscribers_table ->
-				   SJid = Escape(jid:encode({User, LServer, <<>>})),
+				   SJid = ToString(jid:encode({User, LServer, <<>>})),
 				   RoomName = case ODBCType of
 						  sqlite ->
 						      <<"room || '@' || host">>;
@@ -439,11 +432,11 @@ make_sql_query(User, LServer, MAMQuery, RSM, ExtraUsernames) ->
 						      <<"concat(room, '@', host)">>
 					      end,
 				   {<<" username,">>,
-				    [<<" (username = '">>, SUser, <<"'">>,
+				    [<<" (username = ">>, SUser,
 					<<" or username in (select ">>, RoomName,
-					  <<" from muc_room_subscribers where jid='">>, SJid, <<"'">>, HostMatch, <<"))">>]};
+					  <<" from muc_room_subscribers where jid=">>, SJid, HostMatch, <<"))">>]};
 			       _ ->
-				   {<<>>, [<<" username='">>, SUser, <<"'">>]}
+				   {<<>>, [<<" username=">>, SUser]}
 			   end,
 
     Query = [<<"SELECT ">>, TopClause, UserSel,