166 lines
2.7 KiB
C
166 lines
2.7 KiB
C
/*
|
|
*
|
|
* common routines.
|
|
*/
|
|
#pragma noroot
|
|
#pragma optimize 79
|
|
#pragma debug 0x8000
|
|
#pragma lint -1
|
|
|
|
#include <Memory.h>
|
|
#include <MiscTool.h>
|
|
#include <tcpip.h>
|
|
|
|
|
|
#include <stdio.h>
|
|
|
|
#include "options.h"
|
|
#include "prototypes.h"
|
|
#include "connection.h"
|
|
|
|
int read_binary(Word ipid, FILE *file, ReadBlock *dcb)
|
|
{
|
|
Word rv = 0;
|
|
|
|
|
|
if (dcb) dcb->transferCount = 0;
|
|
|
|
IncBusy();
|
|
TCPIPPoll();
|
|
DecBusy();
|
|
|
|
for(;;)
|
|
{
|
|
static char buffer[512];
|
|
rrBuff rb;
|
|
Word count;
|
|
Word tcount;
|
|
|
|
IncBusy();
|
|
rv = TCPIPReadTCP(ipid, 0, (Ref)buffer, 512, &rb);
|
|
DecBusy();
|
|
|
|
count = rb.rrBuffCount;
|
|
|
|
if (!count)
|
|
{
|
|
if (rv) break;
|
|
IncBusy();
|
|
TCPIPPoll();
|
|
DecBusy();
|
|
continue;
|
|
}
|
|
|
|
|
|
|
|
tcount = fwrite(buffer, 1, count, file);
|
|
if (dcb) dcb->transferCount += tcount;
|
|
|
|
if (tcount != count) return -1;
|
|
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
// ReadTCP will only return if the entire request is fulfilled or the connection closes,
|
|
// so it could just do that. For a large request, this is probably nicer.
|
|
int read_binary_size(Word ipid, FILE *file, ReadBlock *dcb)
|
|
{
|
|
Word rv = 0;
|
|
LongWord size;
|
|
if (!dcb) return -1;
|
|
|
|
dcb->transferCount = 0;
|
|
size = dcb->requestCount;
|
|
if (!size) return 0;
|
|
|
|
IncBusy();
|
|
TCPIPPoll();
|
|
DecBusy();
|
|
|
|
for(;;)
|
|
{
|
|
static char buffer[512];
|
|
rrBuff rb;
|
|
Word count;
|
|
Word tcount;
|
|
|
|
count = 512;
|
|
if (count > size) count = size;
|
|
|
|
IncBusy();
|
|
rv = TCPIPReadTCP(ipid, 0, (Ref)buffer, count, &rb);
|
|
DecBusy();
|
|
|
|
count = rb.rrBuffCount;
|
|
|
|
|
|
if (!count)
|
|
{
|
|
if (rv) return -1;
|
|
IncBusy();
|
|
TCPIPPoll();
|
|
DecBusy();
|
|
continue;
|
|
}
|
|
|
|
tcount = fwrite(buffer, 1, count, file);
|
|
|
|
dcb->transferCount += tcount;
|
|
size -= tcount;
|
|
|
|
if (tcount != count) return -1;
|
|
|
|
if (!size) return 0;
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
|
|
int ConnectLoop(char *host, Word port, Connection *connection)
|
|
{
|
|
LongWord qtick;
|
|
|
|
ConnectionInit(connection, MMStartUp(), flags._v ? DisplayCallback : NULL);
|
|
ConnectionOpenC(connection, host, port);
|
|
|
|
// 30 second timeout.
|
|
qtick = GetTick() + 30 * 60;
|
|
while (!ConnectionPoll(connection))
|
|
{
|
|
if (GetTick() >= qtick)
|
|
{
|
|
fprintf(stderr, "Connection timed out.\n");
|
|
|
|
IncBusy();
|
|
TCPIPAbortTCP(connection->ipid);
|
|
TCPIPLogout(connection->ipid);
|
|
DecBusy();
|
|
|
|
return 0;
|
|
}
|
|
}
|
|
|
|
if (connection->state != kConnectionStateConnected)
|
|
{
|
|
fprintf(stderr, "Unable to open host: %s:%u\n",
|
|
host,
|
|
port);
|
|
return 0;
|
|
}
|
|
|
|
return 1;
|
|
}
|
|
|
|
|
|
int CloseLoop(Connection *connection)
|
|
{
|
|
ConnectionClose(connection);
|
|
|
|
while (!ConnectionPoll(connection)) ; // wait for it to close.
|
|
|
|
return 1;
|
|
}
|