tools:iio: catch errors in string allocation

[platform/kernel/linux-starfive.git] / tools / iio / generic_buffer.c

diff --git a/tools/iio/generic_buffer.c b/tools/iio/generic_buffer.c
index 4cd2464..eb89bc2 100644 (file)
--- a/tools/iio/generic_buffer.c
+++ b/tools/iio/generic_buffer.c
@@ -234,7 +234,9 @@ int main(int argc, char **argv)
        }
        printf("iio device number being used is %d\n", dev_num);
 
-       asprintf(&dev_dir_name, "%siio:device%d", iio_dir, dev_num);
+       ret = asprintf(&dev_dir_name, "%siio:device%d", iio_dir, dev_num);
+       if (ret < 0)
+               return -ENOMEM;
 
        if (!notrigger) {
                if (trigger_name == NULL) {