Fix a null dereference issue

Change-Id: I50fe239a1590740e8c15a6a1499baf894c7cbdd9

diff --git a/tests/logger/logger.c b/tests/logger/logger.c
index b7c7b72b785d9234ecce56c3f1ddede47468fa11..d2c4231e0c60f3db9757c3da39d475e0793629fc 100644 (file)
--- a/tests/logger/logger.c
+++ b/tests/logger/logger.c
@@ -285,12 +285,20 @@ int main(int ac, char *av[]) {
                        break;
                case 't':
                        tag = strdup(optarg);
+                       if (!tag) {
+                               fprintf(stderr, "not enough memory!\n");
+                               exit(EXIT_FAILURE);
+                       }
                        break;
                case 'm':
                        test_mask = (unsigned long) strtol(optarg, NULL, 16);
                        break;
                case 'd':
                        device = strdup(optarg);
+                       if (!device) {
+                               fprintf(stderr, "not enough memory!\n");
+                               exit(EXIT_FAILURE);
+                       }
                        break;
                case 'D':
                        dump = true;