transport.c 23.3 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 29 30 31 32 33 34 35
#include <linux/wait.h>
#include <linux/net.h>
#include <linux/delay.h>
#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"
36

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

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

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

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

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

		/*
		 * 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 已提交
74 75 76 77 78 79 80
	}

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

81
void
L
Linus Torvalds 已提交
82 83
DeleteMidQEntry(struct mid_q_entry *midEntry)
{
S
Steve French 已提交
84 85 86
#ifdef CONFIG_CIFS_STATS2
	unsigned long now;
#endif
L
Linus Torvalds 已提交
87
	midEntry->midState = MID_FREE;
88
	atomic_dec(&midCount);
S
Steve French 已提交
89
	if (midEntry->largeBuf)
90 91 92
		cifs_buf_release(midEntry->resp_buf);
	else
		cifs_small_buf_release(midEntry->resp_buf);
S
Steve French 已提交
93 94 95 96
#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 已提交
97 98
	if ((now - midEntry->when_alloc) > HZ) {
		if ((cifsFYI & CIFS_TIMER) &&
S
Steve French 已提交
99 100 101 102 103 104 105 106 107 108
		   (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 已提交
109 110 111
	mempool_free(midEntry, cifs_mid_poolp);
}

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

	DeleteMidQEntry(mid);
}

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

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

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

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

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

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

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

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

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

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

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

244 245 246 247 248 249 250 251 252 253 254 255
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);
}

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

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

			/* 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)
286
				atomic_inc(&server->inFlight);
287 288
			spin_unlock(&GlobalMid_Lock);
			break;
L
Linus Torvalds 已提交
289 290
		}
	}
J
[CIFS]  
Jeremy Allison 已提交
291 292
	return 0;
}
L
Linus Torvalds 已提交
293

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

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

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

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

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

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

335

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

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

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

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

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

	mid->callback = callback;
	mid->callback_data = cbdata;
	mid->midState = MID_REQUEST_SUBMITTED;
380 381

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

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

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

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

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

421 422 423
	return rc;
}

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

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

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

454
	DeleteMidQEntry(mid);
455 456 457
	return rc;
}

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

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

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

	return rc;
}

495 496 497 498 499 500 501 502
int
cifs_check_receive(struct mid_q_entry *mid, struct TCP_Server_Info *server,
		   bool log_error)
{
	dump_smb(mid->resp_buf,
		 min_t(u32, 92, be32_to_cpu(mid->resp_buf->smb_buf_length)));

	/* convert the length into a more usable form */
