transport.c 26.2 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>
30
#include <linux/tcp.h>
31
#include <linux/highmem.h>
L
Linus Torvalds 已提交
32 33 34 35 36 37 38
#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"
39

40 41
void
cifs_wake_up_task(struct mid_q_entry *mid)
42 43 44 45
{
	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
		cifs_dbg(VFS, "Null TCP session in AllocMidQEntry\n");
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
		temp->mid = smb_buffer->Mid;	/* always LE */
		temp->pid = current->pid;
63
		temp->command = cpu_to_le16(smb_buffer->Command);
64
		cifs_dbg(FYI, "For smb_command %d\n", smb_buffer->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
		temp->server = server;
69 70 71 72 73

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

	atomic_inc(&midCount);
79
	temp->mid_state = MID_REQUEST_ALLOCATED;
L
Linus Torvalds 已提交
80 81 82
	return temp;
}

83
void
L
Linus Torvalds 已提交
84 85
DeleteMidQEntry(struct mid_q_entry *midEntry)
{
S
Steve French 已提交
86
#ifdef CONFIG_CIFS_STATS2
87
	__le16 command = midEntry->server->vals->lock_cmd;
S
Steve French 已提交
88 89
	unsigned long now;
#endif
90
	midEntry->mid_state = MID_FREE;
91
	atomic_dec(&midCount);
92
	if (midEntry->large_buf)
93 94 95
		cifs_buf_release(midEntry->resp_buf);
	else
		cifs_small_buf_release(midEntry->resp_buf);
S
Steve French 已提交
96 97 98 99
#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 已提交
100
	if ((now - midEntry->when_alloc) > HZ) {
101
		if ((cifsFYI & CIFS_TIMER) && (midEntry->command != command)) {
102
			printk(KERN_DEBUG " CIFS slow rsp: cmd %d mid %llu",
S
Steve French 已提交
103 104 105 106 107 108 109 110
			       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 已提交
111 112 113
	mempool_free(midEntry, cifs_mid_poolp);
}

P
Pavel Shilovsky 已提交
114 115
void
cifs_delete_mid(struct mid_q_entry *mid)
116 117 118 119 120 121 122 123
{
	spin_lock(&GlobalMid_Lock);
	list_del(&mid->qhead);
	spin_unlock(&GlobalMid_Lock);

	DeleteMidQEntry(mid);
}

124 125 126 127 128 129 130 131 132 133
/*
 * smb_send_kvec - send an array of kvecs to the server
 * @server:	Server to send the data to
 * @iov:	Pointer to array of kvecs
 * @n_vec:	length of kvec array
 * @sent:	amount of data sent on socket is stored here
 *
 * Our basic "send data to server" function. Should be called with srv_mutex
 * held. The caller is responsible for handling the results.
 */
134
static int
135 136
smb_send_kvec(struct TCP_Server_Info *server, struct kvec *iov, size_t n_vec,
		size_t *sent)
L
Linus Torvalds 已提交
137 138 139 140
{
	int rc = 0;
	int i = 0;
	struct msghdr smb_msg;
141 142
	unsigned int remaining;
	size_t first_vec = 0;
143
	struct socket *ssocket = server->ssocket;
144

145 146
	*sent = 0;

147
	smb_msg.msg_name = (struct sockaddr *) &server->dstaddr;
148
	smb_msg.msg_namelen = sizeof(struct sockaddr);
L
Linus Torvalds 已提交
149 150
	smb_msg.msg_control = NULL;
	smb_msg.msg_controllen = 0;
151
	if (server->noblocksnd)
152 153 154
		smb_msg.msg_flags = MSG_DONTWAIT + MSG_NOSIGNAL;
	else
		smb_msg.msg_flags = MSG_NOSIGNAL;
L
Linus Torvalds 已提交
155

156
	remaining = 0;
157
	for (i = 0; i < n_vec; i++)
158
		remaining += iov[i].iov_len;
L
Linus Torvalds 已提交
159

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

S
Steve French 已提交
194
		if (rc < 0)
L
Linus Torvalds 已提交
195
			break;
196

197 198 199 200 201
		/* send was at least partially successful */
		*sent += rc;

		if (rc == remaining) {
			remaining = 0;
S
Steve French 已提交
202
			break;
203 204 205
		}

		if (rc > remaining) {
206
			cifs_dbg(VFS, "sent %d requested %d\n", rc, remaining);
207 208
			break;
		}
209

S
Steve French 已提交
210
		if (rc == 0) {
211 212
			/* should never happen, letting socket clear before
			   retrying is our only obvious option here */
213
			cifs_dbg(VFS, "tcp sent no data\n");
214 215
			msleep(500);
			continue;
216
		}
217 218 219

		remaining -= rc;

220
		/* the line below resets i */
221 222 223 224 225 226 227 228 229 230 231 232
		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;
				}
			}
233
		}
234

235
		i = 0; /* in case we get ENOSPC on the next send */
236
		rc = 0;
L
Linus Torvalds 已提交
237
	}
238 239 240
	return rc;
}

