transport.c 23.5 KB
Newer Older
L
Linus Torvalds 已提交
1 2 3
/*
 *   fs/cifs/transport.c
 *
S
Steve French 已提交
4
 *   Copyright (C) International Business Machines  Corp., 2002,2008
L
Linus Torvalds 已提交
5
 *   Author(s): Steve French (sfrench@us.ibm.com)
6
 *   Jeremy Allison (jra@samba.org) 2006.
S
Steve French 已提交
7
 *
L
Linus Torvalds 已提交
8 9 10 11 12 13 14 15 16 17 18 19
 *   This library is free software; you can redistribute it and/or modify
 *   it under the terms of the GNU Lesser General Public License as published
 *   by the Free Software Foundation; either version 2.1 of the License, or
 *   (at your option) any later version.
 *
 *   This library is distributed in the hope that it will be useful,
 *   but WITHOUT ANY WARRANTY; without even the implied warranty of
 *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See
 *   the GNU Lesser General Public License for more details.
 *
 *   You should have received a copy of the GNU Lesser General Public License
 *   along with this library; if not, write to the Free Software
S
Steve French 已提交
20
 *   Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
L
Linus Torvalds 已提交
21 22 23 24
 */

#include <linux/fs.h>
#include <linux/list.h>
25
#include <linux/gfp.h>
L
Linus Torvalds 已提交
26 27 28
#include <linux/wait.h>
#include <linux/net.h>
#include <linux/delay.h>
29
#include <linux/freezer.h>
L
Linus Torvalds 已提交
30 31 32 33 34 35 36
#include <asm/uaccess.h>
#include <asm/processor.h>
#include <linux/mempool.h>
#include "cifspdu.h"
#include "cifsglob.h"
#include "cifsproto.h"
#include "cifs_debug.h"
37

L
Linus Torvalds 已提交
38 39
extern mempool_t *cifs_mid_poolp;

40 41 42 43 44 45
static void
wake_up_task(struct mid_q_entry *mid)
{
	wake_up_process(mid->callback_data);
}

