systemv/cupsaddsmb.c

   1 /*
   2  * "$Id: cupsaddsmb.c 7033 2007-10-19 02:11:28Z mike $"
   3  *
   4  *   "cupsaddsmb" command for CUPS.
   5  *
   6  *   Copyright 2007-2010 by Apple Inc.
   7  *   Copyright 2001-2006 by Easy Software Products.
   8  *
   9  *   These coded instructions, statements, and computer programs are the
  10  *   property of Apple Inc. and are protected by Federal copyright
  11  *   law.  Distribution and use rights are outlined in the file "LICENSE.txt"
  12  *   which should have been included with this file.  If this file is
  13  *   file is missing or damaged, see the license at "http://www.cups.org/".
  14  *
  15  * Contents:
  16  *
  17  *   main()        - Export printers on the command-line.
  18  *   export_dest() - Export a destination to SAMBA.
  19  *   usage()       - Show program usage and exit...
  20  */
  21
  22 /*
  23  * Include necessary headers...
  24  */
  25
  26 #include <cups/cups-private.h>
  27 #include <cups/adminutil.h>
  28 #include <fcntl.h>
  29 #include <unistd.h>
  30 #include <sys/wait.h>
  31
  32
  33 /*
  34  * Local globals...
  35  */
  36
  37 int             Verbosity = 0;
  38 const char      *SAMBAUser,
  39                 *SAMBAPassword,
  40                 *SAMBAServer;
  41
  42
  43 /*
  44  * Local functions...
  45  */
  46
  47 int     export_dest(http_t *http, const char *dest);
  48 void    usage(void);
  49
  50
  51 /*
  52  * 'main()' - Export printers on the command-line.
  53  */
  54
  55 int                                     /* O - Exit status */
  56 main(int  argc,                         /* I - Number of command-line arguments */
  57      char *argv[])                      /* I - Command-line arguments */
  58 {
  59   int           i, j;                   /* Looping vars */
  60   int           status;                 /* Status from export_dest() */
  61   int           export_all;             /* Export all printers? */
  62   http_t        *http;                  /* Connection to server */
  63   int           num_dests;              /* Number of printers */
  64   cups_dest_t   *dests;                 /* Printers */
  65
  66
  67   _cupsSetLocale(argv);
  68
  69  /*
  70   * Parse command-line arguments...
  71   */
  72
  73   export_all    = 0;
  74   http          = NULL;
  75   SAMBAUser     = cupsUser();
  76   SAMBAPassword = NULL;
  77   SAMBAServer   = NULL;
  78
  79   for (i = 1; i < argc; i ++)
  80     if (!strcmp(argv[i], "-E"))
  81     {
  82 #ifdef HAVE_SSL
  83       cupsSetEncryption(HTTP_ENCRYPT_REQUIRED);
  84 #else
  85       _cupsLangPrintf(stderr,
  86                       _("%s: Sorry, no encryption support compiled in\n"),
  87                       argv[0]);
  88 #endif /* HAVE_SSL */
  89     }
  90     else if (!strcmp(argv[i], "-H"))
  91     {
  92       i ++;
  93       if (i >= argc)
  94         usage();
  95
  96       SAMBAServer = argv[i];
  97     }
  98     else if (!strcmp(argv[i], "-U"))
  99     {
 100       char      *sep;                   /* Separator for password */
 101
 102
 103       i ++;
 104       if (i >= argc)
 105         usage();
 106
 107       SAMBAUser = argv[i];
 108
 109       if ((sep = strchr(argv[i], '%')) != NULL)
 110       {
 111        /*
 112         * Nul-terminate the username at the first % and point the
 113         * password at the rest...
 114         */
 115
 116         *sep++ = '\0';
 117
 118         SAMBAPassword = sep;
 119       }
 120     }
 121     else if (!strcmp(argv[i], "-a"))
 122       export_all = 1;
 123     else if (!strcmp(argv[i], "-h"))
 124     {
 125       i ++;
 126       if (i >= argc)
 127         usage();
 128
 129       cupsSetServer(argv[i]);
 130     }
 131     else if (!strcmp(argv[i], "-v"))
 132       Verbosity = 1;
 133     else if (argv[i][0] != '-')
 134     {
 135       if (!http)
 136       {
 137        /*
 138         * Connect to the server...
 139         */
 140
 141         if ((http = httpConnectEncrypt(cupsServer(), ippPort(),
 142                                        cupsEncryption())) == NULL)
 143         {
 144           _cupsLangPrintf(stderr, _("%s: Unable to connect to server\n"), argv[0]);
 145           exit(1);
 146         }
 147       }
 148
 149       if (SAMBAServer == NULL)
 150       {
 151         SAMBAServer = cupsServer();
 152
 153         if (SAMBAServer[0] == '/')      /* Use localhost instead of domain socket */
 154           SAMBAServer = "localhost";
 155       }
 156
 157       if ((status = export_dest(http, argv[i])) != 0)
 158         return (status);
 159     }
 160     else
 161       usage();
 162
 163  /*
 164   * Connect to the server...
 165   */
 166
 167   if ((http = httpConnectEncrypt(cupsServer(), ippPort(),
 168                                  cupsEncryption())) == NULL)
 169   {
 170     _cupsLangPrintf(stderr, _("%s: Unable to connect to server\n"), argv[0]);
 171     exit(1);
 172   }
 173
 174  /*
 175   * See if the user specified "-a"...
 176   */
 177
 178   if (export_all)
 179   {
 180    /*
 181     * Export all printers...
 182     */
 183
 184     if (SAMBAServer == NULL)
 185     {
 186       SAMBAServer = cupsServer();
 187
 188       if (SAMBAServer[0] == '/')        /* Use localhost instead of domain socket */
 189         SAMBAServer = "localhost";
 190     }
 191
 192     num_dests = cupsGetDests2(http, &dests);
 193
 194     for (j = 0, status = 0; j < num_dests; j ++)
 195       if (!dests[j].instance)
 196       {
 197         if ((status = export_dest(http, dests[j].name)) != 0)
 198           break;
 199       }
 200
 201     cupsFreeDests(num_dests, dests);
 202
 203     if (status)
 204       return (status);
 205   }
 206
 207   return (0);
 208 }
 209
 210
 211 /*
 212  * 'export_dest()' - Export a destination to SAMBA.
 213  */
 214
 215 int                                     /* O - 0 on success, non-zero on error */
 216 export_dest(http_t     *http,           /* I - Connection to server */
 217             const char *dest)           /* I - Destination to export */
 218 {
 219   int           status;                 /* Status of export */
 220   char          ppdfile[1024],          /* PPD file for printer drivers */
 221                 prompt[1024];           /* Password prompt */
 222
 223
 224  /*
 225   * Get the Windows PPD file for the printer...
 226   */
 227
 228   if (!cupsAdminCreateWindowsPPD(http, dest, ppdfile, sizeof(ppdfile)))
 229   {
 230     _cupsLangPrintf(stderr,
 231                     _("cupsaddsmb: No PPD file for printer \"%s\" - "
 232                       "%s\n"),
 233                     dest, cupsLastErrorString());
 234     return (1);
 235   }
 236
 237  /*
 238   * Try to export it...
 239   */
 240
 241   for (status = 0; !status;)
 242   {
 243    /*
 244     * Get the password, as needed...
 245     */
 246
 247     if (!SAMBAPassword)
 248     {
 249       snprintf(prompt, sizeof(prompt),
 250                _cupsLangString(cupsLangDefault(),
 251                                _("Password for %s required to access %s via "
 252                                  "SAMBA: ")),
 253                SAMBAUser, SAMBAServer);
 254
 255       if ((SAMBAPassword = cupsGetPassword(prompt)) == NULL)
 256         break;
 257     }
 258
 259     status = cupsAdminExportSamba(dest, ppdfile, SAMBAServer,
 260                                   SAMBAUser, SAMBAPassword,
 261                                   Verbosity ? stderr : NULL);
 262
 263     if (!status && cupsLastError() == IPP_NOT_FOUND)
 264       break;
 265   }
 266
 267   unlink(ppdfile);
 268
 269   return (!status);
 270 }
 271
 272
 273 /*
 274  * 'usage()' - Show program usage and exit...
 275  */
 276
 277 void
 278 usage(void)
 279 {
 280   _cupsLangPuts(stdout,
 281                 _("Usage: cupsaddsmb [options] printer1 ... printerN\n"
 282                   "       cupsaddsmb [options] -a\n"
 283                   "\n"
 284                   "Options:\n"
 285                   "  -E               Encrypt the connection to the server\n"
 286                   "  -H samba-server  Use the named SAMBA server\n"
 287                   "  -U samba-user    Authenticate using the named SAMBA user\n"
 288                   "  -a               Export all printers\n"
 289                   "  -h cups-server   Use the named CUPS server\n"
 290                   "  -v               Be verbose (show commands)\n"));
 291   exit(1);
 292 }
 293
 294
 295 /*
 296  * End of "$Id: cupsaddsmb.c 7033 2007-10-19 02:11:28Z mike $".
 297  */