26 	struct afs_server *server = (struct afs_server *)rxrpc_kernel_get_peer_data(peer);  in afs_find_server()  local
28 	if (!server)  in afs_find_server()
30 	return afs_use_server(server, false, afs_server_trace_use_cm_call);  in afs_find_server()
39 	struct afs_server *server;  in afs_find_server_by_uuid()  local
47 		server = rb_entry(p, struct afs_server, uuid_rb);  in afs_find_server_by_uuid()
49 		diff = memcmp(uuid, &server->uuid, sizeof(*uuid));  in afs_find_server_by_uuid()
55 			if (test_bit(AFS_SERVER_FL_UNCREATED, &server->flags))  in afs_find_server_by_uuid()
57 			afs_use_server(server, true, afs_server_trace_use_by_uuid);  in afs_find_server_by_uuid()
58 			return server;  in afs_find_server_by_uuid()
72 	struct afs_server *server;  in afs_install_server()  local
85 		server = rb_entry(p, struct afs_server, uuid_rb);  in afs_install_server()
86 		diff = memcmp(&(*candidate)->uuid, &server->uuid, sizeof(uuid_t));  in afs_install_server()
95 	server = *candidate;  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()
100 	hlist_add_head_rcu(&server->proc_link, &net->fs_proc);  in afs_install_server()
106 	afs_use_server(server, true, afs_server_trace_use_install);  in afs_install_server()
107 	return server;  in afs_install_server()
115 	struct afs_server *server;  in afs_alloc_server()  local
120 	server = kzalloc(sizeof(struct afs_server), GFP_KERNEL);  in afs_alloc_server()
121 	if (!server)  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()
145 	_leave(" = %p", server);  in afs_alloc_server()
146 	return server;  in afs_alloc_server()
152 static struct afs_addr_list *afs_vl_lookup_addrs(struct afs_server *server,  in afs_vl_lookup_addrs()  argument
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()
182 	struct afs_server *server, *candidate = NULL;  in afs_lookup_server()  local
189 	server = afs_find_server_by_uuid(cell, uuid);  in afs_lookup_server()
193 	if (server) {  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()
199 		return server;  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()
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()
221 	timer_delete_sync(&server->timer);  in afs_lookup_server()
229 		alist = afs_vl_lookup_addrs(server, key);  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()
249 	return server ?: ERR_PTR(ret);  in afs_lookup_server()
252 	afs_see_server(server, afs_server_trace_wait_create);  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()
258 		server = NULL;  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()
279 	server = NULL;  in afs_lookup_server()
288 static void afs_set_server_timer(struct afs_server *server, unsigned int delay_secs)  in afs_set_server_timer()  argument
290 	mod_timer(&server->timer, jiffies + delay_secs * HZ);  in afs_set_server_timer()
296 struct afs_server *afs_get_server(struct afs_server *server,  in afs_get_server()  argument
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()
305 	return server;  in afs_get_server()
312 struct afs_server *afs_use_server(struct afs_server *server, bool activate,  in afs_use_server()  argument
318 	__refcount_inc(&server->ref, &r);  in afs_use_server()
319 	a = atomic_inc_return(&server->active);  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()
325 	return server;  in afs_use_server()
331 void afs_put_server(struct afs_net *net, struct afs_server *server,  in afs_put_server()  argument
338 	if (!server)  in afs_put_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()
346 		__afs_put_server(net, server);  in afs_put_server()
353 void afs_unuse_server_notime(struct afs_net *net, struct afs_server *server,  in afs_unuse_server_notime()  argument
356 	if (!server)  in afs_unuse_server_notime()
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()
365 	afs_put_server(net, server, reason);  in afs_unuse_server_notime()
371 void afs_unuse_server(struct afs_net *net, struct afs_server *server,  in afs_unuse_server()  argument
374 	if (!server)  in afs_unuse_server()
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()
383 			afs_set_server_timer(server, afs_server_gc_delay);  in afs_unuse_server()
385 			schedule_work(&server->destroyer);  in afs_unuse_server()
389 	afs_put_server(net, server, reason);  in afs_unuse_server()
394 	struct afs_server *server = container_of(rcu, struct afs_server, rcu);  in afs_server_rcu()  local
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()
402 	kfree(server);  in afs_server_rcu()
405 static void __afs_put_server(struct afs_net *net, struct afs_server *server)  in __afs_put_server()  argument
407 	call_rcu(&server->rcu, afs_server_rcu);  in __afs_put_server()
411 static void afs_give_up_callbacks(struct afs_net *net, struct afs_server *server)  in afs_give_up_callbacks()  argument
413 	struct afs_endpoint_state *estate = rcu_access_pointer(server->endpoint_state);  in afs_give_up_callbacks()
416 	afs_fs_give_up_all_callbacks(net, server, &alist->addrs[alist->preferred], NULL);  in afs_give_up_callbacks()
422 static bool afs_has_server_expired(const struct afs_server *server)  in afs_has_server_expired()  argument
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()
447 static bool afs_remove_server_from_cell(struct afs_server *server)  in afs_remove_server_from_cell()  argument
449 	struct afs_cell *cell = server->cell;  in afs_remove_server_from_cell()
453 	if (!afs_has_server_expired(server)) {  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()
460 	afs_see_server(server, afs_server_trace_see_expired);  in afs_remove_server_from_cell()
461 	rb_erase(&server->uuid_rb, &cell->fs_servers);  in afs_remove_server_from_cell()
469 	struct afs_server *server = container_of(work, struct afs_server, destroyer);  in afs_server_destroyer()  local
470 	struct afs_net *net = server->cell->net;  in afs_server_destroyer()
472 	afs_see_server(server, afs_server_trace_see_destroyer);  in afs_server_destroyer()
474 	if (test_bit(AFS_SERVER_FL_EXPIRED, &server->flags))  in afs_server_destroyer()
477 	if (!afs_remove_server_from_cell(server))  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()
484 		afs_give_up_callbacks(net, server);  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()
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()
497 	afs_put_server(net, server, afs_server_trace_destroy);  in afs_server_destroyer()
502 	struct afs_server *server = container_of(timer, struct afs_server, timer);  in afs_server_timer()  local
504 	afs_see_server(server, afs_server_trace_see_timer);  in afs_server_timer()
505 	if (!test_bit(AFS_SERVER_FL_EXPIRED, &server->flags))  in afs_server_timer()
506 		schedule_work(&server->destroyer);  in afs_server_timer()
514 	struct afs_server *server;  in afs_purge_servers()  local
519 		server = rb_entry(rb, struct afs_server, uuid_rb);  in afs_purge_servers()
520 		afs_see_server(server, afs_server_trace_see_purge);  in afs_purge_servers()
521 		schedule_work(&server->destroyer);  in afs_purge_servers()
543 					      struct afs_server *server,  in afs_update_server_record()  argument
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()
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()
586 bool afs_check_server_record(struct afs_operation *op, struct afs_server *server,  in afs_check_server_record()  argument
594 	ASSERT(server);  in afs_check_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()
607 		success = afs_update_server_record(op, server, key);  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()