241 242 243 244 245 246 247 248 249 250
/**
 * rqst_page_to_kvec - Turn a slot in the smb_rqst page array into a kvec
 * @rqst: pointer to smb_rqst
 * @idx: index into the array of the page
 * @iov: pointer to struct kvec that will hold the result
 *
 * Helper function to convert a slot in the rqst->rq_pages array into a kvec.
 * The page will be kmapped and the address placed into iov_base. The length
 * will then be adjusted according to the ptailoff.
 */
251
void
252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 272
cifs_rqst_page_to_kvec(struct smb_rqst *rqst, unsigned int idx,
			struct kvec *iov)
{
	/*
	 * FIXME: We could avoid this kmap altogether if we used
	 * kernel_sendpage instead of kernel_sendmsg. That will only
	 * work if signing is disabled though as sendpage inlines the
	 * page directly into the fraglist. If userspace modifies the
	 * page after we calculate the signature, then the server will
	 * reject it and may break the connection. kernel_sendmsg does
	 * an extra copy of the data and avoids that issue.
	 */
	iov->iov_base = kmap(rqst->rq_pages[idx]);

	/* if last page, don't send beyond this offset into page */
	if (idx == (rqst->rq_npages - 1))
		iov->iov_len = rqst->rq_tailsz;
	else
		iov->iov_len = rqst->rq_pagesz;
}

273 274 275 276 277 278 279
static int
smb_send_rqst(struct TCP_Server_Info *server, struct smb_rqst *rqst)
{
	int rc;
	struct kvec *iov = rqst->rq_iov;
	int n_vec = rqst->rq_nvec;
	unsigned int smb_buf_length = get_rfc1002_length(iov[0].iov_base);
280 281
	unsigned int i;
	size_t total_len = 0, sent;
282 283
	struct socket *ssocket = server->ssocket;
	int val = 1;
284

285 286 287
	if (ssocket == NULL)
		return -ENOTSOCK;

288
	cifs_dbg(FYI, "Sending smb: smb_len=%u\n", smb_buf_length);
289 290
	dump_smb(iov[0].iov_base, iov[0].iov_len);

291 292 293 294
	/* cork the socket */
	kernel_setsockopt(ssocket, SOL_TCP, TCP_CORK,
				(char *)&val, sizeof(val));

295 296 297 298 299 300 301 302 303 304 305 306 307 308 309 310 311 312
	rc = smb_send_kvec(server, iov, n_vec, &sent);
	if (rc < 0)
		goto uncork;

	total_len += sent;

	/* now walk the page array and send each page in it */
	for (i = 0; i < rqst->rq_npages; i++) {
		struct kvec p_iov;

		cifs_rqst_page_to_kvec(rqst, i, &p_iov);
		rc = smb_send_kvec(server, &p_iov, 1, &sent);
		kunmap(rqst->rq_pages[i]);
		if (rc < 0)
			break;

		total_len += sent;
	}
L
Linus Torvalds 已提交
313

314
uncork:
315 316 317 318 319
	/* uncork it */
	val = 0;
	kernel_setsockopt(ssocket, SOL_TCP, TCP_CORK,
				(char *)&val, sizeof(val));

320
	if ((total_len > 0) && (total_len != smb_buf_length + 4)) {
321 322
		cifs_dbg(FYI, "partial send (wanted=%u sent=%zu): terminating session\n",
			 smb_buf_length + 4, total_len);
323 324 325 326 327
		/*
		 * 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
		 */
328 329 330
		server->tcpStatus = CifsNeedReconnect;
	}

331
	if (rc < 0 && rc != -EINTR)
332 333
		cifs_dbg(VFS, "Error %d sending data on socket to server\n",
			 rc);
334
	else
L
Linus Torvalds 已提交
335 336 337 338 339
		rc = 0;

	return rc;
}

340 341 342 343 344 345 346 347 348
static int
smb_sendv(struct TCP_Server_Info *server, struct kvec *iov, int n_vec)
{
	struct smb_rqst rqst = { .rq_iov = iov,
				 .rq_nvec = n_vec };

	return smb_send_rqst(server, &rqst);
}

349 350 351 352 353 354 355 356 357 358 359 360
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 已提交
361
static int
362
wait_for_free_credits(struct TCP_Server_Info *server, const int timeout,
363
		      int *credits)
