eglib/src/gshell.c

   1 /*
   2  * Shell utility functions.
   3  *
   4  * Author:
   5  *   Gonzalo Paniagua Javier (gonzalo@novell.com
   6  *
   7  * (C) 2006 Novell, Inc.
   8  *
   9  * Permission is hereby granted, free of charge, to any person obtaining
  10  * a copy of this software and associated documentation files (the
  11  * "Software"), to deal in the Software without restriction, including
  12  * without limitation the rights to use, copy, modify, merge, publish,
  13  * distribute, sublicense, and/or sell copies of the Software, and to
  14  * permit persons to whom the Software is furnished to do so, subject to
  15  * the following conditions:
  16  *
  17  * The above copyright notice and this permission notice shall be
  18  * included in all copies or substantial portions of the Software.
  19  *
  20  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
  21  * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
  22  * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
  23  * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
  24  * LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
  25  * OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
  26  * WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
  27  */
  28 #include <stdio.h>
  29 #include <glib.h>
  30
  31 static int
  32 split_cmdline (const gchar *cmdline, GPtrArray *array, GError **error)
  33 {
  34         gchar *ptr;
  35         gchar c;
  36         gboolean in_quote = FALSE;
  37         gboolean escaped = FALSE;
  38         gchar quote_char = '\0';
  39         GString *str;
  40
  41         str = g_string_new ("");
  42         ptr = (gchar *) cmdline;
  43         while ((c = *ptr++) != '\0') {
  44                 if (escaped) {
  45                         escaped = FALSE;
  46                         if (!g_ascii_isspace (c))
  47                                 g_string_append_c (str, c);
  48                 } else if (in_quote) {
  49                         if (c == quote_char) {
  50                                 in_quote = FALSE;
  51                                 quote_char = '\0';
  52                                 g_ptr_array_add (array, g_string_free (str, FALSE));
  53                                 str = g_string_new ("");
  54                         } else {
  55                                 g_string_append_c (str, c);
  56                         }
  57                 } else if (g_ascii_isspace (c)) {
  58                         if (str->len > 0) {
  59                                 g_ptr_array_add (array, g_string_free (str, FALSE));
  60                                 str = g_string_new ("");
  61                         }
  62                 } else if (c == '\\') {
  63                         escaped = TRUE;
  64                 } else if (c == '\'' || c == '"') {
  65                         in_quote = TRUE;
  66                         quote_char = c;
  67                 } else {
  68                         g_string_append_c (str, c);
  69                 }
  70         }
  71
  72         if (escaped) {
  73                 if (error)
  74                         *error = g_error_new (G_LOG_DOMAIN, 0, "Unfinished escape.");
  75                 g_string_free (str, TRUE);
  76                 return -1;
  77         }
  78
  79         if (in_quote) {
  80                 if (error)
  81                         *error = g_error_new (G_LOG_DOMAIN, 0, "Unfinished quote.");
  82                 g_string_free (str, TRUE);
  83                 return -1;
  84         }
  85
  86         if (str->len > 0) {
  87                 g_ptr_array_add (array, g_string_free (str, FALSE));
  88         } else {
  89                 g_string_free (str, TRUE);
  90         }
  91         g_ptr_array_add (array, NULL);
  92         return 0;
  93 }
  94
  95 gboolean
  96 g_shell_parse_argv (const gchar *command_line, gint *argcp, gchar ***argvp, GError **error)
  97 {
  98         GPtrArray *array;
  99         gint argc;
 100         gchar **argv;
 101
 102         g_return_val_if_fail (command_line, FALSE);
 103         g_return_val_if_fail (error == NULL || *error == NULL, FALSE);
 104
 105         array = g_ptr_array_new();
 106         if (split_cmdline (command_line, array, error)) {
 107                 g_ptr_array_add (array, NULL);
 108                 g_strfreev ((gchar **) array->pdata);
 109                 g_ptr_array_free (array, FALSE);
 110                 return FALSE;
 111         }
 112
 113         argc = array->len;
 114         argv = (gchar **) array->pdata;
 115
 116         if (argc == 1) {
 117                 g_strfreev (argv);
 118                 g_ptr_array_free (array, FALSE);
 119                 return FALSE;
 120         }
 121
 122         if (argcp) {
 123                 *argcp = array->len - 1;
 124         }
 125
 126         if (argvp) {
 127                 *argvp = argv;
 128         } else {
 129                 g_strfreev (argv);
 130         }
 131
 132         g_ptr_array_free (array, FALSE);
 133         return TRUE;
 134 }
 135
 136 gchar *
 137 g_shell_quote (const gchar *unquoted_string)
 138 {
 139         GString *result = g_string_new ("'");
 140         const gchar *p;
 141
 142         for (p = unquoted_string; *p; p++){
 143                 if (*p == '\'')
 144                         g_string_append (result, "'\\'");
 145                 g_string_append_c (result, *p);
 146         }
 147         g_string_append_c (result, '\'');
 148         return g_string_free (result, FALSE);
 149 }
 150
 151 gchar *
 152 g_shell_unquote (const gchar *quoted_string, GError **error)
 153 {
 154         GString *result;
 155         const char *p;
 156         int do_unquote = 0;
 157
 158         /* Quickly try to determine if we need to unquote or not */
 159         for (p = quoted_string; *p; p++){
 160                 if (*p == '\'' || *p == '"' || *p == '\\'){
 161                         do_unquote = 1;
 162                         break;
 163                 }
 164         }
 165
 166         if (!do_unquote)
 167                 return g_strdup (quoted_string);
 168
 169         /* We do need to unquote */
 170         result = g_string_new ("");
 171         for (p = quoted_string; *p; p++){
 172
 173                 if (*p == '\''){
 174                         /* Process single quote, not even \ is processed by glib's version */
 175                         for (p++; *p; p++){
 176                                 if (*p == '\'')
 177                                         break;
 178                                 g_string_append_c (result, *p);
 179                         }
 180                         if (!*p){
 181                                 g_set_error (error, 0, 0, "Open quote");
 182                                 return NULL;
 183                         }
 184                 } else if (*p == '"'){
 185                         /* Process double quote, allows some escaping */
 186                         for (p++; *p; p++){
 187                                 if (*p == '"')
 188                                         break;
 189                                 if (*p == '\\'){
 190                                         p++;
 191                                         if (*p == 0){
 192                                                 g_set_error (error, 0, 0, "Open quote");
 193                                                 return NULL;
 194                                         }
 195                                         int append = -1;
 196                                         switch (*p){
 197                                         case '$':
 198                                         case '"':
 199                                         case '\\':
 200                                         case '`':
 201                                                 break;
 202                                         default:
 203                                                 g_string_append_c (result, '\\');
 204                                                 break;
 205                                         }
 206                                 }
 207                                 g_string_append_c (result, *p);
 208                         }
 209                         if (!*p){
 210                                 g_set_error (error, 0, 0, "Open quote");
 211                                 return NULL;
 212                         }
 213                 } else if (*p == '\\'){
 214                         p++;
 215                         char c = *p;
 216                         if (!(c == '$' || c == '"' || c == '\\' || c == '`' || c == 0))
 217                                 g_string_append_c (result, '\\');
 218                         if (c == 0)
 219                                 break;
 220                         else
 221                                 g_string_append_c (result, c);
 222                 } else
 223                         g_string_append_c (result, *p);
 224         }
 225         return g_string_free (result, FALSE);
 226 }
 227
 228 #if JOINT_TEST
 229 /*
 230  * This test is designed to be built with the 2 glib/eglib to compare
 231  */
 232
 233 char *args [] = {
 234         "\\",
 235         "\"Foo'bar\"",
 236         "'foo'",
 237         "'fo\'b'",
 238         "'foo\"bar'",
 239         "'foo' dingus bar",
 240         "'foo' 'bar' 'baz'",
 241         "\"foo\" 'bar' \"baz\"",
 242         "\"f\\$\\\'",
 243         "\"\\",
 244         "\\\\",
 245         "'\\\\'",
 246         "\"f\\$\"\\\"\\\\", //  /\\\"\\\\"
 247         "'f\\$'\\\"\\\\",
 248         "'f\\$\\\\'",
 249         NULL
 250 };
 251
 252
 253 int
 254 main ()
 255 {
 256         char **s = args;
 257         int i;
 258
 259         while (*s){
 260                 char *r1 = g_shell_unquote (*s, NULL);
 261                 char *r2 = g2_shell_unquote (*s, NULL);
 262                 char *ok = r1 == r2 ? "ok" : (r1 != NULL && r2 != NULL && strcmp (r1, r2) == 0) ? "ok" : "fail";
 263
 264                 printf ("%s [%s] -> [%s] - [%s]\n", ok, *s, r1, r2);
 265                 s++;
 266         }
 267         return;
 268         char buffer [10];
 269         buffer [0] = '\"';
 270         buffer [1] = '\\';
 271         buffer [3] = '\"';
 272         buffer [4] = 0;
 273
 274         for (i = 32; i < 255; i++){
 275                 buffer [2] = i;
 276                 printf ("%d [%s] -> [%s]\n", i, buffer, g_shell_unquote (buffer, NULL));
 277         }
 278 }
 279 #endif