cups/request.c

   1 /*
   2  * "$Id: request.c 7946 2008-09-16 23:27:54Z mike $"
   3  *
   4  *   IPP utilities for CUPS.
   5  *
   6  *   Copyright 2007-2012 by Apple Inc.
   7  *   Copyright 1997-2007 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  *   This file is subject to the Apple OS-Developed Software exception.
  16  *
  17  * Contents:
  18  *
  19  *   cupsDoFileRequest()    - Do an IPP request with a file.
  20  *   cupsDoIORequest()      - Do an IPP request with file descriptors.
  21  *   cupsDoRequest()        - Do an IPP request.
  22  *   cupsGetResponse()      - Get a response to an IPP request.
  23  *   cupsLastError()        - Return the last IPP status code.
  24  *   cupsLastErrorString()  - Return the last IPP status-message.
  25  *   _cupsNextDelay()       - Return the next retry delay value.
  26  *   cupsReadResponseData() - Read additional data after the IPP response.
  27  *   cupsSendRequest()      - Send an IPP request.
  28  *   cupsWriteRequestData() - Write additional data after an IPP request.
  29  *   _cupsConnect()         - Get the default server connection...
  30  *   _cupsSetError()        - Set the last IPP status code and status-message.
  31  *   _cupsSetHTTPError()    - Set the last error using the HTTP status.
  32  */
  33
  34 /*
  35  * Include necessary headers...
  36  */
  37
  38 #include "cups-private.h"
  39 #include <fcntl.h>
  40 #include <sys/stat.h>
  41 #if defined(WIN32) || defined(__EMX__)
  42 #  include <io.h>
  43 #else
  44 #  include <unistd.h>
  45 #endif /* WIN32 || __EMX__ */
  46 #ifndef O_BINARY
  47 #  define O_BINARY 0
  48 #endif /* O_BINARY */
  49
  50
  51 /*
  52  * 'cupsDoFileRequest()' - Do an IPP request with a file.
  53  *
  54  * This function sends the IPP request and attached file to the specified
  55  * server, retrying and authenticating as necessary.  The request is freed with
  56  * @link ippDelete@.
  57  */
  58
  59 ipp_t *                                 /* O - Response data */
  60 cupsDoFileRequest(http_t     *http,     /* I - Connection to server or @code CUPS_HTTP_DEFAULT@ */
  61                   ipp_t      *request,  /* I - IPP request */
  62                   const char *resource, /* I - HTTP resource for POST */
  63                   const char *filename) /* I - File to send or @code NULL@ for none */
  64 {
  65   ipp_t         *response;              /* IPP response data */
  66   int           infile;                 /* Input file */
  67
  68
  69   DEBUG_printf(("cupsDoFileRequest(http=%p, request=%p(%s), resource=\"%s\", "
  70                 "filename=\"%s\")", http, request,
  71                 request ? ippOpString(request->request.op.operation_id) : "?",
  72                 resource, filename));
  73
  74   if (filename)
  75   {
  76     if ((infile = open(filename, O_RDONLY | O_BINARY)) < 0)
  77     {
  78      /*
  79       * Can't get file information!
  80       */
  81
  82       _cupsSetError(errno == ENOENT ? IPP_NOT_FOUND : IPP_NOT_AUTHORIZED,
  83                     NULL, 0);
  84
  85       ippDelete(request);
  86
  87       return (NULL);
  88     }
  89   }
  90   else
  91     infile = -1;
  92
  93   response = cupsDoIORequest(http, request, resource, infile, -1);
  94
  95   if (infile >= 0)
  96     close(infile);
  97
  98   return (response);
  99 }
 100
 101
 102 /*
 103  * 'cupsDoIORequest()' - Do an IPP request with file descriptors.
 104  *
 105  * This function sends the IPP request with the optional input file "infile" to
 106  * the specified server, retrying and authenticating as necessary.  The request
 107  * is freed with @link ippDelete@.
 108  *
 109  * If "infile" is a valid file descriptor, @code cupsDoIORequest@ copies
 110  * all of the data from the file after the IPP request message.
 111  *
 112  * If "outfile" is a valid file descriptor, @code cupsDoIORequest@ copies
 113  * all of the data after the IPP response message to the file.
 114  *
 115  * @since CUPS 1.3/OS X 10.5@
 116  */
 117
 118 ipp_t *                                 /* O - Response data */
 119 cupsDoIORequest(http_t     *http,       /* I - Connection to server or @code CUPS_HTTP_DEFAULT@ */
 120                 ipp_t      *request,    /* I - IPP request */
 121                 const char *resource,   /* I - HTTP resource for POST */
 122                 int        infile,      /* I - File to read from or -1 for none */
 123                 int        outfile)     /* I - File to write to or -1 for none */
 124 {
 125   ipp_t         *response = NULL;       /* IPP response data */
 126   size_t        length = 0;             /* Content-Length value */
 127   http_status_t status;                 /* Status of HTTP request */
 128   struct stat   fileinfo;               /* File information */
 129   int           bytes;                  /* Number of bytes read/written */
 130   char          buffer[32768];          /* Output buffer */
 131
 132
 133   DEBUG_printf(("cupsDoIORequest(http=%p, request=%p(%s), resource=\"%s\", "
 134                 "infile=%d, outfile=%d)", http, request,
 135                 request ? ippOpString(request->request.op.operation_id) : "?",
 136                 resource, infile, outfile));
 137
 138  /*
 139   * Range check input...
 140   */
 141
 142   if (!request || !resource)
 143   {
 144     ippDelete(request);
 145
 146     _cupsSetError(IPP_INTERNAL_ERROR, strerror(EINVAL), 0);
 147
 148     return (NULL);
 149   }
 150
 151  /*
 152   * Get the default connection as needed...
 153   */
 154
 155   if (!http)
 156     if ((http = _cupsConnect()) == NULL)
 157     {
 158       ippDelete(request);
 159
 160       return (NULL);
 161     }
 162
 163  /*
 164   * See if we have a file to send...
 165   */
 166
 167   if (infile >= 0)
 168   {
 169     if (fstat(infile, &fileinfo))
 170     {
 171      /*
 172       * Can't get file information!
 173       */
 174
 175       _cupsSetError(errno == EBADF ? IPP_NOT_FOUND : IPP_NOT_AUTHORIZED,
 176                     NULL, 0);
 177
 178       ippDelete(request);
 179
 180       return (NULL);
 181     }
 182
 183 #ifdef WIN32
 184     if (fileinfo.st_mode & _S_IFDIR)
 185 #else
 186     if (S_ISDIR(fileinfo.st_mode))
 187 #endif /* WIN32 */
 188     {
 189      /*
 190       * Can't send a directory...
 191       */
 192
 193       ippDelete(request);
 194
 195       _cupsSetError(IPP_NOT_POSSIBLE, strerror(EISDIR), 0);
 196
 197       return (NULL);
 198     }
 199
 200 #ifndef WIN32
 201     if (!S_ISREG(fileinfo.st_mode))
 202       length = 0;                       /* Chunk when piping */
 203     else
 204 #endif /* !WIN32 */
 205     length = ippLength(request) + fileinfo.st_size;
 206   }
 207   else
 208     length = ippLength(request);
 209
 210   DEBUG_printf(("2cupsDoIORequest: Request length=%ld, total length=%ld",
 211                 (long)ippLength(request), (long)length));
 212
 213  /*
 214   * Clear any "Local" authentication data since it is probably stale...
 215   */
 216
 217   if (http->authstring && !strncmp(http->authstring, "Local ", 6))
 218     httpSetAuthString(http, NULL, NULL);
 219
 220  /*
 221   * Loop until we can send the request without authorization problems.
 222   */
 223
 224   while (response == NULL)
 225   {
 226     DEBUG_puts("2cupsDoIORequest: setup...");
 227
 228    /*
 229     * Send the request...
 230     */
 231
 232     status = cupsSendRequest(http, request, resource, length);
 233
 234     DEBUG_printf(("2cupsDoIORequest: status=%d", status));
 235
 236     if (status == HTTP_CONTINUE && request->state == IPP_DATA && infile >= 0)
 237     {
 238       DEBUG_puts("2cupsDoIORequest: file write...");
 239
 240      /*
 241       * Send the file with the request...
 242       */
 243
 244 #ifndef WIN32
 245       if (S_ISREG(fileinfo.st_mode))
 246 #endif /* WIN32 */
 247       lseek(infile, 0, SEEK_SET);
 248
 249       while ((bytes = (int)read(infile, buffer, sizeof(buffer))) > 0)
 250       {
 251         if ((status = cupsWriteRequestData(http, buffer, bytes))
 252                 != HTTP_CONTINUE)
 253           break;
 254       }
 255     }
 256
 257    /*
 258     * Get the server's response...
 259     */
 260
 261     if (status != HTTP_ERROR)
 262     {
 263       response = cupsGetResponse(http, resource);
 264       status   = httpGetStatus(http);
 265     }
 266
 267     DEBUG_printf(("2cupsDoIORequest: status=%d", status));
 268
 269     if (status == HTTP_ERROR ||
 270         (status >= HTTP_BAD_REQUEST && status != HTTP_UNAUTHORIZED &&
 271          status != HTTP_UPGRADE_REQUIRED))
 272     {
 273       _cupsSetHTTPError(status);
 274       break;
 275     }
 276
 277     if (response && outfile >= 0)
 278     {
 279      /*
 280       * Write trailing data to file...
 281       */
 282
 283       while ((bytes = (int)httpRead2(http, buffer, sizeof(buffer))) > 0)
 284         if (write(outfile, buffer, bytes) < bytes)
 285           break;
 286     }
 287
 288     if (http->state != HTTP_WAITING)
 289     {
 290      /*
 291       * Flush any remaining data...
 292       */
 293
 294       httpFlush(http);
 295     }
 296   }
 297
 298  /*
 299   * Delete the original request and return the response...
 300   */
 301
 302   ippDelete(request);
 303
 304   return (response);
 305 }
 306
 307
 308 /*
 309  * 'cupsDoRequest()' - Do an IPP request.
 310  *
 311  * This function sends the IPP request to the specified server, retrying
 312  * and authenticating as necessary.  The request is freed with @link ippDelete@.
 313  */
 314
 315 ipp_t *                                 /* O - Response data */
 316 cupsDoRequest(http_t     *http,         /* I - Connection to server or @code CUPS_HTTP_DEFAULT@ */
 317               ipp_t      *request,      /* I - IPP request */
 318               const char *resource)     /* I - HTTP resource for POST */
 319 {
 320   DEBUG_printf(("cupsDoRequest(http=%p, request=%p(%s), resource=\"%s\")",
 321                 http, request,
 322                 request ? ippOpString(request->request.op.operation_id) : "?",
 323                 resource));
 324
 325   return (cupsDoIORequest(http, request, resource, -1, -1));
 326 }
 327
 328
 329 /*
 330  * 'cupsGetResponse()' - Get a response to an IPP request.
 331  *
 332  * Use this function to get the response for an IPP request sent using
 333  * @link cupsSendRequest@. For requests that return additional data, use
 334  * @link cupsReadResponseData@ after getting a successful response,
 335  * otherwise call @link httpFlush@ to complete the response processing.
 336  *
 337  * @since CUPS 1.4/OS X 10.6@
 338  */
 339
 340 ipp_t *                                 /* O - Response or @code NULL@ on HTTP error */
 341 cupsGetResponse(http_t     *http,       /* I - Connection to server or @code CUPS_HTTP_DEFAULT@ */
 342                 const char *resource)   /* I - HTTP resource for POST */
 343 {
 344   http_status_t status;                 /* HTTP status */
 345   ipp_state_t   state;                  /* IPP read state */
 346   ipp_t         *response = NULL;       /* IPP response */
 347
 348
 349   DEBUG_printf(("cupsGetResponse(http=%p, resource=\"%s\")", http, resource));
 350
 351  /*
 352   * Connect to the default server as needed...
 353   */
 354
 355   if (!http)
 356     http = _cupsConnect();
 357
 358   if (!http || (http->state != HTTP_POST_RECV && http->state != HTTP_POST_SEND))
 359     return (NULL);
 360
 361  /*
 362   * Check for an unfinished chunked request...
 363   */
 364
 365   if (http->data_encoding == HTTP_ENCODE_CHUNKED)
 366   {
 367    /*
 368     * Send a 0-length chunk to finish off the request...
 369     */
 370
 371     DEBUG_puts("2cupsGetResponse: Finishing chunked POST...");
 372
 373     if (httpWrite2(http, "", 0) < 0)
 374       return (NULL);
 375   }
 376
 377  /*
 378   * Wait for a response from the server...
 379   */
 380
 381   DEBUG_printf(("2cupsGetResponse: Update loop, http->status=%d...",
 382                 http->status));
 383
 384   do
 385   {
 386     status = httpUpdate(http);
 387   }
 388   while (status == HTTP_STATUS_CONTINUE);
 389
 390   DEBUG_printf(("2cupsGetResponse: status=%d", status));
 391
 392   if (status == HTTP_OK)
 393   {
 394    /*
 395     * Get the IPP response...
 396     */
 397
 398     response = ippNew();
 399
 400     while ((state = ippRead(http, response)) != IPP_DATA)
 401       if (state == IPP_ERROR)
 402         break;
 403
 404     if (state == IPP_ERROR)
 405     {
 406      /*
 407       * Flush remaining data and delete the response...
 408       */
 409
 410       DEBUG_puts("1cupsGetResponse: IPP read error!");
 411
 412       httpFlush(http);
 413
 414       ippDelete(response);
 415       response = NULL;
 416
 417       http->status = status = HTTP_ERROR;
 418       http->error  = EINVAL;
 419     }
 420   }
 421   else if (status != HTTP_ERROR)
 422   {
 423    /*
 424     * Flush any error message...
 425     */
 426
 427     httpFlush(http);
 428
 429    /*
 430     * Then handle encryption and authentication...
 431     */
 432
 433     if (status == HTTP_UNAUTHORIZED)
 434     {
 435      /*
 436       * See if we can do authentication...
 437       */
 438
 439       DEBUG_puts("2cupsGetResponse: Need authorization...");
 440
 441       if (!cupsDoAuthentication(http, "POST", resource))
 442         httpReconnect(http);
 443       else
 444         http->status = status = HTTP_AUTHORIZATION_CANCELED;
 445     }
 446
 447 #ifdef HAVE_SSL
 448     else if (status == HTTP_UPGRADE_REQUIRED)
 449     {
 450      /*
 451       * Force a reconnect with encryption...
 452       */
 453
 454       DEBUG_puts("2cupsGetResponse: Need encryption...");
 455
 456       if (!httpReconnect(http))
 457         httpEncryption(http, HTTP_ENCRYPT_REQUIRED);
 458     }
 459 #endif /* HAVE_SSL */
 460   }
 461
 462   if (response)
 463   {
 464     ipp_attribute_t     *attr;          /* status-message attribute */
 465
 466
 467     attr = ippFindAttribute(response, "status-message", IPP_TAG_TEXT);
 468
 469     DEBUG_printf(("1cupsGetResponse: status-code=%s, status-message=\"%s\"",
 470                   ippErrorString(response->request.status.status_code),
 471                   attr ? attr->values[0].string.text : ""));
 472
 473     _cupsSetError(response->request.status.status_code,
 474                   attr ? attr->values[0].string.text :
 475                       ippErrorString(response->request.status.status_code), 0);
 476   }
 477
 478   return (response);
 479 }
 480
 481
 482 /*
 483  * 'cupsLastError()' - Return the last IPP status code received on the current
 484  *                     thread.
 485  */
 486
 487 ipp_status_t                            /* O - IPP status code from last request */
 488 cupsLastError(void)
 489 {
 490   return (_cupsGlobals()->last_error);
 491 }
 492
 493
 494 /*
 495  * 'cupsLastErrorString()' - Return the last IPP status-message received on the
 496  *                           current thread.
 497  *
 498  * @since CUPS 1.2/OS X 10.5@
 499  */
 500
 501 const char *                            /* O - status-message text from last request */
 502 cupsLastErrorString(void)
 503 {
 504   return (_cupsGlobals()->last_status_message);
 505 }
 506
 507
 508 /*
 509  * '_cupsNextDelay()' - Return the next retry delay value.
 510  *
 511  * This function currently returns the Fibonacci sequence 1 1 2 3 5 8.
 512  *
 513  * Pass 0 for the current delay value to initialize the sequence.
 514  */
 515
 516 int                                     /* O  - Next delay value */
 517 _cupsNextDelay(int current,             /* I  - Current delay value or 0 */
 518                int *previous)           /* IO - Previous delay value */
 519 {
 520   int   next;                           /* Next delay value */
 521
 522
 523   if (current > 0)
 524   {
 525     next      = (current + *previous) % 12;
 526     *previous = next < current ? 0 : current;
 527   }
 528   else
 529   {
 530     next      = 1;
 531     *previous = 0;
 532   }
 533
 534   return (next);
 535 }
 536
 537
 538 /*
 539  * 'cupsReadResponseData()' - Read additional data after the IPP response.
 540  *
 541  * This function is used after @link cupsGetResponse@ to read the PPD or document
 542  * files from @code CUPS_GET_PPD@ and @code CUPS_GET_DOCUMENT@ requests,
 543  * respectively.
 544  *
 545  * @since CUPS 1.4/OS X 10.6@
 546  */
 547
 548 ssize_t                                 /* O - Bytes read, 0 on EOF, -1 on error */
 549 cupsReadResponseData(
 550     http_t *http,                       /* I - Connection to server or @code CUPS_HTTP_DEFAULT@ */
 551     char   *buffer,                     /* I - Buffer to use */
 552     size_t length)                      /* I - Number of bytes to read */
 553 {
 554  /*
 555   * Get the default connection as needed...
 556   */
 557
 558   DEBUG_printf(("cupsReadResponseData(http=%p, buffer=%p, "
 559                 "length=" CUPS_LLFMT ")", http, buffer, CUPS_LLCAST length));
 560
 561   if (!http)
 562   {
 563     _cups_globals_t *cg = _cupsGlobals();
 564                                         /* Pointer to library globals */
 565
 566     if ((http = cg->http) == NULL)
 567     {
 568       _cupsSetError(IPP_INTERNAL_ERROR, _("No active connection"), 1);
 569       return (-1);
 570     }
 571   }
 572
 573  /*
 574   * Then read from the HTTP connection...
 575   */
 576
 577   return (httpRead2(http, buffer, length));
 578 }
 579
 580
 581 /*
 582  * 'cupsSendRequest()' - Send an IPP request.
 583  *
 584  * Use @link cupsWriteRequestData@ to write any additional data (document, PPD
 585  * file, etc.) for the request, @link cupsGetResponse@ to get the IPP response,
 586  * and @link cupsReadResponseData@ to read any additional data following the
 587  * response. Only one request can be sent/queued at a time per @code http_t@
 588  * connection.
 589  *
 590  * Returns the initial HTTP status code, which will be @code HTTP_CONTINUE@
 591  * on a successful send of the request.
 592  *
 593  * Note: Unlike @link cupsDoFileRequest@, @link cupsDoIORequest@, and
 594  * @link cupsDoRequest@, the request is NOT freed with @link ippDelete@.
 595  *
 596  * @since CUPS 1.4/OS X 10.6@
 597  */
 598
 599 http_status_t                           /* O - Initial HTTP status */
 600 cupsSendRequest(http_t     *http,       /* I - Connection to server or @code CUPS_HTTP_DEFAULT@ */
 601                 ipp_t      *request,    /* I - IPP request */
 602                 const char *resource,   /* I - Resource path */
 603                 size_t     length)      /* I - Length of data to follow or @code CUPS_LENGTH_VARIABLE@ */
 604 {
 605   http_status_t         status;         /* Status of HTTP request */
 606   int                   got_status;     /* Did we get the status? */
 607   ipp_state_t           state;          /* State of IPP processing */
 608   http_status_t         expect;         /* Expect: header to use */
 609
 610
 611   DEBUG_printf(("cupsSendRequest(http=%p, request=%p(%s), resource=\"%s\", "
 612                 "length=" CUPS_LLFMT ")", http, request,
 613                 request ? ippOpString(request->request.op.operation_id) : "?",
 614                 resource, CUPS_LLCAST length));
 615
 616  /*
 617   * Range check input...
 618   */
 619
 620   if (!request || !resource)
 621   {
 622     _cupsSetError(IPP_INTERNAL_ERROR, strerror(EINVAL), 0);
 623
 624     return (HTTP_ERROR);
 625   }
 626
 627  /*
 628   * Get the default connection as needed...
 629   */
 630
 631   if (!http)
 632     if ((http = _cupsConnect()) == NULL)
 633       return (HTTP_SERVICE_UNAVAILABLE);
 634
 635  /*
 636   * If the prior request was not flushed out, do so now...
 637   */
 638
 639   if (http->state == HTTP_GET_SEND ||
 640       http->state == HTTP_POST_SEND)
 641   {
 642     DEBUG_puts("2cupsSendRequest: Flush prior response.");
 643     httpFlush(http);
 644   }
 645   else if (http->state != HTTP_WAITING)
 646   {
 647     DEBUG_printf(("1cupsSendRequest: Unknown HTTP state (%d), "
 648                   "reconnecting.", http->state));
 649     if (httpReconnect(http))
 650       return (HTTP_ERROR);
 651   }
 652
 653 #ifdef HAVE_SSL
 654  /*
 655   * See if we have an auth-info attribute and are communicating over
 656   * a non-local link.  If so, encrypt the link so that we can pass
 657   * the authentication information securely...
 658   */
 659
 660   if (ippFindAttribute(request, "auth-info", IPP_TAG_TEXT) &&
 661       !httpAddrLocalhost(http->hostaddr) && !http->tls &&
 662       httpEncryption(http, HTTP_ENCRYPT_REQUIRED))
 663   {
 664     DEBUG_puts("1cupsSendRequest: Unable to encrypt connection.");
 665     return (HTTP_SERVICE_UNAVAILABLE);
 666   }
 667 #endif /* HAVE_SSL */
 668
 669  /*
 670   * Reconnect if the last response had a "Connection: close"...
 671   */
 672
 673   if (!_cups_strcasecmp(http->fields[HTTP_FIELD_CONNECTION], "close"))
 674   {
 675     DEBUG_puts("2cupsSendRequest: Connection: close");
 676     httpClearFields(http);
 677     if (httpReconnect(http))
 678     {
 679       DEBUG_puts("1cupsSendRequest: Unable to reconnect.");
 680       return (HTTP_SERVICE_UNAVAILABLE);
 681     }
 682   }
 683
 684  /*
 685   * Loop until we can send the request without authorization problems.
 686   */
 687
 688   expect = HTTP_CONTINUE;
 689
 690   for (;;)
 691   {
 692     DEBUG_puts("2cupsSendRequest: Setup...");
 693
 694    /*
 695     * Setup the HTTP variables needed...
 696     */
 697
 698     httpClearFields(http);
 699     httpSetExpect(http, expect);
 700     httpSetField(http, HTTP_FIELD_CONTENT_TYPE, "application/ipp");
 701     httpSetLength(http, length);
 702
 703 #ifdef HAVE_GSSAPI
 704     if (http->authstring && !strncmp(http->authstring, "Negotiate", 9))
 705     {
 706      /*
 707       * Do not use cached Kerberos credentials since they will look like a
 708       * "replay" attack...
 709       */
 710
 711       _cupsSetNegotiateAuthString(http, "POST", resource);
 712     }
 713 #endif /* HAVE_GSSAPI */
 714
 715     httpSetField(http, HTTP_FIELD_AUTHORIZATION, http->authstring);
 716
 717     DEBUG_printf(("2cupsSendRequest: authstring=\"%s\"", http->authstring));
 718
 719    /*
 720     * Try the request...
 721     */
 722
 723     DEBUG_puts("2cupsSendRequest: Sending HTTP POST...");
 724
 725     if (httpPost(http, resource))
 726     {
 727       DEBUG_puts("2cupsSendRequest: POST failed, reconnecting.");
 728       if (httpReconnect(http))
 729       {
 730         DEBUG_puts("1cupsSendRequest: Unable to reconnect.");
 731         return (HTTP_SERVICE_UNAVAILABLE);
 732       }
 733       else
 734         continue;
 735     }
 736
 737    /*
 738     * Send the IPP data...
 739     */
 740
 741     DEBUG_puts("2cupsSendRequest: Writing IPP request...");
 742
 743     request->state = IPP_IDLE;
 744     status         = HTTP_CONTINUE;
 745     got_status     = 0;
 746
 747     while ((state = ippWrite(http, request)) != IPP_DATA)
 748       if (state == IPP_ERROR)
 749         break;
 750       else if (httpCheck(http))
 751       {
 752         got_status = 1;
 753
 754         _httpUpdate(http, &status);
 755         if (status >= HTTP_MULTIPLE_CHOICES)
 756           break;
 757       }
 758
 759     if (state == IPP_ERROR)
 760     {
 761       DEBUG_puts("1cupsSendRequest: Unable to send IPP request.");
 762
 763       http->status = HTTP_ERROR;
 764       http->state  = HTTP_WAITING;
 765
 766       return (HTTP_ERROR);
 767     }
 768
 769    /*
 770     * Wait up to 1 second to get the 100-continue response as needed...
 771     */
 772
 773     if (!got_status)
 774     {
 775       if (expect == HTTP_CONTINUE)
 776       {
 777         DEBUG_puts("2cupsSendRequest: Waiting for 100-continue...");
 778
 779         if (httpWait(http, 1000))
 780           _httpUpdate(http, &status);
 781       }
 782       else if (httpCheck(http))
 783         _httpUpdate(http, &status);
 784     }
 785
 786     DEBUG_printf(("2cupsSendRequest: status=%d", status));
 787
 788    /*
 789     * Process the current HTTP status...
 790     */
 791
 792     if (status >= HTTP_STATUS_MULTIPLE_CHOICES)
 793     {
 794       int temp_status;                  /* Temporary status */
 795
 796       _cupsSetHTTPError(status);
 797
 798       do
 799       {
 800         temp_status = httpUpdate(http);
 801       }
 802       while (temp_status != HTTP_STATUS_ERROR &&
 803              http->state == HTTP_STATE_POST_RECV);
 804
 805       httpFlush(http);
 806     }
 807
 808     switch (status)
 809     {
 810       case HTTP_STATUS_CONTINUE :
 811       case HTTP_STATUS_OK :
 812       case HTTP_STATUS_ERROR :
 813           DEBUG_printf(("1cupsSendRequest: Returning %d.", status));
 814           return (status);
 815
 816       case HTTP_STATUS_UNAUTHORIZED :
 817           if (cupsDoAuthentication(http, "POST", resource))
 818           {
 819             DEBUG_puts("1cupsSendRequest: Returning HTTP_AUTHORIZATION_CANCELED.");
 820             return (HTTP_AUTHORIZATION_CANCELED);
 821           }
 822
 823           DEBUG_puts("2cupsSendRequest: Reconnecting after HTTP_UNAUTHORIZED.");
 824
 825           if (httpReconnect(http))
 826           {
 827             DEBUG_puts("1cupsSendRequest: Unable to reconnect.");
 828             return (HTTP_SERVICE_UNAVAILABLE);
 829           }
 830           break;
 831
 832 #ifdef HAVE_SSL
 833       case HTTP_STATUS_UPGRADE_REQUIRED :
 834          /*
 835           * Flush any error message, reconnect, and then upgrade with
 836           * encryption...
 837           */
 838
 839           DEBUG_puts("2cupsSendRequest: Reconnecting after "
 840                      "HTTP_UPGRADE_REQUIRED.");
 841
 842           if (httpReconnect(http))
 843           {
 844             DEBUG_puts("1cupsSendRequest: Unable to reconnect.");
 845             return (HTTP_SERVICE_UNAVAILABLE);
 846           }
 847
 848           DEBUG_puts("2cupsSendRequest: Upgrading to TLS.");
 849           if (httpEncryption(http, HTTP_ENCRYPT_REQUIRED))
 850           {
 851             DEBUG_puts("1cupsSendRequest: Unable to encrypt connection.");
 852             return (HTTP_SERVICE_UNAVAILABLE);
 853           }
 854           break;
 855 #endif /* HAVE_SSL */
 856
 857       case HTTP_STATUS_EXPECTATION_FAILED :
 858          /*
 859           * Don't try using the Expect: header the next time around...
 860           */
 861
 862           expect = (http_status_t)0;
 863
 864           DEBUG_puts("2cupsSendRequest: Reconnecting after "
 865                      "HTTP_EXPECTATION_FAILED.");
 866
 867           if (httpReconnect(http))
 868           {
 869             DEBUG_puts("1cupsSendRequest: Unable to reconnect.");
 870             return (HTTP_SERVICE_UNAVAILABLE);
 871           }
 872           break;
 873
 874       default :
 875          /*
 876           * Some other error...
 877           */
 878
 879           return (status);
 880     }
 881   }
 882 }
 883
 884
 885 /*
 886  * 'cupsWriteRequestData()' - Write additional data after an IPP request.
 887  *
 888  * This function is used after @link cupsSendRequest@ to provide a PPD and
 889  * after @link cupsStartDocument@ to provide a document file.
 890  *
 891  * @since CUPS 1.4/OS X 10.6@
 892  */
 893
 894 http_status_t                           /* O - @code HTTP_CONTINUE@ if OK or HTTP status on error */
 895 cupsWriteRequestData(
 896     http_t     *http,                   /* I - Connection to server or @code CUPS_HTTP_DEFAULT@ */
 897     const char *buffer,                 /* I - Bytes to write */
 898     size_t     length)                  /* I - Number of bytes to write */
 899 {
 900   int   wused;                          /* Previous bytes in buffer */
 901
 902
 903  /*
 904   * Get the default connection as needed...
 905   */
 906
 907   DEBUG_printf(("cupsWriteRequestData(http=%p, buffer=%p, "
 908                 "length=" CUPS_LLFMT ")", http, buffer, CUPS_LLCAST length));
 909
 910   if (!http)
 911   {
 912     _cups_globals_t *cg = _cupsGlobals();
 913                                         /* Pointer to library globals */
 914
 915     if ((http = cg->http) == NULL)
 916     {
 917       _cupsSetError(IPP_INTERNAL_ERROR, _("No active connection"), 1);
 918       DEBUG_puts("1cupsWriteRequestData: Returning HTTP_ERROR.");
 919       return (HTTP_ERROR);
 920     }
 921   }
 922
 923  /*
 924   * Then write to the HTTP connection...
 925   */
 926
 927   wused = http->wused;
 928
 929   if (httpWrite2(http, buffer, length) < 0)
 930   {
 931     DEBUG_puts("1cupsWriteRequestData: Returning HTTP_ERROR.");
 932     _cupsSetError(IPP_INTERNAL_ERROR, strerror(http->error), 0);
 933     return (HTTP_ERROR);
 934   }
 935
 936  /*
 937   * Finally, check if we have any pending data from the server...
 938   */
 939
 940   if (length >= HTTP_MAX_BUFFER ||
 941       http->wused < wused ||
 942       (wused > 0 && http->wused == length))
 943   {
 944    /*
 945     * We've written something to the server, so check for response data...
 946     */
 947
 948     if (_httpWait(http, 0, 1))
 949     {
 950       http_status_t     status;         /* Status from _httpUpdate */
 951
 952       _httpUpdate(http, &status);
 953       if (status >= HTTP_MULTIPLE_CHOICES)
 954       {
 955         _cupsSetHTTPError(status);
 956
 957         do
 958         {
 959           status = httpUpdate(http);
 960         }
 961         while (status != HTTP_ERROR && http->state == HTTP_POST_RECV);
 962
 963         httpFlush(http);
 964       }
 965
 966       DEBUG_printf(("1cupsWriteRequestData: Returning %d.\n", status));
 967       return (status);
 968     }
 969   }
 970
 971   DEBUG_puts("1cupsWriteRequestData: Returning HTTP_CONTINUE.");
 972   return (HTTP_CONTINUE);
 973 }
 974
 975
 976 /*
 977  * '_cupsConnect()' - Get the default server connection...
 978  */
 979
 980 http_t *                                /* O - HTTP connection */
 981 _cupsConnect(void)
 982 {
 983   _cups_globals_t *cg = _cupsGlobals(); /* Pointer to library globals */
 984
 985
 986  /*
 987   * See if we are connected to the same server...
 988   */
 989
 990   if (cg->http)
 991   {
 992    /*
 993     * Compare the connection hostname, port, and encryption settings to
 994     * the cached defaults; these were initialized the first time we
 995     * connected...
 996     */
 997
 998     if (strcmp(cg->http->hostname, cg->server) ||
 999         cg->ipp_port != httpAddrPort(cg->http->hostaddr) ||
1000         (cg->http->encryption != cg->encryption &&
1001          cg->http->encryption == HTTP_ENCRYPT_NEVER))
1002     {
1003      /*
1004       * Need to close the current connection because something has changed...
1005       */
1006
1007       httpClose(cg->http);
1008       cg->http = NULL;
1009     }
1010   }
1011
1012  /*
1013   * (Re)connect as needed...
1014   */
1015
1016   if (!cg->http)
1017   {
1018     if ((cg->http = httpConnectEncrypt(cupsServer(), ippPort(),
1019                                        cupsEncryption())) == NULL)
1020     {
1021       if (errno)
1022         _cupsSetError(IPP_SERVICE_UNAVAILABLE, NULL, 0);
1023       else
1024         _cupsSetError(IPP_SERVICE_UNAVAILABLE,
1025                       _("Unable to connect to host."), 1);
1026     }
1027   }
1028
1029  /*
1030   * Return the cached connection...
1031   */
1032
1033   return (cg->http);
1034 }
1035
1036
1037 /*
1038  * '_cupsSetError()' - Set the last IPP status code and status-message.
1039  */
1040
1041 void
1042 _cupsSetError(ipp_status_t status,      /* I - IPP status code */
1043               const char   *message,    /* I - status-message value */
1044               int          localize)    /* I - Localize the message? */
1045 {
1046   _cups_globals_t       *cg;            /* Global data */
1047
1048
1049   if (!message && errno)
1050   {
1051     message  = strerror(errno);
1052     localize = 0;
1053   }
1054
1055   cg             = _cupsGlobals();
1056   cg->last_error = status;
1057
1058   if (cg->last_status_message)
1059   {
1060     _cupsStrFree(cg->last_status_message);
1061
1062     cg->last_status_message = NULL;
1063   }
1064
1065   if (message)
1066   {
1067     if (localize)
1068     {
1069      /*
1070       * Get the message catalog...
1071       */
1072
1073       if (!cg->lang_default)
1074         cg->lang_default = cupsLangDefault();
1075
1076       cg->last_status_message = _cupsStrAlloc(_cupsLangString(cg->lang_default,
1077                                                               message));
1078     }
1079     else
1080       cg->last_status_message = _cupsStrAlloc(message);
1081   }
1082
1083   DEBUG_printf(("4_cupsSetError: last_error=%s, last_status_message=\"%s\"",
1084                 ippErrorString(cg->last_error), cg->last_status_message));
1085 }
1086
1087
1088 /*
1089  * '_cupsSetHTTPError()' - Set the last error using the HTTP status.
1090  */
1091
1092 void
1093 _cupsSetHTTPError(http_status_t status) /* I - HTTP status code */
1094 {
1095   switch (status)
1096   {
1097     case HTTP_NOT_FOUND :
1098         _cupsSetError(IPP_NOT_FOUND, httpStatus(status), 0);
1099         break;
1100
1101     case HTTP_UNAUTHORIZED :
1102         _cupsSetError(IPP_NOT_AUTHENTICATED, httpStatus(status), 0);
1103         break;
1104
1105     case HTTP_AUTHORIZATION_CANCELED :
1106         _cupsSetError(IPP_AUTHENTICATION_CANCELED, httpStatus(status), 0);
1107         break;
1108
1109     case HTTP_FORBIDDEN :
1110         _cupsSetError(IPP_FORBIDDEN, httpStatus(status), 0);
1111         break;
1112
1113     case HTTP_BAD_REQUEST :
1114         _cupsSetError(IPP_BAD_REQUEST, httpStatus(status), 0);
1115         break;
1116
1117     case HTTP_REQUEST_TOO_LARGE :
1118         _cupsSetError(IPP_REQUEST_VALUE, httpStatus(status), 0);
1119         break;
1120
1121     case HTTP_NOT_IMPLEMENTED :
1122         _cupsSetError(IPP_OPERATION_NOT_SUPPORTED, httpStatus(status), 0);
1123         break;
1124
1125     case HTTP_NOT_SUPPORTED :
1126         _cupsSetError(IPP_VERSION_NOT_SUPPORTED, httpStatus(status), 0);
1127         break;
1128
1129     case HTTP_UPGRADE_REQUIRED :
1130         _cupsSetError(IPP_UPGRADE_REQUIRED, httpStatus(status), 0);
1131         break;
1132
1133     case HTTP_PKI_ERROR :
1134         _cupsSetError(IPP_PKI_ERROR, httpStatus(status), 0);
1135         break;
1136
1137     case HTTP_ERROR :
1138         _cupsSetError(IPP_INTERNAL_ERROR, strerror(errno), 0);
1139         break;
1140
1141     default :
1142         DEBUG_printf(("4_cupsSetHTTPError: HTTP error %d mapped to "
1143                       "IPP_SERVICE_UNAVAILABLE!", status));
1144         _cupsSetError(IPP_SERVICE_UNAVAILABLE, httpStatus(status), 0);
1145         break;
1146   }
1147 }
1148
1149
1150 /*
1151  * End of "$Id: request.c 7946 2008-09-16 23:27:54Z mike $".
1152  */