// SPDX-License-Identifier: LGPL-2.1
/*
 *
 *   Copyright (C) International Business Machines  Corp., 2002,2008
 *   Author(s): Steve French (sfrench@us.ibm.com)
 *   Jeremy Allison (jra@samba.org) 2006.
 *
 */

#include <linux/fs.h>
#include <linux/list.h>
#include <linux/gfp.h>
#include <linux/wait.h>
#include <linux/net.h>
#include <linux/delay.h>
#include <linux/freezer.h>
#include <linux/tcp.h>
#include <linux/bvec.h>
#include <linux/highmem.h>
#include <linux/uaccess.h>
#include <asm/processor.h>
#include <linux/mempool.h>
#include <linux/sched/signal.h>
#include "cifspdu.h"
#include "cifsglob.h"
#include "cifsproto.h"
#include "cifs_debug.h"
#include "smb2proto.h"
#include "smbdirect.h"

/* Max number of iovectors we can use off the stack when sending requests. */
#define CIFS_MAX_IOV_SIZE 8

void
cifs_wake_up_task(struct mid_q_entry *mid)
{
        wake_up_process(mid->callback_data);
}

struct mid_q_entry *
AllocMidQEntry(const struct smb_hdr *smb_buffer, struct TCP_Server_Info *server)
{
        struct mid_q_entry *temp;

        if (server == NULL) {
                cifs_dbg(VFS, "Null TCP session in AllocMidQEntry\n");
                return NULL;
        }

        temp = mempool_alloc(cifs_mid_poolp, GFP_NOFS);
        memset(temp, 0, sizeof(struct mid_q_entry));
        kref_init(&temp->refcount);
        temp->mid = get_mid(smb_buffer);
        temp->pid = current->pid;
        temp->command = cpu_to_le16(smb_buffer->Command);
        cifs_dbg(FYI, "For smb_command %d\n", smb_buffer->Command);
        /*      do_gettimeofday(&temp->when_sent);*/ /* easier to use jiffies */
        /* when mid allocated can be before when sent */
        temp->when_alloc = jiffies;
        temp->server = server;

        /*
         * The default is for the mid to be synchronous, so the
         * default callback just wakes up the current task.
         */
        get_task_struct(current);
        temp->creator = current;
        temp->callback = cifs_wake_up_task;
        temp->callback_data = current;

        atomic_inc(&midCount);
        temp->mid_state = MID_REQUEST_ALLOCATED;
        return temp;
}

static void _cifs_mid_q_entry_release(struct kref *refcount)
{
        struct mid_q_entry *midEntry =
                        container_of(refcount, struct mid_q_entry, refcount);
#ifdef CONFIG_CIFS_STATS2
        __le16 command = midEntry->server->vals->lock_cmd;
        __u16 smb_cmd = le16_to_cpu(midEntry->command);
        unsigned long now;
        unsigned long roundtrip_time;
#endif
        struct TCP_Server_Info *server = midEntry->server;

        if (midEntry->resp_buf && (midEntry->mid_flags & MID_WAIT_CANCELLED) &&
            midEntry->mid_state == MID_RESPONSE_RECEIVED &&
            server->ops->handle_cancelled_mid)
                server->ops->handle_cancelled_mid(midEntry, server);

        midEntry->mid_state = MID_FREE;
        atomic_dec(&midCount);
        if (midEntry->large_buf)
                cifs_buf_release(midEntry->resp_buf);
        else
                cifs_small_buf_release(midEntry->resp_buf);
#ifdef CONFIG_CIFS_STATS2
        now = jiffies;
        if (now < midEntry->when_alloc)
                cifs_server_dbg(VFS, "Invalid mid allocation time\n");
        roundtrip_time = now - midEntry->when_alloc;

        if (smb_cmd < NUMBER_OF_SMB2_COMMANDS) {
                if (atomic_read(&server->num_cmds[smb_cmd]) == 0) {
                        server->slowest_cmd[smb_cmd] = roundtrip_time;
                        server->fastest_cmd[smb_cmd] = roundtrip_time;
                } else {
                        if (server->slowest_cmd[smb_cmd] < roundtrip_time)
                                server->slowest_cmd[smb_cmd] = roundtrip_time;
                        else if (server->fastest_cmd[smb_cmd] > roundtrip_time)
                                server->fastest_cmd[smb_cmd] = roundtrip_time;
                }
                cifs_stats_inc(&server->num_cmds[smb_cmd]);
                server->time_per_cmd[smb_cmd] += roundtrip_time;
        }
        /*
         * commands taking longer than one second (default) can be indications
         * that something is wrong, unless it is quite a slow link or a very
         * busy server. Note that this calc is unlikely or impossible to wrap
         * as long as slow_rsp_threshold is not set way above recommended max
         * value (32767 ie 9 hours) and is generally harmless even if wrong
         * since only affects debug counters - so leaving the calc as simple
         * comparison rather than doing multiple conversions and overflow
         * checks
         */
        if ((slow_rsp_threshold != 0) &&
            time_after(now, midEntry->when_alloc + (slow_rsp_threshold * HZ)) &&
            (midEntry->command != command)) {
                /*
                 * smb2slowcmd[NUMBER_OF_SMB2_COMMANDS] counts by command
                 * NB: le16_to_cpu returns unsigned so can not be negative below
                 */
                if (smb_cmd < NUMBER_OF_SMB2_COMMANDS)
                        cifs_stats_inc(&server->smb2slowcmd[smb_cmd]);

                trace_smb3_slow_rsp(smb_cmd, midEntry->mid, midEntry->pid,
                               midEntry->when_sent, midEntry->when_received);
                if (cifsFYI & CIFS_TIMER) {
                        pr_debug("slow rsp: cmd %d mid %llu",
                                 midEntry->command, midEntry->mid);
                        cifs_info("A: 0x%lx S: 0x%lx R: 0x%lx\n",
                                  now - midEntry->when_alloc,
                                  now - midEntry->when_sent,
                                  now - midEntry->when_received);
                }
        }
#endif
        put_task_struct(midEntry->creator);

        mempool_free(midEntry, cifs_mid_poolp);
}

void cifs_mid_q_entry_release(struct mid_q_entry *midEntry)
{
        spin_lock(&GlobalMid_Lock);
        kref_put(&midEntry->refcount, _cifs_mid_q_entry_release);
        spin_unlock(&GlobalMid_Lock);
}

void DeleteMidQEntry(struct mid_q_entry *midEntry)
{
        cifs_mid_q_entry_release(midEntry);
}

void
cifs_delete_mid(struct mid_q_entry *mid)
{
        spin_lock(&GlobalMid_Lock);
        if (!(mid->mid_flags & MID_DELETED)) {
                list_del_init(&mid->qhead);
                mid->mid_flags |= MID_DELETED;
        }
        spin_unlock(&GlobalMid_Lock);

        DeleteMidQEntry(mid);
}

