| Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 1 | /* | 
 | 2 |  *  linux/fs/ncpfs/sock.c | 
 | 3 |  * | 
 | 4 |  *  Copyright (C) 1992, 1993  Rick Sladkey | 
 | 5 |  * | 
 | 6 |  *  Modified 1995, 1996 by Volker Lendecke to be usable for ncp | 
 | 7 |  *  Modified 1997 Peter Waltenberg, Bill Hawes, David Woodhouse for 2.1 dcache | 
 | 8 |  * | 
 | 9 |  */ | 
 | 10 |  | 
| Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 11 |  | 
 | 12 | #include <linux/time.h> | 
 | 13 | #include <linux/errno.h> | 
 | 14 | #include <linux/socket.h> | 
 | 15 | #include <linux/fcntl.h> | 
 | 16 | #include <linux/stat.h> | 
| Pierre Ossman | c5f93cf | 2007-02-19 11:34:43 +0100 | [diff] [blame] | 17 | #include <linux/string.h> | 
| Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 18 | #include <asm/uaccess.h> | 
 | 19 | #include <linux/in.h> | 
 | 20 | #include <linux/net.h> | 
 | 21 | #include <linux/mm.h> | 
 | 22 | #include <linux/netdevice.h> | 
 | 23 | #include <linux/signal.h> | 
| Tejun Heo | 5a0e3ad | 2010-03-24 17:04:11 +0900 | [diff] [blame] | 24 | #include <linux/slab.h> | 
| Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 25 | #include <net/scm.h> | 
 | 26 | #include <net/sock.h> | 
 | 27 | #include <linux/ipx.h> | 
 | 28 | #include <linux/poll.h> | 
 | 29 | #include <linux/file.h> | 
 | 30 |  | 
 | 31 | #include <linux/ncp_fs.h> | 
 | 32 |  | 
 | 33 | #include "ncpsign_kernel.h" | 
 | 34 |  | 
 | 35 | static int _recv(struct socket *sock, void *buf, int size, unsigned flags) | 
 | 36 | { | 
 | 37 | 	struct msghdr msg = {NULL, }; | 
 | 38 | 	struct kvec iov = {buf, size}; | 
 | 39 | 	return kernel_recvmsg(sock, &msg, &iov, 1, size, flags); | 
 | 40 | } | 
 | 41 |  | 
 | 42 | static inline int do_send(struct socket *sock, struct kvec *vec, int count, | 
 | 43 | 			  int len, unsigned flags) | 
 | 44 | { | 
 | 45 | 	struct msghdr msg = { .msg_flags = flags }; | 
 | 46 | 	return kernel_sendmsg(sock, &msg, vec, count, len); | 
 | 47 | } | 
 | 48 |  | 
 | 49 | static int _send(struct socket *sock, const void *buff, int len) | 
 | 50 | { | 
 | 51 | 	struct kvec vec; | 
 | 52 | 	vec.iov_base = (void *) buff; | 
 | 53 | 	vec.iov_len = len; | 
 | 54 | 	return do_send(sock, &vec, 1, len, 0); | 
 | 55 | } | 
 | 56 |  | 
 | 57 | struct ncp_request_reply { | 
 | 58 | 	struct list_head req; | 
 | 59 | 	wait_queue_head_t wq; | 
| Pierre Ossman | c5f93cf | 2007-02-19 11:34:43 +0100 | [diff] [blame] | 60 | 	atomic_t refs; | 
 | 61 | 	unsigned char* reply_buf; | 
| Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 62 | 	size_t datalen; | 
 | 63 | 	int result; | 
| Pierre Ossman | c5f93cf | 2007-02-19 11:34:43 +0100 | [diff] [blame] | 64 | 	enum { RQ_DONE, RQ_INPROGRESS, RQ_QUEUED, RQ_IDLE, RQ_ABANDONED } status; | 
| Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 65 | 	struct kvec* tx_ciov; | 
 | 66 | 	size_t tx_totallen; | 
 | 67 | 	size_t tx_iovlen; | 
 | 68 | 	struct kvec tx_iov[3]; | 
 | 69 | 	u_int16_t tx_type; | 
 | 70 | 	u_int32_t sign[6]; | 
 | 71 | }; | 
 | 72 |  | 
| Pierre Ossman | c5f93cf | 2007-02-19 11:34:43 +0100 | [diff] [blame] | 73 | static inline struct ncp_request_reply* ncp_alloc_req(void) | 
 | 74 | { | 
 | 75 | 	struct ncp_request_reply *req; | 
 | 76 |  | 
 | 77 | 	req = kmalloc(sizeof(struct ncp_request_reply), GFP_KERNEL); | 
 | 78 | 	if (!req) | 
 | 79 | 		return NULL; | 
 | 80 |  | 
 | 81 | 	init_waitqueue_head(&req->wq); | 
 | 82 | 	atomic_set(&req->refs, (1)); | 
 | 83 | 	req->status = RQ_IDLE; | 
 | 84 |  | 
 | 85 | 	return req; | 
 | 86 | } | 
 | 87 |  | 
 | 88 | static void ncp_req_get(struct ncp_request_reply *req) | 
 | 89 | { | 
 | 90 | 	atomic_inc(&req->refs); | 
 | 91 | } | 
 | 92 |  | 
 | 93 | static void ncp_req_put(struct ncp_request_reply *req) | 
 | 94 | { | 
 | 95 | 	if (atomic_dec_and_test(&req->refs)) | 
 | 96 | 		kfree(req); | 
 | 97 | } | 
 | 98 |  | 
| Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 99 | void ncp_tcp_data_ready(struct sock *sk, int len) | 
 | 100 | { | 
 | 101 | 	struct ncp_server *server = sk->sk_user_data; | 
 | 102 |  | 
 | 103 | 	server->data_ready(sk, len); | 
 | 104 | 	schedule_work(&server->rcv.tq); | 
 | 105 | } | 
 | 106 |  | 
 | 107 | void ncp_tcp_error_report(struct sock *sk) | 
 | 108 | { | 
 | 109 | 	struct ncp_server *server = sk->sk_user_data; | 
 | 110 | 	 | 
 | 111 | 	server->error_report(sk); | 
 | 112 | 	schedule_work(&server->rcv.tq); | 
 | 113 | } | 
 | 114 |  | 
 | 115 | void ncp_tcp_write_space(struct sock *sk) | 
 | 116 | { | 
 | 117 | 	struct ncp_server *server = sk->sk_user_data; | 
 | 118 | 	 | 
 | 119 | 	/* We do not need any locking: we first set tx.creq, and then we do sendmsg, | 
 | 120 | 	   not vice versa... */ | 
 | 121 | 	server->write_space(sk); | 
 | 122 | 	if (server->tx.creq) | 
 | 123 | 		schedule_work(&server->tx.tq); | 
 | 124 | } | 
 | 125 |  | 
 | 126 | void ncpdgram_timeout_call(unsigned long v) | 
 | 127 | { | 
 | 128 | 	struct ncp_server *server = (void*)v; | 
 | 129 | 	 | 
 | 130 | 	schedule_work(&server->timeout_tq); | 
 | 131 | } | 
 | 132 |  | 
