1 // SPDX-License-Identifier: GPL-2.0-or-later
2 /*
3 * RDMA Network Block Driver
4 *
5 * Copyright (c) 2014 - 2018 ProfitBricks GmbH. All rights reserved.
6 * Copyright (c) 2018 - 2019 1&1 IONOS Cloud GmbH. All rights reserved.
7 * Copyright (c) 2019 - 2020 1&1 IONOS SE. All rights reserved.
8 */
9 #undef pr_fmt
10 #define pr_fmt(fmt) KBUILD_MODNAME " L" __stringify(__LINE__) ": " fmt
11
12 #include <linux/module.h>
13 #include <linux/blkdev.h>
14
15 #include "rnbd-srv.h"
16 #include "rnbd-srv-trace.h"
17
18 MODULE_DESCRIPTION("RDMA Network Block Device Server");
19 MODULE_LICENSE("GPL");
20
21 static u16 port_nr = RTRS_PORT;
22
23 module_param_named(port_nr, port_nr, ushort, 0444);
24 MODULE_PARM_DESC(port_nr,
25 "The port number the server is listening on (default: "
26 __stringify(RTRS_PORT)")");
27
28 #define DEFAULT_DEV_SEARCH_PATH "/"
29
30 static char dev_search_path[PATH_MAX] = DEFAULT_DEV_SEARCH_PATH;
31
dev_search_path_set(const char * val,const struct kernel_param * kp)32 static int dev_search_path_set(const char *val, const struct kernel_param *kp)
33 {
34 const char *p = strrchr(val, '\n') ? : val + strlen(val);
35
36 if (strlen(val) >= sizeof(dev_search_path))
37 return -EINVAL;
38
39 snprintf(dev_search_path, sizeof(dev_search_path), "%.*s",
40 (int)(p - val), val);
41
42 pr_info("dev_search_path changed to '%s'\n", dev_search_path);
43
44 return 0;
45 }
46
47 static struct kparam_string dev_search_path_kparam_str = {
48 .maxlen = sizeof(dev_search_path),
49 .string = dev_search_path
50 };
51
52 static const struct kernel_param_ops dev_search_path_ops = {
53 .set = dev_search_path_set,
54 .get = param_get_string,
55 };
56
57 module_param_cb(dev_search_path, &dev_search_path_ops,
58 &dev_search_path_kparam_str, 0444);
59 MODULE_PARM_DESC(dev_search_path,
60 "Sets the dev_search_path. When a device is mapped this path is prepended to the device path from the map device operation. If %SESSNAME% is specified in a path, then device will be searched in a session namespace. (default: "
61 DEFAULT_DEV_SEARCH_PATH ")");
62
63 static DEFINE_MUTEX(sess_lock);
64 static DEFINE_SPINLOCK(dev_lock);
65
66 static LIST_HEAD(sess_list);
67 static LIST_HEAD(dev_list);
68
69 struct rnbd_io_private {
70 struct rtrs_srv_op *id;
71 struct rnbd_srv_sess_dev *sess_dev;
72 };
73
rnbd_sess_dev_release(struct kref * kref)74 static void rnbd_sess_dev_release(struct kref *kref)
75 {
76 struct rnbd_srv_sess_dev *sess_dev;
77
78 sess_dev = container_of(kref, struct rnbd_srv_sess_dev, kref);
79 complete(sess_dev->destroy_comp);
80 }
81
rnbd_put_sess_dev(struct rnbd_srv_sess_dev * sess_dev)82 static inline void rnbd_put_sess_dev(struct rnbd_srv_sess_dev *sess_dev)
83 {
84 kref_put(&sess_dev->kref, rnbd_sess_dev_release);
85 }
86
87 static struct rnbd_srv_sess_dev *
rnbd_get_sess_dev(int dev_id,struct rnbd_srv_session * srv_sess)88 rnbd_get_sess_dev(int dev_id, struct rnbd_srv_session *srv_sess)
89 {
90 struct rnbd_srv_sess_dev *sess_dev;
91 int ret = 0;
92
93 rcu_read_lock();
94 sess_dev = xa_load(&srv_sess->index_idr, dev_id);
95 if (sess_dev)
96 ret = kref_get_unless_zero(&sess_dev->kref);
97 rcu_read_unlock();
98
99 if (!ret)
100 return ERR_PTR(-ENXIO);
101
102 return sess_dev;
103 }
104
rnbd_dev_bi_end_io(struct bio * bio)105 static void rnbd_dev_bi_end_io(struct bio *bio)
106 {
107 struct rnbd_io_private *rnbd_priv = bio->bi_private;
108 struct rnbd_srv_sess_dev *sess_dev = rnbd_priv->sess_dev;
109
110 rnbd_put_sess_dev(sess_dev);
111 rtrs_srv_resp_rdma(rnbd_priv->id, blk_status_to_errno(bio->bi_status));
112
113 kfree(rnbd_priv);
114 bio_put(bio);
115 }
116
process_rdma(struct rnbd_srv_session * srv_sess,struct rtrs_srv_op * id,void * data,u32 datalen,const void * usr,size_t usrlen)117 static int process_rdma(struct rnbd_srv_session *srv_sess,
118 struct rtrs_srv_op *id, void *data, u32 datalen,
119 const void *usr, size_t usrlen)
120 {
121 const struct rnbd_msg_io *msg = usr;
122 struct rnbd_io_private *priv;
123 struct rnbd_srv_sess_dev *sess_dev;
124 u32 dev_id;
125 int err;
126 struct bio *bio;
127 short prio;
128
129 trace_process_rdma(srv_sess, msg, id, datalen, usrlen);
130
131 priv = kmalloc(sizeof(*priv), GFP_KERNEL);
132 if (!priv)
133 return -ENOMEM;
134
135 dev_id = le32_to_cpu(msg->device_id);
136
137 sess_dev = rnbd_get_sess_dev(dev_id, srv_sess);
138 if (IS_ERR(sess_dev)) {
139 pr_err_ratelimited("Got I/O request on session %s for unknown device id %d: %pe\n",
140 srv_sess->sessname, dev_id, sess_dev);
141 err = -ENOTCONN;
142 goto err;
143 }
144
145 priv->sess_dev = sess_dev;
146 priv->id = id;
147
148 bio = bio_alloc(file_bdev(sess_dev->bdev_file), 1,
149 rnbd_to_bio_flags(le32_to_cpu(msg->rw)), GFP_KERNEL);
150 bio_add_virt_nofail(bio, data, datalen);
151
152 bio->bi_opf = rnbd_to_bio_flags(le32_to_cpu(msg->rw));
153 if (bio_has_data(bio) &&
154 bio->bi_iter.bi_size != le32_to_cpu(msg->bi_size)) {
155 rnbd_srv_err_rl(sess_dev, "Datalen mismatch: bio bi_size (%u), bi_size (%u)\n",
156 bio->bi_iter.bi_size, msg->bi_size);
157 err = -EINVAL;
158 goto bio_put;
159 }
160 bio->bi_end_io = rnbd_dev_bi_end_io;
161 bio->bi_private = priv;
162 bio->bi_iter.bi_sector = le64_to_cpu(msg->sector);
163 prio = srv_sess->ver < RNBD_PROTO_VER_MAJOR ||
164 usrlen < sizeof(*msg) ? 0 : le16_to_cpu(msg->prio);
165 bio->bi_ioprio = prio;
166
167 submit_bio(bio);
168
169 return 0;
170
171 bio_put:
172 bio_put(bio);
173 rnbd_put_sess_dev(sess_dev);
174 err:
175 kfree(priv);
176 return err;
177 }
178
destroy_device(struct kref * kref)179 static void destroy_device(struct kref *kref)
180 {
181 struct rnbd_srv_dev *dev = container_of(kref, struct rnbd_srv_dev, kref);
182
183 WARN_ONCE(!list_empty(&dev->sess_dev_list),
184 "Device %s is being destroyed but still in use!\n",
185 dev->name);
186
187 spin_lock(&dev_lock);
188 list_del(&dev->list);
189 spin_unlock(&dev_lock);
190
191 mutex_destroy(&dev->lock);
192 if (dev->dev_kobj.state_in_sysfs)
193 /*
194 * Destroy kobj only if it was really created.
195 */
196 rnbd_srv_destroy_dev_sysfs(dev);
197 else
198 kfree(dev);
199 }
200
rnbd_put_srv_dev(struct rnbd_srv_dev * dev)201 static void rnbd_put_srv_dev(struct rnbd_srv_dev *dev)
202 {
203 kref_put(&dev->kref, destroy_device);
204 }
205
rnbd_destroy_sess_dev(struct rnbd_srv_sess_dev * sess_dev,bool keep_id)206 void rnbd_destroy_sess_dev(struct rnbd_srv_sess_dev *sess_dev, bool keep_id)
207 {
208 DECLARE_COMPLETION_ONSTACK(dc);
209
210 if (keep_id)
211 /* free the resources for the id but don't */
212 /* allow to re-use the id itself because it */
213 /* is still used by the client */
214 xa_cmpxchg(&sess_dev->sess->index_idr, sess_dev->device_id,
215 sess_dev, NULL, 0);
216 else
217 xa_erase(&sess_dev->sess->index_idr, sess_dev->device_id);
218 synchronize_rcu();
219
220 sess_dev->destroy_comp = &dc;
221 rnbd_put_sess_dev(sess_dev);
222 wait_for_completion(&dc); /* wait for inflights to drop to zero */
223
224 fput(sess_dev->bdev_file);
225 mutex_lock(&sess_dev->dev->lock);
226 list_del(&sess_dev->dev_list);
227 if (!sess_dev->readonly)
228 sess_dev->dev->open_write_cnt--;
229 mutex_unlock(&sess_dev->dev->lock);
230
231 rnbd_put_srv_dev(sess_dev->dev);
232
233 rnbd_srv_info(sess_dev, "Device closed\n");
234 kfree(sess_dev);
235 }
236
destroy_sess(struct rnbd_srv_session * srv_sess)237 static void destroy_sess(struct rnbd_srv_session *srv_sess)
238 {
239 struct rnbd_srv_sess_dev *sess_dev;
240 unsigned long index;
241
242 if (xa_empty(&srv_sess->index_idr))
243 goto out;
244
245 trace_destroy_sess(srv_sess);
246
247 mutex_lock(&srv_sess->lock);
248 xa_for_each(&srv_sess->index_idr, index, sess_dev)
249 rnbd_srv_destroy_dev_session_sysfs(sess_dev);
250 mutex_unlock(&srv_sess->lock);
251
252 out:
253 xa_destroy(&srv_sess->index_idr);
254
255 pr_info("RTRS Session %s disconnected\n", srv_sess->sessname);
256
257 mutex_lock(&sess_lock);
258 list_del(&srv_sess->list);
259 mutex_unlock(&sess_lock);
260
261 mutex_destroy(&srv_sess->lock);
262 kfree(srv_sess);
263 }
264
create_sess(struct rtrs_srv_sess * rtrs)265 static int create_sess(struct rtrs_srv_sess *rtrs)
266 {
267 struct rnbd_srv_session *srv_sess;
268 char pathname[NAME_MAX];
269 int err;
270
271 err = rtrs_srv_get_path_name(rtrs, pathname, sizeof(pathname));
272 if (err) {
273 pr_err("rtrs_srv_get_path_name(%s): %d\n", pathname, err);
274
275 return err;
276 }
277 srv_sess = kzalloc(sizeof(*srv_sess), GFP_KERNEL);
278 if (!srv_sess)
279 return -ENOMEM;
280
281 srv_sess->queue_depth = rtrs_srv_get_queue_depth(rtrs);
282 xa_init_flags(&srv_sess->index_idr, XA_FLAGS_ALLOC);
283 mutex_init(&srv_sess->lock);
284 mutex_lock(&sess_lock);
285 list_add(&srv_sess->list, &sess_list);
286 mutex_unlock(&sess_lock);
287
288 srv_sess->rtrs = rtrs;
289 strscpy(srv_sess->sessname, pathname, sizeof(srv_sess->sessname));
290
291 rtrs_srv_set_sess_priv(rtrs, srv_sess);
292
293 trace_create_sess(srv_sess);
294
295 return 0;
296 }
297
rnbd_srv_link_ev(struct rtrs_srv_sess * rtrs,enum rtrs_srv_link_ev ev,void * priv)298 static int rnbd_srv_link_ev(struct rtrs_srv_sess *rtrs,
299 enum rtrs_srv_link_ev ev, void *priv)
300 {
301 struct rnbd_srv_session *srv_sess = priv;
302
303 switch (ev) {
304 case RTRS_SRV_LINK_EV_CONNECTED:
305 return create_sess(rtrs);
306
307 case RTRS_SRV_LINK_EV_DISCONNECTED:
308 if (WARN_ON_ONCE(!srv_sess))
309 return -EINVAL;
310
311 destroy_sess(srv_sess);
312 return 0;
313
314 default:
315 pr_warn("Received unknown RTRS session event %d from session %s\n",
316 ev, srv_sess->sessname);
317 return -EINVAL;
318 }
319 }
320
rnbd_srv_sess_dev_force_close(struct rnbd_srv_sess_dev * sess_dev,struct kobj_attribute * attr)321 void rnbd_srv_sess_dev_force_close(struct rnbd_srv_sess_dev *sess_dev,
322 struct kobj_attribute *attr)
323 {
324 struct rnbd_srv_session *sess = sess_dev->sess;
325
326 /* It is already started to close by client's close message. */
327 if (!mutex_trylock(&sess->lock))
328 return;
329
330 sess_dev->keep_id = true;
331 /* first remove sysfs itself to avoid deadlock */
332 sysfs_remove_file_self(&sess_dev->kobj, &attr->attr);
333 rnbd_srv_destroy_dev_session_sysfs(sess_dev);
334 mutex_unlock(&sess->lock);
335 }
336
process_msg_close(struct rnbd_srv_session * srv_sess,void * data,size_t datalen,const void * usr,size_t usrlen)337 static void process_msg_close(struct rnbd_srv_session *srv_sess,
338 void *data, size_t datalen, const void *usr,
339 size_t usrlen)
340 {
341 const struct rnbd_msg_close *close_msg = usr;
342 struct rnbd_srv_sess_dev *sess_dev;
343
344 trace_process_msg_close(srv_sess, close_msg);
345
346 sess_dev = rnbd_get_sess_dev(le32_to_cpu(close_msg->device_id),
347 srv_sess);
348 if (IS_ERR(sess_dev))
349 return;
350
351 rnbd_put_sess_dev(sess_dev);
352 mutex_lock(&srv_sess->lock);
353 rnbd_srv_destroy_dev_session_sysfs(sess_dev);
354 mutex_unlock(&srv_sess->lock);
355 }
356
357 static int process_msg_open(struct rnbd_srv_session *srv_sess,
358 const void *msg, size_t len,
359 void *data, size_t datalen);
360
361 static void process_msg_sess_info(struct rnbd_srv_session *srv_sess,
362 const void *msg, size_t len,
363 void *data, size_t datalen);
364
rnbd_srv_rdma_ev(void * priv,struct rtrs_srv_op * id,void * data,size_t datalen,const void * usr,size_t usrlen)365 static int rnbd_srv_rdma_ev(void *priv, struct rtrs_srv_op *id,
366 void *data, size_t datalen,
367 const void *usr, size_t usrlen)
368 {
369 struct rnbd_srv_session *srv_sess = priv;
370 const struct rnbd_msg_hdr *hdr = usr;
371 int ret = 0;
372 u16 type;
373
374 if (WARN_ON_ONCE(!srv_sess))
375 return -ENODEV;
376
377 type = le16_to_cpu(hdr->type);
378
379 switch (type) {
380 case RNBD_MSG_IO:
381 return process_rdma(srv_sess, id, data, datalen, usr, usrlen);
382 case RNBD_MSG_CLOSE:
383 process_msg_close(srv_sess, data, datalen, usr, usrlen);
384 break;
385 case RNBD_MSG_OPEN:
386 ret = process_msg_open(srv_sess, usr, usrlen, data, datalen);
387 break;
388 case RNBD_MSG_SESS_INFO:
389 process_msg_sess_info(srv_sess, usr, usrlen, data, datalen);
390 break;
391 default:
392 pr_warn("Received unexpected message type %d from session %s\n",
393 type, srv_sess->sessname);
394 return -EINVAL;
395 }
396
397 /*
398 * Since ret is passed to rtrs to handle the failure case, we
399 * just return 0 at the end otherwise callers in rtrs would call
400 * send_io_resp_imm again to print redundant err message.
401 */
402 rtrs_srv_resp_rdma(id, ret);
403 return 0;
404 }
405
406 static struct rnbd_srv_sess_dev
rnbd_sess_dev_alloc(struct rnbd_srv_session * srv_sess)407 *rnbd_sess_dev_alloc(struct rnbd_srv_session *srv_sess)
408 {
409 struct rnbd_srv_sess_dev *sess_dev;
410 int error;
411
412 sess_dev = kzalloc(sizeof(*sess_dev), GFP_KERNEL);
413 if (!sess_dev)
414 return ERR_PTR(-ENOMEM);
415
416 error = xa_alloc(&srv_sess->index_idr, &sess_dev->device_id, sess_dev,
417 xa_limit_32b, GFP_NOWAIT);
418 if (error < 0) {
419 pr_warn("Allocating idr failed, err: %d\n", error);
420 kfree(sess_dev);
421 return ERR_PTR(error);
422 }
423
424 return sess_dev;
425 }
426
rnbd_srv_init_srv_dev(struct block_device * bdev)427 static struct rnbd_srv_dev *rnbd_srv_init_srv_dev(struct block_device *bdev)
428 {
429 struct rnbd_srv_dev *dev;
430
431 dev = kzalloc(sizeof(*dev), GFP_KERNEL);
432 if (!dev)
433 return ERR_PTR(-ENOMEM);
434
435 snprintf(dev->name, sizeof(dev->name), "%pg", bdev);
436 kref_init(&dev->kref);
437 INIT_LIST_HEAD(&dev->sess_dev_list);
438 mutex_init(&dev->lock);
439
440 return dev;
441 }
442
443 static struct rnbd_srv_dev *
rnbd_srv_find_or_add_srv_dev(struct rnbd_srv_dev * new_dev)444 rnbd_srv_find_or_add_srv_dev(struct rnbd_srv_dev *new_dev)
445 {
446 struct rnbd_srv_dev *dev;
447
448 spin_lock(&dev_lock);
449 list_for_each_entry(dev, &dev_list, list) {
450 if (!strncmp(dev->name, new_dev->name, sizeof(dev->name))) {
451 if (!kref_get_unless_zero(&dev->kref))
452 /*
453 * We lost the race, device is almost dead.
454 * Continue traversing to find a valid one.
455 */
456 continue;
457 spin_unlock(&dev_lock);
458 return dev;
459 }
460 }
461 list_add(&new_dev->list, &dev_list);
462 spin_unlock(&dev_lock);
463
464 return new_dev;
465 }
466
rnbd_srv_check_update_open_perm(struct rnbd_srv_dev * srv_dev,struct rnbd_srv_session * srv_sess,enum rnbd_access_mode access_mode)467 static int rnbd_srv_check_update_open_perm(struct rnbd_srv_dev *srv_dev,
468 struct rnbd_srv_session *srv_sess,
469 enum rnbd_access_mode access_mode)
470 {
471 int ret = 0;
472
473 mutex_lock(&srv_dev->lock);
474
475 switch (access_mode) {
476 case RNBD_ACCESS_RO:
477 break;
478 case RNBD_ACCESS_RW:
479 if (srv_dev->open_write_cnt == 0) {
480 srv_dev->open_write_cnt++;
481 } else {
482 pr_err("Mapping device '%s' for session %s with RW permissions failed. Device already opened as 'RW' by %d client(s), access mode %s.\n",
483 srv_dev->name, srv_sess->sessname,
484 srv_dev->open_write_cnt,
485 rnbd_access_modes[access_mode].str);
486 ret = -EPERM;
487 }
488 break;
489 case RNBD_ACCESS_MIGRATION:
490 if (srv_dev->open_write_cnt < 2) {
491 srv_dev->open_write_cnt++;
492 } else {
493 pr_err("Mapping device '%s' for session %s with migration permissions failed. Device already opened as 'RW' by %d client(s), access mode %s.\n",
494 srv_dev->name, srv_sess->sessname,
495 srv_dev->open_write_cnt,
496 rnbd_access_modes[access_mode].str);
497 ret = -EPERM;
498 }
499 break;
500 default:
501 pr_err("Received mapping request for device '%s' on session %s with invalid access mode: %d\n",
502 srv_dev->name, srv_sess->sessname, access_mode);
503 ret = -EINVAL;
504 }
505
506 mutex_unlock(&srv_dev->lock);
507
508 return ret;
509 }
510
511 static struct rnbd_srv_dev *
rnbd_srv_get_or_create_srv_dev(struct block_device * bdev,struct rnbd_srv_session * srv_sess,enum rnbd_access_mode access_mode)512 rnbd_srv_get_or_create_srv_dev(struct block_device *bdev,
513 struct rnbd_srv_session *srv_sess,
514 enum rnbd_access_mode access_mode)
515 {
516 int ret;
517 struct rnbd_srv_dev *new_dev, *dev;
518
519 new_dev = rnbd_srv_init_srv_dev(bdev);
520 if (IS_ERR(new_dev))
521 return new_dev;
522
523 dev = rnbd_srv_find_or_add_srv_dev(new_dev);
524 if (dev != new_dev)
525 kfree(new_dev);
526
527 ret = rnbd_srv_check_update_open_perm(dev, srv_sess, access_mode);
528 if (ret) {
529 rnbd_put_srv_dev(dev);
530 return ERR_PTR(ret);
531 }
532
533 return dev;
534 }
535
rnbd_srv_fill_msg_open_rsp(struct rnbd_msg_open_rsp * rsp,struct rnbd_srv_sess_dev * sess_dev)536 static void rnbd_srv_fill_msg_open_rsp(struct rnbd_msg_open_rsp *rsp,
537 struct rnbd_srv_sess_dev *sess_dev)
538 {
539 struct block_device *bdev = file_bdev(sess_dev->bdev_file);
540
541 rsp->hdr.type = cpu_to_le16(RNBD_MSG_OPEN_RSP);
542 rsp->device_id = cpu_to_le32(sess_dev->device_id);
543 rsp->nsectors = cpu_to_le64(bdev_nr_sectors(bdev));
544 rsp->logical_block_size = cpu_to_le16(bdev_logical_block_size(bdev));
545 rsp->physical_block_size = cpu_to_le16(bdev_physical_block_size(bdev));
546 rsp->max_segments = cpu_to_le16(bdev_max_segments(bdev));
547 rsp->max_hw_sectors =
548 cpu_to_le32(queue_max_hw_sectors(bdev_get_queue(bdev)));
549 rsp->max_write_zeroes_sectors =
550 cpu_to_le32(bdev_write_zeroes_sectors(bdev));
551 rsp->max_discard_sectors = cpu_to_le32(bdev_max_discard_sectors(bdev));
552 rsp->discard_granularity = cpu_to_le32(bdev_discard_granularity(bdev));
553 rsp->discard_alignment = cpu_to_le32(bdev_discard_alignment(bdev));
554 rsp->secure_discard = cpu_to_le16(bdev_max_secure_erase_sectors(bdev));
555 rsp->cache_policy = 0;
556 if (bdev_write_cache(bdev))
557 rsp->cache_policy |= RNBD_WRITEBACK;
558 if (bdev_fua(bdev))
559 rsp->cache_policy |= RNBD_FUA;
560 }
561
562 static struct rnbd_srv_sess_dev *
rnbd_srv_create_set_sess_dev(struct rnbd_srv_session * srv_sess,const struct rnbd_msg_open * open_msg,struct file * bdev_file,bool readonly,struct rnbd_srv_dev * srv_dev)563 rnbd_srv_create_set_sess_dev(struct rnbd_srv_session *srv_sess,
564 const struct rnbd_msg_open *open_msg,
565 struct file *bdev_file, bool readonly,
566 struct rnbd_srv_dev *srv_dev)
567 {
568 struct rnbd_srv_sess_dev *sdev = rnbd_sess_dev_alloc(srv_sess);
569
570 if (IS_ERR(sdev))
571 return sdev;
572
573 kref_init(&sdev->kref);
574
575 strscpy(sdev->pathname, open_msg->dev_name, sizeof(sdev->pathname));
576
577 sdev->bdev_file = bdev_file;
578 sdev->sess = srv_sess;
579 sdev->dev = srv_dev;
580 sdev->readonly = readonly;
581 sdev->access_mode = open_msg->access_mode;
582
583 return sdev;
584 }
585
rnbd_srv_get_full_path(struct rnbd_srv_session * srv_sess,const char * dev_name)586 static char *rnbd_srv_get_full_path(struct rnbd_srv_session *srv_sess,
587 const char *dev_name)
588 {
589 char *full_path;
590 char *a, *b;
591 int len;
592
593 full_path = kmalloc(PATH_MAX, GFP_KERNEL);
594 if (!full_path)
595 return ERR_PTR(-ENOMEM);
596
597 /*
598 * Replace %SESSNAME% with a real session name in order to
599 * create device namespace.
600 */
601 a = strnstr(dev_search_path, "%SESSNAME%", sizeof(dev_search_path));
602 if (a) {
603 len = a - dev_search_path;
604
605 len = snprintf(full_path, PATH_MAX, "%.*s/%s/%s", len,
606 dev_search_path, srv_sess->sessname, dev_name);
607 } else {
608 len = snprintf(full_path, PATH_MAX, "%s/%s",
609 dev_search_path, dev_name);
610 }
611 if (len >= PATH_MAX) {
612 pr_err("Too long path: %s, %s, %s\n",
613 dev_search_path, srv_sess->sessname, dev_name);
614 kfree(full_path);
615 return ERR_PTR(-EINVAL);
616 }
617
618 /* eliminitate duplicated slashes */
619 a = strchr(full_path, '/');
620 b = a;
621 while (*b != '\0') {
622 if (*b == '/' && *a == '/') {
623 b++;
624 } else {
625 a++;
626 *a = *b;
627 b++;
628 }
629 }
630 a++;
631 *a = '\0';
632
633 return full_path;
634 }
635
process_msg_sess_info(struct rnbd_srv_session * srv_sess,const void * msg,size_t len,void * data,size_t datalen)636 static void process_msg_sess_info(struct rnbd_srv_session *srv_sess,
637 const void *msg, size_t len,
638 void *data, size_t datalen)
639 {
640 const struct rnbd_msg_sess_info *sess_info_msg = msg;
641 struct rnbd_msg_sess_info_rsp *rsp = data;
642
643 srv_sess->ver = min_t(u8, sess_info_msg->ver, RNBD_PROTO_VER_MAJOR);
644
645 trace_process_msg_sess_info(srv_sess, sess_info_msg);
646
647 rsp->hdr.type = cpu_to_le16(RNBD_MSG_SESS_INFO_RSP);
648 rsp->ver = srv_sess->ver;
649 }
650
651 /**
652 * find_srv_sess_dev() - a dev is already opened by this name
653 * @srv_sess: the session to search.
654 * @dev_name: string containing the name of the device.
655 *
656 * Return struct rnbd_srv_sess_dev if srv_sess already opened the dev_name
657 * NULL if the session didn't open the device yet.
658 */
659 static struct rnbd_srv_sess_dev *
find_srv_sess_dev(struct rnbd_srv_session * srv_sess,const char * dev_name)660 find_srv_sess_dev(struct rnbd_srv_session *srv_sess, const char *dev_name)
661 {
662 struct rnbd_srv_sess_dev *sess_dev;
663 unsigned long index;
664
665 if (xa_empty(&srv_sess->index_idr))
666 return NULL;
667
668 xa_for_each(&srv_sess->index_idr, index, sess_dev)
669 if (!strcmp(sess_dev->pathname, dev_name))
670 return sess_dev;
671
672 return NULL;
673 }
674
process_msg_open(struct rnbd_srv_session * srv_sess,const void * msg,size_t len,void * data,size_t datalen)675 static int process_msg_open(struct rnbd_srv_session *srv_sess,
676 const void *msg, size_t len,
677 void *data, size_t datalen)
678 {
679 int ret;
680 struct rnbd_srv_dev *srv_dev;
681 struct rnbd_srv_sess_dev *srv_sess_dev;
682 const struct rnbd_msg_open *open_msg = msg;
683 struct file *bdev_file;
684 blk_mode_t open_flags = BLK_OPEN_READ;
685 char *full_path;
686 struct rnbd_msg_open_rsp *rsp = data;
687
688 trace_process_msg_open(srv_sess, open_msg);
689
690 if (open_msg->access_mode != RNBD_ACCESS_RO)
691 open_flags |= BLK_OPEN_WRITE;
692
693 mutex_lock(&srv_sess->lock);
694
695 srv_sess_dev = find_srv_sess_dev(srv_sess, open_msg->dev_name);
696 if (srv_sess_dev)
697 goto fill_response;
698
699 if ((strlen(dev_search_path) + strlen(open_msg->dev_name))
700 >= PATH_MAX) {
701 pr_err("Opening device for session %s failed, device path too long. '%s/%s' is longer than PATH_MAX (%d)\n",
702 srv_sess->sessname, dev_search_path, open_msg->dev_name,
703 PATH_MAX);
704 ret = -EINVAL;
705 goto reject;
706 }
707 if (strstr(open_msg->dev_name, "..")) {
708 pr_err("Opening device for session %s failed, device path %s contains relative path ..\n",
709 srv_sess->sessname, open_msg->dev_name);
710 ret = -EINVAL;
711 goto reject;
712 }
713 full_path = rnbd_srv_get_full_path(srv_sess, open_msg->dev_name);
714 if (IS_ERR(full_path)) {
715 ret = PTR_ERR(full_path);
716 pr_err("Opening device '%s' for client %s failed, failed to get device full path, err: %pe\n",
717 open_msg->dev_name, srv_sess->sessname, full_path);
718 goto reject;
719 }
720
721 bdev_file = bdev_file_open_by_path(full_path, open_flags, NULL, NULL);
722 if (IS_ERR(bdev_file)) {
723 ret = PTR_ERR(bdev_file);
724 pr_err("Opening device '%s' on session %s failed, failed to open the block device, err: %pe\n",
725 full_path, srv_sess->sessname, bdev_file);
726 goto free_path;
727 }
728
729 srv_dev = rnbd_srv_get_or_create_srv_dev(file_bdev(bdev_file), srv_sess,
730 open_msg->access_mode);
731 if (IS_ERR(srv_dev)) {
732 pr_err("Opening device '%s' on session %s failed, creating srv_dev failed, err: %pe\n",
733 full_path, srv_sess->sessname, srv_dev);
734 ret = PTR_ERR(srv_dev);
735 goto blkdev_put;
736 }
737
738 srv_sess_dev = rnbd_srv_create_set_sess_dev(srv_sess, open_msg,
739 bdev_file,
740 open_msg->access_mode == RNBD_ACCESS_RO,
741 srv_dev);
742 if (IS_ERR(srv_sess_dev)) {
743 pr_err("Opening device '%s' on session %s failed, creating sess_dev failed, err: %pe\n",
744 full_path, srv_sess->sessname, srv_sess_dev);
745 ret = PTR_ERR(srv_sess_dev);
746 goto srv_dev_put;
747 }
748
749 /* Create the srv_dev sysfs files if they haven't been created yet. The
750 * reason to delay the creation is not to create the sysfs files before
751 * we are sure the device can be opened.
752 */
753 mutex_lock(&srv_dev->lock);
754 if (!srv_dev->dev_kobj.state_in_sysfs) {
755 ret = rnbd_srv_create_dev_sysfs(srv_dev, file_bdev(bdev_file));
756 if (ret) {
757 mutex_unlock(&srv_dev->lock);
758 rnbd_srv_err(srv_sess_dev,
759 "Opening device failed, failed to create device sysfs files, err: %d\n",
760 ret);
761 goto free_srv_sess_dev;
762 }
763 }
764
765 ret = rnbd_srv_create_dev_session_sysfs(srv_sess_dev);
766 if (ret) {
767 mutex_unlock(&srv_dev->lock);
768 rnbd_srv_err(srv_sess_dev,
769 "Opening device failed, failed to create dev client sysfs files, err: %d\n",
770 ret);
771 goto free_srv_sess_dev;
772 }
773
774 list_add(&srv_sess_dev->dev_list, &srv_dev->sess_dev_list);
775 mutex_unlock(&srv_dev->lock);
776
777 rnbd_srv_info(srv_sess_dev, "Opened device '%s'\n", srv_dev->name);
778
779 kfree(full_path);
780
781 fill_response:
782 rnbd_srv_fill_msg_open_rsp(rsp, srv_sess_dev);
783 mutex_unlock(&srv_sess->lock);
784 return 0;
785
786 free_srv_sess_dev:
787 xa_erase(&srv_sess->index_idr, srv_sess_dev->device_id);
788 synchronize_rcu();
789 kfree(srv_sess_dev);
790 srv_dev_put:
791 if (open_msg->access_mode != RNBD_ACCESS_RO) {
792 mutex_lock(&srv_dev->lock);
793 srv_dev->open_write_cnt--;
794 mutex_unlock(&srv_dev->lock);
795 }
796 rnbd_put_srv_dev(srv_dev);
797 blkdev_put:
798 fput(bdev_file);
799 free_path:
800 kfree(full_path);
801 reject:
802 mutex_unlock(&srv_sess->lock);
803 return ret;
804 }
805
806 static struct rtrs_srv_ctx *rtrs_ctx;
807
808 static struct rtrs_srv_ops rtrs_ops;
rnbd_srv_init_module(void)809 static int __init rnbd_srv_init_module(void)
810 {
811 int err = 0;
812
813 BUILD_BUG_ON(sizeof(struct rnbd_msg_hdr) != 4);
814 BUILD_BUG_ON(sizeof(struct rnbd_msg_sess_info) != 36);
815 BUILD_BUG_ON(sizeof(struct rnbd_msg_sess_info_rsp) != 36);
816 BUILD_BUG_ON(sizeof(struct rnbd_msg_open) != 264);
817 BUILD_BUG_ON(sizeof(struct rnbd_msg_close) != 8);
818 BUILD_BUG_ON(sizeof(struct rnbd_msg_open_rsp) != 56);
819 rtrs_ops = (struct rtrs_srv_ops) {
820 .rdma_ev = rnbd_srv_rdma_ev,
821 .link_ev = rnbd_srv_link_ev,
822 };
823 rtrs_ctx = rtrs_srv_open(&rtrs_ops, port_nr);
824 if (IS_ERR(rtrs_ctx)) {
825 pr_err("rtrs_srv_open(), err: %pe\n", rtrs_ctx);
826 return PTR_ERR(rtrs_ctx);
827 }
828
829 err = rnbd_srv_create_sysfs_files();
830 if (err) {
831 pr_err("rnbd_srv_create_sysfs_files(), err: %d\n", err);
832 rtrs_srv_close(rtrs_ctx);
833 }
834
835 return err;
836 }
837
rnbd_srv_cleanup_module(void)838 static void __exit rnbd_srv_cleanup_module(void)
839 {
840 rtrs_srv_close(rtrs_ctx);
841 WARN_ON(!list_empty(&sess_list));
842 rnbd_srv_destroy_sysfs_files();
843 }
844
845 module_init(rnbd_srv_init_module);
846 module_exit(rnbd_srv_cleanup_module);
847