/*
 * smb_send_kvec - send an array of kvecs to the server
 * @server:     Server to send the data to
 * @smb_msg:    Message to send
 * @sent:       amount of data sent on socket is stored here
 *
 * Our basic "send data to server" function. Should be called with srv_mutex
 * held. The caller is responsible for handling the results.
 */
static int
smb_send_kvec(struct TCP_Server_Info *server, struct msghdr *smb_msg,
              size_t *sent)
{
        int rc = 0;
        int retries = 0;
        struct socket *ssocket = server->ssocket;

        *sent = 0;

        smb_msg->msg_name = (struct sockaddr *) &server->dstaddr;
        smb_msg->msg_namelen = sizeof(struct sockaddr);
        smb_msg->msg_control = NULL;
        smb_msg->msg_controllen = 0;
        if (server->noblocksnd)
                smb_msg->msg_flags = MSG_DONTWAIT + MSG_NOSIGNAL;
        else
                smb_msg->msg_flags = MSG_NOSIGNAL;

        while (msg_data_left(smb_msg)) {
                /*
                 * If blocking send, we try 3 times, since each can block
                 * for 5 seconds. For nonblocking  we have to try more
                 * but wait increasing amounts of time allowing time for
                 * socket to clear.  The overall time we wait in either
                 * case to send on the socket is about 15 seconds.
                 * Similarly we wait for 15 seconds for a response from
                 * the server in SendReceive[2] for the server to send
                 * a response back for most types of requests (except
                 * SMB Write past end of file which can be slow, and
                 * blocking lock operations). NFS waits slightly longer
                 * than CIFS, but this can make it take longer for
                 * nonresponsive servers to be detected and 15 seconds
                 * is more than enough time for modern networks to
                 * send a packet.  In most cases if we fail to send
                 * after the retries we will kill the socket and
                 * reconnect which may clear the network problem.
                 */
                rc = sock_sendmsg(ssocket, smb_msg);
                if (rc == -EAGAIN) {
                        retries++;
                        if (retries >= 14 ||
                            (!server->noblocksnd && (retries > 2))) {
                                cifs_server_dbg(VFS, "sends on sock %p stuck for 15 seconds\n",
                                         ssocket);
                                return -EAGAIN;
                        }
                        msleep(1 << retries);
                        continue;
                }

                if (rc < 0)
                        return rc;

                if (rc == 0) {
                        /* should never happen, letting socket clear before
                           retrying is our only obvious option here */
                        cifs_server_dbg(VFS, "tcp sent no data\n");
                        msleep(500);
                        continue;
                }

                /* send was at least partially successful */
                *sent += rc;
                retries = 0; /* in case we get ENOSPC on the next send */
        }
        return 0;
}

unsigned long
smb_rqst_len(struct TCP_Server_Info *server, struct smb_rqst *rqst)
{
        unsigned int i;
        struct kvec *iov;
        int nvec;
        unsigned long buflen = 0;

        if (server->vals->header_preamble_size == 0 &&
            rqst->rq_nvec >= 2 && rqst->rq_iov[0].iov_len == 4) {
                iov = &rqst->rq_iov[1];
                nvec = rqst->rq_nvec - 1;
        } else {
                iov = rqst->rq_iov;
                nvec = rqst->rq_nvec;
        }

        /* total up iov array first */
        for (i = 0; i < nvec; i++)
                buflen += iov[i].iov_len;

        /*
         * Add in the page array if there is one. The caller needs to make
         * sure rq_offset and rq_tailsz are set correctly. If a buffer of
         * multiple pages ends at page boundary, rq_tailsz needs to be set to
         * PAGE_SIZE.
         */
        if (rqst->rq_npages) {
                if (rqst->rq_npages == 1)
                        buflen += rqst->rq_tailsz;
                else {
                        /*
                         * If there is more than one page, calculate the
                         * buffer length based on rq_offset and rq_tailsz
                         */
                        buflen += rqst->rq_pagesz * (rqst->rq_npages - 1) -
                                        rqst->rq_offset;
                        buflen += rqst->rq_tailsz;
                }
        }

        return buflen;
}

static int
__smb_send_rqst(struct TCP_Server_Info *server, int num_rqst,
                struct smb_rqst *rqst)
{
        int rc = 0;
        struct kvec *iov;
        int n_vec;
        unsigned int send_length = 0;
        unsigned int i, j;
        sigset_t mask, oldmask;
        size_t total_len = 0, sent, size;
        struct socket *ssocket = server->ssocket;
        struct msghdr smb_msg;
        __be32 rfc1002_marker;

        if (cifs_rdma_enabled(server)) {
                /* return -EAGAIN when connecting or reconnecting */
                rc = -EAGAIN;
                if (server->smbd_conn)
                        rc = smbd_send(server, num_rqst, rqst);
                goto smbd_done;
        }

        if (ssocket == NULL)
                return -EAGAIN;

        if (fatal_signal_pending(current)) {
                cifs_dbg(FYI, "signal pending before send request\n");
                return -ERESTARTSYS;
        }

        /* cork the socket */
        tcp_sock_set_cork(ssocket->sk, true);

        for (j = 0; j < num_rqst; j++)
                send_length += smb_rqst_len(server, &rqst[j]);
        rfc1002_marker = cpu_to_be32(send_length);

        /*
         * We should not allow signals to interrupt the network send because
         * any partial send will cause session reconnects thus increasing
         * latency of system calls and overload a server with unnecessary
         * requests.
         */

        sigfillset(&mask);
        sigprocmask(SIG_BLOCK, &mask, &oldmask);

        /* Generate a rfc1002 marker for SMB2+ */
        if (server->vals->header_preamble_size == 0) {
                struct kvec hiov = {
                        .iov_base = &rfc1002_marker,
                        .iov_len  = 4
                };
                iov_iter_kvec(&smb_msg.msg_iter, WRITE, &hiov, 1, 4);
                rc = smb_send_kvec(server, &smb_msg, &sent);
                if (rc < 0)
                        goto unmask;

                total_len += sent;
                send_length += 4;
        }

        cifs_dbg(FYI, "Sending smb: smb_len=%u\n", send_length);

        for (j = 0; j < num_rqst; j++) {
                iov = rqst[j].rq_iov;
                n_vec = rqst[j].rq_nvec;

                size = 0;
                for (i = 0; i < n_vec; i++) {
                        dump_smb(iov[i].iov_base, iov[i].iov_len);
                        size += iov[i].iov_len;
                }

                iov_iter_kvec(&smb_msg.msg_iter, WRITE, iov, n_vec, size);

                rc = smb_send_kvec(server, &smb_msg, &sent);
                if (rc < 0)
                        goto unmask;

                total_len += sent;

                /* now walk the page array and send each page in it */
                for (i = 0; i < rqst[j].rq_npages; i++) {
                        struct bio_vec bvec;

                        bvec.bv_page = rqst[j].rq_pages[i];
                        rqst_page_get_length(&rqst[j], i, &bvec.bv_len,
                                             &bvec.bv_offset);

                        iov_iter_bvec(&smb_msg.msg_iter, WRITE,
                                      &bvec, 1, bvec.bv_len);
                        rc = smb_send_kvec(server, &smb_msg, &sent);
                        if (rc < 0)
                                break;

                        total_len += sent;
                }
        }

unmask:
        sigprocmask(SIG_SETMASK, &oldmask, NULL);

