summaryrefslogtreecommitdiff
path: root/src/rabbit_upgrade_functions.erl
blob: 3baac37ebb357741061d2e7e6c115533830b70e3 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
%%   The contents of this file are subject to the Mozilla Public License
%%   Version 1.1 (the "License"); you may not use this file except in
%%   compliance with the License. You may obtain a copy of the License at
%%   http://www.mozilla.org/MPL/
%%
%%   Software distributed under the License is distributed on an "AS IS"
%%   basis, WITHOUT WARRANTY OF ANY KIND, either express or implied. See the
%%   License for the specific language governing rights and limitations
%%   under the License.
%%
%%   The Original Code is RabbitMQ.
%%
%%   The Initial Developers of the Original Code are Rabbit Technologies Ltd.
%%
%%   Copyright (C) 2010 Rabbit Technologies Ltd.
%%
%%   All Rights Reserved.
%%
%%   Contributor(s): ______________________________________.
%%
-module(rabbit_upgrade_functions).

-include("rabbit.hrl").

-compile([export_all]).

-rabbit_upgrade({remove_user_scope,  []}).
-rabbit_upgrade({hash_passwords,     []}).
-rabbit_upgrade({add_ip_to_listener, []}).

%% -------------------------------------------------------------------

-ifdef(use_specs).

-spec(remove_user_scope/0  :: () -> 'ok').
-spec(hash_passwords/0     :: () -> 'ok').
-spec(add_ip_to_listener/0 :: () -> 'ok').

-endif.

%%--------------------------------------------------------------------

remove_user_scope() ->
    mnesia(
      rabbit_user_permission,
      fun (Perm = #user_permission{
             permission = {permission,
                           _Scope, Conf, Write, Read}}) ->
              Perm#user_permission{
                permission = #permission{configure = Conf,
                                         write     = Write,
                                         read      = Read}}
      end,
      record_info(fields, user_permission)).

hash_passwords() ->
    mnesia(
      rabbit_user,
      fun ({user, Username, Password, IsAdmin}) ->
              Hash = rabbit_access_control:hash_password(Password),
              #user{username      = Username,
                    password_hash = Hash,
                    is_admin      = IsAdmin}
      end,
      record_info(fields, user)).

add_ip_to_listener() ->
    mnesia(
      rabbit_listener,
      fun ({listener, Node, Protocol, Host, Port}) ->
              #listener{node       = Node,
                        protocol   = Protocol,
                        host       = Host,
                        ip_address = {0,0,0,0},
                        port       = Port}
      end,
      record_info(fields, listener)).

%%--------------------------------------------------------------------

mnesia(TableName, Fun, RecordInfo) ->
    {atomic, ok} = mnesia:transform_table(TableName, Fun, RecordInfo),
    ok.