Be a bit more robust in the face of out-of-memory errors.

[platform/upstream/libexif.git] / test / test-mem.c

diff --git a/test/test-mem.c b/test/test-mem.c
index defcee9..e94df35 100644 (file)
--- a/test/test-mem.c
+++ b/test/test-mem.c
@@ -39,6 +39,11 @@ main ()
 
        printf ("Creating EXIF data...\n");
        ed = exif_data_new ();
+       if (!ed) {
+               fprintf(stderr, "Out of memory\n");
+               exit(13);
+       }
+
        exif_data_set_data_type (ed, EXIF_DATA_TYPE_UNCOMPRESSED_CHUNKY);
 
        printf ("Fill EXIF data with all necessary entries to follow specs...\n");
@@ -47,6 +52,11 @@ main ()
        /* Add a dummy thumbnail */
        ed->size = 4;
        ed->data = calloc(1, ed->size);
+       if (!ed->data) {
+               fprintf(stderr, "Out of memory\n");
+               exif_data_unref (ed);
+               exit(13);
+       }
 
        exif_data_dump (ed);
 
@@ -56,6 +66,11 @@ main ()
 
        printf ("Writing %i byte(s) EXIF data to loader...\n", ebs);
        loader = exif_loader_new ();
+       if (!loader) {
+               fprintf(stderr, "Out of memory\n");
+               free (eb);
+               exit(13);
+       }
        size[0] = (unsigned char) ebs;
        size[1] = (unsigned char) (ebs >> 8);
        exif_loader_write (loader, size, 2);
@@ -63,6 +78,10 @@ main ()
        printf ("Wrote %i byte(s).\n", i);
        free (eb);
        ed = exif_loader_get_data (loader);
+       if (!ed) {
+               fprintf(stderr, "Out of memory\n");
+               exit(13);
+       }
        exif_loader_unref (loader);
        exif_data_dump (ed);
        exif_data_unref (ed);