        /*
         * If signal is pending but we have already sent the whole packet to
         * the server we need to return success status to allow a corresponding
         * mid entry to be kept in the pending requests queue thus allowing
         * to handle responses from the server by the client.
         *
         * If only part of the packet has been sent there is no need to hide
         * interrupt because the session will be reconnected anyway, so there
         * won't be any response from the server to handle.
         */

        if (signal_pending(current) && (total_len != send_length)) {
                cifs_dbg(FYI, "signal is pending after attempt to send\n");
                rc = -ERESTARTSYS;
        }

        /* uncork it */
        tcp_sock_set_cork(ssocket->sk, false);

        if ((total_len > 0) && (total_len != send_length)) {
                cifs_dbg(FYI, "partial send (wanted=%u sent=%zu): terminating session\n",
                         send_length, total_len);
                /*
                 * If we have only sent part of an SMB then the next SMB could
                 * be taken as the remainder of this one. We need to kill the
                 * socket so the server throws away the partial SMB
                 */
                spin_lock(&GlobalMid_Lock);
                server->tcpStatus = CifsNeedReconnect;
                spin_unlock(&GlobalMid_Lock);
                trace_smb3_partial_send_reconnect(server->CurrentMid,
                                                  server->conn_id, server->hostname);
        }
smbd_done:
        if (rc < 0 && rc != -EINTR)
                cifs_server_dbg(VFS, "Error %d sending data on socket to server\n",
                         rc);
        else if (rc > 0)
                rc = 0;

        return rc;
}

static int
smb_send_rqst(struct TCP_Server_Info *server, int num_rqst,
              struct smb_rqst *rqst, int flags)
{
        struct kvec iov;
        struct smb2_transform_hdr *tr_hdr;
        struct smb_rqst cur_rqst[MAX_COMPOUND];
        int rc;

        if (!(flags & CIFS_TRANSFORM_REQ))
                return __smb_send_rqst(server, num_rqst, rqst);

        if (num_rqst > MAX_COMPOUND - 1)
                return -ENOMEM;

        if (!server->ops->init_transform_rq) {
                cifs_server_dbg(VFS, "Encryption requested but transform callback is missing\n");
                return -EIO;
        }

        tr_hdr = kmalloc(sizeof(*tr_hdr), GFP_NOFS);
        if (!tr_hdr)
                return -ENOMEM;

        memset(&cur_rqst[0], 0, sizeof(cur_rqst));
        memset(&iov, 0, sizeof(iov));
        memset(tr_hdr, 0, sizeof(*tr_hdr));

        iov.iov_base = tr_hdr;
        iov.iov_len = sizeof(*tr_hdr);
        cur_rqst[0].rq_iov = &iov;
        cur_rqst[0].rq_nvec = 1;

        rc = server->ops->init_transform_rq(server, num_rqst + 1,
                                            &cur_rqst[0], rqst);
        if (rc)
                goto out;

        rc = __smb_send_rqst(server, num_rqst + 1, &cur_rqst[0]);
        smb3_free_compound_rqst(num_rqst, &cur_rqst[1]);
out:
        kfree(tr_hdr);
        return rc;
}

int
smb_send(struct TCP_Server_Info *server, struct smb_hdr *smb_buffer,
         unsigned int smb_buf_length)
{
        struct kvec iov[2];
        struct smb_rqst rqst = { .rq_iov = iov,
                                 .rq_nvec = 2 };

        iov[0].iov_base = smb_buffer;
        iov[0].iov_len = 4;
        iov[1].iov_base = (char *)smb_buffer + 4;
        iov[1].iov_len = smb_buf_length;

        return __smb_send_rqst(server, 1, &rqst);
}

static int
wait_for_free_credits(struct TCP_Server_Info *server, const int num_credits,
                      const int timeout, const int flags,
                      unsigned int *instance)
{
        long rc;
        int *credits;
        int optype;
        long int t;
        int scredits, in_flight;

        if (timeout < 0)
                t = MAX_JIFFY_OFFSET;
        else
                t = msecs_to_jiffies(timeout);

        optype = flags & CIFS_OP_MASK;

        *instance = 0;

        credits = server->ops->get_credits_field(server, optype);
        /* Since an echo is already inflight, no need to wait to send another */
        if (*credits <= 0 && optype == CIFS_ECHO_OP)
                return -EAGAIN;

        spin_lock(&server->req_lock);
        if ((flags & CIFS_TIMEOUT_MASK) == CIFS_NON_BLOCKING) {
                /* oplock breaks must not be held up */
                server->in_flight++;
                if (server->in_flight > server->max_in_flight)
                        server->max_in_flight = server->in_flight;
                *credits -= 1;
                *instance = server->reconnect_instance;
                scredits = *credits;
                in_flight = server->in_flight;
                spin_unlock(&server->req_lock);

                trace_smb3_add_credits(server->CurrentMid,
                                server->conn_id, server->hostname, scredits, -1, in_flight);
                cifs_dbg(FYI, "%s: remove %u credits total=%d\n",
                                __func__, 1, scredits);

                return 0;
        }

        while (1) {
                if (*credits < num_credits) {
                        scredits = *credits;
                        spin_unlock(&server->req_lock);

                        cifs_num_waiters_inc(server);
                        rc = wait_event_killable_timeout(server->request_q,
                                has_credits(server, credits, num_credits), t);
                        cifs_num_waiters_dec(server);
                        if (!rc) {
                                spin_lock(&server->req_lock);
                                scredits = *credits;
                                in_flight = server->in_flight;
                                spin_unlock(&server->req_lock);

                                trace_smb3_credit_timeout(server->CurrentMid,
                                                server->conn_id, server->hostname, scredits,
                                                num_credits, in_flight);
                                cifs_server_dbg(VFS, "wait timed out after %d ms\n",
                                                timeout);
                                return -EBUSY;
                        }
                        if (rc == -ERESTARTSYS)
                                return -ERESTARTSYS;
                        spin_lock(&server->req_lock);
                } else {
                        if (server->tcpStatus == CifsExiting) {
                                spin_unlock(&server->req_lock);
                                return -ENOENT;
                        }

                        /*
                         * For normal commands, reserve the last MAX_COMPOUND
                         * credits to compound requests.
                         * Otherwise these compounds could be permanently
                         * starved for credits by single-credit requests.
                         *
                         * To prevent spinning CPU, block this thread until
                         * there are >MAX_COMPOUND credits available.
                         * But only do this is we already have a lot of
                         * credits in flight to avoid triggering this check
                         * for servers that are slow to hand out credits on
                         * new sessions.
                         */
                        if (!optype && num_credits == 1 &&
                            server->in_flight > 2 * MAX_COMPOUND &&
                            *credits <= MAX_COMPOUND) {
                                spin_unlock(&server->req_lock);

                                cifs_num_waiters_inc(server);
                                rc = wait_event_killable_timeout(
                                        server->request_q,
                                        has_credits(server, credits,
                                                    MAX_COMPOUND + 1),
                                        t);
                                cifs_num_waiters_dec(server);
                                if (!rc) {
                                        spin_lock(&server->req_lock);
                                        scredits = *credits;
                                        in_flight = server->in_flight;
                                        spin_unlock(&server->req_lock);

                                        trace_smb3_credit_timeout(
                                                        server->CurrentMid,
                                                        server->conn_id, server->hostname,
                                                        scredits, num_credits, in_flight);
                                        cifs_server_dbg(VFS, "wait timed out after %d ms\n",
                                                        timeout);
                                        return -EBUSY;
                                }
                                if (rc == -ERESTARTSYS)
                                        return -ERESTARTSYS;
                                spin_lock(&server->req_lock);
                                continue;
                        }

                        /*
                         * Can not count locking commands against total
                         * as they are allowed to block on server.
                         */

                        /* update # of requests on the wire to server */
                        if ((flags & CIFS_TIMEOUT_MASK) != CIFS_BLOCKING_OP) {
                                *credits -= num_credits;
                                server->in_flight += num_credits;
                                if (server->in_flight > server->max_in_flight)
                                        server->max_in_flight = server->in_flight;
                                *instance = server->reconnect_instance;
                        }
                        scredits = *credits;
                        in_flight = server->in_flight;
                        spin_unlock(&server->req_lock);

                        trace_smb3_add_credits(server->CurrentMid,
                                        server->conn_id, server->hostname, scredits,
                                        -(num_credits), in_flight);
                        cifs_dbg(FYI, "%s: remove %u credits total=%d\n",
                                        __func__, num_credits, scredits);
                        break;
                }
        }
        return 0;
}

