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
)
116 slprintf(keystr
, sizeof(keystr
) - 1, "STATUS/%s", printer
);
117 result
= tdb_store_int32(tdb
, keystr
, status
);
120 static int get_printer_status(char *printer
)
125 slprintf(keystr
, sizeof(keystr
) - 1, "STATUS/%s", printer
);
127 data
.dptr
= (unsigned char *)keystr
;
128 data
.dsize
= strlen(keystr
) + 1;
130 if (!tdb_exists(tdb
, data
)) {
131 set_printer_status(printer
, LPSTAT_OK
);
135 return tdb_fetch_int32(tdb
, keystr
);
138 /* Display printer queue */
140 static int lpq_command(int argc
, char **argv
)
143 struct vlp_job
*job_list
= NULL
;
144 int i
, num_jobs
, job_count
= 0;
147 printf("Usage: lpq <printername>\n");
153 /* Display printer status */
155 switch (get_printer_status(printer
)) {
160 printf("disabled\n");
168 /* Print queued documents */
170 get_job_list(printer
, &job_list
, &num_jobs
);
172 for (i
= 0; i
< num_jobs
; i
++) {
173 if (job_list
[i
].deleted
) continue;
174 printf("%d\t%d\t%d\t%ld\t%s\t%s\n", job_list
[i
].jobid
,
176 (i
== 0 && job_list
[i
].status
== LPQ_QUEUED
) ?
177 LPQ_SPOOLING
: job_list
[i
].status
,
178 (long int)job_list
[i
].submit_time
, job_list
[i
].owner
,
179 job_list
[i
].jobname
);
190 static int lprm_command(int argc
, char **argv
)
193 int jobid
, num_jobs
, i
;
194 struct vlp_job
*job_list
;
197 printf("Usage: lprm <printername> <jobid>\n");
202 jobid
= atoi(argv
[2]);
204 get_job_list(printer
, &job_list
, &num_jobs
);
206 for (i
= 0; i
< num_jobs
; i
++) {
207 if (job_list
[i
].jobid
== jobid
) {
208 job_list
[i
].deleted
= 1;
209 set_job_list(printer
, job_list
, num_jobs
);
217 /* print command = print-test %p %s */
219 static int print_command(int argc
, char **argv
)
224 TDB_DATA value
, queue
;
228 printf("Usage: print <printername> <jobname>\n");
236 /* Create a job record */
238 slprintf(job
.jobname
, sizeof(job
.jobname
) - 1, "%s", argv
[2]);
240 if (!(pw
= getpwuid(getuid()))) {
244 slprintf(job
.owner
, sizeof(job
.owner
) - 1, "%s", pw
->pw_name
);
246 job
.jobid
= next_jobnum(printer
);
248 job
.submit_time
= time(NULL
);
250 /* Store job entry in queue */
252 slprintf(keystr
, sizeof(keystr
) - 1, "LPQ/%s", printer
);
254 value
= tdb_fetch_bystring(tdb
, keystr
);
258 /* Add job to end of queue */
260 queue
.dptr
= (unsigned char *)malloc(value
.dsize
+ sizeof(struct vlp_job
));
261 if (!queue
.dptr
) return 1;
263 memcpy(queue
.dptr
, value
.dptr
, value
.dsize
);
264 memcpy(queue
.dptr
+ value
.dsize
, &job
, sizeof(struct vlp_job
));
266 queue
.dsize
= value
.dsize
+ sizeof(struct vlp_job
);
268 tdb_store_bystring(tdb
, keystr
, queue
, TDB_REPLACE
);
274 /* Create new queue */
275 queue
.dptr
= (unsigned char *)&job
;
276 queue
.dsize
= sizeof(struct vlp_job
);
278 tdb_store_bystring(tdb
, keystr
, queue
, TDB_REPLACE
);
284 /* Pause the queue */
286 static int queuepause_command(int argc
, char **argv
)
291 printf("Usage: queuepause <printername>\n");
296 set_printer_status(printer
, LPSTAT_STOPPED
);
301 /* Resume the queue */
303 static int queueresume_command(int argc
, char **argv
)
308 printf("Usage: queueresume <printername>\n");
313 set_printer_status(printer
, LPSTAT_OK
);
320 static int lppause_command(int argc
, char **argv
)
322 struct vlp_job
*job_list
;
324 int jobid
, num_jobs
, i
;
327 printf("Usage: lppause <printername> <jobid>\n");
332 jobid
= atoi(argv
[2]);
334 get_job_list(printer
, &job_list
, &num_jobs
);
336 for (i
= 0; i
< num_jobs
; i
++) {
337 if (job_list
[i
].jobid
== jobid
) {
338 job_list
[i
].status
= LPQ_PAUSED
;
339 set_job_list(printer
, job_list
, num_jobs
);
349 static int lpresume_command(int argc
, char **argv
)
351 struct vlp_job
*job_list
;
353 int jobid
, num_jobs
, i
;
356 printf("Usage: lpresume <printername> <jobid>\n");
361 jobid
= atoi(argv
[2]);
363 get_job_list(printer
, &job_list
, &num_jobs
);
365 for (i
= 0; i
< num_jobs
; i
++) {
366 if (job_list
[i
].jobid
== jobid
) {
367 job_list
[i
].status
= LPQ_QUEUED
;
368 set_job_list(printer
, job_list
, num_jobs
);
376 int main(int argc
, char **argv
)
378 /* Parameter check */
379 const char *printdb_path
= NULL
;
386 if (strncmp(argv
[1], "tdbfile", strlen("tdbfile")) != 0) {
391 printdb_path
= get_string_param(argv
[1]);
396 if (!(tdb
= tdb_open(printdb_path
, 0, 0, O_RDWR
| O_CREAT
,
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]);