/*

 * "lpadmin" command for CUPS.

 *

 * Copyright 2007-2016 by Apple Inc.

 * Copyright 1997-2006 by Easy Software Products.

 *

 * These coded instructions, statements, and computer programs are the

 * property of Apple Inc. and are protected by Federal copyright

 * law.  Distribution and use rights are outlined in the file "LICENSE.txt"

 * which should have been included with this file.  If this file is

 * missing or damaged, see the license at "http://www.cups.org/".

 */

/*

 * Include necessary headers...

 */

#define _CUPS_NO_DEPRECATED

#define _PPD_DEPRECATED

#include <cups/cups-private.h>

#include <cups/ppd-private.h>

/*

 * Local functions...

 */

static int		add_printer_to_class(http_t *http, char *printer, char *pclass);

static int		default_printer(http_t *http, char *printer);

static int		delete_printer(http_t *http, char *printer);

static int		delete_printer_from_class(http_t *http, char *printer,

			                          char *pclass);

static int		delete_printer_option(http_t *http, char *printer,

			                      char *option);

static int		enable_printer(http_t *http, char *printer);

static char		*get_printer_ppd(const char *uri, char *buffer, size_t bufsize);

static cups_ptype_t	get_printer_type(http_t *http, char *printer, char *uri,

			                 size_t urisize);

static int		set_printer_options(http_t *http, char *printer,

			                    int num_options, cups_option_t *options,

					    char *file);

static int		validate_name(const char *name);

/*

 * 'main()' - Parse options and configure the scheduler.

 */

int

main(int  argc,			/* I - Number of command-line arguments */

     char *argv[])		/* I - Command-line arguments */

