#ifdef MAC_OS_X_VERSION_MIN_REQUIRED
# undef MAC_OS_X_VERSION_MIN_REQUIRED
#endif
#define MAC_OS_X_VERSION_MIN_REQUIRED MAC_OS_X_VERSION_10_0
#include <config.h>
#include <stdio.h>
#include <stdlib.h>
#include <ctype.h>
#include <termios.h>
#include <unistd.h>
#include <assert.h>
#include <signal.h>
#include <sys/fcntl.h>
#include <sys/param.h>
#include <sys/time.h>
#include <sys/errno.h>
#include <cups/cups.h>
#include <cups/backend.h>
#include <cups/sidechannel.h>
#include <cups/i18n.h>
#include <netat/appletalk.h>
#include <netat/atp.h>
#include <netat/ddp.h>
#include <netat/nbp.h>
#include <netat/pap.h>
#include <libkern/OSByteOrder.h>
#ifdef HAVE_APPLETALK_AT_PROTO_H
# include <AppleTalk/at_proto.h>
#else
# define ZIP_DEF_INTERFACE NULL
enum { RUNNING, NOTLOADED, LOADED, OTHERERROR };
extern int atp_abort(int fd, at_inet_t *dest, u_short tid);
extern int atp_close(int fd);
extern int atp_getreq(int fd, at_inet_t *src, char *buf, int *len, int *userdata,
int *xo, u_short *tid, u_char *bitmap, int nowait);
extern int atp_getresp(int fd, u_short *tid, at_resp_t *resp);
extern int atp_look(int fd);
extern int atp_open(at_socket *sock);
extern int atp_sendreq(int fd, at_inet_t *dest, char *buf, int len,
int userdata, int xo, int xo_relt, u_short *tid,
at_resp_t *resp, at_retry_t *retry, int nowait);
extern int atp_sendrsp(int fd, at_inet_t *dest, int xo, u_short tid,
at_resp_t *resp);
extern int checkATStack();
extern int nbp_lookup(at_entity_t *entity, at_nbptuple_t *buf, int max,
at_retry_t *retry);
extern int nbp_make_entity(at_entity_t *entity, char *obj, char *type,
char *zone);
extern int zip_getmyzone(char *ifName, at_nvestr_t *zone);
#endif
#include <CoreFoundation/CFURL.h>
#include <CoreFoundation/CFNumber.h>
#include <CoreFoundation/CFPreferences.h>
#define MAX_PRINTERS 500
#define PAP_CONNID 0
#define PAP_TYPE 1
#define PAP_EOF 2
#define CONNID_OF(p) (((u_char *)&p)[0])
#define TYPE_OF(p) (((u_char *)&p)[1])
#define SEQUENCE_NUM(p) (((u_char *)&p)[2])
#define IS_PAP_EOF(p) (((u_char *)&p)[2])
#ifndef true
#define true 1
#define false 0
#endif
int gSockfd = 0;
at_inet_t gSessionAddr = { 0 };
u_char gConnID = 0;
u_short gSendDataID = 0;
u_short gTickleID = 0;
int gWaitEOF = false;
int gStatusInterval= 5;
int gErrorlogged = false;
int gDebug = 0;
static int listDevices(void);
static int printFile(char* name, char* type, char* zone, int fdin, int fdout,
int fderr, int copies, int argc);
static int papOpen(at_nbptuple_t* tuple, u_char* connID, int* fd,
at_inet_t* pap_to, u_char* flowQuantum);
static int papClose();
static int papWrite(int sockfd, at_inet_t* dest, u_short tid, u_char connID,
u_char flowQuantum, char* data, int len, int eof);
static int papCloseResp(int sockfd, at_inet_t* dest, int xo, u_short tid,
u_char connID);
static int papSendRequest(int sockfd, at_inet_t* dest, u_char connID,
int function, u_char bitmap, int xo, int seqno);
static int papCancelRequest(int sockfd, u_short tid);
static void sidechannel_request();
static void statusUpdate(char* status, u_char statusLen);
static int parseUri(const char* argv0, char* name, char* type, char* zone);
static int addPercentEscapes(const char* src, char* dst, int dstMax);
static int removePercentEscapes(const char* src, char* dst, int dstMax);
static int nbptuple_compare(const void *p1, const void *p2);
static int okayToUseAppleTalk(void);
static const char *packet_name(u_char x);
static int connectTimeout(void);
static void signalHandler(int sigraised);
int main (int argc, const char * argv[])
{
int err = 0;
FILE *fp;
int copies;
char name[NBP_NVE_STR_SIZE + 1];
char type[NBP_NVE_STR_SIZE + 1];
char zone[NBP_NVE_STR_SIZE + 1];
setbuf(stderr, NULL);
if (argc == 1 || (argc == 2 && strcmp(argv[1], "-discover") == 0))
{
listDevices();
return 0;
}
if (argc < 6 || argc > 7)
{
_cupsLangPrintf(stderr,
_("Usage: %s job-id user title copies options [file]\n"),
argv[0]);
return (CUPS_BACKEND_FAILED);
}
if (argc == 6)
{
fp = stdin;
copies = 1;
}
else
{
fprintf(stderr, "DEBUG: opening print file \"%s\"\n", argv[6]);
if ((fp = fopen(argv[6], "rb")) == NULL)
{
_cupsLangPrintf(stderr,
_("ERROR: Unable to open print file \"%s\": %s\n"),
argv[6], strerror(errno));
return (CUPS_BACKEND_FAILED);
}
copies = atoi(argv[4]);
}
parseUri(cupsBackendDeviceURI((char **)argv), name, type, zone);
err = printFile(name, type, zone, fileno(fp), STDOUT_FILENO, STDERR_FILENO, copies, argc);
if (fp != stdin)
fclose(fp);
if (err == noErr && !gErrorlogged)
fprintf(stderr, "INFO:\n");
return err;
}
static int listDevices(void)
{
int i;
int numberFound;
at_nvestr_t at_zone;
at_entity_t entity;
at_nbptuple_t buf[MAX_PRINTERS];
at_retry_t retry;
char name[NBP_NVE_STR_SIZE+1];
char encodedName[(3 * NBP_NVE_STR_SIZE) + 1];
char zone[NBP_NVE_STR_SIZE+1];
char encodedZone[(3 * NBP_NVE_STR_SIZE) + 1];
if (!okayToUseAppleTalk())
{
_cupsLangPuts(stderr, _("INFO: AppleTalk disabled in System Preferences\n"));
return -1;
}
if (zip_getmyzone(ZIP_DEF_INTERFACE, &at_zone))
{
_cupsLangPrintError(_("ERROR: Unable to get default AppleTalk zone"));
return -2;
}
memcpy(zone, at_zone.str, MIN(at_zone.len, sizeof(zone)-1));
zone[MIN(at_zone.len, sizeof(zone)-1)] = '\0';
_cupsLangPrintf(stderr, _("INFO: Using default AppleTalk zone \"%s\"\n"),
zone);
addPercentEscapes(zone, encodedZone, sizeof(encodedZone));
nbp_make_entity(&entity, "=", "LaserWriter", zone);
retry.retries = 1;
retry.interval = 1;
retry.backoff = 1;
if ((numberFound = nbp_lookup(&entity, buf, MAX_PRINTERS, &retry)) < 0)
{
_cupsLangPrintError(_("ERROR: Unable to lookup AppleTalk printers"));
return numberFound;
}
if (numberFound >= MAX_PRINTERS)
_cupsLangPrintf(stderr,
_("WARNING: Adding only the first %d printers found"),
MAX_PRINTERS);
qsort(buf, numberFound, sizeof(at_nbptuple_t), nbptuple_compare);
for (i = 0; i < numberFound; i++)
{
memcpy(name, buf[i].enu_entity.object.str, MIN(buf[i].enu_entity.object.len, sizeof(name)-1));
name[MIN(buf[i].enu_entity.object.len, sizeof(name)-1)] = '\0';
if (addPercentEscapes(name, encodedName, sizeof(encodedName)) == 0)
{
char make_model[128],
*ptr;
if ((ptr = strchr(name, ' ')) != NULL)
{
if (!strncmp(name, "ET00", 4))
{
strlcpy(make_model, ptr + 1, sizeof(make_model));
}
else
strlcpy(make_model, name, sizeof(make_model));
}
else
strcpy(make_model, "Unknown");
printf("network pap://%s/%s/LaserWriter \"%s\" \"%s AppleTalk\"\n",
encodedZone, encodedName, make_model, name);
}
}
return numberFound;
}
static int printFile(char* name, char* type, char* zone, int fdin, int fdout, int fderr, int copies, int argc)
{
int err;
int rc;
int val;
int len, i;
char fileBuffer[4096];
int fileBufferNbytes;
off_t fileTbytes;
int fileEOFRead;
int fileEOFSent;
char sockBuffer[4096 + 1];
char atpReqBuf[AT_PAP_DATA_SIZE];
fd_set readSet;
int use_sidechannel;
at_nbptuple_t tuple;
at_inet_t sendDataAddr;
at_inet_t src;
at_resp_t resp;
int userdata, xo = 0, reqlen;
u_short tid;
u_char bitmap;
int maxfdp1,
nbp_failures = 0;
struct timeval timeout, *timeoutPtr;
u_char flowQuantum = 1;
time_t now,
start_time,
elasped_time,
sleep_time,
connect_timeout = -1,
nextStatusTime = 0;
at_entity_t entity;
at_retry_t retry;
#if defined(HAVE_SIGACTION) && !defined(HAVE_SIGSET)
struct sigaction action;
#endif
FD_ZERO(&readSet);
FD_SET(CUPS_SC_FD, &readSet);
timeout.tv_sec = 0;
timeout.tv_usec = 0;
if ((select(CUPS_SC_FD+1, &readSet, NULL, NULL, &timeout)) >= 0)
use_sidechannel = 1;
else
use_sidechannel = 0;
if ((err = nbp_make_entity(&entity, name, type, zone)) != noErr)
{
_cupsLangPrintError(_("ERROR: Unable to make AppleTalk address"));
goto Exit;
}
start_time = time(NULL);
retry.interval = 1;
retry.retries = 5;
retry.backoff = 0;
fprintf(stderr, "STATE: +connecting-to-device\n");
for (;;)
{
if (okayToUseAppleTalk())
{
fprintf(stderr, "STATE: -apple-appletalk-disabled-warning\n");
if ((err = nbp_lookup(&entity, &tuple, 1, &retry)) > 0)
{
if (err > 1)
fprintf(stderr, "DEBUG: Found more than one printer with the name \"%s\"\n", name);
if (nbp_failures)
{
fprintf(stderr, "STATE: -apple-nbp-lookup-warning\n");
nbp_failures = 0;
}
if ((err = papOpen(&tuple, &gConnID, &gSockfd, &gSessionAddr, &flowQuantum)) == 0)
break;
_cupsLangPrintf(stderr, _("WARNING: Unable to open \"%s:%s\": %s\n"),
name, zone, strerror(err));
}
else
{
if (++nbp_failures > 2)
{
retry.interval = 2;
retry.retries = 3;
fprintf(stderr, "STATE: +apple-nbp-lookup-warning\n");
_cupsLangPuts(stderr, _("WARNING: Printer not responding\n"));
}
}
}
else
{
fprintf(stderr, "STATE: +apple-appletalk-disabled-warning\n");
_cupsLangPuts(stderr,
_("INFO: AppleTalk disabled in System Preferences.\n"));
}
elasped_time = time(NULL) - start_time;
if (connect_timeout == -1)
connect_timeout = connectTimeout();
if (connect_timeout && elasped_time > connect_timeout)
{
_cupsLangPuts(stderr, _("ERROR: Printer not responding\n"));
err = ETIMEDOUT;
goto Exit;
}
else if (elasped_time < (30 * 60))
sleep_time = 10;
else if (elasped_time < (24 * 60 * 60))
sleep_time = 30;
else
sleep_time = 60;
fprintf(stderr, "DEBUG: sleeping %d seconds...\n", (int)sleep_time);
sleep(sleep_time);
}
fprintf(stderr, "STATE: -connecting-to-device\n");
#ifdef HAVE_SIGSET
sigset(SIGTERM, (argc < 7) ? SIG_IGN : signalHandler);
#elif defined(HAVE_SIGACTION)
memset(&action, 0, sizeof(action));
sigemptyset(&action.sa_mask);
action.sa_handler = (argc < 7) ? SIG_IGN : signalHandler;
sigaction(SIGTERM, &action, NULL);
#else
signal(SIGTERM, (argc < 7) ? SIG_IGN : signalHandler);
#ifdef DEBUG
signal(SIGINT, signalHandler);
#endif
#endif
_cupsLangPuts(stderr, _("INFO: Sending data\n"));
sendDataAddr = tuple.enu_addr;
if ((err = papSendRequest(gSockfd, &gSessionAddr, gConnID, AT_PAP_TYPE_TICKLE, 0, false, false)) < 0)
{
_cupsLangPrintError(_("ERROR: Unable to send PAP tickle request"));
goto Exit;
}
signal(SIGALRM, signalHandler);
alarm(PAP_TIMEOUT);
if ((err = papSendRequest(gSockfd, &gSessionAddr, gConnID, AT_PAP_TYPE_SEND_DATA, 0xFF, true, true)) < 0)
{
_cupsLangPrintError(_("ERROR: Unable to send initial PAP send data request"));
goto Exit;
}
val = fcntl(fdin, F_GETFL, 0);
fcntl(fdin, F_SETFL, val | O_NONBLOCK);
fileBufferNbytes = 0;
fileTbytes = 0;
fileEOFRead = fileEOFSent = false;
maxfdp1 = MAX(fdin, gSockfd) + 1;
if (use_sidechannel && CUPS_SC_FD >= maxfdp1)
maxfdp1 = CUPS_SC_FD + 1;
if (gStatusInterval != 0)
{
timeout.tv_usec = 0;
nextStatusTime = time(NULL) + gStatusInterval;
timeoutPtr = &timeout;
}
else
timeoutPtr = NULL;
for (;;)
{
FD_ZERO(&readSet);
FD_SET(gSockfd, &readSet);
if (fileBufferNbytes == 0 && fileEOFRead == false)
FD_SET(fdin, &readSet);
if (use_sidechannel)
FD_SET(CUPS_SC_FD, &readSet);
if (gStatusInterval != 0)
{
now = time(NULL);
timeout.tv_sec = (nextStatusTime > now) ? nextStatusTime - now : 1;
}
if ((err = select(maxfdp1, &readSet, 0, 0, timeoutPtr)) < 0)
{
_cupsLangPrintError(_("ERROR: select() failed"));
break;
}
if (err == 0 || (gStatusInterval != 0 && time(NULL) >= nextStatusTime))
{
if ((err = papSendRequest(gSockfd, &tuple.enu_addr, 0, AT_PAP_TYPE_SEND_STATUS, 0x01, false, false)) < 0)
_cupsLangPrintError(_("WARNING: Unable to send PAP status request"));
if (gStatusInterval)
nextStatusTime = time(NULL) + gStatusInterval;
}
if (use_sidechannel && FD_ISSET(CUPS_SC_FD, &readSet))
sidechannel_request();
if (FD_ISSET(fdin, &readSet))
{
FD_CLR(fdin, &readSet);
assert(fileBufferNbytes == 0);
fileBufferNbytes = read(fdin, fileBuffer, MIN(sizeof(fileBuffer), AT_PAP_DATA_SIZE * flowQuantum));
if (fileBufferNbytes == 0)
fileEOFRead = true;
if (fileEOFSent == false && fileBufferNbytes >= 0 && gSendDataID != 0)
{
fprintf(stderr, "DEBUG: -> PAP_DATA %d bytes %s\n", fileBufferNbytes, fileEOFRead ? "with EOF" : "");
papWrite(gSockfd, &sendDataAddr, gSendDataID, gConnID, flowQuantum, fileBuffer, fileBufferNbytes, fileEOFRead);
fileTbytes += fileBufferNbytes;
if (argc > 6 && !gErrorlogged)
fprintf(stderr, "DEBUG: Sending print file, %qd bytes\n", (off_t)fileTbytes);
fileBufferNbytes = 0;
gSendDataID = 0;
if (fileEOFRead)
{
fileEOFSent = true;
if (gWaitEOF == false || fileTbytes == 0)
{
err = 0;
goto Exit;
}
}
}
}
if (FD_ISSET(gSockfd, &readSet))
{
if ((rc = atp_look(gSockfd)) < 0)
{
_cupsLangPrintError(_("ERROR: Unable to look for PAP response"));
break;
}
if (rc > 0)
{
resp.resp[0].iov_base = sockBuffer;
resp.resp[0].iov_len = sizeof(sockBuffer) - 1;
resp.bitmap = 0x01;
if ((err = atp_getresp(gSockfd, &tid, &resp)) < 0)
{
_cupsLangPrintError(_("ERROR: Unable to get PAP response"));
break;
}
userdata = resp.userdata[0];
}
else
{
reqlen = sizeof(atpReqBuf);
if ((err = atp_getreq(gSockfd, &src, atpReqBuf, &reqlen, &userdata, &xo, &tid, &bitmap, 0)) < 0)
{
_cupsLangPrintError(_("ERROR: Unable to get PAP request"));
break;
}
}
fprintf(stderr, "DEBUG: <- %s\n", packet_name(TYPE_OF(userdata)));
switch (TYPE_OF(userdata))
{
case AT_PAP_TYPE_SEND_STS_REPLY:
if (resp.bitmap & 1)
{
char *iov_base = (char *)resp.resp[0].iov_base;
statusUpdate(&iov_base[5], iov_base[4]);
}
break;
case AT_PAP_TYPE_SEND_DATA:
sendDataAddr.socket = src.socket;
gSendDataID = tid;
OSReadBigInt16(&SEQUENCE_NUM(userdata), 0);
if ((fileBufferNbytes > 0 || fileEOFRead) && fileEOFSent == false)
{
fprintf(stderr, "DEBUG: -> PAP_DATA %d bytes %s\n", fileBufferNbytes, fileEOFRead ? "with EOF" : "");
papWrite(gSockfd, &sendDataAddr, gSendDataID, gConnID, flowQuantum, fileBuffer, fileBufferNbytes, fileEOFRead);
fileTbytes += fileBufferNbytes;
if (argc > 6 && !gErrorlogged)
fprintf(stderr, "DEBUG: Sending print file, %qd bytes\n", (off_t)fileTbytes);
fileBufferNbytes = 0;
gSendDataID = 0;
if (fileEOFRead)
{
fileEOFSent = true;
if (gWaitEOF == false)
{
err = 0;
goto Exit;
}
}
}
break;
case AT_PAP_TYPE_DATA:
for (len=0, i=0; i < ATP_TRESP_MAX; i++)
{
if (resp.bitmap & (1 << i))
len += resp.resp[i].iov_len;
}
fprintf(stderr, "DEBUG: <- PAP_DATA %d bytes %s\n", len, IS_PAP_EOF(userdata) ? "with EOF" : "");
if (len > 0)
{
char *pLineBegin, *pCommentEnd, *pChar;
char *logLevel;
char logstr[512];
int logstrlen;
cupsBackChannelWrite(sockBuffer, len, 1.0);
sockBuffer[len] = '\0';
pLineBegin = sockBuffer;
while (pLineBegin < sockBuffer + len &&
(pLineBegin = strstr(pLineBegin, "%%[")) != NULL &&
(pCommentEnd = strstr(pLineBegin, "]%%")) != NULL)
{
pCommentEnd += 3;
*pCommentEnd = '\0';
for (pChar = pLineBegin; pChar < pCommentEnd; pChar++)
if (*pChar == '\r' || *pChar == '\n')
*pChar = ' ';
if (strncasecmp(pLineBegin, "%%[ Error:", 10) == 0)
{
logLevel = "DEBUG";
gErrorlogged = true;
}
else if (strncasecmp(pLineBegin, "%%[ Flushing", 12) == 0)
logLevel = "DEBUG";
else
logLevel = "INFO";
if ((logstrlen = snprintf(logstr, sizeof(logstr), "%s: %s\n", logLevel, pLineBegin)) >= sizeof(logstr))
{
logstrlen = sizeof(logstr) - 1;
logstr[logstrlen - 1] = '\n';
}
write(fderr, logstr, logstrlen);
pLineBegin = pCommentEnd + 1;
}
}
if (IS_PAP_EOF(userdata) != 0)
{
if (fileEOFSent == true)
goto Exit;
else
{
_cupsLangPuts(stderr, _("WARNING: Printer sent unexpected EOF\n"));
}
}
if ((err = papSendRequest(gSockfd, &gSessionAddr, gConnID, AT_PAP_TYPE_SEND_DATA, 0xFF, true, true)) < 0)
{
_cupsLangPrintf(stderr,
_("ERROR: Error %d sending PAPSendData request: %s\n"),
err, strerror(errno));
goto Exit;
}
break;
case AT_PAP_TYPE_TICKLE:
break;
case AT_PAP_TYPE_CLOSE_CONN:
papCloseResp(gSockfd, &gSessionAddr, xo, tid, gConnID);
if (fileEOFSent == true)
{
_cupsLangPuts(stderr, _("WARNING: Printer sent unexpected EOF\n"));
}
else
{
_cupsLangPuts(stderr, _("ERROR: Printer sent unexpected EOF\n"));
}
goto Exit;
break;
case AT_PAP_TYPE_OPEN_CONN:
case AT_PAP_TYPE_OPEN_CONN_REPLY:
case AT_PAP_TYPE_SEND_STATUS:
case AT_PAP_TYPE_CLOSE_CONN_REPLY:
_cupsLangPrintf(stderr, _("WARNING: Unexpected PAP packet of type %d\n"),
TYPE_OF(userdata));
break;
default:
_cupsLangPrintf(stderr, _("WARNING: Unknown PAP packet of type %d\n"),
TYPE_OF(userdata));
break;
}
if (CONNID_OF(userdata) == gConnID)
{
alarm(0);
alarm(PAP_TIMEOUT);
}
}
}
Exit:
papClose();
return err;
}
#pragma mark -
static int papOpen(at_nbptuple_t* tuple, u_char* connID, int* fd,
at_inet_t* sessionAddr, u_char* flowQuantum)
{
int result,
open_result,
userdata;
time_t tm,
waitTime;
char data[10],
rdata[ATP_DATA_SIZE];
u_char *puserdata;
at_socket socketfd;
at_resp_t resp;
at_retry_t retry;
result = 0;
socketfd = 0;
puserdata = (u_char *)&userdata;
_cupsLangPuts(stderr, _("INFO: Opening connection\n"));
if ((*fd = atp_open(&socketfd)) < 0)
return -1;
tm = time(NULL);
srand(tm);
*connID = (rand()&0xff) | 0x01;
puserdata[0] = *connID;
puserdata[1] = AT_PAP_TYPE_OPEN_CONN;
puserdata[2] = 0;
puserdata[3] = 0;
retry.interval = 2;
retry.retries = 5;
resp.bitmap = 0x01;
resp.resp[0].iov_base = rdata;
resp.resp[0].iov_len = sizeof(rdata);
data[0] = socketfd;
data[1] = 8;
for (;;)
{
waitTime = time(NULL) - tm;
OSWriteBigInt16(&data[2], 0, (u_short)waitTime);
fprintf(stderr, "DEBUG: -> %s\n", packet_name(AT_PAP_TYPE_OPEN_CONN));
if (atp_sendreq(*fd, &tuple->enu_addr, data, 4, userdata, 1, 0,
0, &resp, &retry, 0) < 0)
{
statusUpdate("Destination unreachable", 23);
result = EHOSTUNREACH;
break;
}
puserdata = (u_char *)&resp.userdata[0];
open_result = OSReadBigInt16(&rdata[2], 0);
fprintf(stderr, "DEBUG: <- %s, status %d\n", packet_name(puserdata[1]),
open_result);
if (puserdata[1] != AT_PAP_TYPE_OPEN_CONN_REPLY ||
(open_result == 0 && (puserdata[0] & 0xff) != *connID))
{
result = EINVAL;
break;
}
statusUpdate(&rdata[5], rdata[4] & 0xff);
if (open_result == 0)
break;
sleep(1);
}
if (result == 0)
{
sessionAddr->net = tuple->enu_addr.net;
sessionAddr->node = tuple->enu_addr.node;
sessionAddr->socket = rdata[0];
*flowQuantum = rdata[1];
}
else
{
atp_close(*fd);
*fd = 0;
sleep(1);
}
return result;
}
static int papClose()
{
int fd;
u_short tmpID;
unsigned char rdata[ATP_DATA_SIZE];
int userdata;
u_char *puserdata = (u_char *)&userdata;
at_resp_t resp;
at_retry_t retry;
if (gSockfd != 0)
{
fd = gSockfd;
gSockfd = 0;
alarm(0);
if (gSendDataID)
{
tmpID = gSendDataID;
gSendDataID = 0;
papCancelRequest(fd, tmpID);
}
if (gTickleID)
{
tmpID = gTickleID;
gTickleID = 0;
papCancelRequest(fd, tmpID);
}
if (gWaitEOF == false)
sleep(2);
fprintf(stderr, "DEBUG: -> %s\n", packet_name(AT_PAP_TYPE_CLOSE_CONN));
puserdata[0] = gConnID;
puserdata[1] = AT_PAP_TYPE_CLOSE_CONN;
puserdata[2] = 0;
puserdata[3] = 0;
retry.interval = 2;
retry.retries = 5;
resp.bitmap = 0x01;
resp.resp[0].iov_base = rdata;
resp.resp[0].iov_len = sizeof(rdata);
atp_sendreq(fd, &gSessionAddr, 0, 0, userdata, 1, 0, 0, &resp, &retry, 0);
close(fd);
}
return noErr;
}
static int papWrite(int sockfd, at_inet_t* dest, u_short tid, u_char connID, u_char flowQuantum, char* data, int len, int eof)
{
int result;
int i;
u_char* puserdata;
at_resp_t resp;
if (len > AT_PAP_DATA_SIZE * flowQuantum)
{
fprintf(stderr, "DEBUG: papWrite() len of %d is too big!\n", len);
errno = E2BIG;
return -1;
}
for (i = 0; i < flowQuantum; i++)
{
resp.userdata[i] = 0;
puserdata = (u_char *)&resp.userdata[i];
puserdata[PAP_CONNID] = connID;
puserdata[PAP_TYPE] = AT_PAP_TYPE_DATA;
puserdata[PAP_EOF] = eof ? 1 : 0;
resp.resp[i].iov_base = (caddr_t)data;
if (data)
data += AT_PAP_DATA_SIZE;
resp.resp[i].iov_len = MIN((int)len, (int)AT_PAP_DATA_SIZE);
len -= resp.resp[i].iov_len;
if (len == 0)
break;
}
resp.bitmap = (1 << (i + 1)) - 1;
errno = 0;
if ((result = atp_sendrsp(sockfd, dest, true, tid, &resp)) < 0)
{
fprintf(stderr, "DEBUG: atp_sendrsp() returns %d, errno %d \"%s\"\n", result, errno, strerror(errno));
return -1;
}
return(0);
}
static int papCloseResp(int sockfd, at_inet_t* dest, int xo, u_short tid, u_char connID)
{
int result;
at_resp_t resp;
resp.bitmap = 1;
resp.userdata[0] = 0;
((u_char*)&resp.userdata[0])[PAP_CONNID] = connID;
((u_char*)&resp.userdata[0])[PAP_TYPE] = AT_PAP_TYPE_CLOSE_CONN_REPLY;
resp.resp[0].iov_base = NULL;
resp.resp[0].iov_len = 0;
if ((result = atp_sendrsp(sockfd, dest, xo, tid, &resp)) < 0)
{
fprintf(stderr, "DEBUG: atp_sendrsp() returns %d, errno %d \"%s\"\n", result, errno, strerror(errno));
return -1;
}
return 0;
}
static int papSendRequest(int sockfd, at_inet_t* dest, u_char connID, int function, u_char bitmap, int xo, int seqno)
{
u_short tid;
int err;
sigset_t sv, osv;
int userdata;
u_char *puserdata = (u_char *)&userdata;
at_retry_t retry;
at_resp_t resp;
static u_short pap_send_count = 0;
fprintf(stderr, "DEBUG: -> %s\n", packet_name(function));
puserdata[0] = connID;
puserdata[1] = function;
resp.bitmap = bitmap;
retry.interval = 10;
retry.retries = -1;
if (seqno)
{
pap_send_count++;
if (pap_send_count == 0)
pap_send_count = 1;
OSWriteBigInt16(&puserdata[2], 0, pap_send_count);
}
else
OSWriteBigInt16(&puserdata[2], 0, 0);
sigemptyset(&sv);
sigaddset(&sv, SIGIO);
sigprocmask(SIG_SETMASK, &sv, &osv);
err = atp_sendreq(sockfd, dest, 0, 0, userdata, xo, 0, &tid, &resp, &retry, 1);
sigprocmask(SIG_SETMASK, &osv, NULL);
return err;
}
int papCancelRequest(int sockfd, u_short tid)
{
sigset_t sv, osv;
sigemptyset(&sv);
sigaddset(&sv, SIGIO);
sigprocmask(SIG_SETMASK, &sv, &osv);
if (atp_abort(sockfd, NULL, tid) < 0)
{
sigprocmask(SIG_SETMASK, &osv, NULL);
return -1;
}
sigprocmask(SIG_SETMASK, &osv, NULL);
return 0;
}
static int
sidechannel_request()
{
cups_sc_command_t command;
cups_sc_status_t status;
char data[2048];
int datalen;
datalen = sizeof(data);
if (cupsSideChannelRead(&command, &status, data, &datalen, 1.0))
return (-1);
switch (command)
{
case CUPS_SC_CMD_GET_BIDI:
data[0] = 1;
return (cupsSideChannelWrite(command, CUPS_SC_STATUS_OK, data, 1, 1.0));
break;
case CUPS_SC_CMD_GET_STATE:
data[0] = CUPS_SC_STATE_ONLINE;
return (cupsSideChannelWrite(command, CUPS_SC_STATUS_OK, data, 1, 1.0));
break;
case CUPS_SC_CMD_DRAIN_OUTPUT:
case CUPS_SC_CMD_SOFT_RESET:
case CUPS_SC_CMD_GET_DEVICE_ID:
default:
return (cupsSideChannelWrite(command, CUPS_SC_STATUS_NOT_IMPLEMENTED,
NULL, 0, 1.0));
break;
}
return (0);
}
#pragma mark -
void statusUpdate(char* status, u_char statusLen)
{
static char status_str[255];
static u_char last_statusLen = 0xFF;
if (statusLen != last_statusLen || memcmp(status, status_str, statusLen) != 0)
{
if (statusLen > sizeof(status_str)-1)
statusLen = sizeof(status_str)-1;
last_statusLen = statusLen;
memcpy(status_str, status, statusLen);
status_str[(int)statusLen] = '\0';
if (statusLen > 3 && memcmp(status, "%%[", 3) == 0)
fprintf(stderr, "INFO: %s\n", status_str);
else
fprintf(stderr, "INFO: %%%%[ %s ]%%%%\n", status_str);
}
return;
}
static int parseUri(const char* argv0, char* name, char* type, char* zone)
{
char method[255],
hostname[1024],
username[255],
resource[1024],
*resourcePtr,
*typePtr,
*options,
*optionName,
*value,
sep;
int port;
int statusInterval;
method[0] = username[0] = hostname[0] = resource[0] = '\0';
port = 0;
httpSeparateURI(HTTP_URI_CODING_NONE, argv0, method, sizeof(method),
username, sizeof(username),
hostname, sizeof(hostname), &port,
resource, sizeof(resource));
if ((options = strchr(resource, '?')) != NULL)
{
*options++ = '\0';
while (*options != '\0')
{
optionName = options;
while (*options && *options != '=' && *options != '+' && *options != '&')
options ++;
if ((sep = *options) != '\0')
*options++ = '\0';
if (sep == '=')
{
value = options;
while (*options && *options != '+' && *options != '&')
options ++;
if (*options)
*options++ = '\0';
}
else
value = (char *)"";
if (!strcasecmp(optionName, "waiteof"))
{
if (!strcasecmp(value, "on") ||
!strcasecmp(value, "yes") ||
!strcasecmp(value, "true"))
{
gWaitEOF = true;
}
else if (!strcasecmp(value, "off") ||
!strcasecmp(value, "no") ||
!strcasecmp(value, "false"))
{
gWaitEOF = false;
}
else
{
_cupsLangPrintf(stderr,
_("WARNING: Boolean expected for waiteof option \"%s\"\n"),
value);
}
}
else if (!strcasecmp(optionName, "status"))
{
statusInterval = atoi(value);
if (value[0] < '0' || value[0] > '9' || statusInterval < 0)
{
_cupsLangPrintf(stderr,
_("WARNING: number expected for status option \"%s\"\n"),
value);
}
else
{
gStatusInterval = statusInterval;
}
}
}
}
resourcePtr = resource;
if (*resourcePtr == '/')
resourcePtr++;
typePtr = strchr(resourcePtr, '/');
if (typePtr != NULL)
{
*typePtr++ = '\0';
}
else
{
typePtr = "LaserWriter";
}
removePercentEscapes(hostname, zone, NBP_NVE_STR_SIZE + 1);
removePercentEscapes(resourcePtr, name, NBP_NVE_STR_SIZE + 1);
removePercentEscapes(typePtr, type, NBP_NVE_STR_SIZE + 1);
return 0;
}
static int addPercentEscapes(const char* src, char* dst, int dstMax)
{
char c;
char *dstEnd = dst + dstMax - 1;
while (*src)
{
c = *src++;
if ((c >= 'a' && c <= 'z') || (c >= 'A' && c <= 'Z') ||
(c >= '0' && c <= '9') || (c == '.' || c == '-' || c == '*' || c == '_'))
{
if (dst >= dstEnd)
return -1;
*dst++ = c;
}
else
{
if (dst >= dstEnd - 2)
return -1;
snprintf(dst, dstEnd - dst, "%%%02x", c);
dst += 3;
}
}
*dst = '\0';
return 0;
}
static int removePercentEscapes(const char* src, char* dst, int dstMax)
{
int c;
const char *dstEnd = dst + dstMax;
while (*src && dst < dstEnd)
{
c = *src++;
if (c == '%')
{
sscanf(src, "%02x", &c);
src += 2;
}
*dst++ = (char)c;
}
if (dst >= dstEnd)
return -1;
*dst = '\0';
return 0;
}
int nbptuple_compare(const void *p1, const void *p2)
{
int result;
int len = MIN(((at_nbptuple_t*)p1)->enu_entity.object.len,
((at_nbptuple_t*)p2)->enu_entity.object.len);
if ((result = memcmp(((at_nbptuple_t*)p1)->enu_entity.object.str, ((at_nbptuple_t*)p2)->enu_entity.object.str, len)) == 0)
{
if (((at_nbptuple_t*)p1)->enu_entity.object.len < ((at_nbptuple_t*)p2)->enu_entity.object.len)
result = -1;
else if (((at_nbptuple_t*)p1)->enu_entity.object.len > ((at_nbptuple_t*)p2)->enu_entity.object.len)
result = 1;
else
result = 0;
}
return result;
}
static int okayToUseAppleTalk()
{
int atStatus = checkATStack();
return atStatus == RUNNING;
}
static const char *packet_name(u_char x)
{
switch (x)
{
case AT_PAP_TYPE_OPEN_CONN: return "PAP_OPEN_CONN";
case AT_PAP_TYPE_OPEN_CONN_REPLY: return "PAP_OPEN_CONN_REPLY";
case AT_PAP_TYPE_SEND_DATA: return "PAP_SEND_DATA";
case AT_PAP_TYPE_DATA: return "PAP_DATA";
case AT_PAP_TYPE_TICKLE: return "PAP_TICKLE";
case AT_PAP_TYPE_CLOSE_CONN: return "PAP_CLOSE_CONN";
case AT_PAP_TYPE_CLOSE_CONN_REPLY: return "PAP_CLOSE_CONN_REPLY";
case AT_PAP_TYPE_SEND_STATUS: return "PAP_SEND_STATUS";
case AT_PAP_TYPE_SEND_STS_REPLY: return "PAP_SEND_STS_REPLY";
case AT_PAP_TYPE_READ_LW: return "PAP_READ_LW";
}
return "<Unknown>";
}
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;
}
static void signalHandler(int sigraised)
{
_cupsLangPuts(stderr, _("ERROR: There was a timeout error while sending data to the printer\n"));
papClose();
_exit(1);
}