Whamcloud - gitweb
- removed not used vars in client_obd_setup()
[fs/lustre-release.git] / lustre / ldlm / ldlm_lib.c
1 /* -*- mode: c; c-basic-offset: 8; indent-tabs-mode: nil; -*-
2  * vim:expandtab:shiftwidth=8:tabstop=8:
3  *
4  *  Copyright (c) 2003 Cluster File Systems, Inc.
5  *
6  *   This file is part of Lustre, http://www.lustre.org.
7  *
8  *   Lustre is free software; you can redistribute it and/or
9  *   modify it under the terms of version 2 of the GNU General Public
10  *   License as published by the Free Software Foundation.
11  *
12  *   Lustre is distributed in the hope that it will be useful,
13  *   but WITHOUT ANY WARRANTY; without even the implied warranty of
14  *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
15  *   GNU General Public License for more details.
16  *
17  *   You should have received a copy of the GNU General Public License
18  *   along with Lustre; if not, write to the Free Software
19  *   Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
20  */
21
22 #ifndef EXPORT_SYMTAB
23 # define EXPORT_SYMTAB
24 #endif
25 #define DEBUG_SUBSYSTEM S_LDLM
26
27 #ifdef __KERNEL__
28 # include <linux/module.h>
29 #else
30 # include <liblustre.h>
31 #endif
32 #include <linux/obd.h>
33 #include <linux/obd_ost.h> /* for LUSTRE_OSC_NAME */
34 #include <linux/lustre_mds.h> /* for LUSTRE_MDC_NAME */
35 #include <linux/lustre_mgmt.h>
36 #include <linux/lustre_dlm.h>
37 #include <linux/lustre_net.h>
38 #include <linux/lustre_sec.h>
39
40 /* @priority: if non-zero, move the selected to the list head
41  * @nocreate: if non-zero, only search in existed connections
42  */
43 static int import_set_conn(struct obd_import *imp, struct obd_uuid *uuid,
44                            int priority, int nocreate)
45 {
46         struct ptlrpc_connection *ptlrpc_conn;
47         struct obd_import_conn *imp_conn = NULL, *item;
48         int rc = 0;
49         ENTRY;
50
51         LASSERT(!(nocreate && !priority));
52
53         ptlrpc_conn = ptlrpc_uuid_to_connection(uuid);
54         if (!ptlrpc_conn) {
55                 CERROR("can't find connection %s\n", uuid->uuid);
56                 RETURN (-EINVAL);
57         }
58
59         if (!nocreate) {
60                 OBD_ALLOC(imp_conn, sizeof(*imp_conn));
61                 if (!imp_conn) {
62                         CERROR("fail to alloc memory\n");
63                         GOTO(out_put, rc = -ENOMEM);
64                 }
65         }
66
67         spin_lock(&imp->imp_lock);
68         list_for_each_entry(item, &imp->imp_conn_list, oic_item) {
69                 if (obd_uuid_equals(uuid, &item->oic_uuid)) {
70                         if (priority) {
71                                 list_del(&item->oic_item);
72                                 list_add(&item->oic_item, &imp->imp_conn_list);
73                                 item->oic_last_attempt = 0;
74                         }
75                         CDEBUG(D_HA, "imp %p@%s: find existed conn %s%s\n",
76                                imp, imp->imp_obd->obd_name, uuid->uuid,
77                                (priority ? ", move to head." : ""));
78                         spin_unlock(&imp->imp_lock);
79                         GOTO(out_free, rc = 0);
80                 }
81         }
82         /* not found */
83         if (!nocreate) {
84                 imp_conn->oic_conn = ptlrpc_conn;
85                 imp_conn->oic_uuid = *uuid;
86                 imp_conn->oic_last_attempt = 0;
87                 if (priority)
88                         list_add(&imp_conn->oic_item, &imp->imp_conn_list);
89                 else
90                         list_add_tail(&imp_conn->oic_item, &imp->imp_conn_list);
91                 CDEBUG(D_HA, "imp %p@%s: add connection %s at %s\n",
92                        imp, imp->imp_obd->obd_name, uuid->uuid,
93                        (priority ? "head" : "tail"));
94         } else
95                 rc = -ENOENT;
96
97         spin_unlock(&imp->imp_lock);
98         RETURN(0);
99 out_free:
100         if (imp_conn)
101                 OBD_FREE(imp_conn, sizeof(*imp_conn));
102 out_put:
103         ptlrpc_put_connection(ptlrpc_conn);
104         RETURN(rc);
105 }
106
107 int import_set_conn_priority(struct obd_import *imp, struct obd_uuid *uuid)
108 {
109         return import_set_conn(imp, uuid, 1, 1);
110 }
111
112 int client_import_add_conn(struct obd_import *imp, struct obd_uuid *uuid,
113                            int priority)
114 {
115         return import_set_conn(imp, uuid, priority, 0);
116 }
117
118 int client_import_del_conn(struct obd_import *imp, struct obd_uuid *uuid)
119 {
120         struct obd_import_conn *imp_conn;
121         struct obd_export *dlmexp;
122         int rc = -ENOENT;
123         ENTRY;
124
125         spin_lock(&imp->imp_lock);
126         if (list_empty(&imp->imp_conn_list)) {
127                 LASSERT(!imp->imp_conn_current);
128                 LASSERT(!imp->imp_connection);
129                 GOTO(out, rc);
130         }
131
132         list_for_each_entry(imp_conn, &imp->imp_conn_list, oic_item) {
133                 if (!obd_uuid_equals(uuid, &imp_conn->oic_uuid))
134                         continue;
135                 LASSERT(imp_conn->oic_conn);
136
137                 /* is current conn? */
138                 if (imp_conn == imp->imp_conn_current) {
139                         LASSERT(imp_conn->oic_conn == imp->imp_connection);
140
141                         if (imp->imp_state != LUSTRE_IMP_CLOSED &&
142                             imp->imp_state != LUSTRE_IMP_DISCON) {
143                                 CERROR("can't remove current connection\n");
144                                 GOTO(out, rc = -EBUSY);
145                         }
146
147                         ptlrpc_put_connection(imp->imp_connection);
148                         imp->imp_connection = NULL;
149
150                         dlmexp = class_conn2export(&imp->imp_dlm_handle);
151                         if (dlmexp && dlmexp->exp_connection) {
152                                 LASSERT(dlmexp->exp_connection ==
153                                         imp_conn->oic_conn);
154                                 ptlrpc_put_connection(dlmexp->exp_connection);
155                                 dlmexp->exp_connection = NULL;
156                         }
157                 }
158
159                 list_del(&imp_conn->oic_item);
160                 ptlrpc_put_connection(imp_conn->oic_conn);
161                 OBD_FREE(imp_conn, sizeof(*imp_conn));
162                 CDEBUG(D_HA, "imp %p@%s: remove connection %s\n",
163                        imp, imp->imp_obd->obd_name, uuid->uuid);
164                 rc = 0;
165                 break;
166         }
167 out:
168         spin_unlock(&imp->imp_lock);
169         if (rc == -ENOENT)
170                 CERROR("connection %s not found\n", uuid->uuid);
171         RETURN(rc);
172 }
173
174 int client_obd_setup(struct obd_device *obddev, obd_count len, void *buf)
175 {
176         struct lustre_cfg* lcfg = buf;
177         struct client_obd *cli = &obddev->u.cli;
178         struct obd_import *imp;
179         struct obd_uuid server_uuid;
180         int rq_portal, rp_portal, connect_op;
181         char *name = obddev->obd_type->typ_name;
182         char *mgmt_name = NULL;
183         int rc;
184         ENTRY;
185
186         /* In a more perfect world, we would hang a ptlrpc_client off of
187          * obd_type and just use the values from there. */
188         if (!strcmp(name, LUSTRE_OSC_NAME)) {
189                 rq_portal = OST_REQUEST_PORTAL;
190                 rp_portal = OSC_REPLY_PORTAL;
191                 connect_op = OST_CONNECT;
192         } else if (!strcmp(name, LUSTRE_MDC_NAME)) {
193                 rq_portal = MDS_REQUEST_PORTAL;
194                 rp_portal = MDC_REPLY_PORTAL;
195                 connect_op = MDS_CONNECT;
196         } else if (!strcmp(name, LUSTRE_MGMTCLI_NAME)) {
197                 rq_portal = MGMT_REQUEST_PORTAL;
198                 rp_portal = MGMT_REPLY_PORTAL;
199                 connect_op = MGMT_CONNECT;
200         } else {
201                 CERROR("unknown client OBD type \"%s\", can't setup\n",
202                        name);
203                 RETURN(-EINVAL);
204         }
205
206
207         if (LUSTRE_CFG_BUFLEN(lcfg, 1) < 1) {
208                 CERROR("requires a TARGET UUID\n");
209                 RETURN(-EINVAL);
210         }
211
212         if (LUSTRE_CFG_BUFLEN(lcfg, 1) > 37) {
213                 CERROR("client UUID must be less than 38 characters\n");
214                 RETURN(-EINVAL);
215         }
216
217         if (LUSTRE_CFG_BUFLEN(lcfg, 2) < 1) {
218                 CERROR("setup requires a SERVER UUID\n");
219                 RETURN(-EINVAL);
220         }
221
222         if (LUSTRE_CFG_BUFLEN(lcfg, 2) > 37) {
223                 CERROR("target UUID must be less than 38 characters\n");
224                 RETURN(-EINVAL);
225         }
226
227         sema_init(&cli->cl_sem, 1);
228         cli->cl_conn_count = 0;
229         memcpy(server_uuid.uuid,  lustre_cfg_buf(lcfg, 2),
230                min_t(unsigned int, LUSTRE_CFG_BUFLEN(lcfg, 2), 
231                sizeof(server_uuid)));
232
233         cli->cl_dirty = 0;
234         cli->cl_avail_grant = 0;
235         /* FIXME: should limit this for the sum of all cl_dirty_max */
236         cli->cl_dirty_max = OSC_MAX_DIRTY_DEFAULT * 1024 * 1024;
237         if (cli->cl_dirty_max >> PAGE_SHIFT > num_physpages / 8)
238                 cli->cl_dirty_max = num_physpages << (PAGE_SHIFT - 3);
239         INIT_LIST_HEAD(&cli->cl_cache_waiters);
240         INIT_LIST_HEAD(&cli->cl_loi_ready_list);
241         INIT_LIST_HEAD(&cli->cl_loi_write_list);
242         INIT_LIST_HEAD(&cli->cl_loi_read_list);
243         spin_lock_init(&cli->cl_loi_list_lock);
244         cli->cl_r_in_flight = 0;
245         cli->cl_w_in_flight = 0;
246         spin_lock_init(&cli->cl_read_rpc_hist.oh_lock);
247         spin_lock_init(&cli->cl_write_rpc_hist.oh_lock);
248         spin_lock_init(&cli->cl_read_page_hist.oh_lock);
249         spin_lock_init(&cli->cl_write_page_hist.oh_lock);
250
251         if (num_physpages >> (20 - PAGE_SHIFT) <= 128) { /* <= 128 MB */
252                 cli->cl_max_pages_per_rpc = PTLRPC_MAX_BRW_PAGES / 4;
253                 cli->cl_max_rpcs_in_flight = OSC_MAX_RIF_DEFAULT / 4;
254 #if 0
255         } else if (num_physpages >> (20 - PAGE_SHIFT) <= 512) { /* <= 512 MB */
256                 cli->cl_max_pages_per_rpc = PTLRPC_MAX_BRW_PAGES / 2;
257                 cli->cl_max_rpcs_in_flight = OSC_MAX_RIF_DEFAULT / 2;
258 #endif
259         } else {
260                 cli->cl_max_pages_per_rpc = PTLRPC_MAX_BRW_PAGES;
261                 cli->cl_max_rpcs_in_flight = OSC_MAX_RIF_DEFAULT;
262         }
263
264         rc = ldlm_get_ref();
265         if (rc) {
266                 CERROR("ldlm_get_ref failed: %d\n", rc);
267                 GOTO(err, rc);
268         }
269
270         ptlrpc_init_client(rq_portal, rp_portal, name,
271                            &obddev->obd_ldlm_client);
272
273         imp = class_new_import();
274         if (imp == NULL) 
275                 GOTO(err_ldlm, rc = -ENOENT);
276         imp->imp_client = &obddev->obd_ldlm_client;
277         imp->imp_obd = obddev;
278         imp->imp_connect_op = connect_op;
279         imp->imp_generation = 0;
280         imp->imp_initial_recov = 1;
281         INIT_LIST_HEAD(&imp->imp_pinger_chain);
282         memcpy(imp->imp_target_uuid.uuid, lustre_cfg_buf(lcfg, 1),
283                LUSTRE_CFG_BUFLEN(lcfg, 1));
284         class_import_put(imp);
285
286         rc = client_import_add_conn(imp, &server_uuid, 1);
287         if (rc) {
288                 CERROR("can't add initial connection\n");
289                 GOTO(err_import, rc);
290         }
291
292         cli->cl_import = imp;
293         cli->cl_max_mds_easize = sizeof(struct lov_mds_md);
294         cli->cl_max_mds_cookiesize = sizeof(struct llog_cookie);
295         cli->cl_sandev = to_kdev_t(0);
296
297         if (LUSTRE_CFG_BUFLEN(lcfg, 3) > 0) {
298                 if (!strcmp(lustre_cfg_string(lcfg, 3), "inactive")) {
299                         CDEBUG(D_HA, "marking %s %s->%s as inactive\n",
300                                name, obddev->obd_name,
301                                imp->imp_target_uuid.uuid);
302                         imp->imp_invalid = 1;
303
304                         if (LUSTRE_CFG_BUFLEN(lcfg, 4) > 0)
305                                 mgmt_name = lustre_cfg_string(lcfg, 4);
306                 } else {
307                         mgmt_name = lustre_cfg_string(lcfg, 3);
308                 }
309         }
310 #if 0
311         if (mgmt_name != NULL) {
312                 /* Register with management client if we need to. */
313                 CDEBUG(D_HA, "%s registering with %s for events about %s\n",
314                        obddev->obd_name, mgmt_name, server_uuid.uuid);
315
316                 mgmt_obd = class_name2obd(mgmt_name);
317                 if (!mgmt_obd) {
318                         CERROR("can't find mgmtcli %s to register\n",
319                                mgmt_name);
320                         GOTO(err_import, rc = -ENOSYS);
321                 }
322
323                 register_f = (mgmtcli_register_for_events_t)symbol_get("mgmtcli_register_for_events");
324                 if (!register_f) {
325                         CERROR("can't i_m_g mgmtcli_register_for_events\n");
326                         GOTO(err_import, rc = -ENOSYS);
327                 }
328
329                 rc = register_f(mgmt_obd, obddev, &imp->imp_target_uuid);
330                 symbol_put("mgmtcli_register_for_events");
331
332                 if (!rc)
333                         cli->cl_mgmtcli_obd = mgmt_obd;
334         }
335 #endif
336         RETURN(rc);
337
338 err_import:
339         class_destroy_import(imp);
340 err_ldlm:
341         ldlm_put_ref(0);
342 err:
343         RETURN(rc);
344
345 }
346
347 int client_obd_cleanup(struct obd_device *obddev, int flags)
348 {
349         struct client_obd *cli = &obddev->u.cli;
350         ENTRY;
351
352         if (!cli->cl_import)
353                 RETURN(-EINVAL);
354         if (cli->cl_mgmtcli_obd) {
355                 mgmtcli_deregister_for_events_t dereg_f;
356
357                 dereg_f = (mgmtcli_deregister_for_events_t)symbol_get("mgmtcli_deregister_for_events");
358                 dereg_f(cli->cl_mgmtcli_obd, obddev);
359                 symbol_put("mgmtcli_deregister_for_events");
360         }
361
362         /* Here we try to drop the security structure after destroy import,
363          * to avoid issue of "sleep in spinlock".
364          */
365         class_import_get(cli->cl_import);
366         class_destroy_import(cli->cl_import);
367         ptlrpcs_import_drop_sec(cli->cl_import);
368         class_import_put(cli->cl_import);
369         cli->cl_import = NULL;
370
371         ldlm_put_ref(flags & OBD_OPT_FORCE);
372
373         RETURN(0);
374 }
375
376 int client_connect_import(struct lustre_handle *dlm_handle,
377                           struct obd_device *obd,
378                           struct obd_uuid *cluuid,
379                           struct obd_connect_data *conn_data,
380                           unsigned long connect_flags)
381 {
382         struct client_obd *cli = &obd->u.cli;
383         struct obd_import *imp = cli->cl_import;
384         struct obd_export *exp;
385         int rc;
386         ENTRY;
387
388         down(&cli->cl_sem);
389         rc = class_connect(dlm_handle, obd, cluuid);
390         if (rc)
391                 GOTO(out_sem, rc);
392
393         cli->cl_conn_count++;
394         if (cli->cl_conn_count > 1)
395                 GOTO(out_sem, rc);
396         exp = class_conn2export(dlm_handle);
397
398         if (obd->obd_namespace != NULL)
399                 CERROR("already have namespace!\n");
400         obd->obd_namespace = ldlm_namespace_new(obd->obd_name,
401                                                 LDLM_NAMESPACE_CLIENT);
402         if (obd->obd_namespace == NULL)
403                 GOTO(out_disco, rc = -ENOMEM);
404
405         rc = ptlrpcs_import_get_sec(imp);
406         if (rc != 0)
407                 GOTO(out_ldlm, rc);
408
409         imp->imp_dlm_handle = *dlm_handle;
410         rc = ptlrpc_init_import(imp);
411         if (rc != 0) 
412                 GOTO(out_ldlm, rc);
413
414         imp->imp_connect_flags = connect_flags;
415         if (conn_data)
416                 memcpy(&imp->imp_connect_data, conn_data, sizeof(*conn_data));
417
418         rc = ptlrpc_connect_import(imp, NULL);
419         if (rc != 0) {
420                 LASSERT (imp->imp_state == LUSTRE_IMP_DISCON);
421                 GOTO(out_ldlm, rc);
422         }
423         LASSERT(exp->exp_connection);
424         ptlrpc_pinger_add_import(imp);
425         EXIT;
426
427         if (rc) {
428 out_ldlm:
429                 ldlm_namespace_free(obd->obd_namespace, 0);
430                 obd->obd_namespace = NULL;
431 out_disco:
432                 cli->cl_conn_count--;
433                 class_disconnect(exp, 0);
434         } else {
435                 class_export_put(exp);
436         }
437 out_sem:
438         up(&cli->cl_sem);
439         return rc;
440 }
441
442 int client_disconnect_export(struct obd_export *exp, unsigned long flags)
443 {
444         struct obd_device *obd = class_exp2obd(exp);
445         struct client_obd *cli = &obd->u.cli;
446         struct obd_import *imp = cli->cl_import;
447         int rc = 0, err;
448         ENTRY;
449
450         if (!obd) {
451                 CERROR("invalid export for disconnect: exp %p cookie "LPX64"\n",
452                        exp, exp ? exp->exp_handle.h_cookie : -1);
453                 RETURN(-EINVAL);
454         }
455
456         down(&cli->cl_sem);
457         if (!cli->cl_conn_count) {
458                 CERROR("disconnecting disconnected device (%s)\n",
459                        obd->obd_name);
460                 GOTO(out_sem, rc = -EINVAL);
461         }
462
463         cli->cl_conn_count--;
464         if (cli->cl_conn_count)
465                 GOTO(out_no_disconnect, rc = 0);
466
467         /* Some non-replayable imports (MDS's OSCs) are pinged, so just
468          * delete it regardless.  (It's safe to delete an import that was
469          * never added.) */
470         (void)ptlrpc_pinger_del_import(imp);
471
472         if (obd->obd_namespace != NULL) {
473                 /* obd_no_recov == local only */
474                 ldlm_cli_cancel_unused(obd->obd_namespace, NULL,
475                                        obd->obd_no_recov, NULL);
476                 ldlm_namespace_free(obd->obd_namespace, obd->obd_no_recov);
477                 obd->obd_namespace = NULL;
478         }
479
480         /* 
481          * Yeah, obd_no_recov also (mainly) means "forced shutdown".
482          */
483         if (obd->obd_no_recov)
484                 ptlrpc_invalidate_import(imp, 0);
485         else
486                 rc = ptlrpc_disconnect_import(imp);
487
488         EXIT;
489  out_no_disconnect:
490         err = class_disconnect(exp, 0);
491         if (!rc && err)
492                 rc = err;
493  out_sem:
494         up(&cli->cl_sem);
495         RETURN(rc);
496 }
497
498 /* --------------------------------------------------------------------------
499  * from old lib/target.c
500  * -------------------------------------------------------------------------- */
501
502 int target_handle_reconnect(struct lustre_handle *conn, struct obd_export *exp,
503                             struct obd_uuid *cluuid, int initial_conn)
504 {
505         if (exp->exp_connection && !initial_conn) {
506                 struct lustre_handle *hdl;
507                 hdl = &exp->exp_imp_reverse->imp_remote_handle;
508                 /* Might be a re-connect after a partition. */
509                 if (!memcmp(&conn->cookie, &hdl->cookie, sizeof conn->cookie)) {
510                         CERROR("%s reconnecting\n", cluuid->uuid);
511                         conn->cookie = exp->exp_handle.h_cookie;
512                         RETURN(EALREADY);
513                 } else {
514                         CERROR("%s reconnecting from %s, "
515                                "handle mismatch (ours "LPX64", theirs "
516                                LPX64")\n", cluuid->uuid,
517                                exp->exp_connection->c_remote_uuid.uuid,
518                                hdl->cookie, conn->cookie);
519                         memset(conn, 0, sizeof *conn);
520                         RETURN(-EALREADY);
521                 }
522         }
523
524         conn->cookie = exp->exp_handle.h_cookie;
525         CDEBUG(D_INFO, "existing export for UUID '%s' at %p\n",
526                cluuid->uuid, exp);
527         CDEBUG(D_IOCTL,"connect: cookie "LPX64"\n", conn->cookie);
528         RETURN(0);
529 }
530
531 static inline int ptlrpc_peer_is_local(struct ptlrpc_peer *peer)
532 {
533         ptl_process_id_t myid;
534
535         PtlGetId(peer->peer_ni->pni_ni_h, &myid);
536         return (memcmp(&peer->peer_id, &myid, sizeof(myid)) == 0);
537 }
538
539 /* To check whether the p_flavor is in deny list or not
540  * rc:
541  *      0           not found, pass
542  *      EPERM       found, refuse
543  */
544
545 static int check_deny_list(struct list_head *head, ptlrpcs_flavor_t *p_flavor)
546 {
547         deny_sec_t *p_deny_sec = NULL;
548         deny_sec_t *n_deny_sec = NULL;
549
550         list_for_each_entry_safe(p_deny_sec, n_deny_sec, head, list) {
551                 if ((p_deny_sec->sec.flavor == p_flavor->flavor) &&
552                     (p_deny_sec->sec.subflavor == p_flavor->subflavor))
553                         return -EPERM;
554         }
555         return 0;
556 }
557
558 int target_check_deny_sec(struct obd_device *target, struct ptlrpc_request *req)
559 {
560         struct gss_svc_data *svcdata;
561         ptlrpcs_flavor_t flavor;
562         int rc = 0;
563
564         /* XXX hacking */
565         svcdata = (struct gss_svc_data *) req->rq_sec_svcdata;
566         if (svcdata == NULL) {
567                 flavor.flavor = PTLRPC_SEC_NULL;
568                 flavor.subflavor = 0;
569         } else {
570                 flavor.flavor = PTLRPC_SEC_GSS;
571                 flavor.subflavor = svcdata->subflavor;
572         }
573
574         if (!strcmp(target->obd_type->typ_name, LUSTRE_MDS_NAME)) {
575                 spin_lock(&target->u.mds.mds_denylist_lock);
576                 rc = check_deny_list(&target->u.mds.mds_denylist, &flavor);
577                 spin_unlock(&target->u.mds.mds_denylist_lock);
578         } else if (!strcmp(target->obd_type->typ_name, "obdfilter")) {
579                 spin_lock(&target->u.filter.fo_denylist_lock);
580                 rc = check_deny_list(&target->u.filter.fo_denylist, &flavor);
581                 spin_unlock(&target->u.filter.fo_denylist_lock);
582         }
583
584         return rc;
585 }
586
587 int target_handle_connect(struct ptlrpc_request *req)
588 {
589         unsigned long connect_flags = 0, *cfp;
590         struct obd_device *target;
591         struct obd_export *export = NULL;
592         struct obd_import *revimp;
593         struct lustre_handle conn;
594         struct obd_uuid tgtuuid;
595         struct obd_uuid cluuid;
596         struct obd_uuid remote_uuid;
597         struct list_head *p;
598         struct obd_connect_data *conn_data;
599         int conn_data_size = sizeof(*conn_data);
600         char *str, *tmp;
601         int rc = 0;
602         unsigned long flags;
603         int initial_conn = 0;
604         char peer_str[PTL_NALFMT_SIZE];
605         const int offset = 1;
606         ENTRY;
607
608         OBD_RACE(OBD_FAIL_TGT_CONN_RACE); 
609
610         LASSERT_REQSWAB (req, offset + 0);
611         str = lustre_msg_string(req->rq_reqmsg, offset + 0,
612                                 sizeof(tgtuuid) - 1);
613         if (str == NULL) {
614                 CERROR("bad target UUID for connect\n");
615                 GOTO(out, rc = -EINVAL);
616         }
617
618         obd_str2uuid (&tgtuuid, str);
619         target = class_uuid2obd(&tgtuuid);
620         if (!target)
621                 target = class_name2obd(str);
622         
623         if (!target || target->obd_stopping || !target->obd_set_up) {
624                 CERROR("UUID '%s' is not available for connect from %s\n",
625                        str, req->rq_peerstr);
626                 GOTO(out, rc = -ENODEV);
627         }
628
629         /* check the secure deny list of mds/ost */
630         rc = target_check_deny_sec(target, req);
631         if (rc != 0)
632                 GOTO(out, rc);
633
634         LASSERT_REQSWAB (req, offset + 1);
635         str = lustre_msg_string(req->rq_reqmsg, offset + 1, sizeof(cluuid) - 1);
636         if (str == NULL) {
637                 CERROR("bad client UUID for connect\n");
638                 GOTO(out, rc = -EINVAL);
639         }
640
641         obd_str2uuid (&cluuid, str);
642
643         /* XXX extract a nettype and format accordingly */
644         switch (sizeof(ptl_nid_t)) {
645                 /* NB the casts only avoid compiler warnings */
646         case 8:
647                 snprintf((char *)remote_uuid.uuid, sizeof(remote_uuid),
648                          "NET_"LPX64"_UUID", (__u64)req->rq_peer.peer_id.nid);
649                 break;
650         case 4:
651                 snprintf((char *)remote_uuid.uuid, sizeof(remote_uuid),
652                          "NET_%x_UUID", (__u32)req->rq_peer.peer_id.nid);
653                 break;
654         default:
655                 LBUG();
656         }
657
658         tmp = lustre_msg_buf(req->rq_reqmsg, offset + 2, sizeof(conn));
659         if (tmp == NULL)
660                 GOTO(out, rc = -EPROTO);
661
662         memcpy(&conn, tmp, sizeof conn);
663
664         cfp = lustre_msg_buf(req->rq_reqmsg, offset + 3, sizeof(unsigned long));
665         LASSERT(cfp != NULL);
666         connect_flags = *cfp;
667
668         conn_data = lustre_swab_reqbuf(req, offset + 4, sizeof(*conn_data),
669                                        lustre_swab_connect);
670         if (!conn_data)
671                 GOTO(out, rc = -EPROTO);
672
673         rc = lustre_pack_reply(req, 1, &conn_data_size, NULL);
674         if (rc)
675                 GOTO(out, rc);
676         
677         if (lustre_msg_get_op_flags(req->rq_reqmsg) & MSG_CONNECT_INITIAL)
678                 initial_conn = 1;
679         
680         /* lctl gets a backstage, all-access pass. */
681         if (obd_uuid_equals(&cluuid, &target->obd_uuid))
682                 goto dont_check_exports;
683
684         spin_lock(&target->obd_dev_lock);
685         list_for_each(p, &target->obd_exports) {
686                 export = list_entry(p, struct obd_export, exp_obd_chain);
687                 if (obd_uuid_equals(&cluuid, &export->exp_client_uuid)) {
688                         spin_unlock(&target->obd_dev_lock);
689                         LASSERT(export->exp_obd == target);
690
691                         rc = target_handle_reconnect(&conn, export, &cluuid,
692                                                      initial_conn);
693                         break;
694                 }
695                 export = NULL;
696         }
697         /* If we found an export, we already unlocked. */
698         if (!export) {
699                 spin_unlock(&target->obd_dev_lock);
700         } else if (req->rq_export == NULL && 
701                    atomic_read(&export->exp_rpc_count) > 0) {
702                 CWARN("%s: refuse connection from %s/%s to 0x%p/%d\n",
703                       target->obd_name, cluuid.uuid,
704                       ptlrpc_peernid2str(&req->rq_peer, peer_str),
705                       export, atomic_read(&export->exp_refcount));
706                 GOTO(out, rc = -EBUSY);
707         } else if (req->rq_export != NULL &&
708                    atomic_read(&export->exp_rpc_count) > 1) {
709                 CWARN("%s: refuse reconnection from %s@%s to 0x%p/%d\n",
710                       target->obd_name, cluuid.uuid,
711                       ptlrpc_peernid2str(&req->rq_peer, peer_str),
712                       export, atomic_read(&export->exp_rpc_count));
713                 GOTO(out, rc = -EBUSY);
714         } else if (req->rq_reqmsg->conn_cnt == 1 && !initial_conn) {
715                 CERROR("%s reconnected with 1 conn_cnt; cookies not random?\n",
716                        cluuid.uuid);
717                 GOTO(out, rc = -EALREADY);
718         }
719
720         /* Tell the client if we're in recovery. */
721         /* If this is the first client, start the recovery timer */
722         CWARN("%s: connection from %s@%s/%lu %st"LPU64"\n", target->obd_name,
723               cluuid.uuid, ptlrpc_peernid2str(&req->rq_peer, peer_str), *cfp,
724               target->obd_recovering ? "recovering/" : "", conn_data->transno);
725
726         if (target->obd_recovering) {
727                 lustre_msg_add_op_flags(req->rq_repmsg, MSG_CONNECT_RECOVERING);
728                 target_start_recovery_timer(target);
729         }
730
731 #if 0
732         /* Tell the client if we support replayable requests */
733         if (target->obd_replayable)
734                 lustre_msg_add_op_flags(req->rq_repmsg, MSG_CONNECT_REPLAYABLE);
735 #endif
736
737         if (export == NULL) {
738                 if (target->obd_recovering) {
739                         CERROR("%s denying connection for new client %s@%s: "
740                                "%d clients in recovery for %lds\n", target->obd_name, 
741                                cluuid.uuid,
742                                ptlrpc_peernid2str(&req->rq_peer, peer_str),
743                                target->obd_recoverable_clients,
744                                (target->obd_recovery_timer.expires-jiffies)/HZ);
745                         rc = -EBUSY;
746                 } else {
747  dont_check_exports:
748                         rc = obd_connect(&conn, target, &cluuid, conn_data,
749                                          connect_flags);
750                 }
751         }
752
753         /* Return only the parts of obd_connect_data that we understand, so the
754          * client knows that we don't understand the rest. */
755         conn_data->ocd_connect_flags &= OBD_CONNECT_SUPPORTED;
756         memcpy(lustre_msg_buf(req->rq_repmsg, 0, sizeof(*conn_data)), conn_data,
757                sizeof(*conn_data));
758
759         /* Tell the client if we support replayable requests */
760         if (target->obd_replayable)
761                 lustre_msg_add_op_flags(req->rq_repmsg, MSG_CONNECT_REPLAYABLE);
762
763         /* If all else goes well, this is our RPC return code. */
764         req->rq_status = 0;
765
766         if (rc && rc != EALREADY)
767                 GOTO(out, rc);
768
769         req->rq_repmsg->handle = conn;
770
771         /* If the client and the server are the same node, we will already
772          * have an export that really points to the client's DLM export,
773          * because we have a shared handles table.
774          *
775          * XXX this will go away when shaver stops sending the "connect" handle
776          * in the real "remote handle" field of the request --phik 24 Apr 2003
777          */
778         if (req->rq_export != NULL)
779                 class_export_put(req->rq_export);
780
781         /* ownership of this export ref transfers to the request */
782         export = req->rq_export = class_conn2export(&conn);
783         LASSERT(export != NULL);
784
785         spin_lock_irqsave(&export->exp_lock, flags);
786         if (initial_conn) {
787                 req->rq_repmsg->conn_cnt = export->exp_conn_cnt + 1;
788         } else if (export->exp_conn_cnt >= req->rq_reqmsg->conn_cnt) {
789                 CERROR("%s@%s: already connected at a higher conn_cnt: %d > %d\n",
790                        cluuid.uuid, ptlrpc_peernid2str(&req->rq_peer, peer_str),
791                        export->exp_conn_cnt, 
792                        req->rq_reqmsg->conn_cnt);
793                 spin_unlock_irqrestore(&export->exp_lock, flags);
794                 GOTO(out, rc = -EALREADY);
795         } 
796         export->exp_conn_cnt = req->rq_reqmsg->conn_cnt;
797         spin_unlock_irqrestore(&export->exp_lock, flags);
798
799         /* request from liblustre? */
800         if (lustre_msg_get_op_flags(req->rq_reqmsg) & MSG_CONNECT_LIBCLIENT)
801                 export->exp_libclient = 1;
802
803         if (!(lustre_msg_get_op_flags(req->rq_reqmsg) & MSG_CONNECT_ASYNC) &&
804             ptlrpc_peer_is_local(&req->rq_peer)) {
805                 CWARN("%s: exp %p set sync\n", target->obd_name, export);
806                 export->exp_sync = 1;
807         } else {
808                 CDEBUG(D_HA, "%s: exp %p set async\n",target->obd_name,export);
809                 export->exp_sync = 0;
810         }
811
812         if (export->exp_connection != NULL)
813                 ptlrpc_put_connection(export->exp_connection);
814         export->exp_connection = ptlrpc_get_connection(&req->rq_peer,
815                                                        &remote_uuid);
816
817         if (rc == EALREADY) {
818                 /* We indicate the reconnection in a flag, not an error code. */
819                 lustre_msg_add_op_flags(req->rq_repmsg, MSG_CONNECT_RECONNECT);
820                 GOTO(out, rc = 0);
821         }
822
823         spin_lock_bh(&target->obd_processing_task_lock);
824         if (target->obd_recovering && export->exp_connected == 0) {
825                 __u64 t = conn_data->transno;
826                 export->exp_connected = 1;
827                 if ((lustre_msg_get_op_flags(req->rq_reqmsg) & MSG_CONNECT_TRANSNO)
828                                 && t < target->obd_next_recovery_transno)
829                         target->obd_next_recovery_transno = t;
830                 target->obd_connected_clients++;
831                 if (target->obd_connected_clients == target->obd_max_recoverable_clients)
832                         wake_up(&target->obd_next_transno_waitq);
833         }
834         spin_unlock_bh(&target->obd_processing_task_lock);
835
836         memcpy(&conn, lustre_msg_buf(req->rq_reqmsg, offset + 2, sizeof(conn)),
837                sizeof(conn));
838
839         if (export->exp_imp_reverse != NULL) {
840                 /* same logic as client_obd_cleanup */
841                 class_import_get(export->exp_imp_reverse);
842                 class_destroy_import(export->exp_imp_reverse);
843                 ptlrpcs_import_drop_sec(export->exp_imp_reverse);
844                 class_import_put(export->exp_imp_reverse);
845         }
846
847         /* for the rest part, we return -ENOTCONN in case of errors
848          * in order to let client initialize connection again.
849          */
850         revimp = export->exp_imp_reverse = class_new_import();
851         if (!revimp) {
852                 CERROR("fail to alloc new reverse import.\n");
853                 GOTO(out, rc = -ENOTCONN);
854         }
855
856         revimp->imp_connection = ptlrpc_connection_addref(export->exp_connection);
857         revimp->imp_client = &export->exp_obd->obd_ldlm_client;
858         revimp->imp_remote_handle = conn;
859         revimp->imp_obd = target;
860         revimp->imp_dlm_fake = 1;
861         revimp->imp_state = LUSTRE_IMP_FULL;
862
863         rc = ptlrpcs_import_get_sec(revimp);
864         if (rc) {
865                 CERROR("reverse import can not get sec: %d\n", rc);
866                 class_destroy_import(revimp);
867                 export->exp_imp_reverse = NULL;
868                 GOTO(out, rc = -ENOTCONN);
869         }
870
871         class_import_put(revimp);
872
873         rc = obd_connect_post(export, initial_conn, connect_flags);
874 out:
875         if (rc)
876                 req->rq_status = rc;
877         RETURN(rc);
878 }
879
880 int target_handle_disconnect(struct ptlrpc_request *req)
881 {
882         struct obd_export *exp;
883         int rc;
884         ENTRY;
885
886         rc = lustre_pack_reply(req, 0, NULL, NULL);
887         if (rc)
888                 RETURN(rc);
889
890         /* keep the rq_export around so we can send the reply */
891         exp = class_export_get(req->rq_export);
892         req->rq_status = obd_disconnect(exp, 0);
893         RETURN(0);
894 }
895
896 void target_destroy_export(struct obd_export *exp)
897 {
898         /* exports created from last_rcvd data, and "fake"
899            exports created by lctl don't have an import */
900         if (exp->exp_imp_reverse != NULL) {
901                 ptlrpcs_import_drop_sec(exp->exp_imp_reverse);
902                 class_destroy_import(exp->exp_imp_reverse);
903         }
904
905         /* We cancel locks at disconnect time, but this will catch any locks
906          * granted in a race with recovery-induced disconnect. */
907         if (exp->exp_obd->obd_namespace != NULL)
908                 ldlm_cancel_locks_for_export(exp);
909 }
910
911 /*
912  * Recovery functions
913  */
914
915 struct ptlrpc_request *
916 ptlrpc_clone_req( struct ptlrpc_request *orig_req) 
917 {
918         struct ptlrpc_request *copy_req;
919         struct lustre_msg *copy_reqmsg;
920
921         OBD_ALLOC(copy_req, sizeof *copy_req);
922         if (!copy_req)
923                 return NULL;
924         OBD_ALLOC(copy_reqmsg, orig_req->rq_reqlen);
925         if (!copy_reqmsg){
926                 OBD_FREE(copy_req, sizeof *copy_req);
927                 return NULL;
928         }
929
930         memcpy(copy_req, orig_req, sizeof *copy_req);
931         memcpy(copy_reqmsg, orig_req->rq_reqmsg, orig_req->rq_reqlen);
932         /* the copied req takes over the reply state and security data */
933         orig_req->rq_reply_state = NULL;
934         orig_req->rq_sec_svcdata = NULL;
935
936         copy_req->rq_reqmsg = copy_reqmsg;
937         class_export_get(copy_req->rq_export);
938         INIT_LIST_HEAD(&copy_req->rq_list);
939
940         return copy_req;
941 }
942
943 void ptlrpc_free_clone( struct ptlrpc_request *req) 
944 {
945         if (req->rq_svcsec)
946                 svcsec_cleanup_req(req);
947
948         class_export_put(req->rq_export);
949         list_del(&req->rq_list);
950         OBD_FREE(req->rq_reqmsg, req->rq_reqlen);
951         OBD_FREE(req, sizeof *req);
952 }
953
954 static void target_release_saved_req(struct ptlrpc_request *req)
955 {
956         if (req->rq_svcsec)
957                 svcsec_cleanup_req(req);
958
959         class_export_put(req->rq_export);
960         OBD_FREE(req->rq_reqmsg, req->rq_reqlen);
961         OBD_FREE(req, sizeof *req);
962 }
963
964 static void target_finish_recovery(struct obd_device *obd)
965 {
966         int rc;
967
968         ldlm_reprocess_all_ns(obd->obd_namespace);
969
970         /* when recovery finished, cleanup orphans on mds and ost */
971         if (OBT(obd) && OBP(obd, postrecov)) {
972                 rc = OBP(obd, postrecov)(obd);
973                 if (rc >= 0)
974                         CWARN("%s: all clients recovered, %d MDS "
975                               "orphans deleted\n", obd->obd_name, rc);
976                 else
977                         CERROR("postrecov failed %d\n", rc);
978         }
979
980         obd->obd_recovery_end = LTIME_S(CURRENT_TIME);
981         return;
982 }
983
984 static void abort_req_replay_queue(struct obd_device *obd)
985 {
986         struct ptlrpc_request *req;
987         struct list_head *tmp, *n;
988         int rc;
989
990         list_for_each_safe(tmp, n, &obd->obd_req_replay_queue) {
991                 req = list_entry(tmp, struct ptlrpc_request, rq_list);
992                 list_del(&req->rq_list);
993                 DEBUG_REQ(D_ERROR, req, "aborted:");
994                 req->rq_status = -ENOTCONN;
995                 req->rq_type = PTL_RPC_MSG_ERR;
996                 rc = lustre_pack_reply(req, 0, NULL, NULL);
997                 if (rc == 0) {
998                         ptlrpc_reply(req);
999                 } else {
1000                         DEBUG_REQ(D_ERROR, req,
1001                                   "packing failed for abort-reply; skipping");
1002                 }
1003                 target_release_saved_req(req);
1004         }
1005 }
1006
1007 static void abort_lock_replay_queue(struct obd_device *obd)
1008 {
1009         struct ptlrpc_request *req;
1010         struct list_head *tmp, *n;
1011         int rc;
1012
1013         list_for_each_safe(tmp, n, &obd->obd_lock_replay_queue) {
1014                 req = list_entry(tmp, struct ptlrpc_request, rq_list);
1015                 list_del(&req->rq_list);
1016                 DEBUG_REQ(D_ERROR, req, "aborted:");
1017                 req->rq_status = -ENOTCONN;
1018                 req->rq_type = PTL_RPC_MSG_ERR;
1019                 rc = lustre_pack_reply(req, 0, NULL, NULL);
1020                 if (rc == 0) {
1021                         ptlrpc_reply(req);
1022                 } else {
1023                         DEBUG_REQ(D_ERROR, req,
1024                                   "packing failed for abort-reply; skipping");
1025                 }
1026                 target_release_saved_req(req);
1027         }
1028 }
1029
1030 /* Called from a cleanup function if the device is being cleaned up
1031    forcefully.  The exports should all have been disconnected already,
1032    the only thing left to do is
1033      - clear the recovery flags
1034      - cancel the timer
1035      - free queued requests and replies, but don't send replies
1036    Because the obd_stopping flag is set, no new requests should be received.
1037
1038 */
1039 void target_cleanup_recovery(struct obd_device *obd)
1040 {
1041         struct list_head *tmp, *n;
1042         struct ptlrpc_request *req;
1043
1044         spin_lock_bh(&obd->obd_processing_task_lock);
1045         if (!obd->obd_recovering) {
1046                 spin_unlock_bh(&obd->obd_processing_task_lock);
1047                 EXIT;
1048                 return;
1049         }
1050         obd->obd_recovering = obd->obd_abort_recovery = 0;
1051         target_cancel_recovery_timer(obd);
1052         spin_unlock_bh(&obd->obd_processing_task_lock);
1053
1054         list_for_each_safe(tmp, n, &obd->obd_req_replay_queue) {
1055                 req = list_entry(tmp, struct ptlrpc_request, rq_list);
1056                 list_del(&req->rq_list);
1057                 LASSERT (req->rq_reply_state == 0);
1058                 target_release_saved_req(req);
1059         }
1060         list_for_each_safe(tmp, n, &obd->obd_lock_replay_queue) {
1061                 req = list_entry(tmp, struct ptlrpc_request, rq_list);
1062                 list_del(&req->rq_list);
1063                 LASSERT (req->rq_reply_state == 0);
1064                 target_release_saved_req(req);
1065         }
1066         list_for_each_safe(tmp, n, &obd->obd_final_req_queue) {
1067                 req = list_entry(tmp, struct ptlrpc_request, rq_list);
1068                 list_del(&req->rq_list);
1069                 LASSERT (req->rq_reply_state == 0);
1070                 target_release_saved_req(req);
1071         }
1072 }
1073
1074 #if 0
1075 static void target_abort_recovery(void *data)
1076 {
1077         struct obd_device *obd = data;
1078
1079         LASSERT(!obd->obd_recovering);
1080
1081         class_disconnect_stale_exports(obd, 0);
1082
1083         CERROR("%s: recovery period over; disconnecting unfinished clients.\n",
1084                obd->obd_name);
1085
1086         abort_recovery_queue(obd);
1087         target_finish_recovery(obd);
1088         ptlrpc_run_recovery_over_upcall(obd);
1089 }
1090 #endif
1091
1092 static void target_recovery_expired(unsigned long castmeharder)
1093 {
1094         struct obd_device *obd = (struct obd_device *)castmeharder;
1095         spin_lock_bh(&obd->obd_processing_task_lock);
1096         if (obd->obd_recovering)
1097                 obd->obd_abort_recovery = 1;
1098
1099         wake_up(&obd->obd_next_transno_waitq);
1100         spin_unlock_bh(&obd->obd_processing_task_lock);
1101 }
1102
1103
1104 /* obd_processing_task_lock should be held */
1105 void target_cancel_recovery_timer(struct obd_device *obd)
1106 {
1107         CDEBUG(D_HA, "%s: cancel recovery timer\n", obd->obd_name);
1108         del_timer(&obd->obd_recovery_timer);
1109 }
1110
1111 #ifdef __KERNEL__
1112 static void reset_recovery_timer(struct obd_device *obd)
1113 {
1114         spin_lock_bh(&obd->obd_processing_task_lock);
1115         if (!obd->obd_recovering) {
1116                 spin_unlock_bh(&obd->obd_processing_task_lock);
1117                 return;
1118         }                
1119         CDEBUG(D_HA, "timer will expire in %u seconds\n",
1120                OBD_RECOVERY_TIMEOUT / HZ);
1121         mod_timer(&obd->obd_recovery_timer, jiffies + OBD_RECOVERY_TIMEOUT);
1122         spin_unlock_bh(&obd->obd_processing_task_lock);
1123 }
1124 #endif
1125
1126 /* Only start it the first time called */
1127 void target_start_recovery_timer(struct obd_device *obd)
1128 {
1129         spin_lock_bh(&obd->obd_processing_task_lock);
1130         if (!obd->obd_recovering || timer_pending(&obd->obd_recovery_timer)) {
1131                 spin_unlock_bh(&obd->obd_processing_task_lock);
1132                 return;
1133         }
1134         CWARN("%s: starting recovery timer (%us)\n", obd->obd_name,
1135                OBD_RECOVERY_TIMEOUT / HZ);
1136         obd->obd_recovery_timer.function = target_recovery_expired;
1137         obd->obd_recovery_timer.data = (unsigned long)obd;
1138         mod_timer(&obd->obd_recovery_timer, jiffies + OBD_RECOVERY_TIMEOUT);
1139         spin_unlock_bh(&obd->obd_processing_task_lock);
1140 }
1141
1142 #ifdef __KERNEL__
1143 static int check_for_next_transno(struct obd_device *obd)
1144 {
1145         struct ptlrpc_request *req = NULL;
1146         int wake_up = 0, connected, completed, queue_len, max;
1147         __u64 next_transno, req_transno;
1148
1149         spin_lock_bh(&obd->obd_processing_task_lock);
1150         if (!list_empty(&obd->obd_req_replay_queue)) {
1151                 req = list_entry(obd->obd_req_replay_queue.next,
1152                                  struct ptlrpc_request, rq_list);
1153                 req_transno = req->rq_reqmsg->transno;
1154         } else {
1155                 req_transno = 0;
1156         }
1157
1158         max = obd->obd_max_recoverable_clients;
1159         connected = obd->obd_connected_clients;
1160         completed = max - obd->obd_recoverable_clients;
1161         queue_len = obd->obd_requests_queued_for_recovery;
1162         next_transno = obd->obd_next_recovery_transno;
1163
1164         CDEBUG(D_HA,"max: %d, connected: %d, completed: %d, queue_len: %d, "
1165                "req_transno: "LPU64", next_transno: "LPU64"\n",
1166                max, connected, completed, queue_len, req_transno, next_transno);
1167         if (obd->obd_abort_recovery) {
1168                 CDEBUG(D_HA, "waking for aborted recovery\n");
1169                 wake_up = 1;
1170         } else if (atomic_read(&obd->obd_req_replay_clients) == 0) {
1171                 CDEBUG(D_HA, "waking for completed recovery\n");
1172                 wake_up = 1;
1173         } else if (req_transno == next_transno) {
1174                 CDEBUG(D_HA, "waking for next ("LPD64")\n", next_transno);
1175                 wake_up = 1;
1176         } else if (queue_len + completed == max) {
1177                 LASSERT(req->rq_reqmsg->transno >= next_transno);
1178                 CDEBUG(req_transno > obd->obd_last_committed ? D_ERROR : D_HA,
1179                        "waking for skipped transno (skip: "LPD64
1180                        ", ql: %d, comp: %d, conn: %d, next: "LPD64")\n",
1181                        next_transno, queue_len, completed, max, req_transno);
1182                 obd->obd_next_recovery_transno = req_transno;
1183                 wake_up = 1;
1184         } else if (queue_len == atomic_read(&obd->obd_req_replay_clients)) {
1185                 /* some clients haven't connected in time, but we need
1186                  * their requests to continue recovery. so, we abort ... */
1187                 CDEBUG(D_ERROR, "abort due to missed clients: queue: %d max: %d\n",
1188                        queue_len, max);
1189                 obd->obd_abort_recovery = 1;
1190                 wake_up = 1;
1191         }
1192         spin_unlock_bh(&obd->obd_processing_task_lock);
1193         
1194         return wake_up;
1195 }
1196
1197 static struct ptlrpc_request *
1198 target_next_replay_req(struct obd_device *obd)
1199 {
1200         struct l_wait_info lwi = { 0 };
1201         struct ptlrpc_request *req;
1202
1203         CDEBUG(D_HA, "Waiting for transno "LPD64"\n",
1204                obd->obd_next_recovery_transno);
1205         l_wait_event(obd->obd_next_transno_waitq,
1206                      check_for_next_transno(obd), &lwi);
1207         
1208         spin_lock_bh(&obd->obd_processing_task_lock);
1209         if (obd->obd_abort_recovery) {
1210                 req = NULL;
1211         } else if (!list_empty(&obd->obd_req_replay_queue)) {
1212                 req = list_entry(obd->obd_req_replay_queue.next,
1213                                  struct ptlrpc_request, rq_list);
1214                 list_del_init(&req->rq_list);
1215                 obd->obd_requests_queued_for_recovery--;
1216         } else {
1217                 req = NULL;
1218         }
1219         spin_unlock_bh(&obd->obd_processing_task_lock);
1220         return req;
1221 }
1222
1223 static int check_for_next_lock(struct obd_device *obd)
1224 {
1225         struct ptlrpc_request *req = NULL;
1226         int wake_up = 0;
1227
1228         spin_lock_bh(&obd->obd_processing_task_lock);
1229         if (!list_empty(&obd->obd_lock_replay_queue)) {
1230                 req = list_entry(obd->obd_lock_replay_queue.next,
1231                                  struct ptlrpc_request, rq_list);
1232                 CDEBUG(D_HA, "waking for next lock\n");
1233                 wake_up = 1;
1234         } else if (atomic_read(&obd->obd_lock_replay_clients) == 0) {
1235                 CDEBUG(D_HA, "waking for completed lock replay\n");
1236                 wake_up = 1;
1237         } else if (obd->obd_abort_recovery) {
1238                 CDEBUG(D_HA, "waking for aborted recovery\n");
1239                 wake_up = 1;
1240         }
1241         spin_unlock_bh(&obd->obd_processing_task_lock);
1242         
1243         return wake_up;
1244 }
1245
1246 static struct ptlrpc_request *
1247 target_next_replay_lock(struct obd_device *obd)
1248 {
1249         struct l_wait_info lwi = { 0 };
1250         struct ptlrpc_request *req;
1251
1252         CDEBUG(D_HA, "Waiting for lock\n");
1253         l_wait_event(obd->obd_next_transno_waitq,
1254                      check_for_next_lock(obd), &lwi);
1255         
1256         spin_lock_bh(&obd->obd_processing_task_lock);
1257         if (obd->obd_abort_recovery) {
1258                 req = NULL;
1259         } else if (!list_empty(&obd->obd_lock_replay_queue)) {
1260                 req = list_entry(obd->obd_lock_replay_queue.next,
1261                                  struct ptlrpc_request, rq_list);
1262                 list_del_init(&req->rq_list);
1263         } else {
1264                 req = NULL;
1265         }
1266         spin_unlock_bh(&obd->obd_processing_task_lock);
1267         return req;
1268 }
1269
1270 static struct ptlrpc_request *
1271 target_next_final_ping(struct obd_device *obd)
1272 {
1273         struct ptlrpc_request *req;
1274
1275         spin_lock_bh(&obd->obd_processing_task_lock);
1276         if (!list_empty(&obd->obd_final_req_queue)) {
1277                 req = list_entry(obd->obd_final_req_queue.next,
1278                                  struct ptlrpc_request, rq_list);
1279                 list_del_init(&req->rq_list);
1280         } else {
1281                 req = NULL;
1282         }
1283         spin_unlock_bh(&obd->obd_processing_task_lock);
1284         return req;
1285 }
1286
1287 static int req_replay_done(struct obd_export *exp)
1288 {
1289         if (exp->exp_req_replay_needed)
1290                 return 0;
1291         return 1;
1292 }
1293
1294 static int lock_replay_done(struct obd_export *exp)
1295 {
1296         if (exp->exp_lock_replay_needed)
1297                 return 0;
1298         return 1;
1299 }
1300
1301 static int connect_done(struct obd_export *exp)
1302 {
1303         if (exp->exp_connected)
1304                 return 1;
1305         return 0;
1306 }
1307
1308 static int check_for_clients(struct obd_device *obd)
1309 {
1310         if (obd->obd_abort_recovery)
1311                 return 1;
1312         LASSERT(obd->obd_connected_clients <= obd->obd_max_recoverable_clients);
1313         if (obd->obd_connected_clients == obd->obd_max_recoverable_clients)
1314                 return 1;
1315         return 0;
1316 }
1317
1318 static int target_recovery_thread(void *arg)
1319 {
1320         struct obd_device *obd = arg;
1321         struct ptlrpc_request *req;
1322         struct target_recovery_data *trd = &obd->obd_recovery_data;
1323         char peer_str[PTL_NALFMT_SIZE];
1324         struct l_wait_info lwi = { 0 };
1325         unsigned long flags;
1326         ENTRY;
1327
1328         kportal_daemonize("tgt-recov");
1329
1330         SIGNAL_MASK_LOCK(current, flags);
1331         sigfillset(&current->blocked);
1332         RECALC_SIGPENDING;
1333         SIGNAL_MASK_UNLOCK(current, flags);
1334
1335         CERROR("%s: started recovery thread pid %d\n", obd->obd_name, 
1336                current->pid);
1337         trd->trd_processing_task = current->pid;
1338
1339         obd->obd_recovering = 1;
1340         complete(&trd->trd_starting);
1341
1342         /* first of all, we have to know the first transno to replay */
1343         obd->obd_abort_recovery = 0;
1344         l_wait_event(obd->obd_next_transno_waitq,
1345                      check_for_clients(obd), &lwi);
1346         
1347         spin_lock_bh(&obd->obd_processing_task_lock);
1348         target_cancel_recovery_timer(obd);
1349         spin_unlock_bh(&obd->obd_processing_task_lock);
1350
1351         /* If some clients haven't connected in time, evict them */
1352         if (obd->obd_abort_recovery) {
1353                 int stale;
1354                 CERROR("some clients haven't connect in time (%d/%d),"
1355                        "evict them ...\n", obd->obd_connected_clients,
1356                        obd->obd_max_recoverable_clients);
1357                 obd->obd_abort_recovery = 0;
1358                 stale = class_disconnect_stale_exports(obd, connect_done, 0);
1359                 atomic_sub(stale, &obd->obd_req_replay_clients);
1360                 atomic_sub(stale, &obd->obd_lock_replay_clients);
1361         }
1362
1363         /* next stage: replay requests */
1364         CWARN("1: request replay stage - %d clients from t"LPU64"\n",
1365               atomic_read(&obd->obd_req_replay_clients),
1366               obd->obd_next_recovery_transno);
1367         while ((req = target_next_replay_req(obd))) {
1368                 LASSERT(trd->trd_processing_task == current->pid);
1369                 DEBUG_REQ(D_HA, req, "processing t"LPD64" from %s: ", 
1370                           req->rq_reqmsg->transno, 
1371                           ptlrpc_peernid2str(&req->rq_peer, peer_str));
1372                 (void)trd->trd_recovery_handler(req);
1373                 obd->obd_replayed_requests++;
1374                 reset_recovery_timer(obd);
1375                 /* bug 1580: decide how to properly sync() in recovery*/
1376                 //mds_fsync_super(mds->mds_sb);
1377                 ptlrpc_free_clone(req);
1378                 spin_lock_bh(&obd->obd_processing_task_lock);
1379                 obd->obd_next_recovery_transno++;
1380                 spin_unlock_bh(&obd->obd_processing_task_lock);
1381         }
1382
1383         spin_lock_bh(&obd->obd_processing_task_lock);
1384         target_cancel_recovery_timer(obd);
1385         spin_unlock_bh(&obd->obd_processing_task_lock);
1386
1387         /* If some clients haven't replayed requests in time, evict them */
1388         if (obd->obd_abort_recovery) {
1389                 int stale;
1390                 CERROR("req replay timed out, aborting ...\n");
1391                 obd->obd_abort_recovery = 0;
1392                 stale = class_disconnect_stale_exports(obd, req_replay_done, 0);
1393                 atomic_sub(stale, &obd->obd_lock_replay_clients);
1394                 abort_req_replay_queue(obd);
1395         }
1396
1397         /* The second stage: replay locks */
1398         CWARN("2: lock replay stage - %d clients\n",
1399               atomic_read(&obd->obd_lock_replay_clients));
1400         while ((req = target_next_replay_lock(obd))) {
1401                 LASSERT(trd->trd_processing_task == current->pid);
1402                 DEBUG_REQ(D_HA, req, "processing lock from %s: ", 
1403                           ptlrpc_peernid2str(&req->rq_peer, peer_str));
1404                 (void)trd->trd_recovery_handler(req);
1405                 reset_recovery_timer(obd);
1406                 ptlrpc_free_clone(req);
1407                 obd->obd_replayed_locks++;
1408         }
1409         
1410         spin_lock_bh(&obd->obd_processing_task_lock);
1411         target_cancel_recovery_timer(obd);
1412         spin_unlock_bh(&obd->obd_processing_task_lock);
1413
1414         /* If some clients haven't replayed requests in time, evict them */
1415         if (obd->obd_abort_recovery) {
1416                 int stale;
1417                 CERROR("lock replay timed out, aborting ...\n");
1418                 obd->obd_abort_recovery = 0;
1419                 stale = class_disconnect_stale_exports(obd, lock_replay_done, 0);
1420                 abort_lock_replay_queue(obd);
1421         }
1422
1423         /* We drop recoverying flag to forward all new requests
1424          * to regular mds_handle() since now */
1425         spin_lock_bh(&obd->obd_processing_task_lock);
1426         obd->obd_recovering = 0;
1427         spin_unlock_bh(&obd->obd_processing_task_lock);
1428
1429         /* The third stage: reply on final pings */
1430         CWARN("3: final stage - process recovery completion pings\n");
1431         while ((req = target_next_final_ping(obd))) {
1432                 LASSERT(trd->trd_processing_task == current->pid);
1433                 DEBUG_REQ(D_HA, req, "processing final ping from %s: ", 
1434                           ptlrpc_peernid2str(&req->rq_peer, peer_str));
1435                 (void)trd->trd_recovery_handler(req);
1436                 ptlrpc_free_clone(req);
1437         }
1438         
1439         CWARN("4: recovery completed - %d/%d reqs/locks replayed\n",
1440               obd->obd_replayed_requests, obd->obd_replayed_locks);
1441         target_finish_recovery(obd);
1442
1443         trd->trd_processing_task = 0;
1444         complete(&trd->trd_finishing);
1445         return 0;
1446 }
1447
1448 int target_start_recovery_thread(struct obd_device *obd, svc_handler_t handler)
1449 {
1450         int rc = 0;
1451         struct target_recovery_data *trd = &obd->obd_recovery_data;
1452
1453         memset(trd, 0, sizeof(*trd));
1454         init_completion(&trd->trd_starting);
1455         init_completion(&trd->trd_finishing);
1456         trd->trd_recovery_handler = handler;
1457
1458         if (kernel_thread(target_recovery_thread, obd, 0) > 0) {
1459                 wait_for_completion(&trd->trd_starting);
1460                 LASSERT(obd->obd_recovering != 0);
1461         } else
1462                 rc = -ECHILD;
1463
1464         return rc;
1465 }
1466
1467 void target_stop_recovery_thread(struct obd_device *obd)
1468 {
1469         spin_lock_bh(&obd->obd_processing_task_lock);
1470         if (obd->obd_recovery_data.trd_processing_task > 0) {
1471                 struct target_recovery_data *trd = &obd->obd_recovery_data;
1472                 CERROR("%s: aborting recovery\n", obd->obd_name);
1473                 obd->obd_abort_recovery = 1;
1474                 wake_up(&obd->obd_next_transno_waitq);
1475                 spin_unlock_bh(&obd->obd_processing_task_lock);
1476                 wait_for_completion(&trd->trd_finishing);
1477         } else {
1478                 spin_unlock_bh(&obd->obd_processing_task_lock);
1479         }
1480 }
1481 #endif
1482
1483 int target_process_req_flags(struct obd_device *obd, struct ptlrpc_request *req)
1484 {
1485         struct obd_export *exp = req->rq_export;
1486         LASSERT(exp != NULL);
1487         if (lustre_msg_get_flags(req->rq_reqmsg) & MSG_REQ_REPLAY_DONE) {
1488                 /* client declares he's ready to replay locks */
1489                 spin_lock_bh(&obd->obd_processing_task_lock);
1490                 if (exp->exp_req_replay_needed) {
1491                         LASSERT(atomic_read(&obd->obd_req_replay_clients) > 0);
1492                         exp->exp_req_replay_needed = 0;
1493                         atomic_dec(&obd->obd_req_replay_clients);
1494                         obd->obd_recoverable_clients--;
1495                         if (atomic_read(&obd->obd_req_replay_clients) == 0)
1496                                 CDEBUG(D_HA, "all clients have replayed reqs\n");
1497                         wake_up(&obd->obd_next_transno_waitq);
1498                 }
1499                 spin_unlock_bh(&obd->obd_processing_task_lock);
1500         }
1501         if (lustre_msg_get_flags(req->rq_reqmsg) & MSG_LOCK_REPLAY_DONE) {
1502                 /* client declares he's ready to complete recovery 
1503                  * so, we put the request on th final queue */
1504                 spin_lock_bh(&obd->obd_processing_task_lock);
1505                 if (exp->exp_lock_replay_needed) {
1506                         LASSERT(atomic_read(&obd->obd_lock_replay_clients) > 0);
1507                         exp->exp_lock_replay_needed = 0;
1508                         atomic_dec(&obd->obd_lock_replay_clients);
1509                         if (atomic_read(&obd->obd_lock_replay_clients) == 0)
1510                                 CDEBUG(D_HA, "all clients have replayed locks\n");
1511                         wake_up(&obd->obd_next_transno_waitq);
1512                 }
1513                 spin_unlock_bh(&obd->obd_processing_task_lock);
1514         }
1515
1516         return 0;
1517 }
1518
1519 int target_queue_recovery_request(struct ptlrpc_request *req,
1520                                   struct obd_device *obd)
1521 {
1522         struct list_head *tmp;
1523         int inserted = 0;
1524         __u64 transno = req->rq_reqmsg->transno;
1525
1526         if (obd->obd_recovery_data.trd_processing_task == current->pid) {
1527                 /* Processing the queue right now, don't re-add. */
1528                 return 1;
1529         }
1530
1531         target_process_req_flags(obd, req);
1532
1533         if (lustre_msg_get_flags(req->rq_reqmsg) & MSG_LOCK_REPLAY_DONE) {
1534                 /* client declares he's ready to complete recovery 
1535                  * so, we put the request on th final queue */
1536                 req = ptlrpc_clone_req(req);
1537                 if (req == NULL)
1538                         return -ENOMEM;
1539                 DEBUG_REQ(D_HA, req, "queue final req");
1540                 spin_lock_bh(&obd->obd_processing_task_lock);
1541                 list_add_tail(&req->rq_list, &obd->obd_final_req_queue);
1542                 spin_unlock_bh(&obd->obd_processing_task_lock);
1543                 return 0;
1544         }
1545         if (lustre_msg_get_flags(req->rq_reqmsg) & MSG_REQ_REPLAY_DONE) {
1546                 /* client declares he's ready to replay locks */
1547                 req = ptlrpc_clone_req(req);
1548                 if (req == NULL)
1549                         return -ENOMEM;
1550                 DEBUG_REQ(D_HA, req, "queue lock replay req");
1551                 spin_lock_bh(&obd->obd_processing_task_lock);
1552                 list_add_tail(&req->rq_list, &obd->obd_lock_replay_queue);
1553                 spin_unlock_bh(&obd->obd_processing_task_lock);
1554                 wake_up(&obd->obd_next_transno_waitq);
1555                 return 0;
1556         }
1557
1558
1559         /* CAVEAT EMPTOR: The incoming request message has been swabbed
1560          * (i.e. buflens etc are in my own byte order), but type-dependent
1561          * buffers (eg mds_body, ost_body etc) have NOT been swabbed. */
1562
1563         if (!transno) {
1564                 INIT_LIST_HEAD(&req->rq_list);
1565                 DEBUG_REQ(D_HA, req, "not queueing");
1566                 return 1;
1567         }
1568
1569
1570         /* If we're processing the queue, we want don't want to queue this
1571          * message.
1572          *
1573          * Also, if this request has a transno less than the one we're waiting
1574          * for, we should process it now.  It could (and currently always will)
1575          * be an open request for a descriptor that was opened some time ago.
1576          *
1577          * Also, a resent, replayed request that has already been
1578          * handled will pass through here and be processed immediately.
1579          */
1580         spin_lock_bh(&obd->obd_processing_task_lock);
1581         if (transno < obd->obd_next_recovery_transno && check_for_clients(obd)) {
1582                 /* Processing the queue right now, don't re-add. */
1583                 LASSERT(list_empty(&req->rq_list));
1584                 spin_unlock_bh(&obd->obd_processing_task_lock);
1585                 return 1;
1586         }
1587         spin_unlock_bh(&obd->obd_processing_task_lock);
1588
1589         /* A resent, replayed request that is still on the queue; just drop it.
1590            The queued request will handle this. */
1591         if ((lustre_msg_get_flags(req->rq_reqmsg) & (MSG_RESENT | MSG_REPLAY))
1592             == (MSG_RESENT | MSG_REPLAY)) {
1593                 DEBUG_REQ(D_ERROR, req, "dropping resent queued req");
1594                 return 0;
1595         }
1596
1597         req = ptlrpc_clone_req(req);
1598         if (req == NULL)
1599                 return -ENOMEM;
1600
1601         spin_lock_bh(&obd->obd_processing_task_lock);
1602
1603         /* XXX O(n^2) */
1604         list_for_each(tmp, &obd->obd_req_replay_queue) {
1605                 struct ptlrpc_request *reqiter =
1606                         list_entry(tmp, struct ptlrpc_request, rq_list);
1607
1608                 if (reqiter->rq_reqmsg->transno > transno) {
1609                         list_add_tail(&req->rq_list, &reqiter->rq_list);
1610                         inserted = 1;
1611                         break;
1612                 }
1613         }
1614
1615         if (!inserted)
1616                 list_add_tail(&req->rq_list, &obd->obd_req_replay_queue);
1617
1618         obd->obd_requests_queued_for_recovery++;
1619         wake_up(&obd->obd_next_transno_waitq);
1620         spin_unlock_bh(&obd->obd_processing_task_lock);
1621         return 0;
1622 }
1623
1624 struct obd_device * target_req2obd(struct ptlrpc_request *req)
1625 {
1626         return req->rq_export->exp_obd;
1627 }
1628
1629 int
1630 target_send_reply_msg (struct ptlrpc_request *req, int rc, int fail_id)
1631 {
1632         if (OBD_FAIL_CHECK(fail_id | OBD_FAIL_ONCE)) {
1633                 obd_fail_loc |= OBD_FAIL_ONCE | OBD_FAILED;
1634                 DEBUG_REQ(D_ERROR, req, "dropping reply");
1635                 /* NB this does _not_ send with ACK disabled, to simulate
1636                  * sending OK, but timing out for the ACK */
1637                 if (req->rq_reply_state != NULL) {
1638                         if (!req->rq_reply_state->rs_difficult) {
1639                                 lustre_free_reply_state (req->rq_reply_state);
1640                                 req->rq_reply_state = NULL;
1641                         } else {
1642                                 struct ptlrpc_service *svc =
1643                                         req->rq_rqbd->rqbd_srv_ni->sni_service;
1644                                 atomic_inc(&svc->srv_outstanding_replies);
1645                         }
1646                 }
1647                 return (-ECOMM);
1648         }
1649
1650         if (rc) {
1651                 req->rq_status = rc;
1652                 return (ptlrpc_error(req));
1653         } else {
1654                 DEBUG_REQ(D_NET, req, "sending reply");
1655         }
1656         
1657         return (ptlrpc_send_reply(req, 1));
1658 }
1659
1660 void 
1661 target_send_reply(struct ptlrpc_request *req, int rc, int fail_id)
1662 {
1663         int                        netrc;
1664         unsigned long              flags;
1665         struct ptlrpc_reply_state *rs;
1666         struct obd_device         *obd;
1667         struct obd_export         *exp;
1668         struct ptlrpc_srv_ni      *sni;
1669         struct ptlrpc_service     *svc;
1670
1671         sni = req->rq_rqbd->rqbd_srv_ni;
1672         svc = sni->sni_service;
1673         
1674         rs = req->rq_reply_state;
1675         if (rs == NULL || !rs->rs_difficult) {
1676                 /* The easy case; no notifiers and reply_out_callback()
1677                  * cleans up (i.e. we can't look inside rs after a
1678                  * successful send) */
1679                 netrc = target_send_reply_msg (req, rc, fail_id);
1680
1681                 LASSERT (netrc == 0 || req->rq_reply_state == NULL);
1682                 return;
1683         }
1684
1685         /* must be an export if locks saved */
1686         LASSERT (req->rq_export != NULL);
1687         /* req/reply consistent */
1688         LASSERT (rs->rs_srv_ni == sni);
1689
1690         /* "fresh" reply */
1691         LASSERT (!rs->rs_scheduled);
1692         LASSERT (!rs->rs_scheduled_ever);
1693         LASSERT (!rs->rs_handled);
1694         LASSERT (!rs->rs_on_net);
1695         LASSERT (rs->rs_export == NULL);
1696         LASSERT (list_empty(&rs->rs_obd_list));
1697         LASSERT (list_empty(&rs->rs_exp_list));
1698
1699         exp = class_export_get (req->rq_export);
1700         obd = exp->exp_obd;
1701
1702         /* disable reply scheduling onto srv_reply_queue while I'm setting up */
1703         rs->rs_scheduled = 1;
1704         rs->rs_on_net    = 1;
1705         rs->rs_xid       = req->rq_xid;
1706         rs->rs_transno   = req->rq_transno;
1707         rs->rs_export    = exp;
1708         
1709         spin_lock_irqsave (&obd->obd_uncommitted_replies_lock, flags);
1710
1711         if (rs->rs_transno > obd->obd_last_committed) {
1712                 /* not committed already */ 
1713                 list_add_tail (&rs->rs_obd_list, 
1714                                &obd->obd_uncommitted_replies);
1715         }
1716
1717         spin_unlock (&obd->obd_uncommitted_replies_lock);
1718         spin_lock (&exp->exp_lock);
1719
1720         list_add_tail (&rs->rs_exp_list, &exp->exp_outstanding_replies);
1721
1722         spin_unlock_irqrestore (&exp->exp_lock, flags);
1723
1724         netrc = target_send_reply_msg (req, rc, fail_id);
1725
1726         spin_lock_irqsave (&svc->srv_lock, flags);
1727
1728         svc->srv_n_difficult_replies++;
1729
1730         if (netrc != 0) /* error sending: reply is off the net */
1731                 rs->rs_on_net = 0;
1732
1733         if (!rs->rs_on_net ||                   /* some notifier */
1734             list_empty(&rs->rs_exp_list) ||     /* completed already */
1735             list_empty(&rs->rs_obd_list)) {
1736                 list_add_tail (&rs->rs_list, &svc->srv_reply_queue);
1737                 wake_up (&svc->srv_waitq);
1738         } else {
1739                 list_add (&rs->rs_list, &sni->sni_active_replies);
1740                 rs->rs_scheduled = 0;           /* allow notifier to schedule */
1741         }
1742
1743         spin_unlock_irqrestore (&svc->srv_lock, flags);
1744 }
1745
1746 int target_handle_ping(struct ptlrpc_request *req)
1747 {
1748         return lustre_pack_reply(req, 0, NULL, NULL);
1749 }