net/9p/trans_fd.c

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