L
Linus Torvalds 已提交
364
{
365 366
	int rc;

P
Pavel Shilovsky 已提交
367
	spin_lock(&server->req_lock);
368
	if (timeout == CIFS_ASYNC_OP) {
L
Linus Torvalds 已提交
369
		/* oplock breaks must not be held up */
P
Pavel Shilovsky 已提交
370
		server->in_flight++;
371
		*credits -= 1;
P
Pavel Shilovsky 已提交
372
		spin_unlock(&server->req_lock);
373 374 375 376
		return 0;
	}

	while (1) {
377
		if (*credits <= 0) {
P
Pavel Shilovsky 已提交
378
			spin_unlock(&server->req_lock);
379
			cifs_num_waiters_inc(server);
380
			rc = wait_event_killable(server->request_q,
381
						 has_credits(server, credits));
382
			cifs_num_waiters_dec(server);
383 384
			if (rc)
				return rc;
P
Pavel Shilovsky 已提交
385
			spin_lock(&server->req_lock);
386
		} else {
387
			if (server->tcpStatus == CifsExiting) {
P
Pavel Shilovsky 已提交
388
				spin_unlock(&server->req_lock);
389
				return -ENOENT;
L
Linus Torvalds 已提交
390
			}
391

392 393 394 395
			/*
			 * Can not count locking commands against total
			 * as they are allowed to block on server.
			 */
396 397

			/* update # of requests on the wire to server */
398
			if (timeout != CIFS_BLOCKING_OP) {
399
				*credits -= 1;
P
Pavel Shilovsky 已提交
400
				server->in_flight++;
401
			}
P
Pavel Shilovsky 已提交
402
			spin_unlock(&server->req_lock);
403
			break;
L
Linus Torvalds 已提交
404 405
		}
	}
J
[CIFS]  
Jeremy Allison 已提交
406 407
	return 0;
}
L
Linus Torvalds 已提交
408

409
static int
410 411
wait_for_free_request(struct TCP_Server_Info *server, const int timeout,
		      const int optype)
412
{
413 414
	return wait_for_free_credits(server, timeout,
				server->ops->get_credits_field(server, optype));
415 416
}

417
static int allocate_mid(struct cifs_ses *ses, struct smb_hdr *in_buf,
J
[CIFS]  
Jeremy Allison 已提交
418 419
			struct mid_q_entry **ppmidQ)
{
L
Linus Torvalds 已提交
420
	if (ses->server->tcpStatus == CifsExiting) {
J
[CIFS]  
Jeremy Allison 已提交
421
		return -ENOENT;
422 423 424
	}

	if (ses->server->tcpStatus == CifsNeedReconnect) {
425
		cifs_dbg(FYI, "tcp session dead - return to caller to retry\n");
J
[CIFS]  
Jeremy Allison 已提交
426
		return -EAGAIN;
427 428 429
	}

	if (ses->status != CifsGood) {
L
Linus Torvalds 已提交
430
		/* check if SMB session is bad because we are setting it up */
S
Steve French 已提交
431
		if ((in_buf->Command != SMB_COM_SESSION_SETUP_ANDX) &&
S
Steve French 已提交
432
			(in_buf->Command != SMB_COM_NEGOTIATE))
J
[CIFS]  
Jeremy Allison 已提交
433
			return -EAGAIN;
S
Steve French 已提交
434
		/* else ok - we are setting up session */
L
Linus Torvalds 已提交
435
	}
436
	*ppmidQ = AllocMidQEntry(in_buf, ses->server);
437
	if (*ppmidQ == NULL)
J
[CIFS]  
Jeremy Allison 已提交
438
		return -ENOMEM;
439 440 441
	spin_lock(&GlobalMid_Lock);
	list_add_tail(&(*ppmidQ)->qhead, &ses->server->pending_mid_q);
	spin_unlock(&GlobalMid_Lock);
J
[CIFS]  
Jeremy Allison 已提交
442 443 444
	return 0;
}

445 446
static int
wait_for_response(struct TCP_Server_Info *server, struct mid_q_entry *midQ)
J
[CIFS]  
Jeremy Allison 已提交
447
{
448
	int error;
J
[CIFS]  
Jeremy Allison 已提交
449

450
	error = wait_event_freezekillable(server->response_q,
451
				    midQ->mid_state != MID_REQUEST_SUBMITTED);
452 453
	if (error < 0)
		return -ERESTARTSYS;
J
[CIFS]  
Jeremy Allison 已提交
454

455
	return 0;
J
[CIFS]  
Jeremy Allison 已提交
456 457
}

