#include <cups/cups.h>
#include <stdio.h>
#include <stdlib.h>
#include <stdarg.h>
#include <ctype.h>
#include <cups/http-private.h>
#include <cups/string.h>
#include <errno.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <signal.h>
#ifdef HAVE_INTTYPES_H
# include <inttypes.h>
#endif
#ifdef WIN32
# include <winsock.h>
#else
# include <sys/socket.h>
# include <netinet/in.h>
# include <arpa/inet.h>
# include <netdb.h>
#endif
#include <CoreFoundation/CFNumber.h>
#include <CoreFoundation/CFPreferences.h>
static char tmpfilename[1024] = "";
#define ORDER_CONTROL_DATA 0
#define ORDER_DATA_CONTROL 1
#define RESERVE_NONE 0
#define RESERVE_RFC1179 1
#define RESERVE_ANY 2
extern int rresvport(int *port);
static int lpd_command(int lpd_fd, int timeout, char *format, ...);
static int lpd_queue(const char *hostname, int port, const char *printer,
const char *filename,
const char *user, const char *title, int copies,
int banner, int format, int order, int reserve,
int manual_copies, int timeout);
static void lpd_timeout(int sig);
static int lpd_write(int lpd_fd, char *buffer, int length);
static void sigterm_handler(int sig);
static int connectTimeout(void);
int
main(int argc,
char *argv[])
{
char method[255],
hostname[1024],
username[255],
resource[1024],
*options,
name[255],
value[255],
*ptr,
*filename,
title[256];
int port;
int status;
int banner;
int format;
int order;
int reserve;
int sanitize_title;
int manual_copies,
timeout,
copies;
#if defined(HAVE_SIGACTION) && !defined(HAVE_SIGSET)
struct sigaction action;
#endif
setbuf(stderr, NULL);
#ifdef HAVE_SIGSET
sigset(SIGPIPE, SIG_IGN);
sigset(SIGTERM, sigterm_handler);
#elif defined(HAVE_SIGACTION)
memset(&action, 0, sizeof(action));
action.sa_handler = SIG_IGN;
sigaction(SIGPIPE, &action, NULL);
sigemptyset(&action.sa_mask);
sigaddset(&action.sa_mask, SIGTERM);
action.sa_handler = sigterm_handler;
sigaction(SIGTERM, &action, NULL);
#else
signal(SIGPIPE, SIG_IGN);
signal(SIGTERM, sigterm_handler);
#endif
if (argc == 1)
{
puts("network lpd \"Unknown\" \"LPD/LPR Host or Printer\"");
return (0);
}
else if (argc < 6 || argc > 7)
{
fprintf(stderr, "Usage: %s job-id user title copies options [file]\n",
argv[0]);
return (1);
}
if (argc == 6)
{
int fd;
char buffer[8192];
int bytes;
if ((fd = cupsTempFd(tmpfilename, sizeof(tmpfilename))) < 0)
{
perror("ERROR: unable to create temporary file");
return (1);
}
while ((bytes = fread(buffer, 1, sizeof(buffer), stdin)) > 0)
if (write(fd, buffer, bytes) < bytes)
{
perror("ERROR: unable to write to temporary file");
close(fd);
unlink(tmpfilename);
return (1);
}
close(fd);
filename = tmpfilename;
}
else
filename = argv[6];
httpSeparate(argv[0], method, username, hostname, &port, resource);
if (!username[0])
{
strlcpy(username, argv[2], sizeof(username));
}
banner = 0;
format = 'l';
order = ORDER_CONTROL_DATA;
reserve = RESERVE_ANY;
manual_copies = 1;
timeout = 300;
sanitize_title = 1;
#if defined(__APPLE__)
sanitize_title= 0;
#endif
if ((options = strchr(resource, '?')) != NULL)
{
*options++ = '\0';
while (*options)
{
for (ptr = name; *options && *options != '=';)
if (ptr < (name + sizeof(name) - 1))
*ptr++ = *options++;
*ptr = '\0';
if (*options == '=')
{
options ++;
for (ptr = value; *options && *options != '+';)
if (ptr < (value + sizeof(value) - 1))
*ptr++ = *options++;
*ptr = '\0';
if (*options == '+')
options ++;
}
else
value[0] = '\0';
if (strcasecmp(name, "banner") == 0)
{
banner = !value[0] ||
strcasecmp(value, "on") == 0 ||
strcasecmp(value, "yes") == 0 ||
strcasecmp(value, "true") == 0;
}
else if (strcasecmp(name, "format") == 0 && value[0])
{
if (strchr("cdfglnoprtv", value[0]) != NULL)
format = value[0];
else
fprintf(stderr, "ERROR: Unknown format character \"%c\"\n", value[0]);
}
else if (strcasecmp(name, "order") == 0 && value[0])
{
if (strcasecmp(value, "control,data") == 0)
order = ORDER_CONTROL_DATA;
else if (strcasecmp(value, "data,control") == 0)
order = ORDER_DATA_CONTROL;
else
fprintf(stderr, "ERROR: Unknown file order \"%s\"\n", value);
}
else if (strcasecmp(name, "reserve") == 0)
{
if (!value[0] ||
!strcasecmp(value, "on") ||
!strcasecmp(value, "yes") ||
!strcasecmp(value, "true") ||
!strcasecmp(value, "rfc1179"))
reserve = RESERVE_RFC1179;
else if (!strcasecmp(value, "any"))
reserve = RESERVE_ANY;
else
reserve = RESERVE_NONE;
}
else if (strcasecmp(name, "manual_copies") == 0)
{
manual_copies = !value[0] ||
strcasecmp(value, "on") == 0 ||
strcasecmp(value, "yes") == 0 ||
strcasecmp(value, "true") == 0;
}
else if (strcasecmp(name, "sanitize_title") == 0)
{
sanitize_title = !value[0] ||
strcasecmp(value, "on") == 0 ||
strcasecmp(value, "yes") == 0 ||
strcasecmp(value, "true") == 0;
}
else if (strcasecmp(name, "timeout") == 0)
{
if (atoi(value) > 0)
timeout = atoi(value);
}
}
}
strlcpy(title, argv[3], sizeof(title));
if (sanitize_title)
{
for (ptr = title; *ptr; ptr ++)
if (!isalnum(*ptr & 255) && !isspace(*ptr & 255))
*ptr = '_';
}
if (argc > 6)
{
if (manual_copies)
{
manual_copies = atoi(argv[4]);
copies = 1;
}
else
{
manual_copies = 1;
copies = atoi(argv[4]);
}
status = lpd_queue(hostname, port, resource + 1, filename,
username, title, copies,
banner, format, order, reserve, manual_copies, timeout);
if (!status)
fprintf(stderr, "PAGE: 1 %d\n", atoi(argv[4]));
}
else
status = lpd_queue(hostname, port, resource + 1, filename,
username, title, 1,
banner, format, order, reserve, 1, timeout);
if (tmpfilename[0])
unlink(tmpfilename);
return (status);
}
static int
lpd_command(int fd,
int timeout,
char *format,
...)
{
va_list ap;
char buf[1024];
int bytes;
char status;
va_start(ap, format);
bytes = vsnprintf(buf, sizeof(buf), format, ap);
va_end(ap);
fprintf(stderr, "DEBUG: lpd_command %2.2x %s", buf[0], buf + 1);
fprintf(stderr, "DEBUG: Sending command string (%d bytes)...\n", bytes);
if (lpd_write(fd, buf, bytes) < bytes)
{
perror("ERROR: Unable to send LPD command");
return (-1);
}
fprintf(stderr, "DEBUG: Reading command status...\n");
alarm(timeout);
if (recv(fd, &status, 1, 0) < 1)
{
fprintf(stderr, "WARNING: Remote host did not respond with command "
"status byte after %d seconds!\n", timeout);
status = errno;
}
alarm(0);
fprintf(stderr, "DEBUG: lpd_command returning %d\n", status);
return (status);
}
static int
lpd_queue(const char *hostname,
int port,
const char *printer,
const char *filename,
const char *user,
const char *title,
int copies,
int banner,
int format,
int order,
int reserve,
int manual_copies,
int timeout)
{
FILE *fp;
char localhost[255];
int error;
struct stat filestats;
int lport;
int fd;
char control[10240],
*cptr;
char status;
struct sockaddr_in addr;
struct hostent *hostaddr;
int copy;
size_t nbytes;
off_t tbytes;
char buffer[8192];
time_t connect_time,
connect_timeout;
int recoverableErrShown;
#if defined(HAVE_SIGACTION) && !defined(HAVE_SIGSET)
struct sigaction action;
#endif
#ifdef HAVE_SIGSET
sigset(SIGALRM, lpd_timeout);
#elif defined(HAVE_SIGACTION)
memset(&action, 0, sizeof(action));
sigemptyset(&action.sa_mask);
action.sa_handler = lpd_timeout;
sigaction(SIGALRM, &action, NULL);
#else
signal(SIGALRM, lpd_timeout);
#endif
recoverableErrShown = 0;
connect_timeout = -1;
connect_time = time(NULL);
for (;;)
{
if ((hostaddr = httpGetHostByName(hostname)) == NULL)
{
fprintf(stderr, "ERROR: Unable to locate printer \'%s\' - %s\n",
hostname, hstrerror(h_errno));
return (1);
}
fprintf(stderr, "INFO: Attempting to connect to host %s for printer %s\n",
hostname, printer);
memset(&addr, 0, sizeof(addr));
memcpy(&(addr.sin_addr), hostaddr->h_addr, hostaddr->h_length);
addr.sin_family = hostaddr->h_addrtype;
addr.sin_port = htons(port);
for (lport = reserve == RESERVE_RFC1179 ? 732 : 1024;;)
{
lport --;
if (lport < 721 && reserve == RESERVE_RFC1179)
lport = 731;
else if (lport < 1)
lport = 1023;
#ifdef HAVE_GETEUID
if (geteuid() || !reserve)
#else
if (getuid() || !reserve)
#endif
{
if ((fd = socket(AF_INET, SOCK_STREAM, 0)) < 0)
{
perror("ERROR: Unable to create socket");
return (1);
}
lport = 0;
}
else
{
if ((fd = rresvport(&lport)) < 0)
{
perror("ERROR: Unable to reserve port");
sleep(1);
continue;
}
}
if (connect(fd, (struct sockaddr *)&addr, sizeof(addr)) < 0)
{
error = errno;
close(fd);
fd = -1;
if (error == ECONNREFUSED || error == EHOSTDOWN ||
error == EHOSTUNREACH)
{
if (connect_timeout == -1)
connect_timeout = connectTimeout();
if (connect_timeout && (time(NULL) - connect_time) > connect_timeout)
{
fprintf(stderr, "ERROR: Printer not responding\n");
return (ETIMEDOUT);
}
recoverableErrShown = true;
fprintf(stderr, "WARNING: recoverable: Network host \'%s\' is busy, down, or unreachable; will retry in 30 seconds...\n",
hostname);
sleep(30);
}
else if (error == EADDRINUSE)
{
sleep(1);
}
else
{
recoverableErrShown = true;
perror("ERROR: recoverable: Unable to connect to printer; will retry in 30 seconds...");
sleep(30);
}
}
else
break;
}
if (recoverableErrShown)
{
fprintf(stderr, "INFO: recovered: \n");
sleep(5);
}
fprintf(stderr, "INFO: Connected to %s...\n", hostname);
fprintf(stderr, "DEBUG: Connected on ports %d (local %d)...\n", port,
lport);
if (stat(filename, &filestats))
{
perror("ERROR: unable to stat print file");
return (1);
}
filestats.st_size *= manual_copies;
if ((fp = fopen(filename, "rb")) == NULL)
{
perror("ERROR: unable to open print file for reading");
return (1);
}
if (lpd_command(fd, timeout, "\002%s\n",
printer))
return (1);
gethostname(localhost, sizeof(localhost));
localhost[31] = '\0';
snprintf(control, sizeof(control), "H%s\nP%s\nJ%s\n", localhost, user,
title);
cptr = control + strlen(control);
if (banner)
{
snprintf(cptr, sizeof(control) - (cptr - control), "C%s\nL%s\n",
localhost, user);
cptr += strlen(cptr);
}
while (copies > 0)
{
snprintf(cptr, sizeof(control) - (cptr - control), "%cdfA%03d%.15s\n", format,
getpid() % 1000, localhost);
cptr += strlen(cptr);
copies --;
}
snprintf(cptr, sizeof(control) - (cptr - control),
"UdfA%03d%.15s\nN%s\n",
getpid() % 1000, localhost, title);
fprintf(stderr, "DEBUG: Control file is:\n%s", control);
if (order == ORDER_CONTROL_DATA)
{
if (lpd_command(fd, timeout, "\002%d cfA%03.3d%.15s\n", strlen(control),
getpid() % 1000, localhost))
return (1);
fprintf(stderr, "INFO: Sending control file (%lu bytes)\n",
(unsigned long)strlen(control));
if (lpd_write(fd, control, strlen(control) + 1) < (strlen(control) + 1))
{
status = errno;
perror("ERROR: Unable to write control file");
}
else
{
alarm(timeout);
if (read(fd, &status, 1) < 1)
{
fprintf(stderr, "WARNING: Remote host did not respond with control "
"status byte after %d seconds!\n", timeout);
status = errno;
}
alarm(0);
}
if (status != 0)
fprintf(stderr, "ERROR: Remote host did not accept control file (%d)\n",
status);
else
fputs("INFO: Control file sent successfully\n", stderr);
}
else
status = 0;
if (status == 0)
{
if (lpd_command(fd, timeout, "\003%" PRIdMAX " dfA%03.3d%.15s\n",
(intmax_t)filestats.st_size, getpid() % 1000,
localhost))
return (1);
fprintf(stderr, "INFO: Sending data file (%" PRIdMAX " bytes)\n",
(intmax_t)filestats.st_size);
tbytes = 0;
for (copy = 0; copy < manual_copies; copy ++)
{
rewind(fp);
while ((nbytes = fread(buffer, 1, sizeof(buffer), fp)) > 0)
{
fprintf(stderr, "INFO: Spooling LPR job, %u%% complete...\n",
(unsigned)(100.0f * tbytes / filestats.st_size));
if (lpd_write(fd, buffer, nbytes) < nbytes)
{
perror("ERROR: Unable to send print file to printer");
break;
}
else
tbytes += nbytes;
}
}
if (tbytes < filestats.st_size)
status = errno;
else if (lpd_write(fd, "", 1) < 1)
{
perror("ERROR: Unable to send trailing nul to printer");
status = errno;
}
else
{
alarm(timeout);
if (recv(fd, &status, 1, 0) < 1)
{
fprintf(stderr, "WARNING: Remote host did not respond with data "
"status byte after %d seconds!\n", timeout);
status = 0;
}
alarm(0);
}
if (status != 0)
fprintf(stderr, "ERROR: Remote host did not accept data file (%d)\n",
status);
else
fputs("INFO: Data file sent successfully\n", stderr);
}
if (status == 0 && order == ORDER_DATA_CONTROL)
{
if (lpd_command(fd, timeout, "\002%d cfA%03.3d%.15s\n", strlen(control),
getpid() % 1000, localhost))
return (1);
fprintf(stderr, "INFO: Sending control file (%lu bytes)\n",
(unsigned long)strlen(control));
if (lpd_write(fd, control, strlen(control) + 1) < (strlen(control) + 1))
{
status = errno;
perror("ERROR: Unable to write control file");
}
else
{
alarm(timeout);
if (read(fd, &status, 1) < 1)
{
fprintf(stderr, "WARNING: Remote host did not respond with control "
"status byte after %d seconds!\n", timeout);
status = errno;
}
alarm(0);
}
if (status != 0)
fprintf(stderr, "ERROR: Remote host did not accept control file (%d)\n",
status);
else
fputs("INFO: Control file sent successfully\n", stderr);
}
close(fd);
fclose(fp);
if (status == 0)
return (0);
sleep(30);
}
}
static void
lpd_timeout(int sig)
{
(void)sig;
#if !defined(HAVE_SIGSET) && !defined(HAVE_SIGACTION)
signal(SIGALRM, lpd_timeout);
#endif
}
static int
lpd_write(int lpd_fd,
char *buffer,
int length)
{
int bytes,
total;
total = 0;
while ((bytes = send(lpd_fd, buffer, length - total, 0)) >= 0)
{
total += bytes;
buffer += bytes;
if (total == length)
break;
}
if (bytes < 0)
return (-1);
else
return (length);
}
#ifndef HAVE_RRESVPORT
int
rresvport(int *port)
{
struct sockaddr_in addr;
int fd;
if ((fd = socket(AF_INET, SOCK_STREAM, 0)) < 0)
return (-1);
memset(&addr, 0, sizeof(addr));
addr.sin_family = AF_INET;
addr.sin_addr.s_addr = INADDR_ANY;
while (*port > 720)
{
addr.sin_port = htons(*port);
if (!bind(fd, (struct sockaddr *)&addr, sizeof(addr)))
return (fd);
if (errno != EADDRINUSE)
{
# ifdef WIN32
closesocket(fd);
# else
close(fd);
# endif
return (-1);
}
(*port)--;
}
# ifdef WIN32
closesocket(fd);
# else
close(fd);
# endif
return (-1);
}
#endif
static void
sigterm_handler(int sig)
{
(void)sig;
if (tmpfilename[0])
unlink(tmpfilename);
exit(1);
}
static int connectTimeout()
{
CFPropertyListRef value;
SInt32 connect_timeout = (7 * 24 * 60 * 60);
value = CFPreferencesCopyValue(CFSTR("timeout"), CFSTR("com.apple.print.backends"),
kCFPreferencesAnyUser, kCFPreferencesCurrentHost);
if (value != NULL)
{
if (CFGetTypeID(value) == CFNumberGetTypeID())
CFNumberGetValue(value, kCFNumberSInt32Type, &connect_timeout);
CFRelease(value);
}
return connect_timeout;
}