netlink.c 2.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139
  1. // SPDX-License-Identifier: GPL-2.0-only
  2. /*
  3. * Copyright (C) 2007 Red Hat, Inc. All rights reserved.
  4. */
  5. #include <net/genetlink.h>
  6. #include <linux/dlm.h>
  7. #include <linux/dlm_netlink.h>
  8. #include <linux/gfp.h>
  9. #include "dlm_internal.h"
  10. static uint32_t dlm_nl_seqnum;
  11. static uint32_t listener_nlportid;
  12. static struct genl_family family;
  13. static int prepare_data(u8 cmd, struct sk_buff **skbp, size_t size)
  14. {
  15. struct sk_buff *skb;
  16. void *data;
  17. skb = genlmsg_new(size, GFP_NOFS);
  18. if (!skb)
  19. return -ENOMEM;
  20. /* add the message headers */
  21. data = genlmsg_put(skb, 0, dlm_nl_seqnum++, &family, 0, cmd);
  22. if (!data) {
  23. nlmsg_free(skb);
  24. return -EINVAL;
  25. }
  26. *skbp = skb;
  27. return 0;
  28. }
  29. static struct dlm_lock_data *mk_data(struct sk_buff *skb)
  30. {
  31. struct nlattr *ret;
  32. ret = nla_reserve(skb, DLM_TYPE_LOCK, sizeof(struct dlm_lock_data));
  33. if (!ret)
  34. return NULL;
  35. return nla_data(ret);
  36. }
  37. static int send_data(struct sk_buff *skb)
  38. {
  39. struct genlmsghdr *genlhdr = nlmsg_data((struct nlmsghdr *)skb->data);
  40. void *data = genlmsg_data(genlhdr);
  41. genlmsg_end(skb, data);
  42. return genlmsg_unicast(&init_net, skb, listener_nlportid);
  43. }
  44. static int user_cmd(struct sk_buff *skb, struct genl_info *info)
  45. {
  46. listener_nlportid = info->snd_portid;
  47. printk("user_cmd nlpid %u\n", listener_nlportid);
  48. return 0;
  49. }
  50. static const struct genl_small_ops dlm_nl_ops[] = {
  51. {
  52. .cmd = DLM_CMD_HELLO,
  53. .validate = GENL_DONT_VALIDATE_STRICT | GENL_DONT_VALIDATE_DUMP,
  54. .doit = user_cmd,
  55. },
  56. };
  57. static struct genl_family family __ro_after_init = {
  58. .name = DLM_GENL_NAME,
  59. .version = DLM_GENL_VERSION,
  60. .small_ops = dlm_nl_ops,
  61. .n_small_ops = ARRAY_SIZE(dlm_nl_ops),
  62. .resv_start_op = DLM_CMD_HELLO + 1,
  63. .module = THIS_MODULE,
  64. };
  65. int __init dlm_netlink_init(void)
  66. {
  67. return genl_register_family(&family);
  68. }
  69. void dlm_netlink_exit(void)
  70. {
  71. genl_unregister_family(&family);
  72. }
  73. static void fill_data(struct dlm_lock_data *data, struct dlm_lkb *lkb)
  74. {
  75. struct dlm_rsb *r = lkb->lkb_resource;
  76. memset(data, 0, sizeof(struct dlm_lock_data));
  77. data->version = DLM_LOCK_DATA_VERSION;
  78. data->nodeid = lkb->lkb_nodeid;
  79. data->ownpid = lkb->lkb_ownpid;
  80. data->id = lkb->lkb_id;
  81. data->remid = lkb->lkb_remid;
  82. data->status = lkb->lkb_status;
  83. data->grmode = lkb->lkb_grmode;
  84. data->rqmode = lkb->lkb_rqmode;
  85. if (lkb->lkb_ua)
  86. data->xid = lkb->lkb_ua->xid;
  87. if (r) {
  88. data->lockspace_id = r->res_ls->ls_global_id;
  89. data->resource_namelen = r->res_length;
  90. memcpy(data->resource_name, r->res_name, r->res_length);
  91. }
  92. }
  93. void dlm_timeout_warn(struct dlm_lkb *lkb)
  94. {
  95. struct sk_buff *send_skb;
  96. struct dlm_lock_data *data;
  97. size_t size;
  98. int rv;
  99. size = nla_total_size(sizeof(struct dlm_lock_data)) +
  100. nla_total_size(0); /* why this? */
  101. rv = prepare_data(DLM_CMD_TIMEOUT, &send_skb, size);
  102. if (rv < 0)
  103. return;
  104. data = mk_data(send_skb);
  105. if (!data) {
  106. nlmsg_free(send_skb);
  107. return;
  108. }
  109. fill_data(data, lkb);
  110. send_data(send_skb);
  111. }