Exposed and tested disconnect notification TCP packets.

2 files changed, 21 insertions, 0 deletions

diff --git a/toxcore/TCP_client.c b/toxcore/TCP_client.c
index 4652b4a6..209c1775 100644
--- a/toxcore/TCP_client.c
+++ b/toxcore/TCP_client.c
@@ -304,6 +304,20 @@ static int send_ping_response(TCP_Client_Connection *con, uint64_t ping_id)
  * return 0 if could not send packet.
  * return -1 on failure (connection must be killed).
  */
+int send_disconnect_request(TCP_Client_Connection *con, uint8_t con_id)
+{
+    if (con_id >= NUM_CLIENT_CONNECTIONS)
+        return -1;
+
+    con->connections[con_id].status = 0;
+    con->connections[con_id].number = 0;
+    return send_disconnect_notification(con, con_id + NUM_RESERVED_PORTS);
+}
+
+/* return 1 on success.
+ * return 0 if could not send packet.
+ * return -1 on failure (connection must be killed).
+ */
 int send_onion_request(TCP_Client_Connection *con, uint8_t *data, uint16_t length)
 {
     uint8_t packet[1 + length];
@@ -564,6 +578,7 @@ void do_TCP_connection(TCP_Client_Connection *TCP_connection)
 
         if (sizeof(data) == len) {
             if (handle_handshake(TCP_connection, data) == 0) {
+                TCP_connection->kill_at = ~0;
                 TCP_connection->status = TCP_CLIENT_CONFIRMED;
             } else {
                 TCP_connection->kill_at = 0;
diff --git a/toxcore/TCP_client.h b/toxcore/TCP_client.h
index 277378b9..768ef9d2 100644
--- a/toxcore/TCP_client.h
+++ b/toxcore/TCP_client.h
@@ -105,6 +105,12 @@ void routing_response_handler(TCP_Client_Connection *con, int (*response_callbac
 void routing_status_handler(TCP_Client_Connection *con, int (*status_callback)(void *object, uint32_t number,
                             uint8_t connection_id, uint8_t status), void *object);
 
+/* return 1 on success.
+ * return 0 if could not send packet.
+ * return -1 on failure (connection must be killed).
+ */
+int send_disconnect_request(TCP_Client_Connection *con, uint8_t con_id);
+
 /* Set the number that will be used as an argument in the callbacks related to con_id.
  *
  * When not set by this function, the number is ~0.