{

  int		i;		/* Looping var */

  http_t	*http;		/* Connection to server */

  char		*printer,	/* Destination printer */

		*pclass,	/* Printer class name */

		*opt,		/* Option pointer */

		*val;		/* Pointer to allow/deny value */

  int		num_options;	/* Number of options */

  cups_option_t	*options;	/* Options */

  char		*file,		/* New PPD file */

		evefile[1024] = "";

				/* IPP Everywhere PPD */

  const char	*ppd_name,	/* ppd-name value */

		*device_uri;	/* device-uri value */

  _cupsSetLocale(argv);

  http        = NULL;

  printer     = NULL;

  num_options = 0;

  options     = NULL;

  file        = NULL;

  for (i = 1; i < argc; i ++)

  {

    if (argv[i][0] == '-')

    {

      for (opt = argv[i] + 1; *opt; opt ++)

      {

	switch (*opt)

	{

	  case 'c' : /* Add printer to class */

	      if (!http)

	      {

		http = httpConnect2(cupsServer(), ippPort(), NULL, AF_UNSPEC, cupsEncryption(), 1, 30000, NULL);

		if (http == NULL)

		{

		  _cupsLangPrintf(stderr, _("lpadmin: Unable to connect to server: %s"), strerror(errno));

		  return (1);

		}

	      }

	      if (printer == NULL)

	      {

		_cupsLangPuts(stderr,

			      _("lpadmin: Unable to add a printer to the class:\n"

				"         You must specify a printer name first."));

		return (1);

	      }

	      if (opt[1] != '\0')

	      {

		pclass = opt + 1;

		opt += strlen(opt) - 1;

	      }

	      else

	      {

		i ++;

		if (i >= argc)

		{

		  _cupsLangPuts(stderr, _("lpadmin: Expected class name after \"-c\" option."));

		  return (1);

		}

		pclass = argv[i];

	      }

	      if (!validate_name(pclass))

	      {

		_cupsLangPuts(stderr,

			      _("lpadmin: Class name can only contain printable "

				"characters."));

		return (1);

	      }

	      if (add_printer_to_class(http, printer, pclass))

		return (1);

	      break;

	  case 'd' : /* Set as default destination */

	      if (!http)

	      {

		http = httpConnect2(cupsServer(), ippPort(), NULL, AF_UNSPEC, cupsEncryption(), 1, 30000, NULL);

		if (http == NULL)

		{

		  _cupsLangPrintf(stderr, _("lpadmin: Unable to connect to server: %s"), strerror(errno));

		  return (1);

		}

	      }

	      if (opt[1] != '\0')

	      {

		printer = opt + 1;

		opt += strlen(opt) - 1;

	      }

	      else

	      {

		i ++;

		if (i >= argc)

		{

		  _cupsLangPuts(stderr, _("lpadmin: Expected printer name after \"-d\" option."));

		  return (1);

		}

		printer = argv[i];

	      }

	      if (!validate_name(printer))

	      {

		_cupsLangPuts(stderr, _("lpadmin: Printer name can only contain printable characters."));

		return (1);

	      }

	      if (default_printer(http, printer))

		return (1);

	      i = argc;

	      break;

	  case 'h' : /* Connect to host */

	      if (http)

	      {

		httpClose(http);

		http = NULL;

	      }

	      if (opt[1] != '\0')

	      {

		cupsSetServer(opt + 1);

		opt += strlen(opt) - 1;

	      }

	      else

	      {

		i ++;

		if (i >= argc)

		{

		  _cupsLangPuts(stderr, _("lpadmin: Expected hostname after \"-h\" option."));

		  return (1);

		}

		cupsSetServer(argv[i]);

	      }

	      break;

	  case 'P' : /* Use the specified PPD file */

	  case 'i' : /* Use the specified PPD file */

	      if (opt[1] != '\0')

	      {

		file = opt + 1;

		opt += strlen(opt) - 1;

	      }

	      else

	      {

		i ++;

		if (i >= argc)

		{

		  _cupsLangPrintf(stderr, _("lpadmin: Expected PPD after \"-%c\" option."), argv[i - 1][1]);

		  return (1);

		}

		file = argv[i];

	      }

	      break;

	  case 'E' : /* Enable the printer/enable encryption */

	      if (printer == NULL)

	      {

#ifdef HAVE_SSL

		cupsSetEncryption(HTTP_ENCRYPTION_REQUIRED);

		if (http)

		  httpEncryption(http, HTTP_ENCRYPTION_REQUIRED);

#else

		_cupsLangPrintf(stderr, _("%s: Sorry, no encryption support."), argv[0]);

#endif /* HAVE_SSL */

		break;

	      }

	      if (!http)

	      {

		http = httpConnect2(cupsServer(), ippPort(), NULL, AF_UNSPEC, cupsEncryption(), 1, 30000, NULL);

		if (http == NULL)

		{

		  _cupsLangPrintf(stderr,

				  _("lpadmin: Unable to connect to server: %s"),

				  strerror(errno));

		  return (1);

		}

	      }

	      if (enable_printer(http, printer))

		return (1);

	      break;

	  case 'm' : /* Use the specified standard script/PPD file */

	      if (opt[1] != '\0')

	      {

		num_options = cupsAddOption("ppd-name", opt + 1, num_options, &options);

		opt += strlen(opt) - 1;

	      }

	      else

	      {

		i ++;

		if (i >= argc)

		{

		  _cupsLangPuts(stderr, _("lpadmin: Expected model after \"-m\" option."));

		  return (1);

		}

		num_options = cupsAddOption("ppd-name", argv[i], num_options, &options);

	      }

	      break;

	  case 'o' : /* Set option */

	      if (opt[1] != '\0')

	      {

		num_options = cupsParseOptions(opt + 1, num_options, &options);

		opt += strlen(opt) - 1;

	      }

	      else

	      {

		i ++;

		if (i >= argc)

		{

		  _cupsLangPuts(stderr, _("lpadmin: Expected name=value after \"-o\" option."));

		  return (1);

		}

		num_options = cupsParseOptions(argv[i], num_options, &options);

	      }

	      break;

	  case 'p' : /* Add/modify a printer */

	      if (opt[1] != '\0')

	      {

		printer = opt + 1;

		opt += strlen(opt) - 1;

	      }

	      else

	      {

		i ++;

		if (i >= argc)

		{

		  _cupsLangPuts(stderr, _("lpadmin: Expected printer after \"-p\" option."));

		  return (1);

		}

		printer = argv[i];

	      }

	      if (!validate_name(printer))

	      {

		_cupsLangPuts(stderr, _("lpadmin: Printer name can only contain printable characters."));

		return (1);

	      }

	      break;

	  case 'r' : /* Remove printer from class */

	      if (!http)

	      {

		http = httpConnect2(cupsServer(), ippPort(), NULL, AF_UNSPEC, cupsEncryption(), 1, 30000, NULL);

		if (http == NULL)

		{

		  _cupsLangPrintf(stderr,

				  _("lpadmin: Unable to connect to server: %s"),

				  strerror(errno));

		  return (1);

		}

	      }

	      if (printer == NULL)

	      {

		_cupsLangPuts(stderr,

			      _("lpadmin: Unable to remove a printer from the class:\n"

				"         You must specify a printer name first."));

		return (1);

	      }

	      if (opt[1] != '\0')

	      {

		pclass = opt + 1;

		opt += strlen(opt) - 1;

	      }

	      else

	      {

		i ++;

		if (i >= argc)

		{

		  _cupsLangPuts(stderr, _("lpadmin: Expected class after \"-r\" option."));

		  return (1);

		}

		pclass = argv[i];

	      }

	      if (!validate_name(pclass))

	      {

		_cupsLangPuts(stderr, _("lpadmin: Class name can only contain printable characters."));

		return (1);

	      }

	      if (delete_printer_from_class(http, printer, pclass))

		return (1);

	      break;

	  case 'R' : /* Remove option */

	      if (!http)

	      {

		http = httpConnect2(cupsServer(), ippPort(), NULL, AF_UNSPEC, cupsEncryption(), 1, 30000, NULL);

		if (http == NULL)

		{

		  _cupsLangPrintf(stderr, _("lpadmin: Unable to connect to server: %s"), strerror(errno));

		  return (1);

		}

	      }

	      if (printer == NULL)

	      {

		_cupsLangPuts(stderr,

			      _("lpadmin: Unable to delete option:\n"

				"         You must specify a printer name first."));

		return (1);

	      }

	      if (opt[1] != '\0')

	      {

		val = opt + 1;

		opt += strlen(opt) - 1;

	      }

	      else

	      {

		i ++;

		if (i >= argc)

		{

		  _cupsLangPuts(stderr, _("lpadmin: Expected name after \"-R\" option."));

		  return (1);

		}

		val = argv[i];

	      }

	      if (delete_printer_option(http, printer, val))

		return (1);

	      break;

	  case 'U' : /* Username */

	      if (opt[1] != '\0')

	      {

		cupsSetUser(opt + 1);

		opt += strlen(opt) - 1;

	      }

	      else

	      {

		i ++;

		if (i >= argc)

		{

		  _cupsLangPrintf(stderr, _("%s: Error - expected username after \"-U\" option."), argv[0]);

		  return (1);

		}

		cupsSetUser(argv[i]);

	      }

	      break;

	  case 'u' : /* Allow/deny users */

	      if (opt[1] != '\0')

	      {

		val = opt + 1;

		opt += strlen(opt) - 1;

	      }

	      else

	      {

		i ++;

		if (i >= argc)

		{

		  _cupsLangPuts(stderr, _("lpadmin: Expected allow/deny:userlist after \"-u\" option."));

		  return (1);

		}

		val = argv[i];

	      }

	      if (!_cups_strncasecmp(val, "allow:", 6))

		num_options = cupsAddOption("requesting-user-name-allowed", val + 6, num_options, &options);

	      else if (!_cups_strncasecmp(val, "deny:", 5))

		num_options = cupsAddOption("requesting-user-name-denied", val + 5, num_options, &options);

	      else

	      {

		_cupsLangPrintf(stderr, _("lpadmin: Unknown allow/deny option \"%s\"."), val);

		return (1);

	      }

	      break;

	  case 'v' : /* Set the device-uri attribute */

	      if (opt[1] != '\0')

	      {

		num_options = cupsAddOption("device-uri", opt + 1, num_options, &options);

		opt += strlen(opt) - 1;

	      }

	      else

	      {

		i ++;

		if (i >= argc)

		{

		  _cupsLangPuts(stderr, _("lpadmin: Expected device URI after \"-v\" option."));

		  return (1);

		}

		num_options = cupsAddOption("device-uri", argv[i], num_options, &options);

	      }

	      break;

	  case 'x' : /* Delete a printer */

	      if (!http)

	      {

		http = httpConnect2(cupsServer(), ippPort(), NULL, AF_UNSPEC, cupsEncryption(), 1, 30000, NULL);

		if (http == NULL)

		{

		  _cupsLangPrintf(stderr,

				  _("lpadmin: Unable to connect to server: %s"),

				  strerror(errno));

		  return (1);

		}

	      }

	      if (opt[1] != '\0')

	      {

		printer = opt + 1;

		opt += strlen(opt) - 1;

	      }

	      else

	      {

		i ++;

		if (i >= argc)

		{

		  _cupsLangPuts(stderr, _("lpadmin: Expected printer or class after \"-x\" option."));

		  return (1);

		}

		printer = argv[i];

	      }

	      if (!validate_name(printer))

	      {

		_cupsLangPuts(stderr, _("lpadmin: Printer name can only contain printable characters."));

		return (1);

	      }

	      if (delete_printer(http, printer))

		return (1);

	      i = argc;

	      break;

	  case 'D' : /* Set the printer-info attribute */

	      if (opt[1] != '\0')

	      {

		num_options = cupsAddOption("printer-info", opt + 1, num_options, &options);

		opt += strlen(opt) - 1;

	      }

	      else

	      {

		i ++;

		if (i >= argc)

		{

		  _cupsLangPuts(stderr, _("lpadmin: Expected description after \"-D\" option."));

		  return (1);

		}

		num_options = cupsAddOption("printer-info", argv[i], num_options, &options);

	      }

	      break;

	  case 'I' : /* Set the supported file types (ignored) */

	      i ++;

	      if (i >= argc)

	      {

		_cupsLangPuts(stderr, _("lpadmin: Expected file type(s) after \"-I\" option."));

		return (1);

	      }

	      _cupsLangPuts(stderr, _("lpadmin: Warning - content type list ignored."));

	      break;

	  case 'L' : /* Set the printer-location attribute */

	      if (opt[1] != '\0')

	      {

		num_options = cupsAddOption("printer-location", opt + 1, num_options, &options);

		opt += strlen(opt) - 1;

	      }

	      else

	      {

		i ++;

		if (i >= argc)

		{

		  _cupsLangPuts(stderr, _("lpadmin: Expected location after \"-L\" option."));

		  return (1);

		}

		num_options = cupsAddOption("printer-location", argv[i], num_options, &options);

	      }

	      break;

	  default :

	      _cupsLangPrintf(stderr, _("lpadmin: Unknown option \"%c\"."), *opt);

	      return (1);

	}

      }

    }

    else

    {

      _cupsLangPrintf(stderr, _("lpadmin: Unknown argument \"%s\"."), argv[i]);

      return (1);

    }

  }

 /*

  * Set options as needed...

  */

  if ((ppd_name = cupsGetOption("ppd-name", num_options, options)) != NULL && !strcmp(ppd_name, "everywhere") && (device_uri = cupsGetOption("device-uri", num_options, options)) != NULL)

  {

    if ((file = get_printer_ppd(device_uri, evefile, sizeof(evefile))) == NULL)

      return (1);

    num_options = cupsRemoveOption("ppd-name", num_options, &options);

  }

  if (num_options || file)

  {

    if (printer == NULL)

    {

      _cupsLangPuts(stderr,

                    _("lpadmin: Unable to set the printer options:\n"

                      "         You must specify a printer name first."));

      return (1);

    }

    if (!http)

    {

      http = httpConnect2(cupsServer(), ippPort(), NULL, AF_UNSPEC,

                          cupsEncryption(), 1, 30000, NULL);

      if (http == NULL) {

        _cupsLangPrintf(stderr, _("lpadmin: Unable to connect to server: %s"),

                        strerror(errno));

        return (1);

      }

    }

    if (set_printer_options(http, printer, num_options, options, file))

      return (1);

  }

  if (evefile[0])

    unlink(evefile);

  if (printer == NULL)

  {

    _cupsLangPuts(stdout,

	          _("Usage:\n"

		    "\n"

		    "    lpadmin [-h server] -d destination\n"

		    "    lpadmin [-h server] -x destination\n"

		    "    lpadmin [-h server] -p printer [-c add-class] "

		    "[-i interface] [-m model]\n"

		    "                       [-r remove-class] [-v device] "

		    "[-D description]\n"

		    "                       [-P ppd-file] [-o name=value]\n"

		    "                       [-u allow:user,user] "

		    "[-u deny:user,user]"));

  }

  if (http)

    httpClose(http);

  return (0);

}

