test_bpf: avoid oopsing the kernel when generate_test_data() fails.

[firefly-linux-kernel-4.4.55.git] / lib / test_bpf.c

diff --git a/lib/test_bpf.c b/lib/test_bpf.c
index 3afddf2026c983d279fe822a0c42b7f1a65f193a..6843d0b6203295910bf500691f7301cb1b695e93 100644 (file)
--- a/lib/test_bpf.c
+++ b/lib/test_bpf.c
@@ -4672,6 +4672,11 @@ static int run_one(const struct bpf_prog *fp, struct bpf_test *test)
                        break;
 
                data = generate_test_data(test, i);
+               if (!data && !(test->aux & FLAG_NO_DATA)) {
+                       pr_cont("data generation failed ");
+                       err_cnt++;
+                       break;
+               }
                ret = __run_one(fp, data, runs, &duration);
                release_test_data(test, data);