genl.c 5.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213
  1. // SPDX-License-Identifier: GPL-2.0-only
  2. /* Copyright (c) 2020, The Linux Foundation. All rights reserved. */
  3. #define pr_fmt(fmt) "cnss_genl: " fmt
  4. #include <linux/err.h>
  5. #include <linux/module.h>
  6. #include <linux/delay.h>
  7. #include <net/netlink.h>
  8. #include <net/genetlink.h>
  9. #include "main.h"
  10. #include "debug.h"
  11. #define ICNSS_GENL_FAMILY_NAME "cnss-genl"
  12. #define ICNSS_GENL_MCAST_GROUP_NAME "cnss-genl-grp"
  13. #define ICNSS_GENL_VERSION 1
  14. #define ICNSS_GENL_DATA_LEN_MAX (15 * 1024)
  15. #define ICNSS_GENL_STR_LEN_MAX 16
  16. enum {
  17. ICNSS_GENL_ATTR_MSG_UNSPEC,
  18. ICNSS_GENL_ATTR_MSG_TYPE,
  19. ICNSS_GENL_ATTR_MSG_FILE_NAME,
  20. ICNSS_GENL_ATTR_MSG_TOTAL_SIZE,
  21. ICNSS_GENL_ATTR_MSG_SEG_ID,
  22. ICNSS_GENL_ATTR_MSG_END,
  23. ICNSS_GENL_ATTR_MSG_DATA_LEN,
  24. ICNSS_GENL_ATTR_MSG_DATA,
  25. __ICNSS_GENL_ATTR_MAX,
  26. };
  27. #define ICNSS_GENL_ATTR_MAX (__ICNSS_GENL_ATTR_MAX - 1)
  28. enum {
  29. ICNSS_GENL_CMD_UNSPEC,
  30. ICNSS_GENL_CMD_MSG,
  31. __ICNSS_GENL_CMD_MAX,
  32. };
  33. #define ICNSS_GENL_CMD_MAX (__ICNSS_GENL_CMD_MAX - 1)
  34. static struct nla_policy icnss_genl_msg_policy[ICNSS_GENL_ATTR_MAX + 1] = {
  35. [ICNSS_GENL_ATTR_MSG_TYPE] = { .type = NLA_U8 },
  36. [ICNSS_GENL_ATTR_MSG_FILE_NAME] = { .type = NLA_NUL_STRING,
  37. .len = ICNSS_GENL_STR_LEN_MAX },
  38. [ICNSS_GENL_ATTR_MSG_TOTAL_SIZE] = { .type = NLA_U32 },
  39. [ICNSS_GENL_ATTR_MSG_SEG_ID] = { .type = NLA_U32 },
  40. [ICNSS_GENL_ATTR_MSG_END] = { .type = NLA_U8 },
  41. [ICNSS_GENL_ATTR_MSG_DATA_LEN] = { .type = NLA_U32 },
  42. [ICNSS_GENL_ATTR_MSG_DATA] = { .type = NLA_BINARY,
  43. .len = ICNSS_GENL_DATA_LEN_MAX },
  44. };
  45. static int icnss_genl_process_msg(struct sk_buff *skb, struct genl_info *info)
  46. {
  47. return 0;
  48. }
  49. static struct genl_ops icnss_genl_ops[] = {
  50. {
  51. .cmd = ICNSS_GENL_CMD_MSG,
  52. .doit = icnss_genl_process_msg,
  53. },
  54. };
  55. static struct genl_multicast_group icnss_genl_mcast_grp[] = {
  56. {
  57. .name = ICNSS_GENL_MCAST_GROUP_NAME,
  58. },
  59. };
  60. static struct genl_family icnss_genl_family = {
  61. .id = 0,
  62. .hdrsize = 0,
  63. .name = ICNSS_GENL_FAMILY_NAME,
  64. .version = ICNSS_GENL_VERSION,
  65. .maxattr = ICNSS_GENL_ATTR_MAX,
  66. .policy = icnss_genl_msg_policy,
  67. .module = THIS_MODULE,
  68. .ops = icnss_genl_ops,
  69. .n_ops = ARRAY_SIZE(icnss_genl_ops),
  70. .mcgrps = icnss_genl_mcast_grp,
  71. .n_mcgrps = ARRAY_SIZE(icnss_genl_mcast_grp),
  72. };
  73. static int icnss_genl_send_data(u8 type, char *file_name, u32 total_size,
  74. u32 seg_id, u8 end, u32 data_len, u8 *msg_buff)
  75. {
  76. struct sk_buff *skb = NULL;
  77. void *msg_header = NULL;
  78. int ret = 0;
  79. char filename[ICNSS_GENL_STR_LEN_MAX + 1];
  80. icnss_pr_dbg("type: %u, file_name %s, total_size: %x, seg_id %u, end %u, data_len %u\n",
  81. type, file_name, total_size, seg_id, end, data_len);
  82. if (!file_name)
  83. strlcpy(filename, "default", sizeof(filename));
  84. else
  85. strlcpy(filename, file_name, sizeof(filename));
  86. skb = genlmsg_new(NLMSG_HDRLEN +
  87. nla_total_size(sizeof(type)) +
  88. nla_total_size(strlen(filename) + 1) +
  89. nla_total_size(sizeof(total_size)) +
  90. nla_total_size(sizeof(seg_id)) +
  91. nla_total_size(sizeof(end)) +
  92. nla_total_size(sizeof(data_len)) +
  93. nla_total_size(data_len), GFP_KERNEL);
  94. if (!skb)
  95. return -ENOMEM;
  96. msg_header = genlmsg_put(skb, 0, 0,
  97. &icnss_genl_family, 0,
  98. ICNSS_GENL_CMD_MSG);
  99. if (!msg_header) {
  100. ret = -ENOMEM;
  101. goto fail;
  102. }
  103. ret = nla_put_u8(skb, ICNSS_GENL_ATTR_MSG_TYPE, type);
  104. if (ret < 0)
  105. goto fail;
  106. ret = nla_put_string(skb, ICNSS_GENL_ATTR_MSG_FILE_NAME, filename);
  107. if (ret < 0)
  108. goto fail;
  109. ret = nla_put_u32(skb, ICNSS_GENL_ATTR_MSG_TOTAL_SIZE, total_size);
  110. if (ret < 0)
  111. goto fail;
  112. ret = nla_put_u32(skb, ICNSS_GENL_ATTR_MSG_SEG_ID, seg_id);
  113. if (ret < 0)
  114. goto fail;
  115. ret = nla_put_u8(skb, ICNSS_GENL_ATTR_MSG_END, end);
  116. if (ret < 0)
  117. goto fail;
  118. ret = nla_put_u32(skb, ICNSS_GENL_ATTR_MSG_DATA_LEN, data_len);
  119. if (ret < 0)
  120. goto fail;
  121. ret = nla_put(skb, ICNSS_GENL_ATTR_MSG_DATA, data_len, msg_buff);
  122. if (ret < 0)
  123. goto fail;
  124. genlmsg_end(skb, msg_header);
  125. ret = genlmsg_multicast(&icnss_genl_family, skb, 0, 0, GFP_KERNEL);
  126. if (ret < 0)
  127. icnss_pr_err("Fail to send genl msg: %d\n", ret);
  128. return ret;
  129. fail:
  130. icnss_pr_err("Fail to generate genl msg: %d\n", ret);
  131. if (skb)
  132. nlmsg_free(skb);
  133. return ret;
  134. }
  135. int icnss_genl_send_msg(void *buff, u8 type, char *file_name, u32 total_size)
  136. {
  137. int ret = 0;
  138. u8 *msg_buff = buff;
  139. u32 remaining = total_size;
  140. u32 seg_id = 0;
  141. u32 data_len = 0;
  142. u8 end = 0;
  143. u8 retry;
  144. icnss_pr_dbg("type: %u, total_size: %x\n", type, total_size);
  145. while (remaining) {
  146. if (remaining > ICNSS_GENL_DATA_LEN_MAX) {
  147. data_len = ICNSS_GENL_DATA_LEN_MAX;
  148. } else {
  149. data_len = remaining;
  150. end = 1;
  151. }
  152. for (retry = 0; retry < 2; retry++) {
  153. ret = icnss_genl_send_data(type, file_name, total_size,
  154. seg_id, end, data_len,
  155. msg_buff);
  156. if (ret >= 0)
  157. break;
  158. msleep(100);
  159. }
  160. if (ret < 0) {
  161. icnss_pr_err("fail to send genl data, ret %d\n", ret);
  162. return ret;
  163. }
  164. remaining -= data_len;
  165. msg_buff += data_len;
  166. seg_id++;
  167. }
  168. return ret;
  169. }
  170. int icnss_genl_init(void)
  171. {
  172. int ret = 0;
  173. ret = genl_register_family(&icnss_genl_family);
  174. if (ret != 0)
  175. icnss_pr_err("genl_register_family fail: %d\n", ret);
  176. return ret;
  177. }
  178. void icnss_genl_exit(void)
  179. {
  180. genl_unregister_family(&icnss_genl_family);
  181. }