s3: Remove unused "retry" from cli_full_connection
[nivanova/samba-autobuild/.git] / source3 / printing / print_cups.c
index b9bed7a138c7d1eebd8a5313dd59b06b009fe0ea..a85fba89974f316c5e49e33f6d1942b5b87a2d2e 100644 (file)
@@ -24,6 +24,7 @@
 
 #include "includes.h"
 #include "printing.h"
+#include "printing/pcap.h"
 
 #ifdef HAVE_CUPS
 #include <cups/cups.h>
@@ -35,7 +36,7 @@ static SIG_ATOMIC_T gotalarm;
  Signal function to tell us we timed out.
 ****************************************************************/
 
-static void gotalarm_sig(void)
+static void gotalarm_sig(int signum)
 {
         gotalarm = 1;
 }
@@ -89,13 +90,18 @@ static http_t *cups_connect(TALLOC_CTX *frame)
        gotalarm = 0;
 
        if (timeout) {
-                CatchSignal(SIGALRM, SIGNAL_CAST gotalarm_sig);
+                CatchSignal(SIGALRM, gotalarm_sig);
                 alarm(timeout);
         }
 
+#ifdef HAVE_HTTPCONNECTENCRYPT
+       http = httpConnectEncrypt(server, port, lp_cups_encrypt());
+#else
        http = httpConnect(server, port);
+#endif
+
 
-       CatchSignal(SIGALRM, SIGNAL_CAST SIG_IGN);
+       CatchSignal(SIGALRM, SIG_IGN);
         alarm(0);
 
        if (http == NULL) {
@@ -106,9 +112,46 @@ static http_t *cups_connect(TALLOC_CTX *frame)
        return http;
 }
 
-bool cups_cache_reload(void)
+static void send_pcap_info(const char *name, const char *info, void *pd)
+{
+       int fd = *(int *)pd;
+       size_t namelen = name ? strlen(name)+1 : 0;
+       size_t infolen = info ? strlen(info)+1 : 0;
+
+       DEBUG(11,("send_pcap_info: writing namelen %u\n", (unsigned int)namelen));
+       if (sys_write(fd, &namelen, sizeof(namelen)) != sizeof(namelen)) {
+               DEBUG(10,("send_pcap_info: namelen write failed %s\n",
+                       strerror(errno)));
+               return;
+       }
+       DEBUG(11,("send_pcap_info: writing infolen %u\n", (unsigned int)infolen));
+       if (sys_write(fd, &infolen, sizeof(infolen)) != sizeof(infolen)) {
+               DEBUG(10,("send_pcap_info: infolen write failed %s\n",
+                       strerror(errno)));
+               return;
+       }
+       if (namelen) {
+               DEBUG(11,("send_pcap_info: writing name %s\n", name));
+               if (sys_write(fd, name, namelen) != namelen) {
+                       DEBUG(10,("send_pcap_info: name write failed %s\n",
+                               strerror(errno)));
+                       return;
+               }
+       }
+       if (infolen) {
+               DEBUG(11,("send_pcap_info: writing info %s\n", info));
+               if (sys_write(fd, info, infolen) != infolen) {
+                       DEBUG(10,("send_pcap_info: info write failed %s\n",
+                               strerror(errno)));
+                       return;
+               }
+       }
+}
+
+static bool cups_cache_reload_async(int fd)
 {
        TALLOC_CTX *frame = talloc_stackframe();
+       struct pcap_cache *tmp_pcap_cache = NULL;
        http_t          *http = NULL;           /* HTTP connection to server */
        ipp_t           *request = NULL,        /* IPP Request */
                        *response = NULL;       /* IPP Response */
@@ -226,7 +269,7 @@ bool cups_cache_reload(void)
                if (name == NULL)
                        break;
 
-               if (!pcap_cache_add(name, info)) {
+               if (!pcap_cache_add_specific(&tmp_pcap_cache, name, info)) {
                        goto out;
                }
        }
@@ -318,7 +361,7 @@ bool cups_cache_reload(void)
                if (name == NULL)
                        break;
 
-               if (!pcap_cache_add(name, info)) {
+               if (!pcap_cache_add_specific(&tmp_pcap_cache, name, info)) {
                        goto out;
                }
        }
@@ -335,10 +378,236 @@ bool cups_cache_reload(void)
        if (http)
                httpClose(http);
 
+       /* Send all the entries up the pipe. */
+       if (tmp_pcap_cache) {
+               pcap_printer_fn_specific(tmp_pcap_cache,
+                               send_pcap_info,
+                               (void *)&fd);
+
+               pcap_cache_destroy_specific(&tmp_pcap_cache);
+       }
        TALLOC_FREE(frame);
        return ret;
 }
 
