2 Unix SMB/Netbios implementation.
4 Virtual lp system for printer testing
6 Copyright (C) Tim Potter 2000
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/>.
23 #include "system/passwd.h"
24 #include "system/filesys.h"
32 #define PRINT_FIRSTJOB "100"
34 static TDB_CONTEXT
*tdb
;
48 static void usage(void)
50 printf("Usage: vlp tdbfile=/tmp/vlp.tdb lpq|lprm|print|queuepause|queueresume|"
51 "lppause|lpresume [args]\n");
54 /* Return an array of vlp jobs that is the printer queue */
56 static void get_job_list(char *printer
, struct vlp_job
**job_list
,
62 slprintf(keystr
, sizeof(keystr
) - 1, "LPQ/%s", printer
);
63 data
= tdb_fetch_bystring(tdb
, keystr
);
65 *job_list
= (struct vlp_job
*)data
.dptr
;
66 *num_jobs
= data
.dsize
/ sizeof(struct vlp_job
);
69 /* Store an array of vl jobs for the queue */
71 static void set_job_list(char *printer
, struct vlp_job
*job_list
,
77 slprintf(keystr
, sizeof(keystr
) - 1, "LPQ/%s", printer
);
79 data
.dptr
= (unsigned char *)job_list
;
80 data
.dsize
= num_jobs
* sizeof(struct vlp_job
);
81 tdb_store_bystring(tdb
, keystr
, data
, TDB_REPLACE
);
84 /* Return the next job number for a printer */
86 static int next_jobnum(char *printer
)
91 slprintf(keystr
, sizeof(keystr
) - 1, "JOBNUM/%s", printer
);
93 tdb_lock_bystring(tdb
, keystr
);
95 jobnum
= tdb_fetch_int32(tdb
, keystr
);
97 /* Create next job index if none exists */
100 jobnum
= atoi(PRINT_FIRSTJOB
);
104 tdb_store_int32(tdb
, keystr
, jobnum
);
106 tdb_unlock_bystring(tdb
, keystr
);
111 static void set_printer_status(char *printer
, int status
)
115 slprintf(keystr
, sizeof(keystr
) - 1, "STATUS/%s", printer
);
116 tdb_store_int32(tdb
, keystr
, status
);
119 static int get_printer_status(char *printer
)
124 slprintf(keystr
, sizeof(keystr
) - 1, "STATUS/%s", printer
);
126 data
.dptr
= (unsigned char *)keystr
;
127 data
.dsize
= strlen(keystr
) + 1;
129 if (!tdb_exists(tdb
, data
)) {
130 set_printer_status(printer
, LPSTAT_OK
);
134 return tdb_fetch_int32(tdb
, keystr
);
137 /* Display printer queue */
139 static int lpq_command(int argc
, char **argv
)
142 struct vlp_job
*job_list
= NULL
;
143 int i
, num_jobs
, job_count
= 0;
146 printf("Usage: lpq <printername>\n");
152 /* Display printer status */
154 switch (get_printer_status(printer
)) {
159 printf("disabled\n");
167 /* Print queued documents */
169 get_job_list(printer
, &job_list
, &num_jobs
);
171 for (i
= 0; i
< num_jobs
; i
++) {
172 if (job_list
[i
].deleted
) continue;
173 printf("%d\t%d\t%d\t%ld\t%s\t%s\n", job_list
[i
].jobid
,
175 (i
== 0 && job_list
[i
].status
== LPQ_QUEUED
) ?
176 LPQ_SPOOLING
: job_list
[i
].status
,
177 (long int)job_list
[i
].submit_time
, job_list
[i
].owner
,
178 job_list
[i
].jobname
);
189 static int lprm_command(int argc
, char **argv
)
192 int jobid
, num_jobs
, i
;
193 struct vlp_job
*job_list
;
196 printf("Usage: lprm <printername> <jobid>\n");
201 jobid
= atoi(argv
[2]);
203 get_job_list(printer
, &job_list
, &num_jobs
);
205 for (i
= 0; i
< num_jobs
; i
++) {
206 if (job_list
[i
].jobid
== jobid
) {
207 job_list
[i
].deleted
= 1;
208 set_job_list(printer
, job_list
, num_jobs
);
216 /* print command = print-test %p %s */
218 static int print_command(int argc
, char **argv
)
223 TDB_DATA value
, queue
;
227 printf("Usage: print <printername> <jobname>\n");
235 /* Create a job record */
237 slprintf(job
.jobname
, sizeof(job
.jobname
) - 1, "%s", argv
[2]);
239 if (!(pw
= getpwuid(getuid()))) {
243 slprintf(job
.owner
, sizeof(job
.owner
) - 1, "%s", pw
->pw_name
);
245 job
.jobid
= next_jobnum(printer
);
247 job
.submit_time
= time(NULL
);
249 /* Store job entry in queue */
251 slprintf(keystr
, sizeof(keystr
) - 1, "LPQ/%s", printer
);
253 value
= tdb_fetch_bystring(tdb
, keystr
);
257 /* Add job to end of queue */
259 queue
.dptr
= (unsigned char *)malloc(value
.dsize
+ sizeof(struct vlp_job
));
260 if (!queue
.dptr
) return 1;
262 memcpy(queue
.dptr
, value
.dptr
, value
.dsize
);
263 memcpy(queue
.dptr
+ value
.dsize
, &job
, sizeof(struct vlp_job
));
265 queue
.dsize
= value
.dsize
+ sizeof(struct vlp_job
);
267 tdb_store_bystring(tdb
, keystr
, queue
, TDB_REPLACE
);
273 /* Create new queue */
274 queue
.dptr
= (unsigned char *)&job
;
275 queue
.dsize
= sizeof(struct vlp_job
);
277 tdb_store_bystring(tdb
, keystr
, queue
, TDB_REPLACE
);
283 /* Pause the queue */
285 static int queuepause_command(int argc
, char **argv
)
290 printf("Usage: queuepause <printername>\n");
295 set_printer_status(printer
, LPSTAT_STOPPED
);
300 /* Resume the queue */
302 static int queueresume_command(int argc
, char **argv
)
307 printf("Usage: queueresume <printername>\n");
312 set_printer_status(printer
, LPSTAT_OK
);
319 static int lppause_command(int argc
, char **argv
)
321 struct vlp_job
*job_list
;
323 int jobid
, num_jobs
, i
;
326 printf("Usage: lppause <printername> <jobid>\n");
331 jobid
= atoi(argv
[2]);
333 get_job_list(printer
, &job_list
, &num_jobs
);
335 for (i
= 0; i
< num_jobs
; i
++) {
336 if (job_list
[i
].jobid
== jobid
) {
337 job_list
[i
].status
= LPQ_PAUSED
;
338 set_job_list(printer
, job_list
, num_jobs
);
348 static int lpresume_command(int argc
, char **argv
)
350 struct vlp_job
*job_list
;
352 int jobid
, num_jobs
, i
;
355 printf("Usage: lpresume <printername> <jobid>\n");
360 jobid
= atoi(argv
[2]);
362 get_job_list(printer
, &job_list
, &num_jobs
);
364 for (i
= 0; i
< num_jobs
; i
++) {
365 if (job_list
[i
].jobid
== jobid
) {
366 job_list
[i
].status
= LPQ_QUEUED
;
367 set_job_list(printer
, job_list
, num_jobs
);
375 int main(int argc
, char **argv
)
377 /* Parameter check */
378 const char *printdb_path
= NULL
;
385 if (strncmp(argv
[1], "tdbfile", strlen("tdbfile")) != 0) {
390 printdb_path
= get_string_param(argv
[1]);
395 /* FIXME: We should *never* open a tdb without logging! */
396 if (!(tdb
= tdb_open_compat(printdb_path
, 0, 0, O_RDWR
| O_CREAT
,
397 0666, NULL
, NULL
))) {
398 printf("%s: unable to open %s\n", argv
[0], printdb_path
);
402 /* Ensure we are modes 666 */
404 chmod(printdb_path
, 0666);
408 if (strcmp(argv
[2], "lpq") == 0) {
409 return lpq_command(argc
- 2, &argv
[2]);
412 if (strcmp(argv
[2], "lprm") == 0) {
413 return lprm_command(argc
- 2, &argv
[2]);
416 if (strcmp(argv
[2], "print") == 0) {
417 return print_command(argc
- 2, &argv
[2]);
420 if (strcmp(argv
[2], "queuepause") == 0) {
421 return queuepause_command(argc
- 2, &argv
[2]);
424 if (strcmp(argv
[2], "queueresume") == 0) {
425 return queueresume_command(argc
- 2, &argv
[2]);
428 if (strcmp(argv
[2], "lppause") == 0) {
429 return lppause_command(argc
- 2, &argv
[2]);
432 if (strcmp(argv
[2], "lpresume") == 0) {
433 return lpresume_command(argc
- 2, &argv
[2]);
436 /* Unknown command */
438 printf("%s: invalid command %s\n", argv
[0], argv
[1]);