| Pierre Ossman | c5f93cf | 2007-02-19 11:34:43 +0100 | [diff] [blame] | 133 | static inline void ncp_finish_request(struct ncp_server *server, struct ncp_request_reply *req, int result) | 
| Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 134 | { | 
 | 135 | 	req->result = result; | 
| Pierre Ossman | c5f93cf | 2007-02-19 11:34:43 +0100 | [diff] [blame] | 136 | 	if (req->status != RQ_ABANDONED) | 
 | 137 | 		memcpy(req->reply_buf, server->rxbuf, req->datalen); | 
| Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 138 | 	req->status = RQ_DONE; | 
 | 139 | 	wake_up_all(&req->wq); | 
| Pierre Ossman | c5f93cf | 2007-02-19 11:34:43 +0100 | [diff] [blame] | 140 | 	ncp_req_put(req); | 
| Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 141 | } | 
 | 142 |  | 
| Pierre Ossman | c5f93cf | 2007-02-19 11:34:43 +0100 | [diff] [blame] | 143 | static void __abort_ncp_connection(struct ncp_server *server) | 
| Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 144 | { | 
 | 145 | 	struct ncp_request_reply *req; | 
 | 146 |  | 
 | 147 | 	ncp_invalidate_conn(server); | 
 | 148 | 	del_timer(&server->timeout_tm); | 
 | 149 | 	while (!list_empty(&server->tx.requests)) { | 
 | 150 | 		req = list_entry(server->tx.requests.next, struct ncp_request_reply, req); | 
 | 151 | 		 | 
 | 152 | 		list_del_init(&req->req); | 
| Pierre Ossman | c5f93cf | 2007-02-19 11:34:43 +0100 | [diff] [blame] | 153 | 		ncp_finish_request(server, req, -EIO); | 
| Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 154 | 	} | 
 | 155 | 	req = server->rcv.creq; | 
 | 156 | 	if (req) { | 
 | 157 | 		server->rcv.creq = NULL; | 
| Pierre Ossman | c5f93cf | 2007-02-19 11:34:43 +0100 | [diff] [blame] | 158 | 		ncp_finish_request(server, req, -EIO); | 
| Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 159 | 		server->rcv.ptr = NULL; | 
 | 160 | 		server->rcv.state = 0; | 
 | 161 | 	} | 
 | 162 | 	req = server->tx.creq; | 
 | 163 | 	if (req) { | 
 | 164 | 		server->tx.creq = NULL; | 
| Pierre Ossman | c5f93cf | 2007-02-19 11:34:43 +0100 | [diff] [blame] | 165 | 		ncp_finish_request(server, req, -EIO); | 
| Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 166 | 	} | 
 | 167 | } | 
 | 168 |  | 
 | 169 | static inline int get_conn_number(struct ncp_reply_header *rp) | 
 | 170 | { | 
 | 171 | 	return rp->conn_low | (rp->conn_high << 8); | 
 | 172 | } | 
 | 173 |  | 
 | 174 | static inline void __ncp_abort_request(struct ncp_server *server, struct ncp_request_reply *req, int err) | 
 | 175 | { | 
 | 176 | 	/* If req is done, we got signal, but we also received answer... */ | 
 | 177 | 	switch (req->status) { | 
 | 178 | 		case RQ_IDLE: | 
 | 179 | 		case RQ_DONE: | 
 | 180 | 			break; | 
 | 181 | 		case RQ_QUEUED: | 
 | 182 | 			list_del_init(&req->req); | 
| Pierre Ossman | c5f93cf | 2007-02-19 11:34:43 +0100 | [diff] [blame] | 183 | 			ncp_finish_request(server, req, err); | 
| Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 184 | 			break; | 
 | 185 | 		case RQ_INPROGRESS: | 
| Pierre Ossman | c5f93cf | 2007-02-19 11:34:43 +0100 | [diff] [blame] | 186 | 			req->status = RQ_ABANDONED; | 
 | 187 | 			break; | 
 | 188 | 		case RQ_ABANDONED: | 
| Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 189 | 			break; | 
 | 190 | 	} | 
 | 191 | } | 
 | 192 |  | 
 | 193 | static inline void ncp_abort_request(struct ncp_server *server, struct ncp_request_reply *req, int err) | 
 | 194 | { | 
| Ingo Molnar | 8e3f904 | 2006-03-23 03:00:43 -0800 | [diff] [blame] | 195 | 	mutex_lock(&server->rcv.creq_mutex); | 
| Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 196 | 	__ncp_abort_request(server, req, err); | 
| Ingo Molnar | 8e3f904 | 2006-03-23 03:00:43 -0800 | [diff] [blame] | 197 | 	mutex_unlock(&server->rcv.creq_mutex); | 
| Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 198 | } | 
 | 199 |  | 
 | 200 | static inline void __ncptcp_abort(struct ncp_server *server) | 
 | 201 | { | 
| Pierre Ossman | c5f93cf | 2007-02-19 11:34:43 +0100 | [diff] [blame] | 202 | 	__abort_ncp_connection(server); | 
| Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 203 | } | 
 | 204 |  | 
 | 205 | static int ncpdgram_send(struct socket *sock, struct ncp_request_reply *req) | 
 | 206 | { | 
 | 207 | 	struct kvec vec[3]; | 
 | 208 | 	/* sock_sendmsg updates iov pointers for us :-( */ | 
 | 209 | 	memcpy(vec, req->tx_ciov, req->tx_iovlen * sizeof(vec[0])); | 
 | 210 | 	return do_send(sock, vec, req->tx_iovlen, | 
 | 211 | 		       req->tx_totallen, MSG_DONTWAIT); | 
 | 212 | } | 
 | 213 |  | 
 | 214 | static void __ncptcp_try_send(struct ncp_server *server) | 
 | 215 | { | 
 | 216 | 	struct ncp_request_reply *rq; | 
 | 217 | 	struct kvec *iov; | 
 | 218 | 	struct kvec iovc[3]; | 
 | 219 | 	int result; | 
 | 220 |  | 
 | 221 | 	rq = server->tx.creq; | 
 | 222 | 	if (!rq) | 
 | 223 | 		return; | 
 | 224 |  | 
 | 225 | 	/* sock_sendmsg updates iov pointers for us :-( */ | 
 | 226 | 	memcpy(iovc, rq->tx_ciov, rq->tx_iovlen * sizeof(iov[0])); | 
 | 227 | 	result = do_send(server->ncp_sock, iovc, rq->tx_iovlen, | 
 | 228 | 			 rq->tx_totallen, MSG_NOSIGNAL | MSG_DONTWAIT); | 
 | 229 |  | 
 | 230 | 	if (result == -EAGAIN) | 
 | 231 | 		return; | 
 | 232 |  | 
 | 233 | 	if (result < 0) { | 
 | 234 | 		printk(KERN_ERR "ncpfs: tcp: Send failed: %d\n", result); | 
 | 235 | 		__ncp_abort_request(server, rq, result); | 
 | 236 | 		return; | 
 | 237 | 	} | 
 | 238 | 	if (result >= rq->tx_totallen) { | 
 | 239 | 		server->rcv.creq = rq; | 
 | 240 | 		server->tx.creq = NULL; | 
 | 241 | 		return; | 
 | 242 | 	} | 
 | 243 | 	rq->tx_totallen -= result; | 
 | 244 | 	iov = rq->tx_ciov; | 
 | 245 | 	while (iov->iov_len <= result) { | 
 | 246 | 		result -= iov->iov_len; | 
 | 247 | 		iov++; | 
 | 248 | 		rq->tx_iovlen--; | 
 | 249 | 	} | 
 | 250 | 	iov->iov_base += result; | 
 | 251 | 	iov->iov_len -= result; | 
 | 252 | 	rq->tx_ciov = iov; | 
 | 253 | } | 
 | 254 |  | 
 | 255 | static inline void ncp_init_header(struct ncp_server *server, struct ncp_request_reply *req, struct ncp_request_header *h) | 
 | 256 | { | 
 | 257 | 	req->status = RQ_INPROGRESS; | 
 | 258 | 	h->conn_low = server->connection; | 
 | 259 | 	h->conn_high = server->connection >> 8; | 
 | 260 | 	h->sequence = ++server->sequence; | 
 | 261 | } | 
 | 262 | 	 | 
 | 263 | static void ncpdgram_start_request(struct ncp_server *server, struct ncp_request_reply *req) | 
 | 264 | { | 
 | 265 | 	size_t signlen; | 
 | 266 | 	struct ncp_request_header* h; | 
 | 267 | 	 | 
 | 268 | 	req->tx_ciov = req->tx_iov + 1; | 
 | 269 |  | 
 | 270 | 	h = req->tx_iov[1].iov_base; | 
 | 271 | 	ncp_init_header(server, req, h); | 
 | 272 | 	signlen = sign_packet(server, req->tx_iov[1].iov_base + sizeof(struct ncp_request_header) - 1,  | 
 | 273 | 			req->tx_iov[1].iov_len - sizeof(struct ncp_request_header) + 1, | 
 | 274 | 			cpu_to_le32(req->tx_totallen), req->sign); | 
 | 275 | 	if (signlen) { | 
 | 276 | 		req->tx_ciov[1].iov_base = req->sign; | 
 | 277 | 		req->tx_ciov[1].iov_len = signlen; | 
 | 278 | 		req->tx_iovlen += 1; | 
 | 279 | 		req->tx_totallen += signlen; | 
 | 280 | 	} | 
 | 281 | 	server->rcv.creq = req; | 
 | 282 | 	server->timeout_last = server->m.time_out; | 
 | 283 | 	server->timeout_retries = server->m.retry_count; | 
 | 284 | 	ncpdgram_send(server->ncp_sock, req); | 
 | 285 | 	mod_timer(&server->timeout_tm, jiffies + server->m.time_out); | 
 | 286 | } | 
 | 287 |  | 
 | 288 | #define NCP_TCP_XMIT_MAGIC	(0x446D6454) | 
 | 289 | #define NCP_TCP_XMIT_VERSION	(1) | 
 | 290 | #define NCP_TCP_RCVD_MAGIC	(0x744E6350) | 
 | 291 |  | 
 | 292 | static void ncptcp_start_request(struct ncp_server *server, struct ncp_request_reply *req) | 
 | 293 | { | 
 | 294 | 	size_t signlen; | 
 | 295 | 	struct ncp_request_header* h; | 
 | 296 |  | 
 | 297 | 	req->tx_ciov = req->tx_iov; | 
 | 298 | 	h = req->tx_iov[1].iov_base; | 
 | 299 | 	ncp_init_header(server, req, h); | 
 | 300 | 	signlen = sign_packet(server, req->tx_iov[1].iov_base + sizeof(struct ncp_request_header) - 1, | 
 | 301 | 			req->tx_iov[1].iov_len - sizeof(struct ncp_request_header) + 1, | 
 | 302 | 			cpu_to_be32(req->tx_totallen + 24), req->sign + 4) + 16; | 
 | 303 |  | 
 | 304 | 	req->sign[0] = htonl(NCP_TCP_XMIT_MAGIC); | 
 | 305 | 	req->sign[1] = htonl(req->tx_totallen + signlen); | 
 | 306 | 	req->sign[2] = htonl(NCP_TCP_XMIT_VERSION); | 
 | 307 | 	req->sign[3] = htonl(req->datalen + 8); | 
 | 308 | 	req->tx_iov[0].iov_base = req->sign; | 
 | 309 | 	req->tx_iov[0].iov_len = signlen; | 
 | 310 | 	req->tx_iovlen += 1; | 
 | 311 | 	req->tx_totallen += signlen; | 
 | 312 |  | 
 | 313 | 	server->tx.creq = req; | 
 | 314 | 	__ncptcp_try_send(server); | 
 | 315 | } | 
 | 316 |  | 
 | 317 | static inline void __ncp_start_request(struct ncp_server *server, struct ncp_request_reply *req) | 
 | 318 | { | 
| Pierre Ossman | c5f93cf | 2007-02-19 11:34:43 +0100 | [diff] [blame] | 319 | 	/* we copy the data so that we do not depend on the caller | 
 | 320 | 	   staying alive */ | 
 | 321 | 	memcpy(server->txbuf, req->tx_iov[1].iov_base, req->tx_iov[1].iov_len); | 
 | 322 | 	req->tx_iov[1].iov_base = server->txbuf; | 
 | 323 |  | 
| Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 324 | 	if (server->ncp_sock->type == SOCK_STREAM) | 
 | 325 | 		ncptcp_start_request(server, req); | 
 | 326 | 	else | 
 | 327 | 		ncpdgram_start_request(server, req); | 
 | 328 | } | 
 | 329 |  | 
 | 330 | static int ncp_add_request(struct ncp_server *server, struct ncp_request_reply *req) | 
 | 331 | { | 
| Ingo Molnar | 8e3f904 | 2006-03-23 03:00:43 -0800 | [diff] [blame] | 332 | 	mutex_lock(&server->rcv.creq_mutex); | 
| Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 333 | 	if (!ncp_conn_valid(server)) { | 
| Ingo Molnar | 8e3f904 | 2006-03-23 03:00:43 -0800 | [diff] [blame] | 334 | 		mutex_unlock(&server->rcv.creq_mutex); | 
| Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 335 | 		printk(KERN_ERR "ncpfs: tcp: Server died\n"); | 
 | 336 | 		return -EIO; | 
 | 337 | 	} | 
| Pierre Ossman | c5f93cf | 2007-02-19 11:34:43 +0100 | [diff] [blame] | 338 | 	ncp_req_get(req); | 
| Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 339 | 	if (server->tx.creq || server->rcv.creq) { | 
 | 340 | 		req->status = RQ_QUEUED; | 
 | 341 | 		list_add_tail(&req->req, &server->tx.requests); | 
| Ingo Molnar | 8e3f904 | 2006-03-23 03:00:43 -0800 | [diff] [blame] | 342 | 		mutex_unlock(&server->rcv.creq_mutex); | 
| Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 343 | 		return 0; | 
 | 344 | 	} | 
 | 345 | 	__ncp_start_request(server, req); | 
| Ingo Molnar | 8e3f904 | 2006-03-23 03:00:43 -0800 | [diff] [blame] | 346 | 	mutex_unlock(&server->rcv.creq_mutex); | 
| Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 347 | 	return 0; | 
 | 348 | } | 
 | 349 |  | 
 | 350 | static void __ncp_next_request(struct ncp_server *server) | 
 | 351 | { | 
 | 352 | 	struct ncp_request_reply *req; | 
 | 353 |  | 
 | 354 | 	server->rcv.creq = NULL; | 
 | 355 | 	if (list_empty(&server->tx.requests)) { | 
 | 356 | 		return; | 
 | 357 | 	} | 
 | 358 | 	req = list_entry(server->tx.requests.next, struct ncp_request_reply, req); | 
 | 359 | 	list_del_init(&req->req); | 
 | 360 | 	__ncp_start_request(server, req); | 
 | 361 | } | 
 | 362 |  | 
 | 363 | static void info_server(struct ncp_server *server, unsigned int id, const void * data, size_t len) | 
 | 364 | { | 
 | 365 | 	if (server->info_sock) { | 
 | 366 | 		struct kvec iov[2]; | 
 | 367 | 		__be32 hdr[2]; | 
 | 368 | 	 | 
 | 369 | 		hdr[0] = cpu_to_be32(len + 8); | 
 | 370 | 		hdr[1] = cpu_to_be32(id); | 
 | 371 | 	 | 
 | 372 | 		iov[0].iov_base = hdr; | 
 | 373 | 		iov[0].iov_len = 8; | 
 | 374 | 		iov[1].iov_base = (void *) data; | 
 | 375 | 		iov[1].iov_len = len; | 
 | 376 |  | 
 | 377 | 		do_send(server->info_sock, iov, 2, len + 8, MSG_NOSIGNAL); | 
 | 378 | 	} | 
 | 379 | } | 
 | 380 |  | 
