/*-

 * Copyright (c) 2012 Michihiro NAKAJIMA 

 * All rights reserved.

 *

 * Redistribution and use in source and binary forms, with or without

 * modification, are permitted provided that the following conditions

 * are met:

 * 1. Redistributions of source code must retain the above copyright

 *    notice, this list of conditions and the following disclaimer.

 * 2. Redistributions in binary form must reproduce the above copyright

 *    notice, this list of conditions and the following disclaimer in the

 *    documentation and/or other materials provided with the distribution.

 *

 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR(S) ``AS IS'' AND ANY EXPRESS OR

 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES

 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.

 * IN NO EVENT SHALL THE AUTHOR(S) BE LIABLE FOR ANY DIRECT, INDIRECT,

 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT

 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,

 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY

 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT

 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF

 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

 */

#include "archive_platform.h"

#ifdef HAVE_STRING_H

#  include <string.h>

#endif

#ifdef HAVE_STDLIB_H

#  include <stdlib.h>

#endif

#include "archive.h"

#include "archive_cmdline_private.h"

#include "archive_string.h"

static int cmdline_set_path(struct archive_cmdline *, const char *);

static int cmdline_add_arg(struct archive_cmdline *, const char *);

static ssize_t

extract_quotation(struct archive_string *as, const char *p)

{

  const char *s;

  for (s = p + 1; *s;) {

    if (*s == '\\') {

      if (s[1] != '\0') {

        archive_strappend_char(as, s[1]);

        s += 2;

      } else

        s++;

    } else if (*s == '"')

      break;

    else {

      archive_strappend_char(as, s[0]);

      s++;

    }

  }

  if (*s != '"')

    return (ARCHIVE_FAILED);/* Invalid sequence. */

  return ((ssize_t)(s + 1 - p));

}

static ssize_t

get_argument(struct archive_string *as, const char *p)

{

  const char *s = p;

  archive_string_empty(as);

  /* Skip beginning space characters. */

  while (*s == ' ')

    s++;

  /* Copy non-space characters. */

  while (*s != '\0' && *s != ' ') {

    if (*s == '\\') {

      if (s[1] != '\0') {

        archive_strappend_char(as, s[1]);

        s += 2;

      } else {

        s++;/* Ignore this character.*/

        break;

      }

    } else if (*s == '"') {

      ssize_t q = extract_quotation(as, s);

      if (q < 0)

        return (ARCHIVE_FAILED);/* Invalid sequence. */

      s += q;

    } else {

      archive_strappend_char(as, s[0]);

      s++;

    }

  }

  return ((ssize_t)(s - p));

}

/*

 * Set up command line arguments.

 * Returns ARCHIVE_OK if everything okey.

 * Returns ARCHIVE_FAILED if there is a lack of the `"' terminator or an

 * empty command line.

 * Returns ARCHIVE_FATAL if no memory.

 */

int

__archive_cmdline_parse(struct archive_cmdline *data, const char *cmd)

{

  struct archive_string as;

  const char *p;

  ssize_t al;

  int r;

  archive_string_init(&as);

  /* Get first argument as a command path. */

  al = get_argument(&as, cmd);

  if (al < 0) {

    r = ARCHIVE_FAILED;/* Invalid sequence. */

    goto exit_function;

  }

  if (archive_strlen(&as) == 0) {

    r = ARCHIVE_FAILED;/* An empty command path. */

    goto exit_function;

  }

  r = cmdline_set_path(data, as.s);

  if (r != ARCHIVE_OK)

    goto exit_function;

  p = strrchr(as.s, '/');

  if (p == NULL)

    p = as.s;

  else

    p++;

  r = cmdline_add_arg(data, p);

  if (r != ARCHIVE_OK)

    goto exit_function;

  cmd += al;

  for (;;) {

    al = get_argument(&as, cmd);

    if (al < 0) {

      r = ARCHIVE_FAILED;/* Invalid sequence. */

      goto exit_function;

    }

    if (al == 0)

      break;

    cmd += al;

    if (archive_strlen(&as) == 0 && *cmd == '\0')

      break;

    r = cmdline_add_arg(data, as.s);

    if (r != ARCHIVE_OK)

      goto exit_function;

  }

  r = ARCHIVE_OK;

exit_function:

  archive_string_free(&as);

  return (r);

}

/*

 * Set the program path.

 */

static int

cmdline_set_path(struct archive_cmdline *data, const char *path)

{

  char *newptr;

  newptr = realloc(data->path, strlen(path) + 1);

  if (newptr == NULL)

    return (ARCHIVE_FATAL);

  data->path = newptr;

  strcpy(data->path, path);

  return (ARCHIVE_OK);

}

/*

 * Add a argument for the program.

 */

static int

cmdline_add_arg(struct archive_cmdline *data, const char *arg)

{

  char **newargv;

  if (data->path == NULL)

    return (ARCHIVE_FAILED);

  newargv = realloc(data->argv, (data->argc + 2) * sizeof(char *));

  if (newargv == NULL)

    return (ARCHIVE_FATAL);

  data->argv = newargv;

  data->argv[data->argc] = strdup(arg);

  if (data->argv[data->argc] == NULL)

    return (ARCHIVE_FATAL);

  /* Set the terminator of argv. */

  data->argv[++data->argc] = NULL;

  return (ARCHIVE_OK);

}

struct archive_cmdline *

__archive_cmdline_allocate(void)

{

  return (struct archive_cmdline *)

    calloc(1, sizeof(struct archive_cmdline));

}

/*

 * Release the resources.

 */

int

__archive_cmdline_free(struct archive_cmdline *data)

{

  if (data) {

    free(data->path);

    if (data->argv != NULL) {

      int i;

      for (i = 0; data->argv[i] != NULL; i++)

        free(data->argv[i]);

      free(data->argv);

    }

    free(data);

  }

  return (ARCHIVE_OK);

}