trans_fd.c 24 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087
  1. /*
  2. * linux/fs/9p/trans_fd.c
  3. *
  4. * Fd transport layer. Includes deprecated socket layer.
  5. *
  6. * Copyright (C) 2006 by Russ Cox <rsc@swtch.com>
  7. * Copyright (C) 2004-2005 by Latchesar Ionkov <lucho@ionkov.net>
  8. * Copyright (C) 2004-2008 by Eric Van Hensbergen <ericvh@gmail.com>
  9. * Copyright (C) 1997-2002 by Ron Minnich <rminnich@sarnoff.com>
  10. *
  11. * This program is free software; you can redistribute it and/or modify
  12. * it under the terms of the GNU General Public License version 2
  13. * as published by the Free Software Foundation.
  14. *
  15. * This program is distributed in the hope that it will be useful,
  16. * but WITHOUT ANY WARRANTY; without even the implied warranty of
  17. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  18. * GNU General Public License for more details.
  19. *
  20. * You should have received a copy of the GNU General Public License
  21. * along with this program; if not, write to:
  22. * Free Software Foundation
  23. * 51 Franklin Street, Fifth Floor
  24. * Boston, MA 02111-1301 USA
  25. *
  26. */
  27. #include <linux/in.h>
  28. #include <linux/module.h>
  29. #include <linux/net.h>
  30. #include <linux/ipv6.h>
  31. #include <linux/kthread.h>
  32. #include <linux/errno.h>
  33. #include <linux/kernel.h>
  34. #include <linux/un.h>
  35. #include <linux/uaccess.h>
  36. #include <linux/inet.h>
  37. #include <linux/idr.h>
  38. #include <linux/file.h>
  39. #include <linux/parser.h>
  40. #include <linux/slab.h>
  41. #include <net/9p/9p.h>
  42. #include <net/9p/client.h>
  43. #include <net/9p/transport.h>
  44. #include <linux/syscalls.h> /* killme */
  45. #define P9_PORT 564
  46. #define MAX_SOCK_BUF (64*1024)
  47. #define MAXPOLLWADDR 2
  48. /**
  49. * struct p9_fd_opts - per-transport options
  50. * @rfd: file descriptor for reading (trans=fd)
  51. * @wfd: file descriptor for writing (trans=fd)
  52. * @port: port to connect to (trans=tcp)
  53. *
  54. */
  55. struct p9_fd_opts {
  56. int rfd;
  57. int wfd;
  58. u16 port;
  59. };
  60. /**
  61. * struct p9_trans_fd - transport state
  62. * @rd: reference to file to read from
  63. * @wr: reference of file to write to
  64. * @conn: connection state reference
  65. *
  66. */
  67. struct p9_trans_fd {
  68. struct file *rd;
  69. struct file *wr;
  70. struct p9_conn *conn;
  71. };
  72. /*
  73. * Option Parsing (code inspired by NFS code)
  74. * - a little lazy - parse all fd-transport options
  75. */
  76. enum {
  77. /* Options that take integer arguments */
  78. Opt_port, Opt_rfdno, Opt_wfdno, Opt_err,
  79. };
  80. static const match_table_t tokens = {
  81. {Opt_port, "port=%u"},
  82. {Opt_rfdno, "rfdno=%u"},
  83. {Opt_wfdno, "wfdno=%u"},
  84. {Opt_err, NULL},
  85. };
  86. enum {
  87. Rworksched = 1, /* read work scheduled or running */
  88. Rpending = 2, /* can read */
  89. Wworksched = 4, /* write work scheduled or running */
  90. Wpending = 8, /* can write */
  91. };
  92. struct p9_poll_wait {
  93. struct p9_conn *conn;
  94. wait_queue_t wait;
  95. wait_queue_head_t *wait_addr;
  96. };
  97. /**
  98. * struct p9_conn - fd mux connection state information
  99. * @mux_list: list link for mux to manage multiple connections (?)
  100. * @client: reference to client instance for this connection
  101. * @err: error state
  102. * @req_list: accounting for requests which have been sent
  103. * @unsent_req_list: accounting for requests that haven't been sent
  104. * @req: current request being processed (if any)
  105. * @tmp_buf: temporary buffer to read in header
  106. * @rsize: amount to read for current frame
  107. * @rpos: read position in current frame
  108. * @rbuf: current read buffer
  109. * @wpos: write position for current frame
  110. * @wsize: amount of data to write for current frame
  111. * @wbuf: current write buffer
  112. * @poll_pending_link: pending links to be polled per conn
  113. * @poll_wait: array of wait_q's for various worker threads
  114. * @pt: poll state
  115. * @rq: current read work
  116. * @wq: current write work
  117. * @wsched: ????
  118. *
  119. */
  120. struct p9_conn {
  121. struct list_head mux_list;
  122. struct p9_client *client;
  123. int err;
  124. struct list_head req_list;
  125. struct list_head unsent_req_list;
  126. struct p9_req_t *req;
  127. char tmp_buf[7];
  128. int rsize;
  129. int rpos;
  130. char *rbuf;
  131. int wpos;
  132. int wsize;
  133. char *wbuf;
  134. struct list_head poll_pending_link;
  135. struct p9_poll_wait poll_wait[MAXPOLLWADDR];
  136. poll_table pt;
  137. struct work_struct rq;
  138. struct work_struct wq;
  139. unsigned long wsched;
  140. };
  141. static void p9_poll_workfn(struct work_struct *work);
  142. static DEFINE_SPINLOCK(p9_poll_lock);
  143. static LIST_HEAD(p9_poll_pending_list);
  144. static DECLARE_WORK(p9_poll_work, p9_poll_workfn);
  145. static void p9_mux_poll_stop(struct p9_conn *m)
  146. {
  147. unsigned long flags;
  148. int i;
  149. for (i = 0; i < ARRAY_SIZE(m->poll_wait); i++) {
  150. struct p9_poll_wait *pwait = &m->poll_wait[i];
  151. if (pwait->wait_addr) {
  152. remove_wait_queue(pwait->wait_addr, &pwait->wait);
  153. pwait->wait_addr = NULL;
  154. }
  155. }
  156. spin_lock_irqsave(&p9_poll_lock, flags);
  157. list_del_init(&m->poll_pending_link);
  158. spin_unlock_irqrestore(&p9_poll_lock, flags);
  159. }
  160. /**
  161. * p9_conn_cancel - cancel all pending requests with error
  162. * @m: mux data
  163. * @err: error code
  164. *
  165. */
  166. static void p9_conn_cancel(struct p9_conn *m, int err)
  167. {
  168. struct p9_req_t *req, *rtmp;
  169. unsigned long flags;
  170. LIST_HEAD(cancel_list);
  171. P9_DPRINTK(P9_DEBUG_ERROR, "mux %p err %d\n", m, err);
  172. spin_lock_irqsave(&m->client->lock, flags);
  173. if (m->err) {
  174. spin_unlock_irqrestore(&m->client->lock, flags);
  175. return;
  176. }
  177. m->err = err;
  178. list_for_each_entry_safe(req, rtmp, &m->req_list, req_list) {
  179. req->status = REQ_STATUS_ERROR;
  180. if (!req->t_err)
  181. req->t_err = err;
  182. list_move(&req->req_list, &cancel_list);
  183. }
  184. list_for_each_entry_safe(req, rtmp, &m->unsent_req_list, req_list) {
  185. req->status = REQ_STATUS_ERROR;
  186. if (!req->t_err)
  187. req->t_err = err;
  188. list_move(&req->req_list, &cancel_list);
  189. }
  190. spin_unlock_irqrestore(&m->client->lock, flags);
  191. list_for_each_entry_safe(req, rtmp, &cancel_list, req_list) {
  192. P9_DPRINTK(P9_DEBUG_ERROR, "call back req %p\n", req);
  193. list_del(&req->req_list);
  194. p9_client_cb(m->client, req);
  195. }
  196. }
  197. static int
  198. p9_fd_poll(struct p9_client *client, struct poll_table_struct *pt)
  199. {
  200. int ret, n;
  201. struct p9_trans_fd *ts = NULL;
  202. if (client && client->status == Connected)
  203. ts = client->trans;
  204. if (!ts)
  205. return -EREMOTEIO;
  206. if (!ts->rd->f_op || !ts->rd->f_op->poll)
  207. return -EIO;
  208. if (!ts->wr->f_op || !ts->wr->f_op->poll)
  209. return -EIO;
  210. ret = ts->rd->f_op->poll(ts->rd, pt);
  211. if (ret < 0)
  212. return ret;
  213. if (ts->rd != ts->wr) {
  214. n = ts->wr->f_op->poll(ts->wr, pt);
  215. if (n < 0)
  216. return n;
  217. ret = (ret & ~POLLOUT) | (n & ~POLLIN);
  218. }
  219. return ret;
  220. }
  221. /**
  222. * p9_fd_read- read from a fd
  223. * @client: client instance
  224. * @v: buffer to receive data into
  225. * @len: size of receive buffer
  226. *
  227. */
  228. static int p9_fd_read(struct p9_client *client, void *v, int len)
  229. {
  230. int ret;
  231. struct p9_trans_fd *ts = NULL;
  232. if (client && client->status != Disconnected)
  233. ts = client->trans;
  234. if (!ts)
  235. return -EREMOTEIO;
  236. if (!(ts->rd->f_flags & O_NONBLOCK))
  237. P9_DPRINTK(P9_DEBUG_ERROR, "blocking read ...\n");
  238. ret = kernel_read(ts->rd, ts->rd->f_pos, v, len);
  239. if (ret <= 0 && ret != -ERESTARTSYS && ret != -EAGAIN)
  240. client->status = Disconnected;
  241. return ret;
  242. }
  243. /**
  244. * p9_read_work - called when there is some data to be read from a transport
  245. * @work: container of work to be done
  246. *
  247. */
  248. static void p9_read_work(struct work_struct *work)
  249. {
  250. int n, err;
  251. struct p9_conn *m;
  252. m = container_of(work, struct p9_conn, rq);
  253. if (m->err < 0)
  254. return;
  255. P9_DPRINTK(P9_DEBUG_TRANS, "start mux %p pos %d\n", m, m->rpos);
  256. if (!m->rbuf) {
  257. m->rbuf = m->tmp_buf;
  258. m->rpos = 0;
  259. m->rsize = 7; /* start by reading header */
  260. }
  261. clear_bit(Rpending, &m->wsched);
  262. P9_DPRINTK(P9_DEBUG_TRANS, "read mux %p pos %d size: %d = %d\n", m,
  263. m->rpos, m->rsize, m->rsize-m->rpos);
  264. err = p9_fd_read(m->client, m->rbuf + m->rpos,
  265. m->rsize - m->rpos);
  266. P9_DPRINTK(P9_DEBUG_TRANS, "mux %p got %d bytes\n", m, err);
  267. if (err == -EAGAIN) {
  268. clear_bit(Rworksched, &m->wsched);
  269. return;
  270. }
  271. if (err <= 0)
  272. goto error;
  273. m->rpos += err;
  274. if ((!m->req) && (m->rpos == m->rsize)) { /* header read in */
  275. u16 tag;
  276. P9_DPRINTK(P9_DEBUG_TRANS, "got new header\n");
  277. n = le32_to_cpu(*(__le32 *) m->rbuf); /* read packet size */
  278. if (n >= m->client->msize) {
  279. P9_DPRINTK(P9_DEBUG_ERROR,
  280. "requested packet size too big: %d\n", n);
  281. err = -EIO;
  282. goto error;
  283. }
  284. tag = le16_to_cpu(*(__le16 *) (m->rbuf+5)); /* read tag */
  285. P9_DPRINTK(P9_DEBUG_TRANS,
  286. "mux %p pkt: size: %d bytes tag: %d\n", m, n, tag);
  287. m->req = p9_tag_lookup(m->client, tag);
  288. if (!m->req || (m->req->status != REQ_STATUS_SENT &&
  289. m->req->status != REQ_STATUS_FLSH)) {
  290. P9_DPRINTK(P9_DEBUG_ERROR, "Unexpected packet tag %d\n",
  291. tag);
  292. err = -EIO;
  293. goto error;
  294. }
  295. if (m->req->rc == NULL) {
  296. m->req->rc = kmalloc(sizeof(struct p9_fcall) +
  297. m->client->msize, GFP_NOFS);
  298. if (!m->req->rc) {
  299. m->req = NULL;
  300. err = -ENOMEM;
  301. goto error;
  302. }
  303. }
  304. m->rbuf = (char *)m->req->rc + sizeof(struct p9_fcall);
  305. memcpy(m->rbuf, m->tmp_buf, m->rsize);
  306. m->rsize = n;
  307. }
  308. /* not an else because some packets (like clunk) have no payload */
  309. if ((m->req) && (m->rpos == m->rsize)) { /* packet is read in */
  310. P9_DPRINTK(P9_DEBUG_TRANS, "got new packet\n");
  311. spin_lock(&m->client->lock);
  312. if (m->req->status != REQ_STATUS_ERROR)
  313. m->req->status = REQ_STATUS_RCVD;
  314. list_del(&m->req->req_list);
  315. spin_unlock(&m->client->lock);
  316. p9_client_cb(m->client, m->req);
  317. m->rbuf = NULL;
  318. m->rpos = 0;
  319. m->rsize = 0;
  320. m->req = NULL;
  321. }
  322. if (!list_empty(&m->req_list)) {
  323. if (test_and_clear_bit(Rpending, &m->wsched))
  324. n = POLLIN;
  325. else
  326. n = p9_fd_poll(m->client, NULL);
  327. if (n & POLLIN) {
  328. P9_DPRINTK(P9_DEBUG_TRANS, "sched read work %p\n", m);
  329. schedule_work(&m->rq);
  330. } else
  331. clear_bit(Rworksched, &m->wsched);
  332. } else
  333. clear_bit(Rworksched, &m->wsched);
  334. return;
  335. error:
  336. p9_conn_cancel(m, err);
  337. clear_bit(Rworksched, &m->wsched);
  338. }
  339. /**
  340. * p9_fd_write - write to a socket
  341. * @client: client instance
  342. * @v: buffer to send data from
  343. * @len: size of send buffer
  344. *
  345. */
  346. static int p9_fd_write(struct p9_client *client, void *v, int len)
  347. {
  348. int ret;
  349. mm_segment_t oldfs;
  350. struct p9_trans_fd *ts = NULL;
  351. if (client && client->status != Disconnected)
  352. ts = client->trans;
  353. if (!ts)
  354. return -EREMOTEIO;
  355. if (!(ts->wr->f_flags & O_NONBLOCK))
  356. P9_DPRINTK(P9_DEBUG_ERROR, "blocking write ...\n");
  357. oldfs = get_fs();
  358. set_fs(get_ds());
  359. /* The cast to a user pointer is valid due to the set_fs() */
  360. ret = vfs_write(ts->wr, (__force void __user *)v, len, &ts->wr->f_pos);
  361. set_fs(oldfs);
  362. if (ret <= 0 && ret != -ERESTARTSYS && ret != -EAGAIN)
  363. client->status = Disconnected;
  364. return ret;
  365. }
  366. /**
  367. * p9_write_work - called when a transport can send some data
  368. * @work: container for work to be done
  369. *
  370. */
  371. static void p9_write_work(struct work_struct *work)
  372. {
  373. int n, err;
  374. struct p9_conn *m;
  375. struct p9_req_t *req;
  376. m = container_of(work, struct p9_conn, wq);
  377. if (m->err < 0) {
  378. clear_bit(Wworksched, &m->wsched);
  379. return;
  380. }
  381. if (!m->wsize) {
  382. if (list_empty(&m->unsent_req_list)) {
  383. clear_bit(Wworksched, &m->wsched);
  384. return;
  385. }
  386. spin_lock(&m->client->lock);
  387. req = list_entry(m->unsent_req_list.next, struct p9_req_t,
  388. req_list);
  389. req->status = REQ_STATUS_SENT;
  390. P9_DPRINTK(P9_DEBUG_TRANS, "move req %p\n", req);
  391. list_move_tail(&req->req_list, &m->req_list);
  392. m->wbuf = req->tc->sdata;
  393. m->wsize = req->tc->size;
  394. m->wpos = 0;
  395. spin_unlock(&m->client->lock);
  396. }
  397. P9_DPRINTK(P9_DEBUG_TRANS, "mux %p pos %d size %d\n", m, m->wpos,
  398. m->wsize);
  399. clear_bit(Wpending, &m->wsched);
  400. err = p9_fd_write(m->client, m->wbuf + m->wpos, m->wsize - m->wpos);
  401. P9_DPRINTK(P9_DEBUG_TRANS, "mux %p sent %d bytes\n", m, err);
  402. if (err == -EAGAIN) {
  403. clear_bit(Wworksched, &m->wsched);
  404. return;
  405. }
  406. if (err < 0)
  407. goto error;
  408. else if (err == 0) {
  409. err = -EREMOTEIO;
  410. goto error;
  411. }
  412. m->wpos += err;
  413. if (m->wpos == m->wsize)
  414. m->wpos = m->wsize = 0;
  415. if (m->wsize == 0 && !list_empty(&m->unsent_req_list)) {
  416. if (test_and_clear_bit(Wpending, &m->wsched))
  417. n = POLLOUT;
  418. else
  419. n = p9_fd_poll(m->client, NULL);
  420. if (n & POLLOUT) {
  421. P9_DPRINTK(P9_DEBUG_TRANS, "sched write work %p\n", m);
  422. schedule_work(&m->wq);
  423. } else
  424. clear_bit(Wworksched, &m->wsched);
  425. } else
  426. clear_bit(Wworksched, &m->wsched);
  427. return;
  428. error:
  429. p9_conn_cancel(m, err);
  430. clear_bit(Wworksched, &m->wsched);
  431. }
  432. static int p9_pollwake(wait_queue_t *wait, unsigned mode, int sync, void *key)
  433. {
  434. struct p9_poll_wait *pwait =
  435. container_of(wait, struct p9_poll_wait, wait);
  436. struct p9_conn *m = pwait->conn;
  437. unsigned long flags;
  438. spin_lock_irqsave(&p9_poll_lock, flags);
  439. if (list_empty(&m->poll_pending_link))
  440. list_add_tail(&m->poll_pending_link, &p9_poll_pending_list);
  441. spin_unlock_irqrestore(&p9_poll_lock, flags);
  442. schedule_work(&p9_poll_work);
  443. return 1;
  444. }
  445. /**
  446. * p9_pollwait - add poll task to the wait queue
  447. * @filp: file pointer being polled
  448. * @wait_address: wait_q to block on
  449. * @p: poll state
  450. *
  451. * called by files poll operation to add v9fs-poll task to files wait queue
  452. */
  453. static void
  454. p9_pollwait(struct file *filp, wait_queue_head_t *wait_address, poll_table *p)
  455. {
  456. struct p9_conn *m = container_of(p, struct p9_conn, pt);
  457. struct p9_poll_wait *pwait = NULL;
  458. int i;
  459. for (i = 0; i < ARRAY_SIZE(m->poll_wait); i++) {
  460. if (m->poll_wait[i].wait_addr == NULL) {
  461. pwait = &m->poll_wait[i];
  462. break;
  463. }
  464. }
  465. if (!pwait) {
  466. P9_DPRINTK(P9_DEBUG_ERROR, "not enough wait_address slots\n");
  467. return;
  468. }
  469. pwait->conn = m;
  470. pwait->wait_addr = wait_address;
  471. init_waitqueue_func_entry(&pwait->wait, p9_pollwake);
  472. add_wait_queue(wait_address, &pwait->wait);
  473. }
  474. /**
  475. * p9_conn_create - allocate and initialize the per-session mux data
  476. * @client: client instance
  477. *
  478. * Note: Creates the polling task if this is the first session.
  479. */
  480. static struct p9_conn *p9_conn_create(struct p9_client *client)
  481. {
  482. int n;
  483. struct p9_conn *m;
  484. P9_DPRINTK(P9_DEBUG_TRANS, "client %p msize %d\n", client,
  485. client->msize);
  486. m = kzalloc(sizeof(struct p9_conn), GFP_KERNEL);
  487. if (!m)
  488. return ERR_PTR(-ENOMEM);
  489. INIT_LIST_HEAD(&m->mux_list);
  490. m->client = client;
  491. INIT_LIST_HEAD(&m->req_list);
  492. INIT_LIST_HEAD(&m->unsent_req_list);
  493. INIT_WORK(&m->rq, p9_read_work);
  494. INIT_WORK(&m->wq, p9_write_work);
  495. INIT_LIST_HEAD(&m->poll_pending_link);
  496. init_poll_funcptr(&m->pt, p9_pollwait);
  497. n = p9_fd_poll(client, &m->pt);
  498. if (n & POLLIN) {
  499. P9_DPRINTK(P9_DEBUG_TRANS, "mux %p can read\n", m);
  500. set_bit(Rpending, &m->wsched);
  501. }
  502. if (n & POLLOUT) {
  503. P9_DPRINTK(P9_DEBUG_TRANS, "mux %p can write\n", m);
  504. set_bit(Wpending, &m->wsched);
  505. }
  506. return m;
  507. }
  508. /**
  509. * p9_poll_mux - polls a mux and schedules read or write works if necessary
  510. * @m: connection to poll
  511. *
  512. */
  513. static void p9_poll_mux(struct p9_conn *m)
  514. {
  515. int n;
  516. if (m->err < 0)
  517. return;
  518. n = p9_fd_poll(m->client, NULL);
  519. if (n < 0 || n & (POLLERR | POLLHUP | POLLNVAL)) {
  520. P9_DPRINTK(P9_DEBUG_TRANS, "error mux %p err %d\n", m, n);
  521. if (n >= 0)
  522. n = -ECONNRESET;
  523. p9_conn_cancel(m, n);
  524. }
  525. if (n & POLLIN) {
  526. set_bit(Rpending, &m->wsched);
  527. P9_DPRINTK(P9_DEBUG_TRANS, "mux %p can read\n", m);
  528. if (!test_and_set_bit(Rworksched, &m->wsched)) {
  529. P9_DPRINTK(P9_DEBUG_TRANS, "sched read work %p\n", m);
  530. schedule_work(&m->rq);
  531. }
  532. }
  533. if (n & POLLOUT) {
  534. set_bit(Wpending, &m->wsched);
  535. P9_DPRINTK(P9_DEBUG_TRANS, "mux %p can write\n", m);
  536. if ((m->wsize || !list_empty(&m->unsent_req_list)) &&
  537. !test_and_set_bit(Wworksched, &m->wsched)) {
  538. P9_DPRINTK(P9_DEBUG_TRANS, "sched write work %p\n", m);
  539. schedule_work(&m->wq);
  540. }
  541. }
  542. }
  543. /**
  544. * p9_fd_request - send 9P request
  545. * The function can sleep until the request is scheduled for sending.
  546. * The function can be interrupted. Return from the function is not
  547. * a guarantee that the request is sent successfully.
  548. *
  549. * @client: client instance
  550. * @req: request to be sent
  551. *
  552. */
  553. static int p9_fd_request(struct p9_client *client, struct p9_req_t *req)
  554. {
  555. int n;
  556. struct p9_trans_fd *ts = client->trans;
  557. struct p9_conn *m = ts->conn;
  558. P9_DPRINTK(P9_DEBUG_TRANS, "mux %p task %p tcall %p id %d\n", m,
  559. current, req->tc, req->tc->id);
  560. if (m->err < 0)
  561. return m->err;
  562. spin_lock(&client->lock);
  563. req->status = REQ_STATUS_UNSENT;
  564. list_add_tail(&req->req_list, &m->unsent_req_list);
  565. spin_unlock(&client->lock);
  566. if (test_and_clear_bit(Wpending, &m->wsched))
  567. n = POLLOUT;
  568. else
  569. n = p9_fd_poll(m->client, NULL);
  570. if (n & POLLOUT && !test_and_set_bit(Wworksched, &m->wsched))
  571. schedule_work(&m->wq);
  572. return 0;
  573. }
  574. static int p9_fd_cancel(struct p9_client *client, struct p9_req_t *req)
  575. {
  576. int ret = 1;
  577. P9_DPRINTK(P9_DEBUG_TRANS, "client %p req %p\n", client, req);
  578. spin_lock(&client->lock);
  579. if (req->status == REQ_STATUS_UNSENT) {
  580. list_del(&req->req_list);
  581. req->status = REQ_STATUS_FLSHD;
  582. ret = 0;
  583. } else if (req->status == REQ_STATUS_SENT)
  584. req->status = REQ_STATUS_FLSH;
  585. spin_unlock(&client->lock);
  586. return ret;
  587. }
  588. /**
  589. * parse_opts - parse mount options into p9_fd_opts structure
  590. * @params: options string passed from mount
  591. * @opts: fd transport-specific structure to parse options into
  592. *
  593. * Returns 0 upon success, -ERRNO upon failure
  594. */
  595. static int parse_opts(char *params, struct p9_fd_opts *opts)
  596. {
  597. char *p;
  598. substring_t args[MAX_OPT_ARGS];
  599. int option;
  600. char *options, *tmp_options;
  601. opts->port = P9_PORT;
  602. opts->rfd = ~0;
  603. opts->wfd = ~0;
  604. if (!params)
  605. return 0;
  606. tmp_options = kstrdup(params, GFP_KERNEL);
  607. if (!tmp_options) {
  608. P9_DPRINTK(P9_DEBUG_ERROR,
  609. "failed to allocate copy of option string\n");
  610. return -ENOMEM;
  611. }
  612. options = tmp_options;
  613. while ((p = strsep(&options, ",")) != NULL) {
  614. int token;
  615. int r;
  616. if (!*p)
  617. continue;
  618. token = match_token(p, tokens, args);
  619. if (token != Opt_err) {
  620. r = match_int(&args[0], &option);
  621. if (r < 0) {
  622. P9_DPRINTK(P9_DEBUG_ERROR,
  623. "integer field, but no integer?\n");
  624. continue;
  625. }
  626. }
  627. switch (token) {
  628. case Opt_port:
  629. opts->port = option;
  630. break;
  631. case Opt_rfdno:
  632. opts->rfd = option;
  633. break;
  634. case Opt_wfdno:
  635. opts->wfd = option;
  636. break;
  637. default:
  638. continue;
  639. }
  640. }
  641. kfree(tmp_options);
  642. return 0;
  643. }
  644. static int p9_fd_open(struct p9_client *client, int rfd, int wfd)
  645. {
  646. struct p9_trans_fd *ts = kmalloc(sizeof(struct p9_trans_fd),
  647. GFP_KERNEL);
  648. if (!ts)
  649. return -ENOMEM;
  650. ts->rd = fget(rfd);
  651. ts->wr = fget(wfd);
  652. if (!ts->rd || !ts->wr) {
  653. if (ts->rd)
  654. fput(ts->rd);
  655. if (ts->wr)
  656. fput(ts->wr);
  657. kfree(ts);
  658. return -EIO;
  659. }
  660. client->trans = ts;
  661. client->status = Connected;
  662. return 0;
  663. }
  664. static int p9_socket_open(struct p9_client *client, struct socket *csocket)
  665. {
  666. struct p9_trans_fd *p;
  667. int ret, fd;
  668. p = kmalloc(sizeof(struct p9_trans_fd), GFP_KERNEL);
  669. if (!p)
  670. return -ENOMEM;
  671. csocket->sk->sk_allocation = GFP_NOIO;
  672. fd = sock_map_fd(csocket, 0);
  673. if (fd < 0) {
  674. P9_EPRINTK(KERN_ERR, "p9_socket_open: failed to map fd\n");
  675. sock_release(csocket);
  676. kfree(p);
  677. return fd;
  678. }
  679. get_file(csocket->file);
  680. get_file(csocket->file);
  681. p->wr = p->rd = csocket->file;
  682. client->trans = p;
  683. client->status = Connected;
  684. sys_close(fd); /* still racy */
  685. p->rd->f_flags |= O_NONBLOCK;
  686. p->conn = p9_conn_create(client);
  687. if (IS_ERR(p->conn)) {
  688. ret = PTR_ERR(p->conn);
  689. p->conn = NULL;
  690. kfree(p);
  691. sockfd_put(csocket);
  692. sockfd_put(csocket);
  693. return ret;
  694. }
  695. return 0;
  696. }
  697. /**
  698. * p9_mux_destroy - cancels all pending requests and frees mux resources
  699. * @m: mux to destroy
  700. *
  701. */
  702. static void p9_conn_destroy(struct p9_conn *m)
  703. {
  704. P9_DPRINTK(P9_DEBUG_TRANS, "mux %p prev %p next %p\n", m,
  705. m->mux_list.prev, m->mux_list.next);
  706. p9_mux_poll_stop(m);
  707. cancel_work_sync(&m->rq);
  708. cancel_work_sync(&m->wq);
  709. p9_conn_cancel(m, -ECONNRESET);
  710. m->client = NULL;
  711. kfree(m);
  712. }
  713. /**
  714. * p9_fd_close - shutdown file descriptor transport
  715. * @client: client instance
  716. *
  717. */
  718. static void p9_fd_close(struct p9_client *client)
  719. {
  720. struct p9_trans_fd *ts;
  721. if (!client)
  722. return;
  723. ts = client->trans;
  724. if (!ts)
  725. return;
  726. client->status = Disconnected;
  727. p9_conn_destroy(ts->conn);
  728. if (ts->rd)
  729. fput(ts->rd);
  730. if (ts->wr)
  731. fput(ts->wr);
  732. kfree(ts);
  733. }
  734. /*
  735. * stolen from NFS - maybe should be made a generic function?
  736. */
  737. static inline int valid_ipaddr4(const char *buf)
  738. {
  739. int rc, count, in[4];
  740. rc = sscanf(buf, "%d.%d.%d.%d", &in[0], &in[1], &in[2], &in[3]);
  741. if (rc != 4)
  742. return -EINVAL;
  743. for (count = 0; count < 4; count++) {
  744. if (in[count] > 255)
  745. return -EINVAL;
  746. }
  747. return 0;
  748. }
  749. static int
  750. p9_fd_create_tcp(struct p9_client *client, const char *addr, char *args)
  751. {
  752. int err;
  753. struct socket *csocket;
  754. struct sockaddr_in sin_server;
  755. struct p9_fd_opts opts;
  756. err = parse_opts(args, &opts);
  757. if (err < 0)
  758. return err;
  759. if (valid_ipaddr4(addr) < 0)
  760. return -EINVAL;
  761. csocket = NULL;
  762. sin_server.sin_family = AF_INET;
  763. sin_server.sin_addr.s_addr = in_aton(addr);
  764. sin_server.sin_port = htons(opts.port);
  765. err = __sock_create(read_pnet(&current->nsproxy->net_ns), PF_INET,
  766. SOCK_STREAM, IPPROTO_TCP, &csocket, 1);
  767. if (err) {
  768. P9_EPRINTK(KERN_ERR, "p9_trans_tcp: problem creating socket\n");
  769. return err;
  770. }
  771. err = csocket->ops->connect(csocket,
  772. (struct sockaddr *)&sin_server,
  773. sizeof(struct sockaddr_in), 0);
  774. if (err < 0) {
  775. P9_EPRINTK(KERN_ERR,
  776. "p9_trans_tcp: problem connecting socket to %s\n",
  777. addr);
  778. sock_release(csocket);
  779. return err;
  780. }
  781. return p9_socket_open(client, csocket);
  782. }
  783. static int
  784. p9_fd_create_unix(struct p9_client *client, const char *addr, char *args)
  785. {
  786. int err;
  787. struct socket *csocket;
  788. struct sockaddr_un sun_server;
  789. csocket = NULL;
  790. if (strlen(addr) >= UNIX_PATH_MAX) {
  791. P9_EPRINTK(KERN_ERR, "p9_trans_unix: address too long: %s\n",
  792. addr);
  793. return -ENAMETOOLONG;
  794. }
  795. sun_server.sun_family = PF_UNIX;
  796. strcpy(sun_server.sun_path, addr);
  797. err = __sock_create(read_pnet(&current->nsproxy->net_ns), PF_UNIX,
  798. SOCK_STREAM, 0, &csocket, 1);
  799. if (err < 0) {
  800. P9_EPRINTK(KERN_ERR, "p9_trans_unix: problem creating socket\n");
  801. return err;
  802. }
  803. err = csocket->ops->connect(csocket, (struct sockaddr *)&sun_server,
  804. sizeof(struct sockaddr_un) - 1, 0);
  805. if (err < 0) {
  806. P9_EPRINTK(KERN_ERR,
  807. "p9_trans_unix: problem connecting socket: %s: %d\n",
  808. addr, err);
  809. sock_release(csocket);
  810. return err;
  811. }
  812. return p9_socket_open(client, csocket);
  813. }
  814. static int
  815. p9_fd_create(struct p9_client *client, const char *addr, char *args)
  816. {
  817. int err;
  818. struct p9_fd_opts opts;
  819. struct p9_trans_fd *p;
  820. parse_opts(args, &opts);
  821. if (opts.rfd == ~0 || opts.wfd == ~0) {
  822. printk(KERN_ERR "v9fs: Insufficient options for proto=fd\n");
  823. return -ENOPROTOOPT;
  824. }
  825. err = p9_fd_open(client, opts.rfd, opts.wfd);
  826. if (err < 0)
  827. return err;
  828. p = (struct p9_trans_fd *) client->trans;
  829. p->conn = p9_conn_create(client);
  830. if (IS_ERR(p->conn)) {
  831. err = PTR_ERR(p->conn);
  832. p->conn = NULL;
  833. fput(p->rd);
  834. fput(p->wr);
  835. return err;
  836. }
  837. return 0;
  838. }
  839. static struct p9_trans_module p9_tcp_trans = {
  840. .name = "tcp",
  841. .maxsize = MAX_SOCK_BUF,
  842. .def = 1,
  843. .create = p9_fd_create_tcp,
  844. .close = p9_fd_close,
  845. .request = p9_fd_request,
  846. .cancel = p9_fd_cancel,
  847. .owner = THIS_MODULE,
  848. };
  849. static struct p9_trans_module p9_unix_trans = {
  850. .name = "unix",
  851. .maxsize = MAX_SOCK_BUF,
  852. .def = 0,
  853. .create = p9_fd_create_unix,
  854. .close = p9_fd_close,
  855. .request = p9_fd_request,
  856. .cancel = p9_fd_cancel,
  857. .owner = THIS_MODULE,
  858. };
  859. static struct p9_trans_module p9_fd_trans = {
  860. .name = "fd",
  861. .maxsize = MAX_SOCK_BUF,
  862. .def = 0,
  863. .create = p9_fd_create,
  864. .close = p9_fd_close,
  865. .request = p9_fd_request,
  866. .cancel = p9_fd_cancel,
  867. .owner = THIS_MODULE,
  868. };
  869. /**
  870. * p9_poll_proc - poll worker thread
  871. * @a: thread state and arguments
  872. *
  873. * polls all v9fs transports for new events and queues the appropriate
  874. * work to the work queue
  875. *
  876. */
  877. static void p9_poll_workfn(struct work_struct *work)
  878. {
  879. unsigned long flags;
  880. P9_DPRINTK(P9_DEBUG_TRANS, "start %p\n", current);
  881. spin_lock_irqsave(&p9_poll_lock, flags);
  882. while (!list_empty(&p9_poll_pending_list)) {
  883. struct p9_conn *conn = list_first_entry(&p9_poll_pending_list,
  884. struct p9_conn,
  885. poll_pending_link);
  886. list_del_init(&conn->poll_pending_link);
  887. spin_unlock_irqrestore(&p9_poll_lock, flags);
  888. p9_poll_mux(conn);
  889. spin_lock_irqsave(&p9_poll_lock, flags);
  890. }
  891. spin_unlock_irqrestore(&p9_poll_lock, flags);
  892. P9_DPRINTK(P9_DEBUG_TRANS, "finish\n");
  893. }
  894. int p9_trans_fd_init(void)
  895. {
  896. v9fs_register_trans(&p9_tcp_trans);
  897. v9fs_register_trans(&p9_unix_trans);
  898. v9fs_register_trans(&p9_fd_trans);
  899. return 0;
  900. }
  901. void p9_trans_fd_exit(void)
  902. {
  903. flush_work_sync(&p9_poll_work);
  904. v9fs_unregister_trans(&p9_tcp_trans);
  905. v9fs_unregister_trans(&p9_unix_trans);
  906. v9fs_unregister_trans(&p9_fd_trans);
  907. }