shell-test.c revision 075945f611290f2b9db9a4ed6cf5433f2fd85785
1#include <stdio.h>
2#include <stdlib.h>
3#define DBUS_COMPILATION
4#include <dbus/dbus-internals.h>
5#include <dbus/dbus-list.h>
6#include <dbus/dbus-memory.h>
7#include <dbus/dbus-shell.h>
8#include <dbus/dbus-string.h>
9
10static dbus_bool_t
11test_command_line (const char *arg1, ...)
12{
13  int i, original_argc, shell_argc;
14  char **shell_argv;
15  char **original_argv;
16  char *command_line, *tmp;
17  DBusString str;
18  DBusList *list = NULL, *node;
19  va_list var_args;
20  DBusError error;
21
22  va_start (var_args, arg1);
23  _dbus_list_append (&list, arg1);
24  do
25    {
26      tmp = va_arg (var_args, char *);
27      if (!tmp)
28        break;
29      _dbus_list_append (&list, tmp);
30    } while (tmp);
31  va_end (var_args);
32
33  original_argc = _dbus_list_get_length (&list);
34  original_argv = dbus_new (char *, original_argc);
35  _dbus_string_init (&str);
36  for (i = 0, node = _dbus_list_get_first_link (&list); i < original_argc && node;
37       i++, node = _dbus_list_get_next_link (&list, node))
38    {
39      original_argv[i] = node->data;
40      if (i > 0)
41        _dbus_string_append_byte (&str, ' ');
42      _dbus_string_append (&str, original_argv[i]);
43    }
44
45  _dbus_list_clear (&list);
46  command_line = _dbus_string_get_data (&str);
47  printf ("\n\nTesting command line '%s'\n", command_line);
48
49  dbus_error_init (&error);
50  if (!_dbus_shell_parse_argv (command_line, &shell_argc, &shell_argv, &error))
51    {
52      fprintf (stderr, "Error parsing command line: %s\n", error.message ? error.message : "");
53      return FALSE;
54    }
55  else
56    {
57      if (shell_argc != original_argc)
58        {
59          printf ("Number of arguments returned (%d) don't match original (%d)\n",
60                  shell_argc, original_argc);
61          return FALSE;
62        }
63      printf ("Number of arguments: %d\n", shell_argc);
64      for (i = 0; i < shell_argc; i++)
65        {
66          char *unquoted;
67
68          unquoted = _dbus_shell_unquote (original_argv[i]);
69          if (strcmp (unquoted ? unquoted : "",
70                      shell_argv[i] ? shell_argv[i] : ""))
71            {
72              printf ("Position %d, returned argument (%s) does not match original (%s)\n",
73                      i, shell_argv[i], unquoted);
74              dbus_free (unquoted);
75              return FALSE;
76            }
77          dbus_free (unquoted);
78          if (shell_argv[i])
79            printf ("Argument %d = %s\n", i, shell_argv[i]);
80        }
81
82      dbus_free_string_array (shell_argv);
83    }
84
85  _dbus_string_free (&str);
86
87  return TRUE;
88}
89
90int
91main (int argc, char **argv)
92{
93  if (!test_command_line ("command", "-s", "--force-shutdown", "\"a string\"", "123", NULL)
94      || !test_command_line ("command", "-s", NULL)
95      || !test_command_line ("/opt/gnome/bin/service-start", NULL)
96      || !test_command_line ("grep", "-l", "-r", "-i", "'whatever'", "files*.c", NULL)
97      || !test_command_line ("/home/boston/johnp/devel-local/dbus/test/test-segfault", NULL)
98      || !test_command_line ("ls", "-l", "-a", "--colors", "/tmp", NULL)
99      || !test_command_line ("rsync-to-server", NULL)
100      || !test_command_line ("test-segfault", "--no-segfault", NULL)
101      || !test_command_line ("evolution", "mailto:pepe@cuco.com", NULL)
102      || !test_command_line ("run", "\"a \n multiline\"", NULL)
103      || test_command_line ("ls", "\"a wrong string'", NULL) /* invalid command line */ )
104    return -1;
105
106  return 0;
107}
108