trans_fd.c 24 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111
  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 <net/9p/9p.h>
  41. #include <net/9p/client.h>
  42. #include <net/9p/transport.h>
  43. #include <linux/syscalls.h> /* killme */
  44. #define P9_PORT 564
  45. #define MAX_SOCK_BUF (64*1024)
  46. #define MAXPOLLWADDR 2
  47. /**
  48. * struct p9_fd_opts - per-transport options
  49. * @rfd: file descriptor for reading (trans=fd)
  50. * @wfd: file descriptor for writing (trans=fd)
  51. * @port: port to connect to (trans=tcp)
  52. *
  53. */
  54. struct p9_fd_opts {
  55. int rfd;
  56. int wfd;
  57. u16 port;
  58. };
  59. /**
  60. * struct p9_trans_fd - transport state
  61. * @rd: reference to file to read from
  62. * @wr: reference of file to write to
  63. * @conn: connection state reference
  64. *
  65. */
  66. struct p9_trans_fd {
  67. struct file *rd;
  68. struct file *wr;
  69. struct p9_conn *conn;
  70. };
  71. /*
  72. * Option Parsing (code inspired by NFS code)
  73. * - a little lazy - parse all fd-transport options
  74. */
  75. enum {
  76. /* Options that take integer arguments */
  77. Opt_port, Opt_rfdno, Opt_wfdno, Opt_err,
  78. };
  79. static const match_table_t tokens = {
  80. {Opt_port, "port=%u"},
  81. {Opt_rfdno, "rfdno=%u"},
  82. {Opt_wfdno, "wfdno=%u"},
  83. {Opt_err, NULL},
  84. };
  85. enum {
  86. Rworksched = 1, /* read work scheduled or running */
  87. Rpending = 2, /* can read */
  88. Wworksched = 4, /* write work scheduled or running */
  89. Wpending = 8, /* can write */
  90. };
  91. struct p9_poll_wait {
  92. struct p9_conn *conn;
  93. wait_queue_t wait;
  94. wait_queue_head_t *wait_addr;
  95. };
  96. /**
  97. * struct p9_conn - fd mux connection state information
  98. * @mux_list: list link for mux to manage multiple connections (?)
  99. * @client: reference to client instance for this connection
  100. * @err: error state
  101. * @req_list: accounting for requests which have been sent
  102. * @unsent_req_list: accounting for requests that haven't been sent
  103. * @req: current request being processed (if any)
  104. * @tmp_buf: temporary buffer to read in header
  105. * @rsize: amount to read for current frame
  106. * @rpos: read position in current frame
  107. * @rbuf: current read buffer
  108. * @wpos: write position for current frame
  109. * @wsize: amount of data to write for current frame
  110. * @wbuf: current write buffer
  111. * @poll_pending_link: pending links to be polled per conn
  112. * @poll_wait: array of wait_q's for various worker threads
  113. * @pt: poll state
  114. * @rq: current read work
  115. * @wq: current write work
  116. * @wsched: ????
  117. *
  118. */
  119. struct p9_conn {
  120. struct list_head mux_list;
  121. struct p9_client *client;
  122. int err;
  123. struct list_head req_list;
  124. struct list_head unsent_req_list;
  125. struct p9_req_t *req;
  126. char tmp_buf[7];
  127. int rsize;
  128. int rpos;
  129. char *rbuf;
  130. int wpos;
  131. int wsize;
  132. char *wbuf;
  133. struct list_head poll_pending_link;
  134. struct p9_poll_wait poll_wait[MAXPOLLWADDR];
  135. poll_table pt;
  136. struct work_struct rq;
  137. struct work_struct wq;
  138. unsigned long wsched;
  139. };
  140. static DEFINE_SPINLOCK(p9_poll_lock);
  141. static LIST_HEAD(p9_poll_pending_list);
  142. static struct workqueue_struct *p9_mux_wq;
  143. static struct task_struct *p9_poll_task;
  144. static void p9_mux_poll_stop(struct p9_conn *m)
  145. {
  146. unsigned long flags;
  147. int i;
  148. for (i = 0; i < ARRAY_SIZE(m->poll_wait); i++) {
  149. struct p9_poll_wait *pwait = &m->poll_wait[i];
  150. if (pwait->wait_addr) {
  151. remove_wait_queue(pwait->wait_addr, &pwait->wait);
  152. pwait->wait_addr = NULL;
  153. }
  154. }
  155. spin_lock_irqsave(&p9_poll_lock, flags);
  156. list_del_init(&m->poll_pending_link);
  157. spin_unlock_irqrestore(&p9_poll_lock, flags);
  158. }
  159. /**
  160. * p9_conn_cancel - cancel all pending requests with error
  161. * @m: mux data
  162. * @err: error code
  163. *
  164. */
  165. static void p9_conn_cancel(struct p9_conn *m, int err)
  166. {
  167. struct p9_req_t *req, *rtmp;
  168. unsigned long flags;
  169. LIST_HEAD(cancel_list);
  170. P9_DPRINTK(P9_DEBUG_ERROR, "mux %p err %d\n", m, err);
  171. spin_lock_irqsave(&m->client->lock, flags);
  172. if (m->err) {
  173. spin_unlock_irqrestore(&m->client->lock, flags);
  174. return;
  175. }
  176. m->err = err;
  177. list_for_each_entry_safe(req, rtmp, &m->req_list, req_list) {
  178. req->status = REQ_STATUS_ERROR;
  179. if (!req->t_err)
  180. req->t_err = err;
  181. list_move(&req->req_list, &cancel_list);
  182. }
  183. list_for_each_entry_safe(req, rtmp, &m->unsent_req_list, req_list) {
  184. req->status = REQ_STATUS_ERROR;
  185. if (!req->t_err)
  186. req->t_err = err;
  187. list_move(&req->req_list, &cancel_list);
  188. }
  189. spin_unlock_irqrestore(&m->client->lock, flags);
  190. list_for_each_entry_safe(req, rtmp, &cancel_list, req_list) {
  191. P9_DPRINTK(P9_DEBUG_ERROR, "call back req %p\n", req);
  192. list_del(&req->req_list);
  193. p9_client_cb(m->client, req);
  194. }
  195. }
  196. static unsigned int
  197. p9_fd_poll(struct p9_client *client, struct poll_table_struct *pt)
  198. {
  199. int ret, n;
  200. struct p9_trans_fd *ts = NULL;
  201. if (client && client->status == Connected)
  202. ts = client->trans;
  203. if (!ts)
  204. return -EREMOTEIO;
  205. if (!ts->rd->f_op || !ts->rd->f_op->poll)
  206. return -EIO;
  207. if (!ts->wr->f_op || !ts->wr->f_op->poll)
  208. return -EIO;
  209. ret = ts->rd->f_op->poll(ts->rd, pt);
  210. if (ret < 0)
  211. return ret;
  212. if (ts->rd != ts->wr) {
  213. n = ts->wr->f_op->poll(ts->wr, pt);
  214. if (n < 0)
  215. return n;
  216. ret = (ret & ~POLLOUT) | (n & ~POLLIN);
  217. }
  218. return ret;
  219. }
  220. /**
  221. * p9_fd_read- read from a fd
  222. * @client: client instance
  223. * @v: buffer to receive data into
  224. * @len: size of receive buffer
  225. *
  226. */
  227. static int p9_fd_read(struct p9_client *client, void *v, int len)
  228. {
  229. int ret;
  230. struct p9_trans_fd *ts = NULL;
  231. if (client && client->status != Disconnected)
  232. ts = client->trans;
  233. if (!ts)
  234. return -EREMOTEIO;
  235. if (!(ts->rd->f_flags & O_NONBLOCK))
  236. P9_DPRINTK(P9_DEBUG_ERROR, "blocking read ...\n");
  237. ret = kernel_read(ts->rd, ts->rd->f_pos, v, len);
  238. if (ret <= 0 && ret != -ERESTARTSYS && ret != -EAGAIN)
  239. client->status = Disconnected;
  240. return ret;
  241. }
  242. /**
  243. * p9_read_work - called when there is some data to be read from a transport
  244. * @work: container of work to be done
  245. *
  246. */
  247. static void p9_read_work(struct work_struct *work)
  248. {
  249. int n, err;
  250. struct p9_conn *m;
  251. m = container_of(work, struct p9_conn, rq);
  252. if (m->err < 0)
  253. return;
  254. P9_DPRINTK(P9_DEBUG_TRANS, "start mux %p pos %d\n", m, m->rpos);
  255. if (!m->rbuf) {
  256. m->rbuf = m->tmp_buf;
  257. m->rpos = 0;
  258. m->rsize = 7; /* start by reading header */
  259. }
  260. clear_bit(Rpending, &m->wsched);
  261. P9_DPRINTK(P9_DEBUG_TRANS, "read mux %p pos %d size: %d = %d\n", m,
  262. m->rpos, m->rsize, m->rsize-m->rpos);
  263. err = p9_fd_read(m->client, m->rbuf + m->rpos,
  264. m->rsize - m->rpos);
  265. P9_DPRINTK(P9_DEBUG_TRANS, "mux %p got %d bytes\n", m, err);
  266. if (err == -EAGAIN) {
  267. clear_bit(Rworksched, &m->wsched);
  268. return;
  269. }
  270. if (err <= 0)
  271. goto error;
  272. m->rpos += err;
  273. if ((!m->req) && (m->rpos == m->rsize)) { /* header read in */
  274. u16 tag;
  275. P9_DPRINTK(P9_DEBUG_TRANS, "got new header\n");
  276. n = le32_to_cpu(*(__le32 *) m->rbuf); /* read packet size */
  277. if (n >= m->client->msize) {
  278. P9_DPRINTK(P9_DEBUG_ERROR,
  279. "requested packet size too big: %d\n", n);
  280. err = -EIO;
  281. goto error;
  282. }
  283. tag = le16_to_cpu(*(__le16 *) (m->rbuf+5)); /* read tag */
  284. P9_DPRINTK(P9_DEBUG_TRANS,
  285. "mux %p pkt: size: %d bytes tag: %d\n", m, n, tag);
  286. m->req = p9_tag_lookup(m->client, tag);
  287. if (!m->req || (m->req->status != REQ_STATUS_SENT &&
  288. m->req->status != REQ_STATUS_FLSH)) {
  289. P9_DPRINTK(P9_DEBUG_ERROR, "Unexpected packet tag %d\n",
  290. tag);
  291. err = -EIO;
  292. goto error;
  293. }
  294. if (m->req->rc == NULL) {
  295. m->req->rc = kmalloc(sizeof(struct p9_fcall) +
  296. m->client->msize, GFP_KERNEL);
  297. if (!m->req->rc) {
  298. m->req = NULL;
  299. err = -ENOMEM;
  300. goto error;
  301. }
  302. }
  303. m->rbuf = (char *)m->req->rc + sizeof(struct p9_fcall);
  304. memcpy(m->rbuf, m->tmp_buf, m->rsize);
  305. m->rsize = n;
  306. }
  307. /* not an else because some packets (like clunk) have no payload */
  308. if ((m->req) && (m->rpos == m->rsize)) { /* packet is read in */
  309. P9_DPRINTK(P9_DEBUG_TRANS, "got new packet\n");
  310. spin_lock(&m->client->lock);
  311. if (m->req->status != REQ_STATUS_ERROR)
  312. m->req->status = REQ_STATUS_RCVD;
  313. list_del(&m->req->req_list);
  314. spin_unlock(&m->client->lock);
  315. p9_client_cb(m->client, m->req);
  316. m->rbuf = NULL;
  317. m->rpos = 0;
  318. m->rsize = 0;
  319. m->req = NULL;
  320. }
  321. if (!list_empty(&m->req_list)) {
  322. if (test_and_clear_bit(Rpending, &m->wsched))
  323. n = POLLIN;
  324. else
  325. n = p9_fd_poll(m->client, NULL);
  326. if (n & POLLIN) {
  327. P9_DPRINTK(P9_DEBUG_TRANS, "sched read work %p\n", m);
  328. queue_work(p9_mux_wq, &m->rq);
  329. } else
  330. clear_bit(Rworksched, &m->wsched);
  331. } else
  332. clear_bit(Rworksched, &m->wsched);
  333. return;
  334. error:
  335. p9_conn_cancel(m, err);
  336. clear_bit(Rworksched, &m->wsched);
  337. }
  338. /**
  339. * p9_fd_write - write to a socket
  340. * @client: client instance
  341. * @v: buffer to send data from
  342. * @len: size of send buffer
  343. *
  344. */
  345. static int p9_fd_write(struct p9_client *client, void *v, int len)
  346. {
  347. int ret;
  348. mm_segment_t oldfs;
  349. struct p9_trans_fd *ts = NULL;
  350. if (client && client->status != Disconnected)
  351. ts = client->trans;
  352. if (!ts)
  353. return -EREMOTEIO;
  354. if (!(ts->wr->f_flags & O_NONBLOCK))
  355. P9_DPRINTK(P9_DEBUG_ERROR, "blocking write ...\n");
  356. oldfs = get_fs();
  357. set_fs(get_ds());
  358. /* The cast to a user pointer is valid due to the set_fs() */
  359. ret = vfs_write(ts->wr, (__force void __user *)v, len, &ts->wr->f_pos);
  360. set_fs(oldfs);
  361. if (ret <= 0 && ret != -ERESTARTSYS && ret != -EAGAIN)
  362. client->status = Disconnected;
  363. return ret;
  364. }
  365. /**
  366. * p9_write_work - called when a transport can send some data
  367. * @work: container for work to be done
  368. *
  369. */
  370. static void p9_write_work(struct work_struct *work)
  371. {
  372. int n, err;
  373. struct p9_conn *m;
  374. struct p9_req_t *req;
  375. m = container_of(work, struct p9_conn, wq);
  376. if (m->err < 0) {
  377. clear_bit(Wworksched, &m->wsched);
  378. return;
  379. }
  380. if (!m->wsize) {
  381. if (list_empty(&m->unsent_req_list)) {
  382. clear_bit(Wworksched, &m->wsched);
  383. return;
  384. }
  385. spin_lock(&m->client->lock);
  386. req = list_entry(m->unsent_req_list.next, struct p9_req_t,
  387. req_list);
  388. req->status = REQ_STATUS_SENT;
  389. P9_DPRINTK(P9_DEBUG_TRANS, "move req %p\n", req);
  390. list_move_tail(&req->req_list, &m->req_list);
  391. m->wbuf = req->tc->sdata;
  392. m->wsize = req->tc->size;
  393. m->wpos = 0;
  394. spin_unlock(&m->client->lock);
  395. }
  396. P9_DPRINTK(P9_DEBUG_TRANS, "mux %p pos %d size %d\n", m, m->wpos,
  397. m->wsize);
  398. clear_bit(Wpending, &m->wsched);
  399. err = p9_fd_write(m->client, m->wbuf + m->wpos, m->wsize - m->wpos);
  400. P9_DPRINTK(P9_DEBUG_TRANS, "mux %p sent %d bytes\n", m, err);
  401. if (err == -EAGAIN) {
  402. clear_bit(Wworksched, &m->wsched);
  403. return;
  404. }
  405. if (err < 0)
  406. goto error;
  407. else if (err == 0) {
  408. err = -EREMOTEIO;
  409. goto error;
  410. }
  411. m->wpos += err;
  412. if (m->wpos == m->wsize)
  413. m->wpos = m->wsize = 0;
  414. if (m->wsize == 0 && !list_empty(&m->unsent_req_list)) {
  415. if (test_and_clear_bit(Wpending, &m->wsched))
  416. n = POLLOUT;
  417. else
  418. n = p9_fd_poll(m->client, NULL);
  419. if (n & POLLOUT) {
  420. P9_DPRINTK(P9_DEBUG_TRANS, "sched write work %p\n", m);
  421. queue_work(p9_mux_wq, &m->wq);
  422. } else
  423. clear_bit(Wworksched, &m->wsched);
  424. } else
  425. clear_bit(Wworksched, &m->wsched);
  426. return;
  427. error:
  428. p9_conn_cancel(m, err);
  429. clear_bit(Wworksched, &m->wsched);
  430. }
  431. static int p9_pollwake(wait_queue_t *wait, unsigned mode, int sync, void *key)
  432. {
  433. struct p9_poll_wait *pwait =
  434. container_of(wait, struct p9_poll_wait, wait);
  435. struct p9_conn *m = pwait->conn;
  436. unsigned long flags;
  437. DECLARE_WAITQUEUE(dummy_wait, p9_poll_task);
  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. /* perform the default wake up operation */
  443. return default_wake_function(&dummy_wait, mode, sync, key);
  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. queue_work(p9_mux_wq, &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. queue_work(p9_mux_wq, &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. queue_work(p9_mux_wq, &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;
  601. int ret;
  602. opts->port = P9_PORT;
  603. opts->rfd = ~0;
  604. opts->wfd = ~0;
  605. if (!params)
  606. return 0;
  607. options = kstrdup(params, GFP_KERNEL);
  608. if (!options) {
  609. P9_DPRINTK(P9_DEBUG_ERROR,
  610. "failed to allocate copy of option string\n");
  611. return -ENOMEM;
  612. }
  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. ret = r;
  625. continue;
  626. }
  627. }
  628. switch (token) {
  629. case Opt_port:
  630. opts->port = option;
  631. break;
  632. case Opt_rfdno:
  633. opts->rfd = option;
  634. break;
  635. case Opt_wfdno:
  636. opts->wfd = option;
  637. break;
  638. default:
  639. continue;
  640. }
  641. }
  642. kfree(options);
  643. return 0;
  644. }
  645. static int p9_fd_open(struct p9_client *client, int rfd, int wfd)
  646. {
  647. struct p9_trans_fd *ts = kmalloc(sizeof(struct p9_trans_fd),
  648. GFP_KERNEL);
  649. if (!ts)
  650. return -ENOMEM;
  651. ts->rd = fget(rfd);
  652. ts->wr = fget(wfd);
  653. if (!ts->rd || !ts->wr) {
  654. if (ts->rd)
  655. fput(ts->rd);
  656. if (ts->wr)
  657. fput(ts->wr);
  658. kfree(ts);
  659. return -EIO;
  660. }
  661. client->trans = ts;
  662. client->status = Connected;
  663. return 0;
  664. }
  665. static int p9_socket_open(struct p9_client *client, struct socket *csocket)
  666. {
  667. struct p9_trans_fd *p;
  668. int ret, fd;
  669. p = kmalloc(sizeof(struct p9_trans_fd), GFP_KERNEL);
  670. if (!p)
  671. return -ENOMEM;
  672. csocket->sk->sk_allocation = GFP_NOIO;
  673. fd = sock_map_fd(csocket, 0);
  674. if (fd < 0) {
  675. P9_EPRINTK(KERN_ERR, "p9_socket_open: failed to map fd\n");
  676. sock_release(csocket);
  677. kfree(p);
  678. return fd;
  679. }
  680. get_file(csocket->file);
  681. get_file(csocket->file);
  682. p->wr = p->rd = csocket->file;
  683. client->trans = p;
  684. client->status = Connected;
  685. sys_close(fd); /* still racy */
  686. p->rd->f_flags |= O_NONBLOCK;
  687. p->conn = p9_conn_create(client);
  688. if (IS_ERR(p->conn)) {
  689. ret = PTR_ERR(p->conn);
  690. p->conn = NULL;
  691. kfree(p);
  692. sockfd_put(csocket);
  693. sockfd_put(csocket);
  694. return ret;
  695. }
  696. return 0;
  697. }
  698. /**
  699. * p9_mux_destroy - cancels all pending requests and frees mux resources
  700. * @m: mux to destroy
  701. *
  702. */
  703. static void p9_conn_destroy(struct p9_conn *m)
  704. {
  705. P9_DPRINTK(P9_DEBUG_TRANS, "mux %p prev %p next %p\n", m,
  706. m->mux_list.prev, m->mux_list.next);
  707. p9_mux_poll_stop(m);
  708. cancel_work_sync(&m->rq);
  709. cancel_work_sync(&m->wq);
  710. p9_conn_cancel(m, -ECONNRESET);
  711. m->client = NULL;
  712. kfree(m);
  713. }
  714. /**
  715. * p9_fd_close - shutdown file descriptor transport
  716. * @client: client instance
  717. *
  718. */
  719. static void p9_fd_close(struct p9_client *client)
  720. {
  721. struct p9_trans_fd *ts;
  722. if (!client)
  723. return;
  724. ts = client->trans;
  725. if (!ts)
  726. return;
  727. client->status = Disconnected;
  728. p9_conn_destroy(ts->conn);
  729. if (ts->rd)
  730. fput(ts->rd);
  731. if (ts->wr)
  732. fput(ts->wr);
  733. kfree(ts);
  734. }
  735. /*
  736. * stolen from NFS - maybe should be made a generic function?
  737. */
  738. static inline int valid_ipaddr4(const char *buf)
  739. {
  740. int rc, count, in[4];
  741. rc = sscanf(buf, "%d.%d.%d.%d", &in[0], &in[1], &in[2], &in[3]);
  742. if (rc != 4)
  743. return -EINVAL;
  744. for (count = 0; count < 4; count++) {
  745. if (in[count] > 255)
  746. return -EINVAL;
  747. }
  748. return 0;
  749. }
  750. static int
  751. p9_fd_create_tcp(struct p9_client *client, const char *addr, char *args)
  752. {
  753. int err;
  754. struct socket *csocket;
  755. struct sockaddr_in sin_server;
  756. struct p9_fd_opts opts;
  757. err = parse_opts(args, &opts);
  758. if (err < 0)
  759. return err;
  760. if (valid_ipaddr4(addr) < 0)
  761. return -EINVAL;
  762. csocket = NULL;
  763. sin_server.sin_family = AF_INET;
  764. sin_server.sin_addr.s_addr = in_aton(addr);
  765. sin_server.sin_port = htons(opts.port);
  766. err = sock_create_kern(PF_INET, SOCK_STREAM, IPPROTO_TCP, &csocket);
  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_kern(PF_UNIX, SOCK_STREAM, 0, &csocket);
  798. if (err < 0) {
  799. P9_EPRINTK(KERN_ERR, "p9_trans_unix: problem creating socket\n");
  800. return err;
  801. }
  802. err = csocket->ops->connect(csocket, (struct sockaddr *)&sun_server,
  803. sizeof(struct sockaddr_un) - 1, 0);
  804. if (err < 0) {
  805. P9_EPRINTK(KERN_ERR,
  806. "p9_trans_unix: problem connecting socket: %s: %d\n",
  807. addr, err);
  808. sock_release(csocket);
  809. return err;
  810. }
  811. return p9_socket_open(client, csocket);
  812. }
  813. static int
  814. p9_fd_create(struct p9_client *client, const char *addr, char *args)
  815. {
  816. int err;
  817. struct p9_fd_opts opts;
  818. struct p9_trans_fd *p;
  819. parse_opts(args, &opts);
  820. if (opts.rfd == ~0 || opts.wfd == ~0) {
  821. printk(KERN_ERR "v9fs: Insufficient options for proto=fd\n");
  822. return -ENOPROTOOPT;
  823. }
  824. err = p9_fd_open(client, opts.rfd, opts.wfd);
  825. if (err < 0)
  826. return err;
  827. p = (struct p9_trans_fd *) client->trans;
  828. p->conn = p9_conn_create(client);
  829. if (IS_ERR(p->conn)) {
  830. err = PTR_ERR(p->conn);
  831. p->conn = NULL;
  832. fput(p->rd);
  833. fput(p->wr);
  834. return err;
  835. }
  836. return 0;
  837. }
  838. static struct p9_trans_module p9_tcp_trans = {
  839. .name = "tcp",
  840. .maxsize = MAX_SOCK_BUF,
  841. .def = 1,
  842. .create = p9_fd_create_tcp,
  843. .close = p9_fd_close,
  844. .request = p9_fd_request,
  845. .cancel = p9_fd_cancel,
  846. .owner = THIS_MODULE,
  847. };
  848. static struct p9_trans_module p9_unix_trans = {
  849. .name = "unix",
  850. .maxsize = MAX_SOCK_BUF,
  851. .def = 0,
  852. .create = p9_fd_create_unix,
  853. .close = p9_fd_close,
  854. .request = p9_fd_request,
  855. .cancel = p9_fd_cancel,
  856. .owner = THIS_MODULE,
  857. };
  858. static struct p9_trans_module p9_fd_trans = {
  859. .name = "fd",
  860. .maxsize = MAX_SOCK_BUF,
  861. .def = 0,
  862. .create = p9_fd_create,
  863. .close = p9_fd_close,
  864. .request = p9_fd_request,
  865. .cancel = p9_fd_cancel,
  866. .owner = THIS_MODULE,
  867. };
  868. /**
  869. * p9_poll_proc - poll worker thread
  870. * @a: thread state and arguments
  871. *
  872. * polls all v9fs transports for new events and queues the appropriate
  873. * work to the work queue
  874. *
  875. */
  876. static int p9_poll_proc(void *a)
  877. {
  878. unsigned long flags;
  879. P9_DPRINTK(P9_DEBUG_TRANS, "start %p\n", current);
  880. repeat:
  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. set_current_state(TASK_INTERRUPTIBLE);
  893. if (list_empty(&p9_poll_pending_list)) {
  894. P9_DPRINTK(P9_DEBUG_TRANS, "sleeping...\n");
  895. schedule();
  896. }
  897. __set_current_state(TASK_RUNNING);
  898. if (!kthread_should_stop())
  899. goto repeat;
  900. P9_DPRINTK(P9_DEBUG_TRANS, "finish\n");
  901. return 0;
  902. }
  903. int p9_trans_fd_init(void)
  904. {
  905. p9_mux_wq = create_workqueue("v9fs");
  906. if (!p9_mux_wq) {
  907. printk(KERN_WARNING "v9fs: mux: creating workqueue failed\n");
  908. return -ENOMEM;
  909. }
  910. p9_poll_task = kthread_run(p9_poll_proc, NULL, "v9fs-poll");
  911. if (IS_ERR(p9_poll_task)) {
  912. destroy_workqueue(p9_mux_wq);
  913. printk(KERN_WARNING "v9fs: mux: creating poll task failed\n");
  914. return PTR_ERR(p9_poll_task);
  915. }
  916. v9fs_register_trans(&p9_tcp_trans);
  917. v9fs_register_trans(&p9_unix_trans);
  918. v9fs_register_trans(&p9_fd_trans);
  919. return 0;
  920. }
  921. void p9_trans_fd_exit(void)
  922. {
  923. kthread_stop(p9_poll_task);
  924. v9fs_unregister_trans(&p9_tcp_trans);
  925. v9fs_unregister_trans(&p9_unix_trans);
  926. v9fs_unregister_trans(&p9_fd_trans);
  927. destroy_workqueue(p9_mux_wq);
  928. }