transport.c 23.4 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);
}

P
Pavel Shilovsky 已提交
257 258
static int
wait_for_free_request(struct TCP_Server_Info *server, const int long_op)
L
Linus Torvalds 已提交
259
{
P
Pavel Shilovsky 已提交
260 261
	spin_lock(&server->req_lock);

262
	if (long_op == CIFS_ASYNC_OP) {
L
Linus Torvalds 已提交
263
		/* oplock breaks must not be held up */
P
Pavel Shilovsky 已提交
264 265
		server->in_flight++;
		spin_unlock(&server->req_lock);
266 267 268 269
		return 0;
	}

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

			/* 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)
P
Pavel Shilovsky 已提交
288 289
				server->in_flight++;
			spin_unlock(&server->req_lock);
290
			break;
L
Linus Torvalds 已提交
291 292
		}
	}
J
[CIFS]  
Jeremy Allison 已提交
293 294
	return 0;
}
L
Linus Torvalds 已提交
295

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

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

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

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

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

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

337

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

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

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

J
Jeff Layton 已提交
359
	mutex_lock(&server->srv_mutex);
360
	mid = AllocMidQEntry(hdr, server);
J
Jeff Layton 已提交
361 362
	if (mid == NULL) {
		mutex_unlock(&server->srv_mutex);
P
Pavel Shilovsky 已提交
363
		dec_in_flight(server);
364
		wake_up(&server->request_q);
J
Jeff Layton 已提交
365 366 367 368 369 370 371 372
		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);

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

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

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

J
Jeff Layton 已提交
390 391 392 393 394 395
	if (rc)
		goto out_err;

	return rc;
out_err:
	delete_mid(mid);
P
Pavel Shilovsky 已提交
396
	dec_in_flight(server);
J
Jeff Layton 已提交
397 398 399 400
	wake_up(&server->request_q);
	return rc;
}

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

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

424 425 426
	return rc;
}

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

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

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

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

461 462 463 464 465 466 467 468 469 470 471 472 473 474 475 476 477
/*
 * 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 */
478
	in_buf->smb_buf_length = cpu_to_be32(sizeof(struct smb_hdr) - 4  + 2);
479 480
	in_buf->Command = SMB_COM_NT_CANCEL;
	in_buf->WordCount = 0;
481
	put_bcc(0, in_buf);
482 483 484 485 486 487 488

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

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

	return rc;
}

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

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

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

		iov.iov_base = mid->resp_buf;
		iov.iov_len = len;
512
		/* FIXME: add code to kill session */
513
		if (cifs_verify_signature(&iov, 1, server,
514 515 516 517 518 519 520 521
					  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 已提交
522
int
523
SendReceive2(const unsigned int xid, struct cifs_ses *ses,
S
Steve French 已提交
524
	     struct kvec *iov, int n_vec, int *pRespBufType /* ret */,
525
	     const int flags)
J
[CIFS]  
Jeremy Allison 已提交
526 527
{
	int rc = 0;
528
	int long_op;
J
[CIFS]  
Jeremy Allison 已提交
529 530
	struct mid_q_entry *midQ;
	struct smb_hdr *in_buf = iov[0].iov_base;
531

532 533
	long_op = flags & CIFS_TIMEOUT_MASK;

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

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

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

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

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

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

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

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

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

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

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

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

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

612 613
	cifs_small_buf_release(in_buf);

614
	rc = cifs_sync_mid_result(midQ, ses->server);
615
	if (rc != 0) {
P
Pavel Shilovsky 已提交
616
		dec_in_flight(ses->server);
J
[CIFS]  
Jeremy Allison 已提交
617
		wake_up(&ses->server->request_q);
618 619
		return rc;
	}
620

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

627 628 629 630 631 632
	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;
633

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

636 637 638
	/* 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 已提交
639
out:
640
	delete_mid(midQ);
P
Pavel Shilovsky 已提交
641
	dec_in_flight(ses->server);
J
[CIFS]  
Jeremy Allison 已提交
642
	wake_up(&ses->server->request_q);
L
Linus Torvalds 已提交
643

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

int
648
SendReceive(const unsigned int xid, struct cifs_ses *ses,
L
Linus Torvalds 已提交
649 650 651 652 653 654 655
	    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) {
656
		cERROR(1, "Null smb session");
L
Linus Torvalds 已提交
657 658
		return -EIO;
	}
S
Steve French 已提交
659
	if (ses->server == NULL) {
660
		cERROR(1, "Null tcp session");
L
Linus Torvalds 已提交
661 662 663
		return -EIO;
	}

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

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

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

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

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

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

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

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

	midQ->midState = MID_REQUEST_SUBMITTED;
704 705

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

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

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

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

732
	rc = cifs_sync_mid_result(midQ, ses->server);
733
	if (rc != 0) {
P
Pavel Shilovsky 已提交
734
		dec_in_flight(ses->server);
J
[CIFS]  
Jeremy Allison 已提交
735
		wake_up(&ses->server->request_q);
L
Linus Torvalds 已提交
736 737
		return rc;
	}
738

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

746 747 748
	*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 已提交
749
out:
750
	delete_mid(midQ);
P
Pavel Shilovsky 已提交
751
	dec_in_flight(ses->server);
J
[CIFS]  
Jeremy Allison 已提交
752
	wake_up(&ses->server->request_q);
L
Linus Torvalds 已提交
753

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

	/* Wait for a reply - allow signals to interrupt. */
	rc = wait_event_interruptible(ses->server->response_q,
S
Steve French 已提交
854
		(!(midQ->midState == MID_REQUEST_SUBMITTED)) ||
J
[CIFS]  
Jeremy Allison 已提交
855 856 857 858 859 860 861 862 863 864 865 866
		((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. */
867
			rc = send_nt_cancel(ses->server, in_buf, midQ);
J
[CIFS]  
Jeremy Allison 已提交
868
			if (rc) {
869
				delete_mid(midQ);
J
[CIFS]  
Jeremy Allison 已提交
870 871 872 873 874 875 876 877 878 879 880
				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) {
881
				delete_mid(midQ);
J
[CIFS]  
Jeremy Allison 已提交
882 883 884 885
				return rc;
			}
		}

886 887
		rc = wait_for_response(ses->server, midQ);
		if (rc) {
888
			send_nt_cancel(ses->server, in_buf, midQ);
889 890 891 892 893 894 895 896
			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 已提交
897
		}
898 899 900

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

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

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

914 915 916
	*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);
917
out:
918
	delete_mid(midQ);
J
[CIFS]  
Jeremy Allison 已提交
919 920
	if (rstart && rc == -EACCES)
		return -ERESTARTSYS;
L
Linus Torvalds 已提交
921 922
	return rc;
}