static int
wait_for_free_request(struct TCP_Server_Info *server, const int flags,
                      unsigned int *instance)
{
        return wait_for_free_credits(server, 1, -1, flags,
                                     instance);
}

static int
wait_for_compound_request(struct TCP_Server_Info *server, int num,
                          const int flags, unsigned int *instance)
{
        int *credits;
        int scredits, in_flight;

        credits = server->ops->get_credits_field(server, flags & CIFS_OP_MASK);

        spin_lock(&server->req_lock);
        scredits = *credits;
        in_flight = server->in_flight;

        if (*credits < num) {
                /*
                 * If the server is tight on resources or just gives us less
                 * credits for other reasons (e.g. requests are coming out of
                 * order and the server delays granting more credits until it
                 * processes a missing mid) and we exhausted most available
                 * credits there may be situations when we try to send
                 * a compound request but we don't have enough credits. At this
                 * point the client needs to decide if it should wait for
                 * additional credits or fail the request. If at least one
                 * request is in flight there is a high probability that the
                 * server will return enough credits to satisfy this compound
                 * request.
                 *
                 * Return immediately if no requests in flight since we will be
                 * stuck on waiting for credits.
                 */
                if (server->in_flight == 0) {
                        spin_unlock(&server->req_lock);
                        trace_smb3_insufficient_credits(server->CurrentMid,
                                        server->conn_id, server->hostname, scredits,
                                        num, in_flight);
                        cifs_dbg(FYI, "%s: %d requests in flight, needed %d total=%d\n",
                                        __func__, in_flight, num, scredits);
                        return -EDEADLK;
                }
        }
        spin_unlock(&server->req_lock);

        return wait_for_free_credits(server, num, 60000, flags,
                                     instance);
}

int
cifs_wait_mtu_credits(struct TCP_Server_Info *server, unsigned int size,
                      unsigned int *num, struct cifs_credits *credits)
{
        *num = size;
        credits->value = 0;
        credits->instance = server->reconnect_instance;
        return 0;
}

static int allocate_mid(struct cifs_ses *ses, struct smb_hdr *in_buf,
                        struct mid_q_entry **ppmidQ)
{
        if (ses->server->tcpStatus == CifsExiting) {
                return -ENOENT;
        }

        if (ses->server->tcpStatus == CifsNeedReconnect) {
                cifs_dbg(FYI, "tcp session dead - return to caller to retry\n");
                return -EAGAIN;
        }

        if (ses->status == CifsNew) {
                if ((in_buf->Command != SMB_COM_SESSION_SETUP_ANDX) &&
                        (in_buf->Command != SMB_COM_NEGOTIATE))
                        return -EAGAIN;
                /* else ok - we are setting up session */
        }

        if (ses->status == CifsExiting) {
                /* check if SMB session is bad because we are setting it up */
                if (in_buf->Command != SMB_COM_LOGOFF_ANDX)
                        return -EAGAIN;
                /* else ok - we are shutting down session */
        }

        *ppmidQ = AllocMidQEntry(in_buf, ses->server);
        if (*ppmidQ == NULL)
                return -ENOMEM;
        spin_lock(&GlobalMid_Lock);
        list_add_tail(&(*ppmidQ)->qhead, &ses->server->pending_mid_q);
        spin_unlock(&GlobalMid_Lock);
        return 0;
}

static int
wait_for_response(struct TCP_Server_Info *server, struct mid_q_entry *midQ)
{
        int error;

        error = wait_event_freezekillable_unsafe(server->response_q,
                                    midQ->mid_state != MID_REQUEST_SUBMITTED);
        if (error < 0)
                return -ERESTARTSYS;

        return 0;
}

struct mid_q_entry *
cifs_setup_async_request(struct TCP_Server_Info *server, struct smb_rqst *rqst)
{
        int rc;
        struct smb_hdr *hdr = (struct smb_hdr *)rqst->rq_iov[0].iov_base;
        struct mid_q_entry *mid;

        if (rqst->rq_iov[0].iov_len != 4 ||
            rqst->rq_iov[0].iov_base + 4 != rqst->rq_iov[1].iov_base)
                return ERR_PTR(-EIO);

        /* enable signing if server requires it */
        if (server->sign)
                hdr->Flags2 |= SMBFLG2_SECURITY_SIGNATURE;

        mid = AllocMidQEntry(hdr, server);
        if (mid == NULL)
                return ERR_PTR(-ENOMEM);

        rc = cifs_sign_rqst(rqst, server, &mid->sequence_number);
        if (rc) {
                DeleteMidQEntry(mid);
                return ERR_PTR(rc);
        }

        return mid;
}

