trans_fd.c 26 KB

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