458 459
struct mid_q_entry *
cifs_setup_async_request(struct TCP_Server_Info *server, struct smb_rqst *rqst)
460 461
{
	int rc;
462
	struct smb_hdr *hdr = (struct smb_hdr *)rqst->rq_iov[0].iov_base;
463 464 465 466 467 468 469 470
	struct mid_q_entry *mid;

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

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

473
	rc = cifs_sign_rqst(rqst, server, &mid->sequence_number);
474 475
	if (rc) {
		DeleteMidQEntry(mid);
476
		return ERR_PTR(rc);
477 478
	}

479
	return mid;
480
}
481

J
Jeff Layton 已提交
482 483 484 485 486
/*
 * Send a SMB request and set the callback function in the mid to handle
 * the result. Caller is responsible for dealing with timeouts.
 */
int
487 488 489
cifs_call_async(struct TCP_Server_Info *server, struct smb_rqst *rqst,
		mid_receive_t *receive, mid_callback_t *callback,
		void *cbdata, const int flags)
J
Jeff Layton 已提交
490
{
491
	int rc, timeout, optype;
J
Jeff Layton 已提交
492 493
	struct mid_q_entry *mid;

494 495 496 497
	timeout = flags & CIFS_TIMEOUT_MASK;
	optype = flags & CIFS_OP_MASK;

	rc = wait_for_free_request(server, timeout, optype);
J
Jeff Layton 已提交
498 499 500 501
	if (rc)
		return rc;

	mutex_lock(&server->srv_mutex);
502 503
	mid = server->ops->setup_async_request(server, rqst);
	if (IS_ERR(mid)) {
J
Jeff Layton 已提交
504
		mutex_unlock(&server->srv_mutex);
505
		add_credits(server, 1, optype);
506
		wake_up(&server->request_q);
507
		return PTR_ERR(mid);
J
Jeff Layton 已提交
508 509
	}

510
	mid->receive = receive;
J
Jeff Layton 已提交
511 512
	mid->callback = callback;
	mid->callback_data = cbdata;
513
	mid->mid_state = MID_REQUEST_SUBMITTED;
514

515 516 517 518 519 520
	/* put it on the pending_mid_q */
	spin_lock(&GlobalMid_Lock);
	list_add_tail(&mid->qhead, &server->pending_mid_q);
	spin_unlock(&GlobalMid_Lock);


521
	cifs_in_send_inc(server);
522
	rc = smb_send_rqst(server, rqst);
523 524
	cifs_in_send_dec(server);
	cifs_save_when_sent(mid);
J
Jeff Layton 已提交
525
	mutex_unlock(&server->srv_mutex);
526

527 528
	if (rc == 0)
		return 0;
J
Jeff Layton 已提交
529

P
Pavel Shilovsky 已提交
530
	cifs_delete_mid(mid);
531
	add_credits(server, 1, optype);
J
Jeff Layton 已提交
532 533 534 535
	wake_up(&server->request_q);
	return rc;
}

536 537 538 539 540 541 542 543 544 545
/*
 *
 * 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
546
SendReceiveNoRsp(const unsigned int xid, struct cifs_ses *ses,
547
		 char *in_buf, int flags)
548 549 550 551 552
{
	int rc;
	struct kvec iov[1];
	int resp_buf_type;

553 554
	iov[0].iov_base = in_buf;
	iov[0].iov_len = get_rfc1002_length(in_buf) + 4;
555 556
	flags |= CIFS_NO_RESP;
	rc = SendReceive2(xid, ses, iov, 1, &resp_buf_type, flags);
557
	cifs_dbg(NOISY, "SendRcvNoRsp flags %d rc %d\n", flags, rc);
558

559 560 561
	return rc;
}

562
static int
563
cifs_sync_mid_result(struct mid_q_entry *mid, struct TCP_Server_Info *server)
564 565 566
{
	int rc = 0;

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

J
Jeff Layton 已提交
570
	spin_lock(&GlobalMid_Lock);
571
	switch (mid->mid_state) {
J
Jeff Layton 已提交
572
	case MID_RESPONSE_RECEIVED:
573 574
		spin_unlock(&GlobalMid_Lock);
		return rc;
J
Jeff Layton 已提交
575 576 577
	case MID_RETRY_NEEDED:
		rc = -EAGAIN;
		break;
578 579 580
	case MID_RESPONSE_MALFORMED:
		rc = -EIO;
		break;
581 582 583
	case MID_SHUTDOWN:
		rc = -EHOSTDOWN;
		break;
J
Jeff Layton 已提交
584
	default:
585
		list_del_init(&mid->qhead);
586 587
		cifs_dbg(VFS, "%s: invalid mid state mid=%llu state=%d\n",
			 __func__, mid->mid, mid->mid_state);
J
Jeff Layton 已提交
588
		rc = -EIO;
589 590 591
	}
	spin_unlock(&GlobalMid_Lock);

592
	DeleteMidQEntry(mid);
593 594 595
	return rc;
}

596 597
static inline int
send_cancel(struct TCP_Server_Info *server, void *buf, struct mid_q_entry *mid)
598
{
599 600
	return server->ops->send_cancel ?
				server->ops->send_cancel(server, buf, mid) : 0;
601 602
}

603 604 605 606
int
cifs_check_receive(struct mid_q_entry *mid, struct TCP_Server_Info *server,
		   bool log_error)
{
607
	unsigned int len = get_rfc1002_length(mid->resp_buf) + 4;
608 609

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

	/* convert the length into a more usable form */