/*
 * Send a SMB request and set the callback function in the mid to handle
 * the result. Caller is responsible for dealing with timeouts.
 */
int
cifs_call_async(struct TCP_Server_Info *server, struct smb_rqst *rqst,
                mid_receive_t *receive, mid_callback_t *callback,
                mid_handle_t *handle, void *cbdata, const int flags,
                const struct cifs_credits *exist_credits)
{
        int rc;
        struct mid_q_entry *mid;
        struct cifs_credits credits = { .value = 0, .instance = 0 };
        unsigned int instance;
        int optype;

        optype = flags & CIFS_OP_MASK;

        if ((flags & CIFS_HAS_CREDITS) == 0) {
                rc = wait_for_free_request(server, flags, &instance);
                if (rc)
                        return rc;
                credits.value = 1;
                credits.instance = instance;
        } else
                instance = exist_credits->instance;

        mutex_lock(&server->srv_mutex);

        /*
         * We can't use credits obtained from the previous session to send this
         * request. Check if there were reconnects after we obtained credits and
         * return -EAGAIN in such cases to let callers handle it.
         */
        if (instance != server->reconnect_instance) {
                mutex_unlock(&server->srv_mutex);
                add_credits_and_wake_if(server, &credits, optype);
                return -EAGAIN;
        }

        mid = server->ops->setup_async_request(server, rqst);
        if (IS_ERR(mid)) {
                mutex_unlock(&server->srv_mutex);
                add_credits_and_wake_if(server, &credits, optype);
                return PTR_ERR(mid);
        }

        mid->receive = receive;
        mid->callback = callback;
        mid->callback_data = cbdata;
        mid->handle = handle;
        mid->mid_state = MID_REQUEST_SUBMITTED;

        /* put it on the pending_mid_q */
        spin_lock(&GlobalMid_Lock);
        list_add_tail(&mid->qhead, &server->pending_mid_q);
        spin_unlock(&GlobalMid_Lock);

        /*
         * Need to store the time in mid before calling I/O. For call_async,
         * I/O response may come back and free the mid entry on another thread.
         */
        cifs_save_when_sent(mid);
        cifs_in_send_inc(server);
        rc = smb_send_rqst(server, 1, rqst, flags);
        cifs_in_send_dec(server);

        if (rc < 0) {
                revert_current_mid(server, mid->credits);
                server->sequence_number -= 2;
                cifs_delete_mid(mid);
        }

        mutex_unlock(&server->srv_mutex);

        if (rc == 0)
                return 0;

        add_credits_and_wake_if(server, &credits, optype);
        return rc;
}

/*
 *
 * Send an SMB Request.  No response info (other than return code)
 * needs to be parsed.
 *
 * flags indicate the type of request buffer and how long to wait
 * and whether to log NT STATUS code (error) before mapping it to POSIX error
 *
 */
int
SendReceiveNoRsp(const unsigned int xid, struct cifs_ses *ses,
                 char *in_buf, int flags)
{
        int rc;
        struct kvec iov[1];
        struct kvec rsp_iov;
        int resp_buf_type;

        iov[0].iov_base = in_buf;
        iov[0].iov_len = get_rfc1002_length(in_buf) + 4;
        flags |= CIFS_NO_RSP_BUF;
        rc = SendReceive2(xid, ses, iov, 1, &resp_buf_type, flags, &rsp_iov);
        cifs_dbg(NOISY, "SendRcvNoRsp flags %d rc %d\n", flags, rc);

        return rc;
}

static int
cifs_sync_mid_result(struct mid_q_entry *mid, struct TCP_Server_Info *server)
{
        int rc = 0;

        cifs_dbg(FYI, "%s: cmd=%d mid=%llu state=%d\n",
                 __func__, le16_to_cpu(mid->command), mid->mid, mid->mid_state);

        spin_lock(&GlobalMid_Lock);
        switch (mid->mid_state) {
        case MID_RESPONSE_RECEIVED:
                spin_unlock(&GlobalMid_Lock);
                return rc;
        case MID_RETRY_NEEDED:
                rc = -EAGAIN;
                break;
        case MID_RESPONSE_MALFORMED:
                rc = -EIO;
                break;
        case MID_SHUTDOWN:
                rc = -EHOSTDOWN;
                break;
        default:
                if (!(mid->mid_flags & MID_DELETED)) {
                        list_del_init(&mid->qhead);
                        mid->mid_flags |= MID_DELETED;
                }
                cifs_server_dbg(VFS, "%s: invalid mid state mid=%llu state=%d\n",
                         __func__, mid->mid, mid->mid_state);
                rc = -EIO;
        }
        spin_unlock(&GlobalMid_Lock);

        DeleteMidQEntry(mid);
        return rc;
}

static inline int
send_cancel(struct TCP_Server_Info *server, struct smb_rqst *rqst,
            struct mid_q_entry *mid)
{
        return server->ops->send_cancel ?
                                server->ops->send_cancel(server, rqst, mid) : 0;
}

int
cifs_check_receive(struct mid_q_entry *mid, struct TCP_Server_Info *server,
                   bool log_error)
{
        unsigned int len = get_rfc1002_length(mid->resp_buf) + 4;

        dump_smb(mid->resp_buf, min_t(u32, 92, len));

        /* convert the length into a more usable form */
        if (server->sign) {
                struct kvec iov[2];
                int rc = 0;
                struct smb_rqst rqst = { .rq_iov = iov,
                                         .rq_nvec = 2 };

                iov[0].iov_base = mid->resp_buf;
                iov[0].iov_len = 4;
                iov[1].iov_base = (char *)mid->resp_buf + 4;
                iov[1].iov_len = len - 4;
                /* FIXME: add code to kill session */
                rc = cifs_verify_signature(&rqst, server,
                                           mid->sequence_number);
                if (rc)
                        cifs_server_dbg(VFS, "SMB signature verification returned error = %d\n",
                                 rc);
        }

        /* BB special case reconnect tid and uid here? */
        return map_and_check_smb_error(mid, log_error);
}

struct mid_q_entry *
cifs_setup_request(struct cifs_ses *ses, struct TCP_Server_Info *ignored,
                   struct smb_rqst *rqst)
{
        int rc;
        struct smb_hdr *hdr = (struct smb_hdr *)rqst->rq_iov[0].iov_base;
        struct mid_q_entry *mid;

        if (rqst->rq_iov[0].iov_len != 4 ||
            rqst->rq_iov[0].iov_base + 4 != rqst->rq_iov[1].iov_base)
                return ERR_PTR(-EIO);

        rc = allocate_mid(ses, hdr, &mid);
        if (rc)
                return ERR_PTR(rc);
        rc = cifs_sign_rqst(rqst, ses->server, &mid->sequence_number);
        if (rc) {

                cifs_delete_mid(mid);
                return ERR_PTR(rc);
        }
        return mid;
}

