| Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 1 | /* main.c: AFS client file system | 
 | 2 |  * | 
 | 3 |  * Copyright (C) 2002 Red Hat, Inc. All Rights Reserved. | 
 | 4 |  * Written by David Howells (dhowells@redhat.com) | 
 | 5 |  * | 
 | 6 |  * This program is free software; you can redistribute it and/or | 
 | 7 |  * modify it under the terms of the GNU General Public License | 
 | 8 |  * as published by the Free Software Foundation; either version | 
 | 9 |  * 2 of the License, or (at your option) any later version. | 
 | 10 |  */ | 
 | 11 |  | 
 | 12 | #include <linux/module.h> | 
 | 13 | #include <linux/moduleparam.h> | 
 | 14 | #include <linux/init.h> | 
 | 15 | #include <linux/sched.h> | 
 | 16 | #include <linux/completion.h> | 
 | 17 | #include <rxrpc/rxrpc.h> | 
 | 18 | #include <rxrpc/transport.h> | 
 | 19 | #include <rxrpc/call.h> | 
 | 20 | #include <rxrpc/peer.h> | 
 | 21 | #include "cache.h" | 
 | 22 | #include "cell.h" | 
 | 23 | #include "server.h" | 
 | 24 | #include "fsclient.h" | 
 | 25 | #include "cmservice.h" | 
 | 26 | #include "kafstimod.h" | 
 | 27 | #include "kafsasyncd.h" | 
 | 28 | #include "internal.h" | 
 | 29 |  | 
 | 30 | struct rxrpc_transport *afs_transport; | 
 | 31 |  | 
 | 32 | static int afs_adding_peer(struct rxrpc_peer *peer); | 
 | 33 | static void afs_discarding_peer(struct rxrpc_peer *peer); | 
 | 34 |  | 
 | 35 |  | 
 | 36 | MODULE_DESCRIPTION("AFS Client File System"); | 
 | 37 | MODULE_AUTHOR("Red Hat, Inc."); | 
 | 38 | MODULE_LICENSE("GPL"); | 
 | 39 |  | 
 | 40 | static char *rootcell; | 
 | 41 |  | 
 | 42 | module_param(rootcell, charp, 0); | 
 | 43 | MODULE_PARM_DESC(rootcell, "root AFS cell name and VL server IP addr list"); | 
 | 44 |  | 
 | 45 |  | 
 | 46 | static struct rxrpc_peer_ops afs_peer_ops = { | 
 | 47 | 	.adding		= afs_adding_peer, | 
 | 48 | 	.discarding	= afs_discarding_peer, | 
 | 49 | }; | 
 | 50 |  | 
 | 51 | struct list_head afs_cb_hash_tbl[AFS_CB_HASH_COUNT]; | 
 | 52 | DEFINE_SPINLOCK(afs_cb_hash_lock); | 
 | 53 |  | 
 | 54 | #ifdef AFS_CACHING_SUPPORT | 
 | 55 | static struct cachefs_netfs_operations afs_cache_ops = { | 
 | 56 | 	.get_page_cookie	= afs_cache_get_page_cookie, | 
 | 57 | }; | 
 | 58 |  | 
 | 59 | struct cachefs_netfs afs_cache_netfs = { | 
 | 60 | 	.name			= "afs", | 
 | 61 | 	.version		= 0, | 
 | 62 | 	.ops			= &afs_cache_ops, | 
 | 63 | }; | 
 | 64 | #endif | 
 | 65 |  | 
 | 66 | /*****************************************************************************/ | 
 | 67 | /* | 
 | 68 |  * initialise the AFS client FS module | 
 | 69 |  */ | 
 | 70 | static int __init afs_init(void) | 
 | 71 | { | 
 | 72 | 	int loop, ret; | 
 | 73 |  | 
 | 74 | 	printk(KERN_INFO "kAFS: Red Hat AFS client v0.1 registering.\n"); | 
 | 75 |  | 
 | 76 | 	/* initialise the callback hash table */ | 
 | 77 | 	spin_lock_init(&afs_cb_hash_lock); | 
 | 78 | 	for (loop = AFS_CB_HASH_COUNT - 1; loop >= 0; loop--) | 
 | 79 | 		INIT_LIST_HEAD(&afs_cb_hash_tbl[loop]); | 
 | 80 |  | 
 | 81 | 	/* register the /proc stuff */ | 
 | 82 | 	ret = afs_proc_init(); | 
 | 83 | 	if (ret < 0) | 
 | 84 | 		return ret; | 
 | 85 |  | 
 | 86 | #ifdef AFS_CACHING_SUPPORT | 
 | 87 | 	/* we want to be able to cache */ | 
 | 88 | 	ret = cachefs_register_netfs(&afs_cache_netfs, | 
 | 89 | 				     &afs_cache_cell_index_def); | 
 | 90 | 	if (ret < 0) | 
 | 91 | 		goto error; | 
 | 92 | #endif | 
 | 93 |  | 
 | 94 | #ifdef CONFIG_KEYS_TURNED_OFF | 
 | 95 | 	ret = afs_key_register(); | 
 | 96 | 	if (ret < 0) | 
 | 97 | 		goto error_cache; | 
 | 98 | #endif | 
 | 99 |  | 
 | 100 | 	/* initialise the cell DB */ | 
 | 101 | 	ret = afs_cell_init(rootcell); | 
 | 102 | 	if (ret < 0) | 
 | 103 | 		goto error_keys; | 
 | 104 |  | 
 | 105 | 	/* start the timeout daemon */ | 
 | 106 | 	ret = afs_kafstimod_start(); | 
 | 107 | 	if (ret < 0) | 
 | 108 | 		goto error_keys; | 
 | 109 |  | 
 | 110 | 	/* start the async operation daemon */ | 
 | 111 | 	ret = afs_kafsasyncd_start(); | 
 | 112 | 	if (ret < 0) | 
 | 113 | 		goto error_kafstimod; | 
 | 114 |  | 
 | 115 | 	/* create the RxRPC transport */ | 
 | 116 | 	ret = rxrpc_create_transport(7001, &afs_transport); | 
 | 117 | 	if (ret < 0) | 
 | 118 | 		goto error_kafsasyncd; | 
 | 119 |  | 
 | 120 | 	afs_transport->peer_ops = &afs_peer_ops; | 
 | 121 |  | 
 | 122 | 	/* register the filesystems */ | 
 | 123 | 	ret = afs_fs_init(); | 
 | 124 | 	if (ret < 0) | 
 | 125 | 		goto error_transport; | 
 | 126 |  | 
 | 127 | 	return ret; | 
 | 128 |  | 
 | 129 |  error_transport: | 
 | 130 | 	rxrpc_put_transport(afs_transport); | 
 | 131 |  error_kafsasyncd: | 
 | 132 | 	afs_kafsasyncd_stop(); | 
 | 133 |  error_kafstimod: | 
 | 134 | 	afs_kafstimod_stop(); | 
 | 135 |  error_keys: | 
 | 136 | #ifdef CONFIG_KEYS_TURNED_OFF | 
 | 137 | 	afs_key_unregister(); | 
 | 138 |  error_cache: | 
 | 139 | #endif | 
 | 140 | #ifdef AFS_CACHING_SUPPORT | 
 | 141 | 	cachefs_unregister_netfs(&afs_cache_netfs); | 
 | 142 |  error: | 
 | 143 | #endif | 
 | 144 | 	afs_cell_purge(); | 
 | 145 | 	afs_proc_cleanup(); | 
 | 146 | 	printk(KERN_ERR "kAFS: failed to register: %d\n", ret); | 
 | 147 | 	return ret; | 
 | 148 | } /* end afs_init() */ | 
 | 149 |  | 
 | 150 | /* XXX late_initcall is kludgy, but the only alternative seems to create | 
 | 151 |  * a transport upon the first mount, which is worse. Or is it? | 
 | 152 |  */ | 
 | 153 | late_initcall(afs_init);	/* must be called after net/ to create socket */ | 
 | 154 | /*****************************************************************************/ | 
 | 155 | /* | 
 | 156 |  * clean up on module removal | 
 | 157 |  */ | 
 | 158 | static void __exit afs_exit(void) | 
 | 159 | { | 
 | 160 | 	printk(KERN_INFO "kAFS: Red Hat AFS client v0.1 unregistering.\n"); | 
 | 161 |  | 
 | 162 | 	afs_fs_exit(); | 
 | 163 | 	rxrpc_put_transport(afs_transport); | 
 | 164 | 	afs_kafstimod_stop(); | 
 | 165 | 	afs_kafsasyncd_stop(); | 
 | 166 | 	afs_cell_purge(); | 
 | 167 | #ifdef CONFIG_KEYS_TURNED_OFF | 
 | 168 | 	afs_key_unregister(); | 
 | 169 | #endif | 
 | 170 | #ifdef AFS_CACHING_SUPPORT | 
 | 171 | 	cachefs_unregister_netfs(&afs_cache_netfs); | 
 | 172 | #endif | 
 | 173 | 	afs_proc_cleanup(); | 
 | 174 |  | 
 | 175 | } /* end afs_exit() */ | 
 | 176 |  | 
 | 177 | module_exit(afs_exit); | 
 | 178 |  | 
 | 179 | /*****************************************************************************/ | 
 | 180 | /* | 
 | 181 |  * notification that new peer record is being added | 
 | 182 |  * - called from krxsecd | 
 | 183 |  * - return an error to induce an abort | 
 | 184 |  * - mustn't sleep (caller holds an rwlock) | 
 | 185 |  */ | 
 | 186 | static int afs_adding_peer(struct rxrpc_peer *peer) | 
 | 187 | { | 
 | 188 | 	struct afs_server *server; | 
 | 189 | 	int ret; | 
 | 190 |  | 
 | 191 | 	_debug("kAFS: Adding new peer %08x\n", ntohl(peer->addr.s_addr)); | 
 | 192 |  | 
 | 193 | 	/* determine which server the peer resides in (if any) */ | 
 | 194 | 	ret = afs_server_find_by_peer(peer, &server); | 
 | 195 | 	if (ret < 0) | 
 | 196 | 		return ret; /* none that we recognise, so abort */ | 
 | 197 |  | 
 | 198 | 	_debug("Server %p{u=%d}\n", server, atomic_read(&server->usage)); | 
 | 199 |  | 
 | 200 | 	_debug("Cell %p{u=%d}\n", | 
 | 201 | 	       server->cell, atomic_read(&server->cell->usage)); | 
 | 202 |  | 
 | 203 | 	/* cross-point the structs under a global lock */ | 
 | 204 | 	spin_lock(&afs_server_peer_lock); | 
 | 205 | 	peer->user = server; | 
 | 206 | 	server->peer = peer; | 
 | 207 | 	spin_unlock(&afs_server_peer_lock); | 
 | 208 |  | 
 | 209 | 	afs_put_server(server); | 
 | 210 |  | 
 | 211 | 	return 0; | 
 | 212 | } /* end afs_adding_peer() */ | 
 | 213 |  | 
 | 214 | /*****************************************************************************/ | 
 | 215 | /* | 
 | 216 |  * notification that a peer record is being discarded | 
 | 217 |  * - called from krxiod or krxsecd | 
 | 218 |  */ | 
 | 219 | static void afs_discarding_peer(struct rxrpc_peer *peer) | 
 | 220 | { | 
 | 221 | 	struct afs_server *server; | 
 | 222 |  | 
 | 223 | 	_enter("%p",peer); | 
 | 224 |  | 
 | 225 | 	_debug("Discarding peer %08x (rtt=%lu.%lumS)\n", | 
 | 226 | 	       ntohl(peer->addr.s_addr), | 
 | 227 | 	       (long) (peer->rtt / 1000), | 
 | 228 | 	       (long) (peer->rtt % 1000)); | 
 | 229 |  | 
 | 230 | 	/* uncross-point the structs under a global lock */ | 
 | 231 | 	spin_lock(&afs_server_peer_lock); | 
 | 232 | 	server = peer->user; | 
 | 233 | 	if (server) { | 
 | 234 | 		peer->user = NULL; | 
 | 235 | 		server->peer = NULL; | 
 | 236 | 	} | 
 | 237 | 	spin_unlock(&afs_server_peer_lock); | 
 | 238 |  | 
 | 239 | 	_leave(""); | 
 | 240 |  | 
 | 241 | } /* end afs_discarding_peer() */ | 
 | 242 |  | 
 | 243 | /*****************************************************************************/ | 
 | 244 | /* | 
 | 245 |  * clear the dead space between task_struct and kernel stack | 
 | 246 |  * - called by supplying -finstrument-functions to gcc | 
 | 247 |  */ | 
 | 248 | #if 0 | 
 | 249 | void __cyg_profile_func_enter (void *this_fn, void *call_site) | 
 | 250 | __attribute__((no_instrument_function)); | 
 | 251 |  | 
 | 252 | void __cyg_profile_func_enter (void *this_fn, void *call_site) | 
 | 253 | { | 
 | 254 |        asm volatile("  movl    %%esp,%%edi     \n" | 
 | 255 |                     "  andl    %0,%%edi        \n" | 
 | 256 |                     "  addl    %1,%%edi        \n" | 
 | 257 |                     "  movl    %%esp,%%ecx     \n" | 
 | 258 |                     "  subl    %%edi,%%ecx     \n" | 
 | 259 |                     "  shrl    $2,%%ecx        \n" | 
 | 260 |                     "  movl    $0xedededed,%%eax     \n" | 
 | 261 |                     "  rep stosl               \n" | 
 | 262 |                     : | 
 | 263 |                     : "i"(~(THREAD_SIZE - 1)), "i"(sizeof(struct thread_info)) | 
 | 264 |                     : "eax", "ecx", "edi", "memory", "cc" | 
 | 265 |                     ); | 
 | 266 | } | 
 | 267 |  | 
 | 268 | void __cyg_profile_func_exit(void *this_fn, void *call_site) | 
 | 269 | __attribute__((no_instrument_function)); | 
 | 270 |  | 
 | 271 | void __cyg_profile_func_exit(void *this_fn, void *call_site) | 
 | 272 | { | 
 | 273 |        asm volatile("  movl    %%esp,%%edi     \n" | 
 | 274 |                     "  andl    %0,%%edi        \n" | 
 | 275 |                     "  addl    %1,%%edi        \n" | 
 | 276 |                     "  movl    %%esp,%%ecx     \n" | 
 | 277 |                     "  subl    %%edi,%%ecx     \n" | 
 | 278 |                     "  shrl    $2,%%ecx        \n" | 
 | 279 |                     "  movl    $0xdadadada,%%eax     \n" | 
 | 280 |                     "  rep stosl               \n" | 
 | 281 |                     : | 
 | 282 |                     : "i"(~(THREAD_SIZE - 1)), "i"(sizeof(struct thread_info)) | 
 | 283 |                     : "eax", "ecx", "edi", "memory", "cc" | 
 | 284 |                     ); | 
 | 285 | } | 
 | 286 | #endif |