From 746c624a09b3a31e7d82aa8cb945d7a6ea3fdb64 Mon Sep 17 00:00:00 2001
From: iphydf <iphydf@users.noreply.github.com>
Date: Wed, 14 Feb 2018 16:33:08 +0000
Subject: [PATCH] Add static asserts for all the struct sizes in toxcore.

This forces us to record size changes (for LP64) and helps estimate
memory consumption of internal data structures.
---
 auto_tests/monolith_test.cpp | 100 +++++++++++++++++++++++++++++++++++
 1 file changed, 100 insertions(+)

diff --git a/auto_tests/monolith_test.cpp b/auto_tests/monolith_test.cpp
index b1e13448..b09d8985 100644
--- a/auto_tests/monolith_test.cpp
+++ b/auto_tests/monolith_test.cpp
@@ -125,7 +125,107 @@ namespace version_test
 #include "version_test.c"
 }
 
+#define PRINT_SIZE 0
+
+template<typename T, size_t Expected, size_t Actual = sizeof(T)>
+void check_size(char const *type)
+{
+#if PRINT_SIZE
+    printf("CHECK_SIZE(%s, %zu);\n", type, Actual);
+#else
+    static_assert(Actual == Expected, "Bad sizeof - see template expansion errors for details");
+#endif
+}
+
+#define CHECK_SIZE(TYPE, SIZE) check_size<TYPE, SIZE>(#TYPE)
+
+/**
+ * The main function static-asserts that we are aware of all the sizes of all
+ * the structs it toxcore. If you find this failing after you make a change,
+ * switch on the PRINT_SIZE above and copy the number into this function.
+ */
 int main(int argc, char *argv[])
 {
+#if defined(__x86_64__) && defined(__LP64__)
+    // toxcore/DHT
+    CHECK_SIZE(Client_data, 496);
+    CHECK_SIZE(Cryptopacket_Handles, 16);
+    CHECK_SIZE(DHT, 676528);
+    CHECK_SIZE(DHT_Friend, 5104);
+    CHECK_SIZE(Hardening, 144);
+    CHECK_SIZE(IPPTs, 40);
+    CHECK_SIZE(IPPTsPng, 232);
+    CHECK_SIZE(NAT, 48);
+    CHECK_SIZE(Node_format, 64);
+    CHECK_SIZE(Shared_Key, 80);
+    CHECK_SIZE(Shared_Keys, 81920);
+    // toxcore/friend_connection
+    CHECK_SIZE(Friend_Conn, 1784);
+    CHECK_SIZE(Friend_Connections, 72);
+    // toxcore/friend_requests
+    CHECK_SIZE(Friend_Requests, 1080);
+    // toxcore/group
+    CHECK_SIZE(Group_c, 728);
+    CHECK_SIZE(Group_Chats, 2112);
+    CHECK_SIZE(Group_Peer, 480);
+    // toxcore/list
+    CHECK_SIZE(BS_LIST, 32);
+    // toxcore/logger
+    CHECK_SIZE(Logger, 24);
+    // toxcore/Messenger
+    CHECK_SIZE(File_Transfers, 72);
+    CHECK_SIZE(Friend, 39264);
+    CHECK_SIZE(Messenger, 2008);
+    CHECK_SIZE(Messenger_Options, 72);
+    CHECK_SIZE(Receipts, 16);
+    // toxcore/net_crypto
+    CHECK_SIZE(Crypto_Connection, 525392);
+    CHECK_SIZE(Net_Crypto, 272);
+    CHECK_SIZE(New_Connection, 168);
+    CHECK_SIZE(Packet_Data, 1384);
+    CHECK_SIZE(Packets_Array, 262152);
+    // toxcore/network
+    CHECK_SIZE(IP, 24);
+    CHECK_SIZE(IP4, 4);
+#if USE_IPV6
+    CHECK_SIZE(IP6, 16);
+#endif
+    CHECK_SIZE(IP_Port, 32);
+    CHECK_SIZE(Networking_Core, 4112);
+    CHECK_SIZE(Packet_Handler, 16);
+    // toxcore/onion_announce
+    CHECK_SIZE(Cmp_data, 296);
+    CHECK_SIZE(Onion_Announce, 128048);
+    CHECK_SIZE(Onion_Announce_Entry, 288);
+    // toxcore/onion_client
+    CHECK_SIZE(Last_Pinged, 40);
+    CHECK_SIZE(Onion_Client, 15816);
+    CHECK_SIZE(Onion_Client_Cmp_data, 176);
+    CHECK_SIZE(Onion_Client_Paths, 2520);
+    CHECK_SIZE(Onion_Friend, 1936);
+    CHECK_SIZE(Onion_Friend, 1936);
+    CHECK_SIZE(Onion_Node, 168);
+    // toxcore/onion
+    CHECK_SIZE(Onion, 245832);
+    CHECK_SIZE(Onion_Path, 392);
+    // toxcore/ping_array
+    CHECK_SIZE(Ping_Array, 24);
+    CHECK_SIZE(Ping_Array_Entry, 32);
+    // toxcore/ping
+    CHECK_SIZE(Ping, 2072);
+    // toxcore/TCP_client
+    CHECK_SIZE(TCP_Client_Connection, 12064);
+    CHECK_SIZE(TCP_Proxy_Info, 40);
+    // toxcore/TCP_connection
+    CHECK_SIZE(TCP_con, 112);
+    CHECK_SIZE(TCP_Connections, 200);
+    CHECK_SIZE(TCP_Connection_to, 112);
+    // toxcore/TCP_server
+    CHECK_SIZE(TCP_Priority_List, 16);
+    CHECK_SIZE(TCP_Secure_Connection, 11816);
+    CHECK_SIZE(TCP_Server, 6049952);  // 6MB!
+    // toxcore/tox
+    CHECK_SIZE(Tox_Options, 64);
+#endif
     return 0;
 }