static void
cifs_compound_callback(struct mid_q_entry *mid)
{
        struct TCP_Server_Info *server = mid->server;
        struct cifs_credits credits;

        credits.value = server->ops->get_credits(mid);
        credits.instance = server->reconnect_instance;

        add_credits(server, &credits, mid->optype);
}

static void
cifs_compound_last_callback(struct mid_q_entry *mid)
{
        cifs_compound_callback(mid);
        cifs_wake_up_task(mid);
}

static void
cifs_cancelled_callback(struct mid_q_entry *mid)
{
        cifs_compound_callback(mid);
        DeleteMidQEntry(mid);
}

/*
 * Return a channel (master if none) of @ses that can be used to send
 * regular requests.
 *
 * If we are currently binding a new channel (negprot/sess.setup),
 * return the new incomplete channel.
 */
struct TCP_Server_Info *cifs_pick_channel(struct cifs_ses *ses)
{
        uint index = 0;

        if (!ses)
                return NULL;

        if (!ses->binding) {
                /* round robin */
                if (ses->chan_count > 1) {
                        index = (uint)atomic_inc_return(&ses->chan_seq);
                        index %= ses->chan_count;
                }
                return ses->chans[index].server;
        } else {
                return cifs_ses_server(ses);
        }
}

int
compound_send_recv(const unsigned int xid, struct cifs_ses *ses,
                   struct TCP_Server_Info *server,
                   const int flags, const int num_rqst, struct smb_rqst *rqst,
                   int *resp_buf_type, struct kvec *resp_iov)
{
        int i, j, optype, rc = 0;
        struct mid_q_entry *midQ[MAX_COMPOUND];
        bool cancelled_mid[MAX_COMPOUND] = {false};
        struct cifs_credits credits[MAX_COMPOUND] = {
                { .value = 0, .instance = 0 }
        };
        unsigned int instance;
        char *buf;

        optype = flags & CIFS_OP_MASK;

        for (i = 0; i < num_rqst; i++)
                resp_buf_type[i] = CIFS_NO_BUFFER;  /* no response buf yet */

        if (!ses || !ses->server || !server) {
                cifs_dbg(VFS, "Null session\n");
                return -EIO;
        }

        if (server->tcpStatus == CifsExiting)
                return -ENOENT;

        /*
         * Wait for all the requests to become available.
         * This approach still leaves the possibility to be stuck waiting for
         * credits if the server doesn't grant credits to the outstanding
         * requests and if the client is completely idle, not generating any
         * other requests.
         * This can be handled by the eventual session reconnect.
         */
        rc = wait_for_compound_request(server, num_rqst, flags,
                                       &instance);
        if (rc)
                return rc;

        for (i = 0; i < num_rqst; i++) {
                credits[i].value = 1;
                credits[i].instance = instance;
        }

        /*
         * Make sure that we sign in the same order that we send on this socket
         * and avoid races inside tcp sendmsg code that could cause corruption
         * of smb data.
         */

        mutex_lock(&server->srv_mutex);

        /*
         * All the parts of the compound chain belong obtained credits from the
         * same session. We can not use credits obtained from the previous
         * session to send this request. Check if there were reconnects after
         * we obtained credits and return -EAGAIN in such cases to let callers
         * handle it.
         */
        if (instance != server->reconnect_instance) {
                mutex_unlock(&server->srv_mutex);
                for (j = 0; j < num_rqst; j++)
                        add_credits(server, &credits[j], optype);
                return -EAGAIN;
        }

        for (i = 0; i < num_rqst; i++) {
                midQ[i] = server->ops->setup_request(ses, server, &rqst[i]);
                if (IS_ERR(midQ[i])) {
                        revert_current_mid(server, i);
                        for (j = 0; j < i; j++)
                                cifs_delete_mid(midQ[j]);
                        mutex_unlock(&server->srv_mutex);

                        /* Update # of requests on wire to server */
                        for (j = 0; j < num_rqst; j++)
                                add_credits(server, &credits[j], optype);
                        return PTR_ERR(midQ[i]);
                }

                midQ[i]->mid_state = MID_REQUEST_SUBMITTED;
                midQ[i]->optype = optype;
                /*
                 * Invoke callback for every part of the compound chain
                 * to calculate credits properly. Wake up this thread only when
                 * the last element is received.
                 */
                if (i < num_rqst - 1)
                        midQ[i]->callback = cifs_compound_callback;
                else
                        midQ[i]->callback = cifs_compound_last_callback;
        }
        cifs_in_send_inc(server);
        rc = smb_send_rqst(server, num_rqst, rqst, flags);
        cifs_in_send_dec(server);

        for (i = 0; i < num_rqst; i++)
                cifs_save_when_sent(midQ[i]);

        if (rc < 0) {
                revert_current_mid(server, num_rqst);
                server->sequence_number -= 2;
        }

        mutex_unlock(&server->srv_mutex);

        /*
         * If sending failed for some reason or it is an oplock break that we
         * will not receive a response to - return credits back
         */
        if (rc < 0 || (flags & CIFS_NO_SRV_RSP)) {
                for (i = 0; i < num_rqst; i++)
                        add_credits(server, &credits[i], optype);
                goto out;
        }

        /*
         * At this point the request is passed to the network stack - we assume
         * that any credits taken from the server structure on the client have
         * been spent and we can't return them back. Once we receive responses
         * we will collect credits granted by the server in the mid callbacks
         * and add those credits to the server structure.
         */

        /*
         * Compounding is never used during session establish.
         */
        if ((ses->status == CifsNew) || (optype & CIFS_NEG_OP) || (optype & CIFS_SESS_OP)) {
                mutex_lock(&server->srv_mutex);
                smb311_update_preauth_hash(ses, rqst[0].rq_iov,
                                           rqst[0].rq_nvec);
                mutex_unlock(&server->srv_mutex);
        }

        for (i = 0; i < num_rqst; i++) {
                rc = wait_for_response(server, midQ[i]);
                if (rc != 0)
                        break;
        }
        if (rc != 0) {
                for (; i < num_rqst; i++) {
                        cifs_server_dbg(FYI, "Cancelling wait for mid %llu cmd: %d\n",
                                 midQ[i]->mid, le16_to_cpu(midQ[i]->command));
                        send_cancel(server, &rqst[i], midQ[i]);
                        spin_lock(&GlobalMid_Lock);
                        midQ[i]->mid_flags |= MID_WAIT_CANCELLED;
                        if (midQ[i]->mid_state == MID_REQUEST_SUBMITTED) {
                                midQ[i]->callback = cifs_cancelled_callback;
                                cancelled_mid[i] = true;
                                credits[i].value = 0;
                        }
                        spin_unlock(&GlobalMid_Lock);
                }
        }

        for (i = 0; i < num_rqst; i++) {
                if (rc < 0)
                        goto out;

                rc = cifs_sync_mid_result(midQ[i], server);
                if (rc != 0) {
                        /* mark this mid as cancelled to not free it below */
                        cancelled_mid[i] = true;
                        goto out;
                }

                if (!midQ[i]->resp_buf ||
                    midQ[i]->mid_state != MID_RESPONSE_RECEIVED) {
                        rc = -EIO;
                        cifs_dbg(FYI, "Bad MID state?\n");
                        goto out;
                }

                buf = (char *)midQ[i]->resp_buf;
                resp_iov[i].iov_base = buf;
                resp_iov[i].iov_len = midQ[i]->resp_buf_size +
                        server->vals->header_preamble_size;

                if (midQ[i]->large_buf)
                        resp_buf_type[i] = CIFS_LARGE_BUFFER;
                else
                        resp_buf_type[i] = CIFS_SMALL_BUFFER;

                rc = server->ops->check_receive(midQ[i], server,
                                                     flags & CIFS_LOG_ERROR);

                /* mark it so buf will not be freed by cifs_delete_mid */
                if ((flags & CIFS_NO_RSP_BUF) == 0)
                        midQ[i]->resp_buf = NULL;

        }

        /*
         * Compounding is never used during session establish.
         */
        if ((ses->status == CifsNew) || (optype & CIFS_NEG_OP) || (optype & CIFS_SESS_OP)) {
                struct kvec iov = {
                        .iov_base = resp_iov[0].iov_base,
                        .iov_len = resp_iov[0].iov_len
                };
                mutex_lock(&server->srv_mutex);
                smb311_update_preauth_hash(ses, &iov, 1);
                mutex_unlock(&server->srv_mutex);
        }

out:
        /*
         * This will dequeue all mids. After this it is important that the
         * demultiplex_thread will not process any of these mids any futher.
         * This is prevented above by using a noop callback that will not
         * wake this thread except for the very last PDU.
         */
        for (i = 0; i < num_rqst; i++) {
                if (!cancelled_mid[i])
                        cifs_delete_mid(midQ[i]);
        }

        return rc;
}

