FrameDecider.c 28 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798
  1. /**
  2. * @file FrameDecider.c
  3. * @author Ambroz Bizjak <ambrop7@gmail.com>
  4. *
  5. * @section LICENSE
  6. *
  7. * Redistribution and use in source and binary forms, with or without
  8. * modification, are permitted provided that the following conditions are met:
  9. * 1. Redistributions of source code must retain the above copyright
  10. * notice, this list of conditions and the following disclaimer.
  11. * 2. Redistributions in binary form must reproduce the above copyright
  12. * notice, this list of conditions and the following disclaimer in the
  13. * documentation and/or other materials provided with the distribution.
  14. * 3. Neither the name of the author nor the
  15. * names of its contributors may be used to endorse or promote products
  16. * derived from this software without specific prior written permission.
  17. *
  18. * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
  19. * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
  20. * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
  21. * DISCLAIMED. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY
  22. * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
  23. * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
  24. * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
  25. * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  26. * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
  27. * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  28. */
  29. #include <string.h>
  30. #include <stddef.h>
  31. #include <inttypes.h>
  32. #include <misc/debug.h>
  33. #include <misc/offset.h>
  34. #include <misc/balloc.h>
  35. #include <misc/ethernet_proto.h>
  36. #include <misc/ipv4_proto.h>
  37. #include <misc/igmp_proto.h>
  38. #include <misc/byteorder.h>
  39. #include <misc/compare.h>
  40. #include <client/FrameDecider.h>
  41. #include <generated/blog_channel_FrameDecider.h>
  42. #define DECIDE_STATE_NONE 1
  43. #define DECIDE_STATE_UNICAST 2
  44. #define DECIDE_STATE_FLOOD 3
  45. #define DECIDE_STATE_MULTICAST 4
  46. #define PeerLog(_o, ...) BLog_LogViaFunc((_o)->logfunc, (_o)->user, BLOG_CURRENT_CHANNEL, __VA_ARGS__)
  47. static int compare_macs (const uint8_t *mac1, const uint8_t *mac2)
  48. {
  49. int c = memcmp(mac1, mac2, 6);
  50. return B_COMPARE(c, 0);
  51. }
  52. #include "FrameDecider_macs_tree.h"
  53. #include <structure/CAvl_impl.h>
  54. #include "FrameDecider_groups_tree.h"
  55. #include <structure/CAvl_impl.h>
  56. static int uint32_comparator (void *user, uint32_t *v1, uint32_t *v2)
  57. {
  58. if (*v1 < *v2) {
  59. return -1;
  60. }
  61. if (*v1 > *v2) {
  62. return 1;
  63. }
  64. return 0;
  65. }
  66. static void add_mac_to_peer (FrameDeciderPeer *o, uint8_t *mac)
  67. {
  68. FrameDecider *d = o->d;
  69. // locate entry in tree
  70. FDMacsTreeRef ref = FDMacsTree_LookupExact(&d->macs_tree, 0, mac);
  71. if (FDMacsTreeIsValidRef(ref)) {
  72. struct _FrameDecider_mac_entry *entry = ref.ptr;
  73. if (entry->peer == o) {
  74. // this is our MAC; only move it to the end of the used list
  75. LinkedList2_Remove(&o->mac_entries_used, &entry->list_node);
  76. LinkedList2_Append(&o->mac_entries_used, &entry->list_node);
  77. return;
  78. }
  79. // some other peer has that MAC; disassociate it
  80. FDMacsTree_Remove(&d->macs_tree, 0, FDMacsTreeDeref(0, entry));
  81. LinkedList2_Remove(&entry->peer->mac_entries_used, &entry->list_node);
  82. LinkedList2_Append(&entry->peer->mac_entries_free, &entry->list_node);
  83. }
  84. // aquire MAC address entry, if there are no free ones reuse the oldest used one
  85. LinkedList2Node *list_node;
  86. struct _FrameDecider_mac_entry *entry;
  87. if (list_node = LinkedList2_GetFirst(&o->mac_entries_free)) {
  88. entry = UPPER_OBJECT(list_node, struct _FrameDecider_mac_entry, list_node);
  89. ASSERT(entry->peer == o)
  90. // remove from free
  91. LinkedList2_Remove(&o->mac_entries_free, &entry->list_node);
  92. } else {
  93. list_node = LinkedList2_GetFirst(&o->mac_entries_used);
  94. ASSERT(list_node)
  95. entry = UPPER_OBJECT(list_node, struct _FrameDecider_mac_entry, list_node);
  96. ASSERT(entry->peer == o)
  97. // remove from used
  98. FDMacsTree_Remove(&d->macs_tree, 0, FDMacsTreeDeref(0, entry));
  99. LinkedList2_Remove(&o->mac_entries_used, &entry->list_node);
  100. }
  101. PeerLog(o, BLOG_INFO, "adding MAC %02"PRIx8":%02"PRIx8":%02"PRIx8":%02"PRIx8":%02"PRIx8":%02"PRIx8"", mac[0], mac[1], mac[2], mac[3], mac[4], mac[5]);
  102. // set MAC in entry
  103. memcpy(entry->mac, mac, sizeof(entry->mac));
  104. // add to used
  105. LinkedList2_Append(&o->mac_entries_used, &entry->list_node);
  106. int res = FDMacsTree_Insert(&d->macs_tree, 0, FDMacsTreeDeref(0, entry), NULL);
  107. ASSERT(res)
  108. }
  109. static uint32_t compute_sig_for_group (uint32_t group)
  110. {
  111. return hton32(ntoh32(group)&0x7FFFFF);
  112. }
  113. static uint32_t compute_sig_for_mac (uint8_t *mac)
  114. {
  115. uint32_t sig;
  116. memcpy(&sig, mac + 2, 4);
  117. sig = hton32(ntoh32(sig)&0x7FFFFF);
  118. return sig;
  119. }
  120. static void add_to_multicast (FrameDecider *d, struct _FrameDecider_group_entry *group_entry)
  121. {
  122. // compute sig
  123. uint32_t sig = compute_sig_for_group(group_entry->group);
  124. BAVLNode *node;
  125. if (node = BAVL_LookupExact(&d->multicast_tree, &sig)) {
  126. // use existing master
  127. struct _FrameDecider_group_entry *master = UPPER_OBJECT(node, struct _FrameDecider_group_entry, master.tree_node);
  128. ASSERT(master->is_master)
  129. // set not master
  130. group_entry->is_master = 0;
  131. // insert to list
  132. LinkedList3Node_InitAfter(&group_entry->sig_list_node, &master->sig_list_node);
  133. } else {
  134. // make this entry master
  135. // set master
  136. group_entry->is_master = 1;
  137. // set sig
  138. group_entry->master.sig = sig;
  139. // insert to multicast tree
  140. ASSERT_EXECUTE(BAVL_Insert(&d->multicast_tree, &group_entry->master.tree_node, NULL))
  141. // init list node
  142. LinkedList3Node_InitLonely(&group_entry->sig_list_node);
  143. }
  144. }
  145. static void remove_from_multicast (FrameDecider *d, struct _FrameDecider_group_entry *group_entry)
  146. {
  147. // compute sig
  148. uint32_t sig = compute_sig_for_group(group_entry->group);
  149. if (group_entry->is_master) {
  150. // remove master from multicast tree
  151. BAVL_Remove(&d->multicast_tree, &group_entry->master.tree_node);
  152. if (!LinkedList3Node_IsLonely(&group_entry->sig_list_node)) {
  153. // at least one more group entry for this sig; make another entry the master
  154. // get an entry
  155. LinkedList3Node *list_node = LinkedList3Node_NextOrPrev(&group_entry->sig_list_node);
  156. struct _FrameDecider_group_entry *newmaster = UPPER_OBJECT(list_node, struct _FrameDecider_group_entry, sig_list_node);
  157. ASSERT(!newmaster->is_master)
  158. // set master
  159. newmaster->is_master = 1;
  160. // set sig
  161. newmaster->master.sig = sig;
  162. // insert to multicast tree
  163. ASSERT_EXECUTE(BAVL_Insert(&d->multicast_tree, &newmaster->master.tree_node, NULL))
  164. }
  165. }
  166. // free linked list node
  167. LinkedList3Node_Free(&group_entry->sig_list_node);
  168. }
  169. static void add_group_to_peer (FrameDeciderPeer *o, uint32_t group)
  170. {
  171. FrameDecider *d = o->d;
  172. struct _FrameDecider_group_entry *group_entry;
  173. // lookup if the peer already has that group
  174. FDGroupsTreeRef ref = FDGroupsTree_LookupExact(&o->groups_tree, 0, group);
  175. if (FDGroupsTreeIsValidRef(ref)) {
  176. group_entry = ref.ptr;
  177. // move to end of used list
  178. LinkedList2_Remove(&o->group_entries_used, &group_entry->list_node);
  179. LinkedList2_Append(&o->group_entries_used, &group_entry->list_node);
  180. } else {
  181. PeerLog(o, BLOG_INFO, "joined group %"PRIu8".%"PRIu8".%"PRIu8".%"PRIu8"",
  182. ((uint8_t *)&group)[0], ((uint8_t *)&group)[1], ((uint8_t *)&group)[2], ((uint8_t *)&group)[3]
  183. );
  184. // aquire group entry, if there are no free ones reuse the earliest used one
  185. LinkedList2Node *node;
  186. if (node = LinkedList2_GetFirst(&o->group_entries_free)) {
  187. group_entry = UPPER_OBJECT(node, struct _FrameDecider_group_entry, list_node);
  188. // remove from free list
  189. LinkedList2_Remove(&o->group_entries_free, &group_entry->list_node);
  190. } else {
  191. node = LinkedList2_GetFirst(&o->group_entries_used);
  192. ASSERT(node)
  193. group_entry = UPPER_OBJECT(node, struct _FrameDecider_group_entry, list_node);
  194. // remove from multicast
  195. remove_from_multicast(d, group_entry);
  196. // remove from peer's groups tree
  197. FDGroupsTree_Remove(&o->groups_tree, 0, FDGroupsTreeDeref(0, group_entry));
  198. // remove from used list
  199. LinkedList2_Remove(&o->group_entries_used, &group_entry->list_node);
  200. }
  201. // add entry to used list
  202. LinkedList2_Append(&o->group_entries_used, &group_entry->list_node);
  203. // set group address
  204. group_entry->group = group;
  205. // insert to peer's groups tree
  206. int res = FDGroupsTree_Insert(&o->groups_tree, 0, FDGroupsTreeDeref(0, group_entry), NULL);
  207. ASSERT(res)
  208. // add to multicast
  209. add_to_multicast(d, group_entry);
  210. }
  211. // set timer
  212. group_entry->timer_endtime = btime_gettime() + d->igmp_group_membership_interval;
  213. BReactor_SetTimerAbsolute(d->reactor, &group_entry->timer, group_entry->timer_endtime);
  214. }
  215. static void remove_group_entry (struct _FrameDecider_group_entry *group_entry)
  216. {
  217. FrameDeciderPeer *peer = group_entry->peer;
  218. FrameDecider *d = peer->d;
  219. uint32_t group = group_entry->group;
  220. PeerLog(peer, BLOG_INFO, "left group %"PRIu8".%"PRIu8".%"PRIu8".%"PRIu8"",
  221. ((uint8_t *)&group)[0], ((uint8_t *)&group)[1], ((uint8_t *)&group)[2], ((uint8_t *)&group)[3]
  222. );
  223. // remove from multicast
  224. remove_from_multicast(d, group_entry);
  225. // remove from peer's groups tree
  226. FDGroupsTree_Remove(&peer->groups_tree, 0, FDGroupsTreeDeref(0, group_entry));
  227. // remove from used list
  228. LinkedList2_Remove(&peer->group_entries_used, &group_entry->list_node);
  229. // add to free list
  230. LinkedList2_Append(&peer->group_entries_free, &group_entry->list_node);
  231. // stop timer
  232. BReactor_RemoveTimer(d->reactor, &group_entry->timer);
  233. }
  234. static void lower_group_timers_to_lmqt (FrameDecider *d, uint32_t group)
  235. {
  236. // have to lower all the group timers of this group down to LMQT
  237. // compute sig
  238. uint32_t sig = compute_sig_for_group(group);
  239. // look up the sig in multicast tree
  240. BAVLNode *tree_node;
  241. if (!(tree_node = BAVL_LookupExact(&d->multicast_tree, &sig))) {
  242. return;
  243. }
  244. struct _FrameDecider_group_entry *master = UPPER_OBJECT(tree_node, struct _FrameDecider_group_entry, master.tree_node);
  245. ASSERT(master->is_master)
  246. // iterate all group entries with this sig
  247. LinkedList3Iterator it;
  248. LinkedList3Iterator_Init(&it, LinkedList3Node_First(&master->sig_list_node), 1);
  249. LinkedList3Node *sig_list_node;
  250. while (sig_list_node = LinkedList3Iterator_Next(&it)) {
  251. struct _FrameDecider_group_entry *group_entry = UPPER_OBJECT(sig_list_node, struct _FrameDecider_group_entry, sig_list_node);
  252. // skip wrong groups
  253. if (group_entry->group != group) {
  254. continue;
  255. }
  256. // lower timer down to LMQT
  257. btime_t now = btime_gettime();
  258. if (group_entry->timer_endtime > now + d->igmp_last_member_query_time) {
  259. group_entry->timer_endtime = now + d->igmp_last_member_query_time;
  260. BReactor_SetTimerAbsolute(d->reactor, &group_entry->timer, group_entry->timer_endtime);
  261. }
  262. }
  263. }
  264. static void group_entry_timer_handler (struct _FrameDecider_group_entry *group_entry)
  265. {
  266. DebugObject_Access(&group_entry->peer->d_obj);
  267. remove_group_entry(group_entry);
  268. }
  269. void FrameDecider_Init (FrameDecider *o, int max_peer_macs, int max_peer_groups, btime_t igmp_group_membership_interval, btime_t igmp_last_member_query_time, BReactor *reactor)
  270. {
  271. ASSERT(max_peer_macs > 0)
  272. ASSERT(max_peer_groups > 0)
  273. // init arguments
  274. o->max_peer_macs = max_peer_macs;
  275. o->max_peer_groups = max_peer_groups;
  276. o->igmp_group_membership_interval = igmp_group_membership_interval;
  277. o->igmp_last_member_query_time = igmp_last_member_query_time;
  278. o->reactor = reactor;
  279. // init peers list
  280. LinkedList2_Init(&o->peers_list);
  281. // init MAC tree
  282. FDMacsTree_Init(&o->macs_tree);
  283. // init multicast tree
  284. BAVL_Init(&o->multicast_tree, OFFSET_DIFF(struct _FrameDecider_group_entry, master.sig, master.tree_node), (BAVL_comparator)uint32_comparator, NULL);
  285. // init decide state
  286. o->decide_state = DECIDE_STATE_NONE;
  287. DebugObject_Init(&o->d_obj);
  288. }
  289. void FrameDecider_Free (FrameDecider *o)
  290. {
  291. ASSERT(BAVL_IsEmpty(&o->multicast_tree))
  292. ASSERT(FDMacsTree_IsEmpty(&o->macs_tree))
  293. ASSERT(LinkedList2_IsEmpty(&o->peers_list))
  294. DebugObject_Free(&o->d_obj);
  295. }
  296. void FrameDecider_AnalyzeAndDecide (FrameDecider *o, const uint8_t *frame, int frame_len)
  297. {
  298. ASSERT(frame_len >= 0)
  299. DebugObject_Access(&o->d_obj);
  300. // reset decide state
  301. switch (o->decide_state) {
  302. case DECIDE_STATE_NONE:
  303. break;
  304. case DECIDE_STATE_UNICAST:
  305. break;
  306. case DECIDE_STATE_FLOOD:
  307. LinkedList2Iterator_Free(&o->decide_flood_it);
  308. break;
  309. case DECIDE_STATE_MULTICAST:
  310. LinkedList3Iterator_Free(&o->decide_multicast_it);
  311. return;
  312. default:
  313. ASSERT(0);
  314. }
  315. o->decide_state = DECIDE_STATE_NONE;
  316. // analyze frame
  317. const uint8_t *pos = frame;
  318. int len = frame_len;
  319. if (len < sizeof(struct ethernet_header)) {
  320. return;
  321. }
  322. struct ethernet_header *eh = (struct ethernet_header *)pos;
  323. pos += sizeof(struct ethernet_header);
  324. len -= sizeof(struct ethernet_header);
  325. int is_igmp = 0;
  326. switch (ntoh16(eh->type)) {
  327. case ETHERTYPE_IPV4: {
  328. // check IPv4 header
  329. struct ipv4_header *ipv4_header;
  330. if (!ipv4_check((uint8_t *)pos, len, &ipv4_header, (uint8_t **)&pos, &len)) {
  331. BLog(BLOG_INFO, "decide: wrong IP packet");
  332. goto out;
  333. }
  334. // check if it's IGMP
  335. if (ntoh8(ipv4_header->protocol) != IPV4_PROTOCOL_IGMP) {
  336. goto out;
  337. }
  338. // remember that it's IGMP; we have to flood IGMP frames
  339. is_igmp = 1;
  340. // check IGMP header
  341. if (len < sizeof(struct igmp_base)) {
  342. BLog(BLOG_INFO, "decide: IGMP: short packet");
  343. goto out;
  344. }
  345. struct igmp_base *igmp_base = (struct igmp_base *)pos;
  346. pos += sizeof(struct igmp_base);
  347. len -= sizeof(struct igmp_base);
  348. switch (ntoh8(igmp_base->type)) {
  349. case IGMP_TYPE_MEMBERSHIP_QUERY: {
  350. if (len == sizeof(struct igmp_v2_extra) && ntoh8(igmp_base->max_resp_code) != 0) {
  351. // V2 query
  352. struct igmp_v2_extra *query = (struct igmp_v2_extra *)pos;
  353. pos += sizeof(struct igmp_v2_extra);
  354. len -= sizeof(struct igmp_v2_extra);
  355. if (ntoh32(query->group) != 0) {
  356. // got a Group-Specific Query, lower group timers to LMQT
  357. lower_group_timers_to_lmqt(o, query->group);
  358. }
  359. }
  360. else if (len >= sizeof(struct igmp_v3_query_extra)) {
  361. // V3 query
  362. struct igmp_v3_query_extra *query = (struct igmp_v3_query_extra *)pos;
  363. pos += sizeof(struct igmp_v3_query_extra);
  364. len -= sizeof(struct igmp_v3_query_extra);
  365. // iterate sources
  366. uint16_t num_sources = ntoh16(query->number_of_sources);
  367. int i;
  368. for (i = 0; i < num_sources; i++) {
  369. // check source
  370. if (len < sizeof(struct igmp_source)) {
  371. BLog(BLOG_NOTICE, "decide: IGMP: short source");
  372. goto out;
  373. }
  374. pos += sizeof(struct igmp_source);
  375. len -= sizeof(struct igmp_source);
  376. }
  377. if (ntoh32(query->group) != 0 && num_sources == 0) {
  378. // got a Group-Specific Query, lower group timers to LMQT
  379. lower_group_timers_to_lmqt(o, query->group);
  380. }
  381. }
  382. } break;
  383. }
  384. } break;
  385. }
  386. out:;
  387. const uint8_t broadcast_mac[] = {0xff, 0xff, 0xff, 0xff, 0xff, 0xff};
  388. const uint8_t multicast_mac_header[] = {0x01, 0x00, 0x5e};
  389. // if it's broadcast or IGMP, flood it
  390. if (is_igmp || !memcmp(eh->dest, broadcast_mac, sizeof(broadcast_mac))) {
  391. o->decide_state = DECIDE_STATE_FLOOD;
  392. LinkedList2Iterator_InitForward(&o->decide_flood_it, &o->peers_list);
  393. return;
  394. }
  395. // if it's multicast, forward to all peers with the given sig
  396. if (!memcmp(eh->dest, multicast_mac_header, sizeof(multicast_mac_header))) {
  397. // extract group's sig from destination MAC
  398. uint32_t sig = compute_sig_for_mac(eh->dest);
  399. // look up the sig in multicast tree
  400. BAVLNode *node;
  401. if (node = BAVL_LookupExact(&o->multicast_tree, &sig)) {
  402. struct _FrameDecider_group_entry *master = UPPER_OBJECT(node, struct _FrameDecider_group_entry, master.tree_node);
  403. ASSERT(master->is_master)
  404. o->decide_state = DECIDE_STATE_MULTICAST;
  405. LinkedList3Iterator_Init(&o->decide_multicast_it, LinkedList3Node_First(&master->sig_list_node), 1);
  406. }
  407. return;
  408. }
  409. // look for MAC entry
  410. FDMacsTreeRef ref = FDMacsTree_LookupExact(&o->macs_tree, 0, eh->dest);
  411. if (FDMacsTreeIsValidRef(ref)) {
  412. struct _FrameDecider_mac_entry *entry = ref.ptr;
  413. o->decide_state = DECIDE_STATE_UNICAST;
  414. o->decide_unicast_peer = entry->peer;
  415. return;
  416. }
  417. // unknown destination MAC, flood
  418. o->decide_state = DECIDE_STATE_FLOOD;
  419. LinkedList2Iterator_InitForward(&o->decide_flood_it, &o->peers_list);
  420. return;
  421. }
  422. FrameDeciderPeer * FrameDecider_NextDestination (FrameDecider *o)
  423. {
  424. DebugObject_Access(&o->d_obj);
  425. switch (o->decide_state) {
  426. case DECIDE_STATE_NONE: {
  427. return NULL;
  428. } break;
  429. case DECIDE_STATE_UNICAST: {
  430. o->decide_state = DECIDE_STATE_NONE;
  431. return o->decide_unicast_peer;
  432. } break;
  433. case DECIDE_STATE_FLOOD: {
  434. LinkedList2Node *list_node = LinkedList2Iterator_Next(&o->decide_flood_it);
  435. if (!list_node) {
  436. o->decide_state = DECIDE_STATE_NONE;
  437. return NULL;
  438. }
  439. FrameDeciderPeer *peer = UPPER_OBJECT(list_node, FrameDeciderPeer, list_node);
  440. return peer;
  441. } break;
  442. case DECIDE_STATE_MULTICAST: {
  443. LinkedList3Node *list_node = LinkedList3Iterator_Next(&o->decide_multicast_it);
  444. if (!list_node) {
  445. o->decide_state = DECIDE_STATE_NONE;
  446. return NULL;
  447. }
  448. struct _FrameDecider_group_entry *group_entry = UPPER_OBJECT(list_node, struct _FrameDecider_group_entry, sig_list_node);
  449. return group_entry->peer;
  450. } break;
  451. default:
  452. ASSERT(0);
  453. return NULL;
  454. }
  455. }
  456. int FrameDeciderPeer_Init (FrameDeciderPeer *o, FrameDecider *d, void *user, BLog_logfunc logfunc)
  457. {
  458. // init arguments
  459. o->d = d;
  460. o->user = user;
  461. o->logfunc = logfunc;
  462. // allocate MAC entries
  463. if (!(o->mac_entries = BAllocArray(d->max_peer_macs, sizeof(struct _FrameDecider_mac_entry)))) {
  464. PeerLog(o, BLOG_ERROR, "failed to allocate MAC entries");
  465. goto fail0;
  466. }
  467. // allocate group entries
  468. if (!(o->group_entries = BAllocArray(d->max_peer_groups, sizeof(struct _FrameDecider_group_entry)))) {
  469. PeerLog(o, BLOG_ERROR, "failed to allocate group entries");
  470. goto fail1;
  471. }
  472. // insert to peers list
  473. LinkedList2_Append(&d->peers_list, &o->list_node);
  474. // init MAC entry lists
  475. LinkedList2_Init(&o->mac_entries_free);
  476. LinkedList2_Init(&o->mac_entries_used);
  477. // initialize MAC entries
  478. for (int i = 0; i < d->max_peer_macs; i++) {
  479. struct _FrameDecider_mac_entry *entry = &o->mac_entries[i];
  480. // set peer
  481. entry->peer = o;
  482. // insert to free list
  483. LinkedList2_Append(&o->mac_entries_free, &entry->list_node);
  484. }
  485. // init group entry lists
  486. LinkedList2_Init(&o->group_entries_free);
  487. LinkedList2_Init(&o->group_entries_used);
  488. // initialize group entries
  489. for (int i = 0; i < d->max_peer_groups; i++) {
  490. struct _FrameDecider_group_entry *entry = &o->group_entries[i];
  491. // set peer
  492. entry->peer = o;
  493. // insert to free list
  494. LinkedList2_Append(&o->group_entries_free, &entry->list_node);
  495. // init timer
  496. BTimer_Init(&entry->timer, 0, (BTimer_handler)group_entry_timer_handler, entry);
  497. }
  498. // initialize groups tree
  499. FDGroupsTree_Init(&o->groups_tree);
  500. DebugObject_Init(&o->d_obj);
  501. return 1;
  502. fail1:
  503. BFree(o->mac_entries);
  504. fail0:
  505. return 0;
  506. }
  507. void FrameDeciderPeer_Free (FrameDeciderPeer *o)
  508. {
  509. DebugObject_Free(&o->d_obj);
  510. FrameDecider *d = o->d;
  511. // remove decide unicast reference
  512. if (d->decide_state == DECIDE_STATE_UNICAST && d->decide_unicast_peer == o) {
  513. d->decide_state = DECIDE_STATE_NONE;
  514. }
  515. LinkedList2Iterator it;
  516. LinkedList2Node *node;
  517. // free group entries
  518. LinkedList2Iterator_InitForward(&it, &o->group_entries_used);
  519. while (node = LinkedList2Iterator_Next(&it)) {
  520. struct _FrameDecider_group_entry *entry = UPPER_OBJECT(node, struct _FrameDecider_group_entry, list_node);
  521. // remove from multicast
  522. remove_from_multicast(d, entry);
  523. // stop timer
  524. BReactor_RemoveTimer(d->reactor, &entry->timer);
  525. }
  526. // remove used MAC entries from tree
  527. LinkedList2Iterator_InitForward(&it, &o->mac_entries_used);
  528. while (node = LinkedList2Iterator_Next(&it)) {
  529. struct _FrameDecider_mac_entry *entry = UPPER_OBJECT(node, struct _FrameDecider_mac_entry, list_node);
  530. // remove from tree
  531. FDMacsTree_Remove(&d->macs_tree, 0, FDMacsTreeDeref(0, entry));
  532. }
  533. // remove from peers list
  534. LinkedList2_Remove(&d->peers_list, &o->list_node);
  535. // free group entries
  536. BFree(o->group_entries);
  537. // free MAC entries
  538. BFree(o->mac_entries);
  539. }
  540. void FrameDeciderPeer_Analyze (FrameDeciderPeer *o, const uint8_t *frame, int frame_len)
  541. {
  542. ASSERT(frame_len >= 0)
  543. DebugObject_Access(&o->d_obj);
  544. const uint8_t *pos = frame;
  545. int len = frame_len;
  546. if (len < sizeof(struct ethernet_header)) {
  547. goto out;
  548. }
  549. struct ethernet_header *eh = (struct ethernet_header *)pos;
  550. pos += sizeof(struct ethernet_header);
  551. len -= sizeof(struct ethernet_header);
  552. // register source MAC address with this peer
  553. add_mac_to_peer(o, eh->source);
  554. switch (ntoh16(eh->type)) {
  555. case ETHERTYPE_IPV4: {
  556. // check IPv4 header
  557. struct ipv4_header *ipv4_header;
  558. if (!ipv4_check((uint8_t *)pos, len, &ipv4_header, (uint8_t **)&pos, &len)) {
  559. PeerLog(o, BLOG_INFO, "analyze: wrong IP packet");
  560. goto out;
  561. }
  562. // check if it's IGMP
  563. if (ntoh8(ipv4_header->protocol) != IPV4_PROTOCOL_IGMP) {
  564. goto out;
  565. }
  566. // check IGMP header
  567. if (len < sizeof(struct igmp_base)) {
  568. PeerLog(o, BLOG_INFO, "analyze: IGMP: short packet");
  569. goto out;
  570. }
  571. struct igmp_base *igmp_base = (struct igmp_base *)pos;
  572. pos += sizeof(struct igmp_base);
  573. len -= sizeof(struct igmp_base);
  574. switch (ntoh8(igmp_base->type)) {
  575. case IGMP_TYPE_V2_MEMBERSHIP_REPORT: {
  576. // check extra
  577. if (len < sizeof(struct igmp_v2_extra)) {
  578. PeerLog(o, BLOG_INFO, "analyze: IGMP: short v2 report");
  579. goto out;
  580. }
  581. struct igmp_v2_extra *report = (struct igmp_v2_extra *)pos;
  582. pos += sizeof(struct igmp_v2_extra);
  583. len -= sizeof(struct igmp_v2_extra);
  584. // add to group
  585. add_group_to_peer(o, report->group);
  586. } break;
  587. case IGMP_TYPE_V3_MEMBERSHIP_REPORT: {
  588. // check extra
  589. if (len < sizeof(struct igmp_v3_report_extra)) {
  590. PeerLog(o, BLOG_INFO, "analyze: IGMP: short v3 report");
  591. goto out;
  592. }
  593. struct igmp_v3_report_extra *report = (struct igmp_v3_report_extra *)pos;
  594. pos += sizeof(struct igmp_v3_report_extra);
  595. len -= sizeof(struct igmp_v3_report_extra);
  596. // iterate records
  597. uint16_t num_records = ntoh16(report->number_of_group_records);
  598. for (int i = 0; i < num_records; i++) {
  599. // check record
  600. if (len < sizeof(struct igmp_v3_report_record)) {
  601. PeerLog(o, BLOG_INFO, "analyze: IGMP: short record header");
  602. goto out;
  603. }
  604. struct igmp_v3_report_record *record = (struct igmp_v3_report_record *)pos;
  605. pos += sizeof(struct igmp_v3_report_record);
  606. len -= sizeof(struct igmp_v3_report_record);
  607. // iterate sources
  608. uint16_t num_sources = ntoh16(record->number_of_sources);
  609. int j;
  610. for (j = 0; j < num_sources; j++) {
  611. // check source
  612. if (len < sizeof(struct igmp_source)) {
  613. PeerLog(o, BLOG_INFO, "analyze: IGMP: short source");
  614. goto out;
  615. }
  616. struct igmp_source *source = (struct igmp_source *)pos;
  617. pos += sizeof(struct igmp_source);
  618. len -= sizeof(struct igmp_source);
  619. }
  620. // check aux data
  621. uint16_t aux_len = ntoh16(record->aux_data_len);
  622. if (len < aux_len) {
  623. PeerLog(o, BLOG_INFO, "analyze: IGMP: short record aux data");
  624. goto out;
  625. }
  626. pos += aux_len;
  627. len -= aux_len;
  628. switch (record->type) {
  629. case IGMP_RECORD_TYPE_MODE_IS_INCLUDE:
  630. case IGMP_RECORD_TYPE_CHANGE_TO_INCLUDE_MODE:
  631. if (num_sources != 0) {
  632. add_group_to_peer(o, record->group);
  633. }
  634. break;
  635. case IGMP_RECORD_TYPE_MODE_IS_EXCLUDE:
  636. case IGMP_RECORD_TYPE_CHANGE_TO_EXCLUDE_MODE:
  637. add_group_to_peer(o, record->group);
  638. break;
  639. }
  640. }
  641. } break;
  642. }
  643. } break;
  644. }
  645. out:;
  646. }