blob: f10a0c89b3f4e542b3901790504a212d612741ef [file] [log] [blame]
Linus Torvalds1da177e2005-04-16 15:20:36 -07001/*
2 * Copyright (C) 2003 Sistina Software.
3 * Copyright (C) 2004 Red Hat, Inc. All rights reserved.
4 *
5 * Module Author: Heinz Mauelshagen
6 *
7 * This file is released under the GPL.
8 *
9 * Path selector registration.
10 */
11
12#include "dm.h"
13#include "dm-path-selector.h"
14
15#include <linux/slab.h>
16
17struct ps_internal {
18 struct path_selector_type pst;
19
20 struct list_head list;
21 long use;
22};
23
24#define pst_to_psi(__pst) container_of((__pst), struct ps_internal, pst)
25
26static LIST_HEAD(_path_selectors);
27static DECLARE_RWSEM(_ps_lock);
28
Alasdair G Kergon5e198d92005-05-05 16:16:09 -070029static struct ps_internal *__find_path_selector_type(const char *name)
Linus Torvalds1da177e2005-04-16 15:20:36 -070030{
31 struct ps_internal *psi;
32
33 list_for_each_entry(psi, &_path_selectors, list) {
34 if (!strcmp(name, psi->pst.name))
35 return psi;
36 }
37
38 return NULL;
39}
40
41static struct ps_internal *get_path_selector(const char *name)
42{
43 struct ps_internal *psi;
44
45 down_read(&_ps_lock);
46 psi = __find_path_selector_type(name);
47 if (psi) {
48 if ((psi->use == 0) && !try_module_get(psi->pst.module))
49 psi = NULL;
50 else
51 psi->use++;
52 }
53 up_read(&_ps_lock);
54
55 return psi;
56}
57
58struct path_selector_type *dm_get_path_selector(const char *name)
59{
60 struct ps_internal *psi;
61
62 if (!name)
63 return NULL;
64
65 psi = get_path_selector(name);
66 if (!psi) {
67 request_module("dm-%s", name);
68 psi = get_path_selector(name);
69 }
70
71 return psi ? &psi->pst : NULL;
72}
73
74void dm_put_path_selector(struct path_selector_type *pst)
75{
76 struct ps_internal *psi;
77
78 if (!pst)
79 return;
80
81 down_read(&_ps_lock);
82 psi = __find_path_selector_type(pst->name);
83 if (!psi)
84 goto out;
85
86 if (--psi->use == 0)
87 module_put(psi->pst.module);
88
Eric Sesterhennc163c722006-03-26 18:21:58 +020089 BUG_ON(psi->use < 0);
Linus Torvalds1da177e2005-04-16 15:20:36 -070090
91out:
92 up_read(&_ps_lock);
93}
94
95static struct ps_internal *_alloc_path_selector(struct path_selector_type *pst)
96{
97 struct ps_internal *psi = kmalloc(sizeof(*psi), GFP_KERNEL);
98
99 if (psi) {
100 memset(psi, 0, sizeof(*psi));
101 psi->pst = *pst;
102 }
103
104 return psi;
105}
106
107int dm_register_path_selector(struct path_selector_type *pst)
108{
109 int r = 0;
110 struct ps_internal *psi = _alloc_path_selector(pst);
111
112 if (!psi)
113 return -ENOMEM;
114
115 down_write(&_ps_lock);
116
117 if (__find_path_selector_type(pst->name)) {
118 kfree(psi);
119 r = -EEXIST;
120 } else
121 list_add(&psi->list, &_path_selectors);
122
123 up_write(&_ps_lock);
124
125 return r;
126}
127
128int dm_unregister_path_selector(struct path_selector_type *pst)
129{
130 struct ps_internal *psi;
131
132 down_write(&_ps_lock);
133
134 psi = __find_path_selector_type(pst->name);
135 if (!psi) {
136 up_write(&_ps_lock);
137 return -EINVAL;
138 }
139
140 if (psi->use) {
141 up_write(&_ps_lock);
142 return -ETXTBSY;
143 }
144
145 list_del(&psi->list);
146
147 up_write(&_ps_lock);
148
149 kfree(psi);
150
151 return 0;
152}
153
154EXPORT_SYMBOL_GPL(dm_register_path_selector);
155EXPORT_SYMBOL_GPL(dm_unregister_path_selector);