Modified Lossless_UDP connections to work with tox_array.

1 files changed, 26 insertions, 12 deletions

diff --git a/testing/misc_tools.h b/testing/misc_tools.h
index 7c37ba38..3b71d8c4 100644
--- a/testing/misc_tools.h
+++ b/testing/misc_tools.h
@@ -117,7 +117,7 @@ static inline void tox_list_remove(tox_list *lst)
  ************************************************************/
 
 typedef struct tox_array {
-    void *data;
+    void *data; /* last elem is data[len-1] */
     uint32_t len;
     size_t elem_size; /* in bytes */
 } tox_array;
@@ -135,34 +135,48 @@ static inline void tox_array_delete(tox_array *arr)
     arr->len = arr->elem_size = 0;
 }
 
-static inline void _tox_array_push(tox_array *arr, uint8_t *item)
+static inline uint8_t tox_array_push_ptr(tox_array *arr, uint8_t *item)
 {
     arr->data = realloc(arr->data, arr->elem_size * (arr->len+1));
 
-    memcpy(arr->data + arr->elem_size*arr->len, item, arr->elem_size);
+    if (arr->data == NULL)
+        return 0;
+
+    if (item != NULL)
+        memcpy(arr->data + arr->elem_size*arr->len, item, arr->elem_size);
     arr->len++;
+
+    return 1;
 }
-#define tox_array_push(arr, item) _tox_array_push(arr, (void*)(&(item)))
+#define tox_array_push(arr, item) tox_array_push_ptr(arr, (uint8_t*)(&(item)))
 
 /* Deletes num items from array.
  * Not same as pop in stacks, because to access elements you use data.
  */
 static inline void tox_array_pop(tox_array *arr, uint32_t num)
 {
-    arr->len--;
+    if (num == 0)
+        num = 1;
+    arr->len -= num;
     arr->data = realloc(arr->data, arr->elem_size*arr->len);
 }
 
+/* TODO: do not use type, since we track the needed info in
+ * elem_size (but array user will have to cast)
+ */
 #define tox_array_get(arr, i, type) (((type*)(arr)->data)[i])
 
-/* TODO: what about nested for loops? */
-#define tox_array_for_each(arr, type) \
+/* This version requires C99 (declaring variables inside for loop)
+#define tox_array_for_each(arr, type, tmp_name) \
     for ( \
-         struct { type tmp; uint32_t i; } tox_array_tmp = { tox_array_get(arr, 0, type), 0 }; \
-         tox_array_tmp.i != (arr)->len; \
-         tox_array_tmp.tmp = tox_array_get(arr, ++tox_array_tmp.i, type) \
+         struct { type val; uint32_t i; } tmp_name = { tox_array_get(arr, 0, type), 0 }; \
+         tmp_name.i != (arr)->len; \
+         tmp_name.val = tox_array_get(arr, ++tmp_name.i, type) \
         )
+*/
 
-#define TOX_ARRAY_TMP (tox_array_tmp.tmp)
+#define tox_array_for_each(arr, type, tmp_name) \
+    type tmp_name; uint32_t tmp_name ## _i = 0; \
+    for (; tmp_name ## _i != (arr)->len; tmp_name = tox_array_get(arr, ++ tmp_name ## _i, type))
 
-#endif // MISC_TOOLS_H
\ No newline at end of file
+#endif // MISC_TOOLS_H