lustre/ptlrpc/import.c

   1 /* -*- mode: c; c-basic-offset: 8; indent-tabs-mode: nil; -*-
   2  * vim:expandtab:shiftwidth=8:tabstop=8:
   3  *
   4  *  Copyright (c) 2002, 2003 Cluster File Systems, Inc.
   5  *   Author: Mike Shaver <shaver@clusterfs.com>
   6  *
   7  *   This file is part of the Lustre file system, http://www.lustre.org
   8  *   Lustre is a trademark of Cluster File Systems, Inc.
   9  *
  10  *   You may have signed or agreed to another license before downloading
  11  *   this software.  If so, you are bound by the terms and conditions
  12  *   of that agreement, and the following does not apply to you.  See the
  13  *   LICENSE file included with this distribution for more information.
  14  *
  15  *   If you did not agree to a different license, then this copy of Lustre
  16  *   is open source software; you can redistribute it and/or modify it
  17  *   under the terms of version 2 of the GNU General Public License as
  18  *   published by the Free Software Foundation.
  19  *
  20  *   In either case, Lustre is distributed in the hope that it will be
  21  *   useful, but WITHOUT ANY WARRANTY; without even the implied warranty
  22  *   of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  23  *   license text for more details.
  24  */
  25
  26 #define DEBUG_SUBSYSTEM S_RPC
  27 #ifndef __KERNEL__
  28 # include <liblustre.h>
  29 #endif
  30
  31 #include <obd_support.h>
  32 #include <lustre_ha.h>
  33 #include <lustre_net.h>
  34 #include <lustre_import.h>
  35 #include <lustre_export.h>
  36 #include <obd.h>
  37 #include <obd_class.h>
  38
  39 #include "ptlrpc_internal.h"
  40
  41 struct ptlrpc_connect_async_args {
  42          __u64 pcaa_peer_committed;
  43         int pcaa_initial_connect;
  44 };
  45
  46 /* A CLOSED import should remain so. */
  47 #define IMPORT_SET_STATE_NOLOCK(imp, state)                                    \
  48 do {                                                                           \
  49         if (imp->imp_state != LUSTRE_IMP_CLOSED) {                             \
  50                CDEBUG(D_HA, "%p %s: changing import state from %s to %s\n",    \
  51                       imp, obd2cli_tgt(imp->imp_obd),                          \
  52                       ptlrpc_import_state_name(imp->imp_state),                \
  53                       ptlrpc_import_state_name(state));                        \
  54                imp->imp_state = state;                                         \
  55         }                                                                      \
  56 } while(0)
  57
  58 #define IMPORT_SET_STATE(imp, state)            \
  59 do {                                            \
  60         spin_lock(&imp->imp_lock);              \
  61         IMPORT_SET_STATE_NOLOCK(imp, state);    \
  62         spin_unlock(&imp->imp_lock);            \
  63 } while(0)
  64
  65
  66 static int ptlrpc_connect_interpret(struct ptlrpc_request *request,
  67                                     void * data, int rc);
  68 int ptlrpc_import_recovery_state_machine(struct obd_import *imp);
  69
  70 /* Only this function is allowed to change the import state when it is
  71  * CLOSED. I would rather refcount the import and free it after
  72  * disconnection like we do with exports. To do that, the client_obd
  73  * will need to save the peer info somewhere other than in the import,
  74  * though. */
  75 int ptlrpc_init_import(struct obd_import *imp)
  76 {
  77         spin_lock(&imp->imp_lock);
  78
  79         imp->imp_generation++;
  80         imp->imp_state =  LUSTRE_IMP_NEW;
  81
  82         spin_unlock(&imp->imp_lock);
  83
  84         return 0;
  85 }
  86 EXPORT_SYMBOL(ptlrpc_init_import);
  87
  88 #define UUID_STR "_UUID"
  89 static void deuuidify(char *uuid, const char *prefix, char **uuid_start,
  90                       int *uuid_len)
  91 {
  92         *uuid_start = !prefix || strncmp(uuid, prefix, strlen(prefix))
  93                 ? uuid : uuid + strlen(prefix);
  94
  95         *uuid_len = strlen(*uuid_start);
  96
  97         if (*uuid_len < strlen(UUID_STR))
  98                 return;
  99
 100         if (!strncmp(*uuid_start + *uuid_len - strlen(UUID_STR),
 101                     UUID_STR, strlen(UUID_STR)))
 102                 *uuid_len -= strlen(UUID_STR);
 103 }
 104
 105 /* Returns true if import was FULL, false if import was already not
 106  * connected.
 107  * @imp - import to be disconnected
 108  * @conn_cnt - connection count (epoch) of the request that timed out
 109  *             and caused the disconnection.  In some cases, multiple
 110  *             inflight requests can fail to a single target (e.g. OST
 111  *             bulk requests) and if one has already caused a reconnection
 112  *             (increasing the import->conn_cnt) the older failure should
 113  *             not also cause a reconnection.  If zero it forces a reconnect.
 114  */
 115 int ptlrpc_set_import_discon(struct obd_import *imp, __u32 conn_cnt)
 116 {
 117         int rc = 0;
 118
 119         spin_lock(&imp->imp_lock);
 120
 121         if (imp->imp_state == LUSTRE_IMP_FULL &&
 122             (conn_cnt == 0 || conn_cnt == imp->imp_conn_cnt)) {
 123                 char *target_start;
 124                 int   target_len;
 125
 126                 deuuidify(obd2cli_tgt(imp->imp_obd), NULL,
 127                           &target_start, &target_len);
 128                 if (imp->imp_replayable) {
 129                         LCONSOLE_WARN("%s: Connection to service %.*s via nid "
 130                                "%s was lost; in progress operations using this "
 131                                "service will wait for recovery to complete.\n",
 132                                imp->imp_obd->obd_name, target_len, target_start,
 133                                libcfs_nid2str(imp->imp_connection->c_peer.nid));
 134                 } else {
 135                         LCONSOLE_ERROR("%s: Connection to service %.*s via nid "
 136                                "%s was lost; in progress operations using this "
 137                                "service will fail.\n",
 138                                imp->imp_obd->obd_name, target_len, target_start,
 139                                libcfs_nid2str(imp->imp_connection->c_peer.nid));
 140                 }
 141                 IMPORT_SET_STATE_NOLOCK(imp, LUSTRE_IMP_DISCON);
 142                 spin_unlock(&imp->imp_lock);
 143
 144                 if (obd_dump_on_timeout)
 145                         libcfs_debug_dumplog();
 146
 147                 obd_import_event(imp->imp_obd, imp, IMP_EVENT_DISCON);
 148                 rc = 1;
 149         } else {
 150                 spin_unlock(&imp->imp_lock);
 151                 CDEBUG(D_HA, "%s: import %p already %s (conn %u, was %u): %s\n",
 152                        imp->imp_client->cli_name, imp,
 153                        (imp->imp_state == LUSTRE_IMP_FULL &&
 154                         imp->imp_conn_cnt > conn_cnt) ?
 155                        "reconnected" : "not connected", imp->imp_conn_cnt,
 156                        conn_cnt, ptlrpc_import_state_name(imp->imp_state));
 157         }
 158
 159         return rc;
 160 }
 161
 162 /*
 163  * This acts as a barrier; all existing requests are rejected, and
 164  * no new requests will be accepted until the import is valid again.
 165  */
 166 void ptlrpc_deactivate_import(struct obd_import *imp)
 167 {
 168         ENTRY;
 169
 170         spin_lock(&imp->imp_lock);
 171         CDEBUG(D_HA, "setting import %s INVALID\n", obd2cli_tgt(imp->imp_obd));
 172         imp->imp_invalid = 1;
 173         imp->imp_generation++;
 174         spin_unlock(&imp->imp_lock);
 175
 176         ptlrpc_abort_inflight(imp);
 177         obd_import_event(imp->imp_obd, imp, IMP_EVENT_INACTIVE);
 178 }
 179
 180 /*
 181  * This function will invalidate the import, if necessary, then block
 182  * for all the RPC completions, and finally notify the obd to
 183  * invalidate its state (ie cancel locks, clear pending requests,
 184  * etc).
 185  */
 186 void ptlrpc_invalidate_import(struct obd_import *imp)
 187 {
 188         struct l_wait_info lwi;
 189         int rc;
 190
 191         if (!imp->imp_invalid)
 192                 ptlrpc_deactivate_import(imp);
 193
 194         LASSERT(imp->imp_invalid);
 195
 196         /* wait for all requests to error out and call completion callbacks */
 197         lwi = LWI_TIMEOUT_INTERVAL(cfs_timeout_cap(cfs_time_seconds(obd_timeout)),
 198                                    HZ, NULL, NULL);
 199         rc = l_wait_event(imp->imp_recovery_waitq,
 200                           (atomic_read(&imp->imp_inflight) == 0),
 201                           &lwi);
 202
 203         if (rc)
 204                 CDEBUG(D_HA, "%s: rc = %d waiting for callback (%d != 0)\n",
 205                        obd2cli_tgt(imp->imp_obd), rc,
 206                        atomic_read(&imp->imp_inflight));
 207
 208         obd_import_event(imp->imp_obd, imp, IMP_EVENT_INVALIDATE);
 209 }
 210
 211 /* unset imp_invalid */
 212 void ptlrpc_activate_import(struct obd_import *imp)
 213 {
 214         struct obd_device *obd = imp->imp_obd;
 215
 216         spin_lock(&imp->imp_lock);
 217         imp->imp_invalid = 0;
 218         spin_unlock(&imp->imp_lock);
 219
 220         obd_import_event(obd, imp, IMP_EVENT_ACTIVE);
 221 }
 222
 223 void ptlrpc_fail_import(struct obd_import *imp, __u32 conn_cnt)
 224 {
 225         ENTRY;
 226
 227         LASSERT(!imp->imp_dlm_fake);
 228
 229         if (ptlrpc_set_import_discon(imp, conn_cnt)) {
 230                 if (!imp->imp_replayable) {
 231                         CDEBUG(D_HA, "import %s@%s for %s not replayable, "
 232                                "auto-deactivating\n",
 233                                obd2cli_tgt(imp->imp_obd),
 234                                imp->imp_connection->c_remote_uuid.uuid,
 235                                imp->imp_obd->obd_name);
 236                         ptlrpc_deactivate_import(imp);
 237                 }
 238
 239                 CDEBUG(D_HA, "%s: waking up pinger\n",
 240                        obd2cli_tgt(imp->imp_obd));
 241
 242                 spin_lock(&imp->imp_lock);
 243                 imp->imp_force_verify = 1;
 244                 spin_unlock(&imp->imp_lock);
 245
 246                 ptlrpc_pinger_wake_up();
 247         }
 248         EXIT;
 249 }
 250
 251 static int import_select_connection(struct obd_import *imp)
 252 {
 253         struct obd_import_conn *imp_conn = NULL, *conn;
 254         struct obd_export *dlmexp;
 255         ENTRY;
 256
 257         spin_lock(&imp->imp_lock);
 258
 259         if (list_empty(&imp->imp_conn_list)) {
 260                 CERROR("%s: no connections available\n",
 261                         imp->imp_obd->obd_name);
 262                 spin_unlock(&imp->imp_lock);
 263                 RETURN(-EINVAL);
 264         }
 265
 266         list_for_each_entry(conn, &imp->imp_conn_list, oic_item) {
 267                 CDEBUG(D_HA, "%s: connect to NID %s last attempt "LPU64"\n",
 268                        imp->imp_obd->obd_name,
 269                        libcfs_nid2str(conn->oic_conn->c_peer.nid),
 270                        conn->oic_last_attempt);
 271                 /* Throttle the reconnect rate to once per RECONNECT_INTERVAL */
 272                 if (cfs_time_before_64(conn->oic_last_attempt +
 273                                        RECONNECT_INTERVAL * HZ,
 274                                        cfs_time_current_64())) {
 275                         /* If we have never tried this connection since the
 276                            the last successful attempt, go with this one */
 277                         if (cfs_time_before_64(conn->oic_last_attempt,
 278                                                imp->imp_last_success_conn)) {
 279                                 imp_conn = conn;
 280                                 break;
 281                         }
 282
 283                         /* Both of these connections have already been tried
 284                            since the last successful connection; just choose the
 285                            least recently used */
 286                         if (!imp_conn)
 287                                 imp_conn = conn;
 288                         else if (cfs_time_before_64(conn->oic_last_attempt,
 289                                                     imp_conn->oic_last_attempt))
 290                                 imp_conn = conn;
 291                 }
 292         }
 293
 294         /* if not found, simply choose the current one */
 295         if (!imp_conn) {
 296                 LASSERT(imp->imp_conn_current);
 297                 imp_conn = imp->imp_conn_current;
 298         }
 299         LASSERT(imp_conn->oic_conn);
 300
 301         imp_conn->oic_last_attempt = cfs_time_current_64();
 302
 303         /* switch connection, don't mind if it's same as the current one */
 304         if (imp->imp_connection)
 305                 ptlrpc_put_connection(imp->imp_connection);
 306         imp->imp_connection = ptlrpc_connection_addref(imp_conn->oic_conn);
 307
 308         dlmexp =  class_conn2export(&imp->imp_dlm_handle);
 309         LASSERT(dlmexp != NULL);
 310         if (dlmexp->exp_connection)
 311                 ptlrpc_put_connection(dlmexp->exp_connection);
 312         dlmexp->exp_connection = ptlrpc_connection_addref(imp_conn->oic_conn);
 313         class_export_put(dlmexp);
 314
 315         if (imp->imp_conn_current != imp_conn) {
 316                 if (imp->imp_conn_current)
 317                         LCONSOLE_INFO("Changing connection for %s to %s/%s\n",
 318                                       imp->imp_obd->obd_name,
 319                                       imp_conn->oic_uuid.uuid,
 320                                       libcfs_nid2str(imp_conn->oic_conn->c_peer.nid));
 321                 imp->imp_conn_current = imp_conn;
 322         }
 323
 324         CDEBUG(D_HA, "%s: import %p using connection %s/%s\n",
 325                imp->imp_obd->obd_name, imp, imp_conn->oic_uuid.uuid,
 326                libcfs_nid2str(imp_conn->oic_conn->c_peer.nid));
 327
 328         spin_unlock(&imp->imp_lock);
 329
 330         RETURN(0);
 331 }
 332
 333 int ptlrpc_connect_import(struct obd_import *imp, char *new_uuid)
 334 {
 335         struct obd_device *obd = imp->imp_obd;
 336         int initial_connect = 0;
 337         int rc;
 338         __u64 committed_before_reconnect = 0;
 339         struct ptlrpc_request *request;
 340         int size[] = { sizeof(struct ptlrpc_body),
 341                        sizeof(imp->imp_obd->u.cli.cl_target_uuid),
 342                        sizeof(obd->obd_uuid),
 343                        sizeof(imp->imp_dlm_handle),
 344                        sizeof(imp->imp_connect_data) };
 345         char *tmp[] = { NULL,
 346                         obd2cli_tgt(imp->imp_obd),
 347                         obd->obd_uuid.uuid,
 348                         (char *)&imp->imp_dlm_handle,
 349                         (char *)&imp->imp_connect_data };
 350         struct ptlrpc_connect_async_args *aa;
 351
 352         ENTRY;
 353         spin_lock(&imp->imp_lock);
 354         if (imp->imp_state == LUSTRE_IMP_CLOSED) {
 355                 spin_unlock(&imp->imp_lock);
 356                 CERROR("can't connect to a closed import\n");
 357                 RETURN(-EINVAL);
 358         } else if (imp->imp_state == LUSTRE_IMP_FULL) {
 359                 spin_unlock(&imp->imp_lock);
 360                 CERROR("already connected\n");
 361                 RETURN(0);
 362         } else if (imp->imp_state == LUSTRE_IMP_CONNECTING) {
 363                 spin_unlock(&imp->imp_lock);
 364                 CERROR("already connecting\n");
 365                 RETURN(-EALREADY);
 366         }
 367
 368         IMPORT_SET_STATE_NOLOCK(imp, LUSTRE_IMP_CONNECTING);
 369
 370         imp->imp_conn_cnt++;
 371         imp->imp_resend_replay = 0;
 372
 373         if (!lustre_handle_is_used(&imp->imp_remote_handle))
 374                 initial_connect = 1;
 375         else
 376                 committed_before_reconnect = imp->imp_peer_committed_transno;
 377
 378         spin_unlock(&imp->imp_lock);
 379
 380         if (new_uuid) {
 381                 struct obd_uuid uuid;
 382
 383                 obd_str2uuid(&uuid, new_uuid);
 384                 rc = import_set_conn_priority(imp, &uuid);
 385                 if (rc)
 386                         GOTO(out, rc);
 387         }
 388
 389         rc = import_select_connection(imp);
 390         if (rc)
 391                 GOTO(out, rc);
 392
 393         /* last in connection list */
 394         if (imp->imp_conn_current->oic_item.next == &imp->imp_conn_list) {
 395                 if (imp->imp_initial_recov_bk && initial_connect) {
 396                         CDEBUG(D_HA, "Last connection attempt (%d) for %s\n",
 397                                imp->imp_conn_cnt, obd2cli_tgt(imp->imp_obd));
 398                         /* Don't retry if connect fails */
 399                         rc = 0;
 400                         obd_set_info_async(obd->obd_self_export,
 401                                            strlen(KEY_INIT_RECOV),
 402                                            KEY_INIT_RECOV,
 403                                            sizeof(rc), &rc, NULL);
 404                 }
 405                 if (imp->imp_recon_bk) {
 406                         CDEBUG(D_HA, "Last reconnection attempt (%d) for %s\n",
 407                                imp->imp_conn_cnt, obd2cli_tgt(imp->imp_obd));
 408                         imp->imp_last_recon = 1;
 409                 }
 410         }
 411
 412         /* Reset connect flags to the originally requested flags, in case
 413          * the server is updated on-the-fly we will get the new features. */
 414         imp->imp_connect_data.ocd_connect_flags = imp->imp_connect_flags_orig;
 415         rc = obd_reconnect(imp->imp_obd->obd_self_export, obd,
 416                            &obd->obd_uuid, &imp->imp_connect_data);
 417         if (rc)
 418                 GOTO(out, rc);
 419
 420         request = ptlrpc_prep_req(imp, LUSTRE_OBD_VERSION, imp->imp_connect_op,
 421                                   5, size, tmp);
 422         if (!request)
 423                 GOTO(out, rc = -ENOMEM);
 424
 425 #ifndef __KERNEL__
 426         lustre_msg_add_op_flags(request->rq_reqmsg, MSG_CONNECT_LIBCLIENT);
 427 #endif
 428         lustre_msg_add_op_flags(request->rq_reqmsg, MSG_CONNECT_NEXT_VER);
 429
 430         request->rq_send_state = LUSTRE_IMP_CONNECTING;
 431         /* Allow a slightly larger reply for future growth compatibility */
 432         size[REPLY_REC_OFF] = sizeof(struct obd_connect_data) +
 433                               16 * sizeof(__u64);
 434         ptlrpc_req_set_repsize(request, 2, size);
 435         request->rq_interpret_reply = ptlrpc_connect_interpret;
 436
 437         CLASSERT(sizeof (*aa) <= sizeof (request->rq_async_args));
 438         aa = (struct ptlrpc_connect_async_args *)&request->rq_async_args;
 439         memset(aa, 0, sizeof *aa);
 440
 441         aa->pcaa_peer_committed = committed_before_reconnect;
 442         aa->pcaa_initial_connect = initial_connect;
 443
 444         if (aa->pcaa_initial_connect) {
 445                 imp->imp_replayable = 1;
 446                 /* On an initial connect, we don't know which one of a
 447                    failover server pair is up.  Don't wait long. */
 448                 request->rq_timeout = max((int)(obd_timeout / 20), 5);
 449         }
 450
 451         DEBUG_REQ(D_RPCTRACE, request, "(re)connect request");
 452         ptlrpcd_add_req(request);
 453         rc = 0;
 454 out:
 455         if (rc != 0) {
 456                 IMPORT_SET_STATE(imp, LUSTRE_IMP_DISCON);
 457         }
 458
 459         RETURN(rc);
 460 }
 461 EXPORT_SYMBOL(ptlrpc_connect_import);
 462
 463 static void ptlrpc_maybe_ping_import_soon(struct obd_import *imp)
 464 {
 465 #ifdef __KERNEL__
 466         struct obd_import_conn *imp_conn;
 467 #endif
 468         int wake_pinger = 0;
 469
 470         ENTRY;
 471
 472         spin_lock(&imp->imp_lock);
 473         if (list_empty(&imp->imp_conn_list))
 474                 GOTO(unlock, 0);
 475
 476 #ifdef __KERNEL__
 477         imp_conn = list_entry(imp->imp_conn_list.prev,
 478                               struct obd_import_conn,
 479                               oic_item);
 480
 481         if (imp->imp_conn_current != imp_conn) {
 482                 ptlrpc_ping_import_soon(imp);
 483                 wake_pinger = 1;
 484         }
 485
 486 #else
 487         /* liblustre has no pinger thead, so we wakup pinger anyway */
 488         wake_pinger = 1;
 489 #endif
 490  unlock:
 491         spin_unlock(&imp->imp_lock);
 492
 493         if (wake_pinger)
 494                 ptlrpc_pinger_wake_up();
 495
 496         EXIT;
 497 }
 498
 499 static int ptlrpc_connect_interpret(struct ptlrpc_request *request,
 500                                     void * data, int rc)
 501 {
 502         struct ptlrpc_connect_async_args *aa = data;
 503         struct obd_import *imp = request->rq_import;
 504         struct client_obd *cli = &imp->imp_obd->u.cli;
 505         struct lustre_handle old_hdl;
 506         int msg_flags;
 507         ENTRY;
 508
 509         spin_lock(&imp->imp_lock);
 510         if (imp->imp_state == LUSTRE_IMP_CLOSED) {
 511                 spin_unlock(&imp->imp_lock);
 512                 RETURN(0);
 513         }
 514         spin_unlock(&imp->imp_lock);
 515
 516         if (rc)
 517                 GOTO(out, rc);
 518
 519         LASSERT(imp->imp_conn_current);
 520
 521         msg_flags = lustre_msg_get_op_flags(request->rq_repmsg);
 522
 523         /* All imports are pingable */
 524         imp->imp_pingable = 1;
 525
 526         if (aa->pcaa_initial_connect) {
 527                 if (msg_flags & MSG_CONNECT_REPLAYABLE) {
 528                         CDEBUG(D_HA, "connected to replayable target: %s\n",
 529                                obd2cli_tgt(imp->imp_obd));
 530                         imp->imp_replayable = 1;
 531                 } else {
 532                         imp->imp_replayable = 0;
 533                 }
 534
 535                 if (msg_flags & MSG_CONNECT_NEXT_VER) {
 536                         imp->imp_msg_magic = LUSTRE_MSG_MAGIC_V2;
 537                         CDEBUG(D_RPCTRACE, "connect to %s with lustre_msg_v2\n",
 538                                obd2cli_tgt(imp->imp_obd));
 539                 } else {
 540                         CDEBUG(D_RPCTRACE, "connect to %s with lustre_msg_v1\n",
 541                                obd2cli_tgt(imp->imp_obd));
 542                 }
 543
 544                 imp->imp_remote_handle =
 545                                 *lustre_msg_get_handle(request->rq_repmsg);
 546
 547                 IMPORT_SET_STATE(imp, LUSTRE_IMP_FULL);
 548                 GOTO(finish, rc = 0);
 549         }
 550
 551         /* Determine what recovery state to move the import to. */
 552         if (MSG_CONNECT_RECONNECT & msg_flags) {
 553                 memset(&old_hdl, 0, sizeof(old_hdl));
 554                 if (!memcmp(&old_hdl, lustre_msg_get_handle(request->rq_repmsg),
 555                             sizeof (old_hdl))) {
 556                         CERROR("%s@%s didn't like our handle "LPX64
 557                                ", failed\n", obd2cli_tgt(imp->imp_obd),
 558                                imp->imp_connection->c_remote_uuid.uuid,
 559                                imp->imp_dlm_handle.cookie);
 560                         GOTO(out, rc = -ENOTCONN);
 561                 }
 562
 563                 if (memcmp(&imp->imp_remote_handle,
 564                            lustre_msg_get_handle(request->rq_repmsg),
 565                            sizeof(imp->imp_remote_handle))) {
 566                         int level = D_ERROR;
 567                         /* Old MGC can reconnect to a restarted MGS */
 568                         if (strcmp(imp->imp_obd->obd_type->typ_name,
 569                                    LUSTRE_MGC_NAME) == 0) {
 570                                 level = D_CONFIG;
 571                         }
 572                         CDEBUG(level,
 573                                "%s@%s changed handle from "LPX64" to "LPX64
 574                                "; copying, but this may foreshadow disaster\n",
 575                                obd2cli_tgt(imp->imp_obd),
 576                                imp->imp_connection->c_remote_uuid.uuid,
 577                                imp->imp_remote_handle.cookie,
 578                                lustre_msg_get_handle(request->rq_repmsg)->
 579                                         cookie);
 580                         imp->imp_remote_handle =
 581                                      *lustre_msg_get_handle(request->rq_repmsg);
 582                 } else {
 583                         CDEBUG(D_HA, "reconnected to %s@%s after partition\n",
 584                                obd2cli_tgt(imp->imp_obd),
 585                                imp->imp_connection->c_remote_uuid.uuid);
 586                 }
 587
 588                 if (imp->imp_invalid) {
 589                         IMPORT_SET_STATE(imp, LUSTRE_IMP_EVICTED);
 590                 } else if (MSG_CONNECT_RECOVERING & msg_flags) {
 591                         CDEBUG(D_HA, "%s: reconnected to %s during replay\n",
 592                                imp->imp_obd->obd_name,
 593                                obd2cli_tgt(imp->imp_obd));
 594                         imp->imp_resend_replay = 1;
 595                         IMPORT_SET_STATE(imp, LUSTRE_IMP_REPLAY);
 596                 } else {
 597                         IMPORT_SET_STATE(imp, LUSTRE_IMP_RECOVER);
 598                 }
 599         } else if ((MSG_CONNECT_RECOVERING & msg_flags) && !imp->imp_invalid) {
 600                 LASSERT(imp->imp_replayable);
 601                 imp->imp_remote_handle =
 602                                 *lustre_msg_get_handle(request->rq_repmsg);
 603                 imp->imp_last_replay_transno = 0;
 604                 IMPORT_SET_STATE(imp, LUSTRE_IMP_REPLAY);
 605         } else {
 606                 imp->imp_remote_handle =
 607                                 *lustre_msg_get_handle(request->rq_repmsg);
 608                 IMPORT_SET_STATE(imp, LUSTRE_IMP_EVICTED);
 609         }
 610
 611         /* Sanity checks for a reconnected import. */
 612         if (!(imp->imp_replayable) != !(msg_flags & MSG_CONNECT_REPLAYABLE)) {
 613                 CERROR("imp_replayable flag does not match server "
 614                        "after reconnect. We should LBUG right here.\n");
 615         }
 616
 617         if (lustre_msg_get_last_committed(request->rq_repmsg) <
 618             aa->pcaa_peer_committed) {
 619                 CERROR("%s went back in time (transno "LPD64
 620                        " was previously committed, server now claims "LPD64
 621                        ")!  See https://bugzilla.clusterfs.com/"
 622                        "long_list.cgi?buglist=9646\n",
 623                        obd2cli_tgt(imp->imp_obd), aa->pcaa_peer_committed,
 624                        lustre_msg_get_last_committed(request->rq_repmsg));
 625         }
 626
 627 finish:
 628         rc = ptlrpc_import_recovery_state_machine(imp);
 629         if (rc != 0) {
 630                 if (rc == -ENOTCONN) {
 631                         CDEBUG(D_HA, "evicted/aborted by %s@%s during recovery;"
 632                                "invalidating and reconnecting\n",
 633                                obd2cli_tgt(imp->imp_obd),
 634                                imp->imp_connection->c_remote_uuid.uuid);
 635                         ptlrpc_connect_import(imp, NULL);
 636                         RETURN(0);
 637                 }
 638         } else {
 639                 struct obd_connect_data *ocd;
 640                 struct obd_export *exp;
 641
 642                 ocd = lustre_swab_repbuf(request, REPLY_REC_OFF, sizeof(*ocd),
 643                                          lustre_swab_connect);
 644
 645                 spin_lock(&imp->imp_lock);
 646                 list_del(&imp->imp_conn_current->oic_item);
 647                 list_add(&imp->imp_conn_current->oic_item, &imp->imp_conn_list);
 648                 imp->imp_last_success_conn =
 649                         imp->imp_conn_current->oic_last_attempt;
 650
 651                 if (ocd == NULL) {
 652                         spin_unlock(&imp->imp_lock);
 653                         CERROR("Wrong connect data from server\n");
 654                         rc = -EPROTO;
 655                         GOTO(out, rc);
 656                 }
 657
 658                 imp->imp_connect_data = *ocd;
 659
 660                 exp = class_conn2export(&imp->imp_dlm_handle);
 661                 spin_unlock(&imp->imp_lock);
 662
 663                 /* check that server granted subset of flags we asked for. */
 664                 LASSERTF((ocd->ocd_connect_flags &
 665                           imp->imp_connect_flags_orig) ==
 666                          ocd->ocd_connect_flags, LPX64" != "LPX64,
 667                          imp->imp_connect_flags_orig, ocd->ocd_connect_flags);
 668
 669                 if (!exp) {
 670                         /* This could happen if export is cleaned during the
 671                            connect attempt */
 672                         spin_unlock(&imp->imp_lock);
 673                         CERROR("Missing export for %s\n",
 674                                imp->imp_obd->obd_name);
 675                         GOTO(out, rc = -ENODEV);
 676                 }
 677                 exp->exp_connect_flags = ocd->ocd_connect_flags;
 678                 class_export_put(exp);
 679
 680                 obd_import_event(imp->imp_obd, imp, IMP_EVENT_OCD);
 681
 682                 if (!ocd->ocd_ibits_known &&
 683                     ocd->ocd_connect_flags & OBD_CONNECT_IBITS)
 684                         CERROR("Inodebits aware server returned zero compatible"
 685                                " bits?\n");
 686
 687                 if ((ocd->ocd_connect_flags & OBD_CONNECT_VERSION) &&
 688                     (ocd->ocd_version > LUSTRE_VERSION_CODE +
 689                     LUSTRE_VERSION_OFFSET_WARN)) {
 690                         /* Sigh, some compilers do not like #ifdef in the middle
 691                            of macro arguments */
 692 #ifdef __KERNEL__
 693                         const char *action = "upgrading this client";
 694 #else
 695                         const char *action = "recompiling this application";
 696 #endif
 697
 698                         CWARN("Server %s version (%d.%d.%d.%d) is much newer. "
 699                               "Consider %s (%s).\n",
 700                               obd2cli_tgt(imp->imp_obd),
 701                               OBD_OCD_VERSION_MAJOR(ocd->ocd_version),
 702                               OBD_OCD_VERSION_MINOR(ocd->ocd_version),
 703                               OBD_OCD_VERSION_PATCH(ocd->ocd_version),
 704                               OBD_OCD_VERSION_FIX(ocd->ocd_version),
 705                               action, LUSTRE_VERSION_STRING);
 706                 }
 707
 708                 if (ocd->ocd_connect_flags & OBD_CONNECT_BRW_SIZE) {
 709                         cli->cl_max_pages_per_rpc =
 710                                 ocd->ocd_brw_size >> CFS_PAGE_SHIFT;
 711                 }
 712
 713                 LASSERT((cli->cl_max_pages_per_rpc <= PTLRPC_MAX_BRW_PAGES) &&
 714                         (cli->cl_max_pages_per_rpc > 0));
 715         }
 716
 717  out:
 718         if (rc != 0) {
 719                 IMPORT_SET_STATE(imp, LUSTRE_IMP_DISCON);
 720                 if (aa->pcaa_initial_connect && !imp->imp_initial_recov)
 721                         ptlrpc_deactivate_import(imp);
 722
 723                 if (imp->imp_recon_bk && imp->imp_last_recon) {
 724                         /* Give up trying to reconnect */
 725                         imp->imp_obd->obd_no_recov = 1;
 726                         ptlrpc_deactivate_import(imp);
 727                 }
 728
 729                 if (rc == -EPROTO) {
 730                         struct obd_connect_data *ocd;
 731                         ocd = lustre_swab_repbuf(request, REPLY_REC_OFF,
 732                                                  sizeof *ocd,
 733                                                  lustre_swab_connect);
 734                         if (ocd &&
 735                             (ocd->ocd_connect_flags & OBD_CONNECT_VERSION) &&
 736                             (ocd->ocd_version != LUSTRE_VERSION_CODE)) {
 737                            /* Actually servers are only supposed to refuse
 738                               connection from liblustre clients, so we should
 739                               never see this from VFS context */
 740                                 CERROR("Server %s version (%d.%d.%d.%d) "
 741                                        "refused connection from this client "
 742                                        "as too old version (%s).  Client must "
 743                                        "be recompiled\n",
 744                                       obd2cli_tgt(imp->imp_obd),
 745                                       OBD_OCD_VERSION_MAJOR(ocd->ocd_version),
 746                                       OBD_OCD_VERSION_MINOR(ocd->ocd_version),
 747                                       OBD_OCD_VERSION_PATCH(ocd->ocd_version),
 748                                       OBD_OCD_VERSION_FIX(ocd->ocd_version),
 749                                       LUSTRE_VERSION_STRING);
 750                                 ptlrpc_deactivate_import(imp);
 751                                 IMPORT_SET_STATE(imp, LUSTRE_IMP_CLOSED);
 752                         }
 753                         RETURN(-EPROTO);
 754                 }
 755
 756                 ptlrpc_maybe_ping_import_soon(imp);
 757
 758                 CDEBUG(D_HA, "recovery of %s on %s failed (%d)\n",
 759                        obd2cli_tgt(imp->imp_obd),
 760                        (char *)imp->imp_connection->c_remote_uuid.uuid, rc);
 761         }
 762
 763         imp->imp_last_recon = 0;
 764
 765         cfs_waitq_signal(&imp->imp_recovery_waitq);
 766         RETURN(rc);
 767 }
 768
 769 static int completed_replay_interpret(struct ptlrpc_request *req,
 770                                     void * data, int rc)
 771 {
 772         ENTRY;
 773         atomic_dec(&req->rq_import->imp_replay_inflight);
 774         if (req->rq_status == 0) {
 775                 ptlrpc_import_recovery_state_machine(req->rq_import);
 776         } else {
 777                 CDEBUG(D_HA, "%s: LAST_REPLAY message error: %d, "
 778                        "reconnecting\n",
 779                        req->rq_import->imp_obd->obd_name, req->rq_status);
 780                 ptlrpc_connect_import(req->rq_import, NULL);
 781         }
 782
 783         RETURN(0);
 784 }
 785
 786 static int signal_completed_replay(struct obd_import *imp)
 787 {
 788         struct ptlrpc_request *req;
 789         ENTRY;
 790
 791         LASSERT(atomic_read(&imp->imp_replay_inflight) == 0);
 792         atomic_inc(&imp->imp_replay_inflight);
 793
 794         req = ptlrpc_prep_req(imp, LUSTRE_OBD_VERSION, OBD_PING, 1, NULL, NULL);
 795         if (!req) {
 796                 atomic_dec(&imp->imp_replay_inflight);
 797                 RETURN(-ENOMEM);
 798         }
 799
 800         ptlrpc_req_set_repsize(req, 1, NULL);
 801         req->rq_send_state = LUSTRE_IMP_REPLAY_WAIT;
 802         lustre_msg_add_flags(req->rq_reqmsg, MSG_LAST_REPLAY);
 803         req->rq_timeout *= 3;
 804         req->rq_interpret_reply = completed_replay_interpret;
 805
 806         ptlrpcd_add_req(req);
 807         RETURN(0);
 808 }
 809
 810 #ifdef __KERNEL__
 811 static int ptlrpc_invalidate_import_thread(void *data)
 812 {
 813         struct obd_import *imp = data;
 814
 815         ENTRY;
 816
 817         ptlrpc_daemonize("ll_imp_inval");
 818
 819         CDEBUG(D_HA, "thread invalidate import %s to %s@%s\n",
 820                imp->imp_obd->obd_name, obd2cli_tgt(imp->imp_obd),
 821                imp->imp_connection->c_remote_uuid.uuid);
 822
 823         ptlrpc_invalidate_import(imp);
 824
 825         if (obd_dump_on_eviction) {
 826                 CERROR("dump the log upon eviction\n");
 827                 libcfs_debug_dumplog();
 828         }
 829
 830         IMPORT_SET_STATE(imp, LUSTRE_IMP_RECOVER);
 831         ptlrpc_import_recovery_state_machine(imp);
 832
 833         RETURN(0);
 834 }
 835 #endif
 836
 837 int ptlrpc_import_recovery_state_machine(struct obd_import *imp)
 838 {
 839         int rc = 0;
 840         int inflight;
 841         char *target_start;
 842         int target_len;
 843
 844         ENTRY;
 845         if (imp->imp_state == LUSTRE_IMP_EVICTED) {
 846                 deuuidify(obd2cli_tgt(imp->imp_obd), NULL,
 847                           &target_start, &target_len);
 848                 /* Don't care about MGC eviction */
 849                 if (strcmp(imp->imp_obd->obd_type->typ_name,
 850                            LUSTRE_MGC_NAME) != 0) {
 851                         LCONSOLE_ERROR("This client was evicted by %.*s; "
 852                                        "in progress operations using this "
 853                                        "service will fail.\n",
 854                                        target_len, target_start);
 855                 }
 856                 CDEBUG(D_HA, "evicted from %s@%s; invalidating\n",
 857                        obd2cli_tgt(imp->imp_obd),
 858                        imp->imp_connection->c_remote_uuid.uuid);
 859
 860 #ifdef __KERNEL__
 861                 rc = cfs_kernel_thread(ptlrpc_invalidate_import_thread, imp,
 862                                    CLONE_VM | CLONE_FILES);
 863                 if (rc < 0)
 864                         CERROR("error starting invalidate thread: %d\n", rc);
 865                 else
 866                         rc = 0;
 867                 RETURN(rc);
 868 #else
 869                 ptlrpc_invalidate_import(imp);
 870
 871                 IMPORT_SET_STATE(imp, LUSTRE_IMP_RECOVER);
 872 #endif
 873         }
 874
 875         if (imp->imp_state == LUSTRE_IMP_REPLAY) {
 876                 CDEBUG(D_HA, "replay requested by %s\n",
 877                        obd2cli_tgt(imp->imp_obd));
 878                 rc = ptlrpc_replay_next(imp, &inflight);
 879                 if (inflight == 0 &&
 880                     atomic_read(&imp->imp_replay_inflight) == 0) {
 881                         IMPORT_SET_STATE(imp, LUSTRE_IMP_REPLAY_LOCKS);
 882                         rc = ldlm_replay_locks(imp);
 883                         if (rc)
 884                                 GOTO(out, rc);
 885                 }
 886                 rc = 0;
 887         }
 888
 889         if (imp->imp_state == LUSTRE_IMP_REPLAY_LOCKS) {
 890                 if (atomic_read(&imp->imp_replay_inflight) == 0) {
 891                         IMPORT_SET_STATE(imp, LUSTRE_IMP_REPLAY_WAIT);
 892                         rc = signal_completed_replay(imp);
 893                         if (rc)
 894                                 GOTO(out, rc);
 895                 }
 896
 897         }
 898
 899         if (imp->imp_state == LUSTRE_IMP_REPLAY_WAIT) {
 900                 if (atomic_read(&imp->imp_replay_inflight) == 0) {
 901                         IMPORT_SET_STATE(imp, LUSTRE_IMP_RECOVER);
 902                 }
 903         }
 904
 905         if (imp->imp_state == LUSTRE_IMP_RECOVER) {
 906                 CDEBUG(D_HA, "reconnected to %s@%s\n",
 907                        obd2cli_tgt(imp->imp_obd),
 908                        imp->imp_connection->c_remote_uuid.uuid);
 909
 910                 rc = ptlrpc_resend(imp);
 911                 if (rc)
 912                         GOTO(out, rc);
 913                 IMPORT_SET_STATE(imp, LUSTRE_IMP_FULL);
 914                 ptlrpc_activate_import(imp);
 915
 916                 deuuidify(obd2cli_tgt(imp->imp_obd), NULL,
 917                           &target_start, &target_len);
 918                 LCONSOLE_INFO("%s: Connection restored to service %.*s "
 919                               "using nid %s.\n", imp->imp_obd->obd_name,
 920                               target_len, target_start,
 921                               libcfs_nid2str(imp->imp_connection->c_peer.nid));
 922         }
 923
 924         if (imp->imp_state == LUSTRE_IMP_FULL) {
 925                 cfs_waitq_signal(&imp->imp_recovery_waitq);
 926                 ptlrpc_wake_delayed(imp);
 927         }
 928
 929  out:
 930         RETURN(rc);
 931 }
 932
 933 static int back_to_sleep(void *unused)
 934 {
 935         return 0;
 936 }
 937
 938 int ptlrpc_disconnect_import(struct obd_import *imp, int noclose)
 939 {
 940         struct ptlrpc_request *req;
 941         int rq_opc, rc = 0;
 942         ENTRY;
 943
 944         switch (imp->imp_connect_op) {
 945         case OST_CONNECT: rq_opc = OST_DISCONNECT; break;
 946         case MDS_CONNECT: rq_opc = MDS_DISCONNECT; break;
 947         case MGS_CONNECT: rq_opc = MGS_DISCONNECT; break;
 948         default:
 949                 CERROR("don't know how to disconnect from %s (connect_op %d)\n",
 950                        obd2cli_tgt(imp->imp_obd), imp->imp_connect_op);
 951                 RETURN(-EINVAL);
 952         }
 953
 954         if (ptlrpc_import_in_recovery(imp)) {
 955                 struct l_wait_info lwi;
 956                 cfs_duration_t timeout = cfs_time_seconds(obd_timeout);
 957
 958                 lwi = LWI_TIMEOUT_INTR(cfs_timeout_cap(timeout),
 959                                        back_to_sleep, LWI_ON_SIGNAL_NOOP, NULL);
 960                 rc = l_wait_event(imp->imp_recovery_waitq,
 961                                   !ptlrpc_import_in_recovery(imp), &lwi);
 962
 963         }
 964
 965         spin_lock(&imp->imp_lock);
 966         if (imp->imp_state != LUSTRE_IMP_FULL)
 967                 GOTO(out, 0);
 968
 969         spin_unlock(&imp->imp_lock);
 970
 971         req = ptlrpc_prep_req(imp, LUSTRE_OBD_VERSION, rq_opc, 1, NULL, NULL);
 972         if (req) {
 973                 /* We are disconnecting, do not retry a failed DISCONNECT rpc if
 974                  * it fails.  We can get through the above with a down server
 975                  * if the client doesn't know the server is gone yet. */
 976                 req->rq_no_resend = 1;
 977                 req->rq_timeout = 5;
 978                 IMPORT_SET_STATE(imp, LUSTRE_IMP_CONNECTING);
 979                 req->rq_send_state =  LUSTRE_IMP_CONNECTING;
 980                 ptlrpc_req_set_repsize(req, 1, NULL);
 981                 rc = ptlrpc_queue_wait(req);
 982                 ptlrpc_req_finished(req);
 983         }
 984
 985         spin_lock(&imp->imp_lock);
 986 out:
 987         if (noclose)
 988                 IMPORT_SET_STATE_NOLOCK(imp, LUSTRE_IMP_DISCON);
 989         else
 990                 IMPORT_SET_STATE_NOLOCK(imp, LUSTRE_IMP_CLOSED);
 991         memset(&imp->imp_remote_handle, 0, sizeof(imp->imp_remote_handle));
 992         spin_unlock(&imp->imp_lock);
 993
 994         RETURN(rc);
 995 }
 996