#include <cups/cups.h>
#include <stdio.h>
#include <stdlib.h>
#include <stdarg.h>
#include <ctype.h>
#include <cups/string.h>
#include <errno.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <signal.h>
#ifdef WIN32
# include <winsock.h>
#else
# include <sys/socket.h>
# include <netinet/in.h>
# include <arpa/inet.h>
# include <netdb.h>
#endif
#define ORDER_CONTROL_DATA 0
#define ORDER_DATA_CONTROL 1
extern int rresvport(int *port);
static int lpd_command(int lpd_fd, char *format, ...);
static int lpd_queue(char *hostname, char *printer, char *filename,
int fromstdin, char *user, char *title, int copies,
int banner, int format, int order);
static int lpd_write(int lpd_fd, char *buffer, int length);
int
main(int argc,
char *argv[])
{
char method[255],
hostname[1024],
username[255],
resource[1024],
*options,
name[255],
value[255],
*ptr,
filename[1024],
title[256];
int port;
int status;
int banner;
int format;
int order;
setbuf(stderr, NULL);
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(filename, sizeof(filename))) < 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(filename);
return (1);
}
close(fd);
}
else
strlcpy(filename, argv[6], sizeof(filename));
httpSeparate(argv[0], method, username, hostname, &port, resource);
banner = 0;
format = 'l';
order = ORDER_CONTROL_DATA;
if ((options = strchr(resource, '?')) != NULL)
{
*options++ = '\0';
while (*options)
{
for (ptr = name; *options && *options != '=';)
*ptr++ = *options++;
*ptr = '\0';
if (*options == '=')
{
options ++;
for (ptr = value; *options && *options != '+';)
*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);
}
}
}
strlcpy(title, argv[3], sizeof(title));
for (ptr = title; *ptr; ptr ++)
if (!isalnum(*ptr) && !isspace(*ptr))
*ptr = '_';
if (argc > 6)
{
status = lpd_queue(hostname, resource + 1, filename, 0,
argv[2] , title, atoi(argv[4]) ,
banner, format, order);
if (!status)
fprintf(stderr, "PAGE: 1 %d\n", atoi(argv[4]));
}
else
status = lpd_queue(hostname, resource + 1, filename, 1,
argv[2] , title, 1, banner, format, order);
if (argc < 7)
unlink(filename);
return (status);
}
static int
lpd_command(int fd,
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)
return (-1);
fprintf(stderr, "DEBUG: Reading command status...\n");
if (recv(fd, &status, 1, 0) < 1)
return (-1);
fprintf(stderr, "DEBUG: lpd_command returning %d\n", status);
return (status);
}
static int
lpd_queue(char *hostname,
char *printer,
char *filename,
int fromstdin,
char *user,
char *title,
int copies,
int banner,
int format,
int order)
{
FILE *fp;
char localhost[255];
int error;
struct stat filestats;
int port;
int fd;
char control[10240],
*cptr;
char status;
struct sockaddr_in addr;
struct hostent *hostaddr;
size_t nbytes,
tbytes;
char buffer[8192];
#if defined(HAVE_SIGACTION) && !defined(HAVE_SIGSET)
struct sigaction action;
#endif
if ((hostaddr = httpGetHostByName(hostname)) == NULL)
{
fprintf(stderr, "ERROR: Unable to locate printer \'%s\' - %s",
hostname, strerror(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(515);
for (port = 732;;)
{
if (getuid())
{
if ((fd = socket(AF_INET, SOCK_STREAM, 0)) < 0)
{
perror("ERROR: Unable to create socket");
return (1);
}
}
else
{
if ((fd = rresvport(&port)) < 0)
{
perror("ERROR: Unable to reserve port");
sleep(30);
continue;
}
}
if (connect(fd, (struct sockaddr *)&addr, sizeof(addr)) < 0)
{
error = errno;
close(fd);
fd = -1;
if (error == ECONNREFUSED || error == EHOSTDOWN ||
error == EHOSTUNREACH)
{
fprintf(stderr, "INFO: Network host \'%s\' is busy; will retry in 30 seconds...",
hostname);
sleep(30);
}
else if (error == EADDRINUSE)
{
port --;
if (port < 721)
port = 732;
}
else
{
perror("ERROR: Unable to connect to printer");
sleep(30);
}
}
else
break;
}
fprintf(stderr, "INFO: Connected from port %d...\n", port);
if (fromstdin)
{
#ifdef HAVE_SIGSET
sigset(SIGTERM, SIG_IGN);
#elif defined(HAVE_SIGACTION)
memset(&action, 0, sizeof(action));
sigemptyset(&action.sa_mask);
action.sa_handler = SIG_IGN;
sigaction(SIGTERM, &action, NULL);
#else
signal(SIGTERM, SIG_IGN);
#endif
}
if (stat(filename, &filestats))
{
perror("ERROR: unable to stat print file");
return (1);
}
if ((fp = fopen(filename, "rb")) == NULL)
{
perror("ERROR: unable to open print file for reading");
return (1);
}
lpd_command(fd, "\002%s\n", printer);
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), "L%s\n", user);
cptr += strlen(cptr);
}
while (copies > 0)
{
snprintf(cptr, sizeof(control) - (cptr - control), "%cdfA%03d%s\n", format,
getpid() % 1000, localhost);
cptr += strlen(cptr);
copies --;
}
snprintf(cptr, sizeof(control) - (cptr - control),
"UdfA%03d%s\nNdfA%03d%s\n",
getpid() % 1000, localhost,
getpid() % 1000, localhost);
fprintf(stderr, "DEBUG: Control file is:\n%s", control);
if (order == ORDER_CONTROL_DATA)
{
lpd_command(fd, "\002%d cfA%03.3d%s\n", strlen(control), getpid() % 1000,
localhost);
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 if (read(fd, &status, 1) < 1)
status = errno;
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)
{
lpd_command(fd, "\003%u dfA%03.3d%s\n", (unsigned)filestats.st_size,
getpid() % 1000, localhost);
fprintf(stderr, "INFO: Sending data file (%u bytes)\n",
(unsigned)filestats.st_size);
tbytes = 0;
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)
status = errno;
else if (recv(fd, &status, 1, 0) < 1)
status = errno;
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)
{
lpd_command(fd, "\002%d cfA%03.3d%s\n", strlen(control), getpid() % 1000,
localhost);
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 if (read(fd, &status, 1) < 1)
status = errno;
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);
return (status);
}
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