+static struct pcap_cache *local_pcap_copy;
+struct fd_event *cache_fd_event;
+
+static bool cups_pcap_load_async(struct tevent_context *ev,
+                                struct messaging_context *msg_ctx,
+                                int *pfd)
+{
+       int fds[2];
+       pid_t pid;
+       NTSTATUS status;
+
+       *pfd = -1;
+
+       if (cache_fd_event) {
+               DEBUG(3,("cups_pcap_load_async: already waiting for "
+                       "a refresh event\n" ));
+               return false;
+       }
+
+       DEBUG(5,("cups_pcap_load_async: asynchronously loading cups printers\n"));
+
+       if (pipe(fds) == -1) {
+               return false;
+       }
+
+       pid = sys_fork();
+       if (pid == (pid_t)-1) {
+               DEBUG(10,("cups_pcap_load_async: fork failed %s\n",
+                       strerror(errno) ));
+               close(fds[0]);
+               close(fds[1]);
+               return false;
+       }
+
+       if (pid) {
+               DEBUG(10,("cups_pcap_load_async: child pid = %u\n",
+                       (unsigned int)pid ));
+               /* Parent. */
+               close(fds[1]);
+               *pfd = fds[0];
+               return true;
+       }
+
+       /* Child. */
+
+       close_all_print_db();
+
+       status = reinit_after_fork(msg_ctx, ev, procid_self(), true);
+       if (!NT_STATUS_IS_OK(status)) {
+               DEBUG(0,("cups_pcap_load_async: reinit_after_fork() failed\n"));
+               smb_panic("cups_pcap_load_async: reinit_after_fork() failed");
+       }
+
+       close(fds[0]);
+       cups_cache_reload_async(fds[1]);
+       close(fds[1]);
+       _exit(0);
+}
+
+static void cups_async_callback(struct event_context *event_ctx,
+                               struct fd_event *event,
+                               uint16 flags,
+                               void *p)
+{
+       TALLOC_CTX *frame = talloc_stackframe();
+       int fd = *(int *)p;
+       struct pcap_cache *tmp_pcap_cache = NULL;
+
+       DEBUG(5,("cups_async_callback: callback received for printer data. "
+               "fd = %d\n", fd));
+
+       while (1) {
+               char *name = NULL, *info = NULL;
+               size_t namelen = 0, infolen = 0;
+               ssize_t ret = -1;
+
+               ret = sys_read(fd, &namelen, sizeof(namelen));
+               if (ret == 0) {
+                       /* EOF */
+                       break;
+               }
+               if (ret != sizeof(namelen)) {
+                       DEBUG(10,("cups_async_callback: namelen read failed %d %s\n",
+                               errno, strerror(errno)));
+                       break;
+               }
+
+               DEBUG(11,("cups_async_callback: read namelen %u\n",
+                       (unsigned int)namelen));
+
+               ret = sys_read(fd, &infolen, sizeof(infolen));
+               if (ret == 0) {
+                       /* EOF */
+                       break;
+               }
+               if (ret != sizeof(infolen)) {
+                       DEBUG(10,("cups_async_callback: infolen read failed %s\n",
+                               strerror(errno)));
+                       break;
+               }
+
+               DEBUG(11,("cups_async_callback: read infolen %u\n",
+                       (unsigned int)infolen));
+
+               if (namelen) {
+                       name = TALLOC_ARRAY(frame, char, namelen);
+                       if (!name) {
+                               break;
+                       }
+                       ret = sys_read(fd, name, namelen);
+                       if (ret == 0) {
+                               /* EOF */
+                               break;
+                       }
+                       if (ret != namelen) {
+                               DEBUG(10,("cups_async_callback: name read failed %s\n",
+                                       strerror(errno)));
+                               break;
+                       }
+                       DEBUG(11,("cups_async_callback: read name %s\n",
+                               name));
+               } else {
+                       name = NULL;
+               }
+               if (infolen) {
+                       info = TALLOC_ARRAY(frame, char, infolen);
+                       if (!info) {
+                               break;
+                       }
+                       ret = sys_read(fd, info, infolen);
+                       if (ret == 0) {
+                               /* EOF */
+                               break;
+                       }
+                       if (ret != infolen) {
+                               DEBUG(10,("cups_async_callback: info read failed %s\n",
+                                       strerror(errno)));
+                               break;
+                       }
+                       DEBUG(11,("cups_async_callback: read info %s\n",
+                               info));
+               } else {
+                       info = NULL;
+               }
+
+               /* Add to our local pcap cache. */
+               pcap_cache_add_specific(&tmp_pcap_cache, name, info);
+               TALLOC_FREE(name);
+               TALLOC_FREE(info);
+       }
+
+       TALLOC_FREE(frame);
+       if (tmp_pcap_cache) {
+               /* We got a namelist, replace our local cache. */
+               pcap_cache_destroy_specific(&local_pcap_copy);
+               local_pcap_copy = tmp_pcap_cache;
+
+               /* And the systemwide pcap cache. */
+               pcap_cache_replace(local_pcap_copy);
+       } else {
+               DEBUG(2,("cups_async_callback: failed to read a new "
+                       "printer list\n"));
+       }
+       close(fd);
+       TALLOC_FREE(p);
+       TALLOC_FREE(cache_fd_event);
+}
+
+bool cups_cache_reload(struct tevent_context *ev,
+                      struct messaging_context *msg_ctx)
+{
+       int *p_pipe_fd = TALLOC_P(NULL, int);
+
+       if (!p_pipe_fd) {
+               return false;
+       }
+
+       *p_pipe_fd = -1;
+
+       /* Set up an async refresh. */
+       if (!cups_pcap_load_async(ev, msg_ctx, p_pipe_fd)) {
+               return false;
+       }
+       if (!local_pcap_copy) {
+               /* We have no local cache, wait directly for
+                * async refresh to complete.
+                */
+               DEBUG(10,("cups_cache_reload: sync read on fd %d\n",
+                       *p_pipe_fd ));
+
+               cups_async_callback(ev, NULL,
+                                       EVENT_FD_READ,
+                                       (void *)p_pipe_fd);
+               if (!local_pcap_copy) {
+                       return false;
+               }
+       } else {
+               /* Replace the system cache with our
+                * local copy. */
+               pcap_cache_replace(local_pcap_copy);
+
+               DEBUG(10,("cups_cache_reload: async read on fd %d\n",
+                       *p_pipe_fd ));
+
+               /* Trigger an event when the pipe can be read. */
+               cache_fd_event = event_add_fd(ev,
+                                       NULL, *p_pipe_fd,
+                                       EVENT_FD_READ,
+                                       cups_async_callback,
+                                       (void *)p_pipe_fd);
+               if (!cache_fd_event) {
+                       close(*p_pipe_fd);
+                       TALLOC_FREE(p_pipe_fd);
+                       return false;
+               }
+       }
+       return true;
+}
 
 /*
  * 'cups_job_delete()' - Delete a job.
@@ -640,9 +909,9 @@ static int cups_job_submit(int snum, struct printjob *pjob)
        http_t          *http = NULL;           /* HTTP connection to server */
        ipp_t           *request = NULL,        /* IPP Request */
                        *response = NULL;       /* IPP Response */
