459 lines
9.9 KiB
C
459 lines
9.9 KiB
C
/***********************************************************
|
|
Copyright 1991, 1992, 1993, 1994 by Stichting Mathematisch Centrum,
|
|
Amsterdam, The Netherlands.
|
|
|
|
All Rights Reserved
|
|
|
|
Permission to use, copy, modify, and distribute this software and its
|
|
documentation for any purpose and without fee is hereby granted,
|
|
provided that the above copyright notice appear in all copies and that
|
|
both that copyright notice and this permission notice appear in
|
|
supporting documentation, and that the names of Stichting Mathematisch
|
|
Centrum or CWI not be used in advertising or publicity pertaining to
|
|
distribution of the software without specific, written prior permission.
|
|
|
|
STICHTING MATHEMATISCH CENTRUM DISCLAIMS ALL WARRANTIES WITH REGARD TO
|
|
THIS SOFTWARE, INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY AND
|
|
FITNESS, IN NO EVENT SHALL STICHTING MATHEMATISCH CENTRUM BE LIABLE
|
|
FOR ANY SPECIAL, INDIRECT OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
|
|
WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
|
|
ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT
|
|
OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
|
|
|
|
******************************************************************/
|
|
|
|
#include "allobjects.h"
|
|
#include "modsupport.h" /* For getargs() etc. */
|
|
|
|
#include <AddressXlation.h>
|
|
#include <Desk.h>
|
|
|
|
#ifndef __MWERKS__
|
|
#define ResultUPP ResultProcPtr
|
|
#define NewResultProc(x) (x)
|
|
/* The '2' has move in this name... */
|
|
#define Result2UPP ResultProc2Ptr
|
|
#define NewResult2Proc(x) (x)
|
|
#endif
|
|
|
|
static object *ErrorObject;
|
|
|
|
/* ----------------------------------------------------- */
|
|
/* Declarations for objects of type MacTCP DNR Result */
|
|
|
|
/* Types of records we have */
|
|
#define DNR_ADDR 0
|
|
#define DNR_HINFO 1
|
|
#define DNR_MX 2
|
|
|
|
typedef struct {
|
|
OB_HEAD
|
|
int type; /* DNR_XXX */
|
|
int waiting; /* True while completion proc not called */
|
|
struct returnRec hinfo;
|
|
} dnrrobject;
|
|
|
|
staticforward typeobject Dnrrtype;
|
|
|
|
#define is_dnrrobject(v) ((v)->ob_type == &Dnrrtype)
|
|
|
|
/* ---------------------------------------------------------------- */
|
|
|
|
static pascal void
|
|
dnrr_done(rrp, udp)
|
|
struct hostInfo *rrp; /* Unused */
|
|
dnrrobject *udp;
|
|
{
|
|
if ( !udp->waiting ) {
|
|
printf("macdnr: dnrr_done: spurious completion call!\n");
|
|
return;
|
|
}
|
|
udp->waiting = 0;
|
|
DECREF(udp);
|
|
}
|
|
|
|
static int dnrwait(self)
|
|
dnrrobject *self;
|
|
{
|
|
while ( self->waiting ) {
|
|
if ( !PyMac_Idle() )
|
|
return 0;
|
|
}
|
|
return 1;
|
|
}
|
|
|
|
static object *
|
|
dnrr_wait(self, args)
|
|
dnrrobject *self;
|
|
object *args;
|
|
{
|
|
if (!newgetargs(args, ""))
|
|
return NULL;
|
|
if ( !dnrwait(self) ) {
|
|
INCREF(None);
|
|
return None;
|
|
}
|
|
if ( self->hinfo.rtnCode ) {
|
|
PyErr_Mac(ErrorObject, self->hinfo.rtnCode);
|
|
return NULL;
|
|
}
|
|
INCREF(None);
|
|
return None;
|
|
}
|
|
|
|
static object *
|
|
dnrr_isdone(self, args)
|
|
dnrrobject *self;
|
|
object *args;
|
|
{
|
|
if (!newgetargs(args, ""))
|
|
return NULL;
|
|
return newintobject(!self->waiting);
|
|
}
|
|
|
|
static struct methodlist dnrr_methods[] = {
|
|
{"wait", (method)dnrr_wait, 1},
|
|
{"isdone", (method)dnrr_isdone, 1},
|
|
|
|
{NULL, NULL} /* sentinel */
|
|
};
|
|
|
|
/* ---------- */
|
|
|
|
static dnrrobject *
|
|
newdnrrobject(tp)
|
|
int tp;
|
|
{
|
|
dnrrobject *self;
|
|
|
|
self = NEWOBJ(dnrrobject, &Dnrrtype);
|
|
if (self == NULL)
|
|
return NULL;
|
|
self->type = tp;
|
|
self->waiting = 0;
|
|
memset(&self->hinfo, 0, sizeof(self->hinfo));
|
|
return self;
|
|
}
|
|
|
|
static void
|
|
dnrr_dealloc(self)
|
|
dnrrobject *self;
|
|
{
|
|
self->waiting = 0; /* Not really needed, since we incref for completion */
|
|
DEL(self);
|
|
}
|
|
|
|
/* Code to access structure members by accessing attributes */
|
|
|
|
#include "structmember.h"
|
|
|
|
#define OFF(x) offsetof(struct returnRec, x)
|
|
|
|
static struct memberlist dnrr_memberlist_addr[] = {
|
|
{ "rtnCode", T_INT, OFF(rtnCode), RO},
|
|
{ "cname", T_STRING_INPLACE, OFF(cname), RO},
|
|
{ "ip0", T_UINT, OFF(rdata.addr[0]), RO},
|
|
{ "ip1", T_UINT, OFF(rdata.addr[1]), RO},
|
|
{ "ip2", T_UINT, OFF(rdata.addr[2]), RO},
|
|
{ "ip3", T_UINT, OFF(rdata.addr[3]), RO},
|
|
{NULL} /* Sentinel */
|
|
};
|
|
|
|
static struct memberlist dnrr_memberlist_hinfo[] = {
|
|
{ "rtnCode", T_INT, OFF(rtnCode), RO},
|
|
{ "cname", T_STRING_INPLACE, OFF(cname), RO},
|
|
{ "cpuType", T_STRING_INPLACE, OFF(rdata.hinfo.cpuType), RO},
|
|
{ "osType", T_STRING_INPLACE, OFF(rdata.hinfo.osType), RO},
|
|
{NULL} /* Sentinel */
|
|
};
|
|
|
|
static struct memberlist dnrr_memberlist_mx[] = {
|
|
{ "rtnCode", T_INT, OFF(rtnCode), RO},
|
|
{ "cname", T_STRING_INPLACE, OFF(cname), RO},
|
|
{ "preference", T_USHORT, OFF(rdata.mx.preference), RO},
|
|
{ "exchange", T_STRING_INPLACE, OFF(rdata.mx.exchange), RO},
|
|
{NULL} /* Sentinel */
|
|
};
|
|
|
|
static struct memberlist *dnrr_mlists[3] = {
|
|
dnrr_memberlist_addr,
|
|
dnrr_memberlist_hinfo,
|
|
dnrr_memberlist_mx
|
|
};
|
|
|
|
static object *
|
|
dnrr_getattr(self, name)
|
|
dnrrobject *self;
|
|
char *name;
|
|
{
|
|
object *rv;
|
|
int tp;
|
|
|
|
rv = findmethod(dnrr_methods, (object *)self, name);
|
|
if ( rv ) return rv;
|
|
err_clear();
|
|
if ( self->waiting )
|
|
if ( !dnrwait(self) ) {
|
|
err_setstr(ErrorObject, "Resolver busy");
|
|
return NULL;
|
|
}
|
|
tp = self->type;
|
|
return getmember((char *)&self->hinfo, dnrr_mlists[tp], name);
|
|
}
|
|
|
|
|
|
static typeobject Dnrrtype = {
|
|
OB_HEAD_INIT(&Typetype)
|
|
0, /*ob_size*/
|
|
"MacTCP DNR Result", /*tp_name*/
|
|
sizeof(dnrrobject), /*tp_basicsize*/
|
|
0, /*tp_itemsize*/
|
|
/* methods */
|
|
(destructor)dnrr_dealloc, /*tp_dealloc*/
|
|
(printfunc)0, /*tp_print*/
|
|
(getattrfunc)dnrr_getattr, /*tp_getattr*/
|
|
(setattrfunc)0, /*tp_setattr*/
|
|
(cmpfunc)0, /*tp_compare*/
|
|
(reprfunc)0, /*tp_repr*/
|
|
0, /*tp_as_number*/
|
|
0, /*tp_as_sequence*/
|
|
0, /*tp_as_mapping*/
|
|
(hashfunc)0, /*tp_hash*/
|
|
};
|
|
|
|
/* End of code for MacTCP DNR Result objects */
|
|
/* -------------------------------------------------------- */
|
|
|
|
int dnr_is_open;
|
|
|
|
static int
|
|
opendnr(fn)
|
|
char *fn;
|
|
{
|
|
OSErr err;
|
|
|
|
if ( dnr_is_open ) return 1;
|
|
if ( (err=OpenResolver(fn)) ) {
|
|
PyErr_Mac(ErrorObject, err);
|
|
return 0;
|
|
}
|
|
dnr_is_open = 1;
|
|
return 1;
|
|
}
|
|
|
|
static object *
|
|
dnr_Open(self, args)
|
|
object *self; /* Not used */
|
|
object *args;
|
|
{
|
|
char *fn = NULL;
|
|
|
|
if (!newgetargs(args, "|s", &fn))
|
|
return NULL;
|
|
if ( dnr_is_open ) {
|
|
err_setstr(ErrorObject, "DNR already open");
|
|
return NULL;
|
|
}
|
|
if ( !opendnr(fn) )
|
|
return NULL;
|
|
INCREF(None);
|
|
return None;
|
|
}
|
|
|
|
static object *
|
|
dnr_Close(self, args)
|
|
object *self; /* Not used */
|
|
object *args;
|
|
{
|
|
OSErr err;
|
|
|
|
if (!newgetargs(args, ""))
|
|
return NULL;
|
|
dnr_is_open = 0;
|
|
if ( (err=CloseResolver()) ) {
|
|
PyErr_Mac(ErrorObject, err);
|
|
return NULL;
|
|
}
|
|
INCREF(None);
|
|
return None;
|
|
}
|
|
|
|
static object *
|
|
dnr_StrToAddr(self, args)
|
|
object *self; /* Not used */
|
|
object *args;
|
|
{
|
|
OSErr err;
|
|
char *hostname;
|
|
dnrrobject *rv;
|
|
ResultUPP cb_upp = NewResultProc(dnrr_done);
|
|
|
|
if (!newgetargs(args, "s", &hostname))
|
|
return NULL;
|
|
if ( !opendnr(NULL) )
|
|
return NULL;
|
|
if ( (rv=newdnrrobject(DNR_ADDR)) == NULL )
|
|
return NULL;
|
|
err = StrToAddr(hostname, (struct hostInfo *)&rv->hinfo, cb_upp, (char *)rv);
|
|
if ( err == cacheFault ) {
|
|
rv->waiting++;
|
|
INCREF(rv);
|
|
} else if ( err ) {
|
|
DECREF(rv);
|
|
PyErr_Mac(ErrorObject, err);
|
|
return NULL;
|
|
}
|
|
return (object *)rv;
|
|
}
|
|
|
|
static object *
|
|
dnr_AddrToName(self, args)
|
|
object *self; /* Not used */
|
|
object *args;
|
|
{
|
|
OSErr err;
|
|
unsigned long ipaddr;
|
|
dnrrobject *rv;
|
|
ResultUPP cb_upp = NewResultProc(dnrr_done);
|
|
|
|
if (!newgetargs(args, "l", &ipaddr))
|
|
return NULL;
|
|
if ( !opendnr(NULL) )
|
|
return NULL;
|
|
if ( (rv=newdnrrobject(DNR_ADDR)) == NULL )
|
|
return NULL;
|
|
err = AddrToName(ipaddr, (struct hostInfo *)&rv->hinfo, cb_upp, (char *)rv);
|
|
if ( err == cacheFault ) {
|
|
rv->waiting++;
|
|
INCREF(rv);
|
|
} else if ( err ) {
|
|
DECREF(rv);
|
|
PyErr_Mac(ErrorObject, err);
|
|
return NULL;
|
|
}
|
|
return (object *)rv;
|
|
}
|
|
|
|
static object *
|
|
dnr_AddrToStr(self, args)
|
|
object *self; /* Not used */
|
|
object *args;
|
|
{
|
|
OSErr err;
|
|
unsigned long ipaddr;
|
|
char ipname[16];
|
|
object *rv;
|
|
|
|
if (!newgetargs(args, "l", &ipaddr))
|
|
return NULL;
|
|
if ( !opendnr(NULL) )
|
|
return NULL;
|
|
if ( (err=AddrToStr(ipaddr, ipname)) ) {
|
|
PyErr_Mac(ErrorObject, err);
|
|
return NULL;
|
|
}
|
|
return newstringobject(ipname);
|
|
}
|
|
|
|
static object *
|
|
dnr_HInfo(self, args)
|
|
object *self; /* Not used */
|
|
object *args;
|
|
{
|
|
OSErr err;
|
|
char *hostname;
|
|
dnrrobject *rv;
|
|
Result2UPP cb_upp = NewResult2Proc(dnrr_done);
|
|
|
|
if (!newgetargs(args, "s", &hostname))
|
|
return NULL;
|
|
if ( !opendnr(NULL) )
|
|
return NULL;
|
|
if ( (rv=newdnrrobject(DNR_HINFO)) == NULL )
|
|
return NULL;
|
|
err = HInfo(hostname, &rv->hinfo, cb_upp, (char *)rv);
|
|
if ( err == cacheFault ) {
|
|
rv->waiting++;
|
|
INCREF(rv);
|
|
} else if ( err ) {
|
|
DECREF(rv);
|
|
PyErr_Mac(ErrorObject, err);
|
|
return NULL;
|
|
}
|
|
return (object *)rv;
|
|
|
|
if (!newgetargs(args, ""))
|
|
return NULL;
|
|
INCREF(None);
|
|
return None;
|
|
}
|
|
|
|
static object *
|
|
dnr_MXInfo(self, args)
|
|
object *self; /* Not used */
|
|
object *args;
|
|
{
|
|
OSErr err;
|
|
char *hostname;
|
|
dnrrobject *rv;
|
|
Result2UPP cb_upp = NewResult2Proc(dnrr_done);
|
|
|
|
if (!newgetargs(args, "s", &hostname))
|
|
return NULL;
|
|
if ( !opendnr(NULL) )
|
|
return NULL;
|
|
if ( (rv=newdnrrobject(DNR_MX)) == NULL )
|
|
return NULL;
|
|
err = MXInfo(hostname, &rv->hinfo, cb_upp, (char *)rv);
|
|
if ( err == cacheFault ) {
|
|
rv->waiting++;
|
|
INCREF(rv);
|
|
} else if ( err ) {
|
|
DECREF(rv);
|
|
PyErr_Mac(ErrorObject, err);
|
|
return NULL;
|
|
}
|
|
return (object *)rv;
|
|
}
|
|
|
|
/* List of methods defined in the module */
|
|
|
|
static struct methodlist dnr_methods[] = {
|
|
{"Open", dnr_Open, 1},
|
|
{"Close", dnr_Close, 1},
|
|
{"StrToAddr", dnr_StrToAddr, 1},
|
|
{"AddrToStr", dnr_AddrToStr, 1},
|
|
{"AddrToName", dnr_AddrToName, 1},
|
|
{"HInfo", dnr_HInfo, 1},
|
|
{"MXInfo", dnr_MXInfo, 1},
|
|
|
|
{NULL, NULL} /* sentinel */
|
|
};
|
|
|
|
|
|
/* Initialization function for the module (*must* be called initmacdnr) */
|
|
|
|
void
|
|
initmacdnr()
|
|
{
|
|
object *m, *d, *o;
|
|
|
|
/* Create the module and add the functions */
|
|
m = initmodule("macdnr", dnr_methods);
|
|
|
|
/* Add some symbolic constants to the module */
|
|
d = getmoduledict(m);
|
|
ErrorObject = newstringobject("macdnr.error");
|
|
dictinsert(d, "error", ErrorObject);
|
|
#if 0
|
|
/* Not needed, after all */
|
|
#define CONST(name, value) o = newintobject(value); dictinsert(d, name, o);
|
|
CONST("ADDR", DNR_ADDR);
|
|
CONST("HINFO", DNR_HINFO);
|
|
CONST("MX", DNR_MX);
|
|
#endif
|
|
/* Check for errors */
|
|
if (err_occurred())
|
|
fatal("can't initialize module macdnr");
|
|
}
|