/*

 * 'add_printer_to_class()' - Add a printer to a class.

 */

static int				/* O - 0 on success, 1 on fail */

add_printer_to_class(http_t *http,	/* I - Server connection */

                     char   *printer,	/* I - Printer to add */

		     char   *pclass)	/* I - Class to add to */

{

  int		i;			/* Looping var */

  ipp_t		*request,		/* IPP Request */

		*response;		/* IPP Response */

  ipp_attribute_t *attr,		/* Current attribute */

		*members;		/* Members in class */

  char		uri[HTTP_MAX_URI];	/* URI for printer/class */

  DEBUG_printf(("add_printer_to_class(%p, \"%s\", \"%s\")\n", http,

                printer, pclass));

 /*

  * Build an IPP_OP_GET_PRINTER_ATTRIBUTES request, which requires the following

  * attributes:

  *

  *    attributes-charset

  *    attributes-natural-language

  *    printer-uri

  *    requesting-user-name

  */

  request = ippNewRequest(IPP_OP_GET_PRINTER_ATTRIBUTES);

  httpAssembleURIf(HTTP_URI_CODING_ALL, uri, sizeof(uri), "ipp", NULL,

                   "localhost", 0, "/classes/%s", pclass);

  ippAddString(request, IPP_TAG_OPERATION, IPP_TAG_URI,

               "printer-uri", NULL, uri);

  ippAddString(request, IPP_TAG_OPERATION, IPP_TAG_NAME, "requesting-user-name",

               NULL, cupsUser());

 /*

  * Do the request and get back a response...

  */

  response = cupsDoRequest(http, request, "/");

 /*

  * Build a CUPS-Add-Modify-Class request, which requires the following

  * attributes:

  *

  *    attributes-charset

  *    attributes-natural-language

  *    printer-uri

  *    requesting-user-name

  *    member-uris

  */

  request = ippNewRequest(IPP_OP_CUPS_ADD_MODIFY_CLASS);

  ippAddString(request, IPP_TAG_OPERATION, IPP_TAG_URI,

               "printer-uri", NULL, uri);

  ippAddString(request, IPP_TAG_OPERATION, IPP_TAG_NAME, "requesting-user-name",

               NULL, cupsUser());

 /*

  * See if the printer is already in the class...

  */

  if (response != NULL &&

      (members = ippFindAttribute(response, "member-names",

                                  IPP_TAG_NAME)) != NULL)

    for (i = 0; i < members->num_values; i ++)

      if (_cups_strcasecmp(printer, members->values[i].string.text) == 0)

      {

        _cupsLangPrintf(stderr,

	                _("lpadmin: Printer %s is already a member of class "

			  "%s."), printer, pclass);

        ippDelete(request);

	ippDelete(response);

	return (0);

      }

 /*

  * OK, the printer isn't part of the class, so add it...

  */

  httpAssembleURIf(HTTP_URI_CODING_ALL, uri, sizeof(uri), "ipp", NULL,

                   "localhost", 0, "/printers/%s", printer);

  if (response != NULL &&

      (members = ippFindAttribute(response, "member-uris",

                                  IPP_TAG_URI)) != NULL)

  {

   /*

    * Add the printer to the existing list...

    */

    attr = ippAddStrings(request, IPP_TAG_PRINTER, IPP_TAG_URI,

                         "member-uris", members->num_values + 1, NULL, NULL);

    for (i = 0; i < members->num_values; i ++)

      attr->values[i].string.text =

          _cupsStrAlloc(members->values[i].string.text);

    attr->values[i].string.text = _cupsStrAlloc(uri);

  }

  else

    ippAddString(request, IPP_TAG_PRINTER, IPP_TAG_URI, "member-uris", NULL,

                 uri);

 /*

  * Then send the request...

  */

  ippDelete(response);

  ippDelete(cupsDoRequest(http, request, "/admin/"));

  if (cupsLastError() > IPP_STATUS_OK_CONFLICTING)

  {

    _cupsLangPrintf(stderr, _("%s: %s"), "lpadmin", cupsLastErrorString());

    return (1);

  }

  else

    return (0);

}