| David Howells | c402895 | 2006-11-22 14:57:56 +0000 | [diff] [blame] | 381 | void ncpdgram_rcv_proc(struct work_struct *work) | 
| Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 382 | { | 
| David Howells | c402895 | 2006-11-22 14:57:56 +0000 | [diff] [blame] | 383 | 	struct ncp_server *server = | 
 | 384 | 		container_of(work, struct ncp_server, rcv.tq); | 
| Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 385 | 	struct socket* sock; | 
 | 386 | 	 | 
 | 387 | 	sock = server->ncp_sock; | 
 | 388 | 	 | 
 | 389 | 	while (1) { | 
 | 390 | 		struct ncp_reply_header reply; | 
 | 391 | 		int result; | 
 | 392 |  | 
 | 393 | 		result = _recv(sock, &reply, sizeof(reply), MSG_PEEK | MSG_DONTWAIT); | 
 | 394 | 		if (result < 0) { | 
 | 395 | 			break; | 
 | 396 | 		} | 
 | 397 | 		if (result >= sizeof(reply)) { | 
 | 398 | 			struct ncp_request_reply *req; | 
 | 399 | 	 | 
 | 400 | 			if (reply.type == NCP_WATCHDOG) { | 
 | 401 | 				unsigned char buf[10]; | 
 | 402 |  | 
 | 403 | 				if (server->connection != get_conn_number(&reply)) { | 
 | 404 | 					goto drop; | 
 | 405 | 				} | 
 | 406 | 				result = _recv(sock, buf, sizeof(buf), MSG_DONTWAIT); | 
 | 407 | 				if (result < 0) { | 
 | 408 | 					DPRINTK("recv failed with %d\n", result); | 
 | 409 | 					continue; | 
 | 410 | 				} | 
 | 411 | 				if (result < 10) { | 
 | 412 | 					DPRINTK("too short (%u) watchdog packet\n", result); | 
 | 413 | 					continue; | 
 | 414 | 				} | 
 | 415 | 				if (buf[9] != '?') { | 
 | 416 | 					DPRINTK("bad signature (%02X) in watchdog packet\n", buf[9]); | 
 | 417 | 					continue; | 
 | 418 | 				} | 
 | 419 | 				buf[9] = 'Y'; | 
 | 420 | 				_send(sock, buf, sizeof(buf)); | 
 | 421 | 				continue; | 
 | 422 | 			} | 
 | 423 | 			if (reply.type != NCP_POSITIVE_ACK && reply.type != NCP_REPLY) { | 
 | 424 | 				result = _recv(sock, server->unexpected_packet.data, sizeof(server->unexpected_packet.data), MSG_DONTWAIT); | 
 | 425 | 				if (result < 0) { | 
 | 426 | 					continue; | 
 | 427 | 				} | 
 | 428 | 				info_server(server, 0, server->unexpected_packet.data, result); | 
 | 429 | 				continue; | 
 | 430 | 			} | 
| Ingo Molnar | 8e3f904 | 2006-03-23 03:00:43 -0800 | [diff] [blame] | 431 | 			mutex_lock(&server->rcv.creq_mutex); | 
| Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 432 | 			req = server->rcv.creq; | 
 | 433 | 			if (req && (req->tx_type == NCP_ALLOC_SLOT_REQUEST || (server->sequence == reply.sequence &&  | 
 | 434 | 					server->connection == get_conn_number(&reply)))) { | 
 | 435 | 				if (reply.type == NCP_POSITIVE_ACK) { | 
 | 436 | 					server->timeout_retries = server->m.retry_count; | 
 | 437 | 					server->timeout_last = NCP_MAX_RPC_TIMEOUT; | 
 | 438 | 					mod_timer(&server->timeout_tm, jiffies + NCP_MAX_RPC_TIMEOUT); | 
 | 439 | 				} else if (reply.type == NCP_REPLY) { | 
| Pierre Ossman | c5f93cf | 2007-02-19 11:34:43 +0100 | [diff] [blame] | 440 | 					result = _recv(sock, server->rxbuf, req->datalen, MSG_DONTWAIT); | 
| Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 441 | #ifdef CONFIG_NCPFS_PACKET_SIGNING | 
 | 442 | 					if (result >= 0 && server->sign_active && req->tx_type != NCP_DEALLOC_SLOT_REQUEST) { | 
 | 443 | 						if (result < 8 + 8) { | 
 | 444 | 							result = -EIO; | 
 | 445 | 						} else { | 
 | 446 | 							unsigned int hdrl; | 
 | 447 | 							 | 
 | 448 | 							result -= 8; | 
 | 449 | 							hdrl = sock->sk->sk_family == AF_INET ? 8 : 6; | 
| Pierre Ossman | c5f93cf | 2007-02-19 11:34:43 +0100 | [diff] [blame] | 450 | 							if (sign_verify_reply(server, server->rxbuf + hdrl, result - hdrl, cpu_to_le32(result), server->rxbuf + result)) { | 
| Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 451 | 								printk(KERN_INFO "ncpfs: Signature violation\n"); | 
 | 452 | 								result = -EIO; | 
 | 453 | 							} | 
 | 454 | 						} | 
 | 455 | 					} | 
 | 456 | #endif | 
 | 457 | 					del_timer(&server->timeout_tm); | 
 | 458 | 				     	server->rcv.creq = NULL; | 
| Pierre Ossman | c5f93cf | 2007-02-19 11:34:43 +0100 | [diff] [blame] | 459 | 					ncp_finish_request(server, req, result); | 
| Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 460 | 					__ncp_next_request(server); | 
| Ingo Molnar | 8e3f904 | 2006-03-23 03:00:43 -0800 | [diff] [blame] | 461 | 					mutex_unlock(&server->rcv.creq_mutex); | 
| Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 462 | 					continue; | 
 | 463 | 				} | 
 | 464 | 			} | 
| Ingo Molnar | 8e3f904 | 2006-03-23 03:00:43 -0800 | [diff] [blame] | 465 | 			mutex_unlock(&server->rcv.creq_mutex); | 
| Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 466 | 		} | 
 | 467 | drop:;		 | 
 | 468 | 		_recv(sock, &reply, sizeof(reply), MSG_DONTWAIT); | 
 | 469 | 	} | 
 | 470 | } | 
 | 471 |  | 
 | 472 | static void __ncpdgram_timeout_proc(struct ncp_server *server) | 
 | 473 | { | 
 | 474 | 	/* If timer is pending, we are processing another request... */ | 
 | 475 | 	if (!timer_pending(&server->timeout_tm)) { | 
 | 476 | 		struct ncp_request_reply* req; | 
 | 477 | 		 | 
 | 478 | 		req = server->rcv.creq; | 
 | 479 | 		if (req) { | 
 | 480 | 			int timeout; | 
 | 481 | 			 | 
 | 482 | 			if (server->m.flags & NCP_MOUNT_SOFT) { | 
 | 483 | 				if (server->timeout_retries-- == 0) { | 
 | 484 | 					__ncp_abort_request(server, req, -ETIMEDOUT); | 
 | 485 | 					return; | 
 | 486 | 				} | 
 | 487 | 			} | 
 | 488 | 			/* Ignore errors */ | 
 | 489 | 			ncpdgram_send(server->ncp_sock, req); | 
 | 490 | 			timeout = server->timeout_last << 1; | 
 | 491 | 			if (timeout > NCP_MAX_RPC_TIMEOUT) { | 
 | 492 | 				timeout = NCP_MAX_RPC_TIMEOUT; | 
 | 493 | 			} | 
 | 494 | 			server->timeout_last = timeout; | 
 | 495 | 			mod_timer(&server->timeout_tm, jiffies + timeout); | 
 | 496 | 		} | 
 | 497 | 	} | 
 | 498 | } | 
 | 499 |  | 
