70 lines
1.7 KiB
Diff
70 lines
1.7 KiB
Diff
diff --git a/src/programs/pkcheck.c b/src/programs/pkcheck.c
|
|
index f1bb4e1..aff4f60 100644
|
|
--- a/src/programs/pkcheck.c
|
|
+++ b/src/programs/pkcheck.c
|
|
@@ -363,6 +363,12 @@ main (int argc, char *argv[])
|
|
local_agent_handle = NULL;
|
|
ret = 126;
|
|
|
|
+ if (argc < 1)
|
|
+ {
|
|
+ help();
|
|
+ exit(1);
|
|
+ }
|
|
+
|
|
/* Disable remote file access from GIO. */
|
|
setenv ("GIO_USE_VFS", "local", 1);
|
|
|
|
diff --git a/src/programs/pkexec.c b/src/programs/pkexec.c
|
|
index 7698c5c..d84dc57 100644
|
|
--- a/src/programs/pkexec.c
|
|
+++ b/src/programs/pkexec.c
|
|
@@ -488,6 +488,17 @@ main (int argc, char *argv[])
|
|
pid_t pid_of_caller;
|
|
gpointer local_agent_handle;
|
|
|
|
+
|
|
+ /*
|
|
+ * If 'pkexec' is called wrong, just show help and bail out.
|
|
+ */
|
|
+ if (argc<1)
|
|
+ {
|
|
+ clearenv();
|
|
+ usage(argc, argv);
|
|
+ exit(1);
|
|
+ }
|
|
+
|
|
ret = 127;
|
|
authority = NULL;
|
|
subject = NULL;
|
|
@@ -614,10 +625,10 @@ main (int argc, char *argv[])
|
|
|
|
path = g_strdup (pwstruct.pw_shell);
|
|
if (!path)
|
|
- {
|
|
+ {
|
|
g_printerr ("No shell configured or error retrieving pw_shell\n");
|
|
goto out;
|
|
- }
|
|
+ }
|
|
/* If you change this, be sure to change the if (!command_line)
|
|
case below too */
|
|
command_line = g_strdup (path);
|
|
@@ -636,7 +647,15 @@ main (int argc, char *argv[])
|
|
goto out;
|
|
}
|
|
g_free (path);
|
|
- argv[n] = path = s;
|
|
+ path = s;
|
|
+
|
|
+ /* argc<2 and pkexec runs just shell, argv is guaranteed to be null-terminated.
|
|
+ * /-less shell shouldn't happen, but let's be defensive and don't write to null-termination
|
|
+ */
|
|
+ if (argv[n] != NULL)
|
|
+ {
|
|
+ argv[n] = path;
|
|
+ }
|
|
}
|
|
if (access (path, F_OK) != 0)
|
|
{
|