612
	if (server->sec_mode & (SECMODE_SIGN_REQUIRED | SECMODE_SIGN_ENABLED)) {
613
		struct kvec iov;
614
		int rc = 0;
615 616
		struct smb_rqst rqst = { .rq_iov = &iov,
					 .rq_nvec = 1 };
617 618 619

		iov.iov_base = mid->resp_buf;
		iov.iov_len = len;
620
		/* FIXME: add code to kill session */
621
		rc = cifs_verify_signature(&rqst, server,
622 623
					   mid->sequence_number + 1);
		if (rc)
624 625
			cifs_dbg(VFS, "SMB signature verification returned error = %d\n",
				 rc);
626 627 628 629 630 631
	}

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

632 633
struct mid_q_entry *
cifs_setup_request(struct cifs_ses *ses, struct smb_rqst *rqst)
634 635
{
	int rc;
636
	struct smb_hdr *hdr = (struct smb_hdr *)rqst->rq_iov[0].iov_base;
637 638 639 640
	struct mid_q_entry *mid;

	rc = allocate_mid(ses, hdr, &mid);
	if (rc)
641 642 643
		return ERR_PTR(rc);
	rc = cifs_sign_rqst(rqst, ses->server, &mid->sequence_number);
	if (rc) {
P
Pavel Shilovsky 已提交
644
		cifs_delete_mid(mid);
645 646 647
		return ERR_PTR(rc);
	}
	return mid;
648 649
}

J
[CIFS]  
Jeremy Allison 已提交
650
int
651
SendReceive2(const unsigned int xid, struct cifs_ses *ses,
652
	     struct kvec *iov, int n_vec, int *resp_buf_type /* ret */,
653
	     const int flags)