| David Howells | c402895 | 2006-11-22 14:57:56 +0000 | [diff] [blame] | 500 | void ncpdgram_timeout_proc(struct work_struct *work) | 
| Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 501 | { | 
| David Howells | c402895 | 2006-11-22 14:57:56 +0000 | [diff] [blame] | 502 | 	struct ncp_server *server = | 
 | 503 | 		container_of(work, struct ncp_server, timeout_tq); | 
| Ingo Molnar | 8e3f904 | 2006-03-23 03:00:43 -0800 | [diff] [blame] | 504 | 	mutex_lock(&server->rcv.creq_mutex); | 
| Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 505 | 	__ncpdgram_timeout_proc(server); | 
| Ingo Molnar | 8e3f904 | 2006-03-23 03:00:43 -0800 | [diff] [blame] | 506 | 	mutex_unlock(&server->rcv.creq_mutex); | 
| Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 507 | } | 
 | 508 |  | 
| Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 509 | static int do_tcp_rcv(struct ncp_server *server, void *buffer, size_t len) | 
 | 510 | { | 
 | 511 | 	int result; | 
 | 512 | 	 | 
 | 513 | 	if (buffer) { | 
 | 514 | 		result = _recv(server->ncp_sock, buffer, len, MSG_DONTWAIT); | 
 | 515 | 	} else { | 
 | 516 | 		static unsigned char dummy[1024]; | 
 | 517 | 			 | 
 | 518 | 		if (len > sizeof(dummy)) { | 
 | 519 | 			len = sizeof(dummy); | 
 | 520 | 		} | 
 | 521 | 		result = _recv(server->ncp_sock, dummy, len, MSG_DONTWAIT); | 
 | 522 | 	} | 
 | 523 | 	if (result < 0) { | 
 | 524 | 		return result; | 
 | 525 | 	} | 
 | 526 | 	if (result > len) { | 
 | 527 | 		printk(KERN_ERR "ncpfs: tcp: bug in recvmsg (%u > %Zu)\n", result, len); | 
 | 528 | 		return -EIO;			 | 
 | 529 | 	} | 
 | 530 | 	return result; | 
 | 531 | }	 | 
 | 532 |  | 
 | 533 | static int __ncptcp_rcv_proc(struct ncp_server *server) | 
 | 534 | { | 
 | 535 | 	/* We have to check the result, so store the complete header */ | 
 | 536 | 	while (1) { | 
 | 537 | 		int result; | 
 | 538 | 		struct ncp_request_reply *req; | 
 | 539 | 		int datalen; | 
 | 540 | 		int type; | 
 | 541 |  | 
 | 542 | 		while (server->rcv.len) { | 
 | 543 | 			result = do_tcp_rcv(server, server->rcv.ptr, server->rcv.len); | 
 | 544 | 			if (result == -EAGAIN) { | 
 | 545 | 				return 0; | 
 | 546 | 			} | 
 | 547 | 			if (result <= 0) { | 
 | 548 | 				req = server->rcv.creq; | 
 | 549 | 				if (req) { | 
 | 550 | 					__ncp_abort_request(server, req, -EIO); | 
 | 551 | 				} else { | 
 | 552 | 					__ncptcp_abort(server); | 
 | 553 | 				} | 
 | 554 | 				if (result < 0) { | 
 | 555 | 					printk(KERN_ERR "ncpfs: tcp: error in recvmsg: %d\n", result); | 
 | 556 | 				} else { | 
 | 557 | 					DPRINTK(KERN_ERR "ncpfs: tcp: EOF\n"); | 
 | 558 | 				} | 
 | 559 | 				return -EIO; | 
 | 560 | 			} | 
 | 561 | 			if (server->rcv.ptr) { | 
 | 562 | 				server->rcv.ptr += result; | 
 | 563 | 			} | 
 | 564 | 			server->rcv.len -= result; | 
 | 565 | 		} | 
 | 566 | 		switch (server->rcv.state) { | 
 | 567 | 			case 0: | 
 | 568 | 				if (server->rcv.buf.magic != htonl(NCP_TCP_RCVD_MAGIC)) { | 
 | 569 | 					printk(KERN_ERR "ncpfs: tcp: Unexpected reply type %08X\n", ntohl(server->rcv.buf.magic)); | 
 | 570 | 					__ncptcp_abort(server); | 
 | 571 | 					return -EIO; | 
 | 572 | 				} | 
 | 573 | 				datalen = ntohl(server->rcv.buf.len) & 0x0FFFFFFF; | 
 | 574 | 				if (datalen < 10) { | 
 | 575 | 					printk(KERN_ERR "ncpfs: tcp: Unexpected reply len %d\n", datalen); | 
 | 576 | 					__ncptcp_abort(server); | 
 | 577 | 					return -EIO; | 
 | 578 | 				} | 
 | 579 | #ifdef CONFIG_NCPFS_PACKET_SIGNING				 | 
 | 580 | 				if (server->sign_active) { | 
 | 581 | 					if (datalen < 18) { | 
 | 582 | 						printk(KERN_ERR "ncpfs: tcp: Unexpected reply len %d\n", datalen); | 
 | 583 | 						__ncptcp_abort(server); | 
 | 584 | 						return -EIO; | 
 | 585 | 					} | 
 | 586 | 					server->rcv.buf.len = datalen - 8; | 
 | 587 | 					server->rcv.ptr = (unsigned char*)&server->rcv.buf.p1; | 
 | 588 | 					server->rcv.len = 8; | 
 | 589 | 					server->rcv.state = 4; | 
 | 590 | 					break; | 
 | 591 | 				} | 
 | 592 | #endif				 | 
 | 593 | 				type = ntohs(server->rcv.buf.type); | 
 | 594 | #ifdef CONFIG_NCPFS_PACKET_SIGNING				 | 
 | 595 | cont:;				 | 
 | 596 | #endif | 
 | 597 | 				if (type != NCP_REPLY) { | 
 | 598 | 					if (datalen - 8 <= sizeof(server->unexpected_packet.data)) { | 
 | 599 | 						*(__u16*)(server->unexpected_packet.data) = htons(type); | 
 | 600 | 						server->unexpected_packet.len = datalen - 8; | 
 | 601 |  | 
 | 602 | 						server->rcv.state = 5; | 
 | 603 | 						server->rcv.ptr = server->unexpected_packet.data + 2; | 
 | 604 | 						server->rcv.len = datalen - 10; | 
 | 605 | 						break; | 
 | 606 | 					}					 | 
 | 607 | 					DPRINTK("ncpfs: tcp: Unexpected NCP type %02X\n", type); | 
 | 608 | skipdata2:; | 
 | 609 | 					server->rcv.state = 2; | 
 | 610 | skipdata:; | 
 | 611 | 					server->rcv.ptr = NULL; | 
 | 612 | 					server->rcv.len = datalen - 10; | 
 | 613 | 					break; | 
 | 614 | 				} | 
 | 615 | 				req = server->rcv.creq; | 
 | 616 | 				if (!req) { | 
 | 617 | 					DPRINTK(KERN_ERR "ncpfs: Reply without appropriate request\n"); | 
 | 618 | 					goto skipdata2; | 
 | 619 | 				} | 
 | 620 | 				if (datalen > req->datalen + 8) { | 
 | 621 | 					printk(KERN_ERR "ncpfs: tcp: Unexpected reply len %d (expected at most %Zd)\n", datalen, req->datalen + 8); | 
 | 622 | 					server->rcv.state = 3; | 
 | 623 | 					goto skipdata; | 
 | 624 | 				} | 
 | 625 | 				req->datalen = datalen - 8; | 
| Pierre Ossman | c5f93cf | 2007-02-19 11:34:43 +0100 | [diff] [blame] | 626 | 				((struct ncp_reply_header*)server->rxbuf)->type = NCP_REPLY; | 
 | 627 | 				server->rcv.ptr = server->rxbuf + 2; | 
| Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 628 | 				server->rcv.len = datalen - 10; | 
 | 629 | 				server->rcv.state = 1; | 
 | 630 | 				break; | 
 | 631 | #ifdef CONFIG_NCPFS_PACKET_SIGNING				 | 
 | 632 | 			case 4: | 
 | 633 | 				datalen = server->rcv.buf.len; | 
 | 634 | 				type = ntohs(server->rcv.buf.type2); | 
 | 635 | 				goto cont; | 
 | 636 | #endif | 
 | 637 | 			case 1: | 
 | 638 | 				req = server->rcv.creq; | 
 | 639 | 				if (req->tx_type != NCP_ALLOC_SLOT_REQUEST) { | 
| Pierre Ossman | c5f93cf | 2007-02-19 11:34:43 +0100 | [diff] [blame] | 640 | 					if (((struct ncp_reply_header*)server->rxbuf)->sequence != server->sequence) { | 
| Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 641 | 						printk(KERN_ERR "ncpfs: tcp: Bad sequence number\n"); | 
 | 642 | 						__ncp_abort_request(server, req, -EIO); | 
 | 643 | 						return -EIO; | 
 | 644 | 					} | 
| Pierre Ossman | c5f93cf | 2007-02-19 11:34:43 +0100 | [diff] [blame] | 645 | 					if ((((struct ncp_reply_header*)server->rxbuf)->conn_low | (((struct ncp_reply_header*)server->rxbuf)->conn_high << 8)) != server->connection) { | 
| Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 646 | 						printk(KERN_ERR "ncpfs: tcp: Connection number mismatch\n"); | 
 | 647 | 						__ncp_abort_request(server, req, -EIO); | 
 | 648 | 						return -EIO; | 
 | 649 | 					} | 
 | 650 | 				} | 
 | 651 | #ifdef CONFIG_NCPFS_PACKET_SIGNING				 | 
 | 652 | 				if (server->sign_active && req->tx_type != NCP_DEALLOC_SLOT_REQUEST) { | 
| Pierre Ossman | c5f93cf | 2007-02-19 11:34:43 +0100 | [diff] [blame] | 653 | 					if (sign_verify_reply(server, server->rxbuf + 6, req->datalen - 6, cpu_to_be32(req->datalen + 16), &server->rcv.buf.type)) { | 
| Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 654 | 						printk(KERN_ERR "ncpfs: tcp: Signature violation\n"); | 
 | 655 | 						__ncp_abort_request(server, req, -EIO); | 
 | 656 | 						return -EIO; | 
 | 657 | 					} | 
 | 658 | 				} | 
 | 659 | #endif				 | 
| Pierre Ossman | c5f93cf | 2007-02-19 11:34:43 +0100 | [diff] [blame] | 660 | 				ncp_finish_request(server, req, req->datalen); | 
| Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 661 | 			nextreq:; | 
 | 662 | 				__ncp_next_request(server); | 
 | 663 | 			case 2: | 
 | 664 | 			next:; | 
 | 665 | 				server->rcv.ptr = (unsigned char*)&server->rcv.buf; | 
 | 666 | 				server->rcv.len = 10; | 
 | 667 | 				server->rcv.state = 0; | 
 | 668 | 				break; | 
 | 669 | 			case 3: | 
| Pierre Ossman | c5f93cf | 2007-02-19 11:34:43 +0100 | [diff] [blame] | 670 | 				ncp_finish_request(server, server->rcv.creq, -EIO); | 
| Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 671 | 				goto nextreq; | 
 | 672 | 			case 5: | 
 | 673 | 				info_server(server, 0, server->unexpected_packet.data, server->unexpected_packet.len); | 
 | 674 | 				goto next; | 
 | 675 | 		} | 
 | 676 | 	} | 
 | 677 | } | 
 | 678 |  | 
