idiag_msg_obj.c 19 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729
  1. /*
  2. * lib/idiag/idiagnl_msg_obj.c Inet Diag Message Object
  3. *
  4. * This library is free software; you can redistribute it and/or
  5. * modify it under the terms of the GNU Lesser General Public
  6. * License as published by the Free Software Foundation version 2.1
  7. * of the License.
  8. *
  9. * Copyright (c) 2013 Sassano Systems LLC <joe@sassanosystems.com>
  10. */
  11. #include <netlink-private/netlink.h>
  12. #include <netlink/idiag/msg.h>
  13. #include <netlink/idiag/meminfo.h>
  14. #include <netlink/idiag/vegasinfo.h>
  15. #include <linux/inet_diag.h>
  16. /**
  17. * @ingroup idiag
  18. * @defgroup idiagnl_msg Inet Diag Messages
  19. *
  20. * @details
  21. * @idiagnl_doc{idiagnl_msg, Inet Diag Message Documentation}
  22. * @{
  23. */
  24. struct idiagnl_msg *idiagnl_msg_alloc(void)
  25. {
  26. return (struct idiagnl_msg *) nl_object_alloc(&idiagnl_msg_obj_ops);
  27. }
  28. void idiagnl_msg_get(struct idiagnl_msg *msg)
  29. {
  30. nl_object_get((struct nl_object *) msg);
  31. }
  32. void idiagnl_msg_put(struct idiagnl_msg *msg)
  33. {
  34. nl_object_put((struct nl_object *) msg);
  35. }
  36. static struct nl_cache_ops idiagnl_msg_ops;
  37. static int idiagnl_msg_parser(struct nl_cache_ops *ops, struct sockaddr_nl *who,
  38. struct nlmsghdr *nlh, struct nl_parser_param *pp)
  39. {
  40. struct idiagnl_msg *msg = NULL;
  41. int err = 0;
  42. if ((err = idiagnl_msg_parse(nlh, &msg)) < 0)
  43. return err;
  44. err = pp->pp_cb((struct nl_object *) msg, pp);
  45. idiagnl_msg_put(msg);
  46. return err;
  47. }
  48. static int idiagnl_request_update(struct nl_cache *cache, struct nl_sock *sk)
  49. {
  50. int family = cache->c_iarg1;
  51. int states = cache->c_iarg2;
  52. return idiagnl_send_simple(sk, 0, family, states, IDIAG_ATTR_ALL);
  53. }
  54. static struct nl_cache_ops idiagnl_msg_ops = {
  55. .co_name = "idiag/idiag",
  56. .co_hdrsize = sizeof(struct inet_diag_msg),
  57. .co_msgtypes = {
  58. { IDIAG_TCPDIAG_GETSOCK, NL_ACT_NEW, "new" },
  59. { IDIAG_DCCPDIAG_GETSOCK, NL_ACT_NEW, "new" },
  60. END_OF_MSGTYPES_LIST,
  61. },
  62. .co_protocol = NETLINK_INET_DIAG,
  63. .co_request_update = idiagnl_request_update,
  64. .co_msg_parser = idiagnl_msg_parser,
  65. .co_obj_ops = &idiagnl_msg_obj_ops,
  66. };
  67. static void __init idiagnl_init(void)
  68. {
  69. nl_cache_mngt_register(&idiagnl_msg_ops);
  70. }
  71. static void __exit idiagnl_exit(void)
  72. {
  73. nl_cache_mngt_unregister(&idiagnl_msg_ops);
  74. }
  75. /**
  76. * @name Cache Management
  77. * @{
  78. */
  79. /**
  80. * Build an inetdiag cache to hold socket state information.
  81. * @arg sk Netlink socket
  82. * @arg family The address family to query
  83. * @arg states Socket states to query
  84. * @arg result Result pointer
  85. *
  86. * @note The caller is responsible for destroying and free the cache after using
  87. * it.
  88. * @return 0 on success of a negative error code.
  89. */
  90. int idiagnl_msg_alloc_cache(struct nl_sock *sk, int family, int states,
  91. struct nl_cache **result)
  92. {
  93. struct nl_cache *cache = NULL;
  94. int err;
  95. if (!(cache = nl_cache_alloc(&idiagnl_msg_ops)))
  96. return -NLE_NOMEM;
  97. cache->c_iarg1 = family;
  98. cache->c_iarg2 = states;
  99. if (sk && (err = nl_cache_refill(sk, cache)) < 0) {
  100. free(cache);
  101. return err;
  102. }
  103. *result = cache;
  104. return 0;
  105. }
  106. /** @} */
  107. /**
  108. * @name Attributes
  109. * @{
  110. */
  111. uint8_t idiagnl_msg_get_family(const struct idiagnl_msg *msg)
  112. {
  113. return msg->idiag_family;
  114. }
  115. void idiagnl_msg_set_family(struct idiagnl_msg *msg, uint8_t family)
  116. {
  117. msg->idiag_family = family;
  118. }
  119. uint8_t idiagnl_msg_get_state(const struct idiagnl_msg *msg)
  120. {
  121. return msg->idiag_state;
  122. }
  123. void idiagnl_msg_set_state(struct idiagnl_msg *msg, uint8_t state)
  124. {
  125. msg->idiag_state = state;
  126. }
  127. uint8_t idiagnl_msg_get_timer(const struct idiagnl_msg *msg)
  128. {
  129. return msg->idiag_timer;
  130. }
  131. void idiagnl_msg_set_timer(struct idiagnl_msg *msg, uint8_t timer)
  132. {
  133. msg->idiag_timer = timer;
  134. }
  135. uint8_t idiagnl_msg_get_retrans(const struct idiagnl_msg *msg)
  136. {
  137. return msg->idiag_retrans;
  138. }
  139. void idiagnl_msg_set_retrans(struct idiagnl_msg *msg, uint8_t retrans)
  140. {
  141. msg->idiag_retrans = retrans;
  142. }
  143. uint16_t idiagnl_msg_get_sport(struct idiagnl_msg *msg)
  144. {
  145. return msg->idiag_sport;
  146. }
  147. void idiagnl_msg_set_sport(struct idiagnl_msg *msg, uint16_t port)
  148. {
  149. msg->idiag_sport = port;
  150. }
  151. uint16_t idiagnl_msg_get_dport(struct idiagnl_msg *msg)
  152. {
  153. return msg->idiag_dport;
  154. }
  155. void idiagnl_msg_set_dport(struct idiagnl_msg *msg, uint16_t port)
  156. {
  157. msg->idiag_dport = port;
  158. }
  159. struct nl_addr *idiagnl_msg_get_src(const struct idiagnl_msg *msg)
  160. {
  161. return msg->idiag_src;
  162. }
  163. int idiagnl_msg_set_src(struct idiagnl_msg *msg, struct nl_addr *addr)
  164. {
  165. if (msg->idiag_src)
  166. nl_addr_put(msg->idiag_src);
  167. nl_addr_get(addr);
  168. msg->idiag_src = addr;
  169. return 0;
  170. }
  171. struct nl_addr *idiagnl_msg_get_dst(const struct idiagnl_msg *msg)
  172. {
  173. return msg->idiag_dst;
  174. }
  175. int idiagnl_msg_set_dst(struct idiagnl_msg *msg, struct nl_addr *addr)
  176. {
  177. if (msg->idiag_dst)
  178. nl_addr_put(msg->idiag_dst);
  179. nl_addr_get(addr);
  180. msg->idiag_dst = addr;
  181. return 0;
  182. }
  183. uint32_t idiagnl_msg_get_ifindex(const struct idiagnl_msg *msg)
  184. {
  185. return msg->idiag_ifindex;
  186. }
  187. void idiagnl_msg_set_ifindex(struct idiagnl_msg *msg, uint32_t ifindex)
  188. {
  189. msg->idiag_ifindex = ifindex;
  190. }
  191. uint32_t idiagnl_msg_get_expires(const struct idiagnl_msg *msg)
  192. {
  193. return msg->idiag_expires;
  194. }
  195. void idiagnl_msg_set_expires(struct idiagnl_msg *msg, uint32_t expires)
  196. {
  197. msg->idiag_expires = expires;
  198. }
  199. uint32_t idiagnl_msg_get_rqueue(const struct idiagnl_msg *msg)
  200. {
  201. return msg->idiag_rqueue;
  202. }
  203. void idiagnl_msg_set_rqueue(struct idiagnl_msg *msg, uint32_t rqueue)
  204. {
  205. msg->idiag_rqueue = rqueue;
  206. }
  207. uint32_t idiagnl_msg_get_wqueue(const struct idiagnl_msg *msg)
  208. {
  209. return msg->idiag_wqueue;
  210. }
  211. void idiagnl_msg_set_wqueue(struct idiagnl_msg *msg, uint32_t wqueue)
  212. {
  213. msg->idiag_wqueue = wqueue;
  214. }
  215. uint32_t idiagnl_msg_get_uid(const struct idiagnl_msg *msg)
  216. {
  217. return msg->idiag_uid;
  218. }
  219. void idiagnl_msg_set_uid(struct idiagnl_msg *msg, uint32_t uid)
  220. {
  221. msg->idiag_uid = uid;
  222. }
  223. uint32_t idiagnl_msg_get_inode(const struct idiagnl_msg *msg)
  224. {
  225. return msg->idiag_inode;
  226. }
  227. void idiagnl_msg_set_inode(struct idiagnl_msg *msg, uint32_t inode)
  228. {
  229. msg->idiag_inode = inode;
  230. }
  231. uint8_t idiagnl_msg_get_tos(const struct idiagnl_msg *msg)
  232. {
  233. return msg->idiag_tos;
  234. }
  235. void idiagnl_msg_set_tos(struct idiagnl_msg *msg, uint8_t tos)
  236. {
  237. msg->idiag_tos = tos;
  238. }
  239. uint8_t idiagnl_msg_get_tclass(const struct idiagnl_msg *msg)
  240. {
  241. return msg->idiag_tclass;
  242. }
  243. void idiagnl_msg_set_tclass(struct idiagnl_msg *msg, uint8_t tclass)
  244. {
  245. msg->idiag_tclass = tclass;
  246. }
  247. uint8_t idiagnl_msg_get_shutdown(const struct idiagnl_msg *msg)
  248. {
  249. return msg->idiag_shutdown;
  250. }
  251. void idiagnl_msg_set_shutdown(struct idiagnl_msg *msg, uint8_t shutdown)
  252. {
  253. msg->idiag_shutdown = shutdown;
  254. }
  255. char *idiagnl_msg_get_cong(const struct idiagnl_msg *msg)
  256. {
  257. return msg->idiag_cong;
  258. }
  259. void idiagnl_msg_set_cong(struct idiagnl_msg *msg, char *cong)
  260. {
  261. msg->idiag_cong = strdup(cong);
  262. }
  263. struct idiagnl_meminfo *idiagnl_msg_get_meminfo(const struct idiagnl_msg *msg)
  264. {
  265. return msg->idiag_meminfo;
  266. }
  267. void idiagnl_msg_set_meminfo(struct idiagnl_msg *msg, struct idiagnl_meminfo
  268. *minfo)
  269. {
  270. if (msg->idiag_meminfo)
  271. idiagnl_meminfo_put(msg->idiag_meminfo);
  272. idiagnl_meminfo_get(minfo);
  273. msg->idiag_meminfo = minfo;
  274. }
  275. struct idiagnl_vegasinfo *idiagnl_msg_get_vegasinfo(const struct idiagnl_msg *msg)
  276. {
  277. return msg->idiag_vegasinfo;
  278. }
  279. void idiagnl_msg_set_vegasinfo(struct idiagnl_msg *msg, struct idiagnl_vegasinfo
  280. *vinfo)
  281. {
  282. if (msg->idiag_vegasinfo)
  283. idiagnl_vegasinfo_put(msg->idiag_vegasinfo);
  284. idiagnl_vegasinfo_get(vinfo);
  285. msg->idiag_vegasinfo = vinfo;
  286. }
  287. struct tcp_info idiagnl_msg_get_tcpinfo(const struct idiagnl_msg *msg)
  288. {
  289. return msg->idiag_tcpinfo;
  290. }
  291. void idiagnl_msg_set_tcpinfo(struct idiagnl_msg *msg, struct tcp_info *tinfo)
  292. {
  293. memcpy(&msg->idiag_tcpinfo, tinfo, sizeof(struct tcp_info));
  294. }
  295. /** @} */
  296. static void idiag_msg_dump_line(struct nl_object *a, struct nl_dump_params *p)
  297. {
  298. struct idiagnl_msg *msg = (struct idiagnl_msg *) a;
  299. char buf[64] = { 0 };
  300. nl_dump_line(p, "family: %s ", nl_af2str(msg->idiag_family, buf, sizeof(buf)));
  301. nl_dump(p, "src: %s:%d ", nl_addr2str(msg->idiag_src, buf, sizeof(buf)),
  302. ntohs(msg->idiag_sport));
  303. nl_dump(p, "dst: %s:%d ", nl_addr2str(msg->idiag_dst, buf, sizeof(buf)),
  304. ntohs(msg->idiag_dport));
  305. nl_dump(p, "iif: %d ", msg->idiag_ifindex);
  306. nl_dump(p, "\n");
  307. }
  308. static void idiag_msg_dump_details(struct nl_object *a, struct nl_dump_params *p)
  309. {
  310. struct idiagnl_msg *msg = (struct idiagnl_msg *) a;
  311. char buf[64], buf2[64];
  312. nl_dump(p, "\nfamily: %s\n", nl_af2str(msg->idiag_family, buf, sizeof(buf)));
  313. nl_dump(p, "state: %s\n",
  314. idiagnl_state2str(msg->idiag_state, buf, sizeof(buf)));
  315. nl_dump(p, "timer (%s, %s, retransmits: %d)\n",
  316. idiagnl_timer2str(msg->idiag_timer, buf, sizeof(buf)),
  317. nl_msec2str(msg->idiag_expires, buf2, sizeof(buf2)),
  318. msg->idiag_retrans);
  319. nl_dump(p, "source: %s:%d\n", nl_addr2str(msg->idiag_src, buf, sizeof(buf)),
  320. ntohs(msg->idiag_sport));
  321. nl_dump(p, "destination: %s:%d\n", nl_addr2str(msg->idiag_dst, buf, sizeof(buf)),
  322. ntohs(msg->idiag_dport));
  323. nl_dump(p, "ifindex: %d\n", msg->idiag_ifindex);
  324. nl_dump(p, "rqueue: %-6d wqueue: %-6d\n", msg->idiag_rqueue, msg->idiag_wqueue);
  325. nl_dump(p, "uid %d\n", msg->idiag_uid);
  326. nl_dump(p, "inode %d\n", msg->idiag_inode);
  327. if (msg->idiag_shutdown) {
  328. nl_dump(p, "socket shutdown: %s\n",
  329. idiagnl_shutdown2str(msg->idiag_shutdown,
  330. buf, sizeof(buf)));
  331. }
  332. nl_dump(p, "tos: 0x%x\n", msg->idiag_tos);
  333. nl_dump(p, "traffic class: %d\n", msg->idiag_tclass);
  334. nl_dump(p, "congestion algorithm: %s\n", msg->idiag_cong);
  335. }
  336. static void idiag_msg_dump_stats(struct nl_object *obj, struct nl_dump_params *p)
  337. {
  338. struct idiagnl_msg *msg = (struct idiagnl_msg *) obj;
  339. char buf[64];
  340. idiag_msg_dump_details(obj, p);
  341. nl_dump(p, "tcp info: [\n");
  342. nl_dump(p, "\tsocket state: %s\n",
  343. idiagnl_state2str(msg->idiag_tcpinfo.tcpi_state,
  344. buf, sizeof(buf)));
  345. nl_dump(p, "\ttcp state: %s\n",
  346. idiagnl_tcpstate2str(msg->idiag_tcpinfo.tcpi_ca_state,
  347. buf, sizeof(buf)));
  348. nl_dump(p, "\tretransmits: %d\n",
  349. msg->idiag_tcpinfo.tcpi_retransmits);
  350. nl_dump(p, "\tprobes: %d\n",
  351. msg->idiag_tcpinfo.tcpi_probes);
  352. nl_dump(p, "\tbackoff: %d\n",
  353. msg->idiag_tcpinfo.tcpi_backoff);
  354. nl_dump(p, "\toptions: %s\n",
  355. idiagnl_tcpopts2str(msg->idiag_tcpinfo.tcpi_options,
  356. buf, sizeof(buf)));
  357. nl_dump(p, "\tsnd_wscale: %d\n", msg->idiag_tcpinfo.tcpi_snd_wscale);
  358. nl_dump(p, "\trcv_wscale: %d\n", msg->idiag_tcpinfo.tcpi_rcv_wscale);
  359. nl_dump(p, "\trto: %d\n", msg->idiag_tcpinfo.tcpi_rto);
  360. nl_dump(p, "\tato: %d\n", msg->idiag_tcpinfo.tcpi_ato);
  361. nl_dump(p, "\tsnd_mss: %s\n", nl_size2str(msg->idiag_tcpinfo.tcpi_snd_mss,
  362. buf, sizeof(buf)));
  363. nl_dump(p, "\trcv_mss: %s\n", nl_size2str(msg->idiag_tcpinfo.tcpi_rcv_mss,
  364. buf, sizeof(buf)));
  365. nl_dump(p, "\tunacked: %d\n", msg->idiag_tcpinfo.tcpi_unacked);
  366. nl_dump(p, "\tsacked: %d\n", msg->idiag_tcpinfo.tcpi_sacked);
  367. nl_dump(p, "\tlost: %d\n", msg->idiag_tcpinfo.tcpi_lost);
  368. nl_dump(p, "\tretransmit segments: %d\n",
  369. msg->idiag_tcpinfo.tcpi_retrans);
  370. nl_dump(p, "\tfackets: %d\n",
  371. msg->idiag_tcpinfo.tcpi_fackets);
  372. nl_dump(p, "\tlast data sent: %s\n",
  373. nl_msec2str(msg->idiag_tcpinfo.tcpi_last_data_sent, buf,
  374. sizeof(buf)));
  375. nl_dump(p, "\tlast ack sent: %s\n",
  376. nl_msec2str(msg->idiag_tcpinfo.tcpi_last_ack_sent, buf, sizeof(buf)));
  377. nl_dump(p, "\tlast data recv: %s\n",
  378. nl_msec2str(msg->idiag_tcpinfo.tcpi_last_data_recv, buf,
  379. sizeof(buf)));
  380. nl_dump(p, "\tlast ack recv: %s\n",
  381. nl_msec2str(msg->idiag_tcpinfo.tcpi_last_ack_recv, buf,
  382. sizeof(buf)));
  383. nl_dump(p, "\tpath mtu: %s\n",
  384. nl_size2str(msg->idiag_tcpinfo.tcpi_pmtu, buf,
  385. sizeof(buf)));
  386. nl_dump(p, "\trcv ss threshold: %d\n",
  387. msg->idiag_tcpinfo.tcpi_rcv_ssthresh);
  388. nl_dump(p, "\tsmoothed round trip time: %d\n",
  389. msg->idiag_tcpinfo.tcpi_rtt);
  390. nl_dump(p, "\tround trip time variation: %d\n",
  391. msg->idiag_tcpinfo.tcpi_rttvar);
  392. nl_dump(p, "\tsnd ss threshold: %s\n",
  393. nl_size2str(msg->idiag_tcpinfo.tcpi_snd_ssthresh, buf,
  394. sizeof(buf)));
  395. nl_dump(p, "\tsend congestion window: %d\n",
  396. msg->idiag_tcpinfo.tcpi_snd_cwnd);
  397. nl_dump(p, "\tadvertised mss: %s\n",
  398. nl_size2str(msg->idiag_tcpinfo.tcpi_advmss, buf,
  399. sizeof(buf)));
  400. nl_dump(p, "\treordering: %d\n",
  401. msg->idiag_tcpinfo.tcpi_reordering);
  402. nl_dump(p, "\trcv rround trip time: %d\n",
  403. msg->idiag_tcpinfo.tcpi_rcv_rtt);
  404. nl_dump(p, "\treceive queue space: %s\n",
  405. nl_size2str(msg->idiag_tcpinfo.tcpi_rcv_space, buf,
  406. sizeof(buf)));
  407. nl_dump(p, "\ttotal retransmits: %d\n",
  408. msg->idiag_tcpinfo.tcpi_total_retrans);
  409. nl_dump(p, "]\n");
  410. if (msg->idiag_meminfo) {
  411. nl_dump(p, "meminfo: [\n");
  412. nl_dump(p, "\trmem: %s\n",
  413. nl_size2str(msg->idiag_meminfo->idiag_rmem,
  414. buf,
  415. sizeof(buf)));
  416. nl_dump(p, "\twmem: %s\n",
  417. nl_size2str(msg->idiag_meminfo->idiag_wmem,
  418. buf,
  419. sizeof(buf)));
  420. nl_dump(p, "\tfmem: %s\n",
  421. nl_size2str(msg->idiag_meminfo->idiag_fmem,
  422. buf,
  423. sizeof(buf)));
  424. nl_dump(p, "\ttmem: %s\n",
  425. nl_size2str(msg->idiag_meminfo->idiag_tmem,
  426. buf,
  427. sizeof(buf)));
  428. nl_dump(p, "]\n");
  429. }
  430. if (msg->idiag_vegasinfo) {
  431. nl_dump(p, "vegasinfo: [\n");
  432. nl_dump(p, "\tvegas enabled: %d\n",
  433. msg->idiag_vegasinfo->tcpv_enabled);
  434. if (msg->idiag_vegasinfo->tcpv_enabled) {
  435. nl_dump(p, "\trtt cnt: %d",
  436. msg->idiag_vegasinfo->tcpv_rttcnt);
  437. nl_dump(p, "\trtt (propagation delay): %d",
  438. msg->idiag_vegasinfo->tcpv_rtt);
  439. nl_dump(p, "\tmin rtt: %d",
  440. msg->idiag_vegasinfo->tcpv_minrtt);
  441. }
  442. nl_dump(p, "]\n");
  443. }
  444. nl_dump(p, "skmeminfo: [\n");
  445. nl_dump(p, "\trmem alloc: %d\n",
  446. msg->idiag_skmeminfo[IDIAG_SK_MEMINFO_RMEM_ALLOC]);
  447. nl_dump(p, "\trcv buf: %s\n",
  448. nl_size2str(msg->idiag_skmeminfo[IDIAG_SK_MEMINFO_RCVBUF],
  449. buf, sizeof(buf)));
  450. nl_dump(p, "\twmem alloc: %d\n",
  451. msg->idiag_skmeminfo[IDIAG_SK_MEMINFO_WMEM_ALLOC]);
  452. nl_dump(p, "\tsnd buf: %s\n",
  453. nl_size2str(msg->idiag_skmeminfo[IDIAG_SK_MEMINFO_SNDBUF],
  454. buf, sizeof(buf)));
  455. nl_dump(p, "\tfwd alloc: %d\n",
  456. msg->idiag_skmeminfo[IDIAG_SK_MEMINFO_FWD_ALLOC]);
  457. nl_dump(p, "\twmem queued: %s\n",
  458. nl_size2str(msg->idiag_skmeminfo[IDIAG_SK_MEMINFO_WMEM_QUEUED],
  459. buf, sizeof(buf)));
  460. nl_dump(p, "\topt mem: %d\n",
  461. msg->idiag_skmeminfo[IDIAG_SK_MEMINFO_OPTMEM]);
  462. nl_dump(p, "\tbacklog: %d\n",
  463. msg->idiag_skmeminfo[IDIAG_SK_MEMINFO_BACKLOG]);
  464. nl_dump(p, "]\n\n");
  465. }
  466. static void idiagnl_msg_free(struct nl_object *a)
  467. {
  468. struct idiagnl_msg *msg = (struct idiagnl_msg *) a;
  469. if (a == NULL)
  470. return;
  471. free(msg->idiag_cong);
  472. nl_addr_put(msg->idiag_src);
  473. nl_addr_put(msg->idiag_dst);
  474. idiagnl_meminfo_put(msg->idiag_meminfo);
  475. idiagnl_vegasinfo_put(msg->idiag_vegasinfo);
  476. }
  477. static int idiagnl_msg_clone(struct nl_object *_dst, struct nl_object *_src)
  478. {
  479. struct idiagnl_msg *dst = (struct idiagnl_msg *) _dst;
  480. struct idiagnl_msg *src = (struct idiagnl_msg *) _src;
  481. if (src->idiag_src)
  482. if (!(dst->idiag_src = nl_addr_clone(src->idiag_src)))
  483. return -NLE_NOMEM;
  484. if (src->idiag_dst)
  485. if (!(dst->idiag_dst = nl_addr_clone(src->idiag_dst)))
  486. return -NLE_NOMEM;
  487. return 0;
  488. }
  489. static struct nla_policy ext_policy[IDIAG_ATTR_MAX] = {
  490. [IDIAG_ATTR_MEMINFO] = { .minlen = sizeof(struct inet_diag_meminfo) },
  491. [IDIAG_ATTR_INFO] = { .minlen = sizeof(struct tcp_info) },
  492. [IDIAG_ATTR_VEGASINFO] = { .minlen = sizeof(struct tcpvegas_info) },
  493. [IDIAG_ATTR_CONG] = { .type = NLA_STRING },
  494. [IDIAG_ATTR_TOS] = { .type = NLA_U8 },
  495. [IDIAG_ATTR_TCLASS] = { .type = NLA_U8 },
  496. [IDIAG_ATTR_SKMEMINFO] = { .minlen = (sizeof(uint32_t) * IDIAG_SK_MEMINFO_VARS) },
  497. [IDIAG_ATTR_SHUTDOWN] = { .type = NLA_U8 },
  498. };
  499. int idiagnl_msg_parse(struct nlmsghdr *nlh, struct idiagnl_msg **result)
  500. {
  501. struct idiagnl_msg *msg = NULL;
  502. struct inet_diag_msg *raw_msg = NULL;
  503. struct nl_addr *src = NULL, *dst = NULL;
  504. struct nlattr *tb[IDIAG_ATTR_MAX];
  505. int err = 0;
  506. msg = idiagnl_msg_alloc();
  507. if (!msg)
  508. goto errout_nomem;
  509. err = nlmsg_parse(nlh, sizeof(struct inet_diag_msg), tb, IDIAG_ATTR_MAX,
  510. ext_policy);
  511. if (err < 0)
  512. goto errout;
  513. raw_msg = nlmsg_data(nlh);
  514. msg->idiag_family = raw_msg->idiag_family;
  515. msg->idiag_state = raw_msg->idiag_state;
  516. msg->idiag_timer = raw_msg->idiag_timer;
  517. msg->idiag_retrans = raw_msg->idiag_retrans;
  518. msg->idiag_expires = raw_msg->idiag_expires;
  519. msg->idiag_rqueue = raw_msg->idiag_rqueue;
  520. msg->idiag_wqueue = raw_msg->idiag_wqueue;
  521. msg->idiag_uid = raw_msg->idiag_uid;
  522. msg->idiag_inode = raw_msg->idiag_inode;
  523. msg->idiag_sport = raw_msg->id.idiag_sport;
  524. msg->idiag_dport = raw_msg->id.idiag_dport;
  525. msg->idiag_ifindex = raw_msg->id.idiag_if;
  526. dst = nl_addr_build(raw_msg->idiag_family, raw_msg->id.idiag_dst,
  527. sizeof(raw_msg->id.idiag_dst));
  528. if (!dst)
  529. goto errout_nomem;
  530. err = idiagnl_msg_set_dst(msg, dst);
  531. if (err < 0)
  532. goto errout;
  533. nl_addr_put(dst);
  534. src = nl_addr_build(raw_msg->idiag_family, raw_msg->id.idiag_src,
  535. sizeof(raw_msg->id.idiag_src));
  536. if (!src)
  537. goto errout_nomem;
  538. err = idiagnl_msg_set_src(msg, src);
  539. if (err < 0)
  540. goto errout;
  541. nl_addr_put(src);
  542. if (tb[IDIAG_ATTR_TOS])
  543. msg->idiag_tos = nla_get_u8(tb[IDIAG_ATTR_TOS]);
  544. if (tb[IDIAG_ATTR_TCLASS])
  545. msg->idiag_tclass = nla_get_u8(tb[IDIAG_ATTR_TCLASS]);
  546. if (tb[IDIAG_ATTR_SHUTDOWN])
  547. msg->idiag_shutdown = nla_get_u8(tb[IDIAG_ATTR_SHUTDOWN]);
  548. if (tb[IDIAG_ATTR_CONG])
  549. msg->idiag_cong = nla_strdup(tb[IDIAG_ATTR_CONG]);
  550. if (tb[IDIAG_ATTR_INFO])
  551. nla_memcpy(&msg->idiag_tcpinfo, tb[IDIAG_ATTR_INFO],
  552. sizeof(msg->idiag_tcpinfo));
  553. if (tb[IDIAG_ATTR_MEMINFO]) {
  554. struct idiagnl_meminfo *minfo = idiagnl_meminfo_alloc();
  555. struct inet_diag_meminfo *raw_minfo = NULL;
  556. if (!minfo)
  557. goto errout_nomem;
  558. raw_minfo = (struct inet_diag_meminfo *)
  559. nla_data(tb[IDIAG_ATTR_MEMINFO]);
  560. idiagnl_meminfo_set_rmem(minfo, raw_minfo->idiag_rmem);
  561. idiagnl_meminfo_set_wmem(minfo, raw_minfo->idiag_wmem);
  562. idiagnl_meminfo_set_fmem(minfo, raw_minfo->idiag_fmem);
  563. idiagnl_meminfo_set_tmem(minfo, raw_minfo->idiag_tmem);
  564. msg->idiag_meminfo = minfo;
  565. }
  566. if (tb[IDIAG_ATTR_VEGASINFO]) {
  567. struct idiagnl_vegasinfo *vinfo = idiagnl_vegasinfo_alloc();
  568. struct tcpvegas_info *raw_vinfo = NULL;
  569. if (!vinfo)
  570. goto errout_nomem;
  571. raw_vinfo = (struct tcpvegas_info *)
  572. nla_data(tb[IDIAG_ATTR_VEGASINFO]);
  573. idiagnl_vegasinfo_set_enabled(vinfo, raw_vinfo->tcpv_enabled);
  574. idiagnl_vegasinfo_set_rttcnt(vinfo, raw_vinfo->tcpv_rttcnt);
  575. idiagnl_vegasinfo_set_rtt(vinfo, raw_vinfo->tcpv_rtt);
  576. idiagnl_vegasinfo_set_minrtt(vinfo, raw_vinfo->tcpv_minrtt);
  577. msg->idiag_vegasinfo = vinfo;
  578. }
  579. if (tb[IDIAG_ATTR_SKMEMINFO])
  580. nla_memcpy(&msg->idiag_skmeminfo, tb[IDIAG_ATTR_SKMEMINFO],
  581. sizeof(msg->idiag_skmeminfo));
  582. *result = msg;
  583. return 0;
  584. errout:
  585. idiagnl_msg_put(msg);
  586. return err;
  587. errout_nomem:
  588. err = -NLE_NOMEM;
  589. goto errout;
  590. }
  591. /** @cond SKIP */
  592. struct nl_object_ops idiagnl_msg_obj_ops = {
  593. .oo_name = "idiag/idiag_msg",
  594. .oo_size = sizeof(struct idiagnl_msg),
  595. .oo_free_data = idiagnl_msg_free,
  596. .oo_clone = idiagnl_msg_clone,
  597. .oo_dump = {
  598. [NL_DUMP_LINE] = idiag_msg_dump_line,
  599. [NL_DUMP_DETAILS] = idiag_msg_dump_details,
  600. [NL_DUMP_STATS] = idiag_msg_dump_stats,
  601. },
  602. .oo_attrs2str = idiagnl_attrs2str,
  603. .oo_id_attrs = (IDIAG_ATTR_INFO)
  604. };
  605. /** @endcond */
  606. /** @} */