/*

 * 'default_printer()' - Set the default printing destination.

 */

static int				/* O - 0 on success, 1 on fail */

default_printer(http_t *http,		/* I - Server connection */

                char   *printer)	/* I - Printer name */

{

  ipp_t		*request;		/* IPP Request */

  char		uri[HTTP_MAX_URI];	/* URI for printer/class */

  DEBUG_printf(("default_printer(%p, \"%s\")\n", http, printer));

 /*

  * Build a CUPS-Set-Default request, which requires the following

  * attributes:

  *

  *    attributes-charset

  *    attributes-natural-language

  *    printer-uri

  *    requesting-user-name

  */

  httpAssembleURIf(HTTP_URI_CODING_ALL, uri, sizeof(uri), "ipp", NULL,

                   "localhost", 0, "/printers/%s", printer);

  request = ippNewRequest(IPP_OP_CUPS_SET_DEFAULT);

  ippAddString(request, IPP_TAG_OPERATION, IPP_TAG_URI,

               "printer-uri", NULL, uri);

  ippAddString(request, IPP_TAG_OPERATION, IPP_TAG_NAME, "requesting-user-name",

               NULL, cupsUser());

 /*

  * Do the request and get back a response...

  */

  ippDelete(cupsDoRequest(http, request, "/admin/"));

  if (cupsLastError() > IPP_STATUS_OK_CONFLICTING)

  {

    _cupsLangPrintf(stderr, _("%s: %s"), "lpadmin", cupsLastErrorString());

    return (1);

  }

  else

    return (0);

}

