X-Git-Url: https://repo.jachan.dev/linux.git/blobdiff_plain/a8dcf12f9e8ba113251c04d602f1695d537fb142..b5b5150977cdfb7173f4468ccd30f9b4a981b902:/lib/argv_split.c

diff --git a/lib/argv_split.c b/lib/argv_split.c
index 4096ed42f490..5205a8dae5bc 100644
--- a/lib/argv_split.c
+++ b/lib/argv_split.c
@@ -4,7 +4,8 @@
 
 #include <linux/kernel.h>
 #include <linux/ctype.h>
-#include <linux/bug.h>
+#include <linux/slab.h>
+#include <linux/module.h>
 
 static const char *skip_sep(const char *cp)
 {
@@ -75,7 +76,9 @@ char **argv_split(gfp_t gfp, const char *str, int *argcp)
 	if (argv == NULL)
 		goto out;
 
-	*argcp = argc;
+	if (argcp)
+		*argcp = argc;
+
 	argvp = argv;
 
 	while (*str) {