lustre/ptlrpc/client.c

   1 /* -*- mode: c; c-basic-offset: 8; indent-tabs-mode: nil; -*-
   2  * vim:expandtab:shiftwidth=8:tabstop=8:
   3  *
   4  *  Copyright (C) 2002 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
  23 #define EXPORT_SYMTAB
  24
  25 #define DEBUG_SUBSYSTEM S_RPC
  26
  27 #include <linux/lustre_ha.h>
  28
  29 void ptlrpc_init_client(struct connmgr_obd *mgr, int req_portal,
  30                         int rep_portal, struct ptlrpc_client *cl)
  31 {
  32         memset(cl, 0, sizeof(*cl));
  33         cl->cli_ha_mgr = mgr;
  34         if (mgr)
  35                 connmgr_cli_manage(mgr, cl);
  36         cl->cli_obd = NULL;
  37         cl->cli_request_portal = req_portal;
  38         cl->cli_reply_portal = rep_portal;
  39         INIT_LIST_HEAD(&cl->cli_sending_head);
  40         INIT_LIST_HEAD(&cl->cli_sent_head);
  41         spin_lock_init(&cl->cli_lock);
  42         sema_init(&cl->cli_rpc_sem, 32);
  43 }
  44
  45 struct ptlrpc_connection *ptlrpc_uuid_to_connection(char *uuid)
  46 {
  47         struct ptlrpc_connection *c;
  48         struct lustre_peer peer;
  49         int err;
  50
  51         err = kportal_uuid_to_peer(uuid, &peer);
  52         if (err != 0) {
  53                 CERROR("cannot find peer %s!\n", uuid);
  54                 return NULL;
  55         }
  56
  57         c = ptlrpc_get_connection(&peer);
  58         if (c)
  59                 c->c_epoch++;
  60
  61         return c;
  62 }
  63
  64 struct ptlrpc_bulk_desc *ptlrpc_prep_bulk(struct ptlrpc_connection *conn)
  65 {
  66         struct ptlrpc_bulk_desc *bulk;
  67
  68         OBD_ALLOC(bulk, sizeof(*bulk));
  69         if (bulk != NULL) {
  70                 bulk->b_connection = ptlrpc_connection_addref(conn);
  71                 init_waitqueue_head(&bulk->b_waitq);
  72         }
  73
  74         return bulk;
  75 }
  76
  77 void ptlrpc_free_bulk(struct ptlrpc_bulk_desc *bulk)
  78 {
  79         if (bulk == NULL)
  80                 return;
  81
  82         ptlrpc_put_connection(bulk->b_connection);
  83
  84         OBD_FREE(bulk, sizeof(*bulk));
  85 }
  86
  87 struct ptlrpc_request *ptlrpc_prep_req(struct ptlrpc_client *cl,
  88                                        struct ptlrpc_connection *conn,
  89                                        int opcode, int count, int *lengths,
  90                                        char **bufs)
  91 {
  92         struct ptlrpc_request *request;
  93         int rc;
  94         ENTRY;
  95
  96         OBD_ALLOC(request, sizeof(*request));
  97         if (!request) {
  98                 CERROR("request allocation out of memory\n");
  99                 RETURN(NULL);
 100         }
 101
 102         rc = lustre_pack_msg(count, lengths, bufs,
 103                              &request->rq_reqlen, &request->rq_reqmsg);
 104         if (rc) {
 105                 CERROR("cannot pack request %d\n", rc);
 106                 RETURN(NULL);
 107         }
 108
 109         request->rq_time = CURRENT_TIME;
 110         request->rq_type = PTL_RPC_REQUEST;
 111         request->rq_connection = ptlrpc_connection_addref(conn);
 112
 113         request->rq_reqmsg->conn = (__u64)(unsigned long)conn->c_remote_conn;
 114         request->rq_reqmsg->token = conn->c_remote_token;
 115         request->rq_reqmsg->opc = HTON__u32(opcode);
 116         request->rq_reqmsg->type = HTON__u32(request->rq_type);
 117         INIT_LIST_HEAD(&request->rq_list);
 118
 119         spin_lock(&conn->c_lock);
 120         request->rq_reqmsg->xid = HTON__u32(++conn->c_xid_out);
 121         spin_unlock(&conn->c_lock);
 122
 123         request->rq_client = cl;
 124
 125         RETURN(request);
 126 }
 127
 128 void ptlrpc_free_req(struct ptlrpc_request *request)
 129 {
 130         if (request == NULL)
 131                 return;
 132
 133         if (request->rq_repmsg != NULL)
 134                 OBD_FREE(request->rq_repmsg, request->rq_replen);
 135
 136         if (request->rq_client) {
 137                 spin_lock(&request->rq_client->cli_lock);
 138                 list_del(&request->rq_list);
 139                 spin_unlock(&request->rq_client->cli_lock);
 140         }
 141
 142         ptlrpc_put_connection(request->rq_connection);
 143
 144         OBD_FREE(request, sizeof(*request));
 145 }
 146
 147 static int ptlrpc_check_reply(struct ptlrpc_request *req)
 148 {
 149         int rc = 0;
 150
 151         schedule_timeout(3 * HZ);  /* 3 second timeout */
 152         if (req->rq_repmsg != NULL) {
 153                 req->rq_flags = PTL_RPC_REPLY;
 154                 GOTO(out, rc = 1);
 155         }
 156
 157         if (CURRENT_TIME - req->rq_time >= 3) {
 158                 CERROR("-- REQ TIMEOUT --\n");
 159                 if (req->rq_client && req->rq_client->cli_ha_mgr)
 160                         connmgr_cli_fail(req->rq_client);
 161                 return 0;
 162         }
 163
 164         if (sigismember(&(current->pending.signal), SIGKILL) ||
 165             sigismember(&(current->pending.signal), SIGTERM) ||
 166             sigismember(&(current->pending.signal), SIGINT)) {
 167                 req->rq_flags = PTL_RPC_INTR;
 168                 GOTO(out, rc = 1);
 169         }
 170
 171  out:
 172         return rc;
 173 }
 174
 175 int ptlrpc_check_status(struct ptlrpc_request *req, int err)
 176 {
 177         ENTRY;
 178
 179         if (err != 0) {
 180                 CERROR("err is %d\n", err);
 181                 RETURN(err);
 182         }
 183
 184         if (req == NULL) {
 185                 CERROR("req == NULL\n");
 186                 RETURN(-ENOMEM);
 187         }
 188
 189         if (req->rq_repmsg == NULL) {
 190                 CERROR("req->rq_repmsg == NULL\n");
 191                 RETURN(-ENOMEM);
 192         }
 193
 194         if (req->rq_repmsg->status != 0) {
 195                 CERROR("req->rq_repmsg->status is %d\n",
 196                        req->rq_repmsg->status);
 197                 /* XXX: translate this error from net to host */
 198                 RETURN(req->rq_repmsg->status);
 199         }
 200
 201         RETURN(0);
 202 }
 203
 204 static void ptlrpc_cleanup_request_buf(struct ptlrpc_request *request)
 205 {
 206         OBD_FREE(request->rq_reqmsg, request->rq_reqlen);
 207         request->rq_reqmsg = NULL;
 208         request->rq_reqlen = 0;
 209 }
 210
 211 /* Abort this request and cleanup any resources associated with it. */
 212 static int ptlrpc_abort(struct ptlrpc_request *request)
 213 {
 214         /* First remove the ME for the reply; in theory, this means
 215          * that we can tear down the buffer safely. */
 216         PtlMEUnlink(request->rq_reply_me_h);
 217         OBD_FREE(request->rq_reply_md.start, request->rq_replen);
 218         request->rq_repmsg = NULL;
 219         request->rq_replen = 0;
 220         return 0;
 221 }
 222
 223 int ptlrpc_queue_wait(struct ptlrpc_request *req)
 224 {
 225         int rc = 0;
 226         ENTRY;
 227
 228         init_waitqueue_head(&req->rq_wait_for_rep);
 229
 230         rc = ptl_send_rpc(req);
 231         if (rc) {
 232                 CERROR("error %d, opcode %d\n", rc, req->rq_reqmsg->opc);
 233                 ptlrpc_cleanup_request_buf(req);
 234                 up(&req->rq_client->cli_rpc_sem);
 235                 RETURN(-rc);
 236         }
 237
 238         CDEBUG(D_OTHER, "-- sleeping\n");
 239         wait_event_interruptible(req->rq_wait_for_rep, ptlrpc_check_reply(req));
 240         CDEBUG(D_OTHER, "-- done\n");
 241         ptlrpc_cleanup_request_buf(req);
 242         up(&req->rq_client->cli_rpc_sem);
 243         if (req->rq_flags == PTL_RPC_INTR) {
 244                 /* Clean up the dangling reply buffers */
 245                 ptlrpc_abort(req);
 246                 GOTO(out, rc = -EINTR);
 247         }
 248
 249         if (req->rq_flags != PTL_RPC_REPLY) {
 250                 CERROR("Unknown reason for wakeup\n");
 251                 /* XXX Phil - I end up here when I kill obdctl */
 252                 ptlrpc_abort(req);
 253                 GOTO(out, rc = -EINTR);
 254         }
 255
 256         rc = lustre_unpack_msg(req->rq_repmsg, req->rq_replen);
 257         if (rc) {
 258                 CERROR("unpack_rep failed: %d\n", rc);
 259                 GOTO(out, rc);
 260         }
 261         CDEBUG(D_NET, "got rep %d\n", req->rq_repmsg->xid);
 262
 263         if (req->rq_repmsg->status == 0)
 264                 CDEBUG(D_NET, "--> buf %p len %d status %d\n", req->rq_repmsg,
 265                        req->rq_replen, req->rq_repmsg->status);
 266
 267         EXIT;
 268  out:
 269         return rc;
 270 }