windows/winutils.c

   1 /*
   2  * winutils.c: miscellaneous Windows utilities for GUI apps
   3  */
   4
   5 #include <stdio.h>
   6 #include <stdlib.h>
   7 #include <ctype.h>
   8
   9 #include "putty.h"
  10 #include "misc.h"
  11
  12 #ifdef TESTMODE
  13 /* Definitions to allow this module to be compiled standalone for testing
  14  * split_into_argv(). */
  15 #define smalloc malloc
  16 #define srealloc realloc
  17 #define sfree free
  18 #endif
  19
  20 /*
  21  * GetOpenFileName/GetSaveFileName tend to muck around with the process'
  22  * working directory on at least some versions of Windows.
  23  * Here's a wrapper that gives more control over this, and hides a little
  24  * bit of other grottiness.
  25  */
  26
  27 struct filereq_tag {
  28     TCHAR cwd[MAX_PATH];
  29 };
  30
  31 /*
  32  * `of' is expected to be initialised with most interesting fields, but
  33  * this function does some administrivia. (assume `of' was memset to 0)
  34  * save==1 -> GetSaveFileName; save==0 -> GetOpenFileName
  35  * `state' is optional.
  36  */
  37 BOOL request_file(filereq *state, OPENFILENAME *of, int preserve, int save)
  38 {
  39     TCHAR cwd[MAX_PATH]; /* process CWD */
  40     BOOL ret;
  41
  42     /* Get process CWD */
  43     if (preserve) {
  44         DWORD r = GetCurrentDirectory(lenof(cwd), cwd);
  45         if (r == 0 || r >= lenof(cwd))
  46             /* Didn't work, oh well. Stop trying to be clever. */
  47             preserve = 0;
  48     }
  49
  50     /* Open the file requester, maybe setting lpstrInitialDir */
  51     {
  52 #ifdef OPENFILENAME_SIZE_VERSION_400
  53         of->lStructSize = OPENFILENAME_SIZE_VERSION_400;
  54 #else
  55         of->lStructSize = sizeof(*of);
  56 #endif
  57         of->lpstrInitialDir = (state && state->cwd[0]) ? state->cwd : NULL;
  58         /* Actually put up the requester. */
  59         ret = save ? GetSaveFileName(of) : GetOpenFileName(of);
  60     }
  61
  62     /* Get CWD left by requester */
  63     if (state) {
  64         DWORD r = GetCurrentDirectory(lenof(state->cwd), state->cwd);
  65         if (r == 0 || r >= lenof(state->cwd))
  66             /* Didn't work, oh well. */
  67             state->cwd[0] = '\0';
  68     }
  69
  70     /* Restore process CWD */
  71     if (preserve)
  72         /* If it fails, there's not much we can do. */
  73         (void) SetCurrentDirectory(cwd);
  74
  75     return ret;
  76 }
  77
  78 filereq *filereq_new(void)
  79 {
  80     filereq *ret = snew(filereq);
  81     ret->cwd[0] = '\0';
  82     return ret;
  83 }
  84
  85 void filereq_free(filereq *state)
  86 {
  87     sfree(state);
  88 }
  89
  90 /*
  91  * Message box with optional context help.
  92  */
  93
  94 /* Callback function to launch context help. */
  95 static VOID CALLBACK message_box_help_callback(LPHELPINFO lpHelpInfo)
  96 {
  97     const char *context = NULL;
  98 #define CHECK_CTX(name) \
  99     do { \
 100         if (lpHelpInfo->dwContextId == WINHELP_CTXID_ ## name) \
 101             context = WINHELP_CTX_ ## name; \
 102     } while (0)
 103     CHECK_CTX(errors_hostkey_absent);
 104     CHECK_CTX(errors_hostkey_changed);
 105     CHECK_CTX(errors_cantloadkey);
 106     CHECK_CTX(option_cleanup);
 107     CHECK_CTX(pgp_fingerprints);
 108 #undef CHECK_CTX
 109     if (context)
 110         launch_help(hwnd, context);
 111 }
 112
 113 int message_box(LPCTSTR text, LPCTSTR caption, DWORD style, DWORD helpctxid)
 114 {
 115     MSGBOXPARAMS mbox;
 116
 117     /*
 118      * We use MessageBoxIndirect() because it allows us to specify a
 119      * callback function for the Help button.
 120      */
 121     mbox.cbSize = sizeof(mbox);
 122     /* Assumes the globals `hinst' and `hwnd' have sensible values. */
 123     mbox.hInstance = hinst;
 124     mbox.hwndOwner = hwnd;
 125     mbox.lpfnMsgBoxCallback = &message_box_help_callback;
 126     mbox.dwLanguageId = LANG_NEUTRAL;
 127     mbox.lpszText = text;
 128     mbox.lpszCaption = caption;
 129     mbox.dwContextHelpId = helpctxid;
 130     mbox.dwStyle = style;
 131     if (helpctxid != 0 && has_help()) mbox.dwStyle |= MB_HELP;
 132     return MessageBoxIndirect(&mbox);
 133 }
 134
 135 /*
 136  * Display the fingerprints of the PGP Master Keys to the user.
 137  */
 138 void pgp_fingerprints(void)
 139 {
 140     message_box("These are the fingerprints of the PuTTY PGP Master Keys. They can\n"
 141                 "be used to establish a trust path from this executable to another\n"
 142                 "one. See the manual for more information.\n"
 143                 "(Note: these fingerprints have nothing to do with SSH!)\n"
 144                 "\n"
 145                 "PuTTY Master Key as of 2015 (RSA, 4096-bit):\n"
 146                 "  " PGP_MASTER_KEY_FP "\n\n"
 147                 "Original PuTTY Master Key (RSA, 1024-bit):\n"
 148                 "  " PGP_RSA_MASTER_KEY_FP "\n"
 149                 "Original PuTTY Master Key (DSA, 1024-bit):\n"
 150                 "  " PGP_DSA_MASTER_KEY_FP,
 151                 "PGP fingerprints", MB_ICONINFORMATION | MB_OK,
 152                 HELPCTXID(pgp_fingerprints));
 153 }
 154
 155 /*
 156  * Handy wrapper around GetDlgItemText which doesn't make you invent
 157  * an arbitrary length limit on the output string. Returned string is
 158  * dynamically allocated; caller must free.
 159  */
 160 char *GetDlgItemText_alloc(HWND hwnd, int id)
 161 {
 162     char *ret = NULL;
 163     int size = 0;
 164
 165     do {
 166         size = size * 4 / 3 + 512;
 167         ret = sresize(ret, size, char);
 168         GetDlgItemText(hwnd, id, ret, size);
 169     } while (!memchr(ret, '\0', size-1));
 170
 171     return ret;
 172 }
 173
 174 /*
 175  * Split a complete command line into argc/argv, attempting to do it
 176  * exactly the same way the Visual Studio C library would do it (so
 177  * that our console utilities, which receive argc and argv already
 178  * broken apart by the C library, will have their command lines
 179  * processed in the same way as the GUI utilities which get a whole
 180  * command line and must call this function).
 181  *
 182  * Does not modify the input command line.
 183  *
 184  * The final parameter (argstart) is used to return a second array
 185  * of char * pointers, the same length as argv, each one pointing
 186  * at the start of the corresponding element of argv in the
 187  * original command line. So if you get half way through processing
 188  * your command line in argc/argv form and then decide you want to
 189  * treat the rest as a raw string, you can. If you don't want to,
 190  * `argstart' can be safely left NULL.
 191  */
 192 void split_into_argv(char *cmdline, int *argc, char ***argv,
 193                      char ***argstart)
 194 {
 195     char *p;
 196     char *outputline, *q;
 197     char **outputargv, **outputargstart;
 198     int outputargc;
 199
 200     /*
 201      * These argument-breaking rules apply to Visual Studio 7, which
 202      * is currently the compiler expected to be used for PuTTY. Visual
 203      * Studio 10 has different rules, lacking the curious mod 3
 204      * behaviour of consecutive quotes described below; I presume they
 205      * fixed a bug. As and when we migrate to a newer compiler, we'll
 206      * have to adjust this to match; however, for the moment we
 207      * faithfully imitate in our GUI utilities what our CLI utilities
 208      * can't be prevented from doing.
 209      *
 210      * When I investigated this, at first glance the rules appeared to
 211      * be:
 212      *
 213      *  - Single quotes are not special characters.
 214      *
 215      *  - Double quotes are removed, but within them spaces cease
 216      *    to be special.
 217      *
 218      *  - Backslashes are _only_ special when a sequence of them
 219      *    appear just before a double quote. In this situation,
 220      *    they are treated like C backslashes: so \" just gives a
 221      *    literal quote, \\" gives a literal backslash and then
 222      *    opens or closes a double-quoted segment, \\\" gives a
 223      *    literal backslash and then a literal quote, \\\\" gives
 224      *    two literal backslashes and then opens/closes a
 225      *    double-quoted segment, and so forth. Note that this
 226      *    behaviour is identical inside and outside double quotes.
 227      *
 228      *  - Two successive double quotes become one literal double
 229      *    quote, but only _inside_ a double-quoted segment.
 230      *    Outside, they just form an empty double-quoted segment
 231      *    (which may cause an empty argument word).
 232      *
 233      *  - That only leaves the interesting question of what happens
 234      *    when one or more backslashes precedes two or more double
 235      *    quotes, starting inside a double-quoted string. And the
 236      *    answer to that appears somewhat bizarre. Here I tabulate
 237      *    number of backslashes (across the top) against number of
 238      *    quotes (down the left), and indicate how many backslashes
 239      *    are output, how many quotes are output, and whether a
 240      *    quoted segment is open at the end of the sequence:
 241      *
 242      *                      backslashes
 243      *
 244      *               0         1      2      3      4
 245      *
 246      *         0   0,0,y  |  1,0,y  2,0,y  3,0,y  4,0,y
 247      *            --------+-----------------------------
 248      *         1   0,0,n  |  0,1,y  1,0,n  1,1,y  2,0,n
 249      *    q    2   0,1,n  |  0,1,n  1,1,n  1,1,n  2,1,n
 250      *    u    3   0,1,y  |  0,2,n  1,1,y  1,2,n  2,1,y
 251      *    o    4   0,1,n  |  0,2,y  1,1,n  1,2,y  2,1,n
 252      *    t    5   0,2,n  |  0,2,n  1,2,n  1,2,n  2,2,n
 253      *    e    6   0,2,y  |  0,3,n  1,2,y  1,3,n  2,2,y
 254      *    s    7   0,2,n  |  0,3,y  1,2,n  1,3,y  2,2,n
 255      *         8   0,3,n  |  0,3,n  1,3,n  1,3,n  2,3,n
 256      *         9   0,3,y  |  0,4,n  1,3,y  1,4,n  2,3,y
 257      *        10   0,3,n  |  0,4,y  1,3,n  1,4,y  2,3,n
 258      *        11   0,4,n  |  0,4,n  1,4,n  1,4,n  2,4,n
 259      *
 260      *
 261      *      [Test fragment was of the form "a\\\"""b c" d.]
 262      *
 263      * There is very weird mod-3 behaviour going on here in the
 264      * number of quotes, and it even applies when there aren't any
 265      * backslashes! How ghastly.
 266      *
 267      * With a bit of thought, this extremely odd diagram suddenly
 268      * coalesced itself into a coherent, if still ghastly, model of
 269      * how things work:
 270      *
 271      *  - As before, backslashes are only special when one or more
 272      *    of them appear contiguously before at least one double
 273      *    quote. In this situation the backslashes do exactly what
 274      *    you'd expect: each one quotes the next thing in front of
 275      *    it, so you end up with n/2 literal backslashes (if n is
 276      *    even) or (n-1)/2 literal backslashes and a literal quote
 277      *    (if n is odd). In the latter case the double quote
 278      *    character right after the backslashes is used up.
 279      *
 280      *  - After that, any remaining double quotes are processed. A
 281      *    string of contiguous unescaped double quotes has a mod-3
 282      *    behaviour:
 283      *
 284      *     * inside a quoted segment, a quote ends the segment.
 285      *     * _immediately_ after ending a quoted segment, a quote
 286      *       simply produces a literal quote.
 287      *     * otherwise, outside a quoted segment, a quote begins a
 288      *       quoted segment.
 289      *
 290      *    So, for example, if we started inside a quoted segment
 291      *    then two contiguous quotes would close the segment and
 292      *    produce a literal quote; three would close the segment,
 293      *    produce a literal quote, and open a new segment. If we
 294      *    started outside a quoted segment, then two contiguous
 295      *    quotes would open and then close a segment, producing no
 296      *    output (but potentially creating a zero-length argument);
 297      *    but three quotes would open and close a segment and then
 298      *    produce a literal quote.
 299      */
 300
 301     /*
 302      * First deal with the simplest of all special cases: if there
 303      * aren't any arguments, return 0,NULL,NULL.
 304      */
 305     while (*cmdline && isspace(*cmdline)) cmdline++;
 306     if (!*cmdline) {
 307         if (argc) *argc = 0;
 308         if (argv) *argv = NULL;
 309         if (argstart) *argstart = NULL;
 310         return;
 311     }
 312
 313     /*
 314      * This will guaranteeably be big enough; we can realloc it
 315      * down later.
 316      */
 317     outputline = snewn(1+strlen(cmdline), char);
 318     outputargv = snewn(strlen(cmdline)+1 / 2, char *);
 319     outputargstart = snewn(strlen(cmdline)+1 / 2, char *);
 320
 321     p = cmdline; q = outputline; outputargc = 0;
 322
 323     while (*p) {
 324         int quote;
 325
 326         /* Skip whitespace searching for start of argument. */
 327         while (*p && isspace(*p)) p++;
 328         if (!*p) break;
 329
 330         /* We have an argument; start it. */
 331         outputargv[outputargc] = q;
 332         outputargstart[outputargc] = p;
 333         outputargc++;
 334         quote = 0;
 335
 336         /* Copy data into the argument until it's finished. */
 337         while (*p) {
 338             if (!quote && isspace(*p))
 339                 break;                 /* argument is finished */
 340
 341             if (*p == '"' || *p == '\\') {
 342                 /*
 343                  * We have a sequence of zero or more backslashes
 344                  * followed by a sequence of zero or more quotes.
 345                  * Count up how many of each, and then deal with
 346                  * them as appropriate.
 347                  */
 348                 int i, slashes = 0, quotes = 0;
 349                 while (*p == '\\') slashes++, p++;
 350                 while (*p == '"') quotes++, p++;
 351
 352                 if (!quotes) {
 353                     /*
 354                      * Special case: if there are no quotes,
 355                      * slashes are not special at all, so just copy
 356                      * n slashes to the output string.
 357                      */
 358                     while (slashes--) *q++ = '\\';
 359                 } else {
 360                     /* Slashes annihilate in pairs. */
 361                     while (slashes >= 2) slashes -= 2, *q++ = '\\';
 362
 363                     /* One remaining slash takes out the first quote. */
 364                     if (slashes) quotes--, *q++ = '"';
 365
 366                     if (quotes > 0) {
 367                         /* Outside a quote segment, a quote starts one. */
 368                         if (!quote) quotes--;
 369
 370                         /* Now we produce (n+1)/3 literal quotes... */
 371                         for (i = 3; i <= quotes+1; i += 3) *q++ = '"';
 372
 373                         /* ... and end in a quote segment iff 3 divides n. */
 374                         quote = (quotes % 3 == 0);
 375                     }
 376                 }
 377             } else {
 378                 *q++ = *p++;
 379             }
 380         }
 381
 382         /* At the end of an argument, just append a trailing NUL. */
 383         *q++ = '\0';
 384     }
 385
 386     outputargv = sresize(outputargv, outputargc, char *);
 387     outputargstart = sresize(outputargstart, outputargc, char *);
 388
 389     if (argc) *argc = outputargc;
 390     if (argv) *argv = outputargv; else sfree(outputargv);
 391     if (argstart) *argstart = outputargstart; else sfree(outputargstart);
 392 }
 393
 394 #ifdef TESTMODE
 395
 396 const struct argv_test {
 397     const char *cmdline;
 398     const char *argv[10];
 399 } argv_tests[] = {
 400     /*
 401      * We generate this set of tests by invoking ourself with
 402      * `-generate'.
 403      */
 404     {"ab c\" d", {"ab", "c d", NULL}},
 405     {"a\"b c\" d", {"ab c", "d", NULL}},
 406     {"a\"\"b c\" d", {"ab", "c d", NULL}},
 407     {"a\"\"\"b c\" d", {"a\"b", "c d", NULL}},
 408     {"a\"\"\"\"b c\" d", {"a\"b c", "d", NULL}},
 409     {"a\"\"\"\"\"b c\" d", {"a\"b", "c d", NULL}},
 410     {"a\"\"\"\"\"\"b c\" d", {"a\"\"b", "c d", NULL}},
 411     {"a\"\"\"\"\"\"\"b c\" d", {"a\"\"b c", "d", NULL}},
 412     {"a\"\"\"\"\"\"\"\"b c\" d", {"a\"\"b", "c d", NULL}},
 413     {"a\\b c\" d", {"a\\b", "c d", NULL}},
 414     {"a\\\"b c\" d", {"a\"b", "c d", NULL}},
 415     {"a\\\"\"b c\" d", {"a\"b c", "d", NULL}},
 416     {"a\\\"\"\"b c\" d", {"a\"b", "c d", NULL}},
 417     {"a\\\"\"\"\"b c\" d", {"a\"\"b", "c d", NULL}},
 418     {"a\\\"\"\"\"\"b c\" d", {"a\"\"b c", "d", NULL}},
 419     {"a\\\"\"\"\"\"\"b c\" d", {"a\"\"b", "c d", NULL}},
 420     {"a\\\"\"\"\"\"\"\"b c\" d", {"a\"\"\"b", "c d", NULL}},
 421     {"a\\\"\"\"\"\"\"\"\"b c\" d", {"a\"\"\"b c", "d", NULL}},
 422     {"a\\\\b c\" d", {"a\\\\b", "c d", NULL}},
 423     {"a\\\\\"b c\" d", {"a\\b c", "d", NULL}},
 424     {"a\\\\\"\"b c\" d", {"a\\b", "c d", NULL}},
 425     {"a\\\\\"\"\"b c\" d", {"a\\\"b", "c d", NULL}},
 426     {"a\\\\\"\"\"\"b c\" d", {"a\\\"b c", "d", NULL}},
 427     {"a\\\\\"\"\"\"\"b c\" d", {"a\\\"b", "c d", NULL}},
 428     {"a\\\\\"\"\"\"\"\"b c\" d", {"a\\\"\"b", "c d", NULL}},
 429     {"a\\\\\"\"\"\"\"\"\"b c\" d", {"a\\\"\"b c", "d", NULL}},
 430     {"a\\\\\"\"\"\"\"\"\"\"b c\" d", {"a\\\"\"b", "c d", NULL}},
 431     {"a\\\\\\b c\" d", {"a\\\\\\b", "c d", NULL}},
 432     {"a\\\\\\\"b c\" d", {"a\\\"b", "c d", NULL}},
 433     {"a\\\\\\\"\"b c\" d", {"a\\\"b c", "d", NULL}},
 434     {"a\\\\\\\"\"\"b c\" d", {"a\\\"b", "c d", NULL}},
 435     {"a\\\\\\\"\"\"\"b c\" d", {"a\\\"\"b", "c d", NULL}},
 436     {"a\\\\\\\"\"\"\"\"b c\" d", {"a\\\"\"b c", "d", NULL}},
 437     {"a\\\\\\\"\"\"\"\"\"b c\" d", {"a\\\"\"b", "c d", NULL}},
 438     {"a\\\\\\\"\"\"\"\"\"\"b c\" d", {"a\\\"\"\"b", "c d", NULL}},
 439     {"a\\\\\\\"\"\"\"\"\"\"\"b c\" d", {"a\\\"\"\"b c", "d", NULL}},
 440     {"a\\\\\\\\b c\" d", {"a\\\\\\\\b", "c d", NULL}},
 441     {"a\\\\\\\\\"b c\" d", {"a\\\\b c", "d", NULL}},
 442     {"a\\\\\\\\\"\"b c\" d", {"a\\\\b", "c d", NULL}},
 443     {"a\\\\\\\\\"\"\"b c\" d", {"a\\\\\"b", "c d", NULL}},
 444     {"a\\\\\\\\\"\"\"\"b c\" d", {"a\\\\\"b c", "d", NULL}},
 445     {"a\\\\\\\\\"\"\"\"\"b c\" d", {"a\\\\\"b", "c d", NULL}},
 446     {"a\\\\\\\\\"\"\"\"\"\"b c\" d", {"a\\\\\"\"b", "c d", NULL}},
 447     {"a\\\\\\\\\"\"\"\"\"\"\"b c\" d", {"a\\\\\"\"b c", "d", NULL}},
 448     {"a\\\\\\\\\"\"\"\"\"\"\"\"b c\" d", {"a\\\\\"\"b", "c d", NULL}},
 449     {"\"ab c\" d", {"ab c", "d", NULL}},
 450     {"\"a\"b c\" d", {"ab", "c d", NULL}},
 451     {"\"a\"\"b c\" d", {"a\"b", "c d", NULL}},
 452     {"\"a\"\"\"b c\" d", {"a\"b c", "d", NULL}},
 453     {"\"a\"\"\"\"b c\" d", {"a\"b", "c d", NULL}},
 454     {"\"a\"\"\"\"\"b c\" d", {"a\"\"b", "c d", NULL}},
 455     {"\"a\"\"\"\"\"\"b c\" d", {"a\"\"b c", "d", NULL}},
 456     {"\"a\"\"\"\"\"\"\"b c\" d", {"a\"\"b", "c d", NULL}},
 457     {"\"a\"\"\"\"\"\"\"\"b c\" d", {"a\"\"\"b", "c d", NULL}},
 458     {"\"a\\b c\" d", {"a\\b c", "d", NULL}},
 459     {"\"a\\\"b c\" d", {"a\"b c", "d", NULL}},
 460     {"\"a\\\"\"b c\" d", {"a\"b", "c d", NULL}},
 461     {"\"a\\\"\"\"b c\" d", {"a\"\"b", "c d", NULL}},
 462     {"\"a\\\"\"\"\"b c\" d", {"a\"\"b c", "d", NULL}},
 463     {"\"a\\\"\"\"\"\"b c\" d", {"a\"\"b", "c d", NULL}},
 464     {"\"a\\\"\"\"\"\"\"b c\" d", {"a\"\"\"b", "c d", NULL}},
 465     {"\"a\\\"\"\"\"\"\"\"b c\" d", {"a\"\"\"b c", "d", NULL}},
 466     {"\"a\\\"\"\"\"\"\"\"\"b c\" d", {"a\"\"\"b", "c d", NULL}},
 467     {"\"a\\\\b c\" d", {"a\\\\b c", "d", NULL}},
 468     {"\"a\\\\\"b c\" d", {"a\\b", "c d", NULL}},
 469     {"\"a\\\\\"\"b c\" d", {"a\\\"b", "c d", NULL}},
 470     {"\"a\\\\\"\"\"b c\" d", {"a\\\"b c", "d", NULL}},
 471     {"\"a\\\\\"\"\"\"b c\" d", {"a\\\"b", "c d", NULL}},
 472     {"\"a\\\\\"\"\"\"\"b c\" d", {"a\\\"\"b", "c d", NULL}},
 473     {"\"a\\\\\"\"\"\"\"\"b c\" d", {"a\\\"\"b c", "d", NULL}},
 474     {"\"a\\\\\"\"\"\"\"\"\"b c\" d", {"a\\\"\"b", "c d", NULL}},
 475     {"\"a\\\\\"\"\"\"\"\"\"\"b c\" d", {"a\\\"\"\"b", "c d", NULL}},
 476     {"\"a\\\\\\b c\" d", {"a\\\\\\b c", "d", NULL}},
 477     {"\"a\\\\\\\"b c\" d", {"a\\\"b c", "d", NULL}},
 478     {"\"a\\\\\\\"\"b c\" d", {"a\\\"b", "c d", NULL}},
 479     {"\"a\\\\\\\"\"\"b c\" d", {"a\\\"\"b", "c d", NULL}},
 480     {"\"a\\\\\\\"\"\"\"b c\" d", {"a\\\"\"b c", "d", NULL}},
 481     {"\"a\\\\\\\"\"\"\"\"b c\" d", {"a\\\"\"b", "c d", NULL}},
 482     {"\"a\\\\\\\"\"\"\"\"\"b c\" d", {"a\\\"\"\"b", "c d", NULL}},
 483     {"\"a\\\\\\\"\"\"\"\"\"\"b c\" d", {"a\\\"\"\"b c", "d", NULL}},
 484     {"\"a\\\\\\\"\"\"\"\"\"\"\"b c\" d", {"a\\\"\"\"b", "c d", NULL}},
 485     {"\"a\\\\\\\\b c\" d", {"a\\\\\\\\b c", "d", NULL}},
 486     {"\"a\\\\\\\\\"b c\" d", {"a\\\\b", "c d", NULL}},
 487     {"\"a\\\\\\\\\"\"b c\" d", {"a\\\\\"b", "c d", NULL}},
 488     {"\"a\\\\\\\\\"\"\"b c\" d", {"a\\\\\"b c", "d", NULL}},
 489     {"\"a\\\\\\\\\"\"\"\"b c\" d", {"a\\\\\"b", "c d", NULL}},
 490     {"\"a\\\\\\\\\"\"\"\"\"b c\" d", {"a\\\\\"\"b", "c d", NULL}},
 491     {"\"a\\\\\\\\\"\"\"\"\"\"b c\" d", {"a\\\\\"\"b c", "d", NULL}},
 492     {"\"a\\\\\\\\\"\"\"\"\"\"\"b c\" d", {"a\\\\\"\"b", "c d", NULL}},
 493     {"\"a\\\\\\\\\"\"\"\"\"\"\"\"b c\" d", {"a\\\\\"\"\"b", "c d", NULL}},
 494 };
 495
 496 int main(int argc, char **argv)
 497 {
 498     int i, j;
 499
 500     if (argc > 1) {
 501         /*
 502          * Generation of tests.
 503          *
 504          * Given `-splat <args>', we print out a C-style
 505          * representation of each argument (in the form "a", "b",
 506          * NULL), backslash-escaping each backslash and double
 507          * quote.
 508          *
 509          * Given `-split <string>', we first doctor `string' by
 510          * turning forward slashes into backslashes, single quotes
 511          * into double quotes and underscores into spaces; and then
 512          * we feed the resulting string to ourself with `-splat'.
 513          *
 514          * Given `-generate', we concoct a variety of fun test
 515          * cases, encode them in quote-safe form (mapping \, " and
 516          * space to /, ' and _ respectively) and feed each one to
 517          * `-split'.
 518          */
 519         if (!strcmp(argv[1], "-splat")) {
 520             int i;
 521             char *p;
 522             for (i = 2; i < argc; i++) {
 523                 putchar('"');
 524                 for (p = argv[i]; *p; p++) {
 525                     if (*p == '\\' || *p == '"')
 526                         putchar('\\');
 527                     putchar(*p);
 528                 }
 529                 printf("\", ");
 530             }
 531             printf("NULL");
 532             return 0;
 533         }
 534
 535         if (!strcmp(argv[1], "-split") && argc > 2) {
 536             char *str = malloc(20 + strlen(argv[0]) + strlen(argv[2]));
 537             char *p, *q;
 538
 539             q = str + sprintf(str, "%s -splat ", argv[0]);
 540             printf("    {\"");
 541             for (p = argv[2]; *p; p++, q++) {
 542                 switch (*p) {
 543                   case '/':  printf("\\\\"); *q = '\\'; break;
 544                   case '\'': printf("\\\""); *q = '"';  break;
 545                   case '_':  printf(" ");    *q = ' ';  break;
 546                   default:   putchar(*p);    *q = *p;   break;
 547                 }
 548             }
 549             *p = '\0';
 550             printf("\", {");
 551             fflush(stdout);
 552
 553             system(str);
 554
 555             printf("}},\n");
 556
 557             return 0;
 558         }
 559
 560         if (!strcmp(argv[1], "-generate")) {
 561             char *teststr, *p;
 562             int i, initialquote, backslashes, quotes;
 563
 564             teststr = malloc(200 + strlen(argv[0]));
 565
 566             for (initialquote = 0; initialquote <= 1; initialquote++) {
 567                 for (backslashes = 0; backslashes < 5; backslashes++) {
 568                     for (quotes = 0; quotes < 9; quotes++) {
 569                         p = teststr + sprintf(teststr, "%s -split ", argv[0]);
 570                         if (initialquote) *p++ = '\'';
 571                         *p++ = 'a';
 572                         for (i = 0; i < backslashes; i++) *p++ = '/';
 573                         for (i = 0; i < quotes; i++) *p++ = '\'';
 574                         *p++ = 'b';
 575                         *p++ = '_';
 576                         *p++ = 'c';
 577                         *p++ = '\'';
 578                         *p++ = '_';
 579                         *p++ = 'd';
 580                         *p = '\0';
 581
 582                         system(teststr);
 583                     }
 584                 }
 585             }
 586             return 0;
 587         }
 588
 589         fprintf(stderr, "unrecognised option: \"%s\"\n", argv[1]);
 590         return 1;
 591     }
 592
 593     /*
 594      * If we get here, we were invoked with no arguments, so just
 595      * run the tests.
 596      */
 597
 598     for (i = 0; i < lenof(argv_tests); i++) {
 599         int ac;
 600         char **av;
 601
 602         split_into_argv(argv_tests[i].cmdline, &ac, &av);
 603
 604         for (j = 0; j < ac && argv_tests[i].argv[j]; j++) {
 605             if (strcmp(av[j], argv_tests[i].argv[j])) {
 606                 printf("failed test %d (|%s|) arg %d: |%s| should be |%s|\n",
 607                        i, argv_tests[i].cmdline,
 608                        j, av[j], argv_tests[i].argv[j]);
 609             }
 610 #ifdef VERBOSE
 611             else {
 612                 printf("test %d (|%s|) arg %d: |%s| == |%s|\n",
 613                        i, argv_tests[i].cmdline,
 614                        j, av[j], argv_tests[i].argv[j]);
 615             }
 616 #endif
 617         }
 618         if (j < ac)
 619             printf("failed test %d (|%s|): %d args returned, should be %d\n",
 620                    i, argv_tests[i].cmdline, ac, j);
 621         if (argv_tests[i].argv[j])
 622             printf("failed test %d (|%s|): %d args returned, should be more\n",
 623                    i, argv_tests[i].cmdline, ac);
 624     }
 625
 626     return 0;
 627 }
 628
 629 #endif