J
[CIFS]  
Jeremy Allison 已提交
654 655
{
	int rc = 0;
656
	int timeout, optype;
J
[CIFS]  
Jeremy Allison 已提交
657
	struct mid_q_entry *midQ;
658
	char *buf = iov[0].iov_base;
659
	unsigned int credits = 1;
660 661
	struct smb_rqst rqst = { .rq_iov = iov,
				 .rq_nvec = n_vec };
662

663 664
	timeout = flags & CIFS_TIMEOUT_MASK;
	optype = flags & CIFS_OP_MASK;
665

666
	*resp_buf_type = CIFS_NO_BUFFER;  /* no response buf yet */
J
[CIFS]  
Jeremy Allison 已提交
667 668

	if ((ses == NULL) || (ses->server == NULL)) {
669
		cifs_small_buf_release(buf);
670
		cifs_dbg(VFS, "Null session\n");
J
[CIFS]  
Jeremy Allison 已提交
671 672 673
		return -EIO;
	}

S
Steve French 已提交
674
	if (ses->server->tcpStatus == CifsExiting) {
675
		cifs_small_buf_release(buf);
J
[CIFS]  
Jeremy Allison 已提交
676 677 678
		return -ENOENT;
	}

679 680 681 682 683
	/*
	 * Ensure that we do not send more than 50 overlapping requests
	 * to the same server. We may make this configurable later or
	 * use ses->maxReq.
	 */
J
[CIFS]  
Jeremy Allison 已提交
684

685
	rc = wait_for_free_request(ses->server, timeout, optype);
J
[CIFS]  
Jeremy Allison 已提交
686
	if (rc) {
687
		cifs_small_buf_release(buf);
J
[CIFS]  
Jeremy Allison 已提交
688 689 690
		return rc;
	}

691 692 693 694 695
	/*
	 * Make sure that we sign in the same order that we send on this socket
	 * and avoid races inside tcp sendmsg code that could cause corruption
	 * of smb data.
	 */
J
[CIFS]  
Jeremy Allison 已提交
696

J
Jeff Layton 已提交
697
	mutex_lock(&ses->server->srv_mutex);
J
[CIFS]  
Jeremy Allison 已提交
698

699 700
	midQ = ses->server->ops->setup_request(ses, &rqst);
	if (IS_ERR(midQ)) {
J
Jeff Layton 已提交
701
		mutex_unlock(&ses->server->srv_mutex);
702
		cifs_small_buf_release(buf);
J
[CIFS]  
Jeremy Allison 已提交
703
		/* Update # of requests on wire to server */
704
		add_credits(ses->server, 1, optype);
705
		return PTR_ERR(midQ);
L
Linus Torvalds 已提交
706 707
	}

708
	midQ->mid_state = MID_REQUEST_SUBMITTED;
709
	cifs_in_send_inc(ses->server);
710
	rc = smb_sendv(ses->server, iov, n_vec);
711 712
	cifs_in_send_dec(ses->server);
	cifs_save_when_sent(midQ);
J
[CIFS]  
Jeremy Allison 已提交
713

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

716
	if (rc < 0) {
717
		cifs_small_buf_release(buf);
J
[CIFS]  
Jeremy Allison 已提交
718
		goto out;
719
	}
720

721
	if (timeout == CIFS_ASYNC_OP) {
722
		cifs_small_buf_release(buf);
723
		goto out;
724
	}
725

726
	rc = wait_for_response(ses->server, midQ);
727
	if (rc != 0) {
728
		send_cancel(ses->server, buf, midQ);
729
		spin_lock(&GlobalMid_Lock);
730
		if (midQ->mid_state == MID_REQUEST_SUBMITTED) {
731 732
			midQ->callback = DeleteMidQEntry;
			spin_unlock(&GlobalMid_Lock);
733
			cifs_small_buf_release(buf);
734
			add_credits(ses->server, 1, optype);
735 736 737 738
			return rc;
		}
		spin_unlock(&GlobalMid_Lock);
	}
739

740
	cifs_small_buf_release(buf);
741

742
	rc = cifs_sync_mid_result(midQ, ses->server);
743
	if (rc != 0) {
744
		add_credits(ses->server, 1, optype);
745 746
		return rc;
	}
747

748
	if (!midQ->resp_buf || midQ->mid_state != MID_RESPONSE_RECEIVED) {
749
		rc = -EIO;
750
		cifs_dbg(FYI, "Bad MID state?\n");
751 752 753
		goto out;
	}

754 755 756
	buf = (char *)midQ->resp_buf;
	iov[0].iov_base = buf;
	iov[0].iov_len = get_rfc1002_length(buf) + 4;
757
	if (midQ->large_buf)
758
		*resp_buf_type = CIFS_LARGE_BUFFER;
759
	else
760 761 762
		*resp_buf_type = CIFS_SMALL_BUFFER;

	credits = ses->server->ops->get_credits(midQ);
763

764 765
	rc = ses->server->ops->check_receive(midQ, ses->server,
					     flags & CIFS_LOG_ERROR);
L
Linus Torvalds 已提交
766

P
Pavel Shilovsky 已提交
767
	/* mark it so buf will not be freed by cifs_delete_mid */
768 769
	if ((flags & CIFS_NO_RESP) == 0)
		midQ->resp_buf = NULL;
J
[CIFS]  
Jeremy Allison 已提交
770
out:
P
Pavel Shilovsky 已提交
771
	cifs_delete_mid(midQ);
772
	add_credits(ses->server, credits, optype);
L
Linus Torvalds 已提交
773

774 775
	return rc;
}
L
Linus Torvalds 已提交
776 777

int
778
SendReceive(const unsigned int xid, struct cifs_ses *ses,
L
Linus Torvalds 已提交
779
	    struct smb_hdr *in_buf, struct smb_hdr *out_buf,
780
	    int *pbytes_returned, const int timeout)
