s3:registry: do not use regdb functions during db upgrade
[Samba/gebeck_regimport.git] / source3 / printing / printing.c
blob019fc65c23a8c7689caac0d73d2e29ff9a421b59
1 /*
2 Unix SMB/Netbios implementation.
3 Version 3.0
4 printing backend routines
5 Copyright (C) Andrew Tridgell 1992-2000
6 Copyright (C) Jeremy Allison 2002
8 This program is free software; you can redistribute it and/or modify
9 it under the terms of the GNU General Public License as published by
10 the Free Software Foundation; either version 3 of the License, or
11 (at your option) any later version.
13 This program is distributed in the hope that it will be useful,
14 but WITHOUT ANY WARRANTY; without even the implied warranty of
15 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
16 GNU General Public License for more details.
18 You should have received a copy of the GNU General Public License
19 along with this program. If not, see <http://www.gnu.org/licenses/>.
22 #include "includes.h"
23 #include "system/syslog.h"
24 #include "system/filesys.h"
25 #include "printing.h"
26 #include "../librpc/gen_ndr/ndr_spoolss.h"
27 #include "nt_printing.h"
28 #include "../librpc/gen_ndr/netlogon.h"
29 #include "printing/notify.h"
30 #include "printing/pcap.h"
31 #include "printing/printer_list.h"
32 #include "printing/queue_process.h"
33 #include "serverid.h"
34 #include "smbd/smbd.h"
35 #include "auth.h"
36 #include "messages.h"
37 #include "util_tdb.h"
38 #include "lib/param/loadparm.h"
40 extern struct current_user current_user;
41 extern userdom_struct current_user_info;
43 /* Current printer interface */
44 static bool remove_from_jobs_added(const char* sharename, uint32 jobid);
47 the printing backend revolves around a tdb database that stores the
48 SMB view of the print queue
50 The key for this database is a jobid - a internally generated number that
51 uniquely identifies a print job
53 reading the print queue involves two steps:
54 - possibly running lpq and updating the internal database from that
55 - reading entries from the database
57 jobids are assigned when a job starts spooling.
60 static TDB_CONTEXT *rap_tdb;
61 static uint16 next_rap_jobid;
62 struct rap_jobid_key {
63 fstring sharename;
64 uint32 jobid;
67 /***************************************************************************
68 Nightmare. LANMAN jobid's are 16 bit numbers..... We must map them to 32
69 bit RPC jobids.... JRA.
70 ***************************************************************************/
72 uint16 pjobid_to_rap(const char* sharename, uint32 jobid)
74 uint16 rap_jobid;
75 TDB_DATA data, key;
76 struct rap_jobid_key jinfo;
77 uint8 buf[2];
79 DEBUG(10,("pjobid_to_rap: called.\n"));
81 if (!rap_tdb) {
82 /* Create the in-memory tdb. */
83 rap_tdb = tdb_open_log(NULL, 0, TDB_INTERNAL, (O_RDWR|O_CREAT), 0644);
84 if (!rap_tdb)
85 return 0;
88 ZERO_STRUCT( jinfo );
89 fstrcpy( jinfo.sharename, sharename );
90 jinfo.jobid = jobid;
91 key.dptr = (uint8 *)&jinfo;
92 key.dsize = sizeof(jinfo);
94 data = tdb_fetch_compat(rap_tdb, key);
95 if (data.dptr && data.dsize == sizeof(uint16)) {
96 rap_jobid = SVAL(data.dptr, 0);
97 SAFE_FREE(data.dptr);
98 DEBUG(10,("pjobid_to_rap: jobid %u maps to RAP jobid %u\n",
99 (unsigned int)jobid, (unsigned int)rap_jobid));
100 return rap_jobid;
102 SAFE_FREE(data.dptr);
103 /* Not found - create and store mapping. */
104 rap_jobid = ++next_rap_jobid;
105 if (rap_jobid == 0)
106 rap_jobid = ++next_rap_jobid;
107 SSVAL(buf,0,rap_jobid);
108 data.dptr = buf;
109 data.dsize = sizeof(rap_jobid);
110 tdb_store(rap_tdb, key, data, TDB_REPLACE);
111 tdb_store(rap_tdb, data, key, TDB_REPLACE);
113 DEBUG(10,("pjobid_to_rap: created jobid %u maps to RAP jobid %u\n",
114 (unsigned int)jobid, (unsigned int)rap_jobid));
115 return rap_jobid;
118 bool rap_to_pjobid(uint16 rap_jobid, fstring sharename, uint32 *pjobid)
120 TDB_DATA data, key;
121 uint8 buf[2];
123 DEBUG(10,("rap_to_pjobid called.\n"));
125 if (!rap_tdb)
126 return False;
128 SSVAL(buf,0,rap_jobid);
129 key.dptr = buf;
130 key.dsize = sizeof(rap_jobid);
131 data = tdb_fetch_compat(rap_tdb, key);
132 if ( data.dptr && data.dsize == sizeof(struct rap_jobid_key) )
134 struct rap_jobid_key *jinfo = (struct rap_jobid_key*)data.dptr;
135 if (sharename != NULL) {
136 fstrcpy( sharename, jinfo->sharename );
138 *pjobid = jinfo->jobid;
139 DEBUG(10,("rap_to_pjobid: jobid %u maps to RAP jobid %u\n",
140 (unsigned int)*pjobid, (unsigned int)rap_jobid));
141 SAFE_FREE(data.dptr);
142 return True;
145 DEBUG(10,("rap_to_pjobid: Failed to lookup RAP jobid %u\n",
146 (unsigned int)rap_jobid));
147 SAFE_FREE(data.dptr);
148 return False;
151 void rap_jobid_delete(const char* sharename, uint32 jobid)
153 TDB_DATA key, data;
154 uint16 rap_jobid;
155 struct rap_jobid_key jinfo;
156 uint8 buf[2];
158 DEBUG(10,("rap_jobid_delete: called.\n"));
160 if (!rap_tdb)
161 return;
163 ZERO_STRUCT( jinfo );
164 fstrcpy( jinfo.sharename, sharename );
165 jinfo.jobid = jobid;
166 key.dptr = (uint8 *)&jinfo;
167 key.dsize = sizeof(jinfo);
169 data = tdb_fetch_compat(rap_tdb, key);
170 if (!data.dptr || (data.dsize != sizeof(uint16))) {
171 DEBUG(10,("rap_jobid_delete: cannot find jobid %u\n",
172 (unsigned int)jobid ));
173 SAFE_FREE(data.dptr);
174 return;
177 DEBUG(10,("rap_jobid_delete: deleting jobid %u\n",
178 (unsigned int)jobid ));
180 rap_jobid = SVAL(data.dptr, 0);
181 SAFE_FREE(data.dptr);
182 SSVAL(buf,0,rap_jobid);
183 data.dptr = buf;
184 data.dsize = sizeof(rap_jobid);
185 tdb_delete(rap_tdb, key);
186 tdb_delete(rap_tdb, data);
189 static int get_queue_status(const char* sharename, print_status_struct *);
191 /****************************************************************************
192 Initialise the printing backend. Called once at startup before the fork().
193 ****************************************************************************/
195 bool print_backend_init(struct messaging_context *msg_ctx)
197 const char *sversion = "INFO/version";
198 int services = lp_numservices();
199 int snum;
201 if (!printer_list_parent_init()) {
202 return false;
205 unlink(cache_path("printing.tdb"));
206 mkdir(cache_path("printing"),0755);
208 /* handle a Samba upgrade */
210 for (snum = 0; snum < services; snum++) {
211 struct tdb_print_db *pdb;
212 if (!lp_print_ok(snum))
213 continue;
215 pdb = get_print_db_byname(lp_const_servicename(snum));
216 if (!pdb)
217 continue;
218 if (tdb_lock_bystring(pdb->tdb, sversion) != 0) {
219 DEBUG(0,("print_backend_init: Failed to open printer %s database\n", lp_const_servicename(snum) ));
220 release_print_db(pdb);
221 return False;
223 if (tdb_fetch_int32(pdb->tdb, sversion) != PRINT_DATABASE_VERSION) {
224 tdb_wipe_all(pdb->tdb);
225 tdb_store_int32(pdb->tdb, sversion, PRINT_DATABASE_VERSION);
227 tdb_unlock_bystring(pdb->tdb, sversion);
228 release_print_db(pdb);
231 close_all_print_db(); /* Don't leave any open. */
233 /* do NT print initialization... */
234 return nt_printing_init(msg_ctx);
237 /****************************************************************************
238 Shut down printing backend. Called once at shutdown to close the tdb.
239 ****************************************************************************/
241 void printing_end(void)
243 close_all_print_db(); /* Don't leave any open. */
246 /****************************************************************************
247 Retrieve the set of printing functions for a given service. This allows
248 us to set the printer function table based on the value of the 'printing'
249 service parameter.
251 Use the generic interface as the default and only use cups interface only
252 when asked for (and only when supported)
253 ****************************************************************************/
255 static struct printif *get_printer_fns_from_type( enum printing_types type )
257 struct printif *printer_fns = &generic_printif;
259 #ifdef HAVE_CUPS
260 if ( type == PRINT_CUPS ) {
261 printer_fns = &cups_printif;
263 #endif /* HAVE_CUPS */
265 #ifdef HAVE_IPRINT
266 if ( type == PRINT_IPRINT ) {
267 printer_fns = &iprint_printif;
269 #endif /* HAVE_IPRINT */
271 printer_fns->type = type;
273 return printer_fns;
276 static struct printif *get_printer_fns( int snum )
278 return get_printer_fns_from_type( (enum printing_types)lp_printing(snum) );
282 /****************************************************************************
283 Useful function to generate a tdb key.
284 ****************************************************************************/
286 static TDB_DATA print_key(uint32 jobid, uint32 *tmp)
288 TDB_DATA ret;
290 SIVAL(tmp, 0, jobid);
291 ret.dptr = (uint8 *)tmp;
292 ret.dsize = sizeof(*tmp);
293 return ret;
296 /****************************************************************************
297 Pack the devicemode to store it in a tdb.
298 ****************************************************************************/
299 static int pack_devicemode(struct spoolss_DeviceMode *devmode, uint8 *buf, int buflen)
301 enum ndr_err_code ndr_err;
302 DATA_BLOB blob;
303 int len = 0;
305 if (devmode) {
306 ndr_err = ndr_push_struct_blob(&blob, talloc_tos(),
307 devmode,
308 (ndr_push_flags_fn_t)
309 ndr_push_spoolss_DeviceMode);
310 if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
311 DEBUG(10, ("pack_devicemode: "
312 "error encoding spoolss_DeviceMode\n"));
313 goto done;
315 } else {
316 ZERO_STRUCT(blob);
319 len = tdb_pack(buf, buflen, "B", blob.length, blob.data);
321 if (devmode) {
322 DEBUG(8, ("Packed devicemode [%s]\n", devmode->formname));
325 done:
326 return len;
329 /****************************************************************************
330 Unpack the devicemode to store it in a tdb.
331 ****************************************************************************/
332 static int unpack_devicemode(TALLOC_CTX *mem_ctx,
333 const uint8 *buf, int buflen,
334 struct spoolss_DeviceMode **devmode)
336 struct spoolss_DeviceMode *dm;
337 enum ndr_err_code ndr_err;
338 char *data = NULL;
339 int data_len = 0;
340 DATA_BLOB blob;
341 int len = 0;
343 *devmode = NULL;
345 len = tdb_unpack(buf, buflen, "B", &data_len, &data);
346 if (!data) {
347 return len;
350 dm = talloc_zero(mem_ctx, struct spoolss_DeviceMode);
351 if (!dm) {
352 goto done;
355 blob = data_blob_const(data, data_len);
357 ndr_err = ndr_pull_struct_blob(&blob, dm, dm,
358 (ndr_pull_flags_fn_t)ndr_pull_spoolss_DeviceMode);
359 if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
360 DEBUG(10, ("unpack_devicemode: "
361 "error parsing spoolss_DeviceMode\n"));
362 goto done;
365 DEBUG(8, ("Unpacked devicemode [%s](%s)\n",
366 dm->devicename, dm->formname));
367 if (dm->driverextra_data.data) {
368 DEBUG(8, ("with a private section of %d bytes\n",
369 dm->__driverextra_length));
372 *devmode = dm;
374 done:
375 SAFE_FREE(data);
376 return len;
379 /***********************************************************************
380 unpack a pjob from a tdb buffer
381 ***********************************************************************/
383 static int unpack_pjob( uint8 *buf, int buflen, struct printjob *pjob )
385 int len = 0;
386 int used;
387 uint32 pjpid, pjsysjob, pjfd, pjstarttime, pjstatus;
388 uint32 pjsize, pjpage_count, pjspooled, pjsmbjob;
390 if ( !buf || !pjob )
391 return -1;
393 len += tdb_unpack(buf+len, buflen-len, "dddddddddfffff",
394 &pjpid,
395 &pjsysjob,
396 &pjfd,
397 &pjstarttime,
398 &pjstatus,
399 &pjsize,
400 &pjpage_count,
401 &pjspooled,
402 &pjsmbjob,
403 pjob->filename,
404 pjob->jobname,
405 pjob->user,
406 pjob->clientmachine,
407 pjob->queuename);
409 if ( len == -1 )
410 return -1;
412 used = unpack_devicemode(NULL, buf+len, buflen-len, &pjob->devmode);
413 if (used == -1) {
414 return -1;
417 len += used;
419 pjob->pid = pjpid;
420 pjob->sysjob = pjsysjob;
421 pjob->fd = pjfd;
422 pjob->starttime = pjstarttime;
423 pjob->status = pjstatus;
424 pjob->size = pjsize;
425 pjob->page_count = pjpage_count;
426 pjob->spooled = pjspooled;
427 pjob->smbjob = pjsmbjob;
429 return len;
433 /****************************************************************************
434 Useful function to find a print job in the database.
435 ****************************************************************************/
437 static struct printjob *print_job_find(const char *sharename, uint32 jobid)
439 static struct printjob pjob;
440 uint32_t tmp;
441 TDB_DATA ret;
442 struct tdb_print_db *pdb = get_print_db_byname(sharename);
444 DEBUG(10,("print_job_find: looking up job %u for share %s\n",
445 (unsigned int)jobid, sharename ));
447 if (!pdb) {
448 return NULL;
451 ret = tdb_fetch_compat(pdb->tdb, print_key(jobid, &tmp));
452 release_print_db(pdb);
454 if (!ret.dptr) {
455 DEBUG(10,("print_job_find: failed to find jobid %u.\n", (unsigned int)jobid ));
456 return NULL;
459 talloc_free(pjob.devmode);
461 ZERO_STRUCT( pjob );
463 if ( unpack_pjob( ret.dptr, ret.dsize, &pjob ) == -1 ) {
464 DEBUG(10,("print_job_find: failed to unpack jobid %u.\n", (unsigned int)jobid ));
465 SAFE_FREE(ret.dptr);
466 return NULL;
469 SAFE_FREE(ret.dptr);
471 DEBUG(10,("print_job_find: returning system job %d for jobid %u.\n",
472 (int)pjob.sysjob, (unsigned int)jobid ));
474 return &pjob;
477 /* Convert a unix jobid to a smb jobid */
479 struct unixjob_traverse_state {
480 int sysjob;
481 uint32 sysjob_to_jobid_value;
484 static int unixjob_traverse_fn(TDB_CONTEXT *the_tdb, TDB_DATA key,
485 TDB_DATA data, void *private_data)
487 struct printjob *pjob;
488 struct unixjob_traverse_state *state =
489 (struct unixjob_traverse_state *)private_data;
491 if (!data.dptr || data.dsize == 0)
492 return 0;
494 pjob = (struct printjob *)data.dptr;
495 if (key.dsize != sizeof(uint32))
496 return 0;
498 if (state->sysjob == pjob->sysjob) {
499 uint32 jobid = IVAL(key.dptr,0);
501 state->sysjob_to_jobid_value = jobid;
502 return 1;
505 return 0;
508 /****************************************************************************
509 This is a *horribly expensive call as we have to iterate through all the
510 current printer tdb's. Don't do this often ! JRA.
511 ****************************************************************************/
513 uint32 sysjob_to_jobid(int unix_jobid)
515 int services = lp_numservices();
516 int snum;
517 struct unixjob_traverse_state state;
519 state.sysjob = unix_jobid;
520 state.sysjob_to_jobid_value = (uint32)-1;
522 for (snum = 0; snum < services; snum++) {
523 struct tdb_print_db *pdb;
524 if (!lp_print_ok(snum))
525 continue;
526 pdb = get_print_db_byname(lp_const_servicename(snum));
527 if (!pdb) {
528 continue;
530 tdb_traverse(pdb->tdb, unixjob_traverse_fn, &state);
531 release_print_db(pdb);
532 if (state.sysjob_to_jobid_value != (uint32)-1)
533 return state.sysjob_to_jobid_value;
535 return (uint32)-1;
538 /****************************************************************************
539 Send notifications based on what has changed after a pjob_store.
540 ****************************************************************************/
542 static const struct {
543 uint32_t lpq_status;
544 uint32_t spoolss_status;
545 } lpq_to_spoolss_status_map[] = {
546 { LPQ_QUEUED, JOB_STATUS_QUEUED },
547 { LPQ_PAUSED, JOB_STATUS_PAUSED },
548 { LPQ_SPOOLING, JOB_STATUS_SPOOLING },
549 { LPQ_PRINTING, JOB_STATUS_PRINTING },
550 { LPQ_DELETING, JOB_STATUS_DELETING },
551 { LPQ_OFFLINE, JOB_STATUS_OFFLINE },
552 { LPQ_PAPEROUT, JOB_STATUS_PAPEROUT },
553 { LPQ_PRINTED, JOB_STATUS_PRINTED },
554 { LPQ_DELETED, JOB_STATUS_DELETED },
555 { LPQ_BLOCKED, JOB_STATUS_BLOCKED_DEVQ },
556 { LPQ_USER_INTERVENTION, JOB_STATUS_USER_INTERVENTION },
557 { (uint32_t)-1, 0 }
560 /* Convert a lpq status value stored in printing.tdb into the
561 appropriate win32 API constant. */
563 static uint32 map_to_spoolss_status(uint32 lpq_status)
565 int i = 0;
567 while (lpq_to_spoolss_status_map[i].lpq_status != -1) {
568 if (lpq_to_spoolss_status_map[i].lpq_status == lpq_status)
569 return lpq_to_spoolss_status_map[i].spoolss_status;
570 i++;
573 return 0;
576 /***************************************************************************
577 Append a jobid to the 'jobs changed' list.
578 ***************************************************************************/
580 static bool add_to_jobs_changed(struct tdb_print_db *pdb, uint32_t jobid)
582 TDB_DATA data;
583 uint32_t store_jobid;
585 SIVAL(&store_jobid, 0, jobid);
586 data.dptr = (uint8 *) &store_jobid;
587 data.dsize = 4;
589 DEBUG(10,("add_to_jobs_added: Added jobid %u\n", (unsigned int)jobid ));
591 return (tdb_append(pdb->tdb, string_tdb_data("INFO/jobs_changed"),
592 data) == 0);
595 /***************************************************************************
596 Remove a jobid from the 'jobs changed' list.
597 ***************************************************************************/
599 static bool remove_from_jobs_changed(const char* sharename, uint32_t jobid)
601 struct tdb_print_db *pdb = get_print_db_byname(sharename);
602 TDB_DATA data, key;
603 size_t job_count, i;
604 bool ret = False;
605 bool gotlock = False;
607 if (!pdb) {
608 return False;
611 ZERO_STRUCT(data);
613 key = string_tdb_data("INFO/jobs_changed");
615 if (tdb_chainlock_with_timeout(pdb->tdb, key, 5) != 0)
616 goto out;
618 gotlock = True;
620 data = tdb_fetch_compat(pdb->tdb, key);
622 if (data.dptr == NULL || data.dsize == 0 || (data.dsize % 4 != 0))
623 goto out;
625 job_count = data.dsize / 4;
626 for (i = 0; i < job_count; i++) {
627 uint32 ch_jobid;
629 ch_jobid = IVAL(data.dptr, i*4);
630 if (ch_jobid == jobid) {
631 if (i < job_count -1 )
632 memmove(data.dptr + (i*4), data.dptr + (i*4) + 4, (job_count - i - 1)*4 );
633 data.dsize -= 4;
634 if (tdb_store(pdb->tdb, key, data, TDB_REPLACE) != 0)
635 goto out;
636 break;
640 ret = True;
641 out:
643 if (gotlock)
644 tdb_chainunlock(pdb->tdb, key);
645 SAFE_FREE(data.dptr);
646 release_print_db(pdb);
647 if (ret)
648 DEBUG(10,("remove_from_jobs_changed: removed jobid %u\n", (unsigned int)jobid ));
649 else
650 DEBUG(10,("remove_from_jobs_changed: Failed to remove jobid %u\n", (unsigned int)jobid ));
651 return ret;
654 static void pjob_store_notify(struct tevent_context *ev,
655 struct messaging_context *msg_ctx,
656 const char* sharename, uint32 jobid,
657 struct printjob *old_data,
658 struct printjob *new_data,
659 bool *pchanged)
661 bool new_job = false;
662 bool changed = false;
664 if (old_data == NULL) {
665 new_job = true;
668 /* ACHTUNG! Due to a bug in Samba's spoolss parsing of the
669 NOTIFY_INFO_DATA buffer, we *have* to send the job submission
670 time first or else we'll end up with potential alignment
671 errors. I don't think the systemtime should be spooled as
672 a string, but this gets us around that error.
673 --jerry (i'll feel dirty for this) */
675 if (new_job) {
676 notify_job_submitted(ev, msg_ctx,
677 sharename, jobid, new_data->starttime);
678 notify_job_username(ev, msg_ctx,
679 sharename, jobid, new_data->user);
680 notify_job_name(ev, msg_ctx,
681 sharename, jobid, new_data->jobname);
682 notify_job_status(ev, msg_ctx,
683 sharename, jobid, map_to_spoolss_status(new_data->status));
684 notify_job_total_bytes(ev, msg_ctx,
685 sharename, jobid, new_data->size);
686 notify_job_total_pages(ev, msg_ctx,
687 sharename, jobid, new_data->page_count);
688 } else {
689 if (!strequal(old_data->jobname, new_data->jobname)) {
690 notify_job_name(ev, msg_ctx, sharename,
691 jobid, new_data->jobname);
692 changed = true;
695 if (old_data->status != new_data->status) {
696 notify_job_status(ev, msg_ctx,
697 sharename, jobid,
698 map_to_spoolss_status(new_data->status));
701 if (old_data->size != new_data->size) {
702 notify_job_total_bytes(ev, msg_ctx,
703 sharename, jobid, new_data->size);
706 if (old_data->page_count != new_data->page_count) {
707 notify_job_total_pages(ev, msg_ctx,
708 sharename, jobid,
709 new_data->page_count);
713 *pchanged = changed;
716 /****************************************************************************
717 Store a job structure back to the database.
718 ****************************************************************************/
720 static bool pjob_store(struct tevent_context *ev,
721 struct messaging_context *msg_ctx,
722 const char* sharename, uint32 jobid,
723 struct printjob *pjob)
725 uint32_t tmp;
726 TDB_DATA old_data, new_data;
727 bool ret = False;
728 struct tdb_print_db *pdb = get_print_db_byname(sharename);
729 uint8 *buf = NULL;
730 int len, newlen, buflen;
733 if (!pdb)
734 return False;
736 /* Get old data */
738 old_data = tdb_fetch_compat(pdb->tdb, print_key(jobid, &tmp));
740 /* Doh! Now we have to pack/unpack data since the NT_DEVICEMODE was added */
742 newlen = 0;
744 do {
745 len = 0;
746 buflen = newlen;
747 len += tdb_pack(buf+len, buflen-len, "dddddddddfffff",
748 (uint32)pjob->pid,
749 (uint32)pjob->sysjob,
750 (uint32)pjob->fd,
751 (uint32)pjob->starttime,
752 (uint32)pjob->status,
753 (uint32)pjob->size,
754 (uint32)pjob->page_count,
755 (uint32)pjob->spooled,
756 (uint32)pjob->smbjob,
757 pjob->filename,
758 pjob->jobname,
759 pjob->user,
760 pjob->clientmachine,
761 pjob->queuename);
763 len += pack_devicemode(pjob->devmode, buf+len, buflen-len);
765 if (buflen != len) {
766 buf = (uint8 *)SMB_REALLOC(buf, len);
767 if (!buf) {
768 DEBUG(0,("pjob_store: failed to enlarge buffer!\n"));
769 goto done;
771 newlen = len;
773 } while ( buflen != len );
776 /* Store new data */
778 new_data.dptr = buf;
779 new_data.dsize = len;
780 ret = (tdb_store(pdb->tdb, print_key(jobid, &tmp), new_data,
781 TDB_REPLACE) == 0);
783 /* Send notify updates for what has changed */
785 if ( ret ) {
786 bool changed = false;
787 struct printjob old_pjob;
789 if ( old_data.dsize )
791 if ( unpack_pjob( old_data.dptr, old_data.dsize, &old_pjob ) != -1 )
793 pjob_store_notify(server_event_context(),
794 msg_ctx,
795 sharename, jobid, &old_pjob,
796 pjob,
797 &changed);
798 talloc_free(old_pjob.devmode);
800 if (changed) {
801 add_to_jobs_changed(pdb, jobid);
806 else {
807 /* new job */
808 pjob_store_notify(server_event_context(), msg_ctx,
809 sharename, jobid, NULL, pjob,
810 &changed);
814 release_print_db(pdb);
815 done:
816 SAFE_FREE( old_data.dptr );
817 SAFE_FREE( buf );
819 return ret;
822 /****************************************************************************
823 Remove a job structure from the database.
824 ****************************************************************************/
826 static void pjob_delete(struct tevent_context *ev,
827 struct messaging_context *msg_ctx,
828 const char* sharename, uint32 jobid)
830 uint32_t tmp;
831 struct printjob *pjob;
832 uint32 job_status = 0;
833 struct tdb_print_db *pdb;
835 pdb = get_print_db_byname( sharename );
837 if (!pdb)
838 return;
840 pjob = print_job_find( sharename, jobid );
842 if (!pjob) {
843 DEBUG(5, ("pjob_delete: we were asked to delete nonexistent job %u\n",
844 (unsigned int)jobid));
845 release_print_db(pdb);
846 return;
849 /* We must cycle through JOB_STATUS_DELETING and
850 JOB_STATUS_DELETED for the port monitor to delete the job
851 properly. */
853 job_status = JOB_STATUS_DELETING|JOB_STATUS_DELETED;
854 notify_job_status(ev, msg_ctx, sharename, jobid, job_status);
856 /* Remove from printing.tdb */
858 tdb_delete(pdb->tdb, print_key(jobid, &tmp));
859 remove_from_jobs_added(sharename, jobid);
860 release_print_db( pdb );
861 rap_jobid_delete(sharename, jobid);
864 /****************************************************************************
865 List a unix job in the print database.
866 ****************************************************************************/
868 static void print_unix_job(struct tevent_context *ev,
869 struct messaging_context *msg_ctx,
870 const char *sharename, print_queue_struct *q,
871 uint32 jobid)
873 struct printjob pj, *old_pj;
875 if (jobid == (uint32)-1)
876 jobid = q->job + UNIX_JOB_START;
878 /* Preserve the timestamp on an existing unix print job */
880 old_pj = print_job_find(sharename, jobid);
882 ZERO_STRUCT(pj);
884 pj.pid = (pid_t)-1;
885 pj.sysjob = q->job;
886 pj.fd = -1;
887 pj.starttime = old_pj ? old_pj->starttime : q->time;
888 pj.status = q->status;
889 pj.size = q->size;
890 pj.spooled = True;
891 fstrcpy(pj.filename, old_pj ? old_pj->filename : "");
892 if (jobid < UNIX_JOB_START) {
893 pj.smbjob = True;
894 fstrcpy(pj.jobname, old_pj ? old_pj->jobname : "Remote Downlevel Document");
895 } else {
896 pj.smbjob = False;
897 fstrcpy(pj.jobname, old_pj ? old_pj->jobname : q->fs_file);
899 fstrcpy(pj.user, old_pj ? old_pj->user : q->fs_user);
900 fstrcpy(pj.queuename, old_pj ? old_pj->queuename : sharename );
902 pjob_store(ev, msg_ctx, sharename, jobid, &pj);
906 struct traverse_struct {
907 print_queue_struct *queue;
908 int qcount, snum, maxcount, total_jobs;
909 const char *sharename;
910 time_t lpq_time;
911 const char *lprm_command;
912 struct printif *print_if;
913 struct tevent_context *ev;
914 struct messaging_context *msg_ctx;
917 /****************************************************************************
918 Utility fn to delete any jobs that are no longer active.
919 ****************************************************************************/
921 static int traverse_fn_delete(TDB_CONTEXT *t, TDB_DATA key, TDB_DATA data, void *state)
923 struct traverse_struct *ts = (struct traverse_struct *)state;
924 struct printjob pjob;
925 uint32 jobid;
926 int i = 0;
928 if ( key.dsize != sizeof(jobid) )
929 return 0;
931 jobid = IVAL(key.dptr, 0);
932 if ( unpack_pjob( data.dptr, data.dsize, &pjob ) == -1 )
933 return 0;
934 talloc_free(pjob.devmode);
937 if (!pjob.smbjob) {
938 /* remove a unix job if it isn't in the system queue any more */
940 for (i=0;i<ts->qcount;i++) {
941 uint32 u_jobid = (ts->queue[i].job + UNIX_JOB_START);
942 if (jobid == u_jobid)
943 break;
945 if (i == ts->qcount) {
946 DEBUG(10,("traverse_fn_delete: pjob %u deleted due to !smbjob\n",
947 (unsigned int)jobid ));
948 pjob_delete(ts->ev, ts->msg_ctx,
949 ts->sharename, jobid);
950 return 0;
953 /* need to continue the the bottom of the function to
954 save the correct attributes */
957 /* maybe it hasn't been spooled yet */
958 if (!pjob.spooled) {
959 /* if a job is not spooled and the process doesn't
960 exist then kill it. This cleans up after smbd
961 deaths */
962 if (!process_exists_by_pid(pjob.pid)) {
963 DEBUG(10,("traverse_fn_delete: pjob %u deleted due to !process_exists (%u)\n",
964 (unsigned int)jobid, (unsigned int)pjob.pid ));
965 pjob_delete(ts->ev, ts->msg_ctx,
966 ts->sharename, jobid);
967 } else
968 ts->total_jobs++;
969 return 0;
972 /* this check only makes sense for jobs submitted from Windows clients */
974 if ( pjob.smbjob ) {
975 for (i=0;i<ts->qcount;i++) {
976 uint32 curr_jobid;
978 if ( pjob.status == LPQ_DELETED )
979 continue;
981 curr_jobid = print_parse_jobid(ts->queue[i].fs_file);
983 if (jobid == curr_jobid) {
985 /* try to clean up any jobs that need to be deleted */
987 if ( pjob.status == LPQ_DELETING ) {
988 int result;
990 result = (*(ts->print_if->job_delete))(
991 ts->sharename, ts->lprm_command, &pjob );
993 if ( result != 0 ) {
994 /* if we can't delete, then reset the job status */
995 pjob.status = LPQ_QUEUED;
996 pjob_store(ts->ev, ts->msg_ctx,
997 ts->sharename, jobid, &pjob);
999 else {
1000 /* if we deleted the job, the remove the tdb record */
1001 pjob_delete(ts->ev,
1002 ts->msg_ctx,
1003 ts->sharename, jobid);
1004 pjob.status = LPQ_DELETED;
1009 break;
1014 /* The job isn't in the system queue - we have to assume it has
1015 completed, so delete the database entry. */
1017 if (i == ts->qcount) {
1019 /* A race can occur between the time a job is spooled and
1020 when it appears in the lpq output. This happens when
1021 the job is added to printing.tdb when another smbd
1022 running print_queue_update() has completed a lpq and
1023 is currently traversing the printing tdb and deleting jobs.
1024 Don't delete the job if it was submitted after the lpq_time. */
1026 if (pjob.starttime < ts->lpq_time) {
1027 DEBUG(10,("traverse_fn_delete: pjob %u deleted due to pjob.starttime (%u) < ts->lpq_time (%u)\n",
1028 (unsigned int)jobid,
1029 (unsigned int)pjob.starttime,
1030 (unsigned int)ts->lpq_time ));
1031 pjob_delete(ts->ev, ts->msg_ctx,
1032 ts->sharename, jobid);
1033 } else
1034 ts->total_jobs++;
1035 return 0;
1038 /* Save the pjob attributes we will store.
1039 FIXME!!! This is the only place where queue->job
1040 represents the SMB jobid --jerry */
1042 ts->queue[i].job = jobid;
1043 ts->queue[i].size = pjob.size;
1044 ts->queue[i].page_count = pjob.page_count;
1045 ts->queue[i].status = pjob.status;
1046 ts->queue[i].priority = 1;
1047 ts->queue[i].time = pjob.starttime;
1048 fstrcpy(ts->queue[i].fs_user, pjob.user);
1049 fstrcpy(ts->queue[i].fs_file, pjob.jobname);
1051 ts->total_jobs++;
1053 return 0;
1056 /****************************************************************************
1057 Check if the print queue has been updated recently enough.
1058 ****************************************************************************/
1060 static void print_cache_flush(const char *sharename)
1062 fstring key;
1063 struct tdb_print_db *pdb = get_print_db_byname(sharename);
1065 if (!pdb)
1066 return;
1067 slprintf(key, sizeof(key)-1, "CACHE/%s", sharename);
1068 tdb_store_int32(pdb->tdb, key, -1);
1069 release_print_db(pdb);
1072 /****************************************************************************
1073 Check if someone already thinks they are doing the update.
1074 ****************************************************************************/
1076 static pid_t get_updating_pid(const char *sharename)
1078 fstring keystr;
1079 TDB_DATA data, key;
1080 pid_t updating_pid;
1081 struct tdb_print_db *pdb = get_print_db_byname(sharename);
1083 if (!pdb)
1084 return (pid_t)-1;
1085 slprintf(keystr, sizeof(keystr)-1, "UPDATING/%s", sharename);
1086 key = string_tdb_data(keystr);
1088 data = tdb_fetch_compat(pdb->tdb, key);
1089 release_print_db(pdb);
1090 if (!data.dptr || data.dsize != sizeof(pid_t)) {
1091 SAFE_FREE(data.dptr);
1092 return (pid_t)-1;
1095 updating_pid = IVAL(data.dptr, 0);
1096 SAFE_FREE(data.dptr);
1098 if (process_exists_by_pid(updating_pid))
1099 return updating_pid;
1101 return (pid_t)-1;
1104 /****************************************************************************
1105 Set the fact that we're doing the update, or have finished doing the update
1106 in the tdb.
1107 ****************************************************************************/
1109 static void set_updating_pid(const fstring sharename, bool updating)
1111 fstring keystr;
1112 TDB_DATA key;
1113 TDB_DATA data;
1114 pid_t updating_pid = sys_getpid();
1115 uint8 buffer[4];
1117 struct tdb_print_db *pdb = get_print_db_byname(sharename);
1119 if (!pdb)
1120 return;
1122 slprintf(keystr, sizeof(keystr)-1, "UPDATING/%s", sharename);
1123 key = string_tdb_data(keystr);
1125 DEBUG(5, ("set_updating_pid: %s updating lpq cache for print share %s\n",
1126 updating ? "" : "not ",
1127 sharename ));
1129 if ( !updating ) {
1130 tdb_delete(pdb->tdb, key);
1131 release_print_db(pdb);
1132 return;
1135 SIVAL( buffer, 0, updating_pid);
1136 data.dptr = buffer;
1137 data.dsize = 4; /* we always assume this is a 4 byte value */
1139 tdb_store(pdb->tdb, key, data, TDB_REPLACE);
1140 release_print_db(pdb);
1143 /****************************************************************************
1144 Sort print jobs by submittal time.
1145 ****************************************************************************/
1147 static int printjob_comp(print_queue_struct *j1, print_queue_struct *j2)
1149 /* Silly cases */
1151 if (!j1 && !j2)
1152 return 0;
1153 if (!j1)
1154 return -1;
1155 if (!j2)
1156 return 1;
1158 /* Sort on job start time */
1160 if (j1->time == j2->time)
1161 return 0;
1162 return (j1->time > j2->time) ? 1 : -1;
1165 /****************************************************************************
1166 Store the sorted queue representation for later portmon retrieval.
1167 Skip deleted jobs
1168 ****************************************************************************/
1170 static void store_queue_struct(struct tdb_print_db *pdb, struct traverse_struct *pts)
1172 TDB_DATA data;
1173 int max_reported_jobs = lp_max_reported_jobs(pts->snum);
1174 print_queue_struct *queue = pts->queue;
1175 size_t len;
1176 size_t i;
1177 unsigned int qcount;
1179 if (max_reported_jobs && (max_reported_jobs < pts->qcount))
1180 pts->qcount = max_reported_jobs;
1181 qcount = 0;
1183 /* Work out the size. */
1184 data.dsize = 0;
1185 data.dsize += tdb_pack(NULL, 0, "d", qcount);
1187 for (i = 0; i < pts->qcount; i++) {
1188 if ( queue[i].status == LPQ_DELETED )
1189 continue;
1191 qcount++;
1192 data.dsize += tdb_pack(NULL, 0, "ddddddff",
1193 (uint32)queue[i].job,
1194 (uint32)queue[i].size,
1195 (uint32)queue[i].page_count,
1196 (uint32)queue[i].status,
1197 (uint32)queue[i].priority,
1198 (uint32)queue[i].time,
1199 queue[i].fs_user,
1200 queue[i].fs_file);
1203 if ((data.dptr = (uint8 *)SMB_MALLOC(data.dsize)) == NULL)
1204 return;
1206 len = 0;
1207 len += tdb_pack(data.dptr + len, data.dsize - len, "d", qcount);
1208 for (i = 0; i < pts->qcount; i++) {
1209 if ( queue[i].status == LPQ_DELETED )
1210 continue;
1212 len += tdb_pack(data.dptr + len, data.dsize - len, "ddddddff",
1213 (uint32)queue[i].job,
1214 (uint32)queue[i].size,
1215 (uint32)queue[i].page_count,
1216 (uint32)queue[i].status,
1217 (uint32)queue[i].priority,
1218 (uint32)queue[i].time,
1219 queue[i].fs_user,
1220 queue[i].fs_file);
1223 tdb_store(pdb->tdb, string_tdb_data("INFO/linear_queue_array"), data,
1224 TDB_REPLACE);
1225 SAFE_FREE(data.dptr);
1226 return;
1229 static TDB_DATA get_jobs_added_data(struct tdb_print_db *pdb)
1231 TDB_DATA data;
1233 ZERO_STRUCT(data);
1235 data = tdb_fetch_compat(pdb->tdb, string_tdb_data("INFO/jobs_added"));
1236 if (data.dptr == NULL || data.dsize == 0 || (data.dsize % 4 != 0)) {
1237 SAFE_FREE(data.dptr);
1238 ZERO_STRUCT(data);
1241 return data;
1244 static void check_job_added(const char *sharename, TDB_DATA data, uint32 jobid)
1246 unsigned int i;
1247 unsigned int job_count = data.dsize / 4;
1249 for (i = 0; i < job_count; i++) {
1250 uint32 ch_jobid;
1252 ch_jobid = IVAL(data.dptr, i*4);
1253 if (ch_jobid == jobid)
1254 remove_from_jobs_added(sharename, jobid);
1258 /****************************************************************************
1259 Check if the print queue has been updated recently enough.
1260 ****************************************************************************/
1262 static bool print_cache_expired(const char *sharename, bool check_pending)
1264 fstring key;
1265 time_t last_qscan_time, time_now = time(NULL);
1266 struct tdb_print_db *pdb = get_print_db_byname(sharename);
1267 bool result = False;
1269 if (!pdb)
1270 return False;
1272 snprintf(key, sizeof(key), "CACHE/%s", sharename);
1273 last_qscan_time = (time_t)tdb_fetch_int32(pdb->tdb, key);
1276 * Invalidate the queue for 3 reasons.
1277 * (1). last queue scan time == -1.
1278 * (2). Current time - last queue scan time > allowed cache time.
1279 * (3). last queue scan time > current time + MAX_CACHE_VALID_TIME (1 hour by default).
1280 * This last test picks up machines for which the clock has been moved
1281 * forward, an lpq scan done and then the clock moved back. Otherwise
1282 * that last lpq scan would stay around for a loooong loooong time... :-). JRA.
1285 if (last_qscan_time == ((time_t)-1)
1286 || (time_now - last_qscan_time) >= lp_lpqcachetime()
1287 || last_qscan_time > (time_now + MAX_CACHE_VALID_TIME))
1289 uint32 u;
1290 time_t msg_pending_time;
1292 DEBUG(4, ("print_cache_expired: cache expired for queue %s "
1293 "(last_qscan_time = %d, time now = %d, qcachetime = %d)\n",
1294 sharename, (int)last_qscan_time, (int)time_now,
1295 (int)lp_lpqcachetime() ));
1297 /* check if another smbd has already sent a message to update the
1298 queue. Give the pending message one minute to clear and
1299 then send another message anyways. Make sure to check for
1300 clocks that have been run forward and then back again. */
1302 snprintf(key, sizeof(key), "MSG_PENDING/%s", sharename);
1304 if ( check_pending
1305 && tdb_fetch_uint32( pdb->tdb, key, &u )
1306 && (msg_pending_time=u) > 0
1307 && msg_pending_time <= time_now
1308 && (time_now - msg_pending_time) < 60 )
1310 DEBUG(4,("print_cache_expired: message already pending for %s. Accepting cache\n",
1311 sharename));
1312 goto done;
1315 result = True;
1318 done:
1319 release_print_db(pdb);
1320 return result;
1323 /****************************************************************************
1324 main work for updating the lpq cache for a printer queue
1325 ****************************************************************************/
1327 static void print_queue_update_internal( struct tevent_context *ev,
1328 struct messaging_context *msg_ctx,
1329 const char *sharename,
1330 struct printif *current_printif,
1331 char *lpq_command, char *lprm_command )
1333 int i, qcount;
1334 print_queue_struct *queue = NULL;
1335 print_status_struct status;
1336 print_status_struct old_status;
1337 struct printjob *pjob;
1338 struct traverse_struct tstruct;
1339 TDB_DATA data, key;
1340 TDB_DATA jcdata;
1341 fstring keystr, cachestr;
1342 struct tdb_print_db *pdb = get_print_db_byname(sharename);
1344 if (!pdb) {
1345 return;
1348 DEBUG(5,("print_queue_update_internal: printer = %s, type = %d, lpq command = [%s]\n",
1349 sharename, current_printif->type, lpq_command));
1352 * Update the cache time FIRST ! Stops others even
1353 * attempting to get the lock and doing this
1354 * if the lpq takes a long time.
1357 slprintf(cachestr, sizeof(cachestr)-1, "CACHE/%s", sharename);
1358 tdb_store_int32(pdb->tdb, cachestr, (int)time(NULL));
1360 /* get the current queue using the appropriate interface */
1361 ZERO_STRUCT(status);
1363 qcount = (*(current_printif->queue_get))(sharename,
1364 current_printif->type,
1365 lpq_command, &queue, &status);
1367 DEBUG(3, ("print_queue_update_internal: %d job%s in queue for %s\n",
1368 qcount, (qcount != 1) ? "s" : "", sharename));
1370 /* Sort the queue by submission time otherwise they are displayed
1371 in hash order. */
1373 TYPESAFE_QSORT(queue, qcount, printjob_comp);
1376 any job in the internal database that is marked as spooled
1377 and doesn't exist in the system queue is considered finished
1378 and removed from the database
1380 any job in the system database but not in the internal database
1381 is added as a unix job
1383 fill in any system job numbers as we go
1386 jcdata = get_jobs_added_data(pdb);
1388 for (i=0; i<qcount; i++) {
1389 uint32 jobid = print_parse_jobid(queue[i].fs_file);
1391 if (jobid == (uint32)-1) {
1392 /* assume its a unix print job */
1393 print_unix_job(ev, msg_ctx,
1394 sharename, &queue[i], jobid);
1395 continue;
1398 /* we have an active SMB print job - update its status */
1399 pjob = print_job_find(sharename, jobid);
1400 if (!pjob) {
1401 /* err, somethings wrong. Probably smbd was restarted
1402 with jobs in the queue. All we can do is treat them
1403 like unix jobs. Pity. */
1404 print_unix_job(ev, msg_ctx,
1405 sharename, &queue[i], jobid);
1406 continue;
1409 pjob->sysjob = queue[i].job;
1411 /* don't reset the status on jobs to be deleted */
1413 if ( pjob->status != LPQ_DELETING )
1414 pjob->status = queue[i].status;
1416 pjob_store(ev, msg_ctx, sharename, jobid, pjob);
1418 check_job_added(sharename, jcdata, jobid);
1421 SAFE_FREE(jcdata.dptr);
1423 /* now delete any queued entries that don't appear in the
1424 system queue */
1425 tstruct.queue = queue;
1426 tstruct.qcount = qcount;
1427 tstruct.snum = -1;
1428 tstruct.total_jobs = 0;
1429 tstruct.lpq_time = time(NULL);
1430 tstruct.sharename = sharename;
1431 tstruct.lprm_command = lprm_command;
1432 tstruct.print_if = current_printif;
1433 tstruct.ev = ev;
1434 tstruct.msg_ctx = msg_ctx;
1436 tdb_traverse(pdb->tdb, traverse_fn_delete, (void *)&tstruct);
1438 /* Store the linearised queue, max jobs only. */
1439 store_queue_struct(pdb, &tstruct);
1441 SAFE_FREE(tstruct.queue);
1443 DEBUG(10,("print_queue_update_internal: printer %s INFO/total_jobs = %d\n",
1444 sharename, tstruct.total_jobs ));
1446 tdb_store_int32(pdb->tdb, "INFO/total_jobs", tstruct.total_jobs);
1448 get_queue_status(sharename, &old_status);
1449 if (old_status.qcount != qcount)
1450 DEBUG(10,("print_queue_update_internal: queue status change %d jobs -> %d jobs for printer %s\n",
1451 old_status.qcount, qcount, sharename));
1453 /* store the new queue status structure */
1454 slprintf(keystr, sizeof(keystr)-1, "STATUS/%s", sharename);
1455 key = string_tdb_data(keystr);
1457 status.qcount = qcount;
1458 data.dptr = (uint8 *)&status;
1459 data.dsize = sizeof(status);
1460 tdb_store(pdb->tdb, key, data, TDB_REPLACE);
1463 * Update the cache time again. We want to do this call
1464 * as little as possible...
1467 slprintf(keystr, sizeof(keystr)-1, "CACHE/%s", sharename);
1468 tdb_store_int32(pdb->tdb, keystr, (int32)time(NULL));
1470 /* clear the msg pending record for this queue */
1472 snprintf(keystr, sizeof(keystr), "MSG_PENDING/%s", sharename);
1474 if ( !tdb_store_uint32( pdb->tdb, keystr, 0 ) ) {
1475 /* log a message but continue on */
1477 DEBUG(0,("print_queue_update: failed to store MSG_PENDING flag for [%s]!\n",
1478 sharename));
1481 release_print_db( pdb );
1483 return;
1486 /****************************************************************************
1487 Update the internal database from the system print queue for a queue.
1488 obtain a lock on the print queue before proceeding (needed when mutiple
1489 smbd processes maytry to update the lpq cache concurrently).
1490 ****************************************************************************/
1492 static void print_queue_update_with_lock( struct tevent_context *ev,
1493 struct messaging_context *msg_ctx,
1494 const char *sharename,
1495 struct printif *current_printif,
1496 char *lpq_command, char *lprm_command )
1498 fstring keystr;
1499 struct tdb_print_db *pdb;
1501 DEBUG(5,("print_queue_update_with_lock: printer share = %s\n", sharename));
1502 pdb = get_print_db_byname(sharename);
1503 if (!pdb)
1504 return;
1506 if ( !print_cache_expired(sharename, False) ) {
1507 DEBUG(5,("print_queue_update_with_lock: print cache for %s is still ok\n", sharename));
1508 release_print_db(pdb);
1509 return;
1513 * Check to see if someone else is doing this update.
1514 * This is essentially a mutex on the update.
1517 if (get_updating_pid(sharename) != -1) {
1518 release_print_db(pdb);
1519 return;
1522 /* Lock the queue for the database update */
1524 slprintf(keystr, sizeof(keystr) - 1, "LOCK/%s", sharename);
1525 /* Only wait 10 seconds for this. */
1526 if (tdb_lock_bystring_with_timeout(pdb->tdb, keystr, 10) != 0) {
1527 DEBUG(0,("print_queue_update_with_lock: Failed to lock printer %s database\n", sharename));
1528 release_print_db(pdb);
1529 return;
1533 * Ensure that no one else got in here.
1534 * If the updating pid is still -1 then we are
1535 * the winner.
1538 if (get_updating_pid(sharename) != -1) {
1540 * Someone else is doing the update, exit.
1542 tdb_unlock_bystring(pdb->tdb, keystr);
1543 release_print_db(pdb);
1544 return;
1548 * We're going to do the update ourselves.
1551 /* Tell others we're doing the update. */
1552 set_updating_pid(sharename, True);
1555 * Allow others to enter and notice we're doing
1556 * the update.
1559 tdb_unlock_bystring(pdb->tdb, keystr);
1561 /* do the main work now */
1563 print_queue_update_internal(ev, msg_ctx,
1564 sharename, current_printif,
1565 lpq_command, lprm_command);
1567 /* Delete our pid from the db. */
1568 set_updating_pid(sharename, False);
1569 release_print_db(pdb);
1572 /****************************************************************************
1573 this is the receive function of the background lpq updater
1574 ****************************************************************************/
1575 void print_queue_receive(struct messaging_context *msg,
1576 void *private_data,
1577 uint32_t msg_type,
1578 struct server_id server_id,
1579 DATA_BLOB *data)
1581 fstring sharename;
1582 char *lpqcommand = NULL, *lprmcommand = NULL;
1583 int printing_type;
1584 size_t len;
1586 len = tdb_unpack( (uint8 *)data->data, data->length, "fdPP",
1587 sharename,
1588 &printing_type,
1589 &lpqcommand,
1590 &lprmcommand );
1592 if ( len == -1 ) {
1593 SAFE_FREE(lpqcommand);
1594 SAFE_FREE(lprmcommand);
1595 DEBUG(0,("print_queue_receive: Got invalid print queue update message\n"));
1596 return;
1599 print_queue_update_with_lock(server_event_context(), msg, sharename,
1600 get_printer_fns_from_type((enum printing_types)printing_type),
1601 lpqcommand, lprmcommand );
1603 SAFE_FREE(lpqcommand);
1604 SAFE_FREE(lprmcommand);
1605 return;
1608 /****************************************************************************
1609 update the internal database from the system print queue for a queue
1610 ****************************************************************************/
1612 extern pid_t background_lpq_updater_pid;
1614 static void print_queue_update(struct messaging_context *msg_ctx,
1615 int snum, bool force)
1617 fstring key;
1618 fstring sharename;
1619 char *lpqcommand = NULL;
1620 char *lprmcommand = NULL;
1621 uint8 *buffer = NULL;
1622 size_t len = 0;
1623 size_t newlen;
1624 struct tdb_print_db *pdb;
1625 int type;
1626 struct printif *current_printif;
1627 TALLOC_CTX *ctx = talloc_tos();
1629 fstrcpy( sharename, lp_const_servicename(snum));
1631 /* don't strip out characters like '$' from the printername */
1633 lpqcommand = talloc_string_sub2(ctx,
1634 lp_lpqcommand(snum),
1635 "%p",
1636 lp_printername(snum),
1637 false, false, false);
1638 if (!lpqcommand) {
1639 return;
1641 lpqcommand = talloc_sub_advanced(ctx,
1642 lp_servicename(snum),
1643 current_user_info.unix_name,
1645 current_user.ut.gid,
1646 get_current_username(),
1647 current_user_info.domain,
1648 lpqcommand);
1649 if (!lpqcommand) {
1650 return;
1653 lprmcommand = talloc_string_sub2(ctx,
1654 lp_lprmcommand(snum),
1655 "%p",
1656 lp_printername(snum),
1657 false, false, false);
1658 if (!lprmcommand) {
1659 return;
1661 lprmcommand = talloc_sub_advanced(ctx,
1662 lp_servicename(snum),
1663 current_user_info.unix_name,
1665 current_user.ut.gid,
1666 get_current_username(),
1667 current_user_info.domain,
1668 lprmcommand);
1669 if (!lprmcommand) {
1670 return;
1674 * Make sure that the background queue process exists.
1675 * Otherwise just do the update ourselves
1678 if ( force || background_lpq_updater_pid == -1 ) {
1679 DEBUG(4,("print_queue_update: updating queue [%s] myself\n", sharename));
1680 current_printif = get_printer_fns( snum );
1681 print_queue_update_with_lock(server_event_context(), msg_ctx,
1682 sharename, current_printif,
1683 lpqcommand, lprmcommand);
1685 return;
1688 type = lp_printing(snum);
1690 /* get the length */
1692 len = tdb_pack( NULL, 0, "fdPP",
1693 sharename,
1694 type,
1695 lpqcommand,
1696 lprmcommand );
1698 buffer = SMB_XMALLOC_ARRAY( uint8, len );
1700 /* now pack the buffer */
1701 newlen = tdb_pack( buffer, len, "fdPP",
1702 sharename,
1703 type,
1704 lpqcommand,
1705 lprmcommand );
1707 SMB_ASSERT( newlen == len );
1709 DEBUG(10,("print_queue_update: Sending message -> printer = %s, "
1710 "type = %d, lpq command = [%s] lprm command = [%s]\n",
1711 sharename, type, lpqcommand, lprmcommand ));
1713 /* here we set a msg pending record for other smbd processes
1714 to throttle the number of duplicate print_queue_update msgs
1715 sent. */
1717 pdb = get_print_db_byname(sharename);
1718 if (!pdb) {
1719 SAFE_FREE(buffer);
1720 return;
1723 snprintf(key, sizeof(key), "MSG_PENDING/%s", sharename);
1725 if ( !tdb_store_uint32( pdb->tdb, key, time(NULL) ) ) {
1726 /* log a message but continue on */
1728 DEBUG(0,("print_queue_update: failed to store MSG_PENDING flag for [%s]!\n",
1729 sharename));
1732 release_print_db( pdb );
1734 /* finally send the message */
1736 messaging_send_buf(msg_ctx, pid_to_procid(background_lpq_updater_pid),
1737 MSG_PRINTER_UPDATE, (uint8 *)buffer, len);
1739 SAFE_FREE( buffer );
1741 return;
1744 /****************************************************************************
1745 Create/Update an entry in the print tdb that will allow us to send notify
1746 updates only to interested smbd's.
1747 ****************************************************************************/
1749 bool print_notify_register_pid(int snum)
1751 TDB_DATA data;
1752 struct tdb_print_db *pdb = NULL;
1753 TDB_CONTEXT *tdb = NULL;
1754 const char *printername;
1755 uint32 mypid = (uint32)sys_getpid();
1756 bool ret = False;
1757 size_t i;
1759 /* if (snum == -1), then the change notify request was
1760 on a print server handle and we need to register on
1761 all print queus */
1763 if (snum == -1)
1765 int num_services = lp_numservices();
1766 int idx;
1768 for ( idx=0; idx<num_services; idx++ ) {
1769 if (lp_snum_ok(idx) && lp_print_ok(idx) )
1770 print_notify_register_pid(idx);
1773 return True;
1775 else /* register for a specific printer */
1777 printername = lp_const_servicename(snum);
1778 pdb = get_print_db_byname(printername);
1779 if (!pdb)
1780 return False;
1781 tdb = pdb->tdb;
1784 if (tdb_lock_bystring_with_timeout(tdb, NOTIFY_PID_LIST_KEY, 10) != 0) {
1785 DEBUG(0,("print_notify_register_pid: Failed to lock printer %s\n",
1786 printername));
1787 if (pdb)
1788 release_print_db(pdb);
1789 return False;
1792 data = get_printer_notify_pid_list( tdb, printername, True );
1794 /* Add ourselves and increase the refcount. */
1796 for (i = 0; i < data.dsize; i += 8) {
1797 if (IVAL(data.dptr,i) == mypid) {
1798 uint32 new_refcount = IVAL(data.dptr, i+4) + 1;
1799 SIVAL(data.dptr, i+4, new_refcount);
1800 break;
1804 if (i == data.dsize) {
1805 /* We weren't in the list. Realloc. */
1806 data.dptr = (uint8 *)SMB_REALLOC(data.dptr, data.dsize + 8);
1807 if (!data.dptr) {
1808 DEBUG(0,("print_notify_register_pid: Relloc fail for printer %s\n",
1809 printername));
1810 goto done;
1812 data.dsize += 8;
1813 SIVAL(data.dptr,data.dsize - 8,mypid);
1814 SIVAL(data.dptr,data.dsize - 4,1); /* Refcount. */
1817 /* Store back the record. */
1818 if (tdb_store_bystring(tdb, NOTIFY_PID_LIST_KEY, data, TDB_REPLACE) != 0) {
1819 DEBUG(0,("print_notify_register_pid: Failed to update pid \
1820 list for printer %s\n", printername));
1821 goto done;
1824 ret = True;
1826 done:
1828 tdb_unlock_bystring(tdb, NOTIFY_PID_LIST_KEY);
1829 if (pdb)
1830 release_print_db(pdb);
1831 SAFE_FREE(data.dptr);
1832 return ret;
1835 /****************************************************************************
1836 Update an entry in the print tdb that will allow us to send notify
1837 updates only to interested smbd's.
1838 ****************************************************************************/
1840 bool print_notify_deregister_pid(int snum)
1842 TDB_DATA data;
1843 struct tdb_print_db *pdb = NULL;
1844 TDB_CONTEXT *tdb = NULL;
1845 const char *printername;
1846 uint32 mypid = (uint32)sys_getpid();
1847 size_t i;
1848 bool ret = False;
1850 /* if ( snum == -1 ), we are deregister a print server handle
1851 which means to deregister on all print queues */
1853 if (snum == -1)
1855 int num_services = lp_numservices();
1856 int idx;
1858 for ( idx=0; idx<num_services; idx++ ) {
1859 if ( lp_snum_ok(idx) && lp_print_ok(idx) )
1860 print_notify_deregister_pid(idx);
1863 return True;
1865 else /* deregister a specific printer */
1867 printername = lp_const_servicename(snum);
1868 pdb = get_print_db_byname(printername);
1869 if (!pdb)
1870 return False;
1871 tdb = pdb->tdb;
1874 if (tdb_lock_bystring_with_timeout(tdb, NOTIFY_PID_LIST_KEY, 10) != 0) {
1875 DEBUG(0,("print_notify_register_pid: Failed to lock \
1876 printer %s database\n", printername));
1877 if (pdb)
1878 release_print_db(pdb);
1879 return False;
1882 data = get_printer_notify_pid_list( tdb, printername, True );
1884 /* Reduce refcount. Remove ourselves if zero. */
1886 for (i = 0; i < data.dsize; ) {
1887 if (IVAL(data.dptr,i) == mypid) {
1888 uint32 refcount = IVAL(data.dptr, i+4);
1890 refcount--;
1892 if (refcount == 0) {
1893 if (data.dsize - i > 8)
1894 memmove( &data.dptr[i], &data.dptr[i+8], data.dsize - i - 8);
1895 data.dsize -= 8;
1896 continue;
1898 SIVAL(data.dptr, i+4, refcount);
1901 i += 8;
1904 if (data.dsize == 0)
1905 SAFE_FREE(data.dptr);
1907 /* Store back the record. */
1908 if (tdb_store_bystring(tdb, NOTIFY_PID_LIST_KEY, data, TDB_REPLACE) != 0) {
1909 DEBUG(0,("print_notify_register_pid: Failed to update pid \
1910 list for printer %s\n", printername));
1911 goto done;
1914 ret = True;
1916 done:
1918 tdb_unlock_bystring(tdb, NOTIFY_PID_LIST_KEY);
1919 if (pdb)
1920 release_print_db(pdb);
1921 SAFE_FREE(data.dptr);
1922 return ret;
1925 /****************************************************************************
1926 Check if a jobid is valid. It is valid if it exists in the database.
1927 ****************************************************************************/
1929 bool print_job_exists(const char* sharename, uint32 jobid)
1931 struct tdb_print_db *pdb = get_print_db_byname(sharename);
1932 bool ret;
1933 uint32_t tmp;
1935 if (!pdb)
1936 return False;
1937 ret = tdb_exists(pdb->tdb, print_key(jobid, &tmp));
1938 release_print_db(pdb);
1939 return ret;
1942 /****************************************************************************
1943 Give the filename used for a jobid.
1944 Only valid for the process doing the spooling and when the job
1945 has not been spooled.
1946 ****************************************************************************/
1948 char *print_job_fname(const char* sharename, uint32 jobid)
1950 struct printjob *pjob = print_job_find(sharename, jobid);
1951 if (!pjob || pjob->spooled || pjob->pid != sys_getpid())
1952 return NULL;
1953 return pjob->filename;
1957 /****************************************************************************
1958 Give the filename used for a jobid.
1959 Only valid for the process doing the spooling and when the job
1960 has not been spooled.
1961 ****************************************************************************/
1963 struct spoolss_DeviceMode *print_job_devmode(const char* sharename, uint32 jobid)
1965 struct printjob *pjob = print_job_find(sharename, jobid);
1967 if ( !pjob )
1968 return NULL;
1970 return pjob->devmode;
1973 /****************************************************************************
1974 Set the name of a job. Only possible for owner.
1975 ****************************************************************************/
1977 bool print_job_set_name(struct tevent_context *ev,
1978 struct messaging_context *msg_ctx,
1979 const char *sharename, uint32 jobid, const char *name)
1981 struct printjob *pjob;
1983 pjob = print_job_find(sharename, jobid);
1984 if (!pjob || pjob->pid != sys_getpid())
1985 return False;
1987 fstrcpy(pjob->jobname, name);
1988 return pjob_store(ev, msg_ctx, sharename, jobid, pjob);
1991 /****************************************************************************
1992 Get the name of a job. Only possible for owner.
1993 ****************************************************************************/
1995 bool print_job_get_name(TALLOC_CTX *mem_ctx, const char *sharename, uint32_t jobid, char **name)
1997 struct printjob *pjob;
1999 pjob = print_job_find(sharename, jobid);
2000 if (!pjob || pjob->pid != sys_getpid()) {
2001 return false;
2004 *name = talloc_strdup(mem_ctx, pjob->jobname);
2005 if (!*name) {
2006 return false;
2009 return true;
2013 /***************************************************************************
2014 Remove a jobid from the 'jobs added' list.
2015 ***************************************************************************/
2017 static bool remove_from_jobs_added(const char* sharename, uint32 jobid)
2019 struct tdb_print_db *pdb = get_print_db_byname(sharename);
2020 TDB_DATA data, key;
2021 size_t job_count, i;
2022 bool ret = False;
2023 bool gotlock = False;
2025 if (!pdb) {
2026 return False;
2029 ZERO_STRUCT(data);
2031 key = string_tdb_data("INFO/jobs_added");
2033 if (tdb_chainlock_with_timeout(pdb->tdb, key, 5) != 0)
2034 goto out;
2036 gotlock = True;
2038 data = tdb_fetch_compat(pdb->tdb, key);
2040 if (data.dptr == NULL || data.dsize == 0 || (data.dsize % 4 != 0))
2041 goto out;
2043 job_count = data.dsize / 4;
2044 for (i = 0; i < job_count; i++) {
2045 uint32 ch_jobid;
2047 ch_jobid = IVAL(data.dptr, i*4);
2048 if (ch_jobid == jobid) {
2049 if (i < job_count -1 )
2050 memmove(data.dptr + (i*4), data.dptr + (i*4) + 4, (job_count - i - 1)*4 );
2051 data.dsize -= 4;
2052 if (tdb_store(pdb->tdb, key, data, TDB_REPLACE) != 0)
2053 goto out;
2054 break;
2058 ret = True;
2059 out:
2061 if (gotlock)
2062 tdb_chainunlock(pdb->tdb, key);
2063 SAFE_FREE(data.dptr);
2064 release_print_db(pdb);
2065 if (ret)
2066 DEBUG(10,("remove_from_jobs_added: removed jobid %u\n", (unsigned int)jobid ));
2067 else
2068 DEBUG(10,("remove_from_jobs_added: Failed to remove jobid %u\n", (unsigned int)jobid ));
2069 return ret;
2072 /****************************************************************************
2073 Delete a print job - don't update queue.
2074 ****************************************************************************/
2076 static bool print_job_delete1(struct tevent_context *ev,
2077 struct messaging_context *msg_ctx,
2078 int snum, uint32 jobid)
2080 const char* sharename = lp_const_servicename(snum);
2081 struct printjob *pjob = print_job_find(sharename, jobid);
2082 int result = 0;
2083 struct printif *current_printif = get_printer_fns( snum );
2085 if (!pjob)
2086 return False;
2089 * If already deleting just return.
2092 if (pjob->status == LPQ_DELETING)
2093 return True;
2095 /* Hrm - we need to be able to cope with deleting a job before it
2096 has reached the spooler. Just mark it as LPQ_DELETING and
2097 let the print_queue_update() code rmeove the record */
2100 if (pjob->sysjob == -1) {
2101 DEBUG(5, ("attempt to delete job %u not seen by lpr\n", (unsigned int)jobid));
2104 /* Set the tdb entry to be deleting. */
2106 pjob->status = LPQ_DELETING;
2107 pjob_store(ev, msg_ctx, sharename, jobid, pjob);
2109 if (pjob->spooled && pjob->sysjob != -1)
2111 result = (*(current_printif->job_delete))(
2112 lp_printername(snum),
2113 lp_lprmcommand(snum),
2114 pjob);
2116 /* Delete the tdb entry if the delete succeeded or the job hasn't
2117 been spooled. */
2119 if (result == 0) {
2120 struct tdb_print_db *pdb = get_print_db_byname(sharename);
2121 int njobs = 1;
2123 if (!pdb)
2124 return False;
2125 pjob_delete(ev, msg_ctx, sharename, jobid);
2126 /* Ensure we keep a rough count of the number of total jobs... */
2127 tdb_change_int32_atomic(pdb->tdb, "INFO/total_jobs", &njobs, -1);
2128 release_print_db(pdb);
2132 remove_from_jobs_added( sharename, jobid );
2134 return (result == 0);
2137 /****************************************************************************
2138 Return true if the current user owns the print job.
2139 ****************************************************************************/
2141 static bool is_owner(const struct auth_session_info *server_info,
2142 const char *servicename,
2143 uint32 jobid)
2145 struct printjob *pjob = print_job_find(servicename, jobid);
2147 if (!pjob || !server_info)
2148 return False;
2150 return strequal(pjob->user, server_info->unix_info->sanitized_username);
2153 /****************************************************************************
2154 Delete a print job.
2155 ****************************************************************************/
2157 WERROR print_job_delete(const struct auth_session_info *server_info,
2158 struct messaging_context *msg_ctx,
2159 int snum, uint32_t jobid)
2161 const char* sharename = lp_const_servicename(snum);
2162 struct printjob *pjob;
2163 bool owner;
2164 char *fname;
2166 owner = is_owner(server_info, lp_const_servicename(snum), jobid);
2168 /* Check access against security descriptor or whether the user
2169 owns their job. */
2171 if (!owner &&
2172 !print_access_check(server_info, msg_ctx, snum,
2173 JOB_ACCESS_ADMINISTER)) {
2174 DEBUG(3, ("delete denied by security descriptor\n"));
2176 /* BEGIN_ADMIN_LOG */
2177 sys_adminlog( LOG_ERR,
2178 "Permission denied-- user not allowed to delete, \
2179 pause, or resume print job. User name: %s. Printer name: %s.",
2180 uidtoname(server_info->unix_token->uid),
2181 lp_printername(snum) );
2182 /* END_ADMIN_LOG */
2184 return WERR_ACCESS_DENIED;
2188 * get the spooled filename of the print job
2189 * if this works, then the file has not been spooled
2190 * to the underlying print system. Just delete the
2191 * spool file & return.
2194 fname = print_job_fname(sharename, jobid);
2195 if (fname != NULL) {
2196 /* remove the spool file */
2197 DEBUG(10, ("print_job_delete: "
2198 "Removing spool file [%s]\n", fname));
2199 if (unlink(fname) == -1) {
2200 return map_werror_from_unix(errno);
2204 if (!print_job_delete1(server_event_context(), msg_ctx, snum, jobid)) {
2205 return WERR_ACCESS_DENIED;
2208 /* force update the database and say the delete failed if the
2209 job still exists */
2211 print_queue_update(msg_ctx, snum, True);
2213 pjob = print_job_find(sharename, jobid);
2214 if (pjob && (pjob->status != LPQ_DELETING)) {
2215 return WERR_ACCESS_DENIED;
2218 return WERR_PRINTER_HAS_JOBS_QUEUED;
2221 /****************************************************************************
2222 Pause a job.
2223 ****************************************************************************/
2225 bool print_job_pause(const struct auth_session_info *server_info,
2226 struct messaging_context *msg_ctx,
2227 int snum, uint32 jobid, WERROR *errcode)
2229 const char* sharename = lp_const_servicename(snum);
2230 struct printjob *pjob;
2231 int ret = -1;
2232 struct printif *current_printif = get_printer_fns( snum );
2234 pjob = print_job_find(sharename, jobid);
2236 if (!pjob || !server_info) {
2237 DEBUG(10, ("print_job_pause: no pjob or user for jobid %u\n",
2238 (unsigned int)jobid ));
2239 return False;
2242 if (!pjob->spooled || pjob->sysjob == -1) {
2243 DEBUG(10, ("print_job_pause: not spooled or bad sysjob = %d for jobid %u\n",
2244 (int)pjob->sysjob, (unsigned int)jobid ));
2245 return False;
2248 if (!is_owner(server_info, lp_const_servicename(snum), jobid) &&
2249 !print_access_check(server_info, msg_ctx, snum,
2250 JOB_ACCESS_ADMINISTER)) {
2251 DEBUG(3, ("pause denied by security descriptor\n"));
2253 /* BEGIN_ADMIN_LOG */
2254 sys_adminlog( LOG_ERR,
2255 "Permission denied-- user not allowed to delete, \
2256 pause, or resume print job. User name: %s. Printer name: %s.",
2257 uidtoname(server_info->unix_token->uid),
2258 lp_printername(snum) );
2259 /* END_ADMIN_LOG */
2261 *errcode = WERR_ACCESS_DENIED;
2262 return False;
2265 /* need to pause the spooled entry */
2266 ret = (*(current_printif->job_pause))(snum, pjob);
2268 if (ret != 0) {
2269 *errcode = WERR_INVALID_PARAM;
2270 return False;
2273 /* force update the database */
2274 print_cache_flush(lp_const_servicename(snum));
2276 /* Send a printer notify message */
2278 notify_job_status(server_event_context(), msg_ctx, sharename, jobid,
2279 JOB_STATUS_PAUSED);
2281 /* how do we tell if this succeeded? */
2283 return True;
2286 /****************************************************************************
2287 Resume a job.
2288 ****************************************************************************/
2290 bool print_job_resume(const struct auth_session_info *server_info,
2291 struct messaging_context *msg_ctx,
2292 int snum, uint32 jobid, WERROR *errcode)
2294 const char *sharename = lp_const_servicename(snum);
2295 struct printjob *pjob;
2296 int ret;
2297 struct printif *current_printif = get_printer_fns( snum );
2299 pjob = print_job_find(sharename, jobid);
2301 if (!pjob || !server_info) {
2302 DEBUG(10, ("print_job_resume: no pjob or user for jobid %u\n",
2303 (unsigned int)jobid ));
2304 return False;
2307 if (!pjob->spooled || pjob->sysjob == -1) {
2308 DEBUG(10, ("print_job_resume: not spooled or bad sysjob = %d for jobid %u\n",
2309 (int)pjob->sysjob, (unsigned int)jobid ));
2310 return False;
2313 if (!is_owner(server_info, lp_const_servicename(snum), jobid) &&
2314 !print_access_check(server_info, msg_ctx, snum,
2315 JOB_ACCESS_ADMINISTER)) {
2316 DEBUG(3, ("resume denied by security descriptor\n"));
2317 *errcode = WERR_ACCESS_DENIED;
2319 /* BEGIN_ADMIN_LOG */
2320 sys_adminlog( LOG_ERR,
2321 "Permission denied-- user not allowed to delete, \
2322 pause, or resume print job. User name: %s. Printer name: %s.",
2323 uidtoname(server_info->unix_token->uid),
2324 lp_printername(snum) );
2325 /* END_ADMIN_LOG */
2326 return False;
2329 ret = (*(current_printif->job_resume))(snum, pjob);
2331 if (ret != 0) {
2332 *errcode = WERR_INVALID_PARAM;
2333 return False;
2336 /* force update the database */
2337 print_cache_flush(lp_const_servicename(snum));
2339 /* Send a printer notify message */
2341 notify_job_status(server_event_context(), msg_ctx, sharename, jobid,
2342 JOB_STATUS_QUEUED);
2344 return True;
2347 /****************************************************************************
2348 Write to a print file.
2349 ****************************************************************************/
2351 ssize_t print_job_write(struct tevent_context *ev,
2352 struct messaging_context *msg_ctx,
2353 int snum, uint32 jobid, const char *buf, size_t size)
2355 const char* sharename = lp_const_servicename(snum);
2356 ssize_t return_code;
2357 struct printjob *pjob;
2359 pjob = print_job_find(sharename, jobid);
2361 if (!pjob)
2362 return -1;
2363 /* don't allow another process to get this info - it is meaningless */
2364 if (pjob->pid != sys_getpid())
2365 return -1;
2367 /* if SMBD is spooling this can't be allowed */
2368 if (pjob->status == PJOB_SMBD_SPOOLING) {
2369 return -1;
2372 return_code = write_data(pjob->fd, buf, size);
2374 if (return_code>0) {
2375 pjob->size += size;
2376 pjob_store(ev, msg_ctx, sharename, jobid, pjob);
2378 return return_code;
2381 /****************************************************************************
2382 Get the queue status - do not update if db is out of date.
2383 ****************************************************************************/
2385 static int get_queue_status(const char* sharename, print_status_struct *status)
2387 fstring keystr;
2388 TDB_DATA data;
2389 struct tdb_print_db *pdb = get_print_db_byname(sharename);
2390 int len;
2392 if (status) {
2393 ZERO_STRUCTP(status);
2396 if (!pdb)
2397 return 0;
2399 if (status) {
2400 fstr_sprintf(keystr, "STATUS/%s", sharename);
2401 data = tdb_fetch_compat(pdb->tdb, string_tdb_data(keystr));
2402 if (data.dptr) {
2403 if (data.dsize == sizeof(print_status_struct))
2404 /* this memcpy is ok since the status struct was
2405 not packed before storing it in the tdb */
2406 memcpy(status, data.dptr, sizeof(print_status_struct));
2407 SAFE_FREE(data.dptr);
2410 len = tdb_fetch_int32(pdb->tdb, "INFO/total_jobs");
2411 release_print_db(pdb);
2412 return (len == -1 ? 0 : len);
2415 /****************************************************************************
2416 Determine the number of jobs in a queue.
2417 ****************************************************************************/
2419 int print_queue_length(struct messaging_context *msg_ctx, int snum,
2420 print_status_struct *pstatus)
2422 const char* sharename = lp_const_servicename( snum );
2423 print_status_struct status;
2424 int len;
2426 ZERO_STRUCT( status );
2428 /* make sure the database is up to date */
2429 if (print_cache_expired(lp_const_servicename(snum), True))
2430 print_queue_update(msg_ctx, snum, False);
2432 /* also fetch the queue status */
2433 memset(&status, 0, sizeof(status));
2434 len = get_queue_status(sharename, &status);
2436 if (pstatus)
2437 *pstatus = status;
2439 return len;
2442 /***************************************************************************
2443 Allocate a jobid. Hold the lock for as short a time as possible.
2444 ***************************************************************************/
2446 static WERROR allocate_print_jobid(struct tdb_print_db *pdb, int snum,
2447 const char *sharename, uint32 *pjobid)
2449 int i;
2450 uint32 jobid;
2451 enum TDB_ERROR terr;
2452 int ret;
2454 *pjobid = (uint32)-1;
2456 for (i = 0; i < 3; i++) {
2457 /* Lock the database - only wait 20 seconds. */
2458 ret = tdb_lock_bystring_with_timeout(pdb->tdb,
2459 "INFO/nextjob", 20);
2460 if (ret != 0) {
2461 DEBUG(0, ("allocate_print_jobid: "
2462 "Failed to lock printing database %s\n",
2463 sharename));
2464 terr = tdb_error(pdb->tdb);
2465 return ntstatus_to_werror(map_nt_error_from_tdb(terr));
2468 if (!tdb_fetch_uint32(pdb->tdb, "INFO/nextjob", &jobid)) {
2469 terr = tdb_error(pdb->tdb);
2470 if (terr != TDB_ERR_NOEXIST) {
2471 DEBUG(0, ("allocate_print_jobid: "
2472 "Failed to fetch INFO/nextjob "
2473 "for print queue %s\n", sharename));
2474 tdb_unlock_bystring(pdb->tdb, "INFO/nextjob");
2475 return ntstatus_to_werror(map_nt_error_from_tdb(terr));
2477 DEBUG(10, ("allocate_print_jobid: "
2478 "No existing jobid in %s\n", sharename));
2479 jobid = 0;
2482 DEBUG(10, ("allocate_print_jobid: "
2483 "Read jobid %u from %s\n", jobid, sharename));
2485 jobid = NEXT_JOBID(jobid);
2487 ret = tdb_store_int32(pdb->tdb, "INFO/nextjob", jobid);
2488 if (ret != 0) {
2489 terr = tdb_error(pdb->tdb);
2490 DEBUG(3, ("allocate_print_jobid: "
2491 "Failed to store INFO/nextjob.\n"));
2492 tdb_unlock_bystring(pdb->tdb, "INFO/nextjob");
2493 return ntstatus_to_werror(map_nt_error_from_tdb(terr));
2496 /* We've finished with the INFO/nextjob lock. */
2497 tdb_unlock_bystring(pdb->tdb, "INFO/nextjob");
2499 if (!print_job_exists(sharename, jobid)) {
2500 break;
2502 DEBUG(10, ("allocate_print_jobid: "
2503 "Found jobid %u in %s\n", jobid, sharename));
2506 if (i > 2) {
2507 DEBUG(0, ("allocate_print_jobid: "
2508 "Failed to allocate a print job for queue %s\n",
2509 sharename));
2510 /* Probably full... */
2511 return WERR_NO_SPOOL_SPACE;
2514 /* Store a dummy placeholder. */
2516 uint32_t tmp;
2517 TDB_DATA dum;
2518 dum.dptr = NULL;
2519 dum.dsize = 0;
2520 if (tdb_store(pdb->tdb, print_key(jobid, &tmp), dum,
2521 TDB_INSERT) != 0) {
2522 DEBUG(3, ("allocate_print_jobid: "
2523 "jobid (%d) failed to store placeholder.\n",
2524 jobid ));
2525 terr = tdb_error(pdb->tdb);
2526 return ntstatus_to_werror(map_nt_error_from_tdb(terr));
2530 *pjobid = jobid;
2531 return WERR_OK;
2534 /***************************************************************************
2535 Append a jobid to the 'jobs added' list.
2536 ***************************************************************************/
2538 static bool add_to_jobs_added(struct tdb_print_db *pdb, uint32 jobid)
2540 TDB_DATA data;
2541 uint32 store_jobid;
2543 SIVAL(&store_jobid, 0, jobid);
2544 data.dptr = (uint8 *)&store_jobid;
2545 data.dsize = 4;
2547 DEBUG(10,("add_to_jobs_added: Added jobid %u\n", (unsigned int)jobid ));
2549 return (tdb_append(pdb->tdb, string_tdb_data("INFO/jobs_added"),
2550 data) == 0);
2554 /***************************************************************************
2555 Do all checks needed to determine if we can start a job.
2556 ***************************************************************************/
2558 static WERROR print_job_checks(const struct auth_session_info *server_info,
2559 struct messaging_context *msg_ctx,
2560 int snum, int *njobs)
2562 const char *sharename = lp_const_servicename(snum);
2563 uint64_t dspace, dsize;
2564 uint64_t minspace;
2565 int ret;
2567 if (!print_access_check(server_info, msg_ctx, snum,
2568 PRINTER_ACCESS_USE)) {
2569 DEBUG(3, ("print_job_checks: "
2570 "job start denied by security descriptor\n"));
2571 return WERR_ACCESS_DENIED;
2574 if (!print_time_access_check(server_info, msg_ctx, sharename)) {
2575 DEBUG(3, ("print_job_checks: "
2576 "job start denied by time check\n"));
2577 return WERR_ACCESS_DENIED;
2580 /* see if we have sufficient disk space */
2581 if (lp_minprintspace(snum)) {
2582 minspace = lp_minprintspace(snum);
2583 ret = sys_fsusage(lp_pathname(snum), &dspace, &dsize);
2584 if (ret == 0 && dspace < 2*minspace) {
2585 DEBUG(3, ("print_job_checks: "
2586 "disk space check failed.\n"));
2587 return WERR_NO_SPOOL_SPACE;
2591 /* for autoloaded printers, check that the printcap entry still exists */
2592 if (lp_autoloaded(snum) && !pcap_printername_ok(sharename)) {
2593 DEBUG(3, ("print_job_checks: printer name %s check failed.\n",
2594 sharename));
2595 return WERR_ACCESS_DENIED;
2598 /* Insure the maximum queue size is not violated */
2599 *njobs = print_queue_length(msg_ctx, snum, NULL);
2600 if (*njobs > lp_maxprintjobs(snum)) {
2601 DEBUG(3, ("print_job_checks: Queue %s number of jobs (%d) "
2602 "larger than max printjobs per queue (%d).\n",
2603 sharename, *njobs, lp_maxprintjobs(snum)));
2604 return WERR_NO_SPOOL_SPACE;
2607 return WERR_OK;
2610 /***************************************************************************
2611 Create a job file.
2612 ***************************************************************************/
2614 static WERROR print_job_spool_file(int snum, uint32_t jobid,
2615 const char *output_file,
2616 struct printjob *pjob)
2618 WERROR werr;
2619 SMB_STRUCT_STAT st;
2620 const char *path;
2621 int len;
2623 /* if this file is within the printer path, it means that smbd
2624 * is spooling it and will pass us control when it is finished.
2625 * Verify that the file name is ok, within path, and it is
2626 * already already there */
2627 if (output_file) {
2628 path = lp_pathname(snum);
2629 len = strlen(path);
2630 if (strncmp(output_file, path, len) == 0 &&
2631 (output_file[len - 1] == '/' || output_file[len] == '/')) {
2633 /* verify path is not too long */
2634 if (strlen(output_file) >= sizeof(pjob->filename)) {
2635 return WERR_INVALID_NAME;
2638 /* verify that the file exists */
2639 if (sys_stat(output_file, &st, false) != 0) {
2640 return WERR_INVALID_NAME;
2643 fstrcpy(pjob->filename, output_file);
2645 DEBUG(3, ("print_job_spool_file:"
2646 "External spooling activated"));
2648 /* we do not open the file until spooling is done */
2649 pjob->fd = -1;
2650 pjob->status = PJOB_SMBD_SPOOLING;
2652 return WERR_OK;
2656 slprintf(pjob->filename, sizeof(pjob->filename)-1,
2657 "%s/%s%.8u.XXXXXX", lp_pathname(snum),
2658 PRINT_SPOOL_PREFIX, (unsigned int)jobid);
2659 pjob->fd = mkstemp(pjob->filename);
2661 if (pjob->fd == -1) {
2662 werr = map_werror_from_unix(errno);
2663 if (W_ERROR_EQUAL(werr, WERR_ACCESS_DENIED)) {
2664 /* Common setup error, force a report. */
2665 DEBUG(0, ("print_job_spool_file: "
2666 "insufficient permissions to open spool "
2667 "file %s.\n", pjob->filename));
2668 } else {
2669 /* Normal case, report at level 3 and above. */
2670 DEBUG(3, ("print_job_spool_file: "
2671 "can't open spool file %s\n",
2672 pjob->filename));
2674 return werr;
2677 return WERR_OK;
2680 /***************************************************************************
2681 Start spooling a job - return the jobid.
2682 ***************************************************************************/
2684 WERROR print_job_start(const struct auth_session_info *server_info,
2685 struct messaging_context *msg_ctx,
2686 const char *clientmachine,
2687 int snum, const char *docname, const char *filename,
2688 struct spoolss_DeviceMode *devmode, uint32_t *_jobid)
2690 uint32_t jobid;
2691 char *path;
2692 struct printjob pjob;
2693 const char *sharename = lp_const_servicename(snum);
2694 struct tdb_print_db *pdb = get_print_db_byname(sharename);
2695 int njobs;
2696 WERROR werr;
2698 if (!pdb) {
2699 return WERR_INTERNAL_DB_CORRUPTION;
2702 path = lp_pathname(snum);
2704 werr = print_job_checks(server_info, msg_ctx, snum, &njobs);
2705 if (!W_ERROR_IS_OK(werr)) {
2706 release_print_db(pdb);
2707 return werr;
2710 DEBUG(10, ("print_job_start: "
2711 "Queue %s number of jobs (%d), max printjobs = %d\n",
2712 sharename, njobs, lp_maxprintjobs(snum)));
2714 werr = allocate_print_jobid(pdb, snum, sharename, &jobid);
2715 if (!W_ERROR_IS_OK(werr)) {
2716 goto fail;
2719 /* create the database entry */
2721 ZERO_STRUCT(pjob);
2723 pjob.pid = sys_getpid();
2724 pjob.sysjob = -1;
2725 pjob.fd = -1;
2726 pjob.starttime = time(NULL);
2727 pjob.status = LPQ_SPOOLING;
2728 pjob.size = 0;
2729 pjob.spooled = False;
2730 pjob.smbjob = True;
2731 pjob.devmode = devmode;
2733 fstrcpy(pjob.jobname, docname);
2735 fstrcpy(pjob.clientmachine, clientmachine);
2737 fstrcpy(pjob.user, lp_printjob_username(snum));
2738 standard_sub_advanced(sharename, server_info->unix_info->sanitized_username,
2739 path, server_info->unix_token->gid,
2740 server_info->unix_info->sanitized_username,
2741 server_info->info->domain_name,
2742 pjob.user, sizeof(pjob.user)-1);
2743 /* ensure NULL termination */
2744 pjob.user[sizeof(pjob.user)-1] = '\0';
2746 fstrcpy(pjob.queuename, lp_const_servicename(snum));
2748 /* we have a job entry - now create the spool file */
2749 werr = print_job_spool_file(snum, jobid, filename, &pjob);
2750 if (!W_ERROR_IS_OK(werr)) {
2751 goto fail;
2754 pjob_store(server_event_context(), msg_ctx, sharename, jobid, &pjob);
2756 /* Update the 'jobs added' entry used by print_queue_status. */
2757 add_to_jobs_added(pdb, jobid);
2759 /* Ensure we keep a rough count of the number of total jobs... */
2760 tdb_change_int32_atomic(pdb->tdb, "INFO/total_jobs", &njobs, 1);
2762 release_print_db(pdb);
2764 *_jobid = jobid;
2765 return WERR_OK;
2767 fail:
2768 if (jobid != -1) {
2769 pjob_delete(server_event_context(), msg_ctx, sharename, jobid);
2772 release_print_db(pdb);
2774 DEBUG(3, ("print_job_start: returning fail. "
2775 "Error = %s\n", win_errstr(werr)));
2776 return werr;
2779 /****************************************************************************
2780 Update the number of pages spooled to jobid
2781 ****************************************************************************/
2783 void print_job_endpage(struct messaging_context *msg_ctx,
2784 int snum, uint32 jobid)
2786 const char* sharename = lp_const_servicename(snum);
2787 struct printjob *pjob;
2789 pjob = print_job_find(sharename, jobid);
2790 if (!pjob)
2791 return;
2792 /* don't allow another process to get this info - it is meaningless */
2793 if (pjob->pid != sys_getpid())
2794 return;
2796 pjob->page_count++;
2797 pjob_store(server_event_context(), msg_ctx, sharename, jobid, pjob);
2800 /****************************************************************************
2801 Print a file - called on closing the file. This spools the job.
2802 If normal close is false then we're tearing down the jobs - treat as an
2803 error.
2804 ****************************************************************************/
2806 NTSTATUS print_job_end(struct messaging_context *msg_ctx, int snum,
2807 uint32 jobid, enum file_close_type close_type)
2809 const char* sharename = lp_const_servicename(snum);
2810 struct printjob *pjob;
2811 int ret;
2812 SMB_STRUCT_STAT sbuf;
2813 struct printif *current_printif = get_printer_fns( snum );
2814 NTSTATUS status = NT_STATUS_UNSUCCESSFUL;
2816 pjob = print_job_find(sharename, jobid);
2818 if (!pjob) {
2819 return NT_STATUS_PRINT_CANCELLED;
2822 if (pjob->spooled || pjob->pid != sys_getpid()) {
2823 return NT_STATUS_ACCESS_DENIED;
2826 if (close_type == NORMAL_CLOSE || close_type == SHUTDOWN_CLOSE) {
2827 if (pjob->status == PJOB_SMBD_SPOOLING) {
2828 /* take over the file now, smbd is done */
2829 if (sys_stat(pjob->filename, &sbuf, false) != 0) {
2830 status = map_nt_error_from_unix(errno);
2831 DEBUG(3, ("print_job_end: "
2832 "stat file failed for jobid %d\n",
2833 jobid));
2834 goto fail;
2837 pjob->status = LPQ_SPOOLING;
2839 } else {
2841 if ((sys_fstat(pjob->fd, &sbuf, false) != 0)) {
2842 status = map_nt_error_from_unix(errno);
2843 close(pjob->fd);
2844 DEBUG(3, ("print_job_end: "
2845 "stat file failed for jobid %d\n",
2846 jobid));
2847 goto fail;
2850 close(pjob->fd);
2853 pjob->size = sbuf.st_ex_size;
2854 } else {
2857 * Not a normal close, something has gone wrong. Cleanup.
2859 if (pjob->fd != -1) {
2860 close(pjob->fd);
2862 goto fail;
2865 /* Technically, this is not quite right. If the printer has a separator
2866 * page turned on, the NT spooler prints the separator page even if the
2867 * print job is 0 bytes. 010215 JRR */
2868 if (pjob->size == 0 || pjob->status == LPQ_DELETING) {
2869 /* don't bother spooling empty files or something being deleted. */
2870 DEBUG(5,("print_job_end: canceling spool of %s (%s)\n",
2871 pjob->filename, pjob->size ? "deleted" : "zero length" ));
2872 unlink(pjob->filename);
2873 pjob_delete(server_event_context(), msg_ctx, sharename, jobid);
2874 return NT_STATUS_OK;
2877 ret = (*(current_printif->job_submit))(snum, pjob);
2879 if (ret) {
2880 status = NT_STATUS_PRINT_CANCELLED;
2881 goto fail;
2884 /* The print job has been successfully handed over to the back-end */
2886 pjob->spooled = True;
2887 pjob->status = LPQ_QUEUED;
2888 pjob_store(server_event_context(), msg_ctx, sharename, jobid, pjob);
2890 /* make sure the database is up to date */
2891 if (print_cache_expired(lp_const_servicename(snum), True))
2892 print_queue_update(msg_ctx, snum, False);
2894 return NT_STATUS_OK;
2896 fail:
2898 /* The print job was not successfully started. Cleanup */
2899 /* Still need to add proper error return propagation! 010122:JRR */
2900 pjob->fd = -1;
2901 unlink(pjob->filename);
2902 pjob_delete(server_event_context(), msg_ctx, sharename, jobid);
2903 return status;
2906 /****************************************************************************
2907 Get a snapshot of jobs in the system without traversing.
2908 ****************************************************************************/
2910 static bool get_stored_queue_info(struct messaging_context *msg_ctx,
2911 struct tdb_print_db *pdb, int snum,
2912 int *pcount, print_queue_struct **ppqueue)
2914 TDB_DATA data, cgdata, jcdata;
2915 print_queue_struct *queue = NULL;
2916 uint32 qcount = 0;
2917 uint32 extra_count = 0;
2918 uint32_t changed_count = 0;
2919 int total_count = 0;
2920 size_t len = 0;
2921 uint32 i;
2922 int max_reported_jobs = lp_max_reported_jobs(snum);
2923 bool ret = False;
2924 const char* sharename = lp_servicename(snum);
2926 /* make sure the database is up to date */
2927 if (print_cache_expired(lp_const_servicename(snum), True))
2928 print_queue_update(msg_ctx, snum, False);
2930 *pcount = 0;
2931 *ppqueue = NULL;
2933 ZERO_STRUCT(data);
2934 ZERO_STRUCT(cgdata);
2936 /* Get the stored queue data. */
2937 data = tdb_fetch_compat(pdb->tdb, string_tdb_data("INFO/linear_queue_array"));
2939 if (data.dptr && data.dsize >= sizeof(qcount))
2940 len += tdb_unpack(data.dptr + len, data.dsize - len, "d", &qcount);
2942 /* Get the added jobs list. */
2943 cgdata = tdb_fetch_compat(pdb->tdb, string_tdb_data("INFO/jobs_added"));
2944 if (cgdata.dptr != NULL && (cgdata.dsize % 4 == 0))
2945 extra_count = cgdata.dsize/4;
2947 /* Get the changed jobs list. */
2948 jcdata = tdb_fetch_compat(pdb->tdb, string_tdb_data("INFO/jobs_changed"));
2949 if (jcdata.dptr != NULL && (jcdata.dsize % 4 == 0))
2950 changed_count = jcdata.dsize / 4;
2952 DEBUG(5,("get_stored_queue_info: qcount = %u, extra_count = %u\n", (unsigned int)qcount, (unsigned int)extra_count));
2954 /* Allocate the queue size. */
2955 if (qcount == 0 && extra_count == 0)
2956 goto out;
2958 if ((queue = SMB_MALLOC_ARRAY(print_queue_struct, qcount + extra_count)) == NULL)
2959 goto out;
2961 /* Retrieve the linearised queue data. */
2963 for( i = 0; i < qcount; i++) {
2964 uint32 qjob, qsize, qpage_count, qstatus, qpriority, qtime;
2965 len += tdb_unpack(data.dptr + len, data.dsize - len, "ddddddff",
2966 &qjob,
2967 &qsize,
2968 &qpage_count,
2969 &qstatus,
2970 &qpriority,
2971 &qtime,
2972 queue[i].fs_user,
2973 queue[i].fs_file);
2974 queue[i].job = qjob;
2975 queue[i].size = qsize;
2976 queue[i].page_count = qpage_count;
2977 queue[i].status = qstatus;
2978 queue[i].priority = qpriority;
2979 queue[i].time = qtime;
2982 total_count = qcount;
2984 /* Add new jobids to the queue. */
2985 for( i = 0; i < extra_count; i++) {
2986 uint32 jobid;
2987 struct printjob *pjob;
2989 jobid = IVAL(cgdata.dptr, i*4);
2990 DEBUG(5,("get_stored_queue_info: added job = %u\n", (unsigned int)jobid));
2991 pjob = print_job_find(lp_const_servicename(snum), jobid);
2992 if (!pjob) {
2993 DEBUG(5,("get_stored_queue_info: failed to find added job = %u\n", (unsigned int)jobid));
2994 remove_from_jobs_added(sharename, jobid);
2995 continue;
2998 queue[total_count].job = jobid;
2999 queue[total_count].size = pjob->size;
3000 queue[total_count].page_count = pjob->page_count;
3001 queue[total_count].status = pjob->status;
3002 queue[total_count].priority = 1;
3003 queue[total_count].time = pjob->starttime;
3004 fstrcpy(queue[total_count].fs_user, pjob->user);
3005 fstrcpy(queue[total_count].fs_file, pjob->jobname);
3006 total_count++;
3009 /* Update the changed jobids. */
3010 for (i = 0; i < changed_count; i++) {
3011 uint32_t jobid = IVAL(jcdata.dptr, i * 4);
3012 uint32_t j;
3013 bool found = false;
3015 for (j = 0; j < total_count; j++) {
3016 if (queue[j].job == jobid) {
3017 found = true;
3018 break;
3022 if (found) {
3023 struct printjob *pjob;
3025 DEBUG(5,("get_stored_queue_info: changed job: %u\n",
3026 (unsigned int) jobid));
3028 pjob = print_job_find(sharename, jobid);
3029 if (pjob == NULL) {
3030 DEBUG(5,("get_stored_queue_info: failed to find "
3031 "changed job = %u\n",
3032 (unsigned int) jobid));
3033 remove_from_jobs_changed(sharename, jobid);
3034 continue;
3037 queue[j].job = jobid;
3038 queue[j].size = pjob->size;
3039 queue[j].page_count = pjob->page_count;
3040 queue[j].status = pjob->status;
3041 queue[j].priority = 1;
3042 queue[j].time = pjob->starttime;
3043 fstrcpy(queue[j].fs_user, pjob->user);
3044 fstrcpy(queue[j].fs_file, pjob->jobname);
3046 DEBUG(5,("get_stored_queue_info: updated queue[%u], jobid: %u, jobname: %s\n",
3047 (unsigned int) j, (unsigned int) jobid, pjob->jobname));
3050 remove_from_jobs_changed(sharename, jobid);
3053 /* Sort the queue by submission time otherwise they are displayed
3054 in hash order. */
3056 TYPESAFE_QSORT(queue, total_count, printjob_comp);
3058 DEBUG(5,("get_stored_queue_info: total_count = %u\n", (unsigned int)total_count));
3060 if (max_reported_jobs && total_count > max_reported_jobs)
3061 total_count = max_reported_jobs;
3063 *ppqueue = queue;
3064 *pcount = total_count;
3066 ret = True;
3068 out:
3070 SAFE_FREE(data.dptr);
3071 SAFE_FREE(cgdata.dptr);
3072 return ret;
3075 /****************************************************************************
3076 Get a printer queue listing.
3077 set queue = NULL and status = NULL if you just want to update the cache
3078 ****************************************************************************/
3080 int print_queue_status(struct messaging_context *msg_ctx, int snum,
3081 print_queue_struct **ppqueue,
3082 print_status_struct *status)
3084 fstring keystr;
3085 TDB_DATA data, key;
3086 const char *sharename;
3087 struct tdb_print_db *pdb;
3088 int count = 0;
3090 /* make sure the database is up to date */
3092 if (print_cache_expired(lp_const_servicename(snum), True))
3093 print_queue_update(msg_ctx, snum, False);
3095 /* return if we are done */
3096 if ( !ppqueue || !status )
3097 return 0;
3099 *ppqueue = NULL;
3100 sharename = lp_const_servicename(snum);
3101 pdb = get_print_db_byname(sharename);
3103 if (!pdb)
3104 return 0;
3107 * Fetch the queue status. We must do this first, as there may
3108 * be no jobs in the queue.
3111 ZERO_STRUCTP(status);
3112 slprintf(keystr, sizeof(keystr)-1, "STATUS/%s", sharename);
3113 key = string_tdb_data(keystr);
3115 data = tdb_fetch_compat(pdb->tdb, key);
3116 if (data.dptr) {
3117 if (data.dsize == sizeof(*status)) {
3118 /* this memcpy is ok since the status struct was
3119 not packed before storing it in the tdb */
3120 memcpy(status, data.dptr, sizeof(*status));
3122 SAFE_FREE(data.dptr);
3126 * Now, fetch the print queue information. We first count the number
3127 * of entries, and then only retrieve the queue if necessary.
3130 if (!get_stored_queue_info(msg_ctx, pdb, snum, &count, ppqueue)) {
3131 release_print_db(pdb);
3132 return 0;
3135 release_print_db(pdb);
3136 return count;
3139 /****************************************************************************
3140 Pause a queue.
3141 ****************************************************************************/
3143 WERROR print_queue_pause(const struct auth_session_info *server_info,
3144 struct messaging_context *msg_ctx, int snum)
3146 int ret;
3147 struct printif *current_printif = get_printer_fns( snum );
3149 if (!print_access_check(server_info, msg_ctx, snum,
3150 PRINTER_ACCESS_ADMINISTER)) {
3151 return WERR_ACCESS_DENIED;
3155 become_root();
3157 ret = (*(current_printif->queue_pause))(snum);
3159 unbecome_root();
3161 if (ret != 0) {
3162 return WERR_INVALID_PARAM;
3165 /* force update the database */
3166 print_cache_flush(lp_const_servicename(snum));
3168 /* Send a printer notify message */
3170 notify_printer_status(server_event_context(), msg_ctx, snum,
3171 PRINTER_STATUS_PAUSED);
3173 return WERR_OK;
3176 /****************************************************************************
3177 Resume a queue.
3178 ****************************************************************************/
3180 WERROR print_queue_resume(const struct auth_session_info *server_info,
3181 struct messaging_context *msg_ctx, int snum)
3183 int ret;
3184 struct printif *current_printif = get_printer_fns( snum );
3186 if (!print_access_check(server_info, msg_ctx, snum,
3187 PRINTER_ACCESS_ADMINISTER)) {
3188 return WERR_ACCESS_DENIED;
3191 become_root();
3193 ret = (*(current_printif->queue_resume))(snum);
3195 unbecome_root();
3197 if (ret != 0) {
3198 return WERR_INVALID_PARAM;
3201 /* make sure the database is up to date */
3202 if (print_cache_expired(lp_const_servicename(snum), True))
3203 print_queue_update(msg_ctx, snum, True);
3205 /* Send a printer notify message */
3207 notify_printer_status(server_event_context(), msg_ctx, snum,
3208 PRINTER_STATUS_OK);
3210 return WERR_OK;
3213 /****************************************************************************
3214 Purge a queue - implemented by deleting all jobs that we can delete.
3215 ****************************************************************************/
3217 WERROR print_queue_purge(const struct auth_session_info *server_info,
3218 struct messaging_context *msg_ctx, int snum)
3220 print_queue_struct *queue;
3221 print_status_struct status;
3222 int njobs, i;
3223 bool can_job_admin;
3225 /* Force and update so the count is accurate (i.e. not a cached count) */
3226 print_queue_update(msg_ctx, snum, True);
3228 can_job_admin = print_access_check(server_info,
3229 msg_ctx,
3230 snum,
3231 JOB_ACCESS_ADMINISTER);
3232 njobs = print_queue_status(msg_ctx, snum, &queue, &status);
3234 if ( can_job_admin )
3235 become_root();
3237 for (i=0;i<njobs;i++) {
3238 bool owner = is_owner(server_info, lp_const_servicename(snum),
3239 queue[i].job);
3241 if (owner || can_job_admin) {
3242 print_job_delete1(server_event_context(), msg_ctx,
3243 snum, queue[i].job);
3247 if ( can_job_admin )
3248 unbecome_root();
3250 /* update the cache */
3251 print_queue_update(msg_ctx, snum, True);
3253 SAFE_FREE(queue);
3255 return WERR_OK;