Thomas Gleixner | 2874c5f | 2019-05-27 08:55:01 +0200 | [diff] [blame] | 1 | // SPDX-License-Identifier: GPL-2.0-or-later |
David Howells | ec26815 | 2007-04-26 15:49:28 -0700 | [diff] [blame] | 2 | /* AFS cell and server record management |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 3 | * |
David Howells | 989782d | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 4 | * Copyright (C) 2002, 2017 Red Hat, Inc. All Rights Reserved. |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 5 | * Written by David Howells (dhowells@redhat.com) |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 6 | */ |
| 7 | |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 8 | #include <linux/slab.h> |
David Howells | 00d3b7a | 2007-04-26 15:57:07 -0700 | [diff] [blame] | 9 | #include <linux/key.h> |
| 10 | #include <linux/ctype.h> |
Wang Lei | 07567a5 | 2010-08-04 15:16:38 +0100 | [diff] [blame] | 11 | #include <linux/dns_resolver.h> |
Alexey Dobriyan | e8edc6e | 2007-05-21 01:22:52 +0400 | [diff] [blame] | 12 | #include <linux/sched.h> |
David Howells | 3838d3e | 2017-11-02 15:27:47 +0000 | [diff] [blame] | 13 | #include <linux/inet.h> |
David Howells | 0da0b7f | 2018-06-15 15:19:22 +0100 | [diff] [blame] | 14 | #include <linux/namei.h> |
David Howells | 00d3b7a | 2007-04-26 15:57:07 -0700 | [diff] [blame] | 15 | #include <keys/rxrpc-type.h> |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 16 | #include "internal.h" |
| 17 | |
David Howells | fe342cf | 2018-04-09 21:12:31 +0100 | [diff] [blame] | 18 | static unsigned __read_mostly afs_cell_gc_delay = 10; |
David Howells | ded2f4c | 2018-10-20 00:57:57 +0100 | [diff] [blame] | 19 | static unsigned __read_mostly afs_cell_min_ttl = 10 * 60; |
| 20 | static unsigned __read_mostly afs_cell_max_ttl = 24 * 60 * 60; |
David Howells | dca54a7 | 2020-10-13 20:51:59 +0100 | [diff] [blame] | 21 | static atomic_t cell_debug_id; |
David Howells | 989782d | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 22 | |
David Howells | 286377f6 | 2020-10-15 11:05:01 +0100 | [diff] [blame] | 23 | static void afs_queue_cell_manager(struct afs_net *); |
David Howells | 88c853c | 2019-07-23 11:24:59 +0100 | [diff] [blame] | 24 | static void afs_manage_cell_work(struct work_struct *); |
David Howells | 989782d | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 25 | |
| 26 | static void afs_dec_cells_outstanding(struct afs_net *net) |
| 27 | { |
| 28 | if (atomic_dec_and_test(&net->cells_outstanding)) |
Peter Zijlstra | ab1fbe3 | 2018-03-15 11:42:28 +0100 | [diff] [blame] | 29 | wake_up_var(&net->cells_outstanding); |
David Howells | 989782d | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 30 | } |
| 31 | |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 32 | /* |
David Howells | 989782d | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 33 | * Set the cell timer to fire after a given delay, assuming it's not already |
| 34 | * set for an earlier time. |
| 35 | */ |
| 36 | static void afs_set_cell_timer(struct afs_net *net, time64_t delay) |
| 37 | { |
| 38 | if (net->live) { |
| 39 | atomic_inc(&net->cells_outstanding); |
| 40 | if (timer_reduce(&net->cells_timer, jiffies + delay * HZ)) |
| 41 | afs_dec_cells_outstanding(net); |
David Howells | 286377f6 | 2020-10-15 11:05:01 +0100 | [diff] [blame] | 42 | } else { |
| 43 | afs_queue_cell_manager(net); |
David Howells | 989782d | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 44 | } |
| 45 | } |
| 46 | |
| 47 | /* |
David Howells | 92e3cc9 | 2020-10-09 14:11:58 +0100 | [diff] [blame] | 48 | * Look up and get an activation reference on a cell record. The caller must |
| 49 | * hold net->cells_lock at least read-locked. |
David Howells | 989782d | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 50 | */ |
David Howells | 92e3cc9 | 2020-10-09 14:11:58 +0100 | [diff] [blame] | 51 | static struct afs_cell *afs_find_cell_locked(struct afs_net *net, |
David Howells | dca54a7 | 2020-10-13 20:51:59 +0100 | [diff] [blame] | 52 | const char *name, unsigned int namesz, |
| 53 | enum afs_cell_trace reason) |
David Howells | 989782d | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 54 | { |
| 55 | struct afs_cell *cell = NULL; |
| 56 | struct rb_node *p; |
David Howells | 92e3cc9 | 2020-10-09 14:11:58 +0100 | [diff] [blame] | 57 | int n; |
David Howells | 989782d | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 58 | |
| 59 | _enter("%*.*s", namesz, namesz, name); |
| 60 | |
| 61 | if (name && namesz == 0) |
| 62 | return ERR_PTR(-EINVAL); |
| 63 | if (namesz > AFS_MAXCELLNAME) |
| 64 | return ERR_PTR(-ENAMETOOLONG); |
| 65 | |
David Howells | 92e3cc9 | 2020-10-09 14:11:58 +0100 | [diff] [blame] | 66 | if (!name) { |
| 67 | cell = net->ws_cell; |
| 68 | if (!cell) |
| 69 | return ERR_PTR(-EDESTADDRREQ); |
David Howells | 88c853c | 2019-07-23 11:24:59 +0100 | [diff] [blame] | 70 | goto found; |
David Howells | 92e3cc9 | 2020-10-09 14:11:58 +0100 | [diff] [blame] | 71 | } |
David Howells | 989782d | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 72 | |
David Howells | 92e3cc9 | 2020-10-09 14:11:58 +0100 | [diff] [blame] | 73 | p = net->cells.rb_node; |
| 74 | while (p) { |
| 75 | cell = rb_entry(p, struct afs_cell, net_node); |
David Howells | 989782d | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 76 | |
David Howells | 92e3cc9 | 2020-10-09 14:11:58 +0100 | [diff] [blame] | 77 | n = strncasecmp(cell->name, name, |
| 78 | min_t(size_t, cell->name_len, namesz)); |
| 79 | if (n == 0) |
| 80 | n = cell->name_len - namesz; |
| 81 | if (n < 0) |
| 82 | p = p->rb_left; |
| 83 | else if (n > 0) |
| 84 | p = p->rb_right; |
| 85 | else |
| 86 | goto found; |
| 87 | } |
David Howells | 989782d | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 88 | |
David Howells | 92e3cc9 | 2020-10-09 14:11:58 +0100 | [diff] [blame] | 89 | return ERR_PTR(-ENOENT); |
David Howells | 989782d | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 90 | |
David Howells | 92e3cc9 | 2020-10-09 14:11:58 +0100 | [diff] [blame] | 91 | found: |
David Howells | dca54a7 | 2020-10-13 20:51:59 +0100 | [diff] [blame] | 92 | return afs_use_cell(cell, reason); |
David Howells | 92e3cc9 | 2020-10-09 14:11:58 +0100 | [diff] [blame] | 93 | } |
David Howells | 989782d | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 94 | |
David Howells | 88c853c | 2019-07-23 11:24:59 +0100 | [diff] [blame] | 95 | /* |
| 96 | * Look up and get an activation reference on a cell record. |
| 97 | */ |
David Howells | 92e3cc9 | 2020-10-09 14:11:58 +0100 | [diff] [blame] | 98 | struct afs_cell *afs_find_cell(struct afs_net *net, |
David Howells | dca54a7 | 2020-10-13 20:51:59 +0100 | [diff] [blame] | 99 | const char *name, unsigned int namesz, |
| 100 | enum afs_cell_trace reason) |
David Howells | 92e3cc9 | 2020-10-09 14:11:58 +0100 | [diff] [blame] | 101 | { |
| 102 | struct afs_cell *cell; |
David Howells | 989782d | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 103 | |
David Howells | 92e3cc9 | 2020-10-09 14:11:58 +0100 | [diff] [blame] | 104 | down_read(&net->cells_lock); |
David Howells | dca54a7 | 2020-10-13 20:51:59 +0100 | [diff] [blame] | 105 | cell = afs_find_cell_locked(net, name, namesz, reason); |
David Howells | 92e3cc9 | 2020-10-09 14:11:58 +0100 | [diff] [blame] | 106 | up_read(&net->cells_lock); |
| 107 | return cell; |
David Howells | 989782d | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 108 | } |
| 109 | |
| 110 | /* |
| 111 | * Set up a cell record and fill in its name, VL server address list and |
David Howells | 00d3b7a | 2007-04-26 15:57:07 -0700 | [diff] [blame] | 112 | * allocate an anonymous key |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 113 | */ |
David Howells | 989782d | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 114 | static struct afs_cell *afs_alloc_cell(struct afs_net *net, |
| 115 | const char *name, unsigned int namelen, |
David Howells | 0a5143f | 2018-10-20 00:57:57 +0100 | [diff] [blame] | 116 | const char *addresses) |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 117 | { |
David Howells | ca1cbbd | 2019-05-07 15:30:34 +0100 | [diff] [blame] | 118 | struct afs_vlserver_list *vllist; |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 119 | struct afs_cell *cell; |
David Howells | 989782d | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 120 | int i, ret; |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 121 | |
David Howells | 989782d | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 122 | ASSERT(name); |
| 123 | if (namelen == 0) |
| 124 | return ERR_PTR(-EINVAL); |
Wang Lei | 07567a5 | 2010-08-04 15:16:38 +0100 | [diff] [blame] | 125 | if (namelen > AFS_MAXCELLNAME) { |
| 126 | _leave(" = -ENAMETOOLONG"); |
David Howells | 00d3b7a | 2007-04-26 15:57:07 -0700 | [diff] [blame] | 127 | return ERR_PTR(-ENAMETOOLONG); |
Wang Lei | 07567a5 | 2010-08-04 15:16:38 +0100 | [diff] [blame] | 128 | } |
David Howells | a45ea48 | 2020-01-26 01:02:53 +0000 | [diff] [blame] | 129 | |
| 130 | /* Prohibit cell names that contain unprintable chars, '/' and '@' or |
| 131 | * that begin with a dot. This also precludes "@cell". |
| 132 | */ |
| 133 | if (name[0] == '.') |
David Howells | 37ab636 | 2018-04-06 14:17:23 +0100 | [diff] [blame] | 134 | return ERR_PTR(-EINVAL); |
David Howells | a45ea48 | 2020-01-26 01:02:53 +0000 | [diff] [blame] | 135 | for (i = 0; i < namelen; i++) { |
| 136 | char ch = name[i]; |
| 137 | if (!isprint(ch) || ch == '/' || ch == '@') |
| 138 | return ERR_PTR(-EINVAL); |
| 139 | } |
David Howells | 00d3b7a | 2007-04-26 15:57:07 -0700 | [diff] [blame] | 140 | |
David Howells | 0a5143f | 2018-10-20 00:57:57 +0100 | [diff] [blame] | 141 | _enter("%*.*s,%s", namelen, namelen, name, addresses); |
David Howells | 989782d | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 142 | |
| 143 | cell = kzalloc(sizeof(struct afs_cell), GFP_KERNEL); |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 144 | if (!cell) { |
| 145 | _leave(" = -ENOMEM"); |
David Howells | 08e0e7c | 2007-04-26 15:55:03 -0700 | [diff] [blame] | 146 | return ERR_PTR(-ENOMEM); |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 147 | } |
| 148 | |
David Howells | 719fdd3 | 2020-06-24 17:00:24 +0100 | [diff] [blame] | 149 | cell->name = kmalloc(namelen + 1, GFP_KERNEL); |
| 150 | if (!cell->name) { |
| 151 | kfree(cell); |
| 152 | return ERR_PTR(-ENOMEM); |
| 153 | } |
| 154 | |
David Howells | f044c88 | 2017-11-02 15:27:45 +0000 | [diff] [blame] | 155 | cell->net = net; |
David Howells | 989782d | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 156 | cell->name_len = namelen; |
| 157 | for (i = 0; i < namelen; i++) |
| 158 | cell->name[i] = tolower(name[i]); |
David Howells | 719fdd3 | 2020-06-24 17:00:24 +0100 | [diff] [blame] | 159 | cell->name[i] = 0; |
David Howells | 989782d | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 160 | |
David Howells | 88c853c | 2019-07-23 11:24:59 +0100 | [diff] [blame] | 161 | atomic_set(&cell->ref, 1); |
| 162 | atomic_set(&cell->active, 0); |
| 163 | INIT_WORK(&cell->manager, afs_manage_cell_work); |
David Howells | 2032596 | 2020-04-30 01:03:49 +0100 | [diff] [blame] | 164 | cell->volumes = RB_ROOT; |
| 165 | INIT_HLIST_HEAD(&cell->proc_volumes); |
| 166 | seqlock_init(&cell->volume_lock); |
| 167 | cell->fs_servers = RB_ROOT; |
| 168 | seqlock_init(&cell->fs_lock); |
David Howells | 6e0e99d | 2021-09-02 16:43:10 +0100 | [diff] [blame] | 169 | INIT_LIST_HEAD(&cell->fs_open_mmaps); |
| 170 | init_rwsem(&cell->fs_open_mmaps_lock); |
David Howells | 0a5143f | 2018-10-20 00:57:57 +0100 | [diff] [blame] | 171 | rwlock_init(&cell->vl_servers_lock); |
David Howells | 8a070a9 | 2020-04-25 10:26:02 +0100 | [diff] [blame] | 172 | cell->flags = (1 << AFS_CELL_FL_CHECK_ALIAS); |
David Howells | 4d9df98 | 2017-11-02 15:27:47 +0000 | [diff] [blame] | 173 | |
David Howells | ca1cbbd | 2019-05-07 15:30:34 +0100 | [diff] [blame] | 174 | /* Provide a VL server list, filling it in if we were given a list of |
| 175 | * addresses to use. |
David Howells | 989782d | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 176 | */ |
David Howells | 0a5143f | 2018-10-20 00:57:57 +0100 | [diff] [blame] | 177 | if (addresses) { |
David Howells | 0a5143f | 2018-10-20 00:57:57 +0100 | [diff] [blame] | 178 | vllist = afs_parse_text_addrs(net, |
| 179 | addresses, strlen(addresses), ':', |
| 180 | VL_SERVICE, AFS_VL_PORT); |
| 181 | if (IS_ERR(vllist)) { |
| 182 | ret = PTR_ERR(vllist); |
David Howells | 8b2a464 | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 183 | goto parse_failed; |
| 184 | } |
David Howells | 989782d | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 185 | |
David Howells | d5c32c8 | 2019-05-07 15:06:36 +0100 | [diff] [blame] | 186 | vllist->source = DNS_RECORD_FROM_CONFIG; |
| 187 | vllist->status = DNS_LOOKUP_NOT_DONE; |
David Howells | 989782d | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 188 | cell->dns_expiry = TIME64_MAX; |
David Howells | ded2f4c | 2018-10-20 00:57:57 +0100 | [diff] [blame] | 189 | } else { |
David Howells | ca1cbbd | 2019-05-07 15:30:34 +0100 | [diff] [blame] | 190 | ret = -ENOMEM; |
| 191 | vllist = afs_alloc_vlserver_list(0); |
| 192 | if (!vllist) |
| 193 | goto error; |
David Howells | d5c32c8 | 2019-05-07 15:06:36 +0100 | [diff] [blame] | 194 | vllist->source = DNS_RECORD_UNAVAILABLE; |
| 195 | vllist->status = DNS_LOOKUP_NOT_DONE; |
David Howells | ded2f4c | 2018-10-20 00:57:57 +0100 | [diff] [blame] | 196 | cell->dns_expiry = ktime_get_real_seconds(); |
Wang Lei | 07567a5 | 2010-08-04 15:16:38 +0100 | [diff] [blame] | 197 | } |
| 198 | |
David Howells | ca1cbbd | 2019-05-07 15:30:34 +0100 | [diff] [blame] | 199 | rcu_assign_pointer(cell->vl_servers, vllist); |
| 200 | |
David Howells | d5c32c8 | 2019-05-07 15:06:36 +0100 | [diff] [blame] | 201 | cell->dns_source = vllist->source; |
| 202 | cell->dns_status = vllist->status; |
| 203 | smp_store_release(&cell->dns_lookup_count, 1); /* vs source/status */ |
David Howells | 88c853c | 2019-07-23 11:24:59 +0100 | [diff] [blame] | 204 | atomic_inc(&net->cells_outstanding); |
David Howells | dca54a7 | 2020-10-13 20:51:59 +0100 | [diff] [blame] | 205 | cell->debug_id = atomic_inc_return(&cell_debug_id); |
| 206 | trace_afs_cell(cell->debug_id, 1, 0, afs_cell_trace_alloc); |
David Howells | d5c32c8 | 2019-05-07 15:06:36 +0100 | [diff] [blame] | 207 | |
David Howells | 00d3b7a | 2007-04-26 15:57:07 -0700 | [diff] [blame] | 208 | _leave(" = %p", cell); |
| 209 | return cell; |
| 210 | |
David Howells | 8b2a464 | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 211 | parse_failed: |
| 212 | if (ret == -EINVAL) |
| 213 | printk(KERN_ERR "kAFS: bad VL server IP address\n"); |
David Howells | ca1cbbd | 2019-05-07 15:30:34 +0100 | [diff] [blame] | 214 | error: |
David Howells | 719fdd3 | 2020-06-24 17:00:24 +0100 | [diff] [blame] | 215 | kfree(cell->name); |
David Howells | 00d3b7a | 2007-04-26 15:57:07 -0700 | [diff] [blame] | 216 | kfree(cell); |
| 217 | _leave(" = %d", ret); |
| 218 | return ERR_PTR(ret); |
| 219 | } |
| 220 | |
| 221 | /* |
David Howells | 989782d | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 222 | * afs_lookup_cell - Look up or create a cell record. |
David Howells | f044c88 | 2017-11-02 15:27:45 +0000 | [diff] [blame] | 223 | * @net: The network namespace |
David Howells | 989782d | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 224 | * @name: The name of the cell. |
| 225 | * @namesz: The strlen of the cell name. |
| 226 | * @vllist: A colon/comma separated list of numeric IP addresses or NULL. |
| 227 | * @excl: T if an error should be given if the cell name already exists. |
| 228 | * |
| 229 | * Look up a cell record by name and query the DNS for VL server addresses if |
| 230 | * needed. Note that that actual DNS query is punted off to the manager thread |
| 231 | * so that this function can return immediately if interrupted whilst allowing |
| 232 | * cell records to be shared even if not yet fully constructed. |
David Howells | 00d3b7a | 2007-04-26 15:57:07 -0700 | [diff] [blame] | 233 | */ |
David Howells | 989782d | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 234 | struct afs_cell *afs_lookup_cell(struct afs_net *net, |
| 235 | const char *name, unsigned int namesz, |
| 236 | const char *vllist, bool excl) |
David Howells | 00d3b7a | 2007-04-26 15:57:07 -0700 | [diff] [blame] | 237 | { |
David Howells | 989782d | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 238 | struct afs_cell *cell, *candidate, *cursor; |
| 239 | struct rb_node *parent, **pp; |
David Howells | d5c32c8 | 2019-05-07 15:06:36 +0100 | [diff] [blame] | 240 | enum afs_cell_state state; |
David Howells | 989782d | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 241 | int ret, n; |
David Howells | 00d3b7a | 2007-04-26 15:57:07 -0700 | [diff] [blame] | 242 | |
David Howells | 989782d | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 243 | _enter("%s,%s", name, vllist); |
David Howells | 00d3b7a | 2007-04-26 15:57:07 -0700 | [diff] [blame] | 244 | |
David Howells | 989782d | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 245 | if (!excl) { |
David Howells | dca54a7 | 2020-10-13 20:51:59 +0100 | [diff] [blame] | 246 | cell = afs_find_cell(net, name, namesz, afs_cell_trace_use_lookup); |
Gustavo A. R. Silva | 6832795 | 2017-11-17 16:40:32 -0600 | [diff] [blame] | 247 | if (!IS_ERR(cell)) |
David Howells | 989782d | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 248 | goto wait_for_cell; |
David Howells | 00d3b7a | 2007-04-26 15:57:07 -0700 | [diff] [blame] | 249 | } |
| 250 | |
David Howells | 989782d | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 251 | /* Assume we're probably going to create a cell and preallocate and |
| 252 | * mostly set up a candidate record. We can then use this to stash the |
| 253 | * name, the net namespace and VL server addresses. |
| 254 | * |
| 255 | * We also want to do this before we hold any locks as it may involve |
| 256 | * upcalling to userspace to make DNS queries. |
| 257 | */ |
| 258 | candidate = afs_alloc_cell(net, name, namesz, vllist); |
| 259 | if (IS_ERR(candidate)) { |
| 260 | _leave(" = %ld", PTR_ERR(candidate)); |
| 261 | return candidate; |
| 262 | } |
| 263 | |
| 264 | /* Find the insertion point and check to see if someone else added a |
| 265 | * cell whilst we were allocating. |
| 266 | */ |
David Howells | 92e3cc9 | 2020-10-09 14:11:58 +0100 | [diff] [blame] | 267 | down_write(&net->cells_lock); |
David Howells | 989782d | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 268 | |
| 269 | pp = &net->cells.rb_node; |
| 270 | parent = NULL; |
| 271 | while (*pp) { |
| 272 | parent = *pp; |
| 273 | cursor = rb_entry(parent, struct afs_cell, net_node); |
| 274 | |
| 275 | n = strncasecmp(cursor->name, name, |
| 276 | min_t(size_t, cursor->name_len, namesz)); |
| 277 | if (n == 0) |
| 278 | n = cursor->name_len - namesz; |
| 279 | if (n < 0) |
| 280 | pp = &(*pp)->rb_left; |
| 281 | else if (n > 0) |
| 282 | pp = &(*pp)->rb_right; |
| 283 | else |
| 284 | goto cell_already_exists; |
| 285 | } |
| 286 | |
| 287 | cell = candidate; |
| 288 | candidate = NULL; |
David Howells | 88c853c | 2019-07-23 11:24:59 +0100 | [diff] [blame] | 289 | atomic_set(&cell->active, 2); |
David Howells | dca54a7 | 2020-10-13 20:51:59 +0100 | [diff] [blame] | 290 | trace_afs_cell(cell->debug_id, atomic_read(&cell->ref), 2, afs_cell_trace_insert); |
David Howells | 989782d | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 291 | rb_link_node_rcu(&cell->net_node, parent, pp); |
| 292 | rb_insert_color(&cell->net_node, &net->cells); |
David Howells | 92e3cc9 | 2020-10-09 14:11:58 +0100 | [diff] [blame] | 293 | up_write(&net->cells_lock); |
David Howells | 989782d | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 294 | |
David Howells | dca54a7 | 2020-10-13 20:51:59 +0100 | [diff] [blame] | 295 | afs_queue_cell(cell, afs_cell_trace_get_queue_new); |
David Howells | 989782d | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 296 | |
| 297 | wait_for_cell: |
David Howells | dca54a7 | 2020-10-13 20:51:59 +0100 | [diff] [blame] | 298 | trace_afs_cell(cell->debug_id, atomic_read(&cell->ref), atomic_read(&cell->active), |
| 299 | afs_cell_trace_wait); |
David Howells | 989782d | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 300 | _debug("wait_for_cell"); |
David Howells | d5c32c8 | 2019-05-07 15:06:36 +0100 | [diff] [blame] | 301 | wait_var_event(&cell->state, |
| 302 | ({ |
| 303 | state = smp_load_acquire(&cell->state); /* vs error */ |
David Howells | 1d0e850 | 2020-10-16 13:21:14 +0100 | [diff] [blame] | 304 | state == AFS_CELL_ACTIVE || state == AFS_CELL_REMOVED; |
David Howells | d5c32c8 | 2019-05-07 15:06:36 +0100 | [diff] [blame] | 305 | })); |
David Howells | 989782d | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 306 | |
David Howells | d5c32c8 | 2019-05-07 15:06:36 +0100 | [diff] [blame] | 307 | /* Check the state obtained from the wait check. */ |
David Howells | 1d0e850 | 2020-10-16 13:21:14 +0100 | [diff] [blame] | 308 | if (state == AFS_CELL_REMOVED) { |
David Howells | 989782d | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 309 | ret = cell->error; |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 310 | goto error; |
David Howells | 989782d | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 311 | } |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 312 | |
David Howells | 989782d | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 313 | _leave(" = %p [cell]", cell); |
David Howells | 08e0e7c | 2007-04-26 15:55:03 -0700 | [diff] [blame] | 314 | return cell; |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 315 | |
David Howells | 989782d | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 316 | cell_already_exists: |
| 317 | _debug("cell exists"); |
| 318 | cell = cursor; |
| 319 | if (excl) { |
| 320 | ret = -EEXIST; |
| 321 | } else { |
David Howells | dca54a7 | 2020-10-13 20:51:59 +0100 | [diff] [blame] | 322 | afs_use_cell(cursor, afs_cell_trace_use_lookup); |
David Howells | 989782d | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 323 | ret = 0; |
wanglei | bec5eb6 | 2010-08-11 09:38:04 +0100 | [diff] [blame] | 324 | } |
David Howells | 92e3cc9 | 2020-10-09 14:11:58 +0100 | [diff] [blame] | 325 | up_write(&net->cells_lock); |
David Howells | 88c853c | 2019-07-23 11:24:59 +0100 | [diff] [blame] | 326 | if (candidate) |
David Howells | dca54a7 | 2020-10-13 20:51:59 +0100 | [diff] [blame] | 327 | afs_put_cell(candidate, afs_cell_trace_put_candidate); |
David Howells | 989782d | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 328 | if (ret == 0) |
| 329 | goto wait_for_cell; |
David Howells | 8b2a464 | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 330 | goto error_noput; |
David Howells | 989782d | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 331 | error: |
David Howells | dca54a7 | 2020-10-13 20:51:59 +0100 | [diff] [blame] | 332 | afs_unuse_cell(net, cell, afs_cell_trace_unuse_lookup); |
David Howells | 8b2a464 | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 333 | error_noput: |
David Howells | 989782d | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 334 | _leave(" = %d [error]", ret); |
| 335 | return ERR_PTR(ret); |
David Howells | ec26815 | 2007-04-26 15:49:28 -0700 | [diff] [blame] | 336 | } |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 337 | |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 338 | /* |
David Howells | 08e0e7c | 2007-04-26 15:55:03 -0700 | [diff] [blame] | 339 | * set the root cell information |
| 340 | * - can be called with a module parameter string |
| 341 | * - can be called from a write to /proc/fs/afs/rootcell |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 342 | */ |
David Howells | 989782d | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 343 | int afs_cell_init(struct afs_net *net, const char *rootcell) |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 344 | { |
| 345 | struct afs_cell *old_root, *new_root; |
David Howells | 989782d | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 346 | const char *cp, *vllist; |
| 347 | size_t len; |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 348 | |
| 349 | _enter(""); |
| 350 | |
| 351 | if (!rootcell) { |
| 352 | /* module is loaded with no parameters, or built statically. |
| 353 | * - in the future we might initialize cell DB here. |
| 354 | */ |
David Howells | 08e0e7c | 2007-04-26 15:55:03 -0700 | [diff] [blame] | 355 | _leave(" = 0 [no root]"); |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 356 | return 0; |
| 357 | } |
| 358 | |
| 359 | cp = strchr(rootcell, ':'); |
David Howells | 989782d | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 360 | if (!cp) { |
Wang Lei | 07567a5 | 2010-08-04 15:16:38 +0100 | [diff] [blame] | 361 | _debug("kAFS: no VL server IP addresses specified"); |
David Howells | 989782d | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 362 | vllist = NULL; |
| 363 | len = strlen(rootcell); |
| 364 | } else { |
| 365 | vllist = cp + 1; |
| 366 | len = cp - rootcell; |
| 367 | } |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 368 | |
| 369 | /* allocate a cell record for the root cell */ |
David Howells | 989782d | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 370 | new_root = afs_lookup_cell(net, rootcell, len, vllist, false); |
David Howells | 08e0e7c | 2007-04-26 15:55:03 -0700 | [diff] [blame] | 371 | if (IS_ERR(new_root)) { |
| 372 | _leave(" = %ld", PTR_ERR(new_root)); |
| 373 | return PTR_ERR(new_root); |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 374 | } |
| 375 | |
David Howells | 17814ae | 2018-04-09 21:12:31 +0100 | [diff] [blame] | 376 | if (!test_and_set_bit(AFS_CELL_FL_NO_GC, &new_root->flags)) |
David Howells | dca54a7 | 2020-10-13 20:51:59 +0100 | [diff] [blame] | 377 | afs_use_cell(new_root, afs_cell_trace_use_pin); |
David Howells | 989782d | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 378 | |
David Howells | 08e0e7c | 2007-04-26 15:55:03 -0700 | [diff] [blame] | 379 | /* install the new cell */ |
David Howells | 92e3cc9 | 2020-10-09 14:11:58 +0100 | [diff] [blame] | 380 | down_write(&net->cells_lock); |
David Howells | dca54a7 | 2020-10-13 20:51:59 +0100 | [diff] [blame] | 381 | afs_see_cell(new_root, afs_cell_trace_see_ws); |
David Howells | 92e3cc9 | 2020-10-09 14:11:58 +0100 | [diff] [blame] | 382 | old_root = net->ws_cell; |
| 383 | net->ws_cell = new_root; |
| 384 | up_write(&net->cells_lock); |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 385 | |
David Howells | dca54a7 | 2020-10-13 20:51:59 +0100 | [diff] [blame] | 386 | afs_unuse_cell(net, old_root, afs_cell_trace_unuse_ws); |
David Howells | 08e0e7c | 2007-04-26 15:55:03 -0700 | [diff] [blame] | 387 | _leave(" = 0"); |
| 388 | return 0; |
David Howells | ec26815 | 2007-04-26 15:49:28 -0700 | [diff] [blame] | 389 | } |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 390 | |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 391 | /* |
David Howells | 989782d | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 392 | * Update a cell's VL server address list from the DNS. |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 393 | */ |
David Howells | d5c32c8 | 2019-05-07 15:06:36 +0100 | [diff] [blame] | 394 | static int afs_update_cell(struct afs_cell *cell) |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 395 | { |
David Howells | d5c32c8 | 2019-05-07 15:06:36 +0100 | [diff] [blame] | 396 | struct afs_vlserver_list *vllist, *old = NULL, *p; |
David Howells | ded2f4c | 2018-10-20 00:57:57 +0100 | [diff] [blame] | 397 | unsigned int min_ttl = READ_ONCE(afs_cell_min_ttl); |
| 398 | unsigned int max_ttl = READ_ONCE(afs_cell_max_ttl); |
| 399 | time64_t now, expiry = 0; |
David Howells | d5c32c8 | 2019-05-07 15:06:36 +0100 | [diff] [blame] | 400 | int ret = 0; |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 401 | |
David Howells | 989782d | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 402 | _enter("%s", cell->name); |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 403 | |
David Howells | 0a5143f | 2018-10-20 00:57:57 +0100 | [diff] [blame] | 404 | vllist = afs_dns_query(cell, &expiry); |
David Howells | d5c32c8 | 2019-05-07 15:06:36 +0100 | [diff] [blame] | 405 | if (IS_ERR(vllist)) { |
| 406 | ret = PTR_ERR(vllist); |
| 407 | |
| 408 | _debug("%s: fail %d", cell->name, ret); |
| 409 | if (ret == -ENOMEM) |
| 410 | goto out_wake; |
| 411 | |
| 412 | ret = -ENOMEM; |
| 413 | vllist = afs_alloc_vlserver_list(0); |
| 414 | if (!vllist) |
| 415 | goto out_wake; |
| 416 | |
| 417 | switch (ret) { |
| 418 | case -ENODATA: |
| 419 | case -EDESTADDRREQ: |
| 420 | vllist->status = DNS_LOOKUP_GOT_NOT_FOUND; |
| 421 | break; |
| 422 | case -EAGAIN: |
| 423 | case -ECONNREFUSED: |
| 424 | vllist->status = DNS_LOOKUP_GOT_TEMP_FAILURE; |
| 425 | break; |
| 426 | default: |
| 427 | vllist->status = DNS_LOOKUP_GOT_LOCAL_FAILURE; |
| 428 | break; |
| 429 | } |
| 430 | } |
| 431 | |
| 432 | _debug("%s: got list %d %d", cell->name, vllist->source, vllist->status); |
| 433 | cell->dns_status = vllist->status; |
David Howells | ded2f4c | 2018-10-20 00:57:57 +0100 | [diff] [blame] | 434 | |
| 435 | now = ktime_get_real_seconds(); |
| 436 | if (min_ttl > max_ttl) |
| 437 | max_ttl = min_ttl; |
| 438 | if (expiry < now + min_ttl) |
| 439 | expiry = now + min_ttl; |
| 440 | else if (expiry > now + max_ttl) |
| 441 | expiry = now + max_ttl; |
| 442 | |
David Howells | d5c32c8 | 2019-05-07 15:06:36 +0100 | [diff] [blame] | 443 | _debug("%s: status %d", cell->name, vllist->status); |
| 444 | if (vllist->source == DNS_RECORD_UNAVAILABLE) { |
| 445 | switch (vllist->status) { |
| 446 | case DNS_LOOKUP_GOT_NOT_FOUND: |
David Howells | ded2f4c | 2018-10-20 00:57:57 +0100 | [diff] [blame] | 447 | /* The DNS said that the cell does not exist or there |
| 448 | * weren't any addresses to be had. |
| 449 | */ |
David Howells | ded2f4c | 2018-10-20 00:57:57 +0100 | [diff] [blame] | 450 | cell->dns_expiry = expiry; |
David Howells | 8b2a464 | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 451 | break; |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 452 | |
David Howells | d5c32c8 | 2019-05-07 15:06:36 +0100 | [diff] [blame] | 453 | case DNS_LOOKUP_BAD: |
| 454 | case DNS_LOOKUP_GOT_LOCAL_FAILURE: |
| 455 | case DNS_LOOKUP_GOT_TEMP_FAILURE: |
| 456 | case DNS_LOOKUP_GOT_NS_FAILURE: |
David Howells | 8b2a464 | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 457 | default: |
David Howells | ded2f4c | 2018-10-20 00:57:57 +0100 | [diff] [blame] | 458 | cell->dns_expiry = now + 10; |
David Howells | 8b2a464 | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 459 | break; |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 460 | } |
David Howells | 8b2a464 | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 461 | } else { |
David Howells | 8b2a464 | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 462 | cell->dns_expiry = expiry; |
David Howells | 8b2a464 | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 463 | } |
| 464 | |
David Howells | d5c32c8 | 2019-05-07 15:06:36 +0100 | [diff] [blame] | 465 | /* Replace the VL server list if the new record has servers or the old |
| 466 | * record doesn't. |
| 467 | */ |
| 468 | write_lock(&cell->vl_servers_lock); |
| 469 | p = rcu_dereference_protected(cell->vl_servers, true); |
| 470 | if (vllist->nr_servers > 0 || p->nr_servers == 0) { |
| 471 | rcu_assign_pointer(cell->vl_servers, vllist); |
| 472 | cell->dns_source = vllist->source; |
| 473 | old = p; |
| 474 | } |
| 475 | write_unlock(&cell->vl_servers_lock); |
| 476 | afs_put_vlserverlist(cell->net, old); |
wanglei | bec5eb6 | 2010-08-11 09:38:04 +0100 | [diff] [blame] | 477 | |
David Howells | d5c32c8 | 2019-05-07 15:06:36 +0100 | [diff] [blame] | 478 | out_wake: |
| 479 | smp_store_release(&cell->dns_lookup_count, |
| 480 | cell->dns_lookup_count + 1); /* vs source/status */ |
| 481 | wake_up_var(&cell->dns_lookup_count); |
| 482 | _leave(" = %d", ret); |
| 483 | return ret; |
David Howells | ec26815 | 2007-04-26 15:49:28 -0700 | [diff] [blame] | 484 | } |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 485 | |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 486 | /* |
David Howells | 989782d | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 487 | * Destroy a cell record |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 488 | */ |
David Howells | 989782d | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 489 | static void afs_cell_destroy(struct rcu_head *rcu) |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 490 | { |
David Howells | 989782d | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 491 | struct afs_cell *cell = container_of(rcu, struct afs_cell, rcu); |
David Howells | 88c853c | 2019-07-23 11:24:59 +0100 | [diff] [blame] | 492 | struct afs_net *net = cell->net; |
| 493 | int u; |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 494 | |
David Howells | 989782d | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 495 | _enter("%p{%s}", cell, cell->name); |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 496 | |
David Howells | 88c853c | 2019-07-23 11:24:59 +0100 | [diff] [blame] | 497 | u = atomic_read(&cell->ref); |
| 498 | ASSERTCMP(u, ==, 0); |
David Howells | dca54a7 | 2020-10-13 20:51:59 +0100 | [diff] [blame] | 499 | trace_afs_cell(cell->debug_id, u, atomic_read(&cell->active), afs_cell_trace_free); |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 500 | |
David Howells | 88c853c | 2019-07-23 11:24:59 +0100 | [diff] [blame] | 501 | afs_put_vlserverlist(net, rcu_access_pointer(cell->vl_servers)); |
David Howells | dca54a7 | 2020-10-13 20:51:59 +0100 | [diff] [blame] | 502 | afs_unuse_cell(net, cell->alias_of, afs_cell_trace_unuse_alias); |
David Howells | 00d3b7a | 2007-04-26 15:57:07 -0700 | [diff] [blame] | 503 | key_put(cell->anonymous_key); |
David Howells | 719fdd3 | 2020-06-24 17:00:24 +0100 | [diff] [blame] | 504 | kfree(cell->name); |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 505 | kfree(cell); |
| 506 | |
David Howells | 88c853c | 2019-07-23 11:24:59 +0100 | [diff] [blame] | 507 | afs_dec_cells_outstanding(net); |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 508 | _leave(" [destroyed]"); |
David Howells | ec26815 | 2007-04-26 15:49:28 -0700 | [diff] [blame] | 509 | } |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 510 | |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 511 | /* |
David Howells | 989782d | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 512 | * Queue the cell manager. |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 513 | */ |
David Howells | 989782d | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 514 | static void afs_queue_cell_manager(struct afs_net *net) |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 515 | { |
David Howells | 989782d | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 516 | int outstanding = atomic_inc_return(&net->cells_outstanding); |
| 517 | |
| 518 | _enter("%d", outstanding); |
| 519 | |
| 520 | if (!queue_work(afs_wq, &net->cells_manager)) |
| 521 | afs_dec_cells_outstanding(net); |
| 522 | } |
| 523 | |
| 524 | /* |
| 525 | * Cell management timer. We have an increment on cells_outstanding that we |
| 526 | * need to pass along to the work item. |
| 527 | */ |
| 528 | void afs_cells_timer(struct timer_list *timer) |
| 529 | { |
| 530 | struct afs_net *net = container_of(timer, struct afs_net, cells_timer); |
| 531 | |
| 532 | _enter(""); |
| 533 | if (!queue_work(afs_wq, &net->cells_manager)) |
| 534 | afs_dec_cells_outstanding(net); |
| 535 | } |
| 536 | |
| 537 | /* |
David Howells | 8b2a464 | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 538 | * Get a reference on a cell record. |
| 539 | */ |
David Howells | dca54a7 | 2020-10-13 20:51:59 +0100 | [diff] [blame] | 540 | struct afs_cell *afs_get_cell(struct afs_cell *cell, enum afs_cell_trace reason) |
David Howells | 8b2a464 | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 541 | { |
David Howells | dca54a7 | 2020-10-13 20:51:59 +0100 | [diff] [blame] | 542 | int u; |
| 543 | |
David Howells | 88c853c | 2019-07-23 11:24:59 +0100 | [diff] [blame] | 544 | if (atomic_read(&cell->ref) <= 0) |
| 545 | BUG(); |
| 546 | |
David Howells | dca54a7 | 2020-10-13 20:51:59 +0100 | [diff] [blame] | 547 | u = atomic_inc_return(&cell->ref); |
| 548 | trace_afs_cell(cell->debug_id, u, atomic_read(&cell->active), reason); |
David Howells | 8b2a464 | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 549 | return cell; |
| 550 | } |
| 551 | |
| 552 | /* |
David Howells | 989782d | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 553 | * Drop a reference on a cell record. |
| 554 | */ |
David Howells | dca54a7 | 2020-10-13 20:51:59 +0100 | [diff] [blame] | 555 | void afs_put_cell(struct afs_cell *cell, enum afs_cell_trace reason) |
David Howells | 88c853c | 2019-07-23 11:24:59 +0100 | [diff] [blame] | 556 | { |
| 557 | if (cell) { |
David Howells | dca54a7 | 2020-10-13 20:51:59 +0100 | [diff] [blame] | 558 | unsigned int debug_id = cell->debug_id; |
David Howells | 88c853c | 2019-07-23 11:24:59 +0100 | [diff] [blame] | 559 | unsigned int u, a; |
| 560 | |
David Howells | dca54a7 | 2020-10-13 20:51:59 +0100 | [diff] [blame] | 561 | a = atomic_read(&cell->active); |
David Howells | 88c853c | 2019-07-23 11:24:59 +0100 | [diff] [blame] | 562 | u = atomic_dec_return(&cell->ref); |
David Howells | dca54a7 | 2020-10-13 20:51:59 +0100 | [diff] [blame] | 563 | trace_afs_cell(debug_id, u, a, reason); |
David Howells | 88c853c | 2019-07-23 11:24:59 +0100 | [diff] [blame] | 564 | if (u == 0) { |
| 565 | a = atomic_read(&cell->active); |
| 566 | WARN(a != 0, "Cell active count %u > 0\n", a); |
| 567 | call_rcu(&cell->rcu, afs_cell_destroy); |
| 568 | } |
| 569 | } |
| 570 | } |
| 571 | |
| 572 | /* |
| 573 | * Note a cell becoming more active. |
| 574 | */ |
David Howells | dca54a7 | 2020-10-13 20:51:59 +0100 | [diff] [blame] | 575 | struct afs_cell *afs_use_cell(struct afs_cell *cell, enum afs_cell_trace reason) |
David Howells | 88c853c | 2019-07-23 11:24:59 +0100 | [diff] [blame] | 576 | { |
David Howells | dca54a7 | 2020-10-13 20:51:59 +0100 | [diff] [blame] | 577 | int u, a; |
| 578 | |
David Howells | 88c853c | 2019-07-23 11:24:59 +0100 | [diff] [blame] | 579 | if (atomic_read(&cell->ref) <= 0) |
| 580 | BUG(); |
| 581 | |
David Howells | dca54a7 | 2020-10-13 20:51:59 +0100 | [diff] [blame] | 582 | u = atomic_read(&cell->ref); |
| 583 | a = atomic_inc_return(&cell->active); |
| 584 | trace_afs_cell(cell->debug_id, u, a, reason); |
David Howells | 88c853c | 2019-07-23 11:24:59 +0100 | [diff] [blame] | 585 | return cell; |
| 586 | } |
| 587 | |
| 588 | /* |
| 589 | * Record a cell becoming less active. When the active counter reaches 1, it |
| 590 | * is scheduled for destruction, but may get reactivated. |
| 591 | */ |
David Howells | dca54a7 | 2020-10-13 20:51:59 +0100 | [diff] [blame] | 592 | void afs_unuse_cell(struct afs_net *net, struct afs_cell *cell, enum afs_cell_trace reason) |
David Howells | 989782d | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 593 | { |
David Howells | acc080d | 2020-10-27 10:42:56 +0000 | [diff] [blame] | 594 | unsigned int debug_id; |
David Howells | 989782d | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 595 | time64_t now, expire_delay; |
David Howells | dca54a7 | 2020-10-13 20:51:59 +0100 | [diff] [blame] | 596 | int u, a; |
David Howells | 989782d | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 597 | |
| 598 | if (!cell) |
| 599 | return; |
| 600 | |
| 601 | _enter("%s", cell->name); |
| 602 | |
| 603 | now = ktime_get_real_seconds(); |
| 604 | cell->last_inactive = now; |
| 605 | expire_delay = 0; |
David Howells | d5c32c8 | 2019-05-07 15:06:36 +0100 | [diff] [blame] | 606 | if (cell->vl_servers->nr_servers) |
David Howells | 989782d | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 607 | expire_delay = afs_cell_gc_delay; |
| 608 | |
David Howells | acc080d | 2020-10-27 10:42:56 +0000 | [diff] [blame] | 609 | debug_id = cell->debug_id; |
David Howells | dca54a7 | 2020-10-13 20:51:59 +0100 | [diff] [blame] | 610 | u = atomic_read(&cell->ref); |
David Howells | 88c853c | 2019-07-23 11:24:59 +0100 | [diff] [blame] | 611 | a = atomic_dec_return(&cell->active); |
David Howells | dca54a7 | 2020-10-13 20:51:59 +0100 | [diff] [blame] | 612 | trace_afs_cell(debug_id, u, a, reason); |
David Howells | 88c853c | 2019-07-23 11:24:59 +0100 | [diff] [blame] | 613 | WARN_ON(a == 0); |
| 614 | if (a == 1) |
| 615 | /* 'cell' may now be garbage collected. */ |
| 616 | afs_set_cell_timer(net, expire_delay); |
| 617 | } |
David Howells | 989782d | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 618 | |
David Howells | 88c853c | 2019-07-23 11:24:59 +0100 | [diff] [blame] | 619 | /* |
David Howells | dca54a7 | 2020-10-13 20:51:59 +0100 | [diff] [blame] | 620 | * Note that a cell has been seen. |
| 621 | */ |
| 622 | void afs_see_cell(struct afs_cell *cell, enum afs_cell_trace reason) |
| 623 | { |
| 624 | int u, a; |
| 625 | |
| 626 | u = atomic_read(&cell->ref); |
| 627 | a = atomic_read(&cell->active); |
| 628 | trace_afs_cell(cell->debug_id, u, a, reason); |
| 629 | } |
| 630 | |
| 631 | /* |
David Howells | 88c853c | 2019-07-23 11:24:59 +0100 | [diff] [blame] | 632 | * Queue a cell for management, giving the workqueue a ref to hold. |
| 633 | */ |
David Howells | dca54a7 | 2020-10-13 20:51:59 +0100 | [diff] [blame] | 634 | void afs_queue_cell(struct afs_cell *cell, enum afs_cell_trace reason) |
David Howells | 88c853c | 2019-07-23 11:24:59 +0100 | [diff] [blame] | 635 | { |
David Howells | dca54a7 | 2020-10-13 20:51:59 +0100 | [diff] [blame] | 636 | afs_get_cell(cell, reason); |
David Howells | 88c853c | 2019-07-23 11:24:59 +0100 | [diff] [blame] | 637 | if (!queue_work(afs_wq, &cell->manager)) |
David Howells | dca54a7 | 2020-10-13 20:51:59 +0100 | [diff] [blame] | 638 | afs_put_cell(cell, afs_cell_trace_put_queue_fail); |
David Howells | 989782d | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 639 | } |
| 640 | |
| 641 | /* |
| 642 | * Allocate a key to use as a placeholder for anonymous user security. |
| 643 | */ |
| 644 | static int afs_alloc_anon_key(struct afs_cell *cell) |
| 645 | { |
| 646 | struct key *key; |
| 647 | char keyname[4 + AFS_MAXCELLNAME + 1], *cp, *dp; |
| 648 | |
| 649 | /* Create a key to represent an anonymous user. */ |
| 650 | memcpy(keyname, "afs@", 4); |
| 651 | dp = keyname + 4; |
| 652 | cp = cell->name; |
| 653 | do { |
| 654 | *dp++ = tolower(*cp); |
| 655 | } while (*cp++); |
| 656 | |
| 657 | key = rxrpc_get_null_key(keyname); |
| 658 | if (IS_ERR(key)) |
| 659 | return PTR_ERR(key); |
| 660 | |
| 661 | cell->anonymous_key = key; |
| 662 | |
| 663 | _debug("anon key %p{%x}", |
| 664 | cell->anonymous_key, key_serial(cell->anonymous_key)); |
| 665 | return 0; |
| 666 | } |
| 667 | |
| 668 | /* |
| 669 | * Activate a cell. |
| 670 | */ |
| 671 | static int afs_activate_cell(struct afs_net *net, struct afs_cell *cell) |
| 672 | { |
David Howells | 6b3944e | 2018-10-11 22:45:49 +0100 | [diff] [blame] | 673 | struct hlist_node **p; |
| 674 | struct afs_cell *pcell; |
David Howells | 989782d | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 675 | int ret; |
| 676 | |
| 677 | if (!cell->anonymous_key) { |
| 678 | ret = afs_alloc_anon_key(cell); |
| 679 | if (ret < 0) |
| 680 | return ret; |
| 681 | } |
| 682 | |
David Howells | 5b86d4f | 2018-05-18 11:46:15 +0100 | [diff] [blame] | 683 | ret = afs_proc_cell_setup(cell); |
David Howells | 989782d | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 684 | if (ret < 0) |
| 685 | return ret; |
David Howells | 0da0b7f | 2018-06-15 15:19:22 +0100 | [diff] [blame] | 686 | |
| 687 | mutex_lock(&net->proc_cells_lock); |
David Howells | 6b3944e | 2018-10-11 22:45:49 +0100 | [diff] [blame] | 688 | for (p = &net->proc_cells.first; *p; p = &(*p)->next) { |
| 689 | pcell = hlist_entry(*p, struct afs_cell, proc_link); |
| 690 | if (strcmp(cell->name, pcell->name) < 0) |
| 691 | break; |
| 692 | } |
| 693 | |
| 694 | cell->proc_link.pprev = p; |
| 695 | cell->proc_link.next = *p; |
| 696 | rcu_assign_pointer(*p, &cell->proc_link.next); |
| 697 | if (cell->proc_link.next) |
| 698 | cell->proc_link.next->pprev = &cell->proc_link.next; |
| 699 | |
David Howells | 0da0b7f | 2018-06-15 15:19:22 +0100 | [diff] [blame] | 700 | afs_dynroot_mkdir(net, cell); |
| 701 | mutex_unlock(&net->proc_cells_lock); |
David Howells | 989782d | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 702 | return 0; |
| 703 | } |
| 704 | |
| 705 | /* |
| 706 | * Deactivate a cell. |
| 707 | */ |
| 708 | static void afs_deactivate_cell(struct afs_net *net, struct afs_cell *cell) |
| 709 | { |
| 710 | _enter("%s", cell->name); |
| 711 | |
David Howells | 5b86d4f | 2018-05-18 11:46:15 +0100 | [diff] [blame] | 712 | afs_proc_cell_remove(cell); |
David Howells | 989782d | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 713 | |
David Howells | 0da0b7f | 2018-06-15 15:19:22 +0100 | [diff] [blame] | 714 | mutex_lock(&net->proc_cells_lock); |
David Howells | 6b3944e | 2018-10-11 22:45:49 +0100 | [diff] [blame] | 715 | hlist_del_rcu(&cell->proc_link); |
David Howells | 0da0b7f | 2018-06-15 15:19:22 +0100 | [diff] [blame] | 716 | afs_dynroot_rmdir(net, cell); |
| 717 | mutex_unlock(&net->proc_cells_lock); |
David Howells | 989782d | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 718 | |
David Howells | 989782d | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 719 | _leave(""); |
| 720 | } |
| 721 | |
| 722 | /* |
| 723 | * Manage a cell record, initialising and destroying it, maintaining its DNS |
| 724 | * records. |
| 725 | */ |
David Howells | 88c853c | 2019-07-23 11:24:59 +0100 | [diff] [blame] | 726 | static void afs_manage_cell(struct afs_cell *cell) |
David Howells | 989782d | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 727 | { |
David Howells | 989782d | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 728 | struct afs_net *net = cell->net; |
David Howells | 88c853c | 2019-07-23 11:24:59 +0100 | [diff] [blame] | 729 | int ret, active; |
David Howells | 989782d | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 730 | |
| 731 | _enter("%s", cell->name); |
| 732 | |
| 733 | again: |
| 734 | _debug("state %u", cell->state); |
| 735 | switch (cell->state) { |
| 736 | case AFS_CELL_INACTIVE: |
| 737 | case AFS_CELL_FAILED: |
David Howells | 92e3cc9 | 2020-10-09 14:11:58 +0100 | [diff] [blame] | 738 | down_write(&net->cells_lock); |
David Howells | 88c853c | 2019-07-23 11:24:59 +0100 | [diff] [blame] | 739 | active = 1; |
David Howells | 1d0e850 | 2020-10-16 13:21:14 +0100 | [diff] [blame] | 740 | if (atomic_try_cmpxchg_relaxed(&cell->active, &active, 0)) { |
David Howells | 989782d | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 741 | rb_erase(&cell->net_node, &net->cells); |
David Howells | dca54a7 | 2020-10-13 20:51:59 +0100 | [diff] [blame] | 742 | trace_afs_cell(cell->debug_id, atomic_read(&cell->ref), 0, |
| 743 | afs_cell_trace_unuse_delete); |
David Howells | 1d0e850 | 2020-10-16 13:21:14 +0100 | [diff] [blame] | 744 | smp_store_release(&cell->state, AFS_CELL_REMOVED); |
David Howells | 88c853c | 2019-07-23 11:24:59 +0100 | [diff] [blame] | 745 | } |
David Howells | 92e3cc9 | 2020-10-09 14:11:58 +0100 | [diff] [blame] | 746 | up_write(&net->cells_lock); |
David Howells | 1d0e850 | 2020-10-16 13:21:14 +0100 | [diff] [blame] | 747 | if (cell->state == AFS_CELL_REMOVED) { |
| 748 | wake_up_var(&cell->state); |
David Howells | 989782d | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 749 | goto final_destruction; |
David Howells | 1d0e850 | 2020-10-16 13:21:14 +0100 | [diff] [blame] | 750 | } |
David Howells | 989782d | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 751 | if (cell->state == AFS_CELL_FAILED) |
| 752 | goto done; |
David Howells | d5c32c8 | 2019-05-07 15:06:36 +0100 | [diff] [blame] | 753 | smp_store_release(&cell->state, AFS_CELL_UNSET); |
| 754 | wake_up_var(&cell->state); |
David Howells | 989782d | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 755 | goto again; |
| 756 | |
| 757 | case AFS_CELL_UNSET: |
David Howells | d5c32c8 | 2019-05-07 15:06:36 +0100 | [diff] [blame] | 758 | smp_store_release(&cell->state, AFS_CELL_ACTIVATING); |
| 759 | wake_up_var(&cell->state); |
David Howells | 989782d | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 760 | goto again; |
| 761 | |
| 762 | case AFS_CELL_ACTIVATING: |
| 763 | ret = afs_activate_cell(net, cell); |
| 764 | if (ret < 0) |
| 765 | goto activation_failed; |
| 766 | |
David Howells | d5c32c8 | 2019-05-07 15:06:36 +0100 | [diff] [blame] | 767 | smp_store_release(&cell->state, AFS_CELL_ACTIVE); |
| 768 | wake_up_var(&cell->state); |
David Howells | 989782d | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 769 | goto again; |
| 770 | |
| 771 | case AFS_CELL_ACTIVE: |
David Howells | 88c853c | 2019-07-23 11:24:59 +0100 | [diff] [blame] | 772 | if (atomic_read(&cell->active) > 1) { |
David Howells | d5c32c8 | 2019-05-07 15:06:36 +0100 | [diff] [blame] | 773 | if (test_and_clear_bit(AFS_CELL_FL_DO_LOOKUP, &cell->flags)) { |
| 774 | ret = afs_update_cell(cell); |
| 775 | if (ret < 0) |
| 776 | cell->error = ret; |
| 777 | } |
David Howells | 989782d | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 778 | goto done; |
| 779 | } |
David Howells | d5c32c8 | 2019-05-07 15:06:36 +0100 | [diff] [blame] | 780 | smp_store_release(&cell->state, AFS_CELL_DEACTIVATING); |
| 781 | wake_up_var(&cell->state); |
David Howells | 989782d | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 782 | goto again; |
| 783 | |
| 784 | case AFS_CELL_DEACTIVATING: |
David Howells | 88c853c | 2019-07-23 11:24:59 +0100 | [diff] [blame] | 785 | if (atomic_read(&cell->active) > 1) |
David Howells | 989782d | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 786 | goto reverse_deactivation; |
| 787 | afs_deactivate_cell(net, cell); |
David Howells | d5c32c8 | 2019-05-07 15:06:36 +0100 | [diff] [blame] | 788 | smp_store_release(&cell->state, AFS_CELL_INACTIVE); |
| 789 | wake_up_var(&cell->state); |
David Howells | 989782d | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 790 | goto again; |
| 791 | |
David Howells | 1d0e850 | 2020-10-16 13:21:14 +0100 | [diff] [blame] | 792 | case AFS_CELL_REMOVED: |
| 793 | goto done; |
| 794 | |
David Howells | 989782d | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 795 | default: |
| 796 | break; |
| 797 | } |
| 798 | _debug("bad state %u", cell->state); |
| 799 | BUG(); /* Unhandled state */ |
| 800 | |
| 801 | activation_failed: |
| 802 | cell->error = ret; |
| 803 | afs_deactivate_cell(net, cell); |
| 804 | |
David Howells | d5c32c8 | 2019-05-07 15:06:36 +0100 | [diff] [blame] | 805 | smp_store_release(&cell->state, AFS_CELL_FAILED); /* vs error */ |
| 806 | wake_up_var(&cell->state); |
David Howells | 989782d | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 807 | goto again; |
| 808 | |
| 809 | reverse_deactivation: |
David Howells | d5c32c8 | 2019-05-07 15:06:36 +0100 | [diff] [blame] | 810 | smp_store_release(&cell->state, AFS_CELL_ACTIVE); |
| 811 | wake_up_var(&cell->state); |
David Howells | 989782d | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 812 | _leave(" [deact->act]"); |
| 813 | return; |
| 814 | |
| 815 | done: |
| 816 | _leave(" [done %u]", cell->state); |
| 817 | return; |
| 818 | |
| 819 | final_destruction: |
David Howells | 88c853c | 2019-07-23 11:24:59 +0100 | [diff] [blame] | 820 | /* The root volume is pinning the cell */ |
| 821 | afs_put_volume(cell->net, cell->root_volume, afs_volume_trace_put_cell_root); |
| 822 | cell->root_volume = NULL; |
David Howells | dca54a7 | 2020-10-13 20:51:59 +0100 | [diff] [blame] | 823 | afs_put_cell(cell, afs_cell_trace_put_destroy); |
David Howells | 88c853c | 2019-07-23 11:24:59 +0100 | [diff] [blame] | 824 | } |
| 825 | |
| 826 | static void afs_manage_cell_work(struct work_struct *work) |
| 827 | { |
| 828 | struct afs_cell *cell = container_of(work, struct afs_cell, manager); |
| 829 | |
| 830 | afs_manage_cell(cell); |
David Howells | dca54a7 | 2020-10-13 20:51:59 +0100 | [diff] [blame] | 831 | afs_put_cell(cell, afs_cell_trace_put_queue_work); |
David Howells | 989782d | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 832 | } |
| 833 | |
| 834 | /* |
| 835 | * Manage the records of cells known to a network namespace. This includes |
| 836 | * updating the DNS records and garbage collecting unused cells that were |
| 837 | * automatically added. |
| 838 | * |
| 839 | * Note that constructed cell records may only be removed from net->cells by |
| 840 | * this work item, so it is safe for this work item to stash a cursor pointing |
| 841 | * into the tree and then return to caller (provided it skips cells that are |
| 842 | * still under construction). |
| 843 | * |
| 844 | * Note also that we were given an increment on net->cells_outstanding by |
| 845 | * whoever queued us that we need to deal with before returning. |
| 846 | */ |
| 847 | void afs_manage_cells(struct work_struct *work) |
| 848 | { |
| 849 | struct afs_net *net = container_of(work, struct afs_net, cells_manager); |
| 850 | struct rb_node *cursor; |
| 851 | time64_t now = ktime_get_real_seconds(), next_manage = TIME64_MAX; |
| 852 | bool purging = !net->live; |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 853 | |
| 854 | _enter(""); |
| 855 | |
David Howells | 989782d | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 856 | /* Trawl the cell database looking for cells that have expired from |
| 857 | * lack of use and cells whose DNS results have expired and dispatch |
| 858 | * their managers. |
| 859 | */ |
David Howells | 92e3cc9 | 2020-10-09 14:11:58 +0100 | [diff] [blame] | 860 | down_read(&net->cells_lock); |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 861 | |
David Howells | 989782d | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 862 | for (cursor = rb_first(&net->cells); cursor; cursor = rb_next(cursor)) { |
| 863 | struct afs_cell *cell = |
| 864 | rb_entry(cursor, struct afs_cell, net_node); |
David Howells | 88c853c | 2019-07-23 11:24:59 +0100 | [diff] [blame] | 865 | unsigned active; |
David Howells | 989782d | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 866 | bool sched_cell = false; |
David Howells | 08e0e7c | 2007-04-26 15:55:03 -0700 | [diff] [blame] | 867 | |
David Howells | 88c853c | 2019-07-23 11:24:59 +0100 | [diff] [blame] | 868 | active = atomic_read(&cell->active); |
David Howells | dca54a7 | 2020-10-13 20:51:59 +0100 | [diff] [blame] | 869 | trace_afs_cell(cell->debug_id, atomic_read(&cell->ref), |
| 870 | active, afs_cell_trace_manage); |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 871 | |
David Howells | 88c853c | 2019-07-23 11:24:59 +0100 | [diff] [blame] | 872 | ASSERTCMP(active, >=, 1); |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 873 | |
David Howells | 989782d | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 874 | if (purging) { |
David Howells | dca54a7 | 2020-10-13 20:51:59 +0100 | [diff] [blame] | 875 | if (test_and_clear_bit(AFS_CELL_FL_NO_GC, &cell->flags)) { |
| 876 | active = atomic_dec_return(&cell->active); |
| 877 | trace_afs_cell(cell->debug_id, atomic_read(&cell->ref), |
| 878 | active, afs_cell_trace_unuse_pin); |
| 879 | } |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 880 | } |
| 881 | |
David Howells | 88c853c | 2019-07-23 11:24:59 +0100 | [diff] [blame] | 882 | if (active == 1) { |
David Howells | d5c32c8 | 2019-05-07 15:06:36 +0100 | [diff] [blame] | 883 | struct afs_vlserver_list *vllist; |
David Howells | 989782d | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 884 | time64_t expire_at = cell->last_inactive; |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 885 | |
David Howells | d5c32c8 | 2019-05-07 15:06:36 +0100 | [diff] [blame] | 886 | read_lock(&cell->vl_servers_lock); |
| 887 | vllist = rcu_dereference_protected( |
| 888 | cell->vl_servers, |
| 889 | lockdep_is_held(&cell->vl_servers_lock)); |
| 890 | if (vllist->nr_servers > 0) |
David Howells | 989782d | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 891 | expire_at += afs_cell_gc_delay; |
David Howells | d5c32c8 | 2019-05-07 15:06:36 +0100 | [diff] [blame] | 892 | read_unlock(&cell->vl_servers_lock); |
David Howells | 989782d | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 893 | if (purging || expire_at <= now) |
| 894 | sched_cell = true; |
| 895 | else if (expire_at < next_manage) |
| 896 | next_manage = expire_at; |
| 897 | } |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 898 | |
David Howells | 989782d | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 899 | if (!purging) { |
David Howells | d5c32c8 | 2019-05-07 15:06:36 +0100 | [diff] [blame] | 900 | if (test_bit(AFS_CELL_FL_DO_LOOKUP, &cell->flags)) |
David Howells | 989782d | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 901 | sched_cell = true; |
David Howells | 989782d | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 902 | } |
| 903 | |
| 904 | if (sched_cell) |
David Howells | dca54a7 | 2020-10-13 20:51:59 +0100 | [diff] [blame] | 905 | afs_queue_cell(cell, afs_cell_trace_get_queue_manage); |
David Howells | 989782d | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 906 | } |
| 907 | |
David Howells | 92e3cc9 | 2020-10-09 14:11:58 +0100 | [diff] [blame] | 908 | up_read(&net->cells_lock); |
David Howells | 989782d | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 909 | |
| 910 | /* Update the timer on the way out. We have to pass an increment on |
| 911 | * cells_outstanding in the namespace that we are in to the timer or |
| 912 | * the work scheduler. |
| 913 | */ |
| 914 | if (!purging && next_manage < TIME64_MAX) { |
| 915 | now = ktime_get_real_seconds(); |
| 916 | |
| 917 | if (next_manage - now <= 0) { |
| 918 | if (queue_work(afs_wq, &net->cells_manager)) |
| 919 | atomic_inc(&net->cells_outstanding); |
| 920 | } else { |
| 921 | afs_set_cell_timer(net, next_manage - now); |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 922 | } |
| 923 | } |
| 924 | |
David Howells | 989782d | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 925 | afs_dec_cells_outstanding(net); |
| 926 | _leave(" [%d]", atomic_read(&net->cells_outstanding)); |
| 927 | } |
| 928 | |
| 929 | /* |
| 930 | * Purge in-memory cell database. |
| 931 | */ |
| 932 | void afs_cell_purge(struct afs_net *net) |
| 933 | { |
| 934 | struct afs_cell *ws; |
| 935 | |
| 936 | _enter(""); |
| 937 | |
David Howells | 92e3cc9 | 2020-10-09 14:11:58 +0100 | [diff] [blame] | 938 | down_write(&net->cells_lock); |
| 939 | ws = net->ws_cell; |
| 940 | net->ws_cell = NULL; |
| 941 | up_write(&net->cells_lock); |
David Howells | dca54a7 | 2020-10-13 20:51:59 +0100 | [diff] [blame] | 942 | afs_unuse_cell(net, ws, afs_cell_trace_unuse_ws); |
David Howells | 989782d | 2017-11-02 15:27:50 +0000 | [diff] [blame] | 943 | |
| 944 | _debug("del timer"); |
| 945 | if (del_timer_sync(&net->cells_timer)) |
| 946 | atomic_dec(&net->cells_outstanding); |
| 947 | |
| 948 | _debug("kick mgr"); |
| 949 | afs_queue_cell_manager(net); |
| 950 | |
| 951 | _debug("wait"); |
Peter Zijlstra | ab1fbe3 | 2018-03-15 11:42:28 +0100 | [diff] [blame] | 952 | wait_var_event(&net->cells_outstanding, |
| 953 | !atomic_read(&net->cells_outstanding)); |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 954 | _leave(""); |
David Howells | ec26815 | 2007-04-26 15:49:28 -0700 | [diff] [blame] | 955 | } |