J
Jeff Layton 已提交
46
struct mid_q_entry *
47
AllocMidQEntry(const struct smb_hdr *smb_buffer, struct TCP_Server_Info *server)
L
Linus Torvalds 已提交
48 49 50
{
	struct mid_q_entry *temp;

51
	if (server == NULL) {
52
		cERROR(1, "Null TCP session in AllocMidQEntry");
L
Linus Torvalds 已提交
53 54
		return NULL;
	}
55

56
	temp = mempool_alloc(cifs_mid_poolp, GFP_NOFS);
L
Linus Torvalds 已提交
57 58 59
	if (temp == NULL)
		return temp;
	else {
60
		memset(temp, 0, sizeof(struct mid_q_entry));
L
Linus Torvalds 已提交
61 62 63
		temp->mid = smb_buffer->Mid;	/* always LE */
		temp->pid = current->pid;
		temp->command = smb_buffer->Command;
64
		cFYI(1, "For smb_command %d", temp->command);
S
Steve French 已提交
65 66 67
	/*	do_gettimeofday(&temp->when_sent);*/ /* easier to use jiffies */
		/* when mid allocated can be before when sent */
		temp->when_alloc = jiffies;
68 69 70 71 72 73 74

		/*
		 * The default is for the mid to be synchronous, so the
		 * default callback just wakes up the current task.
		 */
		temp->callback = wake_up_task;
		temp->callback_data = current;
L
Linus Torvalds 已提交
75 76 77 78 79 80 81
	}

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

82
void
L
Linus Torvalds 已提交
83 84
DeleteMidQEntry(struct mid_q_entry *midEntry)
{
S
Steve French 已提交
85 86 87
#ifdef CONFIG_CIFS_STATS2
	unsigned long now;
#endif
L
Linus Torvalds 已提交
88
	midEntry->midState = MID_FREE;
89
	atomic_dec(&midCount);
S
Steve French 已提交
90
	if (midEntry->largeBuf)
91 92 93
		cifs_buf_release(midEntry->resp_buf);
	else
		cifs_small_buf_release(midEntry->resp_buf);
S
Steve French 已提交
94 95 96 97
#ifdef CONFIG_CIFS_STATS2
	now = jiffies;
	/* commands taking longer than one second are indications that
	   something is wrong, unless it is quite a slow link or server */
S
Steve French 已提交
98 99
	if ((now - midEntry->when_alloc) > HZ) {
		if ((cifsFYI & CIFS_TIMER) &&
S
Steve French 已提交
100 101 102 103 104 105 106 107 108 109
		   (midEntry->command != SMB_COM_LOCKING_ANDX)) {
			printk(KERN_DEBUG " CIFS slow rsp: cmd %d mid %d",
			       midEntry->command, midEntry->mid);
			printk(" A: 0x%lx S: 0x%lx R: 0x%lx\n",
			       now - midEntry->when_alloc,
			       now - midEntry->when_sent,
			       now - midEntry->when_received);
		}
	}
#endif
L
Linus Torvalds 已提交
110 111 112
	mempool_free(midEntry, cifs_mid_poolp);
}

113 114 115 116 117 118 119 120 121 122
static void
delete_mid(struct mid_q_entry *mid)
{
	spin_lock(&GlobalMid_Lock);
	list_del(&mid->qhead);
	spin_unlock(&GlobalMid_Lock);

	DeleteMidQEntry(mid);
}

123
static int
124
smb_sendv(struct TCP_Server_Info *server, struct kvec *iov, int n_vec)
L
Linus Torvalds 已提交
125 126 127 128
{
	int rc = 0;
	int i = 0;
	struct msghdr smb_msg;
129 130 131 132
	struct smb_hdr *smb_buffer = iov[0].iov_base;
	unsigned int len = iov[0].iov_len;
	unsigned int total_len;
	int first_vec = 0;
133
	unsigned int smb_buf_length = be32_to_cpu(smb_buffer->smb_buf_length);
134
	struct socket *ssocket = server->ssocket;
135

S
Steve French 已提交
136
	if (ssocket == NULL)
L
Linus Torvalds 已提交
137
		return -ENOTSOCK; /* BB eventually add reconnect code here */
138

139
	smb_msg.msg_name = (struct sockaddr *) &server->dstaddr;
140
	smb_msg.msg_namelen = sizeof(struct sockaddr);
L
Linus Torvalds 已提交
141 142
	smb_msg.msg_control = NULL;
	smb_msg.msg_controllen = 0;
143
	if (server->noblocksnd)
144 145 146
		smb_msg.msg_flags = MSG_DONTWAIT + MSG_NOSIGNAL;
	else
		smb_msg.msg_flags = MSG_NOSIGNAL;
L
Linus Torvalds 已提交
147

148 149 150 151
	total_len = 0;
	for (i = 0; i < n_vec; i++)
		total_len += iov[i].iov_len;

152
	cFYI(1, "Sending smb:  total_len %d", total_len);
L
Linus Torvalds 已提交
153 154
	dump_smb(smb_buffer, len);

155
	i = 0;
156 157 158
	while (total_len) {
		rc = kernel_sendmsg(ssocket, &smb_msg, &iov[first_vec],
				    n_vec - first_vec, total_len);
L
Linus Torvalds 已提交
159 160
		if ((rc == -ENOSPC) || (rc == -EAGAIN)) {
			i++;
161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179
			/* 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.
			*/
			if ((i >= 14) || (!server->noblocksnd && (i > 2))) {
180 181
				cERROR(1, "sends on sock %p stuck for 15 seconds",
				    ssocket);
L
Linus Torvalds 已提交
182 183 184
				rc = -EAGAIN;
				break;
			}
185
			msleep(1 << i);
L
Linus Torvalds 已提交
186 187
			continue;
		}
S
Steve French 已提交
188
		if (rc < 0)
L
Linus Torvalds 已提交
189
			break;
190

S
Steve French 已提交
191 192 193 194
		if (rc == total_len) {
			total_len = 0;
			break;
		} else if (rc > total_len) {
195
			cERROR(1, "sent %d requested %d", rc, total_len);
196 197
			break;
		}
S
Steve French 已提交
198
		if (rc == 0) {
199 200
			/* should never happen, letting socket clear before
			   retrying is our only obvious option here */
201
			cERROR(1, "tcp sent no data");
202 203
			msleep(500);
			continue;
204
		}
205
		total_len -= rc;
206
		/* the line below resets i */
207 208 209 210 211 212 213 214 215 216 217 218
		for (i = first_vec; i < n_vec; i++) {
			if (iov[i].iov_len) {
				if (rc > iov[i].iov_len) {
					rc -= iov[i].iov_len;
					iov[i].iov_len = 0;
				} else {
					iov[i].iov_base += rc;
					iov[i].iov_len -= rc;
					first_vec = i;
					break;
				}
			}
219
		}
220
		i = 0; /* in case we get ENOSPC on the next send */
L
Linus Torvalds 已提交
221 222
	}

223
	if ((total_len > 0) && (total_len != smb_buf_length + 4)) {
224 225
		cFYI(1, "partial send (%d remaining), terminating session",
			total_len);
226 227 228 229 230 231 232
		/* 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 */
		server->tcpStatus = CifsNeedReconnect;
	}

233
	if (rc < 0 && rc != -EINTR)
234
		cERROR(1, "Error %d sending data on socket to server", rc);
235
	else
L
Linus Torvalds 已提交
236 237
		rc = 0;

J
[CIFS]  
Jeremy Allison 已提交
238 239
	/* Don't want to modify the buffer as a
	   side effect of this call. */
240
	smb_buffer->smb_buf_length = cpu_to_be32(smb_buf_length);
J
[CIFS]  
Jeremy Allison 已提交
241

L
Linus Torvalds 已提交
242 243 244
	return rc;
}

245 246 247 248 249 250 251 252 253 254 255 256
int
smb_send(struct TCP_Server_Info *server, struct smb_hdr *smb_buffer,
	 unsigned int smb_buf_length)
{
	struct kvec iov;

	iov.iov_base = smb_buffer;
	iov.iov_len = smb_buf_length + 4;

	return smb_sendv(server, &iov, 1);
}

257 258
static int wait_for_free_request(struct TCP_Server_Info *server,
				 const int long_op)
L
Linus Torvalds 已提交
259
{
260
	if (long_op == CIFS_ASYNC_OP) {
L
Linus Torvalds 已提交
261
		/* oplock breaks must not be held up */
262
		atomic_inc(&server->inFlight);
263 264 265 266 267
		return 0;
	}

	spin_lock(&GlobalMid_Lock);
	while (1) {
268
		if (atomic_read(&server->inFlight) >= server->maxReq) {
269
			spin_unlock(&GlobalMid_Lock);
270
			cifs_num_waiters_inc(server);
271 272
			wait_event(server->request_q,
				   atomic_read(&server->inFlight)
273
				     < server->maxReq);
274
			cifs_num_waiters_dec(server);
275 276
			spin_lock(&GlobalMid_Lock);
		} else {
277
			if (server->tcpStatus == CifsExiting) {
L
Linus Torvalds 已提交
278
				spin_unlock(&GlobalMid_Lock);
279
				return -ENOENT;
L
Linus Torvalds 已提交
280
			}
281 282 283 284 285 286

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

			/* update # of requests on the wire to server */
			if (long_op != CIFS_BLOCKING_OP)
287
				atomic_inc(&server->inFlight);
288 289
			spin_unlock(&GlobalMid_Lock);
			break;
L
Linus Torvalds 已提交
290 291
		}
	}
J
[CIFS]  
Jeremy Allison 已提交
292 293
	return 0;
}
L
Linus Torvalds 已提交
294

295
static int allocate_mid(struct cifs_ses *ses, struct smb_hdr *in_buf,
J
[CIFS]  
Jeremy Allison 已提交
296 297
			struct mid_q_entry **ppmidQ)
{
L
Linus Torvalds 已提交
298
	if (ses->server->tcpStatus == CifsExiting) {
J
[CIFS]  
Jeremy Allison 已提交
299
		return -ENOENT;
300 301 302
	}

	if (ses->server->tcpStatus == CifsNeedReconnect) {
303
		cFYI(1, "tcp session dead - return to caller to retry");
J
[CIFS]  
Jeremy Allison 已提交
304
		return -EAGAIN;
305 306 307
	}

	if (ses->status != CifsGood) {
L
Linus Torvalds 已提交
308
		/* check if SMB session is bad because we are setting it up */
S
Steve French 已提交
309
		if ((in_buf->Command != SMB_COM_SESSION_SETUP_ANDX) &&
S
Steve French 已提交
310
			(in_buf->Command != SMB_COM_NEGOTIATE))
J
[CIFS]  
Jeremy Allison 已提交
311
			return -EAGAIN;
S
Steve French 已提交
312
		/* else ok - we are setting up session */
L
Linus Torvalds 已提交
313
	}
314
	*ppmidQ = AllocMidQEntry(in_buf, ses->server);
315
	if (*ppmidQ == NULL)
J
[CIFS]  
Jeremy Allison 已提交
316
		return -ENOMEM;
317 318 319
	spin_lock(&GlobalMid_Lock);
	list_add_tail(&(*ppmidQ)->qhead, &ses->server->pending_mid_q);
	spin_unlock(&GlobalMid_Lock);
J
[CIFS]  
Jeremy Allison 已提交
320 321 322
	return 0;
}

323 324
static int
wait_for_response(struct TCP_Server_Info *server, struct mid_q_entry *midQ)
J
[CIFS]  
Jeremy Allison 已提交
325
{
326
	int error;
J
[CIFS]  
Jeremy Allison 已提交
327

328
	error = wait_event_freezekillable(server->response_q,
329 330 331
				    midQ->midState != MID_REQUEST_SUBMITTED);
	if (error < 0)
		return -ERESTARTSYS;
J
[CIFS]  
Jeremy Allison 已提交
332

333
	return 0;
J
[CIFS]  
Jeremy Allison 已提交
334 335
}

336

J
Jeff Layton 已提交
337 338 339 340 341
/*
 * Send a SMB request and set the callback function in the mid to handle
 * the result. Caller is responsible for dealing with timeouts.
 */
int
342
cifs_call_async(struct TCP_Server_Info *server, struct kvec *iov,
343 344
		unsigned int nvec, mid_receive_t *receive,
		mid_callback_t *callback, void *cbdata, bool ignore_pend)
J
Jeff Layton 已提交
345 346 347
{
	int rc;
	struct mid_q_entry *mid;
348
	struct smb_hdr *hdr = (struct smb_hdr *)iov[0].iov_base;
J
Jeff Layton 已提交
349

350
	rc = wait_for_free_request(server, ignore_pend ? CIFS_ASYNC_OP : 0);
J
Jeff Layton 已提交
351 352 353
	if (rc)
		return rc;

354
	/* enable signing if server requires it */
355
	if (server->sec_mode & (SECMODE_SIGN_REQUIRED | SECMODE_SIGN_ENABLED))
356
		hdr->Flags2 |= SMBFLG2_SECURITY_SIGNATURE;
357

J
Jeff Layton 已提交
358
	mutex_lock(&server->srv_mutex);
359
	mid = AllocMidQEntry(hdr, server);
J
Jeff Layton 已提交
360 361
	if (mid == NULL) {
		mutex_unlock(&server->srv_mutex);
362 363
		atomic_dec(&server->inFlight);
		wake_up(&server->request_q);
J
Jeff Layton 已提交
364 365 366 367 368 369 370 371
		return -ENOMEM;
	}

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

372
	rc = cifs_sign_smb2(iov, nvec, server, &mid->sequence_number);
J
Jeff Layton 已提交
373 374 375 376 377
	if (rc) {
		mutex_unlock(&server->srv_mutex);
		goto out_err;
	}

378
	mid->receive = receive;
J
Jeff Layton 已提交
379 380 381
	mid->callback = callback;
	mid->callback_data = cbdata;
	mid->midState = MID_REQUEST_SUBMITTED;
382 383

	cifs_in_send_inc(server);
384
	rc = smb_sendv(server, iov, nvec);
385 386
	cifs_in_send_dec(server);
	cifs_save_when_sent(mid);
J
Jeff Layton 已提交
387
	mutex_unlock(&server->srv_mutex);
388

J
Jeff Layton 已提交
389 390 391 392 393 394 395 396 397 398 399
	if (rc)
		goto out_err;

	return rc;
out_err:
	delete_mid(mid);
	atomic_dec(&server->inFlight);
	wake_up(&server->request_q);
	return rc;
}

400 401 402 403 404 405 406 407 408 409
/*
 *
 * 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
410
SendReceiveNoRsp(const unsigned int xid, struct cifs_ses *ses,
411 412 413 414 415 416 417
		struct smb_hdr *in_buf, int flags)
{
	int rc;
	struct kvec iov[1];
	int resp_buf_type;

	iov[0].iov_base = (char *)in_buf;
418
	iov[0].iov_len = be32_to_cpu(in_buf->smb_buf_length) + 4;
419 420
	flags |= CIFS_NO_RESP;
	rc = SendReceive2(xid, ses, iov, 1, &resp_buf_type, flags);
421
	cFYI(DBG2, "SendRcvNoRsp flags %d rc %d", flags, rc);
422

423 424 425
	return rc;
}

426
static int
427
cifs_sync_mid_result(struct mid_q_entry *mid, struct TCP_Server_Info *server)
428 429 430
{
	int rc = 0;

J
Jeff Layton 已提交
431 432
	cFYI(1, "%s: cmd=%d mid=%d state=%d", __func__, mid->command,
		mid->mid, mid->midState);
433

J
Jeff Layton 已提交
434 435 436
	spin_lock(&GlobalMid_Lock);
	switch (mid->midState) {
	case MID_RESPONSE_RECEIVED:
437 438
		spin_unlock(&GlobalMid_Lock);
		return rc;
J
Jeff Layton 已提交
439 440 441
	case MID_RETRY_NEEDED:
		rc = -EAGAIN;
		break;
442 443 444
	case MID_RESPONSE_MALFORMED:
		rc = -EIO;
		break;
445 446 447
	case MID_SHUTDOWN:
		rc = -EHOSTDOWN;
		break;
J
Jeff Layton 已提交
448
	default:
449
		list_del_init(&mid->qhead);
J
Jeff Layton 已提交
450 451 452
		cERROR(1, "%s: invalid mid state mid=%d state=%d", __func__,
			mid->mid, mid->midState);
		rc = -EIO;
453 454 455
	}
	spin_unlock(&GlobalMid_Lock);

456
	DeleteMidQEntry(mid);
457 458 459
	return rc;
}

460 461 462 463 464 465 466 467 468 469 470 471 472 473 474 475 476
/*
 * An NT cancel request header looks just like the original request except:
 *
 * The Command is SMB_COM_NT_CANCEL
 * The WordCount is zeroed out
 * The ByteCount is zeroed out
 *
 * This function mangles an existing request buffer into a
 * SMB_COM_NT_CANCEL request and then sends it.
 */
static int
send_nt_cancel(struct TCP_Server_Info *server, struct smb_hdr *in_buf,
		struct mid_q_entry *mid)
{
	int rc = 0;

	/* -4 for RFC1001 length and +2 for BCC field */
477
	in_buf->smb_buf_length = cpu_to_be32(sizeof(struct smb_hdr) - 4  + 2);
478 479
	in_buf->Command = SMB_COM_NT_CANCEL;
	in_buf->WordCount = 0;
480
	put_bcc(0, in_buf);
481 482 483 484 485 486 487

	mutex_lock(&server->srv_mutex);
	rc = cifs_sign_smb(in_buf, server, &mid->sequence_number);
	if (rc) {
		mutex_unlock(&server->srv_mutex);
		return rc;
	}
488
	rc = smb_send(server, in_buf, be32_to_cpu(in_buf->smb_buf_length));
489 490 491 492 493 494 495 496
	mutex_unlock(&server->srv_mutex);

	cFYI(1, "issued NT_CANCEL for mid %u, rc = %d",
		in_buf->Mid, rc);

	return rc;
}

497 498 499 500
int
cifs_check_receive(struct mid_q_entry *mid, struct TCP_Server_Info *server,
		   bool log_error)
{
501 502 503
	unsigned int len = be32_to_cpu(mid->resp_buf->smb_buf_length) + 4;

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

	/* convert the length into a more usable form */
506
	if (server->sec_mode & (SECMODE_SIGN_REQUIRED | SECMODE_SIGN_ENABLED)) {
507 508 509 510
		struct kvec iov;

		iov.iov_base = mid->resp_buf;
		iov.iov_len = len;
511
		/* FIXME: add code to kill session */
512
		if (cifs_verify_signature(&iov, 1, server,
513 514 515 516 517 518 519 520
					  mid->sequence_number + 1) != 0)
			cERROR(1, "Unexpected SMB signature");
	}

	/* BB special case reconnect tid and uid here? */
	return map_smb_to_linux_error(mid->resp_buf, log_error);
}

J
[CIFS]  
Jeremy Allison 已提交
521
int
522
SendReceive2(const unsigned int xid, struct cifs_ses *ses,
S
Steve French 已提交
523
	     struct kvec *iov, int n_vec, int *pRespBufType /* ret */,
524
	     const int flags)
J
[CIFS]  
Jeremy Allison 已提交
525 526
{
	int rc = 0;
527
	int long_op;
J
[CIFS]  
Jeremy Allison 已提交
528 529
	struct mid_q_entry *midQ;
	struct smb_hdr *in_buf = iov[0].iov_base;
530

531 532
	long_op = flags & CIFS_TIMEOUT_MASK;

J
[CIFS]  
Jeremy Allison 已提交
533 534 535 536
	*pRespBufType = CIFS_NO_BUFFER;  /* no response buf yet */

	if ((ses == NULL) || (ses->server == NULL)) {
		cifs_small_buf_release(in_buf);
537
		cERROR(1, "Null session");
J
[CIFS]  
Jeremy Allison 已提交
538 539 540
		return -EIO;
	}

S
Steve French 已提交
541
	if (ses->server->tcpStatus == CifsExiting) {
J
[CIFS]  
Jeremy Allison 已提交
542 543 544 545
		cifs_small_buf_release(in_buf);
		return -ENOENT;
	}

S
Steve French 已提交
546
	/* Ensure that we do not send more than 50 overlapping requests
J
[CIFS]  
Jeremy Allison 已提交
547 548 549
	   to the same server. We may make this configurable later or
	   use ses->maxReq */

550
	rc = wait_for_free_request(ses->server, long_op);
J
[CIFS]  
Jeremy Allison 已提交
551 552 553 554 555
	if (rc) {
		cifs_small_buf_release(in_buf);
		return rc;
	}

S
Steve French 已提交
556
	/* make sure that we sign in the same order that we send on this socket
J
[CIFS]  
Jeremy Allison 已提交
557 558 559
	   and avoid races inside tcp sendmsg code that could cause corruption
	   of smb data */

J
Jeff Layton 已提交
560
	mutex_lock(&ses->server->srv_mutex);
J
[CIFS]  
Jeremy Allison 已提交
561 562 563

	rc = allocate_mid(ses, in_buf, &midQ);
	if (rc) {
J
Jeff Layton 已提交
564
		mutex_unlock(&ses->server->srv_mutex);
565
		cifs_small_buf_release(in_buf);
J
[CIFS]  
Jeremy Allison 已提交
566
		/* Update # of requests on wire to server */
S
Steve French 已提交
567
		atomic_dec(&ses->server->inFlight);
J
[CIFS]  
Jeremy Allison 已提交
568 569
		wake_up(&ses->server->request_q);
		return rc;
L
Linus Torvalds 已提交
570
	}
S
Steve French 已提交
571
	rc = cifs_sign_smb2(iov, n_vec, ses->server, &midQ->sequence_number);
572 573 574 575 576
	if (rc) {
		mutex_unlock(&ses->server->srv_mutex);
		cifs_small_buf_release(in_buf);
		goto out;
	}
L
Linus Torvalds 已提交
577 578

	midQ->midState = MID_REQUEST_SUBMITTED;
579
	cifs_in_send_inc(ses->server);
580
	rc = smb_sendv(ses->server, iov, n_vec);
581 582
	cifs_in_send_dec(ses->server);
	cifs_save_when_sent(midQ);
J
[CIFS]  
Jeremy Allison 已提交
583

J
Jeff Layton 已提交
584
	mutex_unlock(&ses->server->srv_mutex);
J
[CIFS]  
Jeremy Allison 已提交
585

586 587
	if (rc < 0) {
		cifs_small_buf_release(in_buf);
J
[CIFS]  
Jeremy Allison 已提交
588
		goto out;
589
	}
590

591 592
	if (long_op == CIFS_ASYNC_OP) {
		cifs_small_buf_release(in_buf);
593
		goto out;
594
	}
595

596
	rc = wait_for_response(ses->server, midQ);
597
	if (rc != 0) {
598
		send_nt_cancel(ses->server, in_buf, midQ);
599 600 601 602
		spin_lock(&GlobalMid_Lock);
		if (midQ->midState == MID_REQUEST_SUBMITTED) {
			midQ->callback = DeleteMidQEntry;
			spin_unlock(&GlobalMid_Lock);
603
			cifs_small_buf_release(in_buf);
604 605 606 607 608 609
			atomic_dec(&ses->server->inFlight);
			wake_up(&ses->server->request_q);
			return rc;
		}
		spin_unlock(&GlobalMid_Lock);
	}
610

611 612
	cifs_small_buf_release(in_buf);

613
	rc = cifs_sync_mid_result(midQ, ses->server);
614
	if (rc != 0) {
S
Steve French 已提交
615
		atomic_dec(&ses->server->inFlight);
J
[CIFS]  
Jeremy Allison 已提交
616
		wake_up(&ses->server->request_q);
617 618
		return rc;
	}
619

620
	if (!midQ->resp_buf || midQ->midState != MID_RESPONSE_RECEIVED) {
621
		rc = -EIO;
622
		cFYI(1, "Bad MID state?");
623 624 625
		goto out;
	}

626 627 628 629 630 631
	iov[0].iov_base = (char *)midQ->resp_buf;
	iov[0].iov_len = be32_to_cpu(midQ->resp_buf->smb_buf_length) + 4;
	if (midQ->largeBuf)
		*pRespBufType = CIFS_LARGE_BUFFER;
	else
		*pRespBufType = CIFS_SMALL_BUFFER;
632

633
	rc = cifs_check_receive(midQ, ses->server, flags & CIFS_LOG_ERROR);
L
Linus Torvalds 已提交
634

635 636 637
	/* mark it so buf will not be freed by delete_mid */
	if ((flags & CIFS_NO_RESP) == 0)
		midQ->resp_buf = NULL;
J
[CIFS]  
Jeremy Allison 已提交
638
out:
639
	delete_mid(midQ);
S
Steve French 已提交
640
	atomic_dec(&ses->server->inFlight);
J
[CIFS]  
Jeremy Allison 已提交
641
	wake_up(&ses->server->request_q);
L
Linus Torvalds 已提交
642

643 644
	return rc;
}
L
Linus Torvalds 已提交
645 646

int
647
SendReceive(const unsigned int xid, struct cifs_ses *ses,
L
Linus Torvalds 已提交
648 649 650 651 652 653 654
	    struct smb_hdr *in_buf, struct smb_hdr *out_buf,
	    int *pbytes_returned, const int long_op)
{
	int rc = 0;
	struct mid_q_entry *midQ;

	if (ses == NULL) {
655
		cERROR(1, "Null smb session");
L
Linus Torvalds 已提交
656 657
		return -EIO;
	}
S
Steve French 已提交
658
	if (ses->server == NULL) {
659
		cERROR(1, "Null tcp session");
L
Linus Torvalds 已提交
660 661 662
		return -EIO;
	}

S
Steve French 已提交
663
	if (ses->server->tcpStatus == CifsExiting)
664 665
		return -ENOENT;

S
Steve French 已提交
666
	/* Ensure that we do not send more than 50 overlapping requests
L
Linus Torvalds 已提交
667 668 669
	   to the same server. We may make this configurable later or
	   use ses->maxReq */

670 671
	if (be32_to_cpu(in_buf->smb_buf_length) > CIFSMaxBufSize +
			MAX_CIFS_HDR_SIZE - 4) {
672
		cERROR(1, "Illegal length, greater than maximum frame, %d",
673
			   be32_to_cpu(in_buf->smb_buf_length));
674 675 676
		return -EIO;
	}

677
	rc = wait_for_free_request(ses->server, long_op);
J
[CIFS]  
Jeremy Allison 已提交
678 679 680
	if (rc)
		return rc;

S
Steve French 已提交
681
	/* make sure that we sign in the same order that we send on this socket
L
Linus Torvalds 已提交
682 683 684
	   and avoid races inside tcp sendmsg code that could cause corruption
	   of smb data */

J
Jeff Layton 已提交
685
	mutex_lock(&ses->server->srv_mutex);
L
Linus Torvalds 已提交
686

J
[CIFS]  
Jeremy Allison 已提交
687 688
	rc = allocate_mid(ses, in_buf, &midQ);
	if (rc) {
J
Jeff Layton 已提交
689
		mutex_unlock(&ses->server->srv_mutex);
J
[CIFS]  
Jeremy Allison 已提交
690
		/* Update # of requests on wire to server */
S
Steve French 已提交
691
		atomic_dec(&ses->server->inFlight);
J
[CIFS]  
Jeremy Allison 已提交
692 693
		wake_up(&ses->server->request_q);
		return rc;
L
Linus Torvalds 已提交
694 695
	}

696
	rc = cifs_sign_smb(in_buf, ses->server, &midQ->sequence_number);
697 698 699 700
	if (rc) {
		mutex_unlock(&ses->server->srv_mutex);
		goto out;
	}
L
Linus Torvalds 已提交
701 702

	midQ->midState = MID_REQUEST_SUBMITTED;
703 704

	cifs_in_send_inc(ses->server);
705
	rc = smb_send(ses->server, in_buf, be32_to_cpu(in_buf->smb_buf_length));
706 707
	cifs_in_send_dec(ses->server);
	cifs_save_when_sent(midQ);
J
Jeff Layton 已提交
708
	mutex_unlock(&ses->server->srv_mutex);
J
[CIFS]  
Jeremy Allison 已提交
709

S
Steve French 已提交
710
	if (rc < 0)
J
[CIFS]  
Jeremy Allison 已提交
711 712
		goto out;

713
	if (long_op == CIFS_ASYNC_OP)
J
[CIFS]  
Jeremy Allison 已提交
714
		goto out;
L
Linus Torvalds 已提交
715

716
	rc = wait_for_response(ses->server, midQ);
717
	if (rc != 0) {
718
		send_nt_cancel(ses->server, in_buf, midQ);
719 720 721 722 723 724 725 726 727 728 729
		spin_lock(&GlobalMid_Lock);
		if (midQ->midState == MID_REQUEST_SUBMITTED) {
			/* no longer considered to be "in-flight" */
			midQ->callback = DeleteMidQEntry;
			spin_unlock(&GlobalMid_Lock);
			atomic_dec(&ses->server->inFlight);
			wake_up(&ses->server->request_q);
			return rc;
		}
		spin_unlock(&GlobalMid_Lock);
	}
L
Linus Torvalds 已提交
730

731
	rc = cifs_sync_mid_result(midQ, ses->server);
732
	if (rc != 0) {
S
Steve French 已提交
733
		atomic_dec(&ses->server->inFlight);
J
[CIFS]  
Jeremy Allison 已提交
734
		wake_up(&ses->server->request_q);
L
Linus Torvalds 已提交
735 736
		return rc;
	}
737

738 739
	if (!midQ->resp_buf || !out_buf ||
	    midQ->midState != MID_RESPONSE_RECEIVED) {
740
		rc = -EIO;
741
		cERROR(1, "Bad MID state?");
742
		goto out;
L
Linus Torvalds 已提交
743
	}
J
[CIFS]  
Jeremy Allison 已提交
744

745 746 747
	*pbytes_returned = be32_to_cpu(midQ->resp_buf->smb_buf_length);
	memcpy(out_buf, midQ->resp_buf, *pbytes_returned + 4);
	rc = cifs_check_receive(midQ, ses->server, 0);
J
[CIFS]  
Jeremy Allison 已提交
748
out:
749
	delete_mid(midQ);
S
Steve French 已提交
750
	atomic_dec(&ses->server->inFlight);
J
[CIFS]  
Jeremy Allison 已提交
751
	wake_up(&ses->server->request_q);
L
Linus Torvalds 已提交
752

J
[CIFS]  
Jeremy Allison 已提交
753 754
	return rc;
}
L
Linus Torvalds 已提交
755

J
[CIFS]  
Jeremy Allison 已提交
756 757 758 759
/* We send a LOCKINGX_CANCEL_LOCK to cause the Windows
   blocking lock to return. */

static int
760
send_lock_cancel(const unsigned int xid, struct cifs_tcon *tcon,
J
[CIFS]  
Jeremy Allison 已提交
761 762 763 764
			struct smb_hdr *in_buf,
			struct smb_hdr *out_buf)
{
	int bytes_returned;
765
	struct cifs_ses *ses = tcon->ses;
J
[CIFS]  
Jeremy Allison 已提交
766 767 768 769 770 771 772 773 774 775 776 777
	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 = GetNextMid(ses->server);

	return SendReceive(xid, ses, in_buf, out_buf,
778
			&bytes_returned, 0);
J
[CIFS]  
Jeremy Allison 已提交
779 780 781
}

int
782
SendReceiveBlockingLock(const unsigned int xid, struct cifs_tcon *tcon,
J
[CIFS]  
Jeremy Allison 已提交
783 784 785 786 787 788
	    struct smb_hdr *in_buf, struct smb_hdr *out_buf,
	    int *pbytes_returned)
{
	int rc = 0;
	int rstart = 0;
	struct mid_q_entry *midQ;
789
	struct cifs_ses *ses;
J
[CIFS]  
Jeremy Allison 已提交
790 791

	if (tcon == NULL || tcon->ses == NULL) {
792
		cERROR(1, "Null smb session");
J
[CIFS]  
Jeremy Allison 已提交
793 794 795 796
		return -EIO;
	}
	ses = tcon->ses;

S
Steve French 已提交
797
	if (ses->server == NULL) {
798
		cERROR(1, "Null tcp session");
J
[CIFS]  
Jeremy Allison 已提交
799 800 801
		return -EIO;
	}

S
Steve French 已提交
802
	if (ses->server->tcpStatus == CifsExiting)
J
[CIFS]  
Jeremy Allison 已提交
803 804
		return -ENOENT;

S
Steve French 已提交
805
	/* Ensure that we do not send more than 50 overlapping requests
J
[CIFS]  
Jeremy Allison 已提交
806 807 808
	   to the same server. We may make this configurable later or
	   use ses->maxReq */

809 810
	if (be32_to_cpu(in_buf->smb_buf_length) > CIFSMaxBufSize +
			MAX_CIFS_HDR_SIZE - 4) {
811
		cERROR(1, "Illegal length, greater than maximum frame, %d",
812
			   be32_to_cpu(in_buf->smb_buf_length));
813 814 815
		return -EIO;
	}

816
	rc = wait_for_free_request(ses->server, CIFS_BLOCKING_OP);
J
[CIFS]  
Jeremy Allison 已提交
817 818 819
	if (rc)
		return rc;

S
Steve French 已提交
820
	/* make sure that we sign in the same order that we send on this socket
J
[CIFS]  
Jeremy Allison 已提交
821 822 823
	   and avoid races inside tcp sendmsg code that could cause corruption
	   of smb data */

J
Jeff Layton 已提交
824
	mutex_lock(&ses->server->srv_mutex);
J
[CIFS]  
Jeremy Allison 已提交
825 826 827

	rc = allocate_mid(ses, in_buf, &midQ);
	if (rc) {
J
Jeff Layton 已提交
828
		mutex_unlock(&ses->server->srv_mutex);
J
[CIFS]  
Jeremy Allison 已提交
829 830 831 832
		return rc;
	}

	rc = cifs_sign_smb(in_buf, ses->server, &midQ->sequence_number);
833
	if (rc) {
834
		delete_mid(midQ);
835 836 837
		mutex_unlock(&ses->server->srv_mutex);
		return rc;
	}
L
Linus Torvalds 已提交
838

J
[CIFS]  
Jeremy Allison 已提交
839
	midQ->midState = MID_REQUEST_SUBMITTED;
840
	cifs_in_send_inc(ses->server);
841
	rc = smb_send(ses->server, in_buf, be32_to_cpu(in_buf->smb_buf_length));
842 843
	cifs_in_send_dec(ses->server);
	cifs_save_when_sent(midQ);
J
Jeff Layton 已提交
844
	mutex_unlock(&ses->server->srv_mutex);
J
[CIFS]  
Jeremy Allison 已提交
845

S
Steve French 已提交
846
	if (rc < 0) {
847
		delete_mid(midQ);
J
[CIFS]  
Jeremy Allison 已提交
848 849 850 851 852
		return rc;
	}

	/* Wait for a reply - allow signals to interrupt. */
	rc = wait_event_interruptible(ses->server->response_q,
S
Steve French 已提交
853
		(!(midQ->midState == MID_REQUEST_SUBMITTED)) ||
J
[CIFS]  
Jeremy Allison 已提交
854 855 856 857 858 859 860 861 862 863 864 865
		((ses->server->tcpStatus != CifsGood) &&
		 (ses->server->tcpStatus != CifsNew)));

	/* Were we interrupted by a signal ? */
	if ((rc == -ERESTARTSYS) &&
		(midQ->midState == MID_REQUEST_SUBMITTED) &&
		((ses->server->tcpStatus == CifsGood) ||
		 (ses->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. */
866
			rc = send_nt_cancel(ses->server, in_buf, midQ);
J
[CIFS]  
Jeremy Allison 已提交
867
			if (rc) {
868
				delete_mid(midQ);
J
[CIFS]  
Jeremy Allison 已提交
869 870 871 872 873 874 875 876 877 878 879
				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) {
880
				delete_mid(midQ);
J
[CIFS]  
Jeremy Allison 已提交
881 882 883 884
				return rc;
			}
		}

885 886
		rc = wait_for_response(ses->server, midQ);
		if (rc) {
887
			send_nt_cancel(ses->server, in_buf, midQ);
888 889 890 891 892 893 894 895
			spin_lock(&GlobalMid_Lock);
			if (midQ->midState == MID_REQUEST_SUBMITTED) {
				/* no longer considered to be "in-flight" */
				midQ->callback = DeleteMidQEntry;
				spin_unlock(&GlobalMid_Lock);
				return rc;
			}
			spin_unlock(&GlobalMid_Lock);
J
[CIFS]  
Jeremy Allison 已提交
896
		}
897 898 899

		/* We got the response - restart system call. */
		rstart = 1;
J
[CIFS]  
Jeremy Allison 已提交
900 901
	}

902
	rc = cifs_sync_mid_result(midQ, ses->server);
903
	if (rc != 0)
J
[CIFS]  
Jeremy Allison 已提交
904
		return rc;
905

906
	/* rcvd frame is ok */
907
	if (out_buf == NULL || midQ->midState != MID_RESPONSE_RECEIVED) {
908
		rc = -EIO;
909
		cERROR(1, "Bad MID state?");
910 911
		goto out;
	}
L
Linus Torvalds 已提交
912

913 914 915
	*pbytes_returned = be32_to_cpu(midQ->resp_buf->smb_buf_length);
	memcpy(out_buf, midQ->resp_buf, *pbytes_returned + 4);
	rc = cifs_check_receive(midQ, ses->server, 0);
916
out:
917
	delete_mid(midQ);
J
[CIFS]  
Jeremy Allison 已提交
918 919
	if (rstart && rc == -EACCES)
		return -ERESTARTSYS;
L
Linus Torvalds 已提交
920 921
	return rc;
}