/*

 * 'delete_printer()' - Delete a printer from the system...

 */

static int				/* O - 0 on success, 1 on fail */

delete_printer(http_t *http,		/* I - Server connection */

               char   *printer)		/* I - Printer to delete */

{

  ipp_t		*request;		/* IPP Request */

  char		uri[HTTP_MAX_URI];	/* URI for printer/class */

  DEBUG_printf(("delete_printer(%p, \"%s\")\n", http, printer));

 /*

  * Build a CUPS-Delete-Printer request, which requires the following

  * attributes:

  *

  *    attributes-charset

  *    attributes-natural-language

  *    printer-uri

  *    requesting-user-name

  */

  request = ippNewRequest(IPP_OP_CUPS_DELETE_PRINTER);

  httpAssembleURIf(HTTP_URI_CODING_ALL, uri, sizeof(uri), "ipp", NULL,

                   "localhost", 0, "/printers/%s", printer);

  ippAddString(request, IPP_TAG_OPERATION, IPP_TAG_URI,

               "printer-uri", NULL, uri);

  ippAddString(request, IPP_TAG_OPERATION, IPP_TAG_NAME, "requesting-user-name",

               NULL, cupsUser());

 /*

  * Do the request and get back a response...

  */

  ippDelete(cupsDoRequest(http, request, "/admin/"));

  if (cupsLastError() > IPP_STATUS_OK_CONFLICTING)

  {

    _cupsLangPrintf(stderr, _("%s: %s"), "lpadmin", cupsLastErrorString());

    return (1);

  }

  else

    return (0);

}