trans_fd.c 29 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340
  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. #define P9_PORT 564
  44. #define MAX_SOCK_BUF (64*1024)
  45. #define MAXPOLLWADDR 2
  46. /**
  47. * struct p9_fd_opts - per-transport options
  48. * @rfd: file descriptor for reading (trans=fd)
  49. * @wfd: file descriptor for writing (trans=fd)
  50. * @port: port to connect to (trans=tcp)
  51. *
  52. */
  53. struct p9_fd_opts {
  54. int rfd;
  55. int wfd;
  56. u16 port;
  57. };
  58. /**
  59. * struct p9_trans_fd - transport state
  60. * @rd: reference to file to read from
  61. * @wr: reference of file to write to
  62. * @conn: connection state reference
  63. *
  64. */
  65. struct p9_trans_fd {
  66. struct file *rd;
  67. struct file *wr;
  68. struct p9_conn *conn;
  69. };
  70. /*
  71. * Option Parsing (code inspired by NFS code)
  72. * - a little lazy - parse all fd-transport options
  73. */
  74. enum {
  75. /* Options that take integer arguments */
  76. Opt_port, Opt_rfdno, Opt_wfdno, Opt_err,
  77. };
  78. static const match_table_t tokens = {
  79. {Opt_port, "port=%u"},
  80. {Opt_rfdno, "rfdno=%u"},
  81. {Opt_wfdno, "wfdno=%u"},
  82. {Opt_err, NULL},
  83. };
  84. enum {
  85. Rworksched = 1, /* read work scheduled or running */
  86. Rpending = 2, /* can read */
  87. Wworksched = 4, /* write work scheduled or running */
  88. Wpending = 8, /* can write */
  89. };
  90. struct p9_poll_wait {
  91. struct p9_conn *conn;
  92. wait_queue_t wait;
  93. wait_queue_head_t *wait_addr;
  94. };
  95. /**
  96. * struct p9_conn - fd mux connection state information
  97. * @mux_list: list link for mux to manage multiple connections (?)
  98. * @client: reference to client instance for this connection
  99. * @err: error state
  100. * @req_list: accounting for requests which have been sent
  101. * @unsent_req_list: accounting for requests that haven't been sent
  102. * @rcall: current response &p9_fcall structure
  103. * @rpos: read position in current frame
  104. * @rbuf: current read buffer
  105. * @wpos: write position for current frame
  106. * @wsize: amount of data to write for current frame
  107. * @wbuf: current write buffer
  108. * @poll_wait: array of wait_q's for various worker threads
  109. * @poll_waddr: ????
  110. * @pt: poll state
  111. * @rq: current read work
  112. * @wq: current write work
  113. * @wsched: ????
  114. *
  115. */
  116. struct p9_conn {
  117. struct list_head mux_list;
  118. struct p9_client *client;
  119. int err;
  120. struct list_head req_list;
  121. struct list_head unsent_req_list;
  122. struct p9_fcall *rcall;
  123. int rpos;
  124. char *rbuf;
  125. int wpos;
  126. int wsize;
  127. char *wbuf;
  128. struct list_head poll_pending_link;
  129. struct p9_poll_wait poll_wait[MAXPOLLWADDR];
  130. poll_table pt;
  131. struct work_struct rq;
  132. struct work_struct wq;
  133. unsigned long wsched;
  134. };
  135. static DEFINE_SPINLOCK(p9_poll_lock);
  136. static LIST_HEAD(p9_poll_pending_list);
  137. static struct workqueue_struct *p9_mux_wq;
  138. static struct task_struct *p9_poll_task;
  139. static void p9_mux_poll_stop(struct p9_conn *m)
  140. {
  141. unsigned long flags;
  142. int i;
  143. for (i = 0; i < ARRAY_SIZE(m->poll_wait); i++) {
  144. struct p9_poll_wait *pwait = &m->poll_wait[i];
  145. if (pwait->wait_addr) {
  146. remove_wait_queue(pwait->wait_addr, &pwait->wait);
  147. pwait->wait_addr = NULL;
  148. }
  149. }
  150. spin_lock_irqsave(&p9_poll_lock, flags);
  151. list_del_init(&m->poll_pending_link);
  152. spin_unlock_irqrestore(&p9_poll_lock, flags);
  153. }
  154. static void p9_conn_rpc_cb(struct p9_client *, struct p9_req_t *);
  155. static void p9_mux_flush_cb(struct p9_client *client, struct p9_req_t *freq)
  156. {
  157. struct p9_conn *m = client->trans;
  158. struct p9_req_t *req;
  159. P9_DPRINTK(P9_DEBUG_MUX, "mux %p tc %p rc %p err %d oldtag %d\n", m,
  160. freq->tc, freq->rc, freq->t_err,
  161. freq->tc->params.tflush.oldtag);
  162. req = p9_tag_lookup(client, freq->tc->params.tflush.oldtag);
  163. if (req) {
  164. req->status = REQ_STATUS_FLSHD;
  165. list_del(&req->req_list);
  166. p9_conn_rpc_cb(client, req);
  167. }
  168. p9_free_req(client, freq);
  169. }
  170. static void p9_conn_rpc_cb(struct p9_client *client, struct p9_req_t *req)
  171. {
  172. P9_DPRINTK(P9_DEBUG_MUX, "req %p\n", req);
  173. if (req->tc->id == P9_TFLUSH) { /* flush callback */
  174. P9_DPRINTK(P9_DEBUG_MUX, "flush req %p\n", req);
  175. p9_mux_flush_cb(client, req);
  176. } else { /* normal wakeup path */
  177. P9_DPRINTK(P9_DEBUG_MUX, "normal req %p\n", req);
  178. if (!req->t_err && (req->status == REQ_STATUS_FLSHD ||
  179. req->status == REQ_STATUS_FLSH))
  180. req->t_err = -ERESTARTSYS;
  181. wake_up(req->wq);
  182. }
  183. }
  184. /**
  185. * p9_conn_cancel - cancel all pending requests with error
  186. * @m: mux data
  187. * @err: error code
  188. *
  189. */
  190. void p9_conn_cancel(struct p9_conn *m, int err)
  191. {
  192. struct p9_req_t *req, *rtmp;
  193. LIST_HEAD(cancel_list);
  194. P9_DPRINTK(P9_DEBUG_ERROR, "mux %p err %d\n", m, err);
  195. m->err = err;
  196. spin_lock(&m->client->lock);
  197. list_for_each_entry_safe(req, rtmp, &m->req_list, req_list) {
  198. req->status = REQ_STATUS_ERROR;
  199. if (!req->t_err)
  200. req->t_err = err;
  201. list_move(&req->req_list, &cancel_list);
  202. }
  203. list_for_each_entry_safe(req, rtmp, &m->unsent_req_list, req_list) {
  204. req->status = REQ_STATUS_ERROR;
  205. if (!req->t_err)
  206. req->t_err = err;
  207. list_move(&req->req_list, &cancel_list);
  208. }
  209. spin_unlock(&m->client->lock);
  210. list_for_each_entry_safe(req, rtmp, &cancel_list, req_list) {
  211. list_del(&req->req_list);
  212. p9_conn_rpc_cb(m->client, req);
  213. }
  214. }
  215. static void process_request(struct p9_conn *m, struct p9_req_t *req)
  216. {
  217. int ecode;
  218. struct p9_str *ename;
  219. if (!req->t_err && req->rc->id == P9_RERROR) {
  220. ecode = req->rc->params.rerror.errno;
  221. ename = &req->rc->params.rerror.error;
  222. P9_DPRINTK(P9_DEBUG_MUX, "Rerror %.*s\n", ename->len,
  223. ename->str);
  224. if (m->client->dotu)
  225. req->t_err = -ecode;
  226. if (!req->t_err) {
  227. req->t_err = p9_errstr2errno(ename->str, ename->len);
  228. /* string match failed */
  229. if (!req->t_err) {
  230. PRINT_FCALL_ERROR("unknown error", req->rc);
  231. req->t_err = -ESERVERFAULT;
  232. }
  233. }
  234. } else if (req->tc && req->rc->id != req->tc->id + 1) {
  235. P9_DPRINTK(P9_DEBUG_ERROR,
  236. "fcall mismatch: expected %d, got %d\n",
  237. req->tc->id + 1, req->rc->id);
  238. if (!req->t_err)
  239. req->t_err = -EIO;
  240. }
  241. }
  242. static unsigned int
  243. p9_fd_poll(struct p9_client *client, struct poll_table_struct *pt)
  244. {
  245. int ret, n;
  246. struct p9_trans_fd *ts = NULL;
  247. if (client && client->status == Connected)
  248. ts = client->trans;
  249. if (!ts)
  250. return -EREMOTEIO;
  251. if (!ts->rd->f_op || !ts->rd->f_op->poll)
  252. return -EIO;
  253. if (!ts->wr->f_op || !ts->wr->f_op->poll)
  254. return -EIO;
  255. ret = ts->rd->f_op->poll(ts->rd, pt);
  256. if (ret < 0)
  257. return ret;
  258. if (ts->rd != ts->wr) {
  259. n = ts->wr->f_op->poll(ts->wr, pt);
  260. if (n < 0)
  261. return n;
  262. ret = (ret & ~POLLOUT) | (n & ~POLLIN);
  263. }
  264. return ret;
  265. }
  266. /**
  267. * p9_fd_read- read from a fd
  268. * @client: client instance
  269. * @v: buffer to receive data into
  270. * @len: size of receive buffer
  271. *
  272. */
  273. static int p9_fd_read(struct p9_client *client, void *v, int len)
  274. {
  275. int ret;
  276. struct p9_trans_fd *ts = NULL;
  277. if (client && client->status != Disconnected)
  278. ts = client->trans;
  279. if (!ts)
  280. return -EREMOTEIO;
  281. if (!(ts->rd->f_flags & O_NONBLOCK))
  282. P9_DPRINTK(P9_DEBUG_ERROR, "blocking read ...\n");
  283. ret = kernel_read(ts->rd, ts->rd->f_pos, v, len);
  284. if (ret <= 0 && ret != -ERESTARTSYS && ret != -EAGAIN)
  285. client->status = Disconnected;
  286. return ret;
  287. }
  288. /**
  289. * p9_read_work - called when there is some data to be read from a transport
  290. * @work: container of work to be done
  291. *
  292. */
  293. static void p9_read_work(struct work_struct *work)
  294. {
  295. int n, err;
  296. struct p9_conn *m;
  297. struct p9_req_t *req;
  298. struct p9_fcall *rcall;
  299. char *rbuf;
  300. m = container_of(work, struct p9_conn, rq);
  301. if (m->err < 0)
  302. return;
  303. rcall = NULL;
  304. P9_DPRINTK(P9_DEBUG_MUX, "start mux %p pos %d\n", m, m->rpos);
  305. if (!m->rcall) {
  306. m->rcall =
  307. kmalloc(sizeof(struct p9_fcall) + m->client->msize,
  308. GFP_KERNEL);
  309. if (!m->rcall) {
  310. err = -ENOMEM;
  311. goto error;
  312. }
  313. m->rbuf = (char *)m->rcall + sizeof(struct p9_fcall);
  314. m->rpos = 0;
  315. }
  316. clear_bit(Rpending, &m->wsched);
  317. err = p9_fd_read(m->client, m->rbuf + m->rpos,
  318. m->client->msize - m->rpos);
  319. P9_DPRINTK(P9_DEBUG_MUX, "mux %p got %d bytes\n", m, err);
  320. if (err == -EAGAIN) {
  321. clear_bit(Rworksched, &m->wsched);
  322. return;
  323. }
  324. if (err <= 0)
  325. goto error;
  326. m->rpos += err;
  327. while (m->rpos > 4) {
  328. n = le32_to_cpu(*(__le32 *) m->rbuf);
  329. if (n >= m->client->msize) {
  330. P9_DPRINTK(P9_DEBUG_ERROR,
  331. "requested packet size too big: %d\n", n);
  332. err = -EIO;
  333. goto error;
  334. }
  335. if (m->rpos < n)
  336. break;
  337. err =
  338. p9_deserialize_fcall(m->rbuf, n, m->rcall, m->client->dotu);
  339. if (err < 0)
  340. goto error;
  341. #ifdef CONFIG_NET_9P_DEBUG
  342. if ((p9_debug_level&P9_DEBUG_FCALL) == P9_DEBUG_FCALL) {
  343. char buf[150];
  344. p9_printfcall(buf, sizeof(buf), m->rcall,
  345. m->client->dotu);
  346. printk(KERN_NOTICE ">>> %p %s\n", m, buf);
  347. }
  348. #endif
  349. rcall = m->rcall;
  350. rbuf = m->rbuf;
  351. if (m->rpos > n) {
  352. m->rcall = kmalloc(sizeof(struct p9_fcall) +
  353. m->client->msize, GFP_KERNEL);
  354. if (!m->rcall) {
  355. err = -ENOMEM;
  356. goto error;
  357. }
  358. m->rbuf = (char *)m->rcall + sizeof(struct p9_fcall);
  359. memmove(m->rbuf, rbuf + n, m->rpos - n);
  360. m->rpos -= n;
  361. } else {
  362. m->rcall = NULL;
  363. m->rbuf = NULL;
  364. m->rpos = 0;
  365. }
  366. P9_DPRINTK(P9_DEBUG_MUX, "mux %p fcall id %d tag %d\n", m,
  367. rcall->id, rcall->tag);
  368. req = p9_tag_lookup(m->client, rcall->tag);
  369. if (req) {
  370. if (req->status != REQ_STATUS_FLSH) {
  371. list_del(&req->req_list);
  372. req->status = REQ_STATUS_RCVD;
  373. }
  374. req->rc = rcall;
  375. process_request(m, req);
  376. if (req->status != REQ_STATUS_FLSH)
  377. p9_conn_rpc_cb(m->client, req);
  378. } else {
  379. if (err >= 0 && rcall->id != P9_RFLUSH)
  380. P9_DPRINTK(P9_DEBUG_ERROR,
  381. "unexpected response mux %p id %d tag %d\n",
  382. m, rcall->id, rcall->tag);
  383. kfree(rcall);
  384. }
  385. }
  386. if (!list_empty(&m->req_list)) {
  387. if (test_and_clear_bit(Rpending, &m->wsched))
  388. n = POLLIN;
  389. else
  390. n = p9_fd_poll(m->client, NULL);
  391. if (n & POLLIN) {
  392. P9_DPRINTK(P9_DEBUG_MUX, "schedule read work %p\n", m);
  393. queue_work(p9_mux_wq, &m->rq);
  394. } else
  395. clear_bit(Rworksched, &m->wsched);
  396. } else
  397. clear_bit(Rworksched, &m->wsched);
  398. return;
  399. error:
  400. p9_conn_cancel(m, err);
  401. clear_bit(Rworksched, &m->wsched);
  402. }
  403. /**
  404. * p9_fd_write - write to a socket
  405. * @client: client instance
  406. * @v: buffer to send data from
  407. * @len: size of send buffer
  408. *
  409. */
  410. static int p9_fd_write(struct p9_client *client, void *v, int len)
  411. {
  412. int ret;
  413. mm_segment_t oldfs;
  414. struct p9_trans_fd *ts = NULL;
  415. if (client && client->status != Disconnected)
  416. ts = client->trans;
  417. if (!ts)
  418. return -EREMOTEIO;
  419. if (!(ts->wr->f_flags & O_NONBLOCK))
  420. P9_DPRINTK(P9_DEBUG_ERROR, "blocking write ...\n");
  421. oldfs = get_fs();
  422. set_fs(get_ds());
  423. /* The cast to a user pointer is valid due to the set_fs() */
  424. ret = vfs_write(ts->wr, (void __user *)v, len, &ts->wr->f_pos);
  425. set_fs(oldfs);
  426. if (ret <= 0 && ret != -ERESTARTSYS && ret != -EAGAIN)
  427. client->status = Disconnected;
  428. return ret;
  429. }
  430. /**
  431. * p9_write_work - called when a transport can send some data
  432. * @work: container for work to be done
  433. *
  434. */
  435. static void p9_write_work(struct work_struct *work)
  436. {
  437. int n, err;
  438. struct p9_conn *m;
  439. struct p9_req_t *req;
  440. m = container_of(work, struct p9_conn, wq);
  441. if (m->err < 0) {
  442. clear_bit(Wworksched, &m->wsched);
  443. return;
  444. }
  445. if (!m->wsize) {
  446. if (list_empty(&m->unsent_req_list)) {
  447. clear_bit(Wworksched, &m->wsched);
  448. return;
  449. }
  450. spin_lock(&m->client->lock);
  451. req = list_entry(m->unsent_req_list.next, struct p9_req_t,
  452. req_list);
  453. req->status = REQ_STATUS_SENT;
  454. list_move_tail(&req->req_list, &m->req_list);
  455. m->wbuf = req->tc->sdata;
  456. m->wsize = req->tc->size;
  457. m->wpos = 0;
  458. spin_unlock(&m->client->lock);
  459. }
  460. P9_DPRINTK(P9_DEBUG_MUX, "mux %p pos %d size %d\n", m, m->wpos,
  461. m->wsize);
  462. clear_bit(Wpending, &m->wsched);
  463. err = p9_fd_write(m->client, m->wbuf + m->wpos, m->wsize - m->wpos);
  464. P9_DPRINTK(P9_DEBUG_MUX, "mux %p sent %d bytes\n", m, err);
  465. if (err == -EAGAIN) {
  466. clear_bit(Wworksched, &m->wsched);
  467. return;
  468. }
  469. if (err < 0)
  470. goto error;
  471. else if (err == 0) {
  472. err = -EREMOTEIO;
  473. goto error;
  474. }
  475. m->wpos += err;
  476. if (m->wpos == m->wsize)
  477. m->wpos = m->wsize = 0;
  478. if (m->wsize == 0 && !list_empty(&m->unsent_req_list)) {
  479. if (test_and_clear_bit(Wpending, &m->wsched))
  480. n = POLLOUT;
  481. else
  482. n = p9_fd_poll(m->client, NULL);
  483. if (n & POLLOUT) {
  484. P9_DPRINTK(P9_DEBUG_MUX, "schedule write work %p\n", m);
  485. queue_work(p9_mux_wq, &m->wq);
  486. } else
  487. clear_bit(Wworksched, &m->wsched);
  488. } else
  489. clear_bit(Wworksched, &m->wsched);
  490. return;
  491. error:
  492. p9_conn_cancel(m, err);
  493. clear_bit(Wworksched, &m->wsched);
  494. }
  495. static int p9_pollwake(wait_queue_t *wait, unsigned mode, int sync, void *key)
  496. {
  497. struct p9_poll_wait *pwait =
  498. container_of(wait, struct p9_poll_wait, wait);
  499. struct p9_conn *m = pwait->conn;
  500. unsigned long flags;
  501. DECLARE_WAITQUEUE(dummy_wait, p9_poll_task);
  502. spin_lock_irqsave(&p9_poll_lock, flags);
  503. if (list_empty(&m->poll_pending_link))
  504. list_add_tail(&m->poll_pending_link, &p9_poll_pending_list);
  505. spin_unlock_irqrestore(&p9_poll_lock, flags);
  506. /* perform the default wake up operation */
  507. return default_wake_function(&dummy_wait, mode, sync, key);
  508. }
  509. /**
  510. * p9_pollwait - add poll task to the wait queue
  511. * @filp: file pointer being polled
  512. * @wait_address: wait_q to block on
  513. * @p: poll state
  514. *
  515. * called by files poll operation to add v9fs-poll task to files wait queue
  516. */
  517. static void
  518. p9_pollwait(struct file *filp, wait_queue_head_t *wait_address, poll_table *p)
  519. {
  520. struct p9_conn *m = container_of(p, struct p9_conn, pt);
  521. struct p9_poll_wait *pwait = NULL;
  522. int i;
  523. for (i = 0; i < ARRAY_SIZE(m->poll_wait); i++) {
  524. if (m->poll_wait[i].wait_addr == NULL) {
  525. pwait = &m->poll_wait[i];
  526. break;
  527. }
  528. }
  529. if (!pwait) {
  530. P9_DPRINTK(P9_DEBUG_ERROR, "not enough wait_address slots\n");
  531. return;
  532. }
  533. if (!wait_address) {
  534. P9_DPRINTK(P9_DEBUG_ERROR, "no wait_address\n");
  535. pwait->wait_addr = ERR_PTR(-EIO);
  536. return;
  537. }
  538. pwait->conn = m;
  539. pwait->wait_addr = wait_address;
  540. init_waitqueue_func_entry(&pwait->wait, p9_pollwake);
  541. add_wait_queue(wait_address, &pwait->wait);
  542. }
  543. /**
  544. * p9_conn_create - allocate and initialize the per-session mux data
  545. * @client: client instance
  546. *
  547. * Note: Creates the polling task if this is the first session.
  548. */
  549. static struct p9_conn *p9_conn_create(struct p9_client *client)
  550. {
  551. int i, n;
  552. struct p9_conn *m;
  553. P9_DPRINTK(P9_DEBUG_MUX, "client %p msize %d\n", client, client->msize);
  554. m = kzalloc(sizeof(struct p9_conn), GFP_KERNEL);
  555. if (!m)
  556. return ERR_PTR(-ENOMEM);
  557. INIT_LIST_HEAD(&m->mux_list);
  558. m->client = client;
  559. INIT_LIST_HEAD(&m->req_list);
  560. INIT_LIST_HEAD(&m->unsent_req_list);
  561. INIT_WORK(&m->rq, p9_read_work);
  562. INIT_WORK(&m->wq, p9_write_work);
  563. INIT_LIST_HEAD(&m->poll_pending_link);
  564. init_poll_funcptr(&m->pt, p9_pollwait);
  565. n = p9_fd_poll(client, &m->pt);
  566. if (n & POLLIN) {
  567. P9_DPRINTK(P9_DEBUG_MUX, "mux %p can read\n", m);
  568. set_bit(Rpending, &m->wsched);
  569. }
  570. if (n & POLLOUT) {
  571. P9_DPRINTK(P9_DEBUG_MUX, "mux %p can write\n", m);
  572. set_bit(Wpending, &m->wsched);
  573. }
  574. for (i = 0; i < ARRAY_SIZE(m->poll_wait); i++) {
  575. if (IS_ERR(m->poll_wait[i].wait_addr)) {
  576. p9_mux_poll_stop(m);
  577. kfree(m);
  578. /* return the error code */
  579. return (void *)m->poll_wait[i].wait_addr;
  580. }
  581. }
  582. return m;
  583. }
  584. /**
  585. * p9_poll_mux - polls a mux and schedules read or write works if necessary
  586. * @m: connection to poll
  587. *
  588. */
  589. static void p9_poll_mux(struct p9_conn *m)
  590. {
  591. int n;
  592. if (m->err < 0)
  593. return;
  594. n = p9_fd_poll(m->client, NULL);
  595. if (n < 0 || n & (POLLERR | POLLHUP | POLLNVAL)) {
  596. P9_DPRINTK(P9_DEBUG_MUX, "error mux %p err %d\n", m, n);
  597. if (n >= 0)
  598. n = -ECONNRESET;
  599. p9_conn_cancel(m, n);
  600. }
  601. if (n & POLLIN) {
  602. set_bit(Rpending, &m->wsched);
  603. P9_DPRINTK(P9_DEBUG_MUX, "mux %p can read\n", m);
  604. if (!test_and_set_bit(Rworksched, &m->wsched)) {
  605. P9_DPRINTK(P9_DEBUG_MUX, "schedule read work %p\n", m);
  606. queue_work(p9_mux_wq, &m->rq);
  607. }
  608. }
  609. if (n & POLLOUT) {
  610. set_bit(Wpending, &m->wsched);
  611. P9_DPRINTK(P9_DEBUG_MUX, "mux %p can write\n", m);
  612. if ((m->wsize || !list_empty(&m->unsent_req_list))
  613. && !test_and_set_bit(Wworksched, &m->wsched)) {
  614. P9_DPRINTK(P9_DEBUG_MUX, "schedule write work %p\n", m);
  615. queue_work(p9_mux_wq, &m->wq);
  616. }
  617. }
  618. }
  619. /**
  620. * p9_send_request - send 9P request
  621. * The function can sleep until the request is scheduled for sending.
  622. * The function can be interrupted. Return from the function is not
  623. * a guarantee that the request is sent successfully. Can return errors
  624. * that can be retrieved by PTR_ERR macros.
  625. *
  626. * @m: mux data
  627. * @tc: request to be sent
  628. *
  629. */
  630. static struct p9_req_t *p9_send_request(struct p9_conn *m, struct p9_fcall *tc)
  631. {
  632. int tag;
  633. int n;
  634. struct p9_req_t *req;
  635. P9_DPRINTK(P9_DEBUG_MUX, "mux %p task %p tcall %p id %d\n", m, current,
  636. tc, tc->id);
  637. if (m->err < 0)
  638. return ERR_PTR(m->err);
  639. tag = P9_NOTAG;
  640. if (tc->id != P9_TVERSION) {
  641. tag = p9_idpool_get(m->client->tagpool);
  642. if (tag < 0)
  643. return ERR_PTR(-ENOMEM);
  644. }
  645. p9_set_tag(tc, tag);
  646. req = p9_tag_alloc(m->client, tag);
  647. #ifdef CONFIG_NET_9P_DEBUG
  648. if ((p9_debug_level&P9_DEBUG_FCALL) == P9_DEBUG_FCALL) {
  649. char buf[150];
  650. p9_printfcall(buf, sizeof(buf), tc, m->client->dotu);
  651. printk(KERN_NOTICE "<<< %p %s\n", m, buf);
  652. }
  653. #endif
  654. req->tag = tag;
  655. req->tc = tc;
  656. req->rc = NULL;
  657. req->t_err = 0;
  658. req->status = REQ_STATUS_UNSENT;
  659. spin_lock(&m->client->lock);
  660. list_add_tail(&req->req_list, &m->unsent_req_list);
  661. spin_unlock(&m->client->lock);
  662. if (test_and_clear_bit(Wpending, &m->wsched))
  663. n = POLLOUT;
  664. else
  665. n = p9_fd_poll(m->client, NULL);
  666. if (n & POLLOUT && !test_and_set_bit(Wworksched, &m->wsched))
  667. queue_work(p9_mux_wq, &m->wq);
  668. return req;
  669. }
  670. static int
  671. p9_mux_flush_request(struct p9_conn *m, struct p9_req_t *req)
  672. {
  673. struct p9_fcall *fc;
  674. struct p9_req_t *rreq, *rptr;
  675. P9_DPRINTK(P9_DEBUG_MUX, "mux %p req %p tag %d\n", m, req, req->tag);
  676. /* if a response was received for a request, do nothing */
  677. if (req->rc || req->t_err) {
  678. P9_DPRINTK(P9_DEBUG_MUX,
  679. "mux %p req %p response already received\n", m, req);
  680. return 0;
  681. }
  682. req->status = REQ_STATUS_FLSH;
  683. spin_lock(&m->client->lock);
  684. /* if the request is not sent yet, just remove it from the list */
  685. list_for_each_entry_safe(rreq, rptr, &m->unsent_req_list, req_list) {
  686. if (rreq->tag == req->tag) {
  687. P9_DPRINTK(P9_DEBUG_MUX,
  688. "mux %p req %p request is not sent yet\n", m, req);
  689. list_del(&rreq->req_list);
  690. req->status = REQ_STATUS_FLSHD;
  691. spin_unlock(&m->client->lock);
  692. p9_conn_rpc_cb(m->client, req);
  693. return 0;
  694. }
  695. }
  696. spin_unlock(&m->client->lock);
  697. clear_thread_flag(TIF_SIGPENDING);
  698. fc = p9_create_tflush(req->tag);
  699. p9_send_request(m, fc);
  700. return 1;
  701. }
  702. /**
  703. * p9_fd_rpc- sends 9P request and waits until a response is available.
  704. * The function can be interrupted.
  705. * @client: client instance
  706. * @tc: request to be sent
  707. * @rc: pointer where a pointer to the response is stored
  708. *
  709. */
  710. int
  711. p9_fd_rpc(struct p9_client *client, struct p9_fcall *tc, struct p9_fcall **rc)
  712. {
  713. struct p9_trans_fd *p = client->trans;
  714. struct p9_conn *m = p->conn;
  715. int err, sigpending;
  716. unsigned long flags;
  717. struct p9_req_t *req;
  718. if (rc)
  719. *rc = NULL;
  720. sigpending = 0;
  721. if (signal_pending(current)) {
  722. sigpending = 1;
  723. clear_thread_flag(TIF_SIGPENDING);
  724. }
  725. req = p9_send_request(m, tc);
  726. if (IS_ERR(req)) {
  727. err = PTR_ERR(req);
  728. P9_DPRINTK(P9_DEBUG_MUX, "error %d\n", err);
  729. return err;
  730. }
  731. err = wait_event_interruptible(*req->wq, req->rc != NULL ||
  732. req->t_err < 0);
  733. if (req->t_err < 0)
  734. err = req->t_err;
  735. if (err == -ERESTARTSYS && client->status == Connected
  736. && m->err == 0) {
  737. if (p9_mux_flush_request(m, req)) {
  738. /* wait until we get response of the flush message */
  739. do {
  740. clear_thread_flag(TIF_SIGPENDING);
  741. err = wait_event_interruptible(*req->wq,
  742. req->rc || req->t_err);
  743. } while (!req->rc && !req->t_err &&
  744. err == -ERESTARTSYS &&
  745. client->status == Connected && !m->err);
  746. err = -ERESTARTSYS;
  747. }
  748. sigpending = 1;
  749. }
  750. if (sigpending) {
  751. spin_lock_irqsave(&current->sighand->siglock, flags);
  752. recalc_sigpending();
  753. spin_unlock_irqrestore(&current->sighand->siglock, flags);
  754. }
  755. if (rc)
  756. *rc = req->rc;
  757. else
  758. kfree(req->rc);
  759. p9_free_req(client, req);
  760. if (err > 0)
  761. err = -EIO;
  762. return err;
  763. }
  764. /**
  765. * parse_options - parse mount options into session structure
  766. * @options: options string passed from mount
  767. * @opts: transport-specific structure to parse options into
  768. *
  769. * Returns 0 upon success, -ERRNO upon failure
  770. */
  771. static int parse_opts(char *params, struct p9_fd_opts *opts)
  772. {
  773. char *p;
  774. substring_t args[MAX_OPT_ARGS];
  775. int option;
  776. char *options;
  777. int ret;
  778. opts->port = P9_PORT;
  779. opts->rfd = ~0;
  780. opts->wfd = ~0;
  781. if (!params)
  782. return 0;
  783. options = kstrdup(params, GFP_KERNEL);
  784. if (!options) {
  785. P9_DPRINTK(P9_DEBUG_ERROR,
  786. "failed to allocate copy of option string\n");
  787. return -ENOMEM;
  788. }
  789. while ((p = strsep(&options, ",")) != NULL) {
  790. int token;
  791. int r;
  792. if (!*p)
  793. continue;
  794. token = match_token(p, tokens, args);
  795. r = match_int(&args[0], &option);
  796. if (r < 0) {
  797. P9_DPRINTK(P9_DEBUG_ERROR,
  798. "integer field, but no integer?\n");
  799. ret = r;
  800. continue;
  801. }
  802. switch (token) {
  803. case Opt_port:
  804. opts->port = option;
  805. break;
  806. case Opt_rfdno:
  807. opts->rfd = option;
  808. break;
  809. case Opt_wfdno:
  810. opts->wfd = option;
  811. break;
  812. default:
  813. continue;
  814. }
  815. }
  816. kfree(options);
  817. return 0;
  818. }
  819. static int p9_fd_open(struct p9_client *client, int rfd, int wfd)
  820. {
  821. struct p9_trans_fd *ts = kmalloc(sizeof(struct p9_trans_fd),
  822. GFP_KERNEL);
  823. if (!ts)
  824. return -ENOMEM;
  825. ts->rd = fget(rfd);
  826. ts->wr = fget(wfd);
  827. if (!ts->rd || !ts->wr) {
  828. if (ts->rd)
  829. fput(ts->rd);
  830. if (ts->wr)
  831. fput(ts->wr);
  832. kfree(ts);
  833. return -EIO;
  834. }
  835. client->trans = ts;
  836. client->status = Connected;
  837. return 0;
  838. }
  839. static int p9_socket_open(struct p9_client *client, struct socket *csocket)
  840. {
  841. int fd, ret;
  842. csocket->sk->sk_allocation = GFP_NOIO;
  843. fd = sock_map_fd(csocket, 0);
  844. if (fd < 0) {
  845. P9_EPRINTK(KERN_ERR, "p9_socket_open: failed to map fd\n");
  846. return fd;
  847. }
  848. ret = p9_fd_open(client, fd, fd);
  849. if (ret < 0) {
  850. P9_EPRINTK(KERN_ERR, "p9_socket_open: failed to open fd\n");
  851. sockfd_put(csocket);
  852. return ret;
  853. }
  854. ((struct p9_trans_fd *)client->trans)->rd->f_flags |= O_NONBLOCK;
  855. return 0;
  856. }
  857. /**
  858. * p9_mux_destroy - cancels all pending requests and frees mux resources
  859. * @m: mux to destroy
  860. *
  861. */
  862. static void p9_conn_destroy(struct p9_conn *m)
  863. {
  864. P9_DPRINTK(P9_DEBUG_MUX, "mux %p prev %p next %p\n", m,
  865. m->mux_list.prev, m->mux_list.next);
  866. p9_mux_poll_stop(m);
  867. cancel_work_sync(&m->rq);
  868. cancel_work_sync(&m->wq);
  869. p9_conn_cancel(m, -ECONNRESET);
  870. m->client = NULL;
  871. kfree(m);
  872. }
  873. /**
  874. * p9_fd_close - shutdown file descriptor transport
  875. * @client: client instance
  876. *
  877. */
  878. static void p9_fd_close(struct p9_client *client)
  879. {
  880. struct p9_trans_fd *ts;
  881. if (!client)
  882. return;
  883. ts = client->trans;
  884. if (!ts)
  885. return;
  886. client->status = Disconnected;
  887. p9_conn_destroy(ts->conn);
  888. if (ts->rd)
  889. fput(ts->rd);
  890. if (ts->wr)
  891. fput(ts->wr);
  892. kfree(ts);
  893. }
  894. /*
  895. * stolen from NFS - maybe should be made a generic function?
  896. */
  897. static inline int valid_ipaddr4(const char *buf)
  898. {
  899. int rc, count, in[4];
  900. rc = sscanf(buf, "%d.%d.%d.%d", &in[0], &in[1], &in[2], &in[3]);
  901. if (rc != 4)
  902. return -EINVAL;
  903. for (count = 0; count < 4; count++) {
  904. if (in[count] > 255)
  905. return -EINVAL;
  906. }
  907. return 0;
  908. }
  909. static int
  910. p9_fd_create_tcp(struct p9_client *client, const char *addr, char *args)
  911. {
  912. int err;
  913. struct socket *csocket;
  914. struct sockaddr_in sin_server;
  915. struct p9_fd_opts opts;
  916. struct p9_trans_fd *p = NULL; /* this gets allocated in p9_fd_open */
  917. err = parse_opts(args, &opts);
  918. if (err < 0)
  919. return err;
  920. if (valid_ipaddr4(addr) < 0)
  921. return -EINVAL;
  922. csocket = NULL;
  923. sin_server.sin_family = AF_INET;
  924. sin_server.sin_addr.s_addr = in_aton(addr);
  925. sin_server.sin_port = htons(opts.port);
  926. sock_create_kern(PF_INET, SOCK_STREAM, IPPROTO_TCP, &csocket);
  927. if (!csocket) {
  928. P9_EPRINTK(KERN_ERR, "p9_trans_tcp: problem creating socket\n");
  929. err = -EIO;
  930. goto error;
  931. }
  932. err = csocket->ops->connect(csocket,
  933. (struct sockaddr *)&sin_server,
  934. sizeof(struct sockaddr_in), 0);
  935. if (err < 0) {
  936. P9_EPRINTK(KERN_ERR,
  937. "p9_trans_tcp: problem connecting socket to %s\n",
  938. addr);
  939. goto error;
  940. }
  941. err = p9_socket_open(client, csocket);
  942. if (err < 0)
  943. goto error;
  944. p = (struct p9_trans_fd *) client->trans;
  945. p->conn = p9_conn_create(client);
  946. if (IS_ERR(p->conn)) {
  947. err = PTR_ERR(p->conn);
  948. p->conn = NULL;
  949. goto error;
  950. }
  951. return 0;
  952. error:
  953. if (csocket)
  954. sock_release(csocket);
  955. kfree(p);
  956. return err;
  957. }
  958. static int
  959. p9_fd_create_unix(struct p9_client *client, const char *addr, char *args)
  960. {
  961. int err;
  962. struct socket *csocket;
  963. struct sockaddr_un sun_server;
  964. struct p9_trans_fd *p = NULL; /* this gets allocated in p9_fd_open */
  965. csocket = NULL;
  966. if (strlen(addr) > UNIX_PATH_MAX) {
  967. P9_EPRINTK(KERN_ERR, "p9_trans_unix: address too long: %s\n",
  968. addr);
  969. err = -ENAMETOOLONG;
  970. goto error;
  971. }
  972. sun_server.sun_family = PF_UNIX;
  973. strcpy(sun_server.sun_path, addr);
  974. sock_create_kern(PF_UNIX, SOCK_STREAM, 0, &csocket);
  975. err = csocket->ops->connect(csocket, (struct sockaddr *)&sun_server,
  976. sizeof(struct sockaddr_un) - 1, 0);
  977. if (err < 0) {
  978. P9_EPRINTK(KERN_ERR,
  979. "p9_trans_unix: problem connecting socket: %s: %d\n",
  980. addr, err);
  981. goto error;
  982. }
  983. err = p9_socket_open(client, csocket);
  984. if (err < 0)
  985. goto error;
  986. p = (struct p9_trans_fd *) client->trans;
  987. p->conn = p9_conn_create(client);
  988. if (IS_ERR(p->conn)) {
  989. err = PTR_ERR(p->conn);
  990. p->conn = NULL;
  991. goto error;
  992. }
  993. return 0;
  994. error:
  995. if (csocket)
  996. sock_release(csocket);
  997. kfree(p);
  998. return err;
  999. }
  1000. static int
  1001. p9_fd_create(struct p9_client *client, const char *addr, char *args)
  1002. {
  1003. int err;
  1004. struct p9_fd_opts opts;
  1005. struct p9_trans_fd *p = NULL; /* this get allocated in p9_fd_open */
  1006. parse_opts(args, &opts);
  1007. if (opts.rfd == ~0 || opts.wfd == ~0) {
  1008. printk(KERN_ERR "v9fs: Insufficient options for proto=fd\n");
  1009. return -ENOPROTOOPT;
  1010. }
  1011. err = p9_fd_open(client, opts.rfd, opts.wfd);
  1012. if (err < 0)
  1013. goto error;
  1014. p = (struct p9_trans_fd *) client->trans;
  1015. p->conn = p9_conn_create(client);
  1016. if (IS_ERR(p->conn)) {
  1017. err = PTR_ERR(p->conn);
  1018. p->conn = NULL;
  1019. goto error;
  1020. }
  1021. return 0;
  1022. error:
  1023. kfree(p);
  1024. return err;
  1025. }
  1026. static struct p9_trans_module p9_tcp_trans = {
  1027. .name = "tcp",
  1028. .maxsize = MAX_SOCK_BUF,
  1029. .def = 1,
  1030. .create = p9_fd_create_tcp,
  1031. .close = p9_fd_close,
  1032. .rpc = p9_fd_rpc,
  1033. .owner = THIS_MODULE,
  1034. };
  1035. static struct p9_trans_module p9_unix_trans = {
  1036. .name = "unix",
  1037. .maxsize = MAX_SOCK_BUF,
  1038. .def = 0,
  1039. .create = p9_fd_create_unix,
  1040. .close = p9_fd_close,
  1041. .rpc = p9_fd_rpc,
  1042. .owner = THIS_MODULE,
  1043. };
  1044. static struct p9_trans_module p9_fd_trans = {
  1045. .name = "fd",
  1046. .maxsize = MAX_SOCK_BUF,
  1047. .def = 0,
  1048. .create = p9_fd_create,
  1049. .close = p9_fd_close,
  1050. .rpc = p9_fd_rpc,
  1051. .owner = THIS_MODULE,
  1052. };
  1053. /**
  1054. * p9_poll_proc - poll worker thread
  1055. * @a: thread state and arguments
  1056. *
  1057. * polls all v9fs transports for new events and queues the appropriate
  1058. * work to the work queue
  1059. *
  1060. */
  1061. static int p9_poll_proc(void *a)
  1062. {
  1063. unsigned long flags;
  1064. P9_DPRINTK(P9_DEBUG_MUX, "start %p\n", current);
  1065. repeat:
  1066. spin_lock_irqsave(&p9_poll_lock, flags);
  1067. while (!list_empty(&p9_poll_pending_list)) {
  1068. struct p9_conn *conn = list_first_entry(&p9_poll_pending_list,
  1069. struct p9_conn,
  1070. poll_pending_link);
  1071. list_del_init(&conn->poll_pending_link);
  1072. spin_unlock_irqrestore(&p9_poll_lock, flags);
  1073. p9_poll_mux(conn);
  1074. spin_lock_irqsave(&p9_poll_lock, flags);
  1075. }
  1076. spin_unlock_irqrestore(&p9_poll_lock, flags);
  1077. set_current_state(TASK_INTERRUPTIBLE);
  1078. if (list_empty(&p9_poll_pending_list)) {
  1079. P9_DPRINTK(P9_DEBUG_MUX, "sleeping...\n");
  1080. schedule();
  1081. }
  1082. __set_current_state(TASK_RUNNING);
  1083. if (!kthread_should_stop())
  1084. goto repeat;
  1085. P9_DPRINTK(P9_DEBUG_MUX, "finish\n");
  1086. return 0;
  1087. }
  1088. int p9_trans_fd_init(void)
  1089. {
  1090. p9_mux_wq = create_workqueue("v9fs");
  1091. if (!p9_mux_wq) {
  1092. printk(KERN_WARNING "v9fs: mux: creating workqueue failed\n");
  1093. return -ENOMEM;
  1094. }
  1095. p9_poll_task = kthread_run(p9_poll_proc, NULL, "v9fs-poll");
  1096. if (IS_ERR(p9_poll_task)) {
  1097. destroy_workqueue(p9_mux_wq);
  1098. printk(KERN_WARNING "v9fs: mux: creating poll task failed\n");
  1099. return PTR_ERR(p9_poll_task);
  1100. }
  1101. v9fs_register_trans(&p9_tcp_trans);
  1102. v9fs_register_trans(&p9_unix_trans);
  1103. v9fs_register_trans(&p9_fd_trans);
  1104. return 0;
  1105. }
  1106. void p9_trans_fd_exit(void)
  1107. {
  1108. kthread_stop(p9_poll_task);
  1109. v9fs_unregister_trans(&p9_tcp_trans);
  1110. v9fs_unregister_trans(&p9_unix_trans);
  1111. v9fs_unregister_trans(&p9_fd_trans);
  1112. destroy_workqueue(p9_mux_wq);
  1113. }