utp_callbacks.cpp 6.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208
  1. // vim:set ts=4 sw=4 ai:
  2. /*
  3. * Copyright (c) 2010-2013 BitTorrent, Inc.
  4. *
  5. * Permission is hereby granted, free of charge, to any person obtaining a copy
  6. * of this software and associated documentation files (the "Software"), to deal
  7. * in the Software without restriction, including without limitation the rights
  8. * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
  9. * copies of the Software, and to permit persons to whom the Software is
  10. * furnished to do so, subject to the following conditions:
  11. *
  12. * The above copyright notice and this permission notice shall be included in
  13. * all copies or substantial portions of the Software.
  14. *
  15. * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  16. * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  17. * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  18. * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  19. * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  20. * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
  21. * THE SOFTWARE.
  22. */
  23. #include "utp_callbacks.h"
  24. int utp_call_on_firewall(utp_context *ctx, const struct sockaddr *address, socklen_t address_len)
  25. {
  26. utp_callback_arguments args;
  27. if (!ctx->callbacks[UTP_ON_FIREWALL]) return 0;
  28. args.callback_type = UTP_ON_FIREWALL;
  29. args.context = ctx;
  30. args.socket = NULL;
  31. args.address = address;
  32. args.address_len = address_len;
  33. return (int)ctx->callbacks[UTP_ON_FIREWALL](&args);
  34. }
  35. void utp_call_on_accept(utp_context *ctx, utp_socket *socket, const struct sockaddr *address, socklen_t address_len)
  36. {
  37. utp_callback_arguments args;
  38. if (!ctx->callbacks[UTP_ON_ACCEPT]) return;
  39. args.callback_type = UTP_ON_ACCEPT;
  40. args.context = ctx;
  41. args.socket = socket;
  42. args.address = address;
  43. args.address_len = address_len;
  44. ctx->callbacks[UTP_ON_ACCEPT](&args);
  45. }
  46. void utp_call_on_connect(utp_context *ctx, utp_socket *socket)
  47. {
  48. utp_callback_arguments args;
  49. if (!ctx->callbacks[UTP_ON_CONNECT]) return;
  50. args.callback_type = UTP_ON_CONNECT;
  51. args.context = ctx;
  52. args.socket = socket;
  53. ctx->callbacks[UTP_ON_CONNECT](&args);
  54. }
  55. void utp_call_on_error(utp_context *ctx, utp_socket *socket, int error_code)
  56. {
  57. utp_callback_arguments args;
  58. if (!ctx->callbacks[UTP_ON_ERROR]) return;
  59. args.callback_type = UTP_ON_ERROR;
  60. args.context = ctx;
  61. args.socket = socket;
  62. args.error_code = error_code;
  63. ctx->callbacks[UTP_ON_ERROR](&args);
  64. }
  65. void utp_call_on_read(utp_context *ctx, utp_socket *socket, const byte *buf, size_t len)
  66. {
  67. utp_callback_arguments args;
  68. if (!ctx->callbacks[UTP_ON_READ]) return;
  69. args.callback_type = UTP_ON_READ;
  70. args.context = ctx;
  71. args.socket = socket;
  72. args.buf = buf;
  73. args.len = len;
  74. ctx->callbacks[UTP_ON_READ](&args);
  75. }
  76. void utp_call_on_overhead_statistics(utp_context *ctx, utp_socket *socket, int send, size_t len, int type)
  77. {
  78. utp_callback_arguments args;
  79. if (!ctx->callbacks[UTP_ON_OVERHEAD_STATISTICS]) return;
  80. args.callback_type = UTP_ON_OVERHEAD_STATISTICS;
  81. args.context = ctx;
  82. args.socket = socket;
  83. args.send = send;
  84. args.len = len;
  85. args.type = type;
  86. ctx->callbacks[UTP_ON_OVERHEAD_STATISTICS](&args);
  87. }
  88. void utp_call_on_delay_sample(utp_context *ctx, utp_socket *socket, int sample_ms)
  89. {
  90. utp_callback_arguments args;
  91. if (!ctx->callbacks[UTP_ON_DELAY_SAMPLE]) return;
  92. args.callback_type = UTP_ON_DELAY_SAMPLE;
  93. args.context = ctx;
  94. args.socket = socket;
  95. args.sample_ms = sample_ms;
  96. ctx->callbacks[UTP_ON_DELAY_SAMPLE](&args);
  97. }
  98. void utp_call_on_state_change(utp_context *ctx, utp_socket *socket, int state)
  99. {
  100. utp_callback_arguments args;
  101. if (!ctx->callbacks[UTP_ON_STATE_CHANGE]) return;
  102. args.callback_type = UTP_ON_STATE_CHANGE;
  103. args.context = ctx;
  104. args.socket = socket;
  105. args.state = state;
  106. ctx->callbacks[UTP_ON_STATE_CHANGE](&args);
  107. }
  108. uint16 utp_call_get_udp_mtu(utp_context *ctx, utp_socket *socket, const struct sockaddr *address, socklen_t address_len)
  109. {
  110. utp_callback_arguments args;
  111. if (!ctx->callbacks[UTP_GET_UDP_MTU]) return 0;
  112. args.callback_type = UTP_GET_UDP_MTU;
  113. args.context = ctx;
  114. args.socket = socket;
  115. args.address = address;
  116. args.address_len = address_len;
  117. return (uint16)ctx->callbacks[UTP_GET_UDP_MTU](&args);
  118. }
  119. uint16 utp_call_get_udp_overhead(utp_context *ctx, utp_socket *socket, const struct sockaddr *address, socklen_t address_len)
  120. {
  121. utp_callback_arguments args;
  122. if (!ctx->callbacks[UTP_GET_UDP_OVERHEAD]) return 0;
  123. args.callback_type = UTP_GET_UDP_OVERHEAD;
  124. args.context = ctx;
  125. args.socket = socket;
  126. args.address = address;
  127. args.address_len = address_len;
  128. return (uint16)ctx->callbacks[UTP_GET_UDP_OVERHEAD](&args);
  129. }
  130. uint64 utp_call_get_milliseconds(utp_context *ctx, utp_socket *socket)
  131. {
  132. utp_callback_arguments args;
  133. if (!ctx->callbacks[UTP_GET_MILLISECONDS]) return 0;
  134. args.callback_type = UTP_GET_MILLISECONDS;
  135. args.context = ctx;
  136. args.socket = socket;
  137. return ctx->callbacks[UTP_GET_MILLISECONDS](&args);
  138. }
  139. uint64 utp_call_get_microseconds(utp_context *ctx, utp_socket *socket)
  140. {
  141. utp_callback_arguments args;
  142. if (!ctx->callbacks[UTP_GET_MICROSECONDS]) return 0;
  143. args.callback_type = UTP_GET_MICROSECONDS;
  144. args.context = ctx;
  145. args.socket = socket;
  146. return ctx->callbacks[UTP_GET_MICROSECONDS](&args);
  147. }
  148. uint32 utp_call_get_random(utp_context *ctx, utp_socket *socket)
  149. {
  150. utp_callback_arguments args;
  151. if (!ctx->callbacks[UTP_GET_RANDOM]) return 0;
  152. args.callback_type = UTP_GET_RANDOM;
  153. args.context = ctx;
  154. args.socket = socket;
  155. return (uint32)ctx->callbacks[UTP_GET_RANDOM](&args);
  156. }
  157. size_t utp_call_get_read_buffer_size(utp_context *ctx, utp_socket *socket)
  158. {
  159. utp_callback_arguments args;
  160. if (!ctx->callbacks[UTP_GET_READ_BUFFER_SIZE]) return 0;
  161. args.callback_type = UTP_GET_READ_BUFFER_SIZE;
  162. args.context = ctx;
  163. args.socket = socket;
  164. return (size_t)ctx->callbacks[UTP_GET_READ_BUFFER_SIZE](&args);
  165. }
  166. void utp_call_log(utp_context *ctx, utp_socket *socket, const byte *buf)
  167. {
  168. utp_callback_arguments args;
  169. if (!ctx->callbacks[UTP_LOG]) return;
  170. args.callback_type = UTP_LOG;
  171. args.context = ctx;
  172. args.socket = socket;
  173. args.buf = buf;
  174. ctx->callbacks[UTP_LOG](&args);
  175. }
  176. void utp_call_sendto(utp_context *ctx, utp_socket *socket, const byte *buf, size_t len, const struct sockaddr *address, socklen_t address_len, uint32 flags)
  177. {
  178. utp_callback_arguments args;
  179. if (!ctx->callbacks[UTP_SENDTO]) return;
  180. args.callback_type = UTP_SENDTO;
  181. args.context = ctx;
  182. args.socket = socket;
  183. args.buf = buf;
  184. args.len = len;
  185. args.address = address;
  186. args.address_len = address_len;
  187. args.flags = flags;
  188. ctx->callbacks[UTP_SENDTO](&args);
  189. }