SwitchPinger.h 4.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127
  1. /* vim: set expandtab ts=4 sw=4: */
  2. /*
  3. * You may redistribute this program and/or modify it under the terms of
  4. * the GNU General Public License as published by the Free Software Foundation,
  5. * either version 3 of the License, or (at your option) any later version.
  6. *
  7. * This program is distributed in the hope that it will be useful,
  8. * but WITHOUT ANY WARRANTY; without even the implied warranty of
  9. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  10. * GNU General Public License for more details.
  11. *
  12. * You should have received a copy of the GNU General Public License
  13. * along with this program. If not, see <http://www.gnu.org/licenses/>.
  14. */
  15. #ifndef SwitchPinger_H
  16. #define SwitchPinger_H
  17. #include "benc/String.h"
  18. #include "dht/Address.h"
  19. #include "crypto/random/Random.h"
  20. #include "interface/Interface.h"
  21. #include "util/events/EventBase.h"
  22. #include "util/log/Log.h"
  23. #include "util/Linker.h"
  24. Linker_require("net/SwitchPinger.c")
  25. #include <stdint.h>
  26. #define SwitchPinger_DEFAULT_MAX_CONCURRENT_PINGS 50
  27. enum SwitchPinger_Result
  28. {
  29. /** Ping responded to ok. */
  30. SwitchPinger_Result_OK,
  31. /** Response label differs from sent label. */
  32. SwitchPinger_Result_LABEL_MISMATCH,
  33. /** Response contains different data than what was sent. */
  34. SwitchPinger_Result_WRONG_DATA,
  35. /** Instead of a normal response, got an error control packet. */
  36. SwitchPinger_Result_ERROR_RESPONSE,
  37. /** A sub-set of ERROR_RESPONSE where the route contains a loop. */
  38. SwitchPinger_Result_LOOP_ROUTE,
  39. /** Ping timeout. */
  40. SwitchPinger_Result_TIMEOUT
  41. };
  42. struct SwitchPinger_Ping;
  43. struct SwitchPinger_Response
  44. {
  45. /** SwitchPinger_Result_OK if all went well, otherwise SwitchPinger_Result_* */
  46. enum SwitchPinger_Result res;
  47. /** the label as of the responding node in host order. */
  48. uint64_t label;
  49. /** the content of the ping response. */
  50. String* data;
  51. /** the number of milliseconds since the original ping was sent. */
  52. uint64_t milliseconds;
  53. /** the version of the node which was pinged. */
  54. uint32_t version;
  55. /** The key for the node which was pinged, if not a keyPing then this is set to 0. */
  56. uint8_t key[32];
  57. struct SwitchPinger_Ping* ping;
  58. };
  59. /** Callback which will be called when the ping response comes back. */
  60. typedef void (* SwitchPinger_ResponseCallback)(struct SwitchPinger_Response* resp, void* userData);
  61. struct SwitchPinger_Ping
  62. {
  63. /**
  64. * The allocator which is set by SwitchPinger_ping() and can be used to
  65. * allocate space which will be freed when the ping completes.
  66. */
  67. struct Allocator* pingAlloc;
  68. /** If true then a key-ping will be sent instead of a legacy ping, default false. */
  69. bool keyPing;
  70. /**
  71. * This is NULL by default and is set by the caller of Pinger_ping(),
  72. * when onResponse() is called, whatever this is, will be given to it.
  73. */
  74. void* onResponseContext;
  75. };
  76. struct SwitchPinger;
  77. String* SwitchPinger_resultString(enum SwitchPinger_Result result);
  78. /**
  79. * Allocate a ping message.
  80. *
  81. * @param label the HOST ORDER label of the node to send the ping message to.
  82. * @param data the content of the ping to send, if NULL, an empty string will be
  83. * returned in the response.
  84. * @param timeoutMilliseconds how long to wait before failing the ping.
  85. * @param onResponse the callback after the on pong or timeout.
  86. * @param alloc free this to cancel the ping.
  87. * @param ctx the pinger
  88. */
  89. struct SwitchPinger_Ping* SwitchPinger_newPing(uint64_t label,
  90. String* data,
  91. uint32_t timeoutMilliseconds,
  92. SwitchPinger_ResponseCallback onResponse,
  93. struct Allocator* alloc,
  94. struct SwitchPinger* ctx);
  95. struct SwitchPinger* SwitchPinger_new(struct Interface* iface,
  96. struct EventBase* eventBase,
  97. struct Random* rand,
  98. struct Log* logger,
  99. struct Address* myAddr,
  100. struct Allocator* alloc);
  101. #endif