diff options
Diffstat (limited to 'printutils/lpr.c')
| -rw-r--r-- | printutils/lpr.c | 200 |
1 files changed, 200 insertions, 0 deletions
diff --git a/printutils/lpr.c b/printutils/lpr.c new file mode 100644 index 0000000..b8c77bf --- /dev/null +++ b/printutils/lpr.c @@ -0,0 +1,200 @@ +/* vi: set sw=4 ts=4: */ +/* + * Copyright 2008 Walter Harms (WHarms@bfs.de) + * + * Licensed under the GPL v2, see the file LICENSE in this tarball. + */ +#include "libbb.h" +#include "lpr.h" + +static char *mygethostname31(void) +{ + char *s = xzalloc(32); + if (gethostname(s, 31) < 0) + bb_perror_msg_and_die("gethostname"); + /* gethostname() does not guarantee NUL-termination. xzalloc does. */ + return s; +} + +static int xmkstemp(char *temp_name) +{ + int fd; + + fd = mkstemp(temp_name); + if (fd < 0) + bb_perror_msg_and_die("mkstemp"); + return fd; +} + +/* lpd server sends NUL byte on success. + * Else read the errormessage and exit. + */ +static void get_response(int server_sock, const char *emsg) +{ + char buf = '\0'; + + safe_read(server_sock, &buf, 1); + if (buf != '\0') { + bb_error_msg("%s. Server said:", emsg); + fputc(buf, stderr); + logmode = 0; /* no errors from bb_copyfd_eof() */ + bb_copyfd_eof(server_sock, STDERR_FILENO); + xfunc_die(); + } +} + +int lpr_main(int argc, char *argv[]) MAIN_EXTERNALLY_VISIBLE; +int lpr_main(int argc, char *argv[]) +{ + struct netprint netprint; + char temp_name[] = "/tmp/lprXXXXXX"; /* for mkstemp */ + char *strings[5]; + const char *netopt; + const char *jobtitle; + const char *hostname; + const char *jobclass; + char *logname; + int pid1000; + int server_sock, tmp_fd; + unsigned opt; + enum { + VERBOSE = 1 << 0, + USE_HEADER = 1 << 1, /* -h banner or header for this job */ + USE_MAIL = 1 << 2, /* send mail to user@hostname */ + OPT_U = 1 << 3, /* -U <username> */ + OPT_J = 1 << 4, /* -J <title> is the jobtitle for the banner page */ + OPT_C = 1 << 5, /* -C <class> job classification */ + OPT_P = 1 << 6, /* -P <queue[@host[:port]]> */ + /* if no -P is given use $PRINTER, then "lp@localhost:515" */ + }; + + /* Set defaults, parse options */ + hostname = mygethostname31(); + netopt = NULL; + logname = getenv("LOGNAME"); + if (logname == NULL) + logname = (char*)"user"; /* TODO: getpwuid(getuid())->pw_name? */ + opt = getopt32(argv, "VhmU:J:C:P:", + &logname, &jobtitle, &jobclass, &netopt); + argv += optind; + parse_prt(netopt, &netprint); + logname = xstrndup(logname, 31); + + /* For stdin we need to save it to a tempfile, + * otherwise we can't know the size. */ + tmp_fd = -1; + if (!*argv) { + if (jobtitle == NULL) + jobtitle = (char *) bb_msg_standard_input; + + tmp_fd = xmkstemp(temp_name); + if (bb_copyfd_eof(STDIN_FILENO, tmp_fd) < 0) + goto del_temp_file; + /* TODO: we actually can have deferred write errors... */ + close(tmp_fd); + argv--; /* back off from NULL */ + *argv = temp_name; + } + + if (opt & VERBOSE) + puts("connect to server"); + server_sock = xconnect_stream(netprint.lsa); + + /* "Receive a printer job" command */ + fdprintf(server_sock, "\x2" "%s\n", netprint.queue); + get_response(server_sock, "set queue failed"); + + pid1000 = getpid() % 1000; + while (*argv) { + char dfa_name[sizeof("dfAnnn") + 32]; + struct stat st; + char **sptr; + unsigned size; + int fd; + + fd = xopen(*argv, O_RDONLY); + + /* "The name ... should start with ASCII "dfA", + * followed by a three digit job number, followed + * by the host name which has constructed the file." */ + snprintf(dfa_name, sizeof(dfa_name), + "dfA%03u%s", pid1000, hostname); + pid1000 = (pid1000 + 1) % 1000; + + /* + * Generate control file contents + */ + /* H HOST, P USER, l DATA_FILE_NAME, J JOBNAME */ + strings[0] = xasprintf("H%.32s\n" "P%.32s\n" "l%.32s\n" + "J%.99s\n", + hostname, logname, dfa_name, + !(opt & OPT_J) ? *argv : jobtitle); + sptr = &strings[1]; + /* C CLASS - printed on banner page (if L cmd is also given) */ + if (opt & OPT_J) /* [1] */ + *sptr++ = xasprintf("C%.32s\n", jobclass); + /* M WHOM_TO_MAIL */ + if (opt & USE_MAIL) /* [2] */ + *sptr++ = xasprintf("M%.32s\n", logname); + /* H USER - print banner page, with given user's name */ + if (opt & USE_HEADER) /* [3] */ + *sptr++ = xasprintf("L%.32s\n", logname); + *sptr = NULL; /* [4] max */ + + /* RFC 1179: "LPR servers MUST be able + * to receive the control file subcommand first + * and SHOULD be able to receive the data file + * subcommand first". + * Ok, we'll send control file first. */ + size = 0; + sptr = strings; + while (*sptr) + size += strlen(*sptr++); + if (opt & VERBOSE) + puts("send control file"); + /* 2: "Receive control file" subcommand */ + fdprintf(server_sock, "\x2" "%u c%s\n", size, dfa_name + 1); + sptr = strings; + while (*sptr) { + xwrite(server_sock, *sptr, strlen(*sptr)); + free(*sptr); + sptr++; + } + free(strings); + /* "Once all of the contents have + * been delivered, an octet of zero bits is sent as + * an indication that the file being sent is complete. + * A second level of acknowledgement processing + * must occur at this point." */ + xwrite(server_sock, "", 1); + get_response(server_sock, "send control file failed"); + + /* Sending data */ + st.st_size = 0; /* paranoia */ + fstat(fd, &st); + if (opt & VERBOSE) + puts("send data file"); + /* 3: "Receive data file" subcommand */ + fdprintf(server_sock, "\x3" "%"OFF_FMT"u %s\n", st.st_size, dfa_name); + /* TODO: if file shrank and we wrote less than st.st_size, + * pad output with NUL bytes? Otherwise server won't know + * that we are done. */ + if (bb_copyfd_size(fd, server_sock, st.st_size) < 0) + xfunc_die(); + close(fd); + xwrite(server_sock, "", 1); + get_response(server_sock, "send file failed"); + + argv++; + } + + if (ENABLE_FEATURE_CLEAN_UP) + close(server_sock); + + if (tmp_fd >= 0) { + del_temp_file: + unlink(temp_name); + } + + return 0; +} |