#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 <netat/appletalk.h>
#include <netat/atp.h>
#include <netat/ddp.h>
#include <netat/nbp.h>
#include <netat/pap.h>
#include <cups/http.h>
#include <AppleTalk/at_proto.h>
#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])
#define PAPPacketStr(x) \
((x) == AT_PAP_TYPE_OPEN_CONN) ? "PAP_OPEN_CONN" : \
((x) == AT_PAP_TYPE_OPEN_CONN_REPLY) ? "PAP_OPEN_CONN_REPLY" : \
((x) == AT_PAP_TYPE_SEND_DATA) ? "PAP_SEND_DATA" : \
((x) == AT_PAP_TYPE_DATA) ? "PAP_DATA" : \
((x) == AT_PAP_TYPE_TICKLE) ? "PAP_TICKLE" : \
((x) == AT_PAP_TYPE_CLOSE_CONN) ? "PAP_CLOSE_CONN" : \
((x) == AT_PAP_TYPE_CLOSE_CONN_REPLY) ? "PAP_CLOSE_CONN_REPLY" : \
((x) == AT_PAP_TYPE_SEND_STATUS) ? "PAP_SEND_STATUS" : \
((x) == AT_PAP_TYPE_SEND_STS_REPLY) ? "PAP_SEND_STS_REPLY" : \
((x) == AT_PAP_TYPE_READ_LW) ? "PAP_READ_LW" : \
"<Unknown>"
#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(int abort);
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, int id);
static void statusUpdate(u_char* status, u_char statusLen);
static int parseUri(const char* argv0, char* name, char* type, char* zone);
static int addPercentEscapes(const unsigned char* src, char* dst, int dstMax);
static int removePercentEscapes(const char* src, unsigned char* dst, int dstMax);
static int nbptuple_compare(const void *p1, const void *p2);
static int okayToUseAppleTalk(void);
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)
{
puts("network pap \"Unknown\" \"AppleTalk Printer Access Protocol (pap)\"");
return 0;
}
if (argc == 2 && strcmp(argv[1], "-discover") == 0)
{
listDevices();
return 0;
}
if (argc < 6 || argc > 7)
{
fprintf(stderr, "argc = %d\n", argc);
for (err = 0; err < argc; err++) {
fprintf(stderr, "%02d:%s\n", err, argv[err]);
}
fprintf(stderr, "Usage: pap job-id user title copies options [file]\n");
exit(EINVAL);
}
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)
{
fprintf(stderr, "ERROR: unable to open print file \"%s\": %s\n", argv[6], strerror(errno));
return (1);
}
copies = atoi(argv[4]);
}
parseUri(argv[0], 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 err = noErr;
int index;
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())
{
fprintf(stderr, "ERROR: AppleTalk disabled in System Preferences\n");
return -1;
}
if ((err = zip_getmyzone(ZIP_DEF_INTERFACE, &at_zone)) != 0)
{
perror("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';
err = 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)
{
perror("ERROR: Unable to lookup AppleTalk printers");
return numberFound;
}
if (numberFound >= MAX_PRINTERS)
fprintf(stderr, "WARNING: Adding only the first %d printers found", MAX_PRINTERS);
qsort(buf, numberFound, sizeof(at_nbptuple_t), nbptuple_compare);
for (index = 0; index < numberFound; index++)
{
memcpy(name, buf[index].enu_entity.object.str, MIN(buf[index].enu_entity.object.len, sizeof(name)-1));
name[MIN(buf[index].enu_entity.object.len, sizeof(name)-1)] = '\0';
if (addPercentEscapes(name, encodedName, sizeof(encodedName)) == 0)
{
printf("network pap://%s/%s/LaserWriter \"Unknown\" \"%s\"\n", encodedZone, encodedName, 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, index;
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;
at_nbptuple_t tuple;
at_inet_t sendDataAddr;
at_inet_t src;
at_resp_t resp;
int userdata, xo, reqlen;
u_short tid;
u_char bitmap;
int maxfdp1;
struct timeval timeout, *timeoutPtr;
u_char flowQuantum;
u_short recvSequence = 0;
time_t now,
connect_time,
elasped_time,
sleep_time,
connect_timeout = -1,
nextStatusTime = 0;
at_entity_t entity;
at_retry_t retry;
Boolean recoverableErrShown = false;
#if defined(HAVE_SIGACTION) && !defined(HAVE_SIGSET)
struct sigaction action;
#endif
if ((err = nbp_make_entity(&entity, name, type, zone)) != noErr)
{
fprintf(stderr, "ERROR: Unable to make AppleTalk address: %s\n", strerror(errno));
goto Exit;
}
connect_time = time(NULL);
retry.interval = 1;
retry.retries = 5;
retry.backoff = 0;
for (;;)
{
if (okayToUseAppleTalk())
{
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 (recoverableErrShown)
{
fprintf(stderr, "INFO: recovered: \n");
sleep(5);
recoverableErrShown = false;
}
if ((err = papOpen(&tuple, &gConnID, &gSockfd, &gSessionAddr, &flowQuantum)) == 0)
break;
fprintf(stderr, "WARNING: Unable to open \"%s:%s\": %s\n", name, zone, strerror(errno));
}
else
{
fprintf(stderr, "WARNING: recoverable: Printer not responding\n");
recoverableErrShown = true;
}
}
else
{
fprintf(stderr, "WARNING: recoverable: AppleTalk disabled in System Preferences.\n");
recoverableErrShown = true;
}
retry.retries = 3;
elasped_time = time(NULL) - connect_time;
if (connect_timeout == -1)
connect_timeout = connectTimeout();
if (connect_timeout && elasped_time > connect_timeout)
{
fprintf(stderr, "ERROR: Printer not responding\n");
err = ETIMEDOUT;
goto Exit;
}
else if (elasped_time < 30 )
sleep_time = 10;
else if (elasped_time < 60 )
sleep_time = 30;
else
sleep_time = 60;
fprintf(stderr, "DEBUG: sleeping %d seconds...\n", (int)sleep_time);
sleep(sleep_time);
}
#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
fprintf(stderr, "INFO: Sending data\n");
sendDataAddr = tuple.enu_addr;
if ((err = papSendRequest(gSockfd, &gSessionAddr, gConnID, AT_PAP_TYPE_TICKLE, 0, false, false)) < 0)
{
perror("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)
{
perror("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 (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 (gStatusInterval != 0)
{
now = time(NULL);
timeout.tv_sec = (nextStatusTime > now) ? nextStatusTime - now : 1;
}
if ((err = select(maxfdp1, &readSet, 0, 0, timeoutPtr)) < 0)
{
perror("ERROR: select");
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)
perror("WARNING: Unable to send PAP status request");
if (gStatusInterval)
nextStatusTime = time(NULL) + gStatusInterval;
}
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)
{
perror("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)
{
perror("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)
{
perror("ERROR: Unable to get PAP request");
break;
}
}
fprintf(stderr, "DEBUG: <- %s\n", PAPPacketStr(TYPE_OF(userdata)));
switch (TYPE_OF(userdata))
{
case AT_PAP_TYPE_SEND_STS_REPLY:
if (resp.bitmap & 1)
{
u_char *iov_base = (u_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;
recvSequence = SEQUENCE_NUM(userdata);
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, index=0; index < ATP_TRESP_MAX; index++)
{
if (resp.bitmap & (1 << index))
len += resp.resp[index].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;
write(fdout, sockBuffer, len);
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
{
fprintf(stderr, "WARNING: Printer sent unexpected EOF\n");
}
}
if ((err = papSendRequest(gSockfd, &gSessionAddr, gConnID, AT_PAP_TYPE_SEND_DATA, 0xFF, true, true)) < 0)
{
fprintf(stderr, "ERROR: Error %d sending PAPSendData resuest: %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)
{
fprintf(stderr, "WARNING: Printer sent unexpected EOF\n");
}
else
{
fprintf(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:
fprintf(stderr, "WARNING: Unexpected PAP packet of type %d\n", TYPE_OF(userdata));
break;
default:
fprintf(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(false);
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;
long tm;
unsigned char data[10], rdata[ATP_DATA_SIZE];
int userdata;
u_char *puserdata = (u_char *)&userdata;
at_socket socket = 0;
int waitTime;
int status;
at_resp_t resp;
at_retry_t retry;
if (tuple == NULL)
{
errno = EINVAL;
return -1;
}
fprintf(stderr, "INFO: Opening connection\n");
errno = 0;
result = 0;
*fd = atp_open(&socket);
if (*fd < 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] = socket;
data[1] = 8;
for (;;)
{
waitTime = time(NULL) - tm;
data[2] = waitTime >> 8;
data[3] = waitTime & 0xff;
fprintf(stderr, "DEBUG: -> %s\n", PAPPacketStr(AT_PAP_TYPE_OPEN_CONN));
status = atp_sendreq(*fd, &tuple->enu_addr, data, 4, userdata, 1, 0, 0, &resp, &retry, 0);
if (status < 0)
{
statusUpdate("Destination unreachable", 23);
result = EHOSTUNREACH;
errno = EHOSTUNREACH;
sleep(1);
goto Exit;
}
else
{
puserdata = (u_char *)&resp.userdata[0];
fprintf(stderr, "DEBUG: <- %s, status %d\n", PAPPacketStr(puserdata[1]), (int)*(short *)&rdata[2]);
if (puserdata[1] != AT_PAP_TYPE_OPEN_CONN_REPLY ||
(*(short *)&rdata[2] == 0 && (puserdata[0]&0xff) != *connID))
{
result = EINVAL;
errno = EINVAL;
goto Exit;
}
statusUpdate(&rdata[5], rdata[4] & 0xff);
if (*(short *)&rdata[2] == 0)
break;
}
sleep(1);
}
sessionAddr->net = tuple->enu_addr.net;
sessionAddr->node = tuple->enu_addr.node;
sessionAddr->socket = rdata[0];
*flowQuantum = rdata[1];
Exit:
if (result != 0)
{
atp_close(*fd);
*fd = 0;
}
return result;
}
static int papClose(int abort)
{
int fd, tmpID;
int result;
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", PAPPacketStr(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);
result = atp_sendreq(fd, &gSessionAddr, 0, 0, userdata, 1, 0, 0, &resp, &retry, 0);
result = 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 index;
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 (index = 0; index < flowQuantum; index++)
{
resp.userdata[index] = 0;
puserdata = (u_char *)&resp.userdata[index];
puserdata[PAP_CONNID] = connID;
puserdata[PAP_TYPE] = AT_PAP_TYPE_DATA;
puserdata[PAP_EOF] = eof ? 1 : 0;
resp.resp[index].iov_base = (caddr_t)data;
if (data)
data += AT_PAP_DATA_SIZE;
resp.resp[index].iov_len = MIN((int)len, (int)AT_PAP_DATA_SIZE);
len -= resp.resp[index].iov_len;
if (len == 0)
break;
}
resp.bitmap = (1 << (index + 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", PAPPacketStr(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;
*(u_short *)&puserdata[2] = pap_send_count;
}
else
*(u_short *)&puserdata[2] = 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 (int)tid;
}
int papCancelRequest(int sockfd, int id)
{
int len;
sigset_t sv, osv;
len = sizeof(int);
sigemptyset(&sv);
sigaddset(&sv, SIGIO);
sigprocmask(SIG_SETMASK, &sv, &osv);
if (at_send_to_dev(sockfd, AT_ATP_CANCEL_REQUEST, (char*)&id, &len) < 0)
{
sigprocmask(SIG_SETMASK, &osv, NULL);
return -1;
}
sigprocmask(SIG_SETMASK, &osv, NULL);
return 0;
}
#pragma mark -
void statusUpdate(u_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[255],
value[255],
*ptr;
int port;
int statusInterval;
method[0] = username[0] = hostname[0] = resource[0] = '\0';
port = 0;
httpSeparateApple(argv0, method, sizeof(method),
username, sizeof(username),
hostname, sizeof(hostname), &port,
resource, sizeof(resource), 0);
if ((options = strchr(resource, '?')) != NULL)
{
*options++ = '\0';
while (*options != '\0')
{
for (ptr = optionName; *options && *options != '=' && *options != '+'; )
*ptr++ = *options++;
*ptr = '\0';
value[0] = '\0';
if (*options == '=')
{
options ++;
for (ptr = value; *options && *options != '+';)
*ptr++ = *options++;
*ptr = '\0';
if (*options == '+')
options ++;
}
else if (*options == '+')
{
options ++;
}
if (strcasecmp(optionName, "waiteof") == 0)
{
if (strcasecmp(value, "on") == 0 ||
strcasecmp(value, "yes") == 0 ||
strcasecmp(value, "true") == 0)
{
gWaitEOF = true;
}
else if (strcasecmp(value, "off") == 0 ||
strcasecmp(value, "no") == 0 ||
strcasecmp(value, "false") == 0)
{
gWaitEOF = false;
}
else
{
fprintf(stderr, "WARNING: Boolean expected for waiteof option \"%s\"\n", value);
}
}
else if (strcasecmp(optionName, "status") == 0)
{
statusInterval = atoi(value);
if (value[0] < '0' || value[0] > '9' ||
statusInterval < 0)
{
fprintf(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 unsigned char* src, char* dst, int dstMax)
{
unsigned 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, unsigned char* dst, int dstMax)
{
int c;
const unsigned 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 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)
{
fprintf(stderr, "ERROR: There was a timeout error while sending data to the printer\n");
papClose(true);
_exit(1);
}