int
cifs_send_recv(const unsigned int xid, struct cifs_ses *ses,
               struct TCP_Server_Info *server,
               struct smb_rqst *rqst, int *resp_buf_type, const int flags,
               struct kvec *resp_iov)
{
        return compound_send_recv(xid, ses, server, flags, 1,
                                  rqst, resp_buf_type, resp_iov);
}

int
SendReceive2(const unsigned int xid, struct cifs_ses *ses,
             struct kvec *iov, int n_vec, int *resp_buf_type /* ret */,
             const int flags, struct kvec *resp_iov)
{
        struct smb_rqst rqst;
        struct kvec s_iov[CIFS_MAX_IOV_SIZE], *new_iov;
        int rc;

        if (n_vec + 1 > CIFS_MAX_IOV_SIZE) {
                new_iov = kmalloc_array(n_vec + 1, sizeof(struct kvec),
                                        GFP_KERNEL);
                if (!new_iov) {
                        /* otherwise cifs_send_recv below sets resp_buf_type */
                        *resp_buf_type = CIFS_NO_BUFFER;
                        return -ENOMEM;
                }
        } else
                new_iov = s_iov;

        /* 1st iov is a RFC1001 length followed by the rest of the packet */
        memcpy(new_iov + 1, iov, (sizeof(struct kvec) * n_vec));

        new_iov[0].iov_base = new_iov[1].iov_base;
        new_iov[0].iov_len = 4;
        new_iov[1].iov_base += 4;
        new_iov[1].iov_len -= 4;

        memset(&rqst, 0, sizeof(struct smb_rqst));
        rqst.rq_iov = new_iov;
        rqst.rq_nvec = n_vec + 1;

        rc = cifs_send_recv(xid, ses, ses->server,
                            &rqst, resp_buf_type, flags, resp_iov);
        if (n_vec + 1 > CIFS_MAX_IOV_SIZE)
                kfree(new_iov);
        return rc;
}

int
SendReceive(const unsigned int xid, struct cifs_ses *ses,
            struct smb_hdr *in_buf, struct smb_hdr *out_buf,
            int *pbytes_returned, const int flags)
{
        int rc = 0;
        struct mid_q_entry *midQ;
        unsigned int len = be32_to_cpu(in_buf->smb_buf_length);
        struct kvec iov = { .iov_base = in_buf, .iov_len = len };
        struct smb_rqst rqst = { .rq_iov = &iov, .rq_nvec = 1 };
        struct cifs_credits credits = { .value = 1, .instance = 0 };
        struct TCP_Server_Info *server;

        if (ses == NULL) {
                cifs_dbg(VFS, "Null smb session\n");
                return -EIO;
        }
        server = ses->server;
        if (server == NULL) {
                cifs_dbg(VFS, "Null tcp session\n");
                return -EIO;
        }

        if (server->tcpStatus == CifsExiting)
                return -ENOENT;

        /* Ensure that we do not send more than 50 overlapping requests
           to the same server. We may make this configurable later or
           use ses->maxReq */

        if (len > CIFSMaxBufSize + MAX_CIFS_HDR_SIZE - 4) {
                cifs_server_dbg(VFS, "Invalid length, greater than maximum frame, %d\n",
                                len);
                return -EIO;
        }

        rc = wait_for_free_request(server, flags, &credits.instance);
        if (rc)
                return rc;

        /* make sure that we sign in the same order that we send on this socket
           and avoid races inside tcp sendmsg code that could cause corruption
           of smb data */

        mutex_lock(&server->srv_mutex);

        rc = allocate_mid(ses, in_buf, &midQ);
        if (rc) {
                mutex_unlock(&server->srv_mutex);
                /* Update # of requests on wire to server */
                add_credits(server, &credits, 0);
                return rc;
        }

        rc = cifs_sign_smb(in_buf, server, &midQ->sequence_number);
        if (rc) {
                mutex_unlock(&server->srv_mutex);
                goto out;
        }

        midQ->mid_state = MID_REQUEST_SUBMITTED;

        cifs_in_send_inc(server);
        rc = smb_send(server, in_buf, len);
        cifs_in_send_dec(server);
        cifs_save_when_sent(midQ);

        if (rc < 0)
                server->sequence_number -= 2;

        mutex_unlock(&server->srv_mutex);

        if (rc < 0)
                goto out;

        rc = wait_for_response(server, midQ);
        if (rc != 0) {
                send_cancel(server, &rqst, midQ);
                spin_lock(&GlobalMid_Lock);
                if (midQ->mid_state == MID_REQUEST_SUBMITTED) {
                        /* no longer considered to be "in-flight" */
                        midQ->callback = DeleteMidQEntry;
                        spin_unlock(&GlobalMid_Lock);
                        add_credits(server, &credits, 0);
                        return rc;
                }
                spin_unlock(&GlobalMid_Lock);
        }

        rc = cifs_sync_mid_result(midQ, server);
        if (rc != 0) {
                add_credits(server, &credits, 0);
                return rc;
        }

        if (!midQ->resp_buf || !out_buf ||
            midQ->mid_state != MID_RESPONSE_RECEIVED) {
                rc = -EIO;
                cifs_server_dbg(VFS, "Bad MID state?\n");
                goto out;
        }

        *pbytes_returned = get_rfc1002_length(midQ->resp_buf);
        memcpy(out_buf, midQ->resp_buf, *pbytes_returned + 4);
        rc = cifs_check_receive(midQ, server, 0);
out:
        cifs_delete_mid(midQ);
        add_credits(server, &credits, 0);

        return rc;
}

