Always check the return value of malloc().

[utcp] / test.c

diff --git a/test.c b/test.c
index d7e1cdc5788db3dc8638e3fba93ecd09a292fb0b..3491e296c9a558a6b587d63cec23702f65ec4b21 100644 (file)
--- a/test.c
+++ b/test.c
@@ -73,6 +73,10 @@ ssize_t do_send(struct utcp *utcp, const void *data, size_t len) {
                        return len;
                if(!reorder_data && drand48() < reorder) {
                        reorder_data = malloc(len);
+                       if(!reorder_data) {
+                               debug("Out of memory\n");
+                               return len;
+                       }
                        reorder_len = len;
                        memcpy(reorder_data, data, len);
                        reorder_countdown = 1 + drand48() * reorder_dist;
@@ -113,6 +117,9 @@ int main(int argc, char *argv[]) {
        reorder = atof(getenv("REORDER") ?: "0");
        reorder_dist = atoi(getenv("REORDER_DIST") ?: "10");
 
+       if(dropto < dropfrom)
+               dropto = 1 << 30;
+
        struct addrinfo *ai;
        struct addrinfo hint = {
                .ai_flags = server ? AI_PASSIVE : 0,