503
	if (server->sec_mode & (SECMODE_SIGN_REQUIRED | SECMODE_SIGN_ENABLED)) {
504 505 506 507 508 509 510 511 512 513
		/* FIXME: add code to kill session */
		if (cifs_verify_signature(mid->resp_buf, server,
					  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 已提交
514
int
515
SendReceive2(const unsigned int xid, struct cifs_ses *ses,
S
Steve French 已提交
516
	     struct kvec *iov, int n_vec, int *pRespBufType /* ret */,
517
	     const int flags)
J
[CIFS]  
Jeremy Allison 已提交
518 519
{
	int rc = 0;
520
	int long_op;
J
[CIFS]  
Jeremy Allison 已提交
521 522
	struct mid_q_entry *midQ;
	struct smb_hdr *in_buf = iov[0].iov_base;
523

524 525
	long_op = flags & CIFS_TIMEOUT_MASK;

J
[CIFS]  
Jeremy Allison 已提交
526 527 528 529
	*pRespBufType = CIFS_NO_BUFFER;  /* no response buf yet */

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

S
Steve French 已提交
534
	if (ses->server->tcpStatus == CifsExiting) {
J
[CIFS]  
Jeremy Allison 已提交
535 536 537 538
		cifs_small_buf_release(in_buf);
		return -ENOENT;
	}

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

543
	rc = wait_for_free_request(ses->server, long_op);
J
[CIFS]  
Jeremy Allison 已提交
544 545 546 547 548
	if (rc) {
		cifs_small_buf_release(in_buf);
		return rc;
	}

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

J
Jeff Layton 已提交
553
	mutex_lock(&ses->server->srv_mutex);
J
[CIFS]  
Jeremy Allison 已提交
554 555 556

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

	midQ->midState = MID_REQUEST_SUBMITTED;
572
	cifs_in_send_inc(ses->server);
573
	rc = smb_sendv(ses->server, iov, n_vec);
574 575
	cifs_in_send_dec(ses->server);
	cifs_save_when_sent(midQ);
J
[CIFS]  
Jeremy Allison 已提交
576

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

579 580
	if (rc < 0) {
		cifs_small_buf_release(in_buf);
J
[CIFS]  
Jeremy Allison 已提交
581
		goto out;
582
	}
583

584 585
	if (long_op == CIFS_ASYNC_OP) {
		cifs_small_buf_release(in_buf);
586
		goto out;
587
	}
588

589
	rc = wait_for_response(ses->server, midQ);
590
	if (rc != 0) {
591
		send_nt_cancel(ses->server, in_buf, midQ);
592 593 594 595
		spin_lock(&GlobalMid_Lock);
		if (midQ->midState == MID_REQUEST_SUBMITTED) {
			midQ->callback = DeleteMidQEntry;
			spin_unlock(&GlobalMid_Lock);
596
			cifs_small_buf_release(in_buf);
597 598 599 600 601 602
			atomic_dec(&ses->server->inFlight);
			wake_up(&ses->server->request_q);
			return rc;
		}
		spin_unlock(&GlobalMid_Lock);
	}
603

604 605
	cifs_small_buf_release(in_buf);

606
	rc = cifs_sync_mid_result(midQ, ses->server);
607
	if (rc != 0) {
S
Steve French 已提交
608
		atomic_dec(&ses->server->inFlight);
J
[CIFS]  
Jeremy Allison 已提交
609
		wake_up(&ses->server->request_q);
610 611
		return rc;
	}
612

613
	if (!midQ->resp_buf || midQ->midState != MID_RESPONSE_RECEIVED) {
614
		rc = -EIO;
615
		cFYI(1, "Bad MID state?");
616 617 618
		goto out;
	}

619 620 621 622 623 624
	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;
625

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

628 629 630
	/* 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 已提交
631
out:
632
	delete_mid(midQ);
S
Steve French 已提交
633
	atomic_dec(&ses->server->inFlight);
J
[CIFS]  
Jeremy Allison 已提交
634
	wake_up(&ses->server->request_q);
L
Linus Torvalds 已提交
635

636 637
	return rc;
}
L
Linus Torvalds 已提交
638 639

int
640
SendReceive(const unsigned int xid, struct cifs_ses *ses,
L
Linus Torvalds 已提交
641 642 643 644 645 646 647
	    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) {
648
		cERROR(1, "Null smb session");
L
Linus Torvalds 已提交
649 650
		return -EIO;
	}
S
Steve French 已提交
651
	if (ses->server == NULL) {
652
		cERROR(1, "Null tcp session");
L
Linus Torvalds 已提交
653 654 655
		return -EIO;
	}

S
Steve French 已提交
656
	if (ses->server->tcpStatus == CifsExiting)
657 658
		return -ENOENT;

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

663 664
	if (be32_to_cpu(in_buf->smb_buf_length) > CIFSMaxBufSize +
			MAX_CIFS_HDR_SIZE - 4) {
665
		cERROR(1, "Illegal length, greater than maximum frame, %d",
666
			   be32_to_cpu(in_buf->smb_buf_length));
667 668 669
		return -EIO;
	}

670
	rc = wait_for_free_request(ses->server, long_op);
J
[CIFS]  
Jeremy Allison 已提交
671 672 673
	if (rc)
		return rc;

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

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

J
[CIFS]  
Jeremy Allison 已提交
680 681
	rc = allocate_mid(ses, in_buf, &midQ);
	if (rc) {
J
Jeff Layton 已提交
682
		mutex_unlock(&ses->server->srv_mutex);
J
[CIFS]  
Jeremy Allison 已提交
683
		/* Update # of requests on wire to server */
S
Steve French 已提交
684
		atomic_dec(&ses->server->inFlight);
J
[CIFS]  
Jeremy Allison 已提交
685 686
		wake_up(&ses->server->request_q);
		return rc;
L
Linus Torvalds 已提交
687 688
	}

689
	rc = cifs_sign_smb(in_buf, ses->server, &midQ->sequence_number);
690 691 692 693
	if (rc) {
		mutex_unlock(&ses->server->srv_mutex);
		goto out;
	}
L
Linus Torvalds 已提交
694 695

	midQ->midState = MID_REQUEST_SUBMITTED;
696 697

	cifs_in_send_inc(ses->server);
698
	rc = smb_send(ses->server, in_buf, be32_to_cpu(in_buf->smb_buf_length));
699 700
	cifs_in_send_dec(ses->server);
	cifs_save_when_sent(midQ);
J
Jeff Layton 已提交
701
	mutex_unlock(&ses->server->srv_mutex);
J
[CIFS]  
Jeremy Allison 已提交
702

S
Steve French 已提交
703
	if (rc < 0)
J
[CIFS]  
Jeremy Allison 已提交
704 705
		goto out;

706
	if (long_op == CIFS_ASYNC_OP)
J
[CIFS]  
Jeremy Allison 已提交
707
		goto out;
L
Linus Torvalds 已提交
708

709
	rc = wait_for_response(ses->server, midQ);
710
	if (rc != 0) {
711
		send_nt_cancel(ses->server, in_buf, midQ);
712 713 714 715 716 717 718 719 720 721 722
		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 已提交
723

724
	rc = cifs_sync_mid_result(midQ, ses->server);
725
	if (rc != 0) {
S
Steve French 已提交
726
		atomic_dec(&ses->server->inFlight);
J
[CIFS]  
Jeremy Allison 已提交
727
		wake_up(&ses->server->request_q);
L
Linus Torvalds 已提交
728 729
		return rc;
	}
730

731 732
	if (!midQ->resp_buf || !out_buf ||
	    midQ->midState != MID_RESPONSE_RECEIVED) {
733
		rc = -EIO;
734
		cERROR(1, "Bad MID state?");
735
		goto out;
L
Linus Torvalds 已提交
736
	}
J
[CIFS]  
Jeremy Allison 已提交
737

738 739 740
	*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 已提交
741
out:
742
	delete_mid(midQ);
S
Steve French 已提交
743
	atomic_dec(&ses->server->inFlight);
J
[CIFS]  
Jeremy Allison 已提交
744
	wake_up(&ses->server->request_q);
L
Linus Torvalds 已提交
745

J
[CIFS]  
Jeremy Allison 已提交
746 747
	return rc;
}
L
Linus Torvalds 已提交
748

J
[CIFS]  
Jeremy Allison 已提交
749 750 751 752
/* We send a LOCKINGX_CANCEL_LOCK to cause the Windows
   blocking lock to return. */

static int
753
send_lock_cancel(const unsigned int xid, struct cifs_tcon *tcon,
J
[CIFS]  
Jeremy Allison 已提交
754 755 756 757
			struct smb_hdr *in_buf,
			struct smb_hdr *out_buf)
{
	int bytes_returned;
758
	struct cifs_ses *ses = tcon->ses;
J
[CIFS]  
Jeremy Allison 已提交
759 760 761 762 763 764 765 766 767 768 769 770
	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,
771
			&bytes_returned, 0);
J
[CIFS]  
Jeremy Allison 已提交
772 773 774
}

int
775
SendReceiveBlockingLock(const unsigned int xid, struct cifs_tcon *tcon,
J
[CIFS]  
Jeremy Allison 已提交
776 777 778 779 780 781
	    struct smb_hdr *in_buf, struct smb_hdr *out_buf,
	    int *pbytes_returned)
{
	int rc = 0;
	int rstart = 0;
	struct mid_q_entry *midQ;
782
	struct cifs_ses *ses;
J
[CIFS]  
Jeremy Allison 已提交
783 784

	if (tcon == NULL || tcon->ses == NULL) {
785
		cERROR(1, "Null smb session");
J
[CIFS]  
Jeremy Allison 已提交
786 787 788 789
		return -EIO;
	}
	ses = tcon->ses;

S
Steve French 已提交
790
	if (ses->server == NULL) {
791
		cERROR(1, "Null tcp session");
J
[CIFS]  
Jeremy Allison 已提交
792 793 794
		return -EIO;
	}

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

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

802 803
	if (be32_to_cpu(in_buf->smb_buf_length) > CIFSMaxBufSize +
			MAX_CIFS_HDR_SIZE - 4) {
804
		cERROR(1, "Illegal length, greater than maximum frame, %d",
805
			   be32_to_cpu(in_buf->smb_buf_length));
806 807 808
		return -EIO;
	}

809
	rc = wait_for_free_request(ses->server, CIFS_BLOCKING_OP);
J
[CIFS]  
Jeremy Allison 已提交
810 811 812
	if (rc)
		return rc;

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

J
Jeff Layton 已提交
817
	mutex_lock(&ses->server->srv_mutex);
J
[CIFS]  
Jeremy Allison 已提交
818 819 820

	rc = allocate_mid(ses, in_buf, &midQ);
	if (rc) {
J
Jeff Layton 已提交
821
		mutex_unlock(&ses->server->srv_mutex);
J
[CIFS]  
Jeremy Allison 已提交
822 823 824 825
		return rc;
	}

	rc = cifs_sign_smb(in_buf, ses->server, &midQ->sequence_number);
826
	if (rc) {
827
		delete_mid(midQ);
828 829 830
		mutex_unlock(&ses->server->srv_mutex);
		return rc;
	}
L
Linus Torvalds 已提交
831

J
[CIFS]  
Jeremy Allison 已提交
832
	midQ->midState = MID_REQUEST_SUBMITTED;
833
	cifs_in_send_inc(ses->server);
834
	rc = smb_send(ses->server, in_buf, be32_to_cpu(in_buf->smb_buf_length));
835 836
	cifs_in_send_dec(ses->server);
	cifs_save_when_sent(midQ);
J
Jeff Layton 已提交
837
	mutex_unlock(&ses->server->srv_mutex);
J
[CIFS]  
Jeremy Allison 已提交
838

S
Steve French 已提交
839
	if (rc < 0) {
840
		delete_mid(midQ);
J
[CIFS]  
Jeremy Allison 已提交
841 842 843 844 845
		return rc;
	}

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

878 879
		rc = wait_for_response(ses->server, midQ);
		if (rc) {
880
			send_nt_cancel(ses->server, in_buf, midQ);
881 882 883 884 885 886 887 888
			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 已提交
889
		}
890 891 892

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

895
	rc = cifs_sync_mid_result(midQ, ses->server);
896
	if (rc != 0)
J
[CIFS]  
Jeremy Allison 已提交
897
		return rc;
898

899
	/* rcvd frame is ok */
900
	if (out_buf == NULL || midQ->midState != MID_RESPONSE_RECEIVED) {
901
		rc = -EIO;
902
		cERROR(1, "Bad MID state?");
903 904
		goto out;
	}
L
Linus Torvalds 已提交
905

906 907 908
	*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);
909
out:
910
	delete_mid(midQ);
J
[CIFS]  
Jeremy Allison 已提交
911 912
	if (rstart && rc == -EACCES)
		return -ERESTARTSYS;
L
Linus Torvalds 已提交
913 914
	return rc;
}