1 // SPDX-License-Identifier: GPL-2.0-or-later
35  * cell->fs_lock.
37 static struct afs_server *afs_find_server_by_uuid(struct afs_cell *cell, const uuid_t *uuid)  in afs_find_server_by_uuid()  argument
45 	p = cell->fs_servers.rb_node;  in afs_find_server_by_uuid()
49 		diff = memcmp(uuid, &server->uuid, sizeof(*uuid));  in afs_find_server_by_uuid()
51 			p = p->rb_left;  in afs_find_server_by_uuid()
53 			p = p->rb_right;  in afs_find_server_by_uuid()
55 			if (test_bit(AFS_SERVER_FL_UNCREATED, &server->flags))  in afs_find_server_by_uuid()
66  * Install a server record in the cell tree.  The caller must hold an exclusive
67  * lock on cell->fs_lock.
69 static struct afs_server *afs_install_server(struct afs_cell *cell,  in afs_install_server()  argument
73 	struct afs_net *net = cell->net;  in afs_install_server()
80 	pp = &cell->fs_servers.rb_node;  in afs_install_server()
84 		_debug("- consider %p", p);  in afs_install_server()
86 		diff = memcmp(&(*candidate)->uuid, &server->uuid, sizeof(uuid_t));  in afs_install_server()
88 			pp = &(*pp)->rb_left;  in afs_install_server()
90 			pp = &(*pp)->rb_right;  in afs_install_server()
97 	rb_link_node(&server->uuid_rb, p, pp);  in afs_install_server()
98 	rb_insert_color(&server->uuid_rb, &cell->fs_servers);  in afs_install_server()
99 	write_seqlock(&net->fs_lock);  in afs_install_server()
100 	hlist_add_head_rcu(&server->proc_link, &net->fs_proc);  in afs_install_server()
101 	write_sequnlock(&net->fs_lock);  in afs_install_server()
103 	afs_get_cell(cell, afs_cell_trace_get_server);  in afs_install_server()
113 static struct afs_server *afs_alloc_server(struct afs_cell *cell, const uuid_t *uuid)  in afs_alloc_server()  argument
116 	struct afs_net *net = cell->net;  in afs_alloc_server()
124 	refcount_set(&server->ref, 1);  in afs_alloc_server()
125 	atomic_set(&server->active, 0);  in afs_alloc_server()
126 	__set_bit(AFS_SERVER_FL_UNCREATED, &server->flags);  in afs_alloc_server()
127 	server->debug_id = atomic_inc_return(&afs_server_debug_id);  in afs_alloc_server()
128 	server->uuid = *uuid;  in afs_alloc_server()
129 	rwlock_init(&server->fs_lock);  in afs_alloc_server()
130 	INIT_WORK(&server->destroyer, &afs_server_destroyer);  in afs_alloc_server()
131 	timer_setup(&server->timer, afs_server_timer, 0);  in afs_alloc_server()
132 	INIT_LIST_HEAD(&server->volumes);  in afs_alloc_server()
133 	init_waitqueue_head(&server->probe_wq);  in afs_alloc_server()
134 	mutex_init(&server->cm_token_lock);  in afs_alloc_server()
135 	INIT_LIST_HEAD(&server->probe_link);  in afs_alloc_server()
136 	INIT_HLIST_NODE(&server->proc_link);  in afs_alloc_server()
137 	spin_lock_init(&server->probe_lock);  in afs_alloc_server()
138 	server->cell = cell;  in afs_alloc_server()
139 	server->rtt = UINT_MAX;  in afs_alloc_server()
140 	server->service_id = FS_SERVICE;  in afs_alloc_server()
141 	server->probe_counter = 1;  in afs_alloc_server()
142 	server->probed_at = jiffies - LONG_MAX / 2;  in afs_alloc_server()
159 	ret = -ERESTARTSYS;  in afs_vl_lookup_addrs()
160 	if (afs_begin_vlserver_operation(&vc, server->cell, key)) {  in afs_vl_lookup_addrs()
162 			if (test_bit(AFS_VLSERVER_FL_IS_YFS, &vc.server->flags))  in afs_vl_lookup_addrs()
163 				alist = afs_yfsvl_get_endpoints(&vc, &server->uuid);  in afs_vl_lookup_addrs()
165 				alist = afs_vl_get_addrs_u(&vc, &server->uuid);  in afs_vl_lookup_addrs()
175  * Get or create a fileserver record and return it with an active-use count on
178 struct afs_server *afs_lookup_server(struct afs_cell *cell, struct key *key,  in afs_lookup_server()  argument
186 	_enter("%p,%pU", cell->net, uuid);  in afs_lookup_server()
188 	down_read(&cell->fs_lock);  in afs_lookup_server()
189 	server = afs_find_server_by_uuid(cell, uuid);  in afs_lookup_server()
191 	up_read(&cell->fs_lock);  in afs_lookup_server()
194 		timer_delete_sync(&server->timer);  in afs_lookup_server()
195 		if (test_bit(AFS_SERVER_FL_CREATING, &server->flags))  in afs_lookup_server()
197 		if (server->addr_version != addr_version)  in afs_lookup_server()
198 			set_bit(AFS_SERVER_FL_NEEDS_UPDATE, &server->flags);  in afs_lookup_server()
202 	candidate = afs_alloc_server(cell, uuid);  in afs_lookup_server()
205 		return ERR_PTR(-ENOMEM);  in afs_lookup_server()
208 	down_write(&cell->fs_lock);  in afs_lookup_server()
209 	server = afs_install_server(cell, &candidate);  in afs_lookup_server()
210 	if (test_bit(AFS_SERVER_FL_CREATING, &server->flags)) {  in afs_lookup_server()
212 		up_write(&cell->fs_lock);  in afs_lookup_server()
215 	if (test_bit(AFS_SERVER_FL_UNCREATED, &server->flags)) {  in afs_lookup_server()
216 		set_bit(AFS_SERVER_FL_CREATING, &server->flags);  in afs_lookup_server()
217 		clear_bit(AFS_SERVER_FL_UNCREATED, &server->flags);  in afs_lookup_server()
220 	up_write(&cell->fs_lock);  in afs_lookup_server()
221 	timer_delete_sync(&server->timer);  in afs_lookup_server()
225 	 * fileserver.  This will make sure the repeat-probing service is  in afs_lookup_server()
235 		ret = afs_fs_probe_fileserver(cell->net, server, alist, key);  in afs_lookup_server()
239 		clear_and_wake_up_bit(AFS_SERVER_FL_CREATING, &server->flags);  in afs_lookup_server()
245 		kfree(rcu_access_pointer(server->endpoint_state));  in afs_lookup_server()
247 		afs_dec_servers_outstanding(cell->net);  in afs_lookup_server()
253 	wait_on_bit(&server->flags, AFS_SERVER_FL_CREATING, TASK_UNINTERRUPTIBLE);  in afs_lookup_server()
254 	if (test_bit_acquire(AFS_SERVER_FL_UNCREATED, &server->flags)) {  in afs_lookup_server()
256 		ret = READ_ONCE(server->create_error);  in afs_lookup_server()
257 		afs_put_server(cell->net, server, afs_server_trace_unuse_create_fail);  in afs_lookup_server()
266 	down_write(&cell->fs_lock);  in afs_lookup_server()
268 	WRITE_ONCE(server->create_error, ret);  in afs_lookup_server()
270 	set_bit(AFS_SERVER_FL_UNCREATED, &server->flags);  in afs_lookup_server()
272 	clear_and_wake_up_bit(AFS_SERVER_FL_CREATING, &server->flags);  in afs_lookup_server()
274 	if (test_bit(AFS_SERVER_FL_UNCREATED, &server->flags)) {  in afs_lookup_server()
275 		clear_bit(AFS_SERVER_FL_UNCREATED, &server->flags);  in afs_lookup_server()
278 	afs_unuse_server(cell->net, server, afs_server_trace_unuse_create_fail);  in afs_lookup_server()
281 	up_write(&cell->fs_lock);  in afs_lookup_server()
290 	mod_timer(&server->timer, jiffies + delay_secs * HZ);  in afs_set_server_timer()
302 	__refcount_inc(&server->ref, &r);  in afs_get_server()
303 	a = atomic_read(&server->active);  in afs_get_server()
304 	trace_afs_server(server->debug_id, r + 1, a, reason);  in afs_get_server()
318 	__refcount_inc(&server->ref, &r);  in afs_use_server()
319 	a = atomic_inc_return(&server->active);  in afs_use_server()
320 	if (a == 1 && activate &&  in afs_use_server()
321 	    !test_bit(AFS_SERVER_FL_EXPIRED, &server->flags))  in afs_use_server()
322 		timer_delete(&server->timer);  in afs_use_server()
324 	trace_afs_server(server->debug_id, r + 1, a, reason);  in afs_use_server()
341 	debug_id = server->debug_id;  in afs_put_server()
342 	a = atomic_read(&server->active);  in afs_put_server()
343 	zero = __refcount_dec_and_test(&server->ref, &r);  in afs_put_server()
344 	trace_afs_server(debug_id, r - 1, a, reason);  in afs_put_server()
350  * Drop an active count on a server object without updating the last-unused
359 	if (atomic_dec_and_test(&server->active)) {  in afs_unuse_server_notime()
360 		if (test_bit(AFS_SERVER_FL_EXPIRED, &server->flags) ||  in afs_unuse_server_notime()
361 		    READ_ONCE(server->cell->state) >= AFS_CELL_REMOVING)  in afs_unuse_server_notime()
362 			schedule_work(&server->destroyer);  in afs_unuse_server_notime()
377 	if (atomic_dec_and_test(&server->active)) {  in afs_unuse_server()
378 		if (!test_bit(AFS_SERVER_FL_EXPIRED, &server->flags) &&  in afs_unuse_server()
379 		    READ_ONCE(server->cell->state) < AFS_CELL_REMOVING) {  in afs_unuse_server()
382 			server->unuse_time = unuse_time;  in afs_unuse_server()
385 			schedule_work(&server->destroyer);  in afs_unuse_server()
396 	trace_afs_server(server->debug_id, refcount_read(&server->ref),  in afs_server_rcu()
397 			 atomic_read(&server->active), afs_server_trace_free);  in afs_server_rcu()
398 	afs_put_endpoint_state(rcu_access_pointer(server->endpoint_state),  in afs_server_rcu()
400 	afs_put_cell(server->cell, afs_cell_trace_put_server);  in afs_server_rcu()
401 	kfree(server->cm_rxgk_appdata.data);  in afs_server_rcu()
407 	call_rcu(&server->rcu, afs_server_rcu);  in __afs_put_server()
413 	struct afs_endpoint_state *estate = rcu_access_pointer(server->endpoint_state);  in afs_give_up_callbacks()
414 	struct afs_addr_list *alist = estate->addresses;  in afs_give_up_callbacks()
416 	afs_fs_give_up_all_callbacks(net, server, &alist->addrs[alist->preferred], NULL);  in afs_give_up_callbacks()
426 	if (atomic_read(&server->active))  in afs_has_server_expired()
429 	if (server->cell->net->live ||  in afs_has_server_expired()
430 	    server->cell->state >= AFS_CELL_REMOVING) {  in afs_has_server_expired()
431 		trace_afs_server(server->debug_id, refcount_read(&server->ref),  in afs_has_server_expired()
436 	expires_at = server->unuse_time;  in afs_has_server_expired()
437 	if (!test_bit(AFS_SERVER_FL_VL_FAIL, &server->flags) &&  in afs_has_server_expired()
438 	    !test_bit(AFS_SERVER_FL_NOT_FOUND, &server->flags))  in afs_has_server_expired()
445  * Remove a server record from it's parent cell's database.
449 	struct afs_cell *cell = server->cell;  in afs_remove_server_from_cell()  local
451 	down_write(&cell->fs_lock);  in afs_remove_server_from_cell()
454 		up_write(&cell->fs_lock);  in afs_remove_server_from_cell()
458 	set_bit(AFS_SERVER_FL_EXPIRED, &server->flags);  in afs_remove_server_from_cell()
459 	_debug("expire %pU %u", &server->uuid, atomic_read(&server->active));  in afs_remove_server_from_cell()
461 	rb_erase(&server->uuid_rb, &cell->fs_servers);  in afs_remove_server_from_cell()
462 	up_write(&cell->fs_lock);  in afs_remove_server_from_cell()
470 	struct afs_net *net = server->cell->net;  in afs_server_destroyer()
474 	if (test_bit(AFS_SERVER_FL_EXPIRED, &server->flags))  in afs_server_destroyer()
480 	timer_shutdown_sync(&server->timer);  in afs_server_destroyer()
481 	cancel_work(&server->destroyer);  in afs_server_destroyer()
483 	if (test_bit(AFS_SERVER_FL_MAY_HAVE_CB, &server->flags))  in afs_server_destroyer()
487 	estate = rcu_access_pointer(server->endpoint_state);  in afs_server_destroyer()
489 		afs_set_peer_appdata(server, estate->addresses, NULL);  in afs_server_destroyer()
491 	write_seqlock(&net->fs_lock);  in afs_server_destroyer()
492 	list_del_init(&server->probe_link);  in afs_server_destroyer()
493 	if (!hlist_unhashed(&server->proc_link))  in afs_server_destroyer()
494 		hlist_del_rcu(&server->proc_link);  in afs_server_destroyer()
495 	write_sequnlock(&net->fs_lock);  in afs_server_destroyer()
505 	if (!test_bit(AFS_SERVER_FL_EXPIRED, &server->flags))  in afs_server_timer()
506 		schedule_work(&server->destroyer);  in afs_server_timer()
510  * Wake up all the servers in a cell so that they can purge themselves.
512 void afs_purge_servers(struct afs_cell *cell)  in afs_purge_servers()  argument
517 	down_read(&cell->fs_lock);  in afs_purge_servers()
518 	for (rb = rb_first(&cell->fs_servers); rb; rb = rb_next(rb)) {  in afs_purge_servers()
521 		schedule_work(&server->destroyer);  in afs_purge_servers()
523 	up_read(&cell->fs_lock);  in afs_purge_servers()
533 	atomic_dec(&net->servers_outstanding);  in afs_wait_for_servers()
534 	wait_var_event(&net->servers_outstanding,  in afs_wait_for_servers()
535 		       !atomic_read(&net->servers_outstanding));  in afs_wait_for_servers()
552 	trace_afs_server(server->debug_id, refcount_read(&server->ref),  in afs_update_server_record()
553 			 atomic_read(&server->active),  in afs_update_server_record()
556 	alist = afs_vl_lookup_addrs(server, op->key);  in afs_update_server_record()
559 		estate = rcu_dereference(server->endpoint_state);  in afs_update_server_record()
560 		has_addrs = estate->addresses;  in afs_update_server_record()
563 		if ((PTR_ERR(alist) == -ERESTARTSYS ||  in afs_update_server_record()
564 		     PTR_ERR(alist) == -EINTR) &&  in afs_update_server_record()
565 		    (op->flags & AFS_OPERATION_UNINTR) &&  in afs_update_server_record()
575 	if (server->addr_version != alist->version)  in afs_update_server_record()
576 		afs_fs_probe_fileserver(op->net, server, alist, key);  in afs_update_server_record()
597 	if (test_bit(AFS_SERVER_FL_UPDATING, &server->flags))  in afs_check_server_record()
599 	if (test_bit(AFS_SERVER_FL_NEEDS_UPDATE, &server->flags))  in afs_check_server_record()
605 	if (!test_and_set_bit_lock(AFS_SERVER_FL_UPDATING, &server->flags)) {  in afs_check_server_record()
606 		clear_bit(AFS_SERVER_FL_NEEDS_UPDATE, &server->flags);  in afs_check_server_record()
608 		clear_bit_unlock(AFS_SERVER_FL_UPDATING, &server->flags);  in afs_check_server_record()
609 		wake_up_bit(&server->flags, AFS_SERVER_FL_UPDATING);  in afs_check_server_record()
615 	ret = wait_on_bit(&server->flags, AFS_SERVER_FL_UPDATING,  in afs_check_server_record()
616 			  (op->flags & AFS_OPERATION_UNINTR) ?  in afs_check_server_record()
618 	if (ret == -ERESTARTSYS) {  in afs_check_server_record()
627 		ret = -ESTALE;  in afs_check_server_record()