DataProto.c 21 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731
  1. /**
  2. * @file DataProto.c
  3. * @author Ambroz Bizjak <ambrop7@gmail.com>
  4. *
  5. * @section LICENSE
  6. *
  7. * This file is part of BadVPN.
  8. *
  9. * BadVPN is free software: you can redistribute it and/or modify
  10. * it under the terms of the GNU General Public License version 2
  11. * as published by the Free Software Foundation.
  12. *
  13. * BadVPN is distributed in the hope that it will be useful,
  14. * but WITHOUT ANY WARRANTY; without even the implied warranty of
  15. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  16. * GNU General Public License for more details.
  17. *
  18. * You should have received a copy of the GNU General Public License along
  19. * with this program; if not, write to the Free Software Foundation, Inc.,
  20. * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  21. */
  22. #include <stdlib.h>
  23. #include <string.h>
  24. #include <limits.h>
  25. #include <protocol/dataproto.h>
  26. #include <misc/offset.h>
  27. #include <misc/byteorder.h>
  28. #include <misc/debug.h>
  29. #include <system/BLog.h>
  30. #include <client/DataProto.h>
  31. #include <generated/blog_channel_DataProto.h>
  32. #define DATAPROTO_TIMEOUT 30000
  33. struct dp_relay_flow {
  34. DataProtoRelaySource *rs;
  35. DataProtoDest *dp;
  36. BestEffortPacketWriteInterface *ainput_if;
  37. PacketBufferAsyncInput ainput;
  38. PacketBuffer buffer;
  39. PacketPassInactivityMonitor monitor;
  40. PacketPassFairQueueFlow qflow;
  41. LinkedList2Node source_list_node;
  42. BAVLNode source_tree_node;
  43. LinkedList2Node dp_list_node;
  44. };
  45. static int peerid_comparator (void *user, peerid_t *val1, peerid_t *val2);
  46. static struct dp_relay_flow * create_relay_flow (DataProtoRelaySource *rs, DataProtoDest *dp, int num_packets);
  47. static void dealloc_relay_flow (struct dp_relay_flow *flow);
  48. static int release_relay_flow (struct dp_relay_flow *flow);
  49. static void flow_monitor_handler (struct dp_relay_flow *flow);
  50. static void monitor_handler (DataProtoDest *o);
  51. static int send_keepalive (DataProtoDest *o);
  52. static void receive_timer_handler (DataProtoDest *o);
  53. static void notifier_handler (DataProtoDest *o, uint8_t *data, int data_len);
  54. static int pointer_comparator (void *user, void **val1, void **val2);
  55. static void keepalive_job_handler (DataProtoDest *o);
  56. int peerid_comparator (void *user, peerid_t *val1, peerid_t *val2)
  57. {
  58. if (*val1 < *val2) {
  59. return -1;
  60. }
  61. if (*val1 > *val2) {
  62. return 1;
  63. }
  64. return 0;
  65. }
  66. struct dp_relay_flow * create_relay_flow (DataProtoRelaySource *rs, DataProtoDest *dp, int num_packets)
  67. {
  68. ASSERT(!BAVL_LookupExact(&rs->relay_flows_tree, &dp))
  69. ASSERT(num_packets > 0)
  70. ASSERT(!dp->d_freeing)
  71. ASSERT(!PacketPassInterface_InClient(dp->d_output))
  72. // allocate flow structure
  73. struct dp_relay_flow *flow = malloc(sizeof(struct dp_relay_flow));
  74. if (!flow) {
  75. BLog(BLOG_ERROR, "failed to allocate flow structure for relay flow from peer %d to %d", (int)rs->source_id, (int)dp->dest_id);
  76. goto fail0;
  77. }
  78. // set source and dp
  79. flow->rs = rs;
  80. flow->dp = dp;
  81. // init queue flow
  82. PacketPassFairQueueFlow_Init(&flow->qflow, &dp->queue);
  83. // init inacitvity monitor
  84. PacketPassInactivityMonitor_Init(&flow->monitor, PacketPassFairQueueFlow_GetInput(&flow->qflow), dp->reactor, DATAPROTO_TIMEOUT, (PacketPassInactivityMonitor_handler)flow_monitor_handler, flow);
  85. // init async input
  86. PacketBufferAsyncInput_Init(&flow->ainput, dp->mtu);
  87. flow->ainput_if = PacketBufferAsyncInput_GetInput(&flow->ainput);
  88. // init buffer
  89. if (!PacketBuffer_Init(&flow->buffer, PacketBufferAsyncInput_GetOutput(&flow->ainput), PacketPassInactivityMonitor_GetInput(&flow->monitor), num_packets, BReactor_PendingGroup(dp->reactor))) {
  90. BLog(BLOG_ERROR, "PacketBuffer_Init failed for relay flow from peer %d to %d", (int)rs->source_id, (int)dp->dest_id);
  91. goto fail1;
  92. }
  93. // insert to source list
  94. LinkedList2_Append(&rs->relay_flows_list, &flow->source_list_node);
  95. // insert to source tree
  96. ASSERT_EXECUTE(BAVL_Insert(&rs->relay_flows_tree, &flow->source_tree_node, NULL))
  97. // insert to dp list
  98. LinkedList2_Append(&dp->relay_flows_list, &flow->dp_list_node);
  99. BLog(BLOG_NOTICE, "created relay flow from peer %d to %d", (int)rs->source_id, (int)dp->dest_id);
  100. return flow;
  101. fail1:
  102. PacketBufferAsyncInput_Free(&flow->ainput);
  103. PacketPassInactivityMonitor_Free(&flow->monitor);
  104. PacketPassFairQueueFlow_Free(&flow->qflow);
  105. free(flow);
  106. fail0:
  107. return NULL;
  108. }
  109. void dealloc_relay_flow (struct dp_relay_flow *flow)
  110. {
  111. #ifndef NDEBUG
  112. if (!flow->dp->d_freeing) {
  113. ASSERT(!PacketPassFairQueueFlow_IsBusy(&flow->qflow))
  114. ASSERT(!PacketPassInterface_InClient(flow->dp->d_output))
  115. }
  116. #endif
  117. DataProtoDest *o = flow->dp;
  118. // remove from dp list
  119. LinkedList2_Remove(&flow->dp->relay_flows_list, &flow->dp_list_node);
  120. // remove from source tree
  121. BAVL_Remove(&flow->rs->relay_flows_tree, &flow->source_tree_node);
  122. // remove from source list
  123. LinkedList2_Remove(&flow->rs->relay_flows_list, &flow->source_list_node);
  124. // free buffer
  125. PacketBuffer_Free(&flow->buffer);
  126. // free async input
  127. PacketBufferAsyncInput_Free(&flow->ainput);
  128. // free inacitvity monitor
  129. PacketPassInactivityMonitor_Free(&flow->monitor);
  130. // free queue flow
  131. PacketPassFairQueueFlow_Free(&flow->qflow);
  132. // free flow structure
  133. free(flow);
  134. }
  135. int release_relay_flow (struct dp_relay_flow *flow)
  136. {
  137. ASSERT(!flow->dp->d_freeing)
  138. ASSERT(!PacketPassInterface_InClient(flow->dp->d_output))
  139. DataProtoDest *o = flow->dp;
  140. if (PacketPassFairQueueFlow_IsBusy(&flow->qflow)) {
  141. // release it
  142. DEAD_ENTER(o->dead)
  143. PacketPassFairQueueFlow_Release(&flow->qflow);
  144. if (DEAD_LEAVE(o->dead)) {
  145. return -1;
  146. }
  147. }
  148. // remove flow
  149. dealloc_relay_flow(flow);
  150. return 0;
  151. }
  152. void flow_monitor_handler (struct dp_relay_flow *flow)
  153. {
  154. ASSERT(!flow->dp->d_freeing)
  155. ASSERT(!PacketPassInterface_InClient(flow->dp->d_output))
  156. BLog(BLOG_NOTICE, "relay flow from peer %d to %d timed out", (int)flow->rs->source_id, (int)flow->dp->dest_id);
  157. release_relay_flow(flow);
  158. return;
  159. }
  160. void monitor_handler (DataProtoDest *o)
  161. {
  162. ASSERT(!o->d_freeing)
  163. ASSERT(!PacketPassInterface_InClient(o->d_output))
  164. DebugObject_Access(&o->d_obj);
  165. send_keepalive(o);
  166. return;
  167. }
  168. int send_keepalive (DataProtoDest *o)
  169. {
  170. ASSERT(!o->d_freeing)
  171. ASSERT(!PacketPassInterface_InClient(o->d_output))
  172. BLog(BLOG_DEBUG, "sending keepalive to peer %d", (int)o->dest_id);
  173. DEAD_ENTER(o->dead)
  174. PacketRecvBlocker_AllowBlockedPacket(&o->ka_blocker);
  175. if (DEAD_LEAVE(o->dead)) {
  176. return -1;
  177. }
  178. return 0;
  179. }
  180. void receive_timer_handler (DataProtoDest *o)
  181. {
  182. DebugObject_Access(&o->d_obj);
  183. BLog(BLOG_DEBUG, "receive timer triggered for peer %d", (int)o->dest_id);
  184. int prev_up = o->up;
  185. // consider down
  186. o->up = 0;
  187. // call handler if up state changed
  188. if (o->handler && o->up != prev_up) {
  189. o->handler(o->user, o->up);
  190. return;
  191. }
  192. }
  193. void notifier_handler (DataProtoDest *o, uint8_t *data, int data_len)
  194. {
  195. ASSERT(data_len >= sizeof(struct dataproto_header))
  196. DebugObject_Access(&o->d_obj);
  197. // modify existing packet here
  198. struct dataproto_header *header = (struct dataproto_header *)data;
  199. header->flags = 0;
  200. // if we are receiving keepalives, set the flag
  201. if (BTimer_IsRunning(&o->receive_timer)) {
  202. header->flags |= DATAPROTO_FLAGS_RECEIVING_KEEPALIVES;
  203. }
  204. }
  205. int pointer_comparator (void *user, void **val1, void **val2)
  206. {
  207. if (*val1 < *val2) {
  208. return -1;
  209. }
  210. if (*val1 > *val2) {
  211. return 1;
  212. }
  213. return 0;
  214. }
  215. void keepalive_job_handler (DataProtoDest *o)
  216. {
  217. ASSERT(!o->d_freeing)
  218. ASSERT(!PacketPassInterface_InClient(o->d_output))
  219. DebugObject_Access(&o->d_obj);
  220. send_keepalive(o);
  221. return;
  222. }
  223. int DataProtoDest_Init (DataProtoDest *o, BReactor *reactor, peerid_t dest_id, PacketPassInterface *output, btime_t keepalive_time, btime_t tolerance_time, DataProtoDest_handler handler, void *user)
  224. {
  225. ASSERT(PacketPassInterface_HasCancel(output))
  226. ASSERT(PacketPassInterface_GetMTU(output) >= sizeof(struct dataproto_header) + sizeof(struct dataproto_peer_id))
  227. // init arguments
  228. o->reactor = reactor;
  229. o->dest_id = dest_id;
  230. o->handler = handler;
  231. o->user = user;
  232. // init dead var
  233. DEAD_INIT(o->dead);
  234. // set MTU
  235. o->mtu = PacketPassInterface_GetMTU(output);
  236. // set frame MTU
  237. o->frame_mtu = o->mtu - (sizeof(struct dataproto_header) + sizeof(struct dataproto_peer_id));
  238. // init notifier
  239. PacketPassNotifier_Init(&o->notifier, output);
  240. PacketPassNotifier_SetHandler(&o->notifier, (PacketPassNotifier_handler_notify)notifier_handler, o);
  241. // init monitor
  242. PacketPassInactivityMonitor_Init(&o->monitor, PacketPassNotifier_GetInput(&o->notifier), o->reactor, keepalive_time, (PacketPassInactivityMonitor_handler)monitor_handler, o);
  243. // init queue
  244. PacketPassFairQueue_Init(&o->queue, PacketPassInactivityMonitor_GetInput(&o->monitor), BReactor_PendingGroup(o->reactor));
  245. PacketPassFairQueue_EnableCancel(&o->queue);
  246. // init keepalive queue flow
  247. PacketPassFairQueueFlow_Init(&o->ka_qflow, &o->queue);
  248. // init keepalive source
  249. DataProtoKeepaliveSource_Init(&o->ka_source);
  250. // init keepalive blocker
  251. PacketRecvBlocker_Init(&o->ka_blocker, DataProtoKeepaliveSource_GetOutput(&o->ka_source));
  252. // init keepalive buffer
  253. if (!SinglePacketBuffer_Init(&o->ka_buffer, PacketRecvBlocker_GetOutput(&o->ka_blocker), PacketPassFairQueueFlow_GetInput(&o->ka_qflow), BReactor_PendingGroup(o->reactor))) {
  254. BLog(BLOG_ERROR, "SinglePacketBuffer_Init failed");
  255. goto fail0;
  256. }
  257. // init receive timer
  258. BTimer_Init(&o->receive_timer, tolerance_time, (BTimer_handler)receive_timer_handler, o);
  259. // set not up
  260. o->up = 0;
  261. // init relay flows list
  262. LinkedList2_Init(&o->relay_flows_list);
  263. // init keepalive job
  264. BPending_Init(&o->keepalive_job, BReactor_PendingGroup(o->reactor), (BPending_handler)keepalive_job_handler, o);
  265. BPending_Set(&o->keepalive_job);
  266. // init flows counter
  267. DebugCounter_Init(&o->flows_counter);
  268. // init debug object
  269. DebugObject_Init(&o->d_obj);
  270. #ifndef NDEBUG
  271. o->d_output = output;
  272. o->d_freeing = 0;
  273. #endif
  274. return 1;
  275. fail0:
  276. PacketRecvBlocker_Free(&o->ka_blocker);
  277. DataProtoKeepaliveSource_Free(&o->ka_source);
  278. PacketPassFairQueueFlow_Free(&o->ka_qflow);
  279. PacketPassFairQueue_Free(&o->queue);
  280. PacketPassInactivityMonitor_Free(&o->monitor);
  281. PacketPassNotifier_Free(&o->notifier);
  282. return 0;
  283. }
  284. void DataProtoDest_Free (DataProtoDest *o)
  285. {
  286. DebugCounter_Free(&o->flows_counter);
  287. DebugObject_Free(&o->d_obj);
  288. // free keepalive job
  289. BPending_Free(&o->keepalive_job);
  290. // allow freeing queue flows
  291. PacketPassFairQueue_PrepareFree(&o->queue);
  292. // free relay flows
  293. LinkedList2Node *node;
  294. while (node = LinkedList2_GetFirst(&o->relay_flows_list)) {
  295. struct dp_relay_flow *flow = UPPER_OBJECT(node, struct dp_relay_flow, dp_list_node);
  296. dealloc_relay_flow(flow);
  297. }
  298. // free receive timer
  299. BReactor_RemoveTimer(o->reactor, &o->receive_timer);
  300. // free keepalive buffer
  301. SinglePacketBuffer_Free(&o->ka_buffer);
  302. // free keepalive blocker
  303. PacketRecvBlocker_Free(&o->ka_blocker);
  304. // free keepalive source
  305. DataProtoKeepaliveSource_Free(&o->ka_source);
  306. // free keepalive queue flow
  307. PacketPassFairQueueFlow_Free(&o->ka_qflow);
  308. // free queue
  309. PacketPassFairQueue_Free(&o->queue);
  310. // free monitor
  311. PacketPassInactivityMonitor_Free(&o->monitor);
  312. // free notifier
  313. PacketPassNotifier_Free(&o->notifier);
  314. // free dead var
  315. DEAD_KILL(o->dead);
  316. }
  317. void DataProtoDest_PrepareFree (DataProtoDest *o)
  318. {
  319. DebugObject_Access(&o->d_obj);
  320. // allow freeing queue flows
  321. PacketPassFairQueue_PrepareFree(&o->queue);
  322. #ifndef NDEBUG
  323. o->d_freeing = 1;
  324. #endif
  325. }
  326. void DataProtoDest_SubmitRelayFrame (DataProtoDest *o, DataProtoRelaySource *rs, uint8_t *data, int data_len, int buffer_num_packets)
  327. {
  328. ASSERT(data_len >= 0)
  329. ASSERT(data_len <= o->frame_mtu)
  330. ASSERT(buffer_num_packets > 0)
  331. ASSERT(!o->d_freeing)
  332. ASSERT(!PacketPassInterface_InClient(o->d_output))
  333. DebugObject_Access(&rs->d_obj);
  334. DebugObject_Access(&o->d_obj);
  335. // lookup relay flow from source to this DataProto
  336. struct dp_relay_flow *flow;
  337. BAVLNode *node = BAVL_LookupExact(&rs->relay_flows_tree, &o);
  338. if (!node) {
  339. // create new flow
  340. if (!(flow = create_relay_flow(rs, o, buffer_num_packets))) {
  341. return;
  342. }
  343. } else {
  344. flow = UPPER_OBJECT(node, struct dp_relay_flow, source_tree_node);
  345. }
  346. // get a buffer
  347. uint8_t *out;
  348. // safe because of PacketBufferAsyncInput
  349. if (!BestEffortPacketWriteInterface_Sender_StartPacket(flow->ainput_if, &out)) {
  350. BLog(BLOG_NOTICE, "out of buffer for relayed frame from peer %d to %d", (int)rs->source_id, (int)o->dest_id);
  351. return;
  352. }
  353. // write header
  354. struct dataproto_header *header = (struct dataproto_header *)out;
  355. // don't set flags, it will be set in notifier_handler
  356. header->from_id = htol16(rs->source_id);
  357. header->num_peer_ids = htol16(1);
  358. struct dataproto_peer_id *id = (struct dataproto_peer_id *)(out + sizeof(struct dataproto_header));
  359. id->id = htol16(o->dest_id);
  360. // write data
  361. memcpy(out + sizeof(struct dataproto_header) + sizeof(struct dataproto_peer_id), data, data_len);
  362. // submit it
  363. BestEffortPacketWriteInterface_Sender_EndPacket(flow->ainput_if, sizeof(struct dataproto_header) + sizeof(struct dataproto_peer_id) + data_len);
  364. return;
  365. }
  366. void DataProtoDest_Received (DataProtoDest *o, int peer_receiving)
  367. {
  368. ASSERT(peer_receiving == 0 || peer_receiving == 1)
  369. ASSERT(!o->d_freeing)
  370. ASSERT(!PacketPassInterface_InClient(o->d_output))
  371. DebugObject_Access(&o->d_obj);
  372. int prev_up = o->up;
  373. // reset receive timer
  374. BReactor_SetTimer(o->reactor, &o->receive_timer);
  375. if (!peer_receiving) {
  376. // peer reports not receiving, consider down
  377. o->up = 0;
  378. // send keep-alive to converge faster
  379. if (send_keepalive(o) < 0) {
  380. return;
  381. }
  382. } else {
  383. // consider up
  384. o->up = 1;
  385. }
  386. // call handler if up state changed
  387. if (o->handler && o->up != prev_up) {
  388. o->handler(o->user, o->up);
  389. return;
  390. }
  391. }
  392. int DataProtoLocalSource_Init (DataProtoLocalSource *o, int frame_mtu, peerid_t source_id, peerid_t dest_id, int num_packets, BReactor *reactor)
  393. {
  394. ASSERT(frame_mtu >= 0)
  395. ASSERT(frame_mtu <= INT_MAX - (sizeof(struct dataproto_header) + sizeof(struct dataproto_peer_id)))
  396. ASSERT(num_packets > 0)
  397. // init arguments
  398. o->frame_mtu = frame_mtu;
  399. o->source_id = source_id;
  400. o->dest_id = dest_id;
  401. // calculate packet MTU
  402. int packet_mtu = o->frame_mtu + sizeof(struct dataproto_header) + sizeof(struct dataproto_peer_id);
  403. // init dead var
  404. DEAD_INIT(o->dead);
  405. // init connector
  406. PacketPassConnector_Init(&o->connector, packet_mtu, BReactor_PendingGroup(reactor));
  407. // init async input
  408. PacketBufferAsyncInput_Init(&o->ainput, packet_mtu);
  409. o->ainput_if = PacketBufferAsyncInput_GetInput(&o->ainput);
  410. // init buffer
  411. if (!PacketBuffer_Init(&o->buffer, PacketBufferAsyncInput_GetOutput(&o->ainput), PacketPassConnector_GetInput(&o->connector), num_packets, BReactor_PendingGroup(reactor))) {
  412. BLog(BLOG_ERROR, "PacketBuffer_Init failed");
  413. goto fail1;
  414. }
  415. // set no DataProto
  416. o->dp = NULL;
  417. // init debug object
  418. DebugObject_Init(&o->d_obj);
  419. return 1;
  420. fail1:
  421. PacketBufferAsyncInput_Free(&o->ainput);
  422. PacketPassConnector_Free(&o->connector);
  423. fail0:
  424. return 0;
  425. }
  426. void DataProtoLocalSource_Free (DataProtoLocalSource *o)
  427. {
  428. ASSERT(!o->dp)
  429. DebugObject_Free(&o->d_obj);
  430. // free buffer
  431. PacketBuffer_Free(&o->buffer);
  432. // free async input
  433. PacketBufferAsyncInput_Free(&o->ainput);
  434. // free connector
  435. PacketPassConnector_Free(&o->connector);
  436. // free dead var
  437. DEAD_KILL(o->dead);
  438. }
  439. void DataProtoLocalSource_SubmitFrame (DataProtoLocalSource *o, uint8_t *data, int data_len)
  440. {
  441. ASSERT(data_len >= 0)
  442. ASSERT(data_len <= o->frame_mtu)
  443. if (o->dp) {
  444. ASSERT(!o->d_dp_released)
  445. ASSERT(!o->dp->d_freeing)
  446. ASSERT(!PacketPassInterface_InClient(o->dp->d_output))
  447. }
  448. DebugObject_Access(&o->d_obj);
  449. // get a buffer
  450. uint8_t *out;
  451. // safe because of PacketBufferAsyncInput
  452. if (!BestEffortPacketWriteInterface_Sender_StartPacket(o->ainput_if, &out)) {
  453. BLog(BLOG_NOTICE, "out of buffer for frame from peer %d to %d", (int)o->source_id, (int)o->dest_id);
  454. return;
  455. }
  456. // write header
  457. struct dataproto_header *header = (struct dataproto_header *)out;
  458. // don't set flags, it will be set in notifier_handler
  459. header->from_id = htol16(o->source_id);
  460. header->num_peer_ids = htol16(1);
  461. struct dataproto_peer_id *id = (struct dataproto_peer_id *)(out + sizeof(struct dataproto_header));
  462. id->id = htol16(o->dest_id);
  463. // write data
  464. memcpy(out + sizeof(struct dataproto_header) + sizeof(struct dataproto_peer_id), data, data_len);
  465. // submit it
  466. BestEffortPacketWriteInterface_Sender_EndPacket(o->ainput_if, sizeof(struct dataproto_header) + sizeof(struct dataproto_peer_id) + data_len);
  467. return;
  468. }
  469. void DataProtoLocalSource_Attach (DataProtoLocalSource *o, DataProtoDest *dp)
  470. {
  471. ASSERT(dp)
  472. ASSERT(!o->dp)
  473. ASSERT(o->frame_mtu <= dp->frame_mtu)
  474. ASSERT(!dp->d_freeing)
  475. ASSERT(!PacketPassInterface_InClient(dp->d_output))
  476. DebugObject_Access(&o->d_obj);
  477. DebugObject_Access(&dp->d_obj);
  478. // set DataProto
  479. o->dp = dp;
  480. // init queue flow
  481. PacketPassFairQueueFlow_Init(&o->dp_qflow, &dp->queue);
  482. // connect to queue flow
  483. PacketPassConnector_ConnectOutput(&o->connector, PacketPassFairQueueFlow_GetInput(&o->dp_qflow));
  484. // increment flows counter
  485. DebugCounter_Increment(&dp->flows_counter);
  486. #ifndef NDEBUG
  487. o->d_dp_released = 0;
  488. #endif
  489. }
  490. void DataProtoLocalSource_Release (DataProtoLocalSource *o)
  491. {
  492. ASSERT(o->dp)
  493. ASSERT(!o->d_dp_released)
  494. ASSERT(!o->dp->d_freeing)
  495. ASSERT(!PacketPassInterface_InClient(o->dp->d_output))
  496. DebugObject_Access(&o->d_obj);
  497. if (PacketPassFairQueueFlow_IsBusy(&o->dp_qflow)) {
  498. DEAD_ENTER(o->dead)
  499. PacketPassFairQueueFlow_Release(&o->dp_qflow);
  500. if (DEAD_LEAVE(o->dead)) {
  501. return;
  502. }
  503. }
  504. #ifndef NDEBUG
  505. o->d_dp_released = 1;
  506. #endif
  507. }
  508. void DataProtoLocalSource_Detach (DataProtoLocalSource *o)
  509. {
  510. #ifndef NDEBUG
  511. ASSERT(o->dp)
  512. ASSERT(o->d_dp_released || o->dp->d_freeing)
  513. if (!o->dp->d_freeing) {
  514. ASSERT(!PacketPassInterface_InClient(o->dp->d_output))
  515. }
  516. #endif
  517. DebugObject_Access(&o->d_obj);
  518. DataProtoDest *dp = o->dp;
  519. // decrement flows counter
  520. DebugCounter_Decrement(&dp->flows_counter);
  521. // disconnect from queue flow
  522. PacketPassConnector_DisconnectOutput(&o->connector);
  523. // free queue flow
  524. PacketPassFairQueueFlow_Free(&o->dp_qflow);
  525. // set no DataProto
  526. o->dp = NULL;
  527. }
  528. void DataProtoRelaySource_Init (DataProtoRelaySource *o, peerid_t source_id)
  529. {
  530. // init arguments
  531. o->source_id = source_id;
  532. // init dead var
  533. DEAD_INIT(o->dead);
  534. // init relay flows list
  535. LinkedList2_Init(&o->relay_flows_list);
  536. // init relay flows tree
  537. BAVL_Init(&o->relay_flows_tree, OFFSET_DIFF(struct dp_relay_flow, dp, source_tree_node), (BAVL_comparator)pointer_comparator, NULL);
  538. // init debug object
  539. DebugObject_Init(&o->d_obj);
  540. }
  541. void DataProtoRelaySource_Free (DataProtoRelaySource *o)
  542. {
  543. ASSERT(BAVL_IsEmpty(&o->relay_flows_tree))
  544. ASSERT(LinkedList2_IsEmpty(&o->relay_flows_list))
  545. DebugObject_Free(&o->d_obj);
  546. // free dead var
  547. DEAD_KILL(o->dead);
  548. }
  549. int DataProtoRelaySource_IsEmpty (DataProtoRelaySource *o)
  550. {
  551. DebugObject_Access(&o->d_obj);
  552. return LinkedList2_IsEmpty(&o->relay_flows_list);
  553. }
  554. void DataProtoRelaySource_Release (DataProtoRelaySource *o)
  555. {
  556. DebugObject_Access(&o->d_obj);
  557. LinkedList2Node *node;
  558. while (node = LinkedList2_GetFirst(&o->relay_flows_list)) {
  559. struct dp_relay_flow *flow = UPPER_OBJECT(node, struct dp_relay_flow, source_list_node);
  560. DEAD_ENTER(o->dead)
  561. release_relay_flow(flow);
  562. if (DEAD_LEAVE(o->dead)) {
  563. return;
  564. }
  565. }
  566. }
  567. void DataProtoRelaySource_FreeRelease (DataProtoRelaySource *o)
  568. {
  569. DebugObject_Access(&o->d_obj);
  570. LinkedList2Node *node;
  571. while (node = LinkedList2_GetFirst(&o->relay_flows_list)) {
  572. struct dp_relay_flow *flow = UPPER_OBJECT(node, struct dp_relay_flow, source_list_node);
  573. DataProtoDest_PrepareFree(flow->dp);
  574. dealloc_relay_flow(flow);
  575. }
  576. }