Blame view

libmailutils/base/argcvjoin.c 2.55 KB
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74
/* Join strings from an array into a single string.
   Copyright (C) 1999, 2000, 2001, 2003, 2004, 2005, 2006, 2010 Free
   Software Foundation, Inc.

   This library is free software; you can redistribute it and/or
   modify it under the terms of the GNU Lesser General Public
   License as published by the Free Software Foundation; either
   version 3 of the License, or (at your option) any later version.

   This library is distributed in the hope that it will be useful,
   but WITHOUT ANY WARRANTY; without even the implied warranty of
   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
   Lesser General Public License for more details.

   You should have received a copy of the GNU Lesser General
   Public License along with this library.  If not, see 
   <http://www.gnu.org/licenses/>. */

#ifdef HAVE_CONFIG_H
# include <config.h>
#endif
#include <stdlib.h>
#include <mailutils/types.h>
#include <mailutils/argcv.h>
#include <mailutils/errno.h>
#include <mailutils/wordsplit.h>

int
mu_argcv_join (int argc, char **argv, char *delim, enum mu_argcv_escape esc,
	       char **pstring)
{
  size_t i, j, len;
  char *buffer;
  size_t delimlen = strlen (delim);
  int quote_hex = 0;
  
  if (pstring == NULL)
    return EINVAL;

  buffer = malloc (1);
  if (buffer == NULL)
    return ENOMEM;
  *buffer = '\0';

  for (len = i = j = 0; i < argc; i++)
    {
      int quote;
      int toklen;

      switch (esc)
	{
	case mu_argcv_escape_no:
	  toklen = strlen (argv[i]);
	  quote = 0;
	  break;

	case mu_argcv_escape_c:
	  toklen = mu_wordsplit_c_quoted_length (argv[i], quote_hex, &quote);
	  break;

#if 0
	case mu_argcv_escape_sh:
	  FIXME
	  toklen = mu_wordsplit_sh_quoted_length (argv[i], &quote);
	  break;
#endif
	default:
	  return EINVAL;
	}
	
      len += toklen + delimlen;
      if (quote)
	len += 2;
      
Sergey Poznyakoff authored
75
      buffer = realloc (buffer, len + 1);
76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116
      if (buffer == NULL)
        return ENOMEM;

      if (i != 0)
	{
	  memcpy (buffer + j, delim, delimlen);
	  j += delimlen;
	}
      if (quote)
	buffer[j++] = '"';
      
      switch (esc)
	{
	case mu_argcv_escape_no:
	  memcpy (buffer + j, argv[i], toklen);
	  break;

	case mu_argcv_escape_c:
	  mu_wordsplit_c_quote_copy (buffer + j, argv[i], quote_hex);
	  break;
#if 0
	case mu_argcv_escape_sh:
	  mu_wordsplit_sh_quote_copy (buffer + j, argv[i]);
#endif
	}

      j += toklen;
      if (quote)
	buffer[j++] = '"';
    }

  buffer[j] = 0;
  *pstring = buffer;
  return 0;
}

int
mu_argcv_string (int argc, char **argv, char **pstring)
{
  return mu_argcv_join (argc, argv, " ", mu_argcv_escape_c, pstring);
}