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         struct obd_device *mgmt_obd;
 185         mgmtcli_register_for_events_t register_f;
 186         ENTRY;
 187
 188         /* In a more perfect world, we would hang a ptlrpc_client off of
 189          * obd_type and just use the values from there. */
 190         if (!strcmp(name, LUSTRE_OSC_NAME)) {
 191                 rq_portal = OST_REQUEST_PORTAL;
 192                 rp_portal = OSC_REPLY_PORTAL;
 193                 connect_op = OST_CONNECT;
 194         } else if (!strcmp(name, LUSTRE_MDC_NAME)) {
 195                 rq_portal = MDS_REQUEST_PORTAL;
 196                 rp_portal = MDC_REPLY_PORTAL;
 197                 connect_op = MDS_CONNECT;
 198         } else if (!strcmp(name, LUSTRE_MGMTCLI_NAME)) {
 199                 rq_portal = MGMT_REQUEST_PORTAL;
 200                 rp_portal = MGMT_REPLY_PORTAL;
 201                 connect_op = MGMT_CONNECT;
 202         } else {
 203                 CERROR("unknown client OBD type \"%s\", can't setup\n",
 204                        name);
 205                 RETURN(-EINVAL);
 206         }
 207
 208
 209         if (LUSTRE_CFG_BUFLEN(lcfg, 1) < 1) {
 210                 CERROR("requires a TARGET UUID\n");
 211                 RETURN(-EINVAL);
 212         }
 213
 214         if (LUSTRE_CFG_BUFLEN(lcfg, 1) > 37) {
 215                 CERROR("client UUID must be less than 38 characters\n");
 216                 RETURN(-EINVAL);
 217         }
 218
 219         if (LUSTRE_CFG_BUFLEN(lcfg, 2) < 1) {
 220                 CERROR("setup requires a SERVER UUID\n");
 221                 RETURN(-EINVAL);
 222         }
 223
 224         if (LUSTRE_CFG_BUFLEN(lcfg, 2) > 37) {
 225                 CERROR("target UUID must be less than 38 characters\n");
 226                 RETURN(-EINVAL);
 227         }
 228
 229         sema_init(&cli->cl_sem, 1);
 230         cli->cl_conn_count = 0;
 231         memcpy(server_uuid.uuid,  lustre_cfg_buf(lcfg, 2),
 232                min_t(unsigned int, LUSTRE_CFG_BUFLEN(lcfg, 2),
 233                sizeof(server_uuid)));
 234
 235         cli->cl_dirty = 0;
 236         cli->cl_avail_grant = 0;
 237         /* FIXME: should limit this for the sum of all cl_dirty_max */
 238         cli->cl_dirty_max = OSC_MAX_DIRTY_DEFAULT * 1024 * 1024;
 239         if (cli->cl_dirty_max >> PAGE_SHIFT > num_physpages / 8)
 240                 cli->cl_dirty_max = num_physpages << (PAGE_SHIFT - 3);
 241         INIT_LIST_HEAD(&cli->cl_cache_waiters);
 242         INIT_LIST_HEAD(&cli->cl_loi_ready_list);
 243         INIT_LIST_HEAD(&cli->cl_loi_write_list);
 244         INIT_LIST_HEAD(&cli->cl_loi_read_list);
 245         spin_lock_init(&cli->cl_loi_list_lock);
 246         cli->cl_r_in_flight = 0;
 247         cli->cl_w_in_flight = 0;
 248         spin_lock_init(&cli->cl_read_rpc_hist.oh_lock);
 249         spin_lock_init(&cli->cl_write_rpc_hist.oh_lock);
 250         spin_lock_init(&cli->cl_read_page_hist.oh_lock);
 251         spin_lock_init(&cli->cl_write_page_hist.oh_lock);
 252
 253         if (num_physpages >> (20 - PAGE_SHIFT) <= 128) { /* <= 128 MB */
 254                 cli->cl_max_pages_per_rpc = PTLRPC_MAX_BRW_PAGES / 4;
 255                 cli->cl_max_rpcs_in_flight = OSC_MAX_RIF_DEFAULT / 4;
 256         } else if (num_physpages >> (20 - PAGE_SHIFT) <= 512) { /* <= 512 MB */
 257                 cli->cl_max_pages_per_rpc = PTLRPC_MAX_BRW_PAGES / 2;
 258                 cli->cl_max_rpcs_in_flight = OSC_MAX_RIF_DEFAULT / 2;
 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
 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
 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 int target_handle_connect(struct ptlrpc_request *req)
 540 {
 541         unsigned long connect_flags = 0, *cfp;
 542         struct obd_device *target;
 543         struct obd_export *export = NULL;
 544         struct obd_import *revimp;
 545         struct lustre_handle conn;
 546         struct obd_uuid tgtuuid;
 547         struct obd_uuid cluuid;
 548         struct obd_uuid remote_uuid;
 549         struct list_head *p;
 550         struct obd_connect_data *conn_data;
 551         int conn_data_size = sizeof(*conn_data);
 552         char *str, *tmp;
 553         int rc = 0;
 554         unsigned long flags;
 555         int initial_conn = 0;
 556         char peer_str[PTL_NALFMT_SIZE];
 557         const int offset = 1;
 558         ENTRY;
 559
 560         OBD_RACE(OBD_FAIL_TGT_CONN_RACE);
 561
 562         LASSERT_REQSWAB (req, offset + 0);
 563         str = lustre_msg_string(req->rq_reqmsg, offset + 0,
 564                                 sizeof(tgtuuid) - 1);
 565         if (str == NULL) {
 566                 CERROR("bad target UUID for connect\n");
 567                 GOTO(out, rc = -EINVAL);
 568         }
 569
 570         obd_str2uuid (&tgtuuid, str);
 571         target = class_uuid2obd(&tgtuuid);
 572         if (!target)
 573                 target = class_name2obd(str);
 574
 575         if (!target || target->obd_stopping || !target->obd_set_up) {
 576                 CERROR("UUID '%s' is not available for connect from %s\n",
 577                        str, req->rq_peerstr);
 578                 GOTO(out, rc = -ENODEV);
 579         }
 580
 581         LASSERT_REQSWAB (req, offset + 1);
 582         str = lustre_msg_string(req->rq_reqmsg, offset + 1, sizeof(cluuid) - 1);
 583         if (str == NULL) {
 584                 CERROR("bad client UUID for connect\n");
 585                 GOTO(out, rc = -EINVAL);
 586         }
 587
 588         obd_str2uuid (&cluuid, str);
 589
 590         /* XXX extract a nettype and format accordingly */
 591         switch (sizeof(ptl_nid_t)) {
 592                 /* NB the casts only avoid compiler warnings */
 593         case 8:
 594                 snprintf((char *)remote_uuid.uuid, sizeof(remote_uuid),
 595                          "NET_"LPX64"_UUID", (__u64)req->rq_peer.peer_id.nid);
 596                 break;
 597         case 4:
 598                 snprintf((char *)remote_uuid.uuid, sizeof(remote_uuid),
 599                          "NET_%x_UUID", (__u32)req->rq_peer.peer_id.nid);
 600                 break;
 601         default:
 602                 LBUG();
 603         }
 604
 605         tmp = lustre_msg_buf(req->rq_reqmsg, offset + 2, sizeof(conn));
 606         if (tmp == NULL)
 607                 GOTO(out, rc = -EPROTO);
 608
 609         memcpy(&conn, tmp, sizeof conn);
 610
 611         cfp = lustre_msg_buf(req->rq_reqmsg, offset + 3, sizeof(unsigned long));
 612         LASSERT(cfp != NULL);
 613         connect_flags = *cfp;
 614
 615         conn_data = lustre_swab_reqbuf(req, offset + 4, sizeof(*conn_data),
 616                                        lustre_swab_connect);
 617         if (!conn_data)
 618                 GOTO(out, rc = -EPROTO);
 619
 620         rc = lustre_pack_reply(req, 1, &conn_data_size, NULL);
 621         if (rc)
 622                 GOTO(out, rc);
 623
 624         if (lustre_msg_get_op_flags(req->rq_reqmsg) & MSG_CONNECT_INITIAL)
 625                 initial_conn = 1;
 626
 627         /* lctl gets a backstage, all-access pass. */
 628         if (obd_uuid_equals(&cluuid, &target->obd_uuid))
 629                 goto dont_check_exports;
 630
 631         spin_lock(&target->obd_dev_lock);
 632         list_for_each(p, &target->obd_exports) {
 633                 export = list_entry(p, struct obd_export, exp_obd_chain);
 634                 if (obd_uuid_equals(&cluuid, &export->exp_client_uuid)) {
 635                         spin_unlock(&target->obd_dev_lock);
 636                         LASSERT(export->exp_obd == target);
 637
 638                         rc = target_handle_reconnect(&conn, export, &cluuid,
 639                                                      initial_conn);
 640                         break;
 641                 }
 642                 export = NULL;
 643         }
 644         /* If we found an export, we already unlocked. */
 645         if (!export) {
 646                 spin_unlock(&target->obd_dev_lock);
 647         } else if (req->rq_export == NULL &&
 648                    atomic_read(&export->exp_rpc_count) > 0) {
 649                 CWARN("%s: refuse connection from %s/%s to 0x%p/%d\n",
 650                       target->obd_name, cluuid.uuid,
 651                       ptlrpc_peernid2str(&req->rq_peer, peer_str),
 652                       export, atomic_read(&export->exp_refcount));
 653                 GOTO(out, rc = -EBUSY);
 654         } else if (req->rq_export != NULL &&
 655                    atomic_read(&export->exp_rpc_count) > 1) {
 656                 CWARN("%s: refuse reconnection from %s@%s to 0x%p/%d\n",
 657                       target->obd_name, cluuid.uuid,
 658                       ptlrpc_peernid2str(&req->rq_peer, peer_str),
 659                       export, atomic_read(&export->exp_rpc_count));
 660                 GOTO(out, rc = -EBUSY);
 661         } else if (req->rq_reqmsg->conn_cnt == 1 && !initial_conn) {
 662                 CERROR("%s reconnected with 1 conn_cnt; cookies not random?\n",
 663                        cluuid.uuid);
 664                 GOTO(out, rc = -EALREADY);
 665         }
 666
 667         /* Tell the client if we're in recovery. */
 668         /* If this is the first client, start the recovery timer */
 669         CWARN("%s: connection from %s@%s/%lu %st"LPU64"\n", target->obd_name, cluuid.uuid,
 670               ptlrpc_peernid2str(&req->rq_peer, peer_str), *cfp,
 671               target->obd_recovering ? "recovering/" : "",
 672               conn_data->transno);
 673
 674         if (target->obd_recovering) {
 675                 lustre_msg_add_op_flags(req->rq_repmsg, MSG_CONNECT_RECOVERING);
 676                 target_start_recovery_timer(target);
 677         }
 678
 679 #if 0
 680         /* Tell the client if we support replayable requests */
 681         if (target->obd_replayable)
 682                 lustre_msg_add_op_flags(req->rq_repmsg, MSG_CONNECT_REPLAYABLE);
 683 #endif
 684
 685         if (export == NULL) {
 686                 if (target->obd_recovering) {
 687                         CERROR("%s denying connection for new client %s@%s: "
 688                                "%d clients in recovery for %lds\n", target->obd_name,
 689                                cluuid.uuid,
 690                                ptlrpc_peernid2str(&req->rq_peer, peer_str),
 691                                target->obd_recoverable_clients,
 692                                (target->obd_recovery_timer.expires-jiffies)/HZ);
 693                         rc = -EBUSY;
 694                 } else {
 695  dont_check_exports:
 696                         rc = obd_connect(&conn, target, &cluuid, conn_data,
 697                                          connect_flags);
 698                 }
 699         }
 700
 701         /* Return only the parts of obd_connect_data that we understand, so the
 702          * client knows that we don't understand the rest. */
 703         conn_data->ocd_connect_flags &= OBD_CONNECT_SUPPORTED;
 704         memcpy(lustre_msg_buf(req->rq_repmsg, 0, sizeof(*conn_data)), conn_data,
 705                sizeof(*conn_data));
 706
 707         /* Tell the client if we support replayable requests */
 708         if (target->obd_replayable)
 709                 lustre_msg_add_op_flags(req->rq_repmsg, MSG_CONNECT_REPLAYABLE);
 710
 711         /* If all else goes well, this is our RPC return code. */
 712         req->rq_status = 0;
 713
 714         if (rc && rc != EALREADY)
 715                 GOTO(out, rc);
 716
 717         req->rq_repmsg->handle = conn;
 718
 719         /* If the client and the server are the same node, we will already
 720          * have an export that really points to the client's DLM export,
 721          * because we have a shared handles table.
 722          *
 723          * XXX this will go away when shaver stops sending the "connect" handle
 724          * in the real "remote handle" field of the request --phik 24 Apr 2003
 725          */
 726         if (req->rq_export != NULL)
 727                 class_export_put(req->rq_export);
 728
 729         /* ownership of this export ref transfers to the request */
 730         export = req->rq_export = class_conn2export(&conn);
 731         LASSERT(export != NULL);
 732
 733         spin_lock_irqsave(&export->exp_lock, flags);
 734         if (initial_conn) {
 735                 req->rq_repmsg->conn_cnt = export->exp_conn_cnt + 1;
 736         } else if (export->exp_conn_cnt >= req->rq_reqmsg->conn_cnt) {
 737                 CERROR("%s@%s: already connected at a higher conn_cnt: %d > %d\n",
 738                        cluuid.uuid, ptlrpc_peernid2str(&req->rq_peer, peer_str),
 739                        export->exp_conn_cnt,
 740                        req->rq_reqmsg->conn_cnt);
 741                 spin_unlock_irqrestore(&export->exp_lock, flags);
 742                 GOTO(out, rc = -EALREADY);
 743         }
 744         export->exp_conn_cnt = req->rq_reqmsg->conn_cnt;
 745         spin_unlock_irqrestore(&export->exp_lock, flags);
 746
 747         /* request from liblustre? */
 748         if (lustre_msg_get_op_flags(req->rq_reqmsg) & MSG_CONNECT_LIBCLIENT)
 749                 export->exp_libclient = 1;
 750
 751         if (!(lustre_msg_get_op_flags(req->rq_reqmsg) & MSG_CONNECT_ASYNC) &&
 752             ptlrpc_peer_is_local(&req->rq_peer)) {
 753                 CWARN("%s: exp %p set sync\n", target->obd_name, export);
 754                 export->exp_sync = 1;
 755         } else {
 756                 CDEBUG(D_HA, "%s: exp %p set async\n",target->obd_name,export);
 757                 export->exp_sync = 0;
 758         }
 759
 760         if (export->exp_connection != NULL)
 761                 ptlrpc_put_connection(export->exp_connection);
 762         export->exp_connection = ptlrpc_get_connection(&req->rq_peer,
 763                                                        &remote_uuid);
 764
 765         if (rc == EALREADY) {
 766                 /* We indicate the reconnection in a flag, not an error code. */
 767                 lustre_msg_add_op_flags(req->rq_repmsg, MSG_CONNECT_RECONNECT);
 768                 GOTO(out, rc = 0);
 769         }
 770
 771         spin_lock_bh(&target->obd_processing_task_lock);
 772         if (target->obd_recovering && export->exp_connected == 0) {
 773                 __u64 t = conn_data->transno;
 774                 export->exp_connected = 1;
 775                 if ((lustre_msg_get_op_flags(req->rq_reqmsg) & MSG_CONNECT_TRANSNO)
 776                                 && t < target->obd_next_recovery_transno)
 777                         target->obd_next_recovery_transno = t;
 778                 target->obd_connected_clients++;
 779                 if (target->obd_connected_clients == target->obd_max_recoverable_clients)
 780                         wake_up(&target->obd_next_transno_waitq);
 781         }
 782         spin_unlock_bh(&target->obd_processing_task_lock);
 783
 784         memcpy(&conn, lustre_msg_buf(req->rq_reqmsg, offset + 2, sizeof(conn)),
 785                sizeof(conn));
 786
 787         if (export->exp_imp_reverse != NULL) {
 788                 /* same logic as client_obd_cleanup */
 789                 class_import_get(export->exp_imp_reverse);
 790                 class_destroy_import(export->exp_imp_reverse);
 791                 ptlrpcs_import_drop_sec(export->exp_imp_reverse);
 792                 class_import_put(export->exp_imp_reverse);
 793         }
 794
 795         /* for the rest part, we return -ENOTCONN in case of errors
 796          * in order to let client initialize connection again.
 797          */
 798         revimp = export->exp_imp_reverse = class_new_import();
 799         if (!revimp) {
 800                 CERROR("fail to alloc new reverse import.\n");
 801                 GOTO(out, rc = -ENOTCONN);
 802         }
 803
 804         revimp->imp_connection = ptlrpc_connection_addref(export->exp_connection);
 805         revimp->imp_client = &export->exp_obd->obd_ldlm_client;
 806         revimp->imp_remote_handle = conn;
 807         revimp->imp_obd = target;
 808         revimp->imp_dlm_fake = 1;
 809         revimp->imp_state = LUSTRE_IMP_FULL;
 810
 811         rc = ptlrpcs_import_get_sec(revimp);
 812         if (rc) {
 813                 CERROR("reverse import can not get sec: %d\n", rc);
 814                 class_destroy_import(revimp);
 815                 export->exp_imp_reverse = NULL;
 816                 GOTO(out, rc = -ENOTCONN);
 817         }
 818
 819         class_import_put(revimp);
 820
 821         rc = obd_connect_post(export, initial_conn, connect_flags);
 822 out:
 823         if (rc)
 824                 req->rq_status = rc;
 825         RETURN(rc);
 826 }
 827
 828 int target_handle_disconnect(struct ptlrpc_request *req)
 829 {
 830         struct obd_export *exp;
 831         int rc;
 832         ENTRY;
 833
 834         rc = lustre_pack_reply(req, 0, NULL, NULL);
 835         if (rc)
 836                 RETURN(rc);
 837
 838         /* keep the rq_export around so we can send the reply */
 839         exp = class_export_get(req->rq_export);
 840         req->rq_status = obd_disconnect(exp, 0);
 841         RETURN(0);
 842 }
 843
 844 void target_destroy_export(struct obd_export *exp)
 845 {
 846         /* exports created from last_rcvd data, and "fake"
 847            exports created by lctl don't have an import */
 848         if (exp->exp_imp_reverse != NULL) {
 849                 ptlrpcs_import_drop_sec(exp->exp_imp_reverse);
 850                 class_destroy_import(exp->exp_imp_reverse);
 851         }
 852
 853         /* We cancel locks at disconnect time, but this will catch any locks
 854          * granted in a race with recovery-induced disconnect. */
 855         if (exp->exp_obd->obd_namespace != NULL)
 856                 ldlm_cancel_locks_for_export(exp);
 857 }
 858
 859 /*
 860  * Recovery functions
 861  */
 862
 863 struct ptlrpc_request *
 864 ptlrpc_clone_req( struct ptlrpc_request *orig_req)
 865 {
 866         struct ptlrpc_request *copy_req;
 867         struct lustre_msg *copy_reqmsg;
 868
 869         OBD_ALLOC(copy_req, sizeof *copy_req);
 870         if (!copy_req)
 871                 return NULL;
 872         OBD_ALLOC(copy_reqmsg, orig_req->rq_reqlen);
 873         if (!copy_reqmsg){
 874                 OBD_FREE(copy_req, sizeof *copy_req);
 875                 return NULL;
 876         }
 877
 878         memcpy(copy_req, orig_req, sizeof *copy_req);
 879         memcpy(copy_reqmsg, orig_req->rq_reqmsg, orig_req->rq_reqlen);
 880         /* the copied req takes over the reply state and security data */
 881         orig_req->rq_reply_state = NULL;
 882         orig_req->rq_sec_svcdata = NULL;
 883
 884         copy_req->rq_reqmsg = copy_reqmsg;
 885         class_export_get(copy_req->rq_export);
 886         INIT_LIST_HEAD(&copy_req->rq_list);
 887
 888         return copy_req;
 889 }
 890
 891 void ptlrpc_free_clone( struct ptlrpc_request *req)
 892 {
 893         if (req->rq_svcsec)
 894                 svcsec_cleanup_req(req);
 895
 896         class_export_put(req->rq_export);
 897         list_del(&req->rq_list);
 898         OBD_FREE(req->rq_reqmsg, req->rq_reqlen);
 899         OBD_FREE(req, sizeof *req);
 900 }
 901
 902 static void target_release_saved_req(struct ptlrpc_request *req)
 903 {
 904         if (req->rq_svcsec)
 905                 svcsec_cleanup_req(req);
 906
 907         class_export_put(req->rq_export);
 908         OBD_FREE(req->rq_reqmsg, req->rq_reqlen);
 909         OBD_FREE(req, sizeof *req);
 910 }
 911
 912 static void target_finish_recovery(struct obd_device *obd)
 913 {
 914         int rc;
 915
 916         ldlm_reprocess_all_ns(obd->obd_namespace);
 917
 918         /* when recovery finished, cleanup orphans on mds and ost */
 919         if (OBT(obd) && OBP(obd, postrecov)) {
 920                 rc = OBP(obd, postrecov)(obd);
 921                 if (rc >= 0)
 922                         CWARN("%s: all clients recovered, %d MDS "
 923                               "orphans deleted\n", obd->obd_name, rc);
 924                 else
 925                         CERROR("postrecov failed %d\n", rc);
 926         }
 927
 928         obd->obd_recovery_end = LTIME_S(CURRENT_TIME);
 929         return;
 930 }
 931
 932 static void abort_req_replay_queue(struct obd_device *obd)
 933 {
 934         struct ptlrpc_request *req;
 935         struct list_head *tmp, *n;
 936         int rc;
 937
 938         list_for_each_safe(tmp, n, &obd->obd_req_replay_queue) {
 939                 req = list_entry(tmp, struct ptlrpc_request, rq_list);
 940                 list_del(&req->rq_list);
 941                 DEBUG_REQ(D_ERROR, req, "aborted:");
 942                 req->rq_status = -ENOTCONN;
 943                 req->rq_type = PTL_RPC_MSG_ERR;
 944                 rc = lustre_pack_reply(req, 0, NULL, NULL);
 945                 if (rc == 0) {
 946                         ptlrpc_reply(req);
 947                 } else {
 948                         DEBUG_REQ(D_ERROR, req,
 949                                   "packing failed for abort-reply; skipping");
 950                 }
 951                 target_release_saved_req(req);
 952         }
 953 }
 954
 955 static void abort_lock_replay_queue(struct obd_device *obd)
 956 {
 957         struct ptlrpc_request *req;
 958         struct list_head *tmp, *n;
 959         int rc;
 960
 961         list_for_each_safe(tmp, n, &obd->obd_lock_replay_queue) {
 962                 req = list_entry(tmp, struct ptlrpc_request, rq_list);
 963                 list_del(&req->rq_list);
 964                 DEBUG_REQ(D_ERROR, req, "aborted:");
 965                 req->rq_status = -ENOTCONN;
 966                 req->rq_type = PTL_RPC_MSG_ERR;
 967                 rc = lustre_pack_reply(req, 0, NULL, NULL);
 968                 if (rc == 0) {
 969                         ptlrpc_reply(req);
 970                 } else {
 971                         DEBUG_REQ(D_ERROR, req,
 972                                   "packing failed for abort-reply; skipping");
 973                 }
 974                 target_release_saved_req(req);
 975         }
 976 }
 977
 978 /* Called from a cleanup function if the device is being cleaned up
 979    forcefully.  The exports should all have been disconnected already,
 980    the only thing left to do is
 981      - clear the recovery flags
 982      - cancel the timer
 983      - free queued requests and replies, but don't send replies
 984    Because the obd_stopping flag is set, no new requests should be received.
 985
 986 */
 987 void target_cleanup_recovery(struct obd_device *obd)
 988 {
 989         struct list_head *tmp, *n;
 990         struct ptlrpc_request *req;
 991
 992         spin_lock_bh(&obd->obd_processing_task_lock);
 993         if (!obd->obd_recovering) {
 994                 spin_unlock_bh(&obd->obd_processing_task_lock);
 995                 EXIT;
 996                 return;
 997         }
 998         obd->obd_recovering = obd->obd_abort_recovery = 0;
 999         target_cancel_recovery_timer(obd);
1000         spin_unlock_bh(&obd->obd_processing_task_lock);
1001
1002         list_for_each_safe(tmp, n, &obd->obd_req_replay_queue) {
1003                 req = list_entry(tmp, struct ptlrpc_request, rq_list);
1004                 list_del(&req->rq_list);
1005                 LASSERT (req->rq_reply_state == 0);
1006                 target_release_saved_req(req);
1007         }
1008         list_for_each_safe(tmp, n, &obd->obd_lock_replay_queue) {
1009                 req = list_entry(tmp, struct ptlrpc_request, rq_list);
1010                 list_del(&req->rq_list);
1011                 LASSERT (req->rq_reply_state == 0);
1012                 target_release_saved_req(req);
1013         }
1014         list_for_each_safe(tmp, n, &obd->obd_final_req_queue) {
1015                 req = list_entry(tmp, struct ptlrpc_request, rq_list);
1016                 list_del(&req->rq_list);
1017                 LASSERT (req->rq_reply_state == 0);
1018                 target_release_saved_req(req);
1019         }
1020 }
1021
1022 #if 0
1023 static void target_abort_recovery(void *data)
1024 {
1025         struct obd_device *obd = data;
1026
1027         LASSERT(!obd->obd_recovering);
1028
1029         class_disconnect_stale_exports(obd, 0);
1030
1031         CERROR("%s: recovery period over; disconnecting unfinished clients.\n",
1032                obd->obd_name);
1033
1034         abort_recovery_queue(obd);
1035         target_finish_recovery(obd);
1036         ptlrpc_run_recovery_over_upcall(obd);
1037 }
1038 #endif
1039
1040 static void target_recovery_expired(unsigned long castmeharder)
1041 {
1042         struct obd_device *obd = (struct obd_device *)castmeharder;
1043         spin_lock_bh(&obd->obd_processing_task_lock);
1044         if (obd->obd_recovering)
1045                 obd->obd_abort_recovery = 1;
1046
1047         wake_up(&obd->obd_next_transno_waitq);
1048         spin_unlock_bh(&obd->obd_processing_task_lock);
1049 }
1050
1051
1052 /* obd_processing_task_lock should be held */
1053 void target_cancel_recovery_timer(struct obd_device *obd)
1054 {
1055         CDEBUG(D_HA, "%s: cancel recovery timer\n", obd->obd_name);
1056         del_timer(&obd->obd_recovery_timer);
1057 }
1058
1059 #ifdef __KERNEL__
1060 static void reset_recovery_timer(struct obd_device *obd)
1061 {
1062         spin_lock_bh(&obd->obd_processing_task_lock);
1063         if (!obd->obd_recovering) {
1064                 spin_unlock_bh(&obd->obd_processing_task_lock);
1065                 return;
1066         }
1067         CDEBUG(D_HA, "timer will expire in %u seconds\n",
1068                OBD_RECOVERY_TIMEOUT / HZ);
1069         mod_timer(&obd->obd_recovery_timer, jiffies + OBD_RECOVERY_TIMEOUT);
1070         spin_unlock_bh(&obd->obd_processing_task_lock);
1071 }
1072 #endif
1073
1074 /* Only start it the first time called */
1075 void target_start_recovery_timer(struct obd_device *obd)
1076 {
1077         spin_lock_bh(&obd->obd_processing_task_lock);
1078         if (!obd->obd_recovering || timer_pending(&obd->obd_recovery_timer)) {
1079                 spin_unlock_bh(&obd->obd_processing_task_lock);
1080                 return;
1081         }
1082         CWARN("%s: starting recovery timer (%us)\n", obd->obd_name,
1083                OBD_RECOVERY_TIMEOUT / HZ);
1084         obd->obd_recovery_timer.function = target_recovery_expired;
1085         obd->obd_recovery_timer.data = (unsigned long)obd;
1086         mod_timer(&obd->obd_recovery_timer, jiffies + OBD_RECOVERY_TIMEOUT);
1087         spin_unlock_bh(&obd->obd_processing_task_lock);
1088 }
1089
1090 #ifdef __KERNEL__
1091 static int check_for_next_transno(struct obd_device *obd)
1092 {
1093         struct ptlrpc_request *req = NULL;
1094         int wake_up = 0, connected, completed, queue_len, max;
1095         __u64 next_transno, req_transno;
1096
1097         spin_lock_bh(&obd->obd_processing_task_lock);
1098         if (!list_empty(&obd->obd_req_replay_queue)) {
1099                 req = list_entry(obd->obd_req_replay_queue.next,
1100                                  struct ptlrpc_request, rq_list);
1101                 req_transno = req->rq_reqmsg->transno;
1102         } else {
1103                 req_transno = 0;
1104         }
1105
1106         max = obd->obd_max_recoverable_clients;
1107         connected = obd->obd_connected_clients;
1108         completed = max - obd->obd_recoverable_clients;
1109         queue_len = obd->obd_requests_queued_for_recovery;
1110         next_transno = obd->obd_next_recovery_transno;
1111
1112         CDEBUG(D_HA,"max: %d, connected: %d, completed: %d, queue_len: %d, "
1113                "req_transno: "LPU64", next_transno: "LPU64"\n",
1114                max, connected, completed, queue_len, req_transno, next_transno);
1115         if (obd->obd_abort_recovery) {
1116                 CDEBUG(D_HA, "waking for aborted recovery\n");
1117                 wake_up = 1;
1118         } else if (atomic_read(&obd->obd_req_replay_clients) == 0) {
1119                 CDEBUG(D_HA, "waking for completed recovery\n");
1120                 wake_up = 1;
1121         } else if (req_transno == next_transno) {
1122                 CDEBUG(D_HA, "waking for next ("LPD64")\n", next_transno);
1123                 wake_up = 1;
1124         } else if (queue_len + completed == max) {
1125                 LASSERT(req->rq_reqmsg->transno >= next_transno);
1126                 CDEBUG(req_transno > obd->obd_last_committed ? D_ERROR : D_HA,
1127                        "waking for skipped transno (skip: "LPD64
1128                        ", ql: %d, comp: %d, conn: %d, next: "LPD64")\n",
1129                        next_transno, queue_len, completed, max, req_transno);
1130                 obd->obd_next_recovery_transno = req_transno;
1131                 wake_up = 1;
1132         } else if (queue_len == atomic_read(&obd->obd_req_replay_clients)) {
1133                 /* some clients haven't connected in time, but we need
1134                  * their requests to continue recovery. so, we abort ... */
1135                 CDEBUG(D_ERROR, "abort due to missed clients: queue: %d max: %d\n",
1136                        queue_len, max);
1137                 obd->obd_abort_recovery = 1;
1138                 wake_up = 1;
1139         }
1140         spin_unlock_bh(&obd->obd_processing_task_lock);
1141
1142         return wake_up;
1143 }
1144
1145 static struct ptlrpc_request *
1146 target_next_replay_req(struct obd_device *obd)
1147 {
1148         struct l_wait_info lwi = { 0 };
1149         struct ptlrpc_request *req;
1150
1151         CDEBUG(D_HA, "Waiting for transno "LPD64"\n",
1152                obd->obd_next_recovery_transno);
1153         l_wait_event(obd->obd_next_transno_waitq,
1154                      check_for_next_transno(obd), &lwi);
1155
1156         spin_lock_bh(&obd->obd_processing_task_lock);
1157         if (obd->obd_abort_recovery) {
1158                 req = NULL;
1159         } else if (!list_empty(&obd->obd_req_replay_queue)) {
1160                 req = list_entry(obd->obd_req_replay_queue.next,
1161                                  struct ptlrpc_request, rq_list);
1162                 list_del_init(&req->rq_list);
1163                 obd->obd_requests_queued_for_recovery--;
1164         } else {
1165                 req = NULL;
1166         }
1167         spin_unlock_bh(&obd->obd_processing_task_lock);
1168         return req;
1169 }
1170
1171 static int check_for_next_lock(struct obd_device *obd)
1172 {
1173         struct ptlrpc_request *req = NULL;
1174         int wake_up = 0;
1175
1176         spin_lock_bh(&obd->obd_processing_task_lock);
1177         if (!list_empty(&obd->obd_lock_replay_queue)) {
1178                 req = list_entry(obd->obd_lock_replay_queue.next,
1179                                  struct ptlrpc_request, rq_list);
1180                 CDEBUG(D_HA, "waking for next lock\n");
1181                 wake_up = 1;
1182         } else if (atomic_read(&obd->obd_lock_replay_clients) == 0) {
1183                 CDEBUG(D_HA, "waking for completed lock replay\n");
1184                 wake_up = 1;
1185         } else if (obd->obd_abort_recovery) {
1186                 CDEBUG(D_HA, "waking for aborted recovery\n");
1187                 wake_up = 1;
1188         }
1189         spin_unlock_bh(&obd->obd_processing_task_lock);
1190
1191         return wake_up;
1192 }
1193
1194 static struct ptlrpc_request *
1195 target_next_replay_lock(struct obd_device *obd)
1196 {
1197         struct l_wait_info lwi = { 0 };
1198         struct ptlrpc_request *req;
1199
1200         CDEBUG(D_HA, "Waiting for lock\n");
1201         l_wait_event(obd->obd_next_transno_waitq,
1202                      check_for_next_lock(obd), &lwi);
1203
1204         spin_lock_bh(&obd->obd_processing_task_lock);
1205         if (obd->obd_abort_recovery) {
1206                 req = NULL;
1207         } else if (!list_empty(&obd->obd_lock_replay_queue)) {
1208                 req = list_entry(obd->obd_lock_replay_queue.next,
1209                                  struct ptlrpc_request, rq_list);
1210                 list_del_init(&req->rq_list);
1211         } else {
1212                 req = NULL;
1213         }
1214         spin_unlock_bh(&obd->obd_processing_task_lock);
1215         return req;
1216 }
1217
1218 static struct ptlrpc_request *
1219 target_next_final_ping(struct obd_device *obd)
1220 {
1221         struct ptlrpc_request *req;
1222
1223         spin_lock_bh(&obd->obd_processing_task_lock);
1224         if (!list_empty(&obd->obd_final_req_queue)) {
1225                 req = list_entry(obd->obd_final_req_queue.next,
1226                                  struct ptlrpc_request, rq_list);
1227                 list_del_init(&req->rq_list);
1228         } else {
1229                 req = NULL;
1230         }
1231         spin_unlock_bh(&obd->obd_processing_task_lock);
1232         return req;
1233 }
1234
1235 static int req_replay_done(struct obd_export *exp)
1236 {
1237         if (exp->exp_req_replay_needed)
1238                 return 0;
1239         return 1;
1240 }
1241
1242 static int lock_replay_done(struct obd_export *exp)
1243 {
1244         if (exp->exp_lock_replay_needed)
1245                 return 0;
1246         return 1;
1247 }
1248
1249 static int connect_done(struct obd_export *exp)
1250 {
1251         if (exp->exp_connected)
1252                 return 1;
1253         return 0;
1254 }
1255
1256 static int check_for_clients(struct obd_device *obd)
1257 {
1258         if (obd->obd_abort_recovery)
1259                 return 1;
1260         LASSERT(obd->obd_connected_clients <= obd->obd_max_recoverable_clients);
1261         if (obd->obd_connected_clients == obd->obd_max_recoverable_clients)
1262                 return 1;
1263         return 0;
1264 }
1265
1266 static int target_recovery_thread(void *arg)
1267 {
1268         struct obd_device *obd = arg;
1269         struct ptlrpc_request *req;
1270         struct target_recovery_data *trd = &obd->obd_recovery_data;
1271         char peer_str[PTL_NALFMT_SIZE];
1272         struct l_wait_info lwi = { 0 };
1273         unsigned long flags;
1274         ENTRY;
1275
1276         kportal_daemonize("tgt-recov");
1277
1278         SIGNAL_MASK_LOCK(current, flags);
1279         sigfillset(&current->blocked);
1280         RECALC_SIGPENDING;
1281         SIGNAL_MASK_UNLOCK(current, flags);
1282
1283         CERROR("%s: started recovery thread pid %d\n", obd->obd_name,
1284                current->pid);
1285         trd->trd_processing_task = current->pid;
1286
1287         obd->obd_recovering = 1;
1288         complete(&trd->trd_starting);
1289
1290         /* first of all, we have to know the first transno to replay */
1291         obd->obd_abort_recovery = 0;
1292         l_wait_event(obd->obd_next_transno_waitq,
1293                      check_for_clients(obd), &lwi);
1294
1295         spin_lock_bh(&obd->obd_processing_task_lock);
1296         target_cancel_recovery_timer(obd);
1297         spin_unlock_bh(&obd->obd_processing_task_lock);
1298
1299         /* If some clients haven't connected in time, evict them */
1300         if (obd->obd_abort_recovery) {
1301                 int stale;
1302                 CERROR("some clients haven't connect in time, evict them ...\n");
1303                 obd->obd_abort_recovery = 0;
1304                 stale = class_disconnect_stale_exports(obd, connect_done, 0);
1305                 atomic_sub(stale, &obd->obd_req_replay_clients);
1306                 atomic_sub(stale, &obd->obd_lock_replay_clients);
1307         }
1308
1309         /* next stage: replay requests */
1310         CWARN("1: request replay stage - %d clients from t"LPU64"\n",
1311               atomic_read(&obd->obd_req_replay_clients),
1312               obd->obd_next_recovery_transno);
1313         while ((req = target_next_replay_req(obd))) {
1314                 LASSERT(trd->trd_processing_task == current->pid);
1315                 DEBUG_REQ(D_HA, req, "processing t"LPD64" from %s: ",
1316                           req->rq_reqmsg->transno,
1317                           ptlrpc_peernid2str(&req->rq_peer, peer_str));
1318                 (void)trd->trd_recovery_handler(req);
1319                 obd->obd_replayed_requests++;
1320                 reset_recovery_timer(obd);
1321                 /* bug 1580: decide how to properly sync() in recovery*/
1322                 //mds_fsync_super(mds->mds_sb);
1323                 ptlrpc_free_clone(req);
1324                 spin_lock_bh(&obd->obd_processing_task_lock);
1325                 obd->obd_next_recovery_transno++;
1326                 spin_unlock_bh(&obd->obd_processing_task_lock);
1327         }
1328
1329         spin_lock_bh(&obd->obd_processing_task_lock);
1330         target_cancel_recovery_timer(obd);
1331         spin_unlock_bh(&obd->obd_processing_task_lock);
1332
1333         /* If some clients haven't replayed requests in time, evict them */
1334         if (obd->obd_abort_recovery) {
1335                 int stale;
1336                 CERROR("req replay timed out, aborting ...\n");
1337                 obd->obd_abort_recovery = 0;
1338                 stale = class_disconnect_stale_exports(obd, req_replay_done, 0);
1339                 atomic_sub(stale, &obd->obd_lock_replay_clients);
1340                 abort_req_replay_queue(obd);
1341         }
1342
1343         /* The second stage: replay locks */
1344         CWARN("2: lock replay stage - %d clients\n",
1345               atomic_read(&obd->obd_lock_replay_clients));
1346         while ((req = target_next_replay_lock(obd))) {
1347                 LASSERT(trd->trd_processing_task == current->pid);
1348                 DEBUG_REQ(D_HA, req, "processing lock from %s: ",
1349                           ptlrpc_peernid2str(&req->rq_peer, peer_str));
1350                 (void)trd->trd_recovery_handler(req);
1351                 reset_recovery_timer(obd);
1352                 ptlrpc_free_clone(req);
1353                 obd->obd_replayed_locks++;
1354         }
1355
1356         spin_lock_bh(&obd->obd_processing_task_lock);
1357         target_cancel_recovery_timer(obd);
1358         spin_unlock_bh(&obd->obd_processing_task_lock);
1359
1360         /* If some clients haven't replayed requests in time, evict them */
1361         if (obd->obd_abort_recovery) {
1362                 int stale;
1363                 CERROR("lock replay timed out, aborting ...\n");
1364                 obd->obd_abort_recovery = 0;
1365                 stale = class_disconnect_stale_exports(obd, lock_replay_done, 0);
1366                 abort_lock_replay_queue(obd);
1367         }
1368
1369         /* We drop recoverying flag to forward all new requests
1370          * to regular mds_handle() since now */
1371         spin_lock_bh(&obd->obd_processing_task_lock);
1372         obd->obd_recovering = 0;
1373         spin_unlock_bh(&obd->obd_processing_task_lock);
1374
1375         /* The third stage: reply on final pings */
1376         CWARN("3: final stage - process recovery completion pings\n");
1377         while ((req = target_next_final_ping(obd))) {
1378                 LASSERT(trd->trd_processing_task == current->pid);
1379                 DEBUG_REQ(D_HA, req, "processing final ping from %s: ",
1380                           ptlrpc_peernid2str(&req->rq_peer, peer_str));
1381                 (void)trd->trd_recovery_handler(req);
1382                 ptlrpc_free_clone(req);
1383         }
1384
1385         CWARN("4: recovery completed - %d/%d reqs/locks replayed\n",
1386               obd->obd_replayed_requests, obd->obd_replayed_locks);
1387         target_finish_recovery(obd);
1388
1389         trd->trd_processing_task = 0;
1390         complete(&trd->trd_finishing);
1391         return 0;
1392 }
1393
1394 int target_start_recovery_thread(struct obd_device *obd, svc_handler_t handler)
1395 {
1396         int rc = 0;
1397         struct target_recovery_data *trd = &obd->obd_recovery_data;
1398
1399         memset(trd, 0, sizeof(*trd));
1400         init_completion(&trd->trd_starting);
1401         init_completion(&trd->trd_finishing);
1402         trd->trd_recovery_handler = handler;
1403
1404         if (kernel_thread(target_recovery_thread, obd, 0) > 0) {
1405                 wait_for_completion(&trd->trd_starting);
1406                 LASSERT(obd->obd_recovering != 0);
1407         } else
1408                 rc = -ECHILD;
1409
1410         return rc;
1411 }
1412
1413 void target_stop_recovery_thread(struct obd_device *obd)
1414 {
1415         spin_lock_bh(&obd->obd_processing_task_lock);
1416         if (obd->obd_recovery_data.trd_processing_task > 0) {
1417                 struct target_recovery_data *trd = &obd->obd_recovery_data;
1418                 CERROR("%s: aborting recovery\n", obd->obd_name);
1419                 obd->obd_abort_recovery = 1;
1420                 wake_up(&obd->obd_next_transno_waitq);
1421                 spin_unlock_bh(&obd->obd_processing_task_lock);
1422                 wait_for_completion(&trd->trd_finishing);
1423         } else {
1424                 spin_unlock_bh(&obd->obd_processing_task_lock);
1425         }
1426 }
1427 #endif
1428
1429 int target_process_req_flags(struct obd_device *obd, struct ptlrpc_request *req)
1430 {
1431         struct obd_export *exp = req->rq_export;
1432         LASSERT(exp != NULL);
1433         if (lustre_msg_get_flags(req->rq_reqmsg) & MSG_REQ_REPLAY_DONE) {
1434                 /* client declares he's ready to replay locks */
1435                 spin_lock_bh(&obd->obd_processing_task_lock);
1436                 if (exp->exp_req_replay_needed) {
1437                         LASSERT(atomic_read(&obd->obd_req_replay_clients) > 0);
1438                         exp->exp_req_replay_needed = 0;
1439                         atomic_dec(&obd->obd_req_replay_clients);
1440                         obd->obd_recoverable_clients--;
1441                         if (atomic_read(&obd->obd_req_replay_clients) == 0) {
1442                                 CDEBUG(D_HA, "all clients have replayed reqs\n");
1443                                 wake_up(&obd->obd_next_transno_waitq);
1444                         }
1445                 }
1446                 spin_unlock_bh(&obd->obd_processing_task_lock);
1447         }
1448         if (lustre_msg_get_flags(req->rq_reqmsg) & MSG_LOCK_REPLAY_DONE) {
1449                 /* client declares he's ready to complete recovery
1450                  * so, we put the request on th final queue */
1451                 spin_lock_bh(&obd->obd_processing_task_lock);
1452                 if (exp->exp_lock_replay_needed) {
1453                         LASSERT(atomic_read(&obd->obd_lock_replay_clients) > 0);
1454                         exp->exp_lock_replay_needed = 0;
1455                         atomic_dec(&obd->obd_lock_replay_clients);
1456                         if (atomic_read(&obd->obd_lock_replay_clients) == 0) {
1457                                 CDEBUG(D_HA, "all clients have replayed locks\n");
1458                                 wake_up(&obd->obd_next_transno_waitq);
1459                         }
1460                 }
1461                 spin_unlock_bh(&obd->obd_processing_task_lock);
1462         }
1463
1464         return 0;
1465 }
1466
1467 int target_queue_recovery_request(struct ptlrpc_request *req,
1468                                   struct obd_device *obd)
1469 {
1470         struct list_head *tmp;
1471         int inserted = 0;
1472         __u64 transno = req->rq_reqmsg->transno;
1473
1474         if (obd->obd_recovery_data.trd_processing_task == current->pid) {
1475                 /* Processing the queue right now, don't re-add. */
1476                 return 1;
1477         }
1478
1479         target_process_req_flags(obd, req);
1480
1481         if (lustre_msg_get_flags(req->rq_reqmsg) & MSG_LOCK_REPLAY_DONE) {
1482                 /* client declares he's ready to complete recovery
1483                  * so, we put the request on th final queue */
1484                 req = ptlrpc_clone_req(req);
1485                 if (req == NULL)
1486                         return -ENOMEM;
1487                 DEBUG_REQ(D_HA, req, "queue final req");
1488                 spin_lock_bh(&obd->obd_processing_task_lock);
1489                 list_add_tail(&req->rq_list, &obd->obd_final_req_queue);
1490                 spin_unlock_bh(&obd->obd_processing_task_lock);
1491                 return 0;
1492         }
1493         if (lustre_msg_get_flags(req->rq_reqmsg) & MSG_REQ_REPLAY_DONE) {
1494                 /* client declares he's ready to replay locks */
1495                 req = ptlrpc_clone_req(req);
1496                 if (req == NULL)
1497                         return -ENOMEM;
1498                 DEBUG_REQ(D_HA, req, "queue lock replay req");
1499                 spin_lock_bh(&obd->obd_processing_task_lock);
1500                 list_add_tail(&req->rq_list, &obd->obd_lock_replay_queue);
1501                 spin_unlock_bh(&obd->obd_processing_task_lock);
1502                 wake_up(&obd->obd_next_transno_waitq);
1503                 return 0;
1504         }
1505
1506
1507         /* CAVEAT EMPTOR: The incoming request message has been swabbed
1508          * (i.e. buflens etc are in my own byte order), but type-dependent
1509          * buffers (eg mds_body, ost_body etc) have NOT been swabbed. */
1510
1511         if (!transno) {
1512                 INIT_LIST_HEAD(&req->rq_list);
1513                 DEBUG_REQ(D_HA, req, "not queueing");
1514                 return 1;
1515         }
1516
1517
1518         /* If we're processing the queue, we want don't want to queue this
1519          * message.
1520          *
1521          * Also, if this request has a transno less than the one we're waiting
1522          * for, we should process it now.  It could (and currently always will)
1523          * be an open request for a descriptor that was opened some time ago.
1524          *
1525          * Also, a resent, replayed request that has already been
1526          * handled will pass through here and be processed immediately.
1527          */
1528         spin_lock_bh(&obd->obd_processing_task_lock);
1529         if (transno < obd->obd_next_recovery_transno && check_for_clients(obd)) {
1530                 /* Processing the queue right now, don't re-add. */
1531                 LASSERT(list_empty(&req->rq_list));
1532                 spin_unlock_bh(&obd->obd_processing_task_lock);
1533                 return 1;
1534         }
1535         spin_unlock_bh(&obd->obd_processing_task_lock);
1536
1537         /* A resent, replayed request that is still on the queue; just drop it.
1538            The queued request will handle this. */
1539         if ((lustre_msg_get_flags(req->rq_reqmsg) & (MSG_RESENT | MSG_REPLAY))
1540             == (MSG_RESENT | MSG_REPLAY)) {
1541                 DEBUG_REQ(D_ERROR, req, "dropping resent queued req");
1542                 return 0;
1543         }
1544
1545         req = ptlrpc_clone_req(req);
1546         if (req == NULL)
1547                 return -ENOMEM;
1548
1549         spin_lock_bh(&obd->obd_processing_task_lock);
1550
1551         /* XXX O(n^2) */
1552         list_for_each(tmp, &obd->obd_req_replay_queue) {
1553                 struct ptlrpc_request *reqiter =
1554                         list_entry(tmp, struct ptlrpc_request, rq_list);
1555
1556                 if (reqiter->rq_reqmsg->transno > transno) {
1557                         list_add_tail(&req->rq_list, &reqiter->rq_list);
1558                         inserted = 1;
1559                         break;
1560                 }
1561         }
1562
1563         if (!inserted)
1564                 list_add_tail(&req->rq_list, &obd->obd_req_replay_queue);
1565
1566         obd->obd_requests_queued_for_recovery++;
1567         wake_up(&obd->obd_next_transno_waitq);
1568         spin_unlock_bh(&obd->obd_processing_task_lock);
1569         return 0;
1570 }
1571
1572 struct obd_device * target_req2obd(struct ptlrpc_request *req)
1573 {
1574         return req->rq_export->exp_obd;
1575 }
1576
1577 int
1578 target_send_reply_msg (struct ptlrpc_request *req, int rc, int fail_id)
1579 {
1580         if (OBD_FAIL_CHECK(fail_id | OBD_FAIL_ONCE)) {
1581                 obd_fail_loc |= OBD_FAIL_ONCE | OBD_FAILED;
1582                 DEBUG_REQ(D_ERROR, req, "dropping reply");
1583                 /* NB this does _not_ send with ACK disabled, to simulate
1584                  * sending OK, but timing out for the ACK */
1585                 if (req->rq_reply_state != NULL) {
1586                         if (!req->rq_reply_state->rs_difficult) {
1587                                 lustre_free_reply_state (req->rq_reply_state);
1588                                 req->rq_reply_state = NULL;
1589                         } else {
1590                                 struct ptlrpc_service *svc =
1591                                         req->rq_rqbd->rqbd_srv_ni->sni_service;
1592                                 atomic_inc(&svc->srv_outstanding_replies);
1593                         }
1594                 }
1595                 return (-ECOMM);
1596         }
1597
1598         if (rc) {
1599                 req->rq_status = rc;
1600                 return (ptlrpc_error(req));
1601         } else {
1602                 DEBUG_REQ(D_NET, req, "sending reply");
1603         }
1604
1605         return (ptlrpc_send_reply(req, 1));
1606 }
1607
1608 void
1609 target_send_reply(struct ptlrpc_request *req, int rc, int fail_id)
1610 {
1611         int                        netrc;
1612         unsigned long              flags;
1613         struct ptlrpc_reply_state *rs;
1614         struct obd_device         *obd;
1615         struct obd_export         *exp;
1616         struct ptlrpc_srv_ni      *sni;
1617         struct ptlrpc_service     *svc;
1618
1619         sni = req->rq_rqbd->rqbd_srv_ni;
1620         svc = sni->sni_service;
1621
1622         rs = req->rq_reply_state;
1623         if (rs == NULL || !rs->rs_difficult) {
1624                 /* The easy case; no notifiers and reply_out_callback()
1625                  * cleans up (i.e. we can't look inside rs after a
1626                  * successful send) */
1627                 netrc = target_send_reply_msg (req, rc, fail_id);
1628
1629                 LASSERT (netrc == 0 || req->rq_reply_state == NULL);
1630                 return;
1631         }
1632
1633         /* must be an export if locks saved */
1634         LASSERT (req->rq_export != NULL);
1635         /* req/reply consistent */
1636         LASSERT (rs->rs_srv_ni == sni);
1637
1638         /* "fresh" reply */
1639         LASSERT (!rs->rs_scheduled);
1640         LASSERT (!rs->rs_scheduled_ever);
1641         LASSERT (!rs->rs_handled);
1642         LASSERT (!rs->rs_on_net);
1643         LASSERT (rs->rs_export == NULL);
1644         LASSERT (list_empty(&rs->rs_obd_list));
1645         LASSERT (list_empty(&rs->rs_exp_list));
1646
1647         exp = class_export_get (req->rq_export);
1648         obd = exp->exp_obd;
1649
1650         /* disable reply scheduling onto srv_reply_queue while I'm setting up */
1651         rs->rs_scheduled = 1;
1652         rs->rs_on_net    = 1;
1653         rs->rs_xid       = req->rq_xid;
1654         rs->rs_transno   = req->rq_transno;
1655         rs->rs_export    = exp;
1656
1657         spin_lock_irqsave (&obd->obd_uncommitted_replies_lock, flags);
1658
1659         if (rs->rs_transno > obd->obd_last_committed) {
1660                 /* not committed already */
1661                 list_add_tail (&rs->rs_obd_list,
1662                                &obd->obd_uncommitted_replies);
1663         }
1664
1665         spin_unlock (&obd->obd_uncommitted_replies_lock);
1666         spin_lock (&exp->exp_lock);
1667
1668         list_add_tail (&rs->rs_exp_list, &exp->exp_outstanding_replies);
1669
1670         spin_unlock_irqrestore (&exp->exp_lock, flags);
1671
1672         netrc = target_send_reply_msg (req, rc, fail_id);
1673
1674         spin_lock_irqsave (&svc->srv_lock, flags);
1675
1676         svc->srv_n_difficult_replies++;
1677
1678         if (netrc != 0) /* error sending: reply is off the net */
1679                 rs->rs_on_net = 0;
1680
1681         if (!rs->rs_on_net ||                   /* some notifier */
1682             list_empty(&rs->rs_exp_list) ||     /* completed already */
1683             list_empty(&rs->rs_obd_list)) {
1684                 list_add_tail (&rs->rs_list, &svc->srv_reply_queue);
1685                 wake_up (&svc->srv_waitq);
1686         } else {
1687                 list_add (&rs->rs_list, &sni->sni_active_replies);
1688                 rs->rs_scheduled = 0;           /* allow notifier to schedule */
1689         }
1690
1691         spin_unlock_irqrestore (&svc->srv_lock, flags);
1692 }
1693
1694 int target_handle_ping(struct ptlrpc_request *req)
1695 {
1696         return lustre_pack_reply(req, 0, NULL, NULL);
1697 }