/* We send a LOCKINGX_CANCEL_LOCK to cause the Windows
   blocking lock to return. */

static int
send_lock_cancel(const unsigned int xid, struct cifs_tcon *tcon,
                        struct smb_hdr *in_buf,
                        struct smb_hdr *out_buf)
{
        int bytes_returned;
        struct cifs_ses *ses = tcon->ses;
        LOCK_REQ *pSMB = (LOCK_REQ *)in_buf;

        /* We just modify the current in_buf to change
           the type of lock from LOCKING_ANDX_SHARED_LOCK
           or LOCKING_ANDX_EXCLUSIVE_LOCK to
           LOCKING_ANDX_CANCEL_LOCK. */

        pSMB->LockType = LOCKING_ANDX_CANCEL_LOCK|LOCKING_ANDX_LARGE_FILES;
        pSMB->Timeout = 0;
        pSMB->hdr.Mid = get_next_mid(ses->server);

        return SendReceive(xid, ses, in_buf, out_buf,
                        &bytes_returned, 0);
}

int
SendReceiveBlockingLock(const unsigned int xid, struct cifs_tcon *tcon,
            struct smb_hdr *in_buf, struct smb_hdr *out_buf,
            int *pbytes_returned)
{
        int rc = 0;
        int rstart = 0;
        struct mid_q_entry *midQ;
        struct cifs_ses *ses;
        unsigned int len = be32_to_cpu(in_buf->smb_buf_length);
        struct kvec iov = { .iov_base = in_buf, .iov_len = len };
        struct smb_rqst rqst = { .rq_iov = &iov, .rq_nvec = 1 };
        unsigned int instance;
        struct TCP_Server_Info *server;

        if (tcon == NULL || tcon->ses == NULL) {
                cifs_dbg(VFS, "Null smb session\n");
                return -EIO;
        }
        ses = tcon->ses;
        server = ses->server;

        if (server == NULL) {
                cifs_dbg(VFS, "Null tcp session\n");
                return -EIO;
        }

        if (server->tcpStatus == CifsExiting)
                return -ENOENT;

        /* Ensure that we do not send more than 50 overlapping requests
           to the same server. We may make this configurable later or
           use ses->maxReq */

        if (len > CIFSMaxBufSize + MAX_CIFS_HDR_SIZE - 4) {
                cifs_tcon_dbg(VFS, "Invalid length, greater than maximum frame, %d\n",
                              len);
                return -EIO;
        }

        rc = wait_for_free_request(server, CIFS_BLOCKING_OP, &instance);
        if (rc)
                return rc;

        /* make sure that we sign in the same order that we send on this socket
           and avoid races inside tcp sendmsg code that could cause corruption
           of smb data */

        mutex_lock(&server->srv_mutex);

        rc = allocate_mid(ses, in_buf, &midQ);
        if (rc) {
                mutex_unlock(&server->srv_mutex);
                return rc;
        }

        rc = cifs_sign_smb(in_buf, server, &midQ->sequence_number);
        if (rc) {
                cifs_delete_mid(midQ);
                mutex_unlock(&server->srv_mutex);
                return rc;
        }

        midQ->mid_state = MID_REQUEST_SUBMITTED;
        cifs_in_send_inc(server);
        rc = smb_send(server, in_buf, len);
        cifs_in_send_dec(server);
        cifs_save_when_sent(midQ);

        if (rc < 0)
                server->sequence_number -= 2;

        mutex_unlock(&server->srv_mutex);

        if (rc < 0) {
                cifs_delete_mid(midQ);
                return rc;
        }

        /* Wait for a reply - allow signals to interrupt. */
        rc = wait_event_interruptible(server->response_q,
                (!(midQ->mid_state == MID_REQUEST_SUBMITTED)) ||
                ((server->tcpStatus != CifsGood) &&
                 (server->tcpStatus != CifsNew)));

        /* Were we interrupted by a signal ? */
        if ((rc == -ERESTARTSYS) &&
                (midQ->mid_state == MID_REQUEST_SUBMITTED) &&
                ((server->tcpStatus == CifsGood) ||
                 (server->tcpStatus == CifsNew))) {

                if (in_buf->Command == SMB_COM_TRANSACTION2) {
                        /* POSIX lock. We send a NT_CANCEL SMB to cause the
                           blocking lock to return. */
                        rc = send_cancel(server, &rqst, midQ);
                        if (rc) {
                                cifs_delete_mid(midQ);
                                return rc;
                        }
                } else {
                        /* Windows lock. We send a LOCKINGX_CANCEL_LOCK
                           to cause the blocking lock to return. */

                        rc = send_lock_cancel(xid, tcon, in_buf, out_buf);

                        /* If we get -ENOLCK back the lock may have
                           already been removed. Don't exit in this case. */
                        if (rc && rc != -ENOLCK) {
                                cifs_delete_mid(midQ);
                                return rc;
                        }
                }

                rc = wait_for_response(server, midQ);
                if (rc) {
                        send_cancel(server, &rqst, midQ);
                        spin_lock(&GlobalMid_Lock);
                        if (midQ->mid_state == MID_REQUEST_SUBMITTED) {
                                /* no longer considered to be "in-flight" */
                                midQ->callback = DeleteMidQEntry;
                                spin_unlock(&GlobalMid_Lock);
                                return rc;
                        }
                        spin_unlock(&GlobalMid_Lock);
                }

                /* We got the response - restart system call. */
                rstart = 1;
        }

        rc = cifs_sync_mid_result(midQ, server);
        if (rc != 0)
                return rc;

        /* rcvd frame is ok */
        if (out_buf == NULL || midQ->mid_state != MID_RESPONSE_RECEIVED) {
                rc = -EIO;
                cifs_tcon_dbg(VFS, "Bad MID state?\n");
                goto out;
        }

        *pbytes_returned = get_rfc1002_length(midQ->resp_buf);
        memcpy(out_buf, midQ->resp_buf, *pbytes_returned + 4);
        rc = cifs_check_receive(midQ, server, 0);
out:
        cifs_delete_mid(midQ);
        if (rstart && rc == -EACCES)
                return -ERESTARTSYS;
        return rc;
}