L
Linus Torvalds 已提交
781 782 783 784 785
{
	int rc = 0;
	struct mid_q_entry *midQ;

	if (ses == NULL) {
786
		cifs_dbg(VFS, "Null smb session\n");
L
Linus Torvalds 已提交
787 788
		return -EIO;
	}
S
Steve French 已提交
789
	if (ses->server == NULL) {
790
		cifs_dbg(VFS, "Null tcp session\n");
L
Linus Torvalds 已提交
791 792 793
		return -EIO;
	}

S
Steve French 已提交
794
	if (ses->server->tcpStatus == CifsExiting)
795 796
		return -ENOENT;

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

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

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

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

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

J
[CIFS]  
Jeremy Allison 已提交
818 819
	rc = allocate_mid(ses, in_buf, &midQ);
	if (rc) {
J
Jeff Layton 已提交
820
		mutex_unlock(&ses->server->srv_mutex);
J
[CIFS]  
Jeremy Allison 已提交
821
		/* Update # of requests on wire to server */
822
		add_credits(ses->server, 1, 0);
J
[CIFS]  
Jeremy Allison 已提交
823
		return rc;
L
Linus Torvalds 已提交
824 825
	}

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

832
	midQ->mid_state = MID_REQUEST_SUBMITTED;
833 834

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

S
Steve French 已提交
840
	if (rc < 0)
J
[CIFS]  
Jeremy Allison 已提交
841 842
		goto out;

843
	if (timeout == CIFS_ASYNC_OP)
J
[CIFS]  
Jeremy Allison 已提交
844
		goto out;
L
Linus Torvalds 已提交
845

846
	rc = wait_for_response(ses->server, midQ);
847
	if (rc != 0) {
848
		send_cancel(ses->server, in_buf, midQ);
849
		spin_lock(&GlobalMid_Lock);
850
		if (midQ->mid_state == MID_REQUEST_SUBMITTED) {
851 852 853
			/* no longer considered to be "in-flight" */
			midQ->callback = DeleteMidQEntry;
			spin_unlock(&GlobalMid_Lock);
854
			add_credits(ses->server, 1, 0);
855 856 857 858
			return rc;
		}
		spin_unlock(&GlobalMid_Lock);
	}
L
Linus Torvalds 已提交
859

860
	rc = cifs_sync_mid_result(midQ, ses->server);
861
	if (rc != 0) {
862
		add_credits(ses->server, 1, 0);
L
Linus Torvalds 已提交
863 864
		return rc;
	}
865

866
	if (!midQ->resp_buf || !out_buf ||
867
	    midQ->mid_state != MID_RESPONSE_RECEIVED) {
868
		rc = -EIO;
869
		cifs_dbg(VFS, "Bad MID state?\n");
870
		goto out;
L
Linus Torvalds 已提交
871
	}
J
[CIFS]  
Jeremy Allison 已提交
872

873
	*pbytes_returned = get_rfc1002_length(midQ->resp_buf);
874 875
	memcpy(out_buf, midQ->resp_buf, *pbytes_returned + 4);
	rc = cifs_check_receive(midQ, ses->server, 0);
J
[CIFS]  
Jeremy Allison 已提交
876
out:
P
Pavel Shilovsky 已提交
877
	cifs_delete_mid(midQ);
878
	add_credits(ses->server, 1, 0);
L
Linus Torvalds 已提交
879

J
[CIFS]  
Jeremy Allison 已提交
880 881
	return rc;
}
L
Linus Torvalds 已提交
882

J
[CIFS]  
Jeremy Allison 已提交
883 884 885 886
/* We send a LOCKINGX_CANCEL_LOCK to cause the Windows
   blocking lock to return. */

static int
887
send_lock_cancel(const unsigned int xid, struct cifs_tcon *tcon,
J
[CIFS]  
Jeremy Allison 已提交
888 889 890 891
			struct smb_hdr *in_buf,
			struct smb_hdr *out_buf)
{
	int bytes_returned;
892
	struct cifs_ses *ses = tcon->ses;
J
[CIFS]  
Jeremy Allison 已提交
893 894 895 896 897 898 899 900 901
	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;
902
	pSMB->hdr.Mid = get_next_mid(ses->server);
J
[CIFS]  
Jeremy Allison 已提交
903 904

	return SendReceive(xid, ses, in_buf, out_buf,
905
			&bytes_returned, 0);
J
[CIFS]  
Jeremy Allison 已提交
906 907 908
}

