source: branches/samba-3.3.x/source/printing/printing.c@ 693

Last change on this file since 693 was 578, checked in by Herwig Bauernfeind, 14 years ago

Update Samba 3.3 to 3.3.15 (security update)

File size: 80.3 KB
Line 
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
7
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.
12
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.
17
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/>.
20*/
21
22#include "includes.h"
23#include "printing.h"
24
25extern SIG_ATOMIC_T got_sig_term;
26extern SIG_ATOMIC_T reload_after_sighup;
27extern struct current_user current_user;
28extern userdom_struct current_user_info;
29
30/* Current printer interface */
31static bool remove_from_jobs_changed(const char* sharename, uint32 jobid);
32
33/*
34 the printing backend revolves around a tdb database that stores the
35 SMB view of the print queue
36
37 The key for this database is a jobid - a internally generated number that
38 uniquely identifies a print job
39
40 reading the print queue involves two steps:
41 - possibly running lpq and updating the internal database from that
42 - reading entries from the database
43
44 jobids are assigned when a job starts spooling.
45*/
46
47static TDB_CONTEXT *rap_tdb;
48static uint16 next_rap_jobid;
49struct rap_jobid_key {
50 fstring sharename;
51 uint32 jobid;
52};
53
54/***************************************************************************
55 Nightmare. LANMAN jobid's are 16 bit numbers..... We must map them to 32
56 bit RPC jobids.... JRA.
57***************************************************************************/
58
59uint16 pjobid_to_rap(const char* sharename, uint32 jobid)
60{
61 uint16 rap_jobid;
62 TDB_DATA data, key;
63 struct rap_jobid_key jinfo;
64 uint8 buf[2];
65
66 DEBUG(10,("pjobid_to_rap: called.\n"));
67
68 if (!rap_tdb) {
69 /* Create the in-memory tdb. */
70 rap_tdb = tdb_open_log(NULL, 0, TDB_INTERNAL, (O_RDWR|O_CREAT), 0644);
71 if (!rap_tdb)
72 return 0;
73 }
74
75 ZERO_STRUCT( jinfo );
76 fstrcpy( jinfo.sharename, sharename );
77 jinfo.jobid = jobid;
78 key.dptr = (uint8 *)&jinfo;
79 key.dsize = sizeof(jinfo);
80
81 data = tdb_fetch(rap_tdb, key);
82 if (data.dptr && data.dsize == sizeof(uint16)) {
83 rap_jobid = SVAL(data.dptr, 0);
84 SAFE_FREE(data.dptr);
85 DEBUG(10,("pjobid_to_rap: jobid %u maps to RAP jobid %u\n",
86 (unsigned int)jobid, (unsigned int)rap_jobid));
87 return rap_jobid;
88 }
89 SAFE_FREE(data.dptr);
90 /* Not found - create and store mapping. */
91 rap_jobid = ++next_rap_jobid;
92 if (rap_jobid == 0)
93 rap_jobid = ++next_rap_jobid;
94 SSVAL(buf,0,rap_jobid);
95 data.dptr = buf;
96 data.dsize = sizeof(rap_jobid);
97 tdb_store(rap_tdb, key, data, TDB_REPLACE);
98 tdb_store(rap_tdb, data, key, TDB_REPLACE);
99
100 DEBUG(10,("pjobid_to_rap: created jobid %u maps to RAP jobid %u\n",
101 (unsigned int)jobid, (unsigned int)rap_jobid));
102 return rap_jobid;
103}
104
105bool rap_to_pjobid(uint16 rap_jobid, fstring sharename, uint32 *pjobid)
106{
107 TDB_DATA data, key;
108 uint8 buf[2];
109
110 DEBUG(10,("rap_to_pjobid called.\n"));
111
112 if (!rap_tdb)
113 return False;
114
115 SSVAL(buf,0,rap_jobid);
116 key.dptr = buf;
117 key.dsize = sizeof(rap_jobid);
118 data = tdb_fetch(rap_tdb, key);
119 if ( data.dptr && data.dsize == sizeof(struct rap_jobid_key) )
120 {
121 struct rap_jobid_key *jinfo = (struct rap_jobid_key*)data.dptr;
122 fstrcpy( sharename, jinfo->sharename );
123 *pjobid = jinfo->jobid;
124 DEBUG(10,("rap_to_pjobid: jobid %u maps to RAP jobid %u\n",
125 (unsigned int)*pjobid, (unsigned int)rap_jobid));
126 SAFE_FREE(data.dptr);
127 return True;
128 }
129
130 DEBUG(10,("rap_to_pjobid: Failed to lookup RAP jobid %u\n",
131 (unsigned int)rap_jobid));
132 SAFE_FREE(data.dptr);
133 return False;
134}
135
136static void rap_jobid_delete(const char* sharename, uint32 jobid)
137{
138 TDB_DATA key, data;
139 uint16 rap_jobid;
140 struct rap_jobid_key jinfo;
141 uint8 buf[2];
142
143 DEBUG(10,("rap_jobid_delete: called.\n"));
144
145 if (!rap_tdb)
146 return;
147
148 ZERO_STRUCT( jinfo );
149 fstrcpy( jinfo.sharename, sharename );
150 jinfo.jobid = jobid;
151 key.dptr = (uint8 *)&jinfo;
152 key.dsize = sizeof(jinfo);
153
154 data = tdb_fetch(rap_tdb, key);
155 if (!data.dptr || (data.dsize != sizeof(uint16))) {
156 DEBUG(10,("rap_jobid_delete: cannot find jobid %u\n",
157 (unsigned int)jobid ));
158 SAFE_FREE(data.dptr);
159 return;
160 }
161
162 DEBUG(10,("rap_jobid_delete: deleting jobid %u\n",
163 (unsigned int)jobid ));
164
165 rap_jobid = SVAL(data.dptr, 0);
166 SAFE_FREE(data.dptr);
167 SSVAL(buf,0,rap_jobid);
168 data.dptr = buf;
169 data.dsize = sizeof(rap_jobid);
170 tdb_delete(rap_tdb, key);
171 tdb_delete(rap_tdb, data);
172}
173
174static int get_queue_status(const char* sharename, print_status_struct *);
175
176/****************************************************************************
177 Initialise the printing backend. Called once at startup before the fork().
178****************************************************************************/
179
180bool print_backend_init(struct messaging_context *msg_ctx)
181{
182 const char *sversion = "INFO/version";
183 int services = lp_numservices();
184 int snum;
185
186 unlink(lock_path("printing.tdb"));
187 mkdir(lock_path("printing"),0755);
188
189 /* handle a Samba upgrade */
190
191 for (snum = 0; snum < services; snum++) {
192 struct tdb_print_db *pdb;
193 if (!lp_print_ok(snum))
194 continue;
195
196 pdb = get_print_db_byname(lp_const_servicename(snum));
197 if (!pdb)
198 continue;
199 if (tdb_lock_bystring(pdb->tdb, sversion) == -1) {
200 DEBUG(0,("print_backend_init: Failed to open printer %s database\n", lp_const_servicename(snum) ));
201 release_print_db(pdb);
202 return False;
203 }
204 if (tdb_fetch_int32(pdb->tdb, sversion) != PRINT_DATABASE_VERSION) {
205 tdb_wipe_all(pdb->tdb);
206 tdb_store_int32(pdb->tdb, sversion, PRINT_DATABASE_VERSION);
207 }
208 tdb_unlock_bystring(pdb->tdb, sversion);
209 release_print_db(pdb);
210 }
211
212 close_all_print_db(); /* Don't leave any open. */
213
214 /* do NT print initialization... */
215 return nt_printing_init(msg_ctx);
216}
217
218/****************************************************************************
219 Shut down printing backend. Called once at shutdown to close the tdb.
220****************************************************************************/
221
222void printing_end(void)
223{
224 close_all_print_db(); /* Don't leave any open. */
225}
226
227/****************************************************************************
228 Retrieve the set of printing functions for a given service. This allows
229 us to set the printer function table based on the value of the 'printing'
230 service parameter.
231
232 Use the generic interface as the default and only use cups interface only
233 when asked for (and only when supported)
234****************************************************************************/
235
236static struct printif *get_printer_fns_from_type( enum printing_types type )
237{
238 struct printif *printer_fns = &generic_printif;
239
240#ifdef HAVE_CUPS
241 if ( type == PRINT_CUPS ) {
242 printer_fns = &cups_printif;
243 }
244#endif /* HAVE_CUPS */
245
246#ifdef HAVE_IPRINT
247 if ( type == PRINT_IPRINT ) {
248 printer_fns = &iprint_printif;
249 }
250#endif /* HAVE_IPRINT */
251
252 printer_fns->type = type;
253
254 return printer_fns;
255}
256
257static struct printif *get_printer_fns( int snum )
258{
259 return get_printer_fns_from_type( (enum printing_types)lp_printing(snum) );
260}
261
262
263/****************************************************************************
264 Useful function to generate a tdb key.
265****************************************************************************/
266
267static TDB_DATA print_key(uint32 jobid, uint32 *tmp)
268{
269 TDB_DATA ret;
270
271 SIVAL(tmp, 0, jobid);
272 ret.dptr = (uint8 *)tmp;
273 ret.dsize = sizeof(*tmp);
274 return ret;
275}
276
277/***********************************************************************
278 unpack a pjob from a tdb buffer
279***********************************************************************/
280
281int unpack_pjob( uint8 *buf, int buflen, struct printjob *pjob )
282{
283 int len = 0;
284 int used;
285 uint32 pjpid, pjsysjob, pjfd, pjstarttime, pjstatus;
286 uint32 pjsize, pjpage_count, pjspooled, pjsmbjob;
287
288 if ( !buf || !pjob )
289 return -1;
290
291 len += tdb_unpack(buf+len, buflen-len, "dddddddddffff",
292 &pjpid,
293 &pjsysjob,
294 &pjfd,
295 &pjstarttime,
296 &pjstatus,
297 &pjsize,
298 &pjpage_count,
299 &pjspooled,
300 &pjsmbjob,
301 pjob->filename,
302 pjob->jobname,
303 pjob->user,
304 pjob->queuename);
305
306 if ( len == -1 )
307 return -1;
308
309 if ( (used = unpack_devicemode(&pjob->nt_devmode, buf+len, buflen-len)) == -1 )
310 return -1;
311
312 len += used;
313
314 pjob->pid = pjpid;
315 pjob->sysjob = pjsysjob;
316 pjob->fd = pjfd;
317 pjob->starttime = pjstarttime;
318 pjob->status = pjstatus;
319 pjob->size = pjsize;
320 pjob->page_count = pjpage_count;
321 pjob->spooled = pjspooled;
322 pjob->smbjob = pjsmbjob;
323
324 return len;
325
326}
327
328/****************************************************************************
329 Useful function to find a print job in the database.
330****************************************************************************/
331
332static struct printjob *print_job_find(const char *sharename, uint32 jobid)
333{
334 static struct printjob pjob;
335 uint32_t tmp;
336 TDB_DATA ret;
337 struct tdb_print_db *pdb = get_print_db_byname(sharename);
338
339 DEBUG(10,("print_job_find: looking up job %u for share %s\n",
340 (unsigned int)jobid, sharename ));
341
342 if (!pdb) {
343 return NULL;
344 }
345
346 ret = tdb_fetch(pdb->tdb, print_key(jobid, &tmp));
347 release_print_db(pdb);
348
349 if (!ret.dptr) {
350 DEBUG(10,("print_job_find: failed to find jobid %u.\n", (unsigned int)jobid ));
351 return NULL;
352 }
353
354 if ( pjob.nt_devmode ) {
355 free_nt_devicemode( &pjob.nt_devmode );
356 }
357
358 ZERO_STRUCT( pjob );
359
360 if ( unpack_pjob( ret.dptr, ret.dsize, &pjob ) == -1 ) {
361 DEBUG(10,("print_job_find: failed to unpack jobid %u.\n", (unsigned int)jobid ));
362 SAFE_FREE(ret.dptr);
363 return NULL;
364 }
365
366 SAFE_FREE(ret.dptr);
367
368 DEBUG(10,("print_job_find: returning system job %d for jobid %u.\n",
369 (int)pjob.sysjob, (unsigned int)jobid ));
370
371 return &pjob;
372}
373
374/* Convert a unix jobid to a smb jobid */
375
376struct unixjob_traverse_state {
377 int sysjob;
378 uint32 sysjob_to_jobid_value;
379};
380
381static int unixjob_traverse_fn(TDB_CONTEXT *the_tdb, TDB_DATA key,
382 TDB_DATA data, void *private_data)
383{
384 struct printjob *pjob;
385 struct unixjob_traverse_state *state =
386 (struct unixjob_traverse_state *)private_data;
387
388 if (!data.dptr || data.dsize == 0)
389 return 0;
390
391 pjob = (struct printjob *)data.dptr;
392 if (key.dsize != sizeof(uint32))
393 return 0;
394
395 if (state->sysjob == pjob->sysjob) {
396 uint32 jobid = IVAL(key.dptr,0);
397
398 state->sysjob_to_jobid_value = jobid;
399 return 1;
400 }
401
402 return 0;
403}
404
405/****************************************************************************
406 This is a *horribly expensive call as we have to iterate through all the
407 current printer tdb's. Don't do this often ! JRA.
408****************************************************************************/
409
410uint32 sysjob_to_jobid(int unix_jobid)
411{
412 int services = lp_numservices();
413 int snum;
414 struct unixjob_traverse_state state;
415
416 state.sysjob = unix_jobid;
417 state.sysjob_to_jobid_value = (uint32)-1;
418
419 for (snum = 0; snum < services; snum++) {
420 struct tdb_print_db *pdb;
421 if (!lp_print_ok(snum))
422 continue;
423 pdb = get_print_db_byname(lp_const_servicename(snum));
424 if (!pdb) {
425 continue;
426 }
427 tdb_traverse(pdb->tdb, unixjob_traverse_fn, &state);
428 release_print_db(pdb);
429 if (state.sysjob_to_jobid_value != (uint32)-1)
430 return state.sysjob_to_jobid_value;
431 }
432 return (uint32)-1;
433}
434
435/****************************************************************************
436 Send notifications based on what has changed after a pjob_store.
437****************************************************************************/
438
439static const struct {
440 uint32 lpq_status;
441 uint32 spoolss_status;
442} lpq_to_spoolss_status_map[] = {
443 { LPQ_QUEUED, JOB_STATUS_QUEUED },
444 { LPQ_PAUSED, JOB_STATUS_PAUSED },
445 { LPQ_SPOOLING, JOB_STATUS_SPOOLING },
446 { LPQ_PRINTING, JOB_STATUS_PRINTING },
447 { LPQ_DELETING, JOB_STATUS_DELETING },
448 { LPQ_OFFLINE, JOB_STATUS_OFFLINE },
449 { LPQ_PAPEROUT, JOB_STATUS_PAPEROUT },
450 { LPQ_PRINTED, JOB_STATUS_PRINTED },
451 { LPQ_DELETED, JOB_STATUS_DELETED },
452 { LPQ_BLOCKED, JOB_STATUS_BLOCKED },
453 { LPQ_USER_INTERVENTION, JOB_STATUS_USER_INTERVENTION },
454 { -1, 0 }
455};
456
457/* Convert a lpq status value stored in printing.tdb into the
458 appropriate win32 API constant. */
459
460static uint32 map_to_spoolss_status(uint32 lpq_status)
461{
462 int i = 0;
463
464 while (lpq_to_spoolss_status_map[i].lpq_status != -1) {
465 if (lpq_to_spoolss_status_map[i].lpq_status == lpq_status)
466 return lpq_to_spoolss_status_map[i].spoolss_status;
467 i++;
468 }
469
470 return 0;
471}
472
473static void pjob_store_notify(const char* sharename, uint32 jobid, struct printjob *old_data,
474 struct printjob *new_data)
475{
476 bool new_job = False;
477
478 if (!old_data)
479 new_job = True;
480
481 /* Job attributes that can't be changed. We only send
482 notification for these on a new job. */
483
484 /* ACHTUNG! Due to a bug in Samba's spoolss parsing of the
485 NOTIFY_INFO_DATA buffer, we *have* to send the job submission
486 time first or else we'll end up with potential alignment
487 errors. I don't think the systemtime should be spooled as
488 a string, but this gets us around that error.
489 --jerry (i'll feel dirty for this) */
490
491 if (new_job) {
492 notify_job_submitted(sharename, jobid, new_data->starttime);
493 notify_job_username(sharename, jobid, new_data->user);
494 }
495
496 if (new_job || !strequal(old_data->jobname, new_data->jobname))
497 notify_job_name(sharename, jobid, new_data->jobname);
498
499 /* Job attributes of a new job or attributes that can be
500 modified. */
501
502 if (new_job || !strequal(old_data->jobname, new_data->jobname))
503 notify_job_name(sharename, jobid, new_data->jobname);
504
505 if (new_job || old_data->status != new_data->status)
506 notify_job_status(sharename, jobid, map_to_spoolss_status(new_data->status));
507
508 if (new_job || old_data->size != new_data->size)
509 notify_job_total_bytes(sharename, jobid, new_data->size);
510
511 if (new_job || old_data->page_count != new_data->page_count)
512 notify_job_total_pages(sharename, jobid, new_data->page_count);
513}
514
515/****************************************************************************
516 Store a job structure back to the database.
517****************************************************************************/
518
519static bool pjob_store(const char* sharename, uint32 jobid, struct printjob *pjob)
520{
521 uint32_t tmp;
522 TDB_DATA old_data, new_data;
523 bool ret = False;
524 struct tdb_print_db *pdb = get_print_db_byname(sharename);
525 uint8 *buf = NULL;
526 int len, newlen, buflen;
527
528
529 if (!pdb)
530 return False;
531
532 /* Get old data */
533
534 old_data = tdb_fetch(pdb->tdb, print_key(jobid, &tmp));
535
536 /* Doh! Now we have to pack/unpack data since the NT_DEVICEMODE was added */
537
538 newlen = 0;
539
540 do {
541 len = 0;
542 buflen = newlen;
543 len += tdb_pack(buf+len, buflen-len, "dddddddddffff",
544 (uint32)pjob->pid,
545 (uint32)pjob->sysjob,
546 (uint32)pjob->fd,
547 (uint32)pjob->starttime,
548 (uint32)pjob->status,
549 (uint32)pjob->size,
550 (uint32)pjob->page_count,
551 (uint32)pjob->spooled,
552 (uint32)pjob->smbjob,
553 pjob->filename,
554 pjob->jobname,
555 pjob->user,
556 pjob->queuename);
557
558 len += pack_devicemode(pjob->nt_devmode, buf+len, buflen-len);
559
560 if (buflen != len) {
561 buf = (uint8 *)SMB_REALLOC(buf, len);
562 if (!buf) {
563 DEBUG(0,("pjob_store: failed to enlarge buffer!\n"));
564 goto done;
565 }
566 newlen = len;
567 }
568 } while ( buflen != len );
569
570
571 /* Store new data */
572
573 new_data.dptr = buf;
574 new_data.dsize = len;
575 ret = (tdb_store(pdb->tdb, print_key(jobid, &tmp), new_data,
576 TDB_REPLACE) == 0);
577
578 release_print_db(pdb);
579
580 /* Send notify updates for what has changed */
581
582 if ( ret ) {
583 struct printjob old_pjob;
584
585 if ( old_data.dsize )
586 {
587 if ( unpack_pjob( old_data.dptr, old_data.dsize, &old_pjob ) != -1 )
588 {
589 pjob_store_notify( sharename, jobid, &old_pjob , pjob );
590 free_nt_devicemode( &old_pjob.nt_devmode );
591 }
592 }
593 else {
594 /* new job */
595 pjob_store_notify( sharename, jobid, NULL, pjob );
596 }
597 }
598
599done:
600 SAFE_FREE( old_data.dptr );
601 SAFE_FREE( buf );
602
603 return ret;
604}
605
606/****************************************************************************
607 Remove a job structure from the database.
608****************************************************************************/
609
610void pjob_delete(const char* sharename, uint32 jobid)
611{
612 uint32_t tmp;
613 struct printjob *pjob;
614 uint32 job_status = 0;
615 struct tdb_print_db *pdb;
616
617 pdb = get_print_db_byname( sharename );
618
619 if (!pdb)
620 return;
621
622 pjob = print_job_find( sharename, jobid );
623
624 if (!pjob) {
625 DEBUG(5, ("pjob_delete: we were asked to delete nonexistent job %u\n",
626 (unsigned int)jobid));
627 release_print_db(pdb);
628 return;
629 }
630
631 /* We must cycle through JOB_STATUS_DELETING and
632 JOB_STATUS_DELETED for the port monitor to delete the job
633 properly. */
634
635 job_status = JOB_STATUS_DELETING|JOB_STATUS_DELETED;
636 notify_job_status(sharename, jobid, job_status);
637
638 /* Remove from printing.tdb */
639
640 tdb_delete(pdb->tdb, print_key(jobid, &tmp));
641 remove_from_jobs_changed(sharename, jobid);
642 release_print_db( pdb );
643 rap_jobid_delete(sharename, jobid);
644}
645
646/****************************************************************************
647 Parse a file name from the system spooler to generate a jobid.
648****************************************************************************/
649
650static uint32 print_parse_jobid(char *fname)
651{
652 int jobid;
653
654 if (strncmp(fname,PRINT_SPOOL_PREFIX,strlen(PRINT_SPOOL_PREFIX)) != 0)
655 return (uint32)-1;
656 fname += strlen(PRINT_SPOOL_PREFIX);
657
658 jobid = atoi(fname);
659 if (jobid <= 0)
660 return (uint32)-1;
661
662 return (uint32)jobid;
663}
664
665/****************************************************************************
666 List a unix job in the print database.
667****************************************************************************/
668
669static void print_unix_job(const char *sharename, print_queue_struct *q, uint32 jobid)
670{
671 struct printjob pj, *old_pj;
672
673 if (jobid == (uint32)-1)
674 jobid = q->job + UNIX_JOB_START;
675
676 /* Preserve the timestamp on an existing unix print job */
677
678 old_pj = print_job_find(sharename, jobid);
679
680 ZERO_STRUCT(pj);
681
682 pj.pid = (pid_t)-1;
683 pj.sysjob = q->job;
684 pj.fd = -1;
685 pj.starttime = old_pj ? old_pj->starttime : q->time;
686 pj.status = q->status;
687 pj.size = q->size;
688 pj.spooled = True;
689 fstrcpy(pj.filename, old_pj ? old_pj->filename : "");
690 if (jobid < UNIX_JOB_START) {
691 pj.smbjob = True;
692 fstrcpy(pj.jobname, old_pj ? old_pj->jobname : "Remote Downlevel Document");
693 } else {
694 pj.smbjob = False;
695 fstrcpy(pj.jobname, old_pj ? old_pj->jobname : q->fs_file);
696 }
697 fstrcpy(pj.user, old_pj ? old_pj->user : q->fs_user);
698 fstrcpy(pj.queuename, old_pj ? old_pj->queuename : sharename );
699
700 pjob_store(sharename, jobid, &pj);
701}
702
703
704struct traverse_struct {
705 print_queue_struct *queue;
706 int qcount, snum, maxcount, total_jobs;
707 const char *sharename;
708 time_t lpq_time;
709 const char *lprm_command;
710 struct printif *print_if;
711};
712
713/****************************************************************************
714 Utility fn to delete any jobs that are no longer active.
715****************************************************************************/
716
717static int traverse_fn_delete(TDB_CONTEXT *t, TDB_DATA key, TDB_DATA data, void *state)
718{
719 struct traverse_struct *ts = (struct traverse_struct *)state;
720 struct printjob pjob;
721 uint32 jobid;
722 int i = 0;
723
724 if ( key.dsize != sizeof(jobid) )
725 return 0;
726
727 jobid = IVAL(key.dptr, 0);
728 if ( unpack_pjob( data.dptr, data.dsize, &pjob ) == -1 )
729 return 0;
730 free_nt_devicemode( &pjob.nt_devmode );
731
732
733 if (!pjob.smbjob) {
734 /* remove a unix job if it isn't in the system queue any more */
735
736 for (i=0;i<ts->qcount;i++) {
737 uint32 u_jobid = (ts->queue[i].job + UNIX_JOB_START);
738 if (jobid == u_jobid)
739 break;
740 }
741 if (i == ts->qcount) {
742 DEBUG(10,("traverse_fn_delete: pjob %u deleted due to !smbjob\n",
743 (unsigned int)jobid ));
744 pjob_delete(ts->sharename, jobid);
745 return 0;
746 }
747
748 /* need to continue the the bottom of the function to
749 save the correct attributes */
750 }
751
752 /* maybe it hasn't been spooled yet */
753 if (!pjob.spooled) {
754 /* if a job is not spooled and the process doesn't
755 exist then kill it. This cleans up after smbd
756 deaths */
757 if (!process_exists_by_pid(pjob.pid)) {
758 DEBUG(10,("traverse_fn_delete: pjob %u deleted due to !process_exists (%u)\n",
759 (unsigned int)jobid, (unsigned int)pjob.pid ));
760 pjob_delete(ts->sharename, jobid);
761 } else
762 ts->total_jobs++;
763 return 0;
764 }
765
766 /* this check only makes sense for jobs submitted from Windows clients */
767
768 if ( pjob.smbjob ) {
769 for (i=0;i<ts->qcount;i++) {
770 uint32 curr_jobid;
771
772 if ( pjob.status == LPQ_DELETED )
773 continue;
774
775 curr_jobid = print_parse_jobid(ts->queue[i].fs_file);
776
777 if (jobid == curr_jobid) {
778
779 /* try to clean up any jobs that need to be deleted */
780
781 if ( pjob.status == LPQ_DELETING ) {
782 int result;
783
784 result = (*(ts->print_if->job_delete))(
785 ts->sharename, ts->lprm_command, &pjob );
786
787 if ( result != 0 ) {
788 /* if we can't delete, then reset the job status */
789 pjob.status = LPQ_QUEUED;
790 pjob_store(ts->sharename, jobid, &pjob);
791 }
792 else {
793 /* if we deleted the job, the remove the tdb record */
794 pjob_delete(ts->sharename, jobid);
795 pjob.status = LPQ_DELETED;
796 }
797
798 }
799
800 break;
801 }
802 }
803 }
804
805 /* The job isn't in the system queue - we have to assume it has
806 completed, so delete the database entry. */
807
808 if (i == ts->qcount) {
809
810 /* A race can occur between the time a job is spooled and
811 when it appears in the lpq output. This happens when
812 the job is added to printing.tdb when another smbd
813 running print_queue_update() has completed a lpq and
814 is currently traversing the printing tdb and deleting jobs.
815 Don't delete the job if it was submitted after the lpq_time. */
816
817 if (pjob.starttime < ts->lpq_time) {
818 DEBUG(10,("traverse_fn_delete: pjob %u deleted due to pjob.starttime (%u) < ts->lpq_time (%u)\n",
819 (unsigned int)jobid,
820 (unsigned int)pjob.starttime,
821 (unsigned int)ts->lpq_time ));
822 pjob_delete(ts->sharename, jobid);
823 } else
824 ts->total_jobs++;
825 return 0;
826 }
827
828 /* Save the pjob attributes we will store.
829 FIXME!!! This is the only place where queue->job
830 represents the SMB jobid --jerry */
831
832 ts->queue[i].job = jobid;
833 ts->queue[i].size = pjob.size;
834 ts->queue[i].page_count = pjob.page_count;
835 ts->queue[i].status = pjob.status;
836 ts->queue[i].priority = 1;
837 ts->queue[i].time = pjob.starttime;
838 fstrcpy(ts->queue[i].fs_user, pjob.user);
839 fstrcpy(ts->queue[i].fs_file, pjob.jobname);
840
841 ts->total_jobs++;
842
843 return 0;
844}
845
846/****************************************************************************
847 Check if the print queue has been updated recently enough.
848****************************************************************************/
849
850static void print_cache_flush(const char *sharename)
851{
852 fstring key;
853 struct tdb_print_db *pdb = get_print_db_byname(sharename);
854
855 if (!pdb)
856 return;
857 slprintf(key, sizeof(key)-1, "CACHE/%s", sharename);
858 tdb_store_int32(pdb->tdb, key, -1);
859 release_print_db(pdb);
860}
861
862/****************************************************************************
863 Check if someone already thinks they are doing the update.
864****************************************************************************/
865
866static pid_t get_updating_pid(const char *sharename)
867{
868 fstring keystr;
869 TDB_DATA data, key;
870 pid_t updating_pid;
871 struct tdb_print_db *pdb = get_print_db_byname(sharename);
872
873 if (!pdb)
874 return (pid_t)-1;
875 slprintf(keystr, sizeof(keystr)-1, "UPDATING/%s", sharename);
876 key = string_tdb_data(keystr);
877
878 data = tdb_fetch(pdb->tdb, key);
879 release_print_db(pdb);
880 if (!data.dptr || data.dsize != sizeof(pid_t)) {
881 SAFE_FREE(data.dptr);
882 return (pid_t)-1;
883 }
884
885 updating_pid = IVAL(data.dptr, 0);
886 SAFE_FREE(data.dptr);
887
888 if (process_exists_by_pid(updating_pid))
889 return updating_pid;
890
891 return (pid_t)-1;
892}
893
894/****************************************************************************
895 Set the fact that we're doing the update, or have finished doing the update
896 in the tdb.
897****************************************************************************/
898
899static void set_updating_pid(const fstring sharename, bool updating)
900{
901 fstring keystr;
902 TDB_DATA key;
903 TDB_DATA data;
904 pid_t updating_pid = sys_getpid();
905 uint8 buffer[4];
906
907 struct tdb_print_db *pdb = get_print_db_byname(sharename);
908
909 if (!pdb)
910 return;
911
912 slprintf(keystr, sizeof(keystr)-1, "UPDATING/%s", sharename);
913 key = string_tdb_data(keystr);
914
915 DEBUG(5, ("set_updating_pid: %s updating lpq cache for print share %s\n",
916 updating ? "" : "not ",
917 sharename ));
918
919 if ( !updating ) {
920 tdb_delete(pdb->tdb, key);
921 release_print_db(pdb);
922 return;
923 }
924
925 SIVAL( buffer, 0, updating_pid);
926 data.dptr = buffer;
927 data.dsize = 4; /* we always assume this is a 4 byte value */
928
929 tdb_store(pdb->tdb, key, data, TDB_REPLACE);
930 release_print_db(pdb);
931}
932
933/****************************************************************************
934 Sort print jobs by submittal time.
935****************************************************************************/
936
937static int printjob_comp(print_queue_struct *j1, print_queue_struct *j2)
938{
939 /* Silly cases */
940
941 if (!j1 && !j2)
942 return 0;
943 if (!j1)
944 return -1;
945 if (!j2)
946 return 1;
947
948 /* Sort on job start time */
949
950 if (j1->time == j2->time)
951 return 0;
952 return (j1->time > j2->time) ? 1 : -1;
953}
954
955/****************************************************************************
956 Store the sorted queue representation for later portmon retrieval.
957 Skip deleted jobs
958****************************************************************************/
959
960static void store_queue_struct(struct tdb_print_db *pdb, struct traverse_struct *pts)
961{
962 TDB_DATA data;
963 int max_reported_jobs = lp_max_reported_jobs(pts->snum);
964 print_queue_struct *queue = pts->queue;
965 size_t len;
966 size_t i;
967 unsigned int qcount;
968
969 if (max_reported_jobs && (max_reported_jobs < pts->qcount))
970 pts->qcount = max_reported_jobs;
971 qcount = 0;
972
973 /* Work out the size. */
974 data.dsize = 0;
975 data.dsize += tdb_pack(NULL, 0, "d", qcount);
976
977 for (i = 0; i < pts->qcount; i++) {
978 if ( queue[i].status == LPQ_DELETED )
979 continue;
980
981 qcount++;
982 data.dsize += tdb_pack(NULL, 0, "ddddddff",
983 (uint32)queue[i].job,
984 (uint32)queue[i].size,
985 (uint32)queue[i].page_count,
986 (uint32)queue[i].status,
987 (uint32)queue[i].priority,
988 (uint32)queue[i].time,
989 queue[i].fs_user,
990 queue[i].fs_file);
991 }
992
993 if ((data.dptr = (uint8 *)SMB_MALLOC(data.dsize)) == NULL)
994 return;
995
996 len = 0;
997 len += tdb_pack(data.dptr + len, data.dsize - len, "d", qcount);
998 for (i = 0; i < pts->qcount; i++) {
999 if ( queue[i].status == LPQ_DELETED )
1000 continue;
1001
1002 len += tdb_pack(data.dptr + len, data.dsize - len, "ddddddff",
1003 (uint32)queue[i].job,
1004 (uint32)queue[i].size,
1005 (uint32)queue[i].page_count,
1006 (uint32)queue[i].status,
1007 (uint32)queue[i].priority,
1008 (uint32)queue[i].time,
1009 queue[i].fs_user,
1010 queue[i].fs_file);
1011 }
1012
1013 tdb_store(pdb->tdb, string_tdb_data("INFO/linear_queue_array"), data,
1014 TDB_REPLACE);
1015 SAFE_FREE(data.dptr);
1016 return;
1017}
1018
1019static TDB_DATA get_jobs_changed_data(struct tdb_print_db *pdb)
1020{
1021 TDB_DATA data;
1022
1023 ZERO_STRUCT(data);
1024
1025 data = tdb_fetch(pdb->tdb, string_tdb_data("INFO/jobs_changed"));
1026 if (data.dptr == NULL || data.dsize == 0 || (data.dsize % 4 != 0)) {
1027 SAFE_FREE(data.dptr);
1028 ZERO_STRUCT(data);
1029 }
1030
1031 return data;
1032}
1033
1034static void check_job_changed(const char *sharename, TDB_DATA data, uint32 jobid)
1035{
1036 unsigned int i;
1037 unsigned int job_count = data.dsize / 4;
1038
1039 for (i = 0; i < job_count; i++) {
1040 uint32 ch_jobid;
1041
1042 ch_jobid = IVAL(data.dptr, i*4);
1043 if (ch_jobid == jobid)
1044 remove_from_jobs_changed(sharename, jobid);
1045 }
1046}
1047
1048/****************************************************************************
1049 Check if the print queue has been updated recently enough.
1050****************************************************************************/
1051
1052static bool print_cache_expired(const char *sharename, bool check_pending)
1053{
1054 fstring key;
1055 time_t last_qscan_time, time_now = time(NULL);
1056 struct tdb_print_db *pdb = get_print_db_byname(sharename);
1057 bool result = False;
1058
1059 if (!pdb)
1060 return False;
1061
1062 snprintf(key, sizeof(key), "CACHE/%s", sharename);
1063 last_qscan_time = (time_t)tdb_fetch_int32(pdb->tdb, key);
1064
1065 /*
1066 * Invalidate the queue for 3 reasons.
1067 * (1). last queue scan time == -1.
1068 * (2). Current time - last queue scan time > allowed cache time.
1069 * (3). last queue scan time > current time + MAX_CACHE_VALID_TIME (1 hour by default).
1070 * This last test picks up machines for which the clock has been moved
1071 * forward, an lpq scan done and then the clock moved back. Otherwise
1072 * that last lpq scan would stay around for a loooong loooong time... :-). JRA.
1073 */
1074
1075 if (last_qscan_time == ((time_t)-1)
1076 || (time_now - last_qscan_time) >= lp_lpqcachetime()
1077 || last_qscan_time > (time_now + MAX_CACHE_VALID_TIME))
1078 {
1079 uint32 u;
1080 time_t msg_pending_time;
1081
1082 DEBUG(4, ("print_cache_expired: cache expired for queue %s "
1083 "(last_qscan_time = %d, time now = %d, qcachetime = %d)\n",
1084 sharename, (int)last_qscan_time, (int)time_now,
1085 (int)lp_lpqcachetime() ));
1086
1087 /* check if another smbd has already sent a message to update the
1088 queue. Give the pending message one minute to clear and
1089 then send another message anyways. Make sure to check for
1090 clocks that have been run forward and then back again. */
1091
1092 snprintf(key, sizeof(key), "MSG_PENDING/%s", sharename);
1093
1094 if ( check_pending
1095 && tdb_fetch_uint32( pdb->tdb, key, &u )
1096 && (msg_pending_time=u) > 0
1097 && msg_pending_time <= time_now
1098 && (time_now - msg_pending_time) < 60 )
1099 {
1100 DEBUG(4,("print_cache_expired: message already pending for %s. Accepting cache\n",
1101 sharename));
1102 goto done;
1103 }
1104
1105 result = True;
1106 }
1107
1108done:
1109 release_print_db(pdb);
1110 return result;
1111}
1112
1113/****************************************************************************
1114 main work for updating the lpq cahe for a printer queue
1115****************************************************************************/
1116
1117static void print_queue_update_internal( const char *sharename,
1118 struct printif *current_printif,
1119 char *lpq_command, char *lprm_command )
1120{
1121 int i, qcount;
1122 print_queue_struct *queue = NULL;
1123 print_status_struct status;
1124 print_status_struct old_status;
1125 struct printjob *pjob;
1126 struct traverse_struct tstruct;
1127 TDB_DATA data, key;
1128 TDB_DATA jcdata;
1129 fstring keystr, cachestr;
1130 struct tdb_print_db *pdb = get_print_db_byname(sharename);
1131
1132 if (!pdb) {
1133 return;
1134 }
1135
1136 DEBUG(5,("print_queue_update_internal: printer = %s, type = %d, lpq command = [%s]\n",
1137 sharename, current_printif->type, lpq_command));
1138
1139 /*
1140 * Update the cache time FIRST ! Stops others even
1141 * attempting to get the lock and doing this
1142 * if the lpq takes a long time.
1143 */
1144
1145 slprintf(cachestr, sizeof(cachestr)-1, "CACHE/%s", sharename);
1146 tdb_store_int32(pdb->tdb, cachestr, (int)time(NULL));
1147
1148 /* get the current queue using the appropriate interface */
1149 ZERO_STRUCT(status);
1150
1151 qcount = (*(current_printif->queue_get))(sharename,
1152 current_printif->type,
1153 lpq_command, &queue, &status);
1154
1155 DEBUG(3, ("print_queue_update_internal: %d job%s in queue for %s\n",
1156 qcount, (qcount != 1) ? "s" : "", sharename));
1157
1158 /* Sort the queue by submission time otherwise they are displayed
1159 in hash order. */
1160
1161 qsort(queue, qcount, sizeof(print_queue_struct),
1162 QSORT_CAST(printjob_comp));
1163
1164 /*
1165 any job in the internal database that is marked as spooled
1166 and doesn't exist in the system queue is considered finished
1167 and removed from the database
1168
1169 any job in the system database but not in the internal database
1170 is added as a unix job
1171
1172 fill in any system job numbers as we go
1173 */
1174
1175 jcdata = get_jobs_changed_data(pdb);
1176
1177 for (i=0; i<qcount; i++) {
1178 uint32 jobid = print_parse_jobid(queue[i].fs_file);
1179
1180 if (jobid == (uint32)-1) {
1181 /* assume its a unix print job */
1182 print_unix_job(sharename, &queue[i], jobid);
1183 continue;
1184 }
1185
1186 /* we have an active SMB print job - update its status */
1187 pjob = print_job_find(sharename, jobid);
1188 if (!pjob) {
1189 /* err, somethings wrong. Probably smbd was restarted
1190 with jobs in the queue. All we can do is treat them
1191 like unix jobs. Pity. */
1192 print_unix_job(sharename, &queue[i], jobid);
1193 continue;
1194 }
1195
1196 pjob->sysjob = queue[i].job;
1197
1198 /* don't reset the status on jobs to be deleted */
1199
1200 if ( pjob->status != LPQ_DELETING )
1201 pjob->status = queue[i].status;
1202
1203 pjob_store(sharename, jobid, pjob);
1204
1205 check_job_changed(sharename, jcdata, jobid);
1206 }
1207
1208 SAFE_FREE(jcdata.dptr);
1209
1210 /* now delete any queued entries that don't appear in the
1211 system queue */
1212 tstruct.queue = queue;
1213 tstruct.qcount = qcount;
1214 tstruct.snum = -1;
1215 tstruct.total_jobs = 0;
1216 tstruct.lpq_time = time(NULL);
1217 tstruct.sharename = sharename;
1218 tstruct.lprm_command = lprm_command;
1219 tstruct.print_if = current_printif;
1220
1221 tdb_traverse(pdb->tdb, traverse_fn_delete, (void *)&tstruct);
1222
1223 /* Store the linearised queue, max jobs only. */
1224 store_queue_struct(pdb, &tstruct);
1225
1226 SAFE_FREE(tstruct.queue);
1227
1228 DEBUG(10,("print_queue_update_internal: printer %s INFO/total_jobs = %d\n",
1229 sharename, tstruct.total_jobs ));
1230
1231 tdb_store_int32(pdb->tdb, "INFO/total_jobs", tstruct.total_jobs);
1232
1233 get_queue_status(sharename, &old_status);
1234 if (old_status.qcount != qcount)
1235 DEBUG(10,("print_queue_update_internal: queue status change %d jobs -> %d jobs for printer %s\n",
1236 old_status.qcount, qcount, sharename));
1237
1238 /* store the new queue status structure */
1239 slprintf(keystr, sizeof(keystr)-1, "STATUS/%s", sharename);
1240 key = string_tdb_data(keystr);
1241
1242 status.qcount = qcount;
1243 data.dptr = (uint8 *)&status;
1244 data.dsize = sizeof(status);
1245 tdb_store(pdb->tdb, key, data, TDB_REPLACE);
1246
1247 /*
1248 * Update the cache time again. We want to do this call
1249 * as little as possible...
1250 */
1251
1252 slprintf(keystr, sizeof(keystr)-1, "CACHE/%s", sharename);
1253 tdb_store_int32(pdb->tdb, keystr, (int32)time(NULL));
1254
1255 /* clear the msg pending record for this queue */
1256
1257 snprintf(keystr, sizeof(keystr), "MSG_PENDING/%s", sharename);
1258
1259 if ( !tdb_store_uint32( pdb->tdb, keystr, 0 ) ) {
1260 /* log a message but continue on */
1261
1262 DEBUG(0,("print_queue_update: failed to store MSG_PENDING flag for [%s]!\n",
1263 sharename));
1264 }
1265
1266 release_print_db( pdb );
1267
1268 return;
1269}
1270
1271/****************************************************************************
1272 Update the internal database from the system print queue for a queue.
1273 obtain a lock on the print queue before proceeding (needed when mutiple
1274 smbd processes maytry to update the lpq cache concurrently).
1275****************************************************************************/
1276
1277static void print_queue_update_with_lock( const char *sharename,
1278 struct printif *current_printif,
1279 char *lpq_command, char *lprm_command )
1280{
1281 fstring keystr;
1282 struct tdb_print_db *pdb;
1283
1284 DEBUG(5,("print_queue_update_with_lock: printer share = %s\n", sharename));
1285 pdb = get_print_db_byname(sharename);
1286 if (!pdb)
1287 return;
1288
1289 if ( !print_cache_expired(sharename, False) ) {
1290 DEBUG(5,("print_queue_update_with_lock: print cache for %s is still ok\n", sharename));
1291 release_print_db(pdb);
1292 return;
1293 }
1294
1295 /*
1296 * Check to see if someone else is doing this update.
1297 * This is essentially a mutex on the update.
1298 */
1299
1300 if (get_updating_pid(sharename) != -1) {
1301 release_print_db(pdb);
1302 return;
1303 }
1304
1305 /* Lock the queue for the database update */
1306
1307 slprintf(keystr, sizeof(keystr) - 1, "LOCK/%s", sharename);
1308 /* Only wait 10 seconds for this. */
1309 if (tdb_lock_bystring_with_timeout(pdb->tdb, keystr, 10) == -1) {
1310 DEBUG(0,("print_queue_update_with_lock: Failed to lock printer %s database\n", sharename));
1311 release_print_db(pdb);
1312 return;
1313 }
1314
1315 /*
1316 * Ensure that no one else got in here.
1317 * If the updating pid is still -1 then we are
1318 * the winner.
1319 */
1320
1321 if (get_updating_pid(sharename) != -1) {
1322 /*
1323 * Someone else is doing the update, exit.
1324 */
1325 tdb_unlock_bystring(pdb->tdb, keystr);
1326 release_print_db(pdb);
1327 return;
1328 }
1329
1330 /*
1331 * We're going to do the update ourselves.
1332 */
1333
1334 /* Tell others we're doing the update. */
1335 set_updating_pid(sharename, True);
1336
1337 /*
1338 * Allow others to enter and notice we're doing
1339 * the update.
1340 */
1341
1342 tdb_unlock_bystring(pdb->tdb, keystr);
1343
1344 /* do the main work now */
1345
1346 print_queue_update_internal( sharename, current_printif,
1347 lpq_command, lprm_command );
1348
1349 /* Delete our pid from the db. */
1350 set_updating_pid(sharename, False);
1351 release_print_db(pdb);
1352}
1353
1354/****************************************************************************
1355this is the receive function of the background lpq updater
1356****************************************************************************/
1357static void print_queue_receive(struct messaging_context *msg,
1358 void *private_data,
1359 uint32_t msg_type,
1360 struct server_id server_id,
1361 DATA_BLOB *data)
1362{
1363 fstring sharename;
1364 char *lpqcommand = NULL, *lprmcommand = NULL;
1365 int printing_type;
1366 size_t len;
1367
1368 len = tdb_unpack( (uint8 *)data->data, data->length, "fdPP",
1369 sharename,
1370 &printing_type,
1371 &lpqcommand,
1372 &lprmcommand );
1373
1374 if ( len == -1 ) {
1375 SAFE_FREE(lpqcommand);
1376 SAFE_FREE(lprmcommand);
1377 DEBUG(0,("print_queue_receive: Got invalid print queue update message\n"));
1378 return;
1379 }
1380
1381 print_queue_update_with_lock(sharename,
1382 get_printer_fns_from_type((enum printing_types)printing_type),
1383 lpqcommand, lprmcommand );
1384
1385 SAFE_FREE(lpqcommand);
1386 SAFE_FREE(lprmcommand);
1387 return;
1388}
1389
1390static pid_t background_lpq_updater_pid = -1;
1391
1392/****************************************************************************
1393main thread of the background lpq updater
1394****************************************************************************/
1395void start_background_queue(void)
1396{
1397 /* Use local variables for this as we don't
1398 * need to save the parent side of this, just
1399 * ensure it closes when the process exits.
1400 */
1401 int pause_pipe[2];
1402
1403 DEBUG(3,("start_background_queue: Starting background LPQ thread\n"));
1404
1405#ifndef __OS2__
1406 if (pipe(pause_pipe) == -1) {
1407#else
1408 // on OS2 we have no select() on pipes, so we use socketpair() instead
1409 if (socketpair(AF_UNIX, SOCK_STREAM,0, pause_pipe) < 0) {
1410#endif
1411 DEBUG(5,("start_background_queue: cannot create pipe. %s\n", strerror(errno) ));
1412 exit(1);
1413 }
1414
1415 if (pause_pipe[1] < 0 || pause_pipe[1] >= FD_SETSIZE) {
1416 DEBUG(5,("start_background_queue: pipe fd out of range.\n"));
1417 exit(1);
1418 }
1419
1420 background_lpq_updater_pid = sys_fork();
1421
1422 if (background_lpq_updater_pid == -1) {
1423 DEBUG(5,("start_background_queue: background LPQ thread failed to start. %s\n", strerror(errno) ));
1424 exit(1);
1425 }
1426
1427 if(background_lpq_updater_pid == 0) {
1428 /* Child. */
1429 DEBUG(5,("start_background_queue: background LPQ thread started\n"));
1430
1431 close(pause_pipe[0]);
1432 pause_pipe[0] = -1;
1433
1434 if (!reinit_after_fork(smbd_messaging_context(),
1435 smbd_event_context(), true)) {
1436 DEBUG(0,("reinit_after_fork() failed\n"));
1437 smb_panic("reinit_after_fork() failed");
1438 }
1439
1440 claim_connection( NULL, "smbd lpq backend",
1441 FLAG_MSG_GENERAL|FLAG_MSG_SMBD|FLAG_MSG_PRINT_GENERAL);
1442
1443 if (!locking_init()) {
1444 exit(1);
1445 }
1446
1447 messaging_register(smbd_messaging_context(), NULL,
1448 MSG_PRINTER_UPDATE, print_queue_receive);
1449
1450 DEBUG(5,("start_background_queue: background LPQ thread waiting for messages\n"));
1451 while (1) {
1452 fd_set pause_fds;
1453 int pause_select;
1454
1455 FD_ZERO(&pause_fds);
1456 FD_SET(pause_pipe[1], &pause_fds);
1457 pause_select = sys_select(pause_pipe[1]+1, &pause_fds, NULL, NULL, NULL);
1458 /* If pause_pipe[0] is closed it means the parent smbd
1459 * and children exited or aborted. */
1460 if (pause_select == 1) {
1461 exit_server_cleanly(NULL);
1462 }
1463
1464 /* check for some essential signals first */
1465
1466 if (got_sig_term) {
1467 exit_server_cleanly(NULL);
1468 }
1469
1470 if (reload_after_sighup) {
1471 change_to_root_user();
1472 DEBUG(1,("Reloading services after SIGHUP\n"));
1473 reload_services(False);
1474 reload_after_sighup = 0;
1475 }
1476
1477 /* now check for messages */
1478
1479 DEBUG(10,("start_background_queue: background LPQ thread got a message\n"));
1480 message_dispatch(smbd_messaging_context());
1481
1482 /* process any pending print change notify messages */
1483
1484 print_notify_send_messages(smbd_messaging_context(),
1485 0);
1486 }
1487 }
1488
1489 close(pause_pipe[1]);
1490}
1491
1492/****************************************************************************
1493update the internal database from the system print queue for a queue
1494****************************************************************************/
1495
1496static void print_queue_update(int snum, bool force)
1497{
1498 fstring key;
1499 fstring sharename;
1500 char *lpqcommand = NULL;
1501 char *lprmcommand = NULL;
1502 uint8 *buffer = NULL;
1503 size_t len = 0;
1504 size_t newlen;
1505 struct tdb_print_db *pdb;
1506 int type;
1507 struct printif *current_printif;
1508 TALLOC_CTX *ctx = talloc_tos();
1509
1510 fstrcpy( sharename, lp_const_servicename(snum));
1511
1512 /* don't strip out characters like '$' from the printername */
1513
1514 lpqcommand = talloc_string_sub2(ctx,
1515 lp_lpqcommand(snum),
1516 "%p",
1517 PRINTERNAME(snum),
1518 false, false, false);
1519 if (!lpqcommand) {
1520 return;
1521 }
1522 lpqcommand = talloc_sub_advanced(ctx,
1523 lp_servicename(snum),
1524 current_user_info.unix_name,
1525 "",
1526 current_user.ut.gid,
1527 get_current_username(),
1528 current_user_info.domain,
1529 lpqcommand);
1530 if (!lpqcommand) {
1531 return;
1532 }
1533
1534 lprmcommand = talloc_string_sub2(ctx,
1535 lp_lprmcommand(snum),
1536 "%p",
1537 PRINTERNAME(snum),
1538 false, false, false);
1539 if (!lprmcommand) {
1540 return;
1541 }
1542 lprmcommand = talloc_sub_advanced(ctx,
1543 lp_servicename(snum),
1544 current_user_info.unix_name,
1545 "",
1546 current_user.ut.gid,
1547 get_current_username(),
1548 current_user_info.domain,
1549 lprmcommand);
1550 if (!lprmcommand) {
1551 return;
1552 }
1553
1554 /*
1555 * Make sure that the background queue process exists.
1556 * Otherwise just do the update ourselves
1557 */
1558
1559 if ( force || background_lpq_updater_pid == -1 ) {
1560 DEBUG(4,("print_queue_update: updating queue [%s] myself\n", sharename));
1561 current_printif = get_printer_fns( snum );
1562 print_queue_update_with_lock( sharename, current_printif, lpqcommand, lprmcommand );
1563
1564 return;
1565 }
1566
1567 type = lp_printing(snum);
1568
1569 /* get the length */
1570
1571 len = tdb_pack( NULL, 0, "fdPP",
1572 sharename,
1573 type,
1574 lpqcommand,
1575 lprmcommand );
1576
1577 buffer = SMB_XMALLOC_ARRAY( uint8, len );
1578
1579 /* now pack the buffer */
1580 newlen = tdb_pack( buffer, len, "fdPP",
1581 sharename,
1582 type,
1583 lpqcommand,
1584 lprmcommand );
1585
1586 SMB_ASSERT( newlen == len );
1587
1588 DEBUG(10,("print_queue_update: Sending message -> printer = %s, "
1589 "type = %d, lpq command = [%s] lprm command = [%s]\n",
1590 sharename, type, lpqcommand, lprmcommand ));
1591
1592 /* here we set a msg pending record for other smbd processes
1593 to throttle the number of duplicate print_queue_update msgs
1594 sent. */
1595
1596 pdb = get_print_db_byname(sharename);
1597 if (!pdb) {
1598 SAFE_FREE(buffer);
1599 return;
1600 }
1601
1602 snprintf(key, sizeof(key), "MSG_PENDING/%s", sharename);
1603
1604 if ( !tdb_store_uint32( pdb->tdb, key, time(NULL) ) ) {
1605 /* log a message but continue on */
1606
1607 DEBUG(0,("print_queue_update: failed to store MSG_PENDING flag for [%s]!\n",
1608 sharename));
1609 }
1610
1611 release_print_db( pdb );
1612
1613 /* finally send the message */
1614
1615 messaging_send_buf(smbd_messaging_context(),
1616 pid_to_procid(background_lpq_updater_pid),
1617 MSG_PRINTER_UPDATE, (uint8 *)buffer, len);
1618
1619 SAFE_FREE( buffer );
1620
1621 return;
1622}
1623
1624/****************************************************************************
1625 Create/Update an entry in the print tdb that will allow us to send notify
1626 updates only to interested smbd's.
1627****************************************************************************/
1628
1629bool print_notify_register_pid(int snum)
1630{
1631 TDB_DATA data;
1632 struct tdb_print_db *pdb = NULL;
1633 TDB_CONTEXT *tdb = NULL;
1634 const char *printername;
1635 uint32 mypid = (uint32)sys_getpid();
1636 bool ret = False;
1637 size_t i;
1638
1639 /* if (snum == -1), then the change notify request was
1640 on a print server handle and we need to register on
1641 all print queus */
1642
1643 if (snum == -1)
1644 {
1645 int num_services = lp_numservices();
1646 int idx;
1647
1648 for ( idx=0; idx<num_services; idx++ ) {
1649 if (lp_snum_ok(idx) && lp_print_ok(idx) )
1650 print_notify_register_pid(idx);
1651 }
1652
1653 return True;
1654 }
1655 else /* register for a specific printer */
1656 {
1657 printername = lp_const_servicename(snum);
1658 pdb = get_print_db_byname(printername);
1659 if (!pdb)
1660 return False;
1661 tdb = pdb->tdb;
1662 }
1663
1664 if (tdb_lock_bystring_with_timeout(tdb, NOTIFY_PID_LIST_KEY, 10) == -1) {
1665 DEBUG(0,("print_notify_register_pid: Failed to lock printer %s\n",
1666 printername));
1667 if (pdb)
1668 release_print_db(pdb);
1669 return False;
1670 }
1671
1672 data = get_printer_notify_pid_list( tdb, printername, True );
1673
1674 /* Add ourselves and increase the refcount. */
1675
1676 for (i = 0; i < data.dsize; i += 8) {
1677 if (IVAL(data.dptr,i) == mypid) {
1678 uint32 new_refcount = IVAL(data.dptr, i+4) + 1;
1679 SIVAL(data.dptr, i+4, new_refcount);
1680 break;
1681 }
1682 }
1683
1684 if (i == data.dsize) {
1685 /* We weren't in the list. Realloc. */
1686 data.dptr = (uint8 *)SMB_REALLOC(data.dptr, data.dsize + 8);
1687 if (!data.dptr) {
1688 DEBUG(0,("print_notify_register_pid: Relloc fail for printer %s\n",
1689 printername));
1690 goto done;
1691 }
1692 data.dsize += 8;
1693 SIVAL(data.dptr,data.dsize - 8,mypid);
1694 SIVAL(data.dptr,data.dsize - 4,1); /* Refcount. */
1695 }
1696
1697 /* Store back the record. */
1698 if (tdb_store_bystring(tdb, NOTIFY_PID_LIST_KEY, data, TDB_REPLACE) == -1) {
1699 DEBUG(0,("print_notify_register_pid: Failed to update pid \
1700list for printer %s\n", printername));
1701 goto done;
1702 }
1703
1704 ret = True;
1705
1706 done:
1707
1708 tdb_unlock_bystring(tdb, NOTIFY_PID_LIST_KEY);
1709 if (pdb)
1710 release_print_db(pdb);
1711 SAFE_FREE(data.dptr);
1712 return ret;
1713}
1714
1715/****************************************************************************
1716 Update an entry in the print tdb that will allow us to send notify
1717 updates only to interested smbd's.
1718****************************************************************************/
1719
1720bool print_notify_deregister_pid(int snum)
1721{
1722 TDB_DATA data;
1723 struct tdb_print_db *pdb = NULL;
1724 TDB_CONTEXT *tdb = NULL;
1725 const char *printername;
1726 uint32 mypid = (uint32)sys_getpid();
1727 size_t i;
1728 bool ret = False;
1729
1730 /* if ( snum == -1 ), we are deregister a print server handle
1731 which means to deregister on all print queues */
1732
1733 if (snum == -1)
1734 {
1735 int num_services = lp_numservices();
1736 int idx;
1737
1738 for ( idx=0; idx<num_services; idx++ ) {
1739 if ( lp_snum_ok(idx) && lp_print_ok(idx) )
1740 print_notify_deregister_pid(idx);
1741 }
1742
1743 return True;
1744 }
1745 else /* deregister a specific printer */
1746 {
1747 printername = lp_const_servicename(snum);
1748 pdb = get_print_db_byname(printername);
1749 if (!pdb)
1750 return False;
1751 tdb = pdb->tdb;
1752 }
1753
1754 if (tdb_lock_bystring_with_timeout(tdb, NOTIFY_PID_LIST_KEY, 10) == -1) {
1755 DEBUG(0,("print_notify_register_pid: Failed to lock \
1756printer %s database\n", printername));
1757 if (pdb)
1758 release_print_db(pdb);
1759 return False;
1760 }
1761
1762 data = get_printer_notify_pid_list( tdb, printername, True );
1763
1764 /* Reduce refcount. Remove ourselves if zero. */
1765
1766 for (i = 0; i < data.dsize; ) {
1767 if (IVAL(data.dptr,i) == mypid) {
1768 uint32 refcount = IVAL(data.dptr, i+4);
1769
1770 refcount--;
1771
1772 if (refcount == 0) {
1773 if (data.dsize - i > 8)
1774 memmove( &data.dptr[i], &data.dptr[i+8], data.dsize - i - 8);
1775 data.dsize -= 8;
1776 continue;
1777 }
1778 SIVAL(data.dptr, i+4, refcount);
1779 }
1780
1781 i += 8;
1782 }
1783
1784 if (data.dsize == 0)
1785 SAFE_FREE(data.dptr);
1786
1787 /* Store back the record. */
1788 if (tdb_store_bystring(tdb, NOTIFY_PID_LIST_KEY, data, TDB_REPLACE) == -1) {
1789 DEBUG(0,("print_notify_register_pid: Failed to update pid \
1790list for printer %s\n", printername));
1791 goto done;
1792 }
1793
1794 ret = True;
1795
1796 done:
1797
1798 tdb_unlock_bystring(tdb, NOTIFY_PID_LIST_KEY);
1799 if (pdb)
1800 release_print_db(pdb);
1801 SAFE_FREE(data.dptr);
1802 return ret;
1803}
1804
1805/****************************************************************************
1806 Check if a jobid is valid. It is valid if it exists in the database.
1807****************************************************************************/
1808
1809bool print_job_exists(const char* sharename, uint32 jobid)
1810{
1811 struct tdb_print_db *pdb = get_print_db_byname(sharename);
1812 bool ret;
1813 uint32_t tmp;
1814
1815 if (!pdb)
1816 return False;
1817 ret = tdb_exists(pdb->tdb, print_key(jobid, &tmp));
1818 release_print_db(pdb);
1819 return ret;
1820}
1821
1822/****************************************************************************
1823 Give the fd used for a jobid.
1824****************************************************************************/
1825
1826int print_job_fd(const char* sharename, uint32 jobid)
1827{
1828 struct printjob *pjob = print_job_find(sharename, jobid);
1829 if (!pjob)
1830 return -1;
1831 /* don't allow another process to get this info - it is meaningless */
1832 if (pjob->pid != sys_getpid())
1833 return -1;
1834 return pjob->fd;
1835}
1836
1837/****************************************************************************
1838 Give the filename used for a jobid.
1839 Only valid for the process doing the spooling and when the job
1840 has not been spooled.
1841****************************************************************************/
1842
1843char *print_job_fname(const char* sharename, uint32 jobid)
1844{
1845 struct printjob *pjob = print_job_find(sharename, jobid);
1846 if (!pjob || pjob->spooled || pjob->pid != sys_getpid())
1847 return NULL;
1848 return pjob->filename;
1849}
1850
1851
1852/****************************************************************************
1853 Give the filename used for a jobid.
1854 Only valid for the process doing the spooling and when the job
1855 has not been spooled.
1856****************************************************************************/
1857
1858NT_DEVICEMODE *print_job_devmode(const char* sharename, uint32 jobid)
1859{
1860 struct printjob *pjob = print_job_find(sharename, jobid);
1861
1862 if ( !pjob )
1863 return NULL;
1864
1865 return pjob->nt_devmode;
1866}
1867
1868/****************************************************************************
1869 Set the place in the queue for a job.
1870****************************************************************************/
1871
1872bool print_job_set_place(const char *sharename, uint32 jobid, int place)
1873{
1874 DEBUG(2,("print_job_set_place not implemented yet\n"));
1875 return False;
1876}
1877
1878/****************************************************************************
1879 Set the name of a job. Only possible for owner.
1880****************************************************************************/
1881
1882bool print_job_set_name(const char *sharename, uint32 jobid, char *name)
1883{
1884 struct printjob *pjob;
1885
1886 pjob = print_job_find(sharename, jobid);
1887 if (!pjob || pjob->pid != sys_getpid())
1888 return False;
1889
1890 fstrcpy(pjob->jobname, name);
1891 return pjob_store(sharename, jobid, pjob);
1892}
1893
1894/***************************************************************************
1895 Remove a jobid from the 'jobs changed' list.
1896***************************************************************************/
1897
1898static bool remove_from_jobs_changed(const char* sharename, uint32 jobid)
1899{
1900 struct tdb_print_db *pdb = get_print_db_byname(sharename);
1901 TDB_DATA data, key;
1902 size_t job_count, i;
1903 bool ret = False;
1904 bool gotlock = False;
1905
1906 if (!pdb) {
1907 return False;
1908 }
1909
1910 ZERO_STRUCT(data);
1911
1912 key = string_tdb_data("INFO/jobs_changed");
1913
1914 if (tdb_chainlock_with_timeout(pdb->tdb, key, 5) == -1)
1915 goto out;
1916
1917 gotlock = True;
1918
1919 data = tdb_fetch(pdb->tdb, key);
1920
1921 if (data.dptr == NULL || data.dsize == 0 || (data.dsize % 4 != 0))
1922 goto out;
1923
1924 job_count = data.dsize / 4;
1925 for (i = 0; i < job_count; i++) {
1926 uint32 ch_jobid;
1927
1928 ch_jobid = IVAL(data.dptr, i*4);
1929 if (ch_jobid == jobid) {
1930 if (i < job_count -1 )
1931 memmove(data.dptr + (i*4), data.dptr + (i*4) + 4, (job_count - i - 1)*4 );
1932 data.dsize -= 4;
1933 if (tdb_store(pdb->tdb, key, data, TDB_REPLACE) == -1)
1934 goto out;
1935 break;
1936 }
1937 }
1938
1939 ret = True;
1940 out:
1941
1942 if (gotlock)
1943 tdb_chainunlock(pdb->tdb, key);
1944 SAFE_FREE(data.dptr);
1945 release_print_db(pdb);
1946 if (ret)
1947 DEBUG(10,("remove_from_jobs_changed: removed jobid %u\n", (unsigned int)jobid ));
1948 else
1949 DEBUG(10,("remove_from_jobs_changed: Failed to remove jobid %u\n", (unsigned int)jobid ));
1950 return ret;
1951}
1952
1953/****************************************************************************
1954 Delete a print job - don't update queue.
1955****************************************************************************/
1956
1957static bool print_job_delete1(int snum, uint32 jobid)
1958{
1959 const char* sharename = lp_const_servicename(snum);
1960 struct printjob *pjob = print_job_find(sharename, jobid);
1961 int result = 0;
1962 struct printif *current_printif = get_printer_fns( snum );
1963
1964 if (!pjob)
1965 return False;
1966
1967 /*
1968 * If already deleting just return.
1969 */
1970
1971 if (pjob->status == LPQ_DELETING)
1972 return True;
1973
1974 /* Hrm - we need to be able to cope with deleting a job before it
1975 has reached the spooler. Just mark it as LPQ_DELETING and
1976 let the print_queue_update() code rmeove the record */
1977
1978
1979 if (pjob->sysjob == -1) {
1980 DEBUG(5, ("attempt to delete job %u not seen by lpr\n", (unsigned int)jobid));
1981 }
1982
1983 /* Set the tdb entry to be deleting. */
1984
1985 pjob->status = LPQ_DELETING;
1986 pjob_store(sharename, jobid, pjob);
1987
1988 if (pjob->spooled && pjob->sysjob != -1)
1989 {
1990 result = (*(current_printif->job_delete))(
1991 PRINTERNAME(snum),
1992 lp_lprmcommand(snum),
1993 pjob);
1994
1995 /* Delete the tdb entry if the delete succeeded or the job hasn't
1996 been spooled. */
1997
1998 if (result == 0) {
1999 struct tdb_print_db *pdb = get_print_db_byname(sharename);
2000 int njobs = 1;
2001
2002 if (!pdb)
2003 return False;
2004 pjob_delete(sharename, jobid);
2005 /* Ensure we keep a rough count of the number of total jobs... */
2006 tdb_change_int32_atomic(pdb->tdb, "INFO/total_jobs", &njobs, -1);
2007 release_print_db(pdb);
2008 }
2009 }
2010
2011 remove_from_jobs_changed( sharename, jobid );
2012
2013 return (result == 0);
2014}
2015
2016/****************************************************************************
2017 Return true if the current user owns the print job.
2018****************************************************************************/
2019
2020static bool is_owner(struct auth_serversupplied_info *server_info,
2021 const char *servicename,
2022 uint32 jobid)
2023{
2024 struct printjob *pjob = print_job_find(servicename, jobid);
2025
2026 if (!pjob || !server_info)
2027 return False;
2028
2029 return strequal(pjob->user, server_info->sanitized_username);
2030}
2031
2032/****************************************************************************
2033 Delete a print job.
2034****************************************************************************/
2035
2036bool print_job_delete(struct auth_serversupplied_info *server_info, int snum,
2037 uint32 jobid, WERROR *errcode)
2038{
2039 const char* sharename = lp_const_servicename( snum );
2040 struct printjob *pjob;
2041 bool owner;
2042 char *fname;
2043
2044 *errcode = WERR_OK;
2045
2046 owner = is_owner(server_info, lp_const_servicename(snum), jobid);
2047
2048 /* Check access against security descriptor or whether the user
2049 owns their job. */
2050
2051 if (!owner &&
2052 !print_access_check(server_info, snum, JOB_ACCESS_ADMINISTER)) {
2053 DEBUG(3, ("delete denied by security descriptor\n"));
2054 *errcode = WERR_ACCESS_DENIED;
2055
2056 /* BEGIN_ADMIN_LOG */
2057 sys_adminlog( LOG_ERR,
2058 "Permission denied-- user not allowed to delete, \
2059pause, or resume print job. User name: %s. Printer name: %s.",
2060 uidtoname(server_info->utok.uid),
2061 PRINTERNAME(snum) );
2062 /* END_ADMIN_LOG */
2063
2064 return False;
2065 }
2066
2067 /*
2068 * get the spooled filename of the print job
2069 * if this works, then the file has not been spooled
2070 * to the underlying print system. Just delete the
2071 * spool file & return.
2072 */
2073
2074 if ( (fname = print_job_fname( sharename, jobid )) != NULL )
2075 {
2076 /* remove the spool file */
2077 DEBUG(10,("print_job_delete: Removing spool file [%s]\n", fname ));
2078 if ( unlink( fname ) == -1 ) {
2079 *errcode = map_werror_from_unix(errno);
2080 return False;
2081 }
2082 }
2083
2084 if (!print_job_delete1(snum, jobid)) {
2085 *errcode = WERR_ACCESS_DENIED;
2086 return False;
2087 }
2088
2089 /* force update the database and say the delete failed if the
2090 job still exists */
2091
2092 print_queue_update(snum, True);
2093
2094 pjob = print_job_find(sharename, jobid);
2095 if ( pjob && (pjob->status != LPQ_DELETING) )
2096 *errcode = WERR_ACCESS_DENIED;
2097
2098 return (pjob == NULL );
2099}
2100
2101/****************************************************************************
2102 Pause a job.
2103****************************************************************************/
2104
2105bool print_job_pause(struct auth_serversupplied_info *server_info, int snum,
2106 uint32 jobid, WERROR *errcode)
2107{
2108 const char* sharename = lp_const_servicename(snum);
2109 struct printjob *pjob;
2110 int ret = -1;
2111 struct printif *current_printif = get_printer_fns( snum );
2112
2113 pjob = print_job_find(sharename, jobid);
2114
2115 if (!pjob || !server_info) {
2116 DEBUG(10, ("print_job_pause: no pjob or user for jobid %u\n",
2117 (unsigned int)jobid ));
2118 return False;
2119 }
2120
2121 if (!pjob->spooled || pjob->sysjob == -1) {
2122 DEBUG(10, ("print_job_pause: not spooled or bad sysjob = %d for jobid %u\n",
2123 (int)pjob->sysjob, (unsigned int)jobid ));
2124 return False;
2125 }
2126
2127 if (!is_owner(server_info, lp_const_servicename(snum), jobid) &&
2128 !print_access_check(server_info, snum, JOB_ACCESS_ADMINISTER)) {
2129 DEBUG(3, ("pause denied by security descriptor\n"));
2130
2131 /* BEGIN_ADMIN_LOG */
2132 sys_adminlog( LOG_ERR,
2133 "Permission denied-- user not allowed to delete, \
2134pause, or resume print job. User name: %s. Printer name: %s.",
2135 uidtoname(server_info->utok.uid),
2136 PRINTERNAME(snum) );
2137 /* END_ADMIN_LOG */
2138
2139 *errcode = WERR_ACCESS_DENIED;
2140 return False;
2141 }
2142
2143 /* need to pause the spooled entry */
2144 ret = (*(current_printif->job_pause))(snum, pjob);
2145
2146 if (ret != 0) {
2147 *errcode = WERR_INVALID_PARAM;
2148 return False;
2149 }
2150
2151 /* force update the database */
2152 print_cache_flush(lp_const_servicename(snum));
2153
2154 /* Send a printer notify message */
2155
2156 notify_job_status(sharename, jobid, JOB_STATUS_PAUSED);
2157
2158 /* how do we tell if this succeeded? */
2159
2160 return True;
2161}
2162
2163/****************************************************************************
2164 Resume a job.
2165****************************************************************************/
2166
2167bool print_job_resume(struct auth_serversupplied_info *server_info, int snum,
2168 uint32 jobid, WERROR *errcode)
2169{
2170 const char *sharename = lp_const_servicename(snum);
2171 struct printjob *pjob;
2172 int ret;
2173 struct printif *current_printif = get_printer_fns( snum );
2174
2175 pjob = print_job_find(sharename, jobid);
2176
2177 if (!pjob || !server_info) {
2178 DEBUG(10, ("print_job_resume: no pjob or user for jobid %u\n",
2179 (unsigned int)jobid ));
2180 return False;
2181 }
2182
2183 if (!pjob->spooled || pjob->sysjob == -1) {
2184 DEBUG(10, ("print_job_resume: not spooled or bad sysjob = %d for jobid %u\n",
2185 (int)pjob->sysjob, (unsigned int)jobid ));
2186 return False;
2187 }
2188
2189 if (!is_owner(server_info, lp_const_servicename(snum), jobid) &&
2190 !print_access_check(server_info, snum, JOB_ACCESS_ADMINISTER)) {
2191 DEBUG(3, ("resume denied by security descriptor\n"));
2192 *errcode = WERR_ACCESS_DENIED;
2193
2194 /* BEGIN_ADMIN_LOG */
2195 sys_adminlog( LOG_ERR,
2196 "Permission denied-- user not allowed to delete, \
2197pause, or resume print job. User name: %s. Printer name: %s.",
2198 uidtoname(server_info->utok.uid),
2199 PRINTERNAME(snum) );
2200 /* END_ADMIN_LOG */
2201 return False;
2202 }
2203
2204 ret = (*(current_printif->job_resume))(snum, pjob);
2205
2206 if (ret != 0) {
2207 *errcode = WERR_INVALID_PARAM;
2208 return False;
2209 }
2210
2211 /* force update the database */
2212 print_cache_flush(lp_const_servicename(snum));
2213
2214 /* Send a printer notify message */
2215
2216 notify_job_status(sharename, jobid, JOB_STATUS_QUEUED);
2217
2218 return True;
2219}
2220
2221/****************************************************************************
2222 Write to a print file.
2223****************************************************************************/
2224
2225ssize_t print_job_write(int snum, uint32 jobid, const char *buf, SMB_OFF_T pos, size_t size)
2226{
2227 const char* sharename = lp_const_servicename(snum);
2228 int return_code;
2229 struct printjob *pjob;
2230
2231 pjob = print_job_find(sharename, jobid);
2232
2233 if (!pjob)
2234 return -1;
2235 /* don't allow another process to get this info - it is meaningless */
2236 if (pjob->pid != sys_getpid())
2237 return -1;
2238
2239 return_code = write_data_at_offset(pjob->fd, buf, size, pos);
2240
2241 if (return_code>0) {
2242 pjob->size += size;
2243 pjob_store(sharename, jobid, pjob);
2244 }
2245 return return_code;
2246}
2247
2248/****************************************************************************
2249 Get the queue status - do not update if db is out of date.
2250****************************************************************************/
2251
2252static int get_queue_status(const char* sharename, print_status_struct *status)
2253{
2254 fstring keystr;
2255 TDB_DATA data;
2256 struct tdb_print_db *pdb = get_print_db_byname(sharename);
2257 int len;
2258
2259 if (status) {
2260 ZERO_STRUCTP(status);
2261 }
2262
2263 if (!pdb)
2264 return 0;
2265
2266 if (status) {
2267 fstr_sprintf(keystr, "STATUS/%s", sharename);
2268 data = tdb_fetch(pdb->tdb, string_tdb_data(keystr));
2269 if (data.dptr) {
2270 if (data.dsize == sizeof(print_status_struct))
2271 /* this memcpy is ok since the status struct was
2272 not packed before storing it in the tdb */
2273 memcpy(status, data.dptr, sizeof(print_status_struct));
2274 SAFE_FREE(data.dptr);
2275 }
2276 }
2277 len = tdb_fetch_int32(pdb->tdb, "INFO/total_jobs");
2278 release_print_db(pdb);
2279 return (len == -1 ? 0 : len);
2280}
2281
2282/****************************************************************************
2283 Determine the number of jobs in a queue.
2284****************************************************************************/
2285
2286int print_queue_length(int snum, print_status_struct *pstatus)
2287{
2288 const char* sharename = lp_const_servicename( snum );
2289 print_status_struct status;
2290 int len;
2291
2292 ZERO_STRUCT( status );
2293
2294 /* make sure the database is up to date */
2295 if (print_cache_expired(lp_const_servicename(snum), True))
2296 print_queue_update(snum, False);
2297
2298 /* also fetch the queue status */
2299 memset(&status, 0, sizeof(status));
2300 len = get_queue_status(sharename, &status);
2301
2302 if (pstatus)
2303 *pstatus = status;
2304
2305 return len;
2306}
2307
2308/***************************************************************************
2309 Allocate a jobid. Hold the lock for as short a time as possible.
2310***************************************************************************/
2311
2312static bool allocate_print_jobid(struct tdb_print_db *pdb, int snum, const char *sharename, uint32 *pjobid)
2313{
2314 int i;
2315 uint32 jobid;
2316
2317 *pjobid = (uint32)-1;
2318
2319 for (i = 0; i < 3; i++) {
2320 /* Lock the database - only wait 20 seconds. */
2321 if (tdb_lock_bystring_with_timeout(pdb->tdb, "INFO/nextjob", 20) == -1) {
2322 DEBUG(0,("allocate_print_jobid: failed to lock printing database %s\n", sharename));
2323 return False;
2324 }
2325
2326 if (!tdb_fetch_uint32(pdb->tdb, "INFO/nextjob", &jobid)) {
2327 if (tdb_error(pdb->tdb) != TDB_ERR_NOEXIST) {
2328 DEBUG(0, ("allocate_print_jobid: failed to fetch INFO/nextjob for print queue %s\n",
2329 sharename));
2330 return False;
2331 }
2332 jobid = 0;
2333 }
2334
2335 jobid = NEXT_JOBID(jobid);
2336
2337 if (tdb_store_int32(pdb->tdb, "INFO/nextjob", jobid)==-1) {
2338 DEBUG(3, ("allocate_print_jobid: failed to store INFO/nextjob.\n"));
2339 tdb_unlock_bystring(pdb->tdb, "INFO/nextjob");
2340 return False;
2341 }
2342
2343 /* We've finished with the INFO/nextjob lock. */
2344 tdb_unlock_bystring(pdb->tdb, "INFO/nextjob");
2345
2346 if (!print_job_exists(sharename, jobid))
2347 break;
2348 }
2349
2350 if (i > 2) {
2351 DEBUG(0, ("allocate_print_jobid: failed to allocate a print job for queue %s\n",
2352 sharename));
2353 /* Probably full... */
2354 errno = ENOSPC;
2355 return False;
2356 }
2357
2358 /* Store a dummy placeholder. */
2359 {
2360 uint32_t tmp;
2361 TDB_DATA dum;
2362 dum.dptr = NULL;
2363 dum.dsize = 0;
2364 if (tdb_store(pdb->tdb, print_key(jobid, &tmp), dum,
2365 TDB_INSERT) == -1) {
2366 DEBUG(3, ("allocate_print_jobid: jobid (%d) failed to store placeholder.\n",
2367 jobid ));
2368 return False;
2369 }
2370 }
2371
2372 *pjobid = jobid;
2373 return True;
2374}
2375
2376/***************************************************************************
2377 Append a jobid to the 'jobs changed' list.
2378***************************************************************************/
2379
2380static bool add_to_jobs_changed(struct tdb_print_db *pdb, uint32 jobid)
2381{
2382 TDB_DATA data;
2383 uint32 store_jobid;
2384
2385 SIVAL(&store_jobid, 0, jobid);
2386 data.dptr = (uint8 *)&store_jobid;
2387 data.dsize = 4;
2388
2389 DEBUG(10,("add_to_jobs_changed: Added jobid %u\n", (unsigned int)jobid ));
2390
2391 return (tdb_append(pdb->tdb, string_tdb_data("INFO/jobs_changed"),
2392 data) == 0);
2393}
2394
2395/***************************************************************************
2396 Start spooling a job - return the jobid.
2397***************************************************************************/
2398
2399uint32 print_job_start(struct auth_serversupplied_info *server_info, int snum,
2400 char *jobname, NT_DEVICEMODE *nt_devmode )
2401{
2402 uint32 jobid;
2403 char *path;
2404 struct printjob pjob;
2405 const char *sharename = lp_const_servicename(snum);
2406 struct tdb_print_db *pdb = get_print_db_byname(sharename);
2407 int njobs;
2408
2409 errno = 0;
2410
2411 if (!pdb)
2412 return (uint32)-1;
2413
2414 if (!print_access_check(server_info, snum, PRINTER_ACCESS_USE)) {
2415 DEBUG(3, ("print_job_start: job start denied by security descriptor\n"));
2416 release_print_db(pdb);
2417 return (uint32)-1;
2418 }
2419
2420 if (!print_time_access_check(lp_servicename(snum))) {
2421 DEBUG(3, ("print_job_start: job start denied by time check\n"));
2422 release_print_db(pdb);
2423 return (uint32)-1;
2424 }
2425
2426 path = lp_pathname(snum);
2427
2428 /* see if we have sufficient disk space */
2429 if (lp_minprintspace(snum)) {
2430 SMB_BIG_UINT dspace, dsize;
2431 if (sys_fsusage(path, &dspace, &dsize) == 0 &&
2432 dspace < 2*(SMB_BIG_UINT)lp_minprintspace(snum)) {
2433 DEBUG(3, ("print_job_start: disk space check failed.\n"));
2434 release_print_db(pdb);
2435 errno = ENOSPC;
2436 return (uint32)-1;
2437 }
2438 }
2439
2440 /* for autoloaded printers, check that the printcap entry still exists */
2441 if (lp_autoloaded(snum) && !pcap_printername_ok(lp_const_servicename(snum))) {
2442 DEBUG(3, ("print_job_start: printer name %s check failed.\n", lp_const_servicename(snum) ));
2443 release_print_db(pdb);
2444 errno = ENOENT;
2445 return (uint32)-1;
2446 }
2447
2448 /* Insure the maximum queue size is not violated */
2449 if ((njobs = print_queue_length(snum,NULL)) > lp_maxprintjobs(snum)) {
2450 DEBUG(3, ("print_job_start: Queue %s number of jobs (%d) larger than max printjobs per queue (%d).\n",
2451 sharename, njobs, lp_maxprintjobs(snum) ));
2452 release_print_db(pdb);
2453 errno = ENOSPC;
2454 return (uint32)-1;
2455 }
2456
2457 DEBUG(10,("print_job_start: Queue %s number of jobs (%d), max printjobs = %d\n",
2458 sharename, njobs, lp_maxprintjobs(snum) ));
2459
2460 if (!allocate_print_jobid(pdb, snum, sharename, &jobid))
2461 goto fail;
2462
2463 /* create the database entry */
2464
2465 ZERO_STRUCT(pjob);
2466
2467 pjob.pid = sys_getpid();
2468 pjob.sysjob = -1;
2469 pjob.fd = -1;
2470 pjob.starttime = time(NULL);
2471 pjob.status = LPQ_SPOOLING;
2472 pjob.size = 0;
2473 pjob.spooled = False;
2474 pjob.smbjob = True;
2475 pjob.nt_devmode = nt_devmode;
2476
2477 fstrcpy(pjob.jobname, jobname);
2478
2479 fstrcpy(pjob.user, lp_printjob_username(snum));
2480 standard_sub_advanced(sharename, server_info->sanitized_username,
2481 path, server_info->utok.gid,
2482 server_info->sanitized_username,
2483 pdb_get_domain(server_info->sam_account),
2484 pjob.user, sizeof(pjob.user)-1);
2485 /* ensure NULL termination */
2486 pjob.user[sizeof(pjob.user)-1] = '\0';
2487
2488 fstrcpy(pjob.queuename, lp_const_servicename(snum));
2489
2490 /* we have a job entry - now create the spool file */
2491 slprintf(pjob.filename, sizeof(pjob.filename)-1, "%s/%s%.8u.XXXXXX",
2492 path, PRINT_SPOOL_PREFIX, (unsigned int)jobid);
2493 pjob.fd = smb_mkstemp(pjob.filename);
2494
2495 if (pjob.fd == -1) {
2496 if (errno == EACCES) {
2497 /* Common setup error, force a report. */
2498 DEBUG(0, ("print_job_start: insufficient permissions \
2499to open spool file %s.\n", pjob.filename));
2500 } else {
2501 /* Normal case, report at level 3 and above. */
2502 DEBUG(3, ("print_job_start: can't open spool file %s,\n", pjob.filename));
2503 DEBUGADD(3, ("errno = %d (%s).\n", errno, strerror(errno)));
2504 }
2505 goto fail;
2506 }
2507
2508 pjob_store(sharename, jobid, &pjob);
2509
2510 /* Update the 'jobs changed' entry used by print_queue_status. */
2511 add_to_jobs_changed(pdb, jobid);
2512
2513 /* Ensure we keep a rough count of the number of total jobs... */
2514 tdb_change_int32_atomic(pdb->tdb, "INFO/total_jobs", &njobs, 1);
2515
2516 release_print_db(pdb);
2517
2518 return jobid;
2519
2520 fail:
2521 if (jobid != -1)
2522 pjob_delete(sharename, jobid);
2523
2524 release_print_db(pdb);
2525
2526 DEBUG(3, ("print_job_start: returning fail. Error = %s\n", strerror(errno) ));
2527 return (uint32)-1;
2528}
2529
2530/****************************************************************************
2531 Update the number of pages spooled to jobid
2532****************************************************************************/
2533
2534void print_job_endpage(int snum, uint32 jobid)
2535{
2536 const char* sharename = lp_const_servicename(snum);
2537 struct printjob *pjob;
2538
2539 pjob = print_job_find(sharename, jobid);
2540 if (!pjob)
2541 return;
2542 /* don't allow another process to get this info - it is meaningless */
2543 if (pjob->pid != sys_getpid())
2544 return;
2545
2546 pjob->page_count++;
2547 pjob_store(sharename, jobid, pjob);
2548}
2549
2550/****************************************************************************
2551 Print a file - called on closing the file. This spools the job.
2552 If normal close is false then we're tearing down the jobs - treat as an
2553 error.
2554****************************************************************************/
2555
2556bool print_job_end(int snum, uint32 jobid, enum file_close_type close_type)
2557{
2558 const char* sharename = lp_const_servicename(snum);
2559 struct printjob *pjob;
2560 int ret;
2561 SMB_STRUCT_STAT sbuf;
2562 struct printif *current_printif = get_printer_fns( snum );
2563
2564 pjob = print_job_find(sharename, jobid);
2565
2566 if (!pjob)
2567 return False;
2568
2569 if (pjob->spooled || pjob->pid != sys_getpid())
2570 return False;
2571
2572 if ((close_type == NORMAL_CLOSE || close_type == SHUTDOWN_CLOSE) &&
2573 (sys_fstat(pjob->fd, &sbuf) == 0)) {
2574 pjob->size = sbuf.st_size;
2575 close(pjob->fd);
2576 pjob->fd = -1;
2577 } else {
2578
2579 /*
2580 * Not a normal close or we couldn't stat the job file,
2581 * so something has gone wrong. Cleanup.
2582 */
2583 close(pjob->fd);
2584 pjob->fd = -1;
2585 DEBUG(3,("print_job_end: failed to stat file for jobid %d\n", jobid ));
2586 goto fail;
2587 }
2588
2589 /* Technically, this is not quite right. If the printer has a separator
2590 * page turned on, the NT spooler prints the separator page even if the
2591 * print job is 0 bytes. 010215 JRR */
2592 if (pjob->size == 0 || pjob->status == LPQ_DELETING) {
2593 /* don't bother spooling empty files or something being deleted. */
2594 DEBUG(5,("print_job_end: canceling spool of %s (%s)\n",
2595 pjob->filename, pjob->size ? "deleted" : "zero length" ));
2596 unlink(pjob->filename);
2597 pjob_delete(sharename, jobid);
2598 return True;
2599 }
2600
2601 pjob->smbjob = jobid;
2602
2603 ret = (*(current_printif->job_submit))(snum, pjob);
2604
2605 if (ret)
2606 goto fail;
2607
2608 /* The print job has been successfully handed over to the back-end */
2609
2610 pjob->spooled = True;
2611 pjob->status = LPQ_QUEUED;
2612 pjob_store(sharename, jobid, pjob);
2613
2614 /* make sure the database is up to date */
2615 if (print_cache_expired(lp_const_servicename(snum), True))
2616 print_queue_update(snum, False);
2617
2618 return True;
2619
2620fail:
2621
2622 /* The print job was not successfully started. Cleanup */
2623 /* Still need to add proper error return propagation! 010122:JRR */
2624 unlink(pjob->filename);
2625 pjob_delete(sharename, jobid);
2626 return False;
2627}
2628
2629/****************************************************************************
2630 Get a snapshot of jobs in the system without traversing.
2631****************************************************************************/
2632
2633static bool get_stored_queue_info(struct tdb_print_db *pdb, int snum, int *pcount, print_queue_struct **ppqueue)
2634{
2635 TDB_DATA data, cgdata;
2636 print_queue_struct *queue = NULL;
2637 uint32 qcount = 0;
2638 uint32 extra_count = 0;
2639 int total_count = 0;
2640 size_t len = 0;
2641 uint32 i;
2642 int max_reported_jobs = lp_max_reported_jobs(snum);
2643 bool ret = False;
2644 const char* sharename = lp_servicename(snum);
2645
2646 /* make sure the database is up to date */
2647 if (print_cache_expired(lp_const_servicename(snum), True))
2648 print_queue_update(snum, False);
2649
2650 *pcount = 0;
2651 *ppqueue = NULL;
2652
2653 ZERO_STRUCT(data);
2654 ZERO_STRUCT(cgdata);
2655
2656 /* Get the stored queue data. */
2657 data = tdb_fetch(pdb->tdb, string_tdb_data("INFO/linear_queue_array"));
2658
2659 if (data.dptr && data.dsize >= sizeof(qcount))
2660 len += tdb_unpack(data.dptr + len, data.dsize - len, "d", &qcount);
2661
2662 /* Get the changed jobs list. */
2663 cgdata = tdb_fetch(pdb->tdb, string_tdb_data("INFO/jobs_changed"));
2664 if (cgdata.dptr != NULL && (cgdata.dsize % 4 == 0))
2665 extra_count = cgdata.dsize/4;
2666
2667 DEBUG(5,("get_stored_queue_info: qcount = %u, extra_count = %u\n", (unsigned int)qcount, (unsigned int)extra_count));
2668
2669 /* Allocate the queue size. */
2670 if (qcount == 0 && extra_count == 0)
2671 goto out;
2672
2673 if ((queue = SMB_MALLOC_ARRAY(print_queue_struct, qcount + extra_count)) == NULL)
2674 goto out;
2675
2676 /* Retrieve the linearised queue data. */
2677
2678 for( i = 0; i < qcount; i++) {
2679 uint32 qjob, qsize, qpage_count, qstatus, qpriority, qtime;
2680 len += tdb_unpack(data.dptr + len, data.dsize - len, "ddddddff",
2681 &qjob,
2682 &qsize,
2683 &qpage_count,
2684 &qstatus,
2685 &qpriority,
2686 &qtime,
2687 queue[i].fs_user,
2688 queue[i].fs_file);
2689 queue[i].job = qjob;
2690 queue[i].size = qsize;
2691 queue[i].page_count = qpage_count;
2692 queue[i].status = qstatus;
2693 queue[i].priority = qpriority;
2694 queue[i].time = qtime;
2695 }
2696
2697 total_count = qcount;
2698
2699 /* Add in the changed jobids. */
2700 for( i = 0; i < extra_count; i++) {
2701 uint32 jobid;
2702 struct printjob *pjob;
2703
2704 jobid = IVAL(cgdata.dptr, i*4);
2705 DEBUG(5,("get_stored_queue_info: changed job = %u\n", (unsigned int)jobid));
2706 pjob = print_job_find(lp_const_servicename(snum), jobid);
2707 if (!pjob) {
2708 DEBUG(5,("get_stored_queue_info: failed to find changed job = %u\n", (unsigned int)jobid));
2709 remove_from_jobs_changed(sharename, jobid);
2710 continue;
2711 }
2712
2713 queue[total_count].job = jobid;
2714 queue[total_count].size = pjob->size;
2715 queue[total_count].page_count = pjob->page_count;
2716 queue[total_count].status = pjob->status;
2717 queue[total_count].priority = 1;
2718 queue[total_count].time = pjob->starttime;
2719 fstrcpy(queue[total_count].fs_user, pjob->user);
2720 fstrcpy(queue[total_count].fs_file, pjob->jobname);
2721 total_count++;
2722 }
2723
2724 /* Sort the queue by submission time otherwise they are displayed
2725 in hash order. */
2726
2727 qsort(queue, total_count, sizeof(print_queue_struct), QSORT_CAST(printjob_comp));
2728
2729 DEBUG(5,("get_stored_queue_info: total_count = %u\n", (unsigned int)total_count));
2730
2731 if (max_reported_jobs && total_count > max_reported_jobs)
2732 total_count = max_reported_jobs;
2733
2734 *ppqueue = queue;
2735 *pcount = total_count;
2736
2737 ret = True;
2738
2739 out:
2740
2741 SAFE_FREE(data.dptr);
2742 SAFE_FREE(cgdata.dptr);
2743 return ret;
2744}
2745
2746/****************************************************************************
2747 Get a printer queue listing.
2748 set queue = NULL and status = NULL if you just want to update the cache
2749****************************************************************************/
2750
2751int print_queue_status(int snum,
2752 print_queue_struct **ppqueue,
2753 print_status_struct *status)
2754{
2755 fstring keystr;
2756 TDB_DATA data, key;
2757 const char *sharename;
2758 struct tdb_print_db *pdb;
2759 int count = 0;
2760
2761 /* make sure the database is up to date */
2762
2763 if (print_cache_expired(lp_const_servicename(snum), True))
2764 print_queue_update(snum, False);
2765
2766 /* return if we are done */
2767 if ( !ppqueue || !status )
2768 return 0;
2769
2770 *ppqueue = NULL;
2771 sharename = lp_const_servicename(snum);
2772 pdb = get_print_db_byname(sharename);
2773
2774 if (!pdb)
2775 return 0;
2776
2777 /*
2778 * Fetch the queue status. We must do this first, as there may
2779 * be no jobs in the queue.
2780 */
2781
2782 ZERO_STRUCTP(status);
2783 slprintf(keystr, sizeof(keystr)-1, "STATUS/%s", sharename);
2784 key = string_tdb_data(keystr);
2785
2786 data = tdb_fetch(pdb->tdb, key);
2787 if (data.dptr) {
2788 if (data.dsize == sizeof(*status)) {
2789 /* this memcpy is ok since the status struct was
2790 not packed before storing it in the tdb */
2791 memcpy(status, data.dptr, sizeof(*status));
2792 }
2793 SAFE_FREE(data.dptr);
2794 }
2795
2796 /*
2797 * Now, fetch the print queue information. We first count the number
2798 * of entries, and then only retrieve the queue if necessary.
2799 */
2800
2801 if (!get_stored_queue_info(pdb, snum, &count, ppqueue)) {
2802 release_print_db(pdb);
2803 return 0;
2804 }
2805
2806 release_print_db(pdb);
2807 return count;
2808}
2809
2810/****************************************************************************
2811 Pause a queue.
2812****************************************************************************/
2813
2814bool print_queue_pause(struct auth_serversupplied_info *server_info, int snum,
2815 WERROR *errcode)
2816{
2817 int ret;
2818 struct printif *current_printif = get_printer_fns( snum );
2819
2820 if (!print_access_check(server_info, snum,
2821 PRINTER_ACCESS_ADMINISTER)) {
2822 *errcode = WERR_ACCESS_DENIED;
2823 return False;
2824 }
2825
2826
2827 become_root();
2828
2829 ret = (*(current_printif->queue_pause))(snum);
2830
2831 unbecome_root();
2832
2833 if (ret != 0) {
2834 *errcode = WERR_INVALID_PARAM;
2835 return False;
2836 }
2837
2838 /* force update the database */
2839 print_cache_flush(lp_const_servicename(snum));
2840
2841 /* Send a printer notify message */
2842
2843 notify_printer_status(snum, PRINTER_STATUS_PAUSED);
2844
2845 return True;
2846}
2847
2848/****************************************************************************
2849 Resume a queue.
2850****************************************************************************/
2851
2852bool print_queue_resume(struct auth_serversupplied_info *server_info, int snum,
2853 WERROR *errcode)
2854{
2855 int ret;
2856 struct printif *current_printif = get_printer_fns( snum );
2857
2858 if (!print_access_check(server_info, snum,
2859 PRINTER_ACCESS_ADMINISTER)) {
2860 *errcode = WERR_ACCESS_DENIED;
2861 return False;
2862 }
2863
2864 become_root();
2865
2866 ret = (*(current_printif->queue_resume))(snum);
2867
2868 unbecome_root();
2869
2870 if (ret != 0) {
2871 *errcode = WERR_INVALID_PARAM;
2872 return False;
2873 }
2874
2875 /* make sure the database is up to date */
2876 if (print_cache_expired(lp_const_servicename(snum), True))
2877 print_queue_update(snum, True);
2878
2879 /* Send a printer notify message */
2880
2881 notify_printer_status(snum, PRINTER_STATUS_OK);
2882
2883 return True;
2884}
2885
2886/****************************************************************************
2887 Purge a queue - implemented by deleting all jobs that we can delete.
2888****************************************************************************/
2889
2890bool print_queue_purge(struct auth_serversupplied_info *server_info, int snum,
2891 WERROR *errcode)
2892{
2893 print_queue_struct *queue;
2894 print_status_struct status;
2895 int njobs, i;
2896 bool can_job_admin;
2897
2898 /* Force and update so the count is accurate (i.e. not a cached count) */
2899 print_queue_update(snum, True);
2900
2901 can_job_admin = print_access_check(server_info, snum,
2902 JOB_ACCESS_ADMINISTER);
2903 njobs = print_queue_status(snum, &queue, &status);
2904
2905 if ( can_job_admin )
2906 become_root();
2907
2908 for (i=0;i<njobs;i++) {
2909 bool owner = is_owner(server_info, lp_const_servicename(snum),
2910 queue[i].job);
2911
2912 if (owner || can_job_admin) {
2913 print_job_delete1(snum, queue[i].job);
2914 }
2915 }
2916
2917 if ( can_job_admin )
2918 unbecome_root();
2919
2920 /* update the cache */
2921 print_queue_update( snum, True );
2922
2923 SAFE_FREE(queue);
2924
2925 return True;
2926}
Note: See TracBrowser for help on using the repository browser.