marignotti/marignotti.c

1050 lines
20 KiB
C

/*
* MariGNOtti
*
*/
#pragma optimize 79
#include <memory.h>
#include <tcpip.h>
#include <texttool.h>
#include <misctool.h>
#include <locator.h>
#include <stdio.h>
#include <stdlib.h>
#include <signal.h>
//#include <gno/kerntool.h>
#include <gno/gno.h>
#include <sys/errno.h>
#include <sys/socket.h>
#include <sys/ioctl.h>
#include <unistd.h>
#include <ctype.h>
#include <string.h>
//#include <netinet/in.h>
#include "net.h"
volatile int QuitFlag;
static Word MyID;
int semID;
// this will be used to hold extra info (sockopts, etc)
// as well as to keep a list of ipids waiting to close.
struct flags {
// shutdown(2)
Word _SHUT_RD:1;
Word _SHUT_WR:1;
// fcntl(2)
Word _NONBLOCK:1;
// setsockopt()
Word _OOBINLINE:1;
Word _LINGER:1;
Word _NOSIGPIPE:1;
};
typedef struct ipidHash {
Word ipid;
struct flags flags;
Word _SNDLOWAT;
Word _RCVLOWAT;
struct ipidHash *next;
} ipidHash;
ipidHash *htable[64] = { };
ipidHash *dlist = NULL;
ipidHash *findIpidHash(Word ipid)
{
ipidHash *e;
e = htable[ipid & (64 - 1)];
while (e)
{
if (e->ipid == ipid) return e;
e = e->next;
}
return NULL;
}
// not yet sure what this is for...
typedef void (*selwakeup)(int col_flag, int pid);
void debug_str(const char *cp)
{
static Word Sweet16 = -1;
if (Sweet16 == -1)
{
//volatile char *ereg = (char *)0x00c04f;
/*
// Apple II tech note 201
*/
word emu_id = 0;
word emu_version = 0;
asm {
// the lda ..,x is to prevent
// the 2nd read from being optimized out.
//brk 0xea
lda #0
ldx #0
sep #0x20
sta >0x00c04f,x
lda >0x00c04f,x
sta <emu_id
lda >0x00c04f,x
sta <emu_version
rep #0x20
}
if (emu_id == 0x16 && emu_version >= 0x23) Sweet16 = 1;
else Sweet16 = 0;
//*EMUBYTE = 0;
//if (*EMUBYTE == 0x16) Sweet16 = 1;
//else Sweet16 = 0;
#undef EMUBYTE
}
if (Sweet16 == 0) return;
asm {
ldx <cp+2
ldy <cp
cop 0x84
}
}
void dump_bytes(const char *bytes, Word length)
{
static const char *HexMap = "0123456789abcdef";
static char buffer[80];
while (length)
{
Word i, j, l;
l = 16;
if (l > length) l = length;
memset(buffer, ' ', sizeof(buffer));
for (i = 0, j = 0; i < l; ++i)
{
unsigned x = bytes[i];
buffer[j++] = HexMap[x >> 4];
buffer[j++] = HexMap[x & 0x0f];
j++;
if (i == 7) j++;
buffer[50 + i] = isascii(x) && isprint(x) ? x : '.';
}
buffer[50 + 16 + 1] = 0;
debug_str(buffer);
length -= l;
bytes += l;
}
}
void dump_srbuff(const srBuff *sb)
{
static char buffer[128];
sprintf(buffer, "%04x %04x %08lx %08lx %08lx %04x %04x %04x",
sb->srState,
sb->srNetworkError,
sb->srSndQueued,
sb->srRcvQueued,
sb->srDestIP,
sb->srDestPort,
sb->srConnectType,
sb->srAcceptCount
);
debug_str(buffer);
}
int do_attach(
int socknum,
void *m, size_t *m_len,
struct sockaddr *addr, int *addrlen,
void *rights
)
{
// returns errno or 0.
Word ipid;
ipidHash *e;
int type = socknum;
int *sockptr = (int *)m;
selwakeup *fx = (selwakeup *)m_len;
int *protocol = (int *)addr;
//WriteLine("\pdo_attach");
debug_str("do_attach");
if (type != SOCK_STREAM) return EPROTONOSUPPORT;
ipid = TCPIPLogin(MyID, 0, 0, 0, 0x0040);
if (_toolErr) return EPFNOSUPPORT;
// semaphore ...
swait(semID);
e = (ipidHash *)calloc(sizeof(ipidHash), 1);
e->ipid = ipid;
e->_RCVLOWAT = 1;
e->_SNDLOWAT = 1024;
e->flags._OOBINLINE = 1;
e->next = htable[ipid & (64-1)];
htable[ipid & (64 - 1)] = e;
ssignal(semID);
// end semaphore
*sockptr = ipid;
return 0;
}
// netinet/in.h is giving me compile errors...
struct xsockaddr_in {
short sin_family;
unsigned short sin_port;
unsigned long sin_addr;
char sin_zero[8];
};
int do_bind(
int socknum,
void *m, size_t *m_len,
struct sockaddr *addr, int *addrlen,
void *rights)
{
// freebsd 2.0 whois does a bind before
// the connect.
// this was removed in freebsd 3.0
Word ipid = (Word)socknum;
struct xsockaddr_in *sin = (struct xsockaddr_in *)addr;
port = sin->sin_port;
asm {
lda <port
xba
sta <port
}
// ?
// bind is usually called for the server side.
TCPIPSetNewDestination(ipid,
sin->sin_addr,
port);
if (_toolErr) return EPFNOSUPPORT;
return 0;
}
int do_connect(
int socknum,
void *m, size_t *m_len,
struct sockaddr *addr, int *addrlen,
void *rights)
{
// returns errno or 0.
Word terr;
Longword tick, qtick;
Word port;
Word ipid = (Word)socknum;
struct xsockaddr_in *sin = (struct xsockaddr_in *)addr;
//WriteLine("\pdo_connect");
debug_str("do_connect");
// port is byte-swapped.
port = sin->sin_port;
asm {
lda <port
xba
sta <port
}
TCPIPSetNewDestination(ipid,
sin->sin_addr,
port);
if (_toolErr) return EPFNOSUPPORT;
terr = TCPIPOpenTCP(ipid);
if (_toolErr || terr)
{
return EPFNOSUPPORT;
}
// wait to loop until actually connected.
// 30-second timeout.
qtick = GetTick() + 60 * 30;
for(;;)
{
srBuff sr;
terr = TCPIPStatusTCP(ipid, &sr);
dump_srbuff(&sr);
if (sr.srState == TCPSESTABLISHED)
return 0;
if (sr.srState == TCPSCLOSED)
return ECONNREFUSED;
if (sr.srState > TCPSESTABLISHED)
return ECONNREFUSED;
if (GetTick() >= qtick)
return ETIMEDOUT;
// is this safe?
sleep(1);
}
return 0;
}
int do_disconnect(
int socknum,
void *m, size_t *m_len,
struct sockaddr *addr, int *addrlen,
void *rights)
{
Word ipid = socknum;
debug_str("do_disconnect");
//WriteLine("\pdo_disconnect");
// TODO -- SO_LINGER / SO_LINGER_SEC
// monitor the out queue and don't return until all
// data is sent (or it times out).
// return value ignored.
return 0;
}
int do_detach(
int socknum,
void *m, size_t *m_len,
struct sockaddr *addr, int *addrlen,
void *rights)
{
Word ipid = socknum;
Word terr;
ipidHash *e;
ipidHash *prev;
// returns a GS/OS error.
// remove from hashtable.
// begin disconnect sequence.
// add to dlist.
// need to use semaphore to block other thread...
//WriteLine("\pdo_detach");
debug_str("do_detach");
debug_str("TCPIPCloseTCP");
terr = TCPIPCloseTCP(ipid);
// semaphore...
debug_str("swait");
swait(semID);
prev = NULL;
e = htable[ipid & (64-1)];
while (e)
{
if (e->ipid == ipid) break;
prev = e;
e = e->next;
}
if (e)
{
if (prev) prev->next = e->next;
else htable[ipid & (64 - 1)] = e->next;
e->next = dlist;
dlist = e;
}
debug_str("ssignal");
ssignal(semID);
// end semaphore.
debug_str("return 0");
return 0;
}
int do_send(
int socknum,
void *m, size_t *m_len,
struct sockaddr *addr, int *addrlen,
void *rights)
{
// called from GS/OS, busyflag set
// (seems like a bug to me....)
Word terr;
Word ipid = (Word)socknum;
void *data = (void *)m;
size_t *len = (size_t *)m_len;
//WriteLine("\pdo_send");
{
static char buffer[64];
sprintf(buffer, "do_send: ipid = %04x len = %04x",
ipid, (Word)*len
);
debug_str(buffer);
}
terr = TCPIPWriteTCP(ipid, data, *len, 0, 0);
// closing == eof.
if (terr == tcperrConClosing)
{
*len = 0;
return ECONNRESET; // ? EPIPE?
}
return 0;
}
int do_rcvd(
int socknum,
void *m, size_t *m_len,
struct sockaddr *addr, int *addrlen,
void *rights)
{
// called from GS/OS, busy flag is set.
// todo -- EWOULDBLOCK / EAGAIN
// Marinetti will not read less than the requested
// size unless there is a push or the connection is closing.
// TODO -- SO_RCVLOWAT (default is 1?)
// if SO_RCVLOWAT < queued data < requested length,
// read SO_RCVLOWAT bytes
Word terr;
Word ipid = (Word)socknum;
void *data = (void *)m;
size_t *len = (size_t *)m_len;
ipidHash *e;
rrBuff rr;
Word readCount;
Word minCount;
if (!len || !data) return EFAULT;
e = findIpidHash(ipid);
readCount = *len;
minCount = e ? e->_RCVLOWAT;
//WriteLine("\pdo_rcvd");
{
static char buffer[64];
sprintf(buffer, "do_rcvd: ipid = %04x len = %04x",
ipid, (Word)*len
);
debug_str(buffer);
}
// todo -- should block unless O_NONBLOCK set.
// todo -- could the read return 0 if there was a push at byte 0
// but data afterwards?
for (;;)
{
srBuff sr;
terr = TCPIPStatusTCP(ipid, &sr);
dump_srbuff(&sr);
if (!sr.srRcvQueued && e && e->flags._NONBLOCK)
{
*len = 0;
return EAGAIN;
}
//
if (sr.srRcvQueued < readCount && sr.srRcvQueued >= minCount)
{
readCount = minCount;
}
terr = TCPIPReadTCP(ipid, 0, (Ref)data, readCount, &rr);
if (rr.rrBuffCount)
{
{
static char buffer[80];
sprintf(buffer, "read %04x bytes %04x %04x %04x",
(Word)rr.rrBuffCount,
rr.rrMoreFlag,
rr.rrPushFlag,
rr.rrUrgentFlag
);
debug_str(buffer);
dump_bytes(data, rr.rrBuffCount);
}
*len = rr.rrBuffCount;
return 0;
}
if (terr == tcperrBadConnection)
return ENOTCONN;
// no data, no more data, then closing.
if (terr == tcperrConClosing && !rr.rrMoreFlag)
return ECONNRESET;
//asm {
// cop 0x7f
//}
// poll in main loop not called, apparently.
//TCPIPPoll(); //
sleep(1);
}
return 0;
}
int do_select(
int socknum,
void *m, size_t *m_len,
struct sockaddr *addr, int *addrlen,
void *rights)
{
ipidHash *e;
srBuff sr;
Word terr;
Word ipid = (Word)sock;
int *pid = (int *)m_len;
int *flag = (int *)(addr);
// *flag is return value as well as input value
// (which select type)
// SEL_READ = 0
// SEL_WRITE = 1
// SEL_EXCEPT = 2
if (*flag == 0)
{
// sel read.
terr = TCPIPStatusTCP(ipid, &sr);
if (sr.srRcvQueued)
*flag = 1;
else
*flag = 0;
return 0;
}
if (*flag == 1)
{
// sel write.
// check the high water mark.
terr = TCPIPStatusTCP(ipid, &sr);
e = findIpidHash(ipid);
if (e && sr.srSndQueued >= e->_SNDLOWAT)
*flag = 0;
else
*flag = 1;
}
if (*flag == 2)
{
// exceptions??
*flag = 0;
}
return 0;
}
int do_getsockopt(
Word ipid,
void *p1, void *p2,
void *p3, void *p4,
void *p5)
{
ipidHash *e;
int level = *(int *)p1;
int optname = *(int *)p2;
void *optval = (void *)p3;
int *optlen = (int *)p4;
if (!optval) return EINVAL;
e = findIpidHash(ipid);
switch (optname)
{
case SO_NREAD:
// return the amount of data available to read.
break;
case SO_NWRITE:
// return the amount of data queued for writing.
break;
case SO_TYPE:
*(Word *)optval = SOCK_STREAM;
return 0;
case SO_NOSIGPIPE:
if (e) *(Word *)optval = e->flags._NOSIGPIPE;
else *(Word *)optval = 0;
return 0;
break;
case SO_RCVLOWAT:
if (e) *(Word)optval = e->_RCVLOWAT;
else *(Word)optval = 1;
return 0;
case SO_SNDLOWAT:
if (e) *(Word)optval = e->_SNDLOWAT;
else *(Word)optval = 1024;
return 0;
case SO_LINGER:
if (e) *(Word *)optval = e->flags._LINGER;
else *(Word *)optval = 0;
return 0;
break;
case SO_OOBINLINE:
if (e) *(Word *)optval = e->flags._OOBINLINE;
else *(Word *)optval = 1;
return 0;
break;
// unsupported ish.
case SO_SNDBUF:
case SO_RCVBUF:
// input/output buffer size
case SO_RCVTIMEO:
case SO_SNDTIMEO:
// read/send timeout.
case SO_ERROR:
// return any pending error and clear the error status.
default:
return ENOPROTOOPT;
}
return ENOPROTOOPT;
}
int do_shutdown(
Word ipid,
void *p1, void *p2,
void *p3, void *p4,
void *p5)
{
ipidHash *e = findIpidHash(ipid);
int how = *(int *)p1;
// mark the read/write op disabled
// TODO -- main thread should read & ignore any
// incoming data if SHUT_RD?
if (!e) return EINVAL;
switch(how)
{
case 0:
e->flags._SHUT_RD = 1;
break;
case 1:
e->flags._SHUT_WR = 1;
break;
case 2:
e->flags._SHUT_RD = 1;
e->flags._SHUT_WR = 1;
break;
default:
return EINVAL;
}
return 0;
}
#pragma databank 1
static int driver(
int socknum, int req,
void *m, size_t *m_len,
struct sockaddr *addr, int *addrlen,
void *rights)
{
int rv;
{
static char buffer[64];
sprintf(buffer, "driver: req = %d", req);
debug_str(buffer);
}
switch (req)
{
case PRU_ABORT:
break;
case PRU_ACCEPT:
break;
case PRU_ATTACH:
// KERNsocket(int domain, int type, int protocol, int *ERRNO);
return do_attach(socknum, m, m_len, addr, addrlen, rights);
break;
case PRU_BIND:
// KERNbind(int fd, struct sockaddr *my_addr, int addrlen, int *ERRNO)
return do_bind(socknum, m, m_len, addr, addrlen, rights);
break;
case PRU_CONNECT:
// KERNconnect(int fd, struct sockaddr *serv_addr,
// int addrlen, int *ERRNO)
return do_connect(socknum, m, m_len, addr, addrlen, rights);
break;
case PRU_CONNECT2:
break;
case PRU_CONTROL:
break;
case PRU_DETACH:
// called from GS/OS
// int SOCKclose(int sock) part 2
return do_detach(socknum, m, m_len, addr, addrlen, rights);
break;
case PRU_DISCONNECT:
// called from GS/OS
// int SOCKclose(int sock) part 1
return do_disconnect(socknum, m, m_len, addr, addrlen, rights);
break;
case PRU_LISTEN:
break;
case PRU_PEERADDR:
break;
case PRU_RCVD:
// called from GS/OS.
// may block, so be nice.
// SOCKrdwr(struct rwPBlock *pb, word cmd, int sock)
asm {
// dec busy.
jsl 0xE10068
}
rv = do_rcvd(socknum, m, m_len, addr, addrlen, rights);
asm {
// inc busy.
jsl 0xE10064
}
return rv;
break;
case PRU_RCVOOB:
break;
case PRU_SEND:
// SOCKrdwr(struct rwPBlock *pb, word cmd, int sock)
return do_send(socknum, m, m_len, addr, addrlen, rights);
break;
case PRU_SENDOOB:
break;
case PRU_SENSE:
break;
case PRU_SHUTDOWN:
break;
case PRU_SOCKADDR:
break;
case PRU_CO_GETOPT:
break;
case PRU_CO_SETOPT:
break;
case PRU_SELECT:
// int SOCKselect(int pid, int fl, int sock)
return do_select(socknum, m, m_len, addr, addrlen, rights);
break;
}
return EINVAL;
}
void signal_handler(int sig, int code)
{
WriteLine("\pWe got signal!");
QuitFlag = 1;
}
void DisplayMessage(const char *string)
{
if (string) WriteLine(string);
}
#pragma databank 0
// startup/shutdown flags.
enum {
kLoaded = 1,
kStarted = 2,
kConnected = 4
};
Word StartUp(displayPtr fx)
{
word status;
word flags = 0;
// TCPIP is an init, not a tool, so it should always
// be loaded.
status = TCPIPStatus();
if (_toolErr)
{
LoadOneTool(54, 0x0300);
if (_toolErr) return -1;
status = 0;
flags |= kLoaded;
}
#if 1
// require 3.0b3
if (TCPIPLongVersion() < 0x03006003)
{
if (fx) fx("Marinetti 3.0b3 is required.");
if (flags & kLoaded)
UnloadOneTool(54);
return -1;
}
#endif
if (!status)
{
TCPIPStartUp();
if (_toolErr) return -1;
flags |= kStarted;
}
status = TCPIPGetConnectStatus();
if (!status)
{
TCPIPConnect(fx);
flags |= kConnected;
}
return flags;
}
void ShutDown(word flags, Boolean force, displayPtr fx)
{
if (flags & kConnected)
{
TCPIPDisconnect(force, fx);
if (_toolErr) return;
}
if (flags & kStarted)
{
TCPIPShutDown();
if (_toolErr) return;
}
if (flags & kLoaded)
{
UnloadOneTool(54);
}
}
int main(int argc, char **argv)
{
int flags;
MyID = MMStartUp();
flags = StartUp(DisplayMessage);
if (flags == -1) exit(1);
semID = screate(1);
InstallNetDriver(driver, 0);
QuitFlag = 0;
signal(SIGQUIT, signal_handler);
signal(SIGINT, signal_handler);
signal(SIGHUP, signal_handler);
while(!QuitFlag)
{
ipidHash *e;
ipidHash *prev;
debug_str("TCPIPPoll() begin");
if (dlist)
{
// suspend so we can be fg.
kill(getpid(),SIGSTOP);
asm {
brk 0xea
}
}
TCPIPPoll();
debug_str("TCPIPPoll() end");
// monitor disconnects...
// semaphore ...
swait(semID);
if (errno == EINTR) continue;
e = dlist;
prev = NULL;
while (e)
{
Word terr;
srBuff sr;
ipidHash *next;
next = e->next;
terr = TCPIPStatusTCP(e->ipid, &sr);
dump_srbuff(&sr);
// timed wait or closed ...
if (sr.srState == TCPSCLOSED)
{
debug_str("TCPLogout");
TCPIPLogout(e->ipid);
free(e);
if (prev)
{
prev->next = next;
}
else
{
dlist = next;
}
}
else
{
prev = e;
}
e = next;
}
ssignal(semID);
// end semaphore.
sleep(1); // need a sleep10 to match alarm10
}
InstallNetDriver(NULL, 0);
sdelete(semID);
ShutDown(flags, 0, DisplayMessage);
return 0;
}