int
909
SendReceiveBlockingLock(const unsigned int xid, struct cifs_tcon *tcon,
J
[CIFS]  
Jeremy Allison 已提交
910 911 912 913 914 915
	    struct smb_hdr *in_buf, struct smb_hdr *out_buf,
	    int *pbytes_returned)
{
	int rc = 0;
	int rstart = 0;
	struct mid_q_entry *midQ;
916
	struct cifs_ses *ses;
J
[CIFS]  
Jeremy Allison 已提交
917 918

	if (tcon == NULL || tcon->ses == NULL) {
919
		cifs_dbg(VFS, "Null smb session\n");
J
[CIFS]  
Jeremy Allison 已提交
920 921 922 923
		return -EIO;
	}
	ses = tcon->ses;

S
Steve French 已提交
924
	if (ses->server == NULL) {
925
		cifs_dbg(VFS, "Null tcp session\n");
J
[CIFS]  
Jeremy Allison 已提交
926 927 928
		return -EIO;
	}

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

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

936 937
	if (be32_to_cpu(in_buf->smb_buf_length) > CIFSMaxBufSize +
			MAX_CIFS_HDR_SIZE - 4) {
938 939
		cifs_dbg(VFS, "Illegal length, greater than maximum frame, %d\n",
			 be32_to_cpu(in_buf->smb_buf_length));
940 941 942
		return -EIO;
	}

943
	rc = wait_for_free_request(ses->server, CIFS_BLOCKING_OP, 0);
J
[CIFS]  
Jeremy Allison 已提交
944 945 946
	if (rc)
		return rc;

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

J
Jeff Layton 已提交
951
	mutex_lock(&ses->server->srv_mutex);
J
[CIFS]  
Jeremy Allison 已提交
952 953 954

	rc = allocate_mid(ses, in_buf, &midQ);
	if (rc) {
J
Jeff Layton 已提交
955
		mutex_unlock(&ses->server->srv_mutex);
J
[CIFS]  
Jeremy Allison 已提交
956 957 958 959
		return rc;
	}

	rc = cifs_sign_smb(in_buf, ses->server, &midQ->sequence_number);
960
	if (rc) {
P
Pavel Shilovsky 已提交
961
		cifs_delete_mid(midQ);
962 963 964
		mutex_unlock(&ses->server->srv_mutex);
		return rc;
	}
L
Linus Torvalds 已提交
965

966
	midQ->mid_state = MID_REQUEST_SUBMITTED;
967
	cifs_in_send_inc(ses->server);
968
	rc = smb_send(ses->server, in_buf, be32_to_cpu(in_buf->smb_buf_length));
969 970
	cifs_in_send_dec(ses->server);
	cifs_save_when_sent(midQ);
J
Jeff Layton 已提交
971
	mutex_unlock(&ses->server->srv_mutex);
J
[CIFS]  
Jeremy Allison 已提交
972

S
Steve French 已提交
973
	if (rc < 0) {
P
Pavel Shilovsky 已提交
974
		cifs_delete_mid(midQ);
J
[CIFS]  
Jeremy Allison 已提交
975 976 977 978 979
		return rc;
	}

	/* Wait for a reply - allow signals to interrupt. */
	rc = wait_event_interruptible(ses->server->response_q,
980
		(!(midQ->mid_state == MID_REQUEST_SUBMITTED)) ||
J
[CIFS]  
Jeremy Allison 已提交
981 982 983 984 985
		((ses->server->tcpStatus != CifsGood) &&
		 (ses->server->tcpStatus != CifsNew)));

	/* Were we interrupted by a signal ? */
	if ((rc == -ERESTARTSYS) &&
986
		(midQ->mid_state == MID_REQUEST_SUBMITTED) &&
J
[CIFS]  
Jeremy Allison 已提交
987 988 989 990 991 992
		((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. */
993
			rc = send_cancel(ses->server, in_buf, midQ);
J
[CIFS]  
Jeremy Allison 已提交
994
			if (rc) {
P
Pavel Shilovsky 已提交
995
				cifs_delete_mid(midQ);
J
[CIFS]  
Jeremy Allison 已提交
996 997 998 999 1000 1001 1002 1003 1004 1005 1006
				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) {
P
Pavel Shilovsky 已提交
1007
				cifs_delete_mid(midQ);
J
[CIFS]  
Jeremy Allison 已提交
1008 1009 1010 1011
				return rc;
			}
		}

1012 1013
		rc = wait_for_response(ses->server, midQ);
		if (rc) {
1014
			send_cancel(ses->server, in_buf, midQ);
1015
			spin_lock(&GlobalMid_Lock);
1016
			if (midQ->mid_state == MID_REQUEST_SUBMITTED) {
1017 1018 1019 1020 1021 1022
				/* no longer considered to be "in-flight" */
				midQ->callback = DeleteMidQEntry;
				spin_unlock(&GlobalMid_Lock);
				return rc;
			}
			spin_unlock(&GlobalMid_Lock);
J
[CIFS]  
Jeremy Allison 已提交
1023
		}
1024 1025 1026

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

1029
	rc = cifs_sync_mid_result(midQ, ses->server);
1030
	if (rc != 0)
J
[CIFS]  
Jeremy Allison 已提交
1031
		return rc;
1032

1033
	/* rcvd frame is ok */
1034
	if (out_buf == NULL || midQ->mid_state != MID_RESPONSE_RECEIVED) {
1035
		rc = -EIO;
1036
		cifs_dbg(VFS, "Bad MID state?\n");
1037 1038
		goto out;
	}
L
Linus Torvalds 已提交
1039

1040
	*pbytes_returned = get_rfc1002_length(midQ->resp_buf);
1041 1042
	memcpy(out_buf, midQ->resp_buf, *pbytes_returned + 4);
	rc = cifs_check_receive(midQ, ses->server, 0);
1043
out:
P
Pavel Shilovsky 已提交
1044
	cifs_delete_mid(midQ);
J
[CIFS]  
Jeremy Allison 已提交
1045 1046
	if (rstart && rc == -EACCES)
		return -ERESTARTSYS;
L
Linus Torvalds 已提交
1047 1048
	return rc;
}