| David Howells | c402895 | 2006-11-22 14:57:56 +0000 | [diff] [blame] | 679 | void ncp_tcp_rcv_proc(struct work_struct *work) | 
| Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 680 | { | 
| David Howells | c402895 | 2006-11-22 14:57:56 +0000 | [diff] [blame] | 681 | 	struct ncp_server *server = | 
 | 682 | 		container_of(work, struct ncp_server, rcv.tq); | 
| Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 683 |  | 
| Ingo Molnar | 8e3f904 | 2006-03-23 03:00:43 -0800 | [diff] [blame] | 684 | 	mutex_lock(&server->rcv.creq_mutex); | 
| Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 685 | 	__ncptcp_rcv_proc(server); | 
| Ingo Molnar | 8e3f904 | 2006-03-23 03:00:43 -0800 | [diff] [blame] | 686 | 	mutex_unlock(&server->rcv.creq_mutex); | 
| Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 687 | } | 
 | 688 |  | 
| David Howells | c402895 | 2006-11-22 14:57:56 +0000 | [diff] [blame] | 689 | void ncp_tcp_tx_proc(struct work_struct *work) | 
| Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 690 | { | 
| David Howells | c402895 | 2006-11-22 14:57:56 +0000 | [diff] [blame] | 691 | 	struct ncp_server *server = | 
 | 692 | 		container_of(work, struct ncp_server, tx.tq); | 
| Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 693 | 	 | 
| Ingo Molnar | 8e3f904 | 2006-03-23 03:00:43 -0800 | [diff] [blame] | 694 | 	mutex_lock(&server->rcv.creq_mutex); | 
| Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 695 | 	__ncptcp_try_send(server); | 
| Ingo Molnar | 8e3f904 | 2006-03-23 03:00:43 -0800 | [diff] [blame] | 696 | 	mutex_unlock(&server->rcv.creq_mutex); | 
| Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 697 | } | 
 | 698 |  | 
 | 699 | static int do_ncp_rpc_call(struct ncp_server *server, int size, | 
| Pierre Ossman | c5f93cf | 2007-02-19 11:34:43 +0100 | [diff] [blame] | 700 | 		unsigned char* reply_buf, int max_reply_size) | 
| Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 701 | { | 
 | 702 | 	int result; | 
| Pierre Ossman | c5f93cf | 2007-02-19 11:34:43 +0100 | [diff] [blame] | 703 | 	struct ncp_request_reply *req; | 
| Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 704 |  | 
| Pierre Ossman | c5f93cf | 2007-02-19 11:34:43 +0100 | [diff] [blame] | 705 | 	req = ncp_alloc_req(); | 
 | 706 | 	if (!req) | 
 | 707 | 		return -ENOMEM; | 
| Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 708 |  | 
| Pierre Ossman | c5f93cf | 2007-02-19 11:34:43 +0100 | [diff] [blame] | 709 | 	req->reply_buf = reply_buf; | 
 | 710 | 	req->datalen = max_reply_size; | 
 | 711 | 	req->tx_iov[1].iov_base = server->packet; | 
 | 712 | 	req->tx_iov[1].iov_len = size; | 
 | 713 | 	req->tx_iovlen = 1; | 
 | 714 | 	req->tx_totallen = size; | 
 | 715 | 	req->tx_type = *(u_int16_t*)server->packet; | 
 | 716 |  | 
 | 717 | 	result = ncp_add_request(server, req); | 
 | 718 | 	if (result < 0) | 
 | 719 | 		goto out; | 
 | 720 |  | 
 | 721 | 	if (wait_event_interruptible(req->wq, req->status == RQ_DONE)) { | 
 | 722 | 		ncp_abort_request(server, req, -EINTR); | 
 | 723 | 		result = -EINTR; | 
 | 724 | 		goto out; | 
| Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 725 | 	} | 
| Pierre Ossman | c5f93cf | 2007-02-19 11:34:43 +0100 | [diff] [blame] | 726 |  | 
 | 727 | 	result = req->result; | 
 | 728 |  | 
 | 729 | out: | 
 | 730 | 	ncp_req_put(req); | 
 | 731 |  | 
 | 732 | 	return result; | 
| Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 733 | } | 
 | 734 |  | 
 | 735 | /* | 
 | 736 |  * We need the server to be locked here, so check! | 
 | 737 |  */ | 
 | 738 |  | 
 | 739 | static int ncp_do_request(struct ncp_server *server, int size, | 
 | 740 | 		void* reply, int max_reply_size) | 
 | 741 | { | 
 | 742 | 	int result; | 
 | 743 |  | 
 | 744 | 	if (server->lock == 0) { | 
 | 745 | 		printk(KERN_ERR "ncpfs: Server not locked!\n"); | 
 | 746 | 		return -EIO; | 
 | 747 | 	} | 
 | 748 | 	if (!ncp_conn_valid(server)) { | 
 | 749 | 		printk(KERN_ERR "ncpfs: Connection invalid!\n"); | 
 | 750 | 		return -EIO; | 
 | 751 | 	} | 
 | 752 | 	{ | 
 | 753 | 		sigset_t old_set; | 
 | 754 | 		unsigned long mask, flags; | 
 | 755 |  | 
 | 756 | 		spin_lock_irqsave(¤t->sighand->siglock, flags); | 
 | 757 | 		old_set = current->blocked; | 
 | 758 | 		if (current->flags & PF_EXITING) | 
 | 759 | 			mask = 0; | 
 | 760 | 		else | 
 | 761 | 			mask = sigmask(SIGKILL); | 
 | 762 | 		if (server->m.flags & NCP_MOUNT_INTR) { | 
 | 763 | 			/* FIXME: This doesn't seem right at all.  So, like, | 
 | 764 | 			   we can't handle SIGINT and get whatever to stop? | 
 | 765 | 			   What if we've blocked it ourselves?  What about | 
 | 766 | 			   alarms?  Why, in fact, are we mucking with the | 
 | 767 | 			   sigmask at all? -- r~ */ | 
 | 768 | 			if (current->sighand->action[SIGINT - 1].sa.sa_handler == SIG_DFL) | 
 | 769 | 				mask |= sigmask(SIGINT); | 
 | 770 | 			if (current->sighand->action[SIGQUIT - 1].sa.sa_handler == SIG_DFL) | 
 | 771 | 				mask |= sigmask(SIGQUIT); | 
 | 772 | 		} | 
 | 773 | 		siginitsetinv(¤t->blocked, mask); | 
 | 774 | 		recalc_sigpending(); | 
 | 775 | 		spin_unlock_irqrestore(¤t->sighand->siglock, flags); | 
 | 776 | 		 | 
 | 777 | 		result = do_ncp_rpc_call(server, size, reply, max_reply_size); | 
 | 778 |  | 
 | 779 | 		spin_lock_irqsave(¤t->sighand->siglock, flags); | 
 | 780 | 		current->blocked = old_set; | 
 | 781 | 		recalc_sigpending(); | 
 | 782 | 		spin_unlock_irqrestore(¤t->sighand->siglock, flags); | 
 | 783 | 	} | 
 | 784 |  | 
 | 785 | 	DDPRINTK("do_ncp_rpc_call returned %d\n", result); | 
 | 786 |  | 
| Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 787 | 	return result; | 
 | 788 | } | 
 | 789 |  | 
 | 790 | /* ncp_do_request assures that at least a complete reply header is | 
 | 791 |  * received. It assumes that server->current_size contains the ncp | 
 | 792 |  * request size | 
 | 793 |  */ | 
 | 794 | int ncp_request2(struct ncp_server *server, int function,  | 
 | 795 | 		void* rpl, int size) | 
 | 796 | { | 
 | 797 | 	struct ncp_request_header *h; | 
 | 798 | 	struct ncp_reply_header* reply = rpl; | 
 | 799 | 	int result; | 
 | 800 |  | 
 | 801 | 	h = (struct ncp_request_header *) (server->packet); | 
 | 802 | 	if (server->has_subfunction != 0) { | 
 | 803 | 		*(__u16 *) & (h->data[0]) = htons(server->current_size - sizeof(*h) - 2); | 
 | 804 | 	} | 
 | 805 | 	h->type = NCP_REQUEST; | 
 | 806 | 	/* | 
 | 807 | 	 * The server shouldn't know or care what task is making a | 
 | 808 | 	 * request, so we always use the same task number. | 
 | 809 | 	 */ | 
 | 810 | 	h->task = 2; /* (current->pid) & 0xff; */ | 
 | 811 | 	h->function = function; | 
 | 812 |  | 
 | 813 | 	result = ncp_do_request(server, server->current_size, reply, size); | 
 | 814 | 	if (result < 0) { | 
 | 815 | 		DPRINTK("ncp_request_error: %d\n", result); | 
 | 816 | 		goto out; | 
 | 817 | 	} | 
 | 818 | 	server->completion = reply->completion_code; | 
 | 819 | 	server->conn_status = reply->connection_state; | 
 | 820 | 	server->reply_size = result; | 
 | 821 | 	server->ncp_reply_size = result - sizeof(struct ncp_reply_header); | 
 | 822 |  | 
 | 823 | 	result = reply->completion_code; | 
 | 824 |  | 
 | 825 | 	if (result != 0) | 
 | 826 | 		PPRINTK("ncp_request: completion code=%x\n", result); | 
 | 827 | out: | 
 | 828 | 	return result; | 
 | 829 | } | 
 | 830 |  | 
 | 831 | int ncp_connect(struct ncp_server *server) | 
 | 832 | { | 
 | 833 | 	struct ncp_request_header *h; | 
 | 834 | 	int result; | 
 | 835 |  | 
 | 836 | 	server->connection = 0xFFFF; | 
 | 837 | 	server->sequence = 255; | 
 | 838 |  | 
 | 839 | 	h = (struct ncp_request_header *) (server->packet); | 
 | 840 | 	h->type = NCP_ALLOC_SLOT_REQUEST; | 
 | 841 | 	h->task		= 2; /* see above */ | 
 | 842 | 	h->function	= 0; | 
 | 843 |  | 
 | 844 | 	result = ncp_do_request(server, sizeof(*h), server->packet, server->packet_size); | 
 | 845 | 	if (result < 0) | 
 | 846 | 		goto out; | 
 | 847 | 	server->connection = h->conn_low + (h->conn_high * 256); | 
 | 848 | 	result = 0; | 
 | 849 | out: | 
 | 850 | 	return result; | 
 | 851 | } | 
 | 852 |  | 
 | 853 | int ncp_disconnect(struct ncp_server *server) | 
 | 854 | { | 
 | 855 | 	struct ncp_request_header *h; | 
 | 856 |  | 
 | 857 | 	h = (struct ncp_request_header *) (server->packet); | 
 | 858 | 	h->type = NCP_DEALLOC_SLOT_REQUEST; | 
 | 859 | 	h->task		= 2; /* see above */ | 
 | 860 | 	h->function	= 0; | 
 | 861 |  | 
 | 862 | 	return ncp_do_request(server, sizeof(*h), server->packet, server->packet_size); | 
 | 863 | } | 
 | 864 |  | 
 | 865 | void ncp_lock_server(struct ncp_server *server) | 
 | 866 | { | 
| Ingo Molnar | 8e3f904 | 2006-03-23 03:00:43 -0800 | [diff] [blame] | 867 | 	mutex_lock(&server->mutex); | 
| Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 868 | 	if (server->lock) | 
 | 869 | 		printk(KERN_WARNING "ncp_lock_server: was locked!\n"); | 
 | 870 | 	server->lock = 1; | 
 | 871 | } | 
 | 872 |  | 
 | 873 | void ncp_unlock_server(struct ncp_server *server) | 
 | 874 | { | 
 | 875 | 	if (!server->lock) { | 
 | 876 | 		printk(KERN_WARNING "ncp_unlock_server: was not locked!\n"); | 
 | 877 | 		return; | 
 | 878 | 	} | 
 | 879 | 	server->lock = 0; | 
| Ingo Molnar | 8e3f904 | 2006-03-23 03:00:43 -0800 | [diff] [blame] | 880 | 	mutex_unlock(&server->mutex); | 
| Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 881 | } |