+       ipp_attribute_t *attr_job_id = NULL;    /* IPP Attribute "job-id" */
        cups_lang_t     *language = NULL;       /* Default language */
        char            uri[HTTP_MAX_URI]; /* printer-uri attribute */
-       const char      *clientname = NULL;     /* hostname of client for job-originating-host attribute */
        char *new_jobname = NULL;
        int             num_options = 0;
        cups_option_t   *options = NULL;
@@ -652,9 +921,9 @@ static int cups_job_submit(int snum, struct printjob *pjob)
        char *cupsoptions = NULL;
        char *filename = NULL;
        size_t size;
-       char addr[INET6_ADDRSTRLEN];
+       uint32_t jobid = (uint32_t)-1;
 
-       DEBUG(5,("cups_job_submit(%d, %p (%d))\n", snum, pjob, pjob->sysjob));
+       DEBUG(5,("cups_job_submit(%d, %p)\n", snum, pjob));
 
        /*
         * Make sure we don't ask for passwords...
@@ -694,7 +963,8 @@ static int cups_job_submit(int snum, struct printjob *pjob)
        ippAddString(request, IPP_TAG_OPERATION, IPP_TAG_LANGUAGE,
                     "attributes-natural-language", NULL, language->language);
 
-       if (!push_utf8_talloc(frame, &printername, PRINTERNAME(snum), &size)) {
+       if (!push_utf8_talloc(frame, &printername, lp_printername(snum),
+                             &size)) {
                goto out;
        }
        slprintf(uri, sizeof(uri) - 1, "ipp://localhost/printers/%s",
@@ -709,21 +979,24 @@ static int cups_job_submit(int snum, struct printjob *pjob)
        ippAddString(request, IPP_TAG_OPERATION, IPP_TAG_NAME, "requesting-user-name",
                     NULL, user);
 
-       clientname = client_name(get_client_fd());
-       if (strcmp(clientname, "UNKNOWN") == 0) {
-               clientname = client_addr(get_client_fd(),addr,sizeof(addr));
-       }
-
        ippAddString(request, IPP_TAG_OPERATION, IPP_TAG_NAME,
                     "job-originating-host-name", NULL,
-                     clientname);
+                    pjob->clientmachine);
+
+       /* Get the jobid from the filename. */
+       jobid = print_parse_jobid(pjob->filename);
+       if (jobid == (uint32_t)-1) {
+               DEBUG(0,("cups_job_submit: failed to parse jobid from name %s\n",
+                               pjob->filename ));
+               jobid = 0;
+       }
 
        if (!push_utf8_talloc(frame, &jobname, pjob->jobname, &size)) {
                goto out;
        }
        new_jobname = talloc_asprintf(frame,
                        "%s%.8u %s", PRINT_SPOOL_PREFIX,
-                       (unsigned int)pjob->smbjob,
+                       (unsigned int)jobid,
                        jobname);
        if (new_jobname == NULL) {
                goto out;
@@ -757,13 +1030,22 @@ static int cups_job_submit(int snum, struct printjob *pjob)
        }
        if ((response = cupsDoFileRequest(http, request, uri, pjob->filename)) != NULL) {
                if (response->request.status.status_code >= IPP_OK_CONFLICT) {
-                       DEBUG(0,("Unable to print file to %s - %s\n", PRINTERNAME(snum),
+                       DEBUG(0,("Unable to print file to %s - %s\n",
+                                lp_printername(snum),
                                 ippErrorString(cupsLastError())));
                } else {
                        ret = 0;
+                       attr_job_id = ippFindAttribute(response, "job-id", IPP_TAG_INTEGER);
+                       if(attr_job_id) {
+                               pjob->sysjob = attr_job_id->values[0].integer;
+                               DEBUG(5,("cups_job_submit: job-id %d\n", pjob->sysjob));
+                       } else {
+                               DEBUG(0,("Missing job-id attribute in IPP response"));
+                       }
                }
        } else {
-               DEBUG(0,("Unable to print file to `%s' - %s\n", PRINTERNAME(snum),
+               DEBUG(0,("Unable to print file to `%s' - %s\n",
+                        lp_printername(snum),
                         ippErrorString(cupsLastError())));
        }
 
@@ -1077,14 +1359,12 @@ static int cups_queue_get(const char *sharename,
        if ((response = cupsDoRequest(http, request, "/")) == NULL) {
                DEBUG(0,("Unable to get printer status for %s - %s\n", printername,
                         ippErrorString(cupsLastError())));
-               *q = queue;
                goto out;
        }
 
        if (response->request.status.status_code >= IPP_OK_CONFLICT) {
                DEBUG(0,("Unable to get printer status for %s - %s\n", printername,
                         ippErrorString(response->request.status.status_code)));
-               *q = queue;
                goto out;
        }
 
@@ -1105,18 +1385,21 @@ static int cups_queue_get(const char *sharename,
                if (!pull_utf8_talloc(frame, &msg,
                                attr->values[0].string.text,
                                &size)) {
+                       SAFE_FREE(queue);
+                       qcount = 0;
                        goto out;
                }
                fstrcpy(status->message, msg);
        }
 
+ out:
+
        /*
         * Return the job queue...
        */
 
        *q = queue;
 
- out:
        if (response)
                ippDelete(response);
 
@@ -1187,7 +1470,8 @@ static int cups_queue_pause(int snum)
        ippAddString(request, IPP_TAG_OPERATION, IPP_TAG_LANGUAGE,
                     "attributes-natural-language", NULL, language->language);
 
-       if (!push_utf8_talloc(frame, &printername, PRINTERNAME(snum), &size)) {
+       if (!push_utf8_talloc(frame, &printername, lp_printername(snum),
+                             &size)) {
                goto out;
        }
        slprintf(uri, sizeof(uri) - 1, "ipp://localhost/printers/%s",
@@ -1207,13 +1491,15 @@ static int cups_queue_pause(int snum)
 
        if ((response = cupsDoRequest(http, request, "/admin/")) != NULL) {
                if (response->request.status.status_code >= IPP_OK_CONFLICT) {
-                       DEBUG(0,("Unable to pause printer %s - %s\n", PRINTERNAME(snum),
+                       DEBUG(0,("Unable to pause printer %s - %s\n",
+                                lp_printername(snum),
                                ippErrorString(cupsLastError())));
                } else {
                        ret = 0;
                }
        } else {
-               DEBUG(0,("Unable to pause printer %s - %s\n", PRINTERNAME(snum),
+               DEBUG(0,("Unable to pause printer %s - %s\n",
+                        lp_printername(snum),
                        ippErrorString(cupsLastError())));
        }
 
@@ -1288,7 +1574,8 @@ static int cups_queue_resume(int snum)
        ippAddString(request, IPP_TAG_OPERATION, IPP_TAG_LANGUAGE,
                     "attributes-natural-language", NULL, language->language);
 
-       if (!push_utf8_talloc(frame, &printername, PRINTERNAME(snum), &size)) {
+       if (!push_utf8_talloc(frame, &printername, lp_printername(snum),
+                             &size)) {
                goto out;
        }
        slprintf(uri, sizeof(uri) - 1, "ipp://localhost/printers/%s",
@@ -1308,13 +1595,15 @@ static int cups_queue_resume(int snum)
 
        if ((response = cupsDoRequest(http, request, "/admin/")) != NULL) {
                if (response->request.status.status_code >= IPP_OK_CONFLICT) {
-                       DEBUG(0,("Unable to resume printer %s - %s\n", PRINTERNAME(snum),
+                       DEBUG(0,("Unable to resume printer %s - %s\n",
+                                lp_printername(snum),
                                ippErrorString(cupsLastError())));
                } else {
                        ret = 0;
                }
        } else {
-               DEBUG(0,("Unable to resume printer %s - %s\n", PRINTERNAME(snum),
+               DEBUG(0,("Unable to resume printer %s - %s\n",
+                        lp_printername(snum),
                        ippErrorString(cupsLastError())));
        }
 
@@ -1348,7 +1637,10 @@ struct printif   cups_printif =
        cups_job_submit,
 };
 
-bool cups_pull_comment_location(NT_PRINTER_INFO_LEVEL_2 *printer)
+bool cups_pull_comment_location(TALLOC_CTX *mem_ctx,
+                               const char *printername,
+                               char **comment,
+                               char **location)
 {
        TALLOC_CTX *frame = talloc_stackframe();
        http_t          *http = NULL;           /* HTTP connection to server */
@@ -1369,7 +1661,7 @@ bool cups_pull_comment_location(NT_PRINTER_INFO_LEVEL_2 *printer)
        bool ret = False;
        size_t size;
 
-       DEBUG(5, ("pulling %s location\n", printer->sharename));
+       DEBUG(5, ("pulling %s location\n", printername));
 
        /*
         * Make sure we don't ask for passwords...
@@ -1408,7 +1700,7 @@ bool cups_pull_comment_location(NT_PRINTER_INFO_LEVEL_2 *printer)
        if (server) {
                goto out;
        }
-       if (!push_utf8_talloc(frame, &sharename, printer->sharename, &size)) {
+       if (!push_utf8_talloc(frame, &sharename, printername, &size)) {
                goto out;
        }
        slprintf(uri, sizeof(uri) - 1, "ipp://%s/printers/%s",
@@ -1460,40 +1752,30 @@ bool cups_pull_comment_location(NT_PRINTER_INFO_LEVEL_2 *printer)
 
                        /* Grab the comment if we don't have one */
                        if ( (strcmp(attr->name, "printer-info") == 0)
-                            && (attr->value_tag == IPP_TAG_TEXT)
-                            && !strlen(printer->comment) )
+                            && (attr->value_tag == IPP_TAG_TEXT))
                        {
-                               char *comment = NULL;
-                               if (!pull_utf8_talloc(frame,
-                                               &comment,
+                               if (!pull_utf8_talloc(mem_ctx,
+                                               comment,
                                                attr->values[0].string.text,
                                                &size)) {
                                        goto out;
                                }
                                DEBUG(5,("cups_pull_comment_location: Using cups comment: %s\n",
-                                        comment));
-                               strlcpy(printer->comment,
-                                       comment,
-                                       sizeof(printer->comment));
+                                        *comment));
                        }
 
                        /* Grab the location if we don't have one */
                        if ( (strcmp(attr->name, "printer-location") == 0)
-                            && (attr->value_tag == IPP_TAG_TEXT)
-                            && !strlen(printer->location) )
+                            && (attr->value_tag == IPP_TAG_TEXT))
                        {
-                               char *location = NULL;
-                               if (!pull_utf8_talloc(frame,
-                                               &location,
+                               if (!pull_utf8_talloc(mem_ctx,
+                                               location,
                                                attr->values[0].string.text,
                                                &size)) {
                                        goto out;
                                }
                                DEBUG(5,("cups_pull_comment_location: Using cups location: %s\n",
-                                        location));
-                               strlcpy(printer->location,
-                                       location,
-                                       sizeof(printer->location));
+                                        *location));
                        }
 
                        attr = attr->next;
@@ -1513,6 +1795,10 @@ bool cups_pull_comment_location(NT_PRINTER_INFO_LEVEL_2 *printer)
        if (response)
                ippDelete(response);
 
+       if (request) {
+               ippDelete(request);
+       }
+
        if (language)
                cupsLangFree(language);