bb02cc | 1997-09-15 | Fredrik Hübinette (Hubbe) | |
|
a7c144 | 1999-10-30 | Mirar (Pontus Hagland) | |
|
bb02cc | 1997-09-15 | Fredrik Hübinette (Hubbe) | | constant NOERROR=0;
constant FORMERR=1;
constant SERVFAIL=2;
constant NXDOMAIN=3;
constant NOTIMPL=4;
constant NXRRSET=8;
constant QUERY=0;
constant C_IN=1;
constant C_ANY=255;
constant T_A=1;
constant T_NS=2;
constant T_MD=3;
constant T_MF=4;
constant T_CNAME=5;
constant T_SOA=6;
constant T_MB=7;
constant T_PTR=12;
constant T_HINFO=13;
constant T_MINFO=14;
constant T_MX=15;
constant T_TXT=16;
constant T_AAAA=28;
class protocol
{
string mklabel(string s)
{
if(strlen(s)>63)
|
dafeda | 1999-08-10 | Martin Stjernholm | | throw(({"Too long component in domain name\n",backtrace()}));
|
bb02cc | 1997-09-15 | Fredrik Hübinette (Hubbe) | | return sprintf("%c%s",strlen(s),s);
}
|
3d3ed7 | 1997-09-15 | Fredrik Hübinette (Hubbe) | | string low_mkquery(int id,
string dname,
int cl,
int type)
|
bb02cc | 1997-09-15 | Fredrik Hübinette (Hubbe) | | {
|
383392 | 1999-12-22 | Henrik Grubbström (Grubba) | | if ( dname[-1] == '.') dname = dname[..sizeof(dname)-2];
|
bb02cc | 1997-09-15 | Fredrik Hübinette (Hubbe) | | return sprintf("%2c%c%c%2c%2c%2c%2c%s\000%2c%2c",
id,
1,0,
1,
0,
0,
0,
Array.map(dname/".",mklabel)*"",
type,cl);
}
|
3d3ed7 | 1997-09-15 | Fredrik Hübinette (Hubbe) | |
string mkquery(string dname,
int cl,
int type)
{
return low_mkquery(random(65536),dname,cl,type);
}
|
4755b7 | 2000-02-18 | Henrik Grubbström (Grubba) | | string decode_domain(string msg, array(int) n)
|
bb02cc | 1997-09-15 | Fredrik Hübinette (Hubbe) | | {
|
4755b7 | 2000-02-18 | Henrik Grubbström (Grubba) | | array(string) domains=({});
|
bb02cc | 1997-09-15 | Fredrik Hübinette (Hubbe) | |
int pos=n[0];
int next=-1;
|
4755b7 | 2000-02-18 | Henrik Grubbström (Grubba) | | array(string) ret=({});
|
b95656 | 1998-05-20 | Henrik Grubbström (Grubba) | | while(pos < sizeof(msg))
|
bb02cc | 1997-09-15 | Fredrik Hübinette (Hubbe) | | {
switch(int len=msg[pos])
{
case 0:
if(next==-1) next=pos+1;
n[0]=next;
return ret*".";
case 1..63:
pos+=len+1;
ret+=({msg[pos-len..pos-1]});
continue;
default:
if(next==-1) next=pos+2;
pos=((len&63)<<8) + msg[pos+1];
continue;
}
break;
}
}
|
4755b7 | 2000-02-18 | Henrik Grubbström (Grubba) | | string decode_string(string s, array(int) next)
|
bb02cc | 1997-09-15 | Fredrik Hübinette (Hubbe) | | {
int len=s[next[0]];
next[0]+=len+1;
return s[next[0]-len..next[0]-1];
}
|
4755b7 | 2000-02-18 | Henrik Grubbström (Grubba) | | int decode_short(string s, array(int) next)
|
bb02cc | 1997-09-15 | Fredrik Hübinette (Hubbe) | | {
sscanf(s[next[0]..next[0]+1],"%2c",int ret);
next[0]+=2;
return ret;
}
|
4755b7 | 2000-02-18 | Henrik Grubbström (Grubba) | | int decode_int(string s, array(int) next)
|
bb02cc | 1997-09-15 | Fredrik Hübinette (Hubbe) | | {
sscanf(s[next[0]..next[0]+1],"%2c",int ret);
next[0]+=2;
return ret;
}
|
4755b7 | 2000-02-18 | Henrik Grubbström (Grubba) | | array decode_entries(string s,int num, array(int) next)
|
bb02cc | 1997-09-15 | Fredrik Hübinette (Hubbe) | | {
|
4755b7 | 2000-02-18 | Henrik Grubbström (Grubba) | | array(string) ret=({});
|
a87b6b | 1998-04-29 | Fredrik Hübinette (Hubbe) | | for(int e=0;e<num && next[0]<strlen(s);e++)
|
bb02cc | 1997-09-15 | Fredrik Hübinette (Hubbe) | | {
mapping m=([]);
m->name=decode_domain(s,next);
sscanf(s[next[0]..next[0]+10],
"%2c%2c%4c%2c",
m->type,m->cl,m->ttl,m->len);
next[0]+=10;
int tmp=next[0];
switch(m->type)
{
case T_CNAME:
m->cname=decode_domain(s,next);
break;
case T_PTR:
m->ptr=decode_domain(s,next);
break;
case T_NS:
m->ns=decode_domain(s,next);
break;
case T_MX:
m->preference=decode_short(s,next);
m->mx=decode_domain(s,next);
break;
case T_HINFO:
m->cpu=decode_string(s,next);
m->os=decode_string(s,next);
break;
case T_A:
case T_AAAA:
m->a=sprintf("%{.%d%}",values(s[next[0]..next[0]+m->len-1]))[1..];
break;
case T_SOA:
m->mname=decode_domain(s,next);
m->rname=decode_domain(s,next);
m->serial=decode_int(s,next);
m->refresh=decode_int(s,next);
m->retry=decode_int(s,next);
m->expire=decode_int(s,next);
m->minimum=decode_int(s,next);
break;
}
next[0]=tmp+m->len;
ret+=({m});
}
return ret;
}
mapping decode_res(string s)
{
mapping m=([]);
sscanf(s,"%2c%c%c%2c%2c%2c%2c",
m->id,
m->c1,
m->c2,
m->qdcount,
m->ancount,
m->nscount,
m->arcount);
m->rd=(m->c1>>7)&1;
m->tc=(m->c1>>6)&1;
|
22f177 | 1998-01-16 | Fredrik Hübinette (Hubbe) | | m->aa=(m->c1>>5)&1;
|
bb02cc | 1997-09-15 | Fredrik Hübinette (Hubbe) | | m->opcode=(m->c1>>1)&15;
m->qr=m->c1&1;
m->rcode=(m->c2>>4)&15;
m->cd=(m->c2>>3)&1;
m->ad=(m->c2>>2)&1;
m->ra=(m->c2)&1;
m->length=strlen(s);
|
4755b7 | 2000-02-18 | Henrik Grubbström (Grubba) | | array(string) tmp=({});
|
bb02cc | 1997-09-15 | Fredrik Hübinette (Hubbe) | | int e;
if(m->qdcount!=1)
return m;
|
4755b7 | 2000-02-18 | Henrik Grubbström (Grubba) | | array(int) next=({12});
|
bb02cc | 1997-09-15 | Fredrik Hübinette (Hubbe) | | m->qd=decode_domain(s,next);
sscanf(s[next[0]..next[0]+3],"%2c%2c",m->type, m->cl);
next[0]+=4;
m->an=decode_entries(s,m->ancount,next);
m->ns=decode_entries(s,m->nscount,next);
m->ar=decode_entries(s,m->arcount,next);
return m;
}
};
|
eabcd5 | 1999-07-25 | Henrik Grubbström (Grubba) | | #define RETRIES 12
#define RETRY_DELAY 5
|
a7c144 | 1999-10-30 | Mirar (Pontus Hagland) | | class client
{
|
bb02cc | 1997-09-15 | Fredrik Hübinette (Hubbe) | | inherit protocol;
|
667644 | 1998-01-25 | Henrik Grubbström (Grubba) | | static private int is_ip(string ip)
{
|
b9b6ec | 1998-01-25 | Henrik Grubbström (Grubba) | | return(replace(ip,
|
667644 | 1998-01-25 | Henrik Grubbström (Grubba) | | ({ "0", "1", "2", "3", "4", "5", "6", "7", "8", "9", "." }),
|
b9b6ec | 1998-01-25 | Henrik Grubbström (Grubba) | | ({ "", "", "", "", "", "", "", "", "", "", "" })) == "");
|
667644 | 1998-01-25 | Henrik Grubbström (Grubba) | | }
static private mapping etc_hosts;
|
37ffea | 1998-04-06 | Fredrik Hübinette (Hubbe) | | #ifdef __NT__
string get_tcpip_param(string val)
|
a7c144 | 1999-10-30 | Mirar (Pontus Hagland) | | {
foreach(({
"SYSTEM\\CurrentControlSet\\Services\\Tcpip\\Parameters",
"SYSTEM\\CurrentControlSet\\Services\\VxD\\MSTCP"
}),string key)
|
37ffea | 1998-04-06 | Fredrik Hübinette (Hubbe) | | {
|
a7c144 | 1999-10-30 | Mirar (Pontus Hagland) | | catch {
return RegGetValue(HKEY_LOCAL_MACHINE, key, val);
};
|
37ffea | 1998-04-06 | Fredrik Hübinette (Hubbe) | | }
|
a7c144 | 1999-10-30 | Mirar (Pontus Hagland) | | }
|
37ffea | 1998-04-06 | Fredrik Hübinette (Hubbe) | | #endif
|
667644 | 1998-01-25 | Henrik Grubbström (Grubba) | | static private string match_etc_hosts(string host)
|
a7c144 | 1999-10-30 | Mirar (Pontus Hagland) | | {
if (!etc_hosts) {
string raw;
|
9f6f4d | 1998-02-27 | Fredrik Hübinette (Hubbe) | | #ifdef __NT__
|
a7c144 | 1999-10-30 | Mirar (Pontus Hagland) | | raw=get_tcpip_param("DataBasePath")+"\\hosts";
|
9f6f4d | 1998-02-27 | Fredrik Hübinette (Hubbe) | | #else
|
a7c144 | 1999-10-30 | Mirar (Pontus Hagland) | | raw="/etc/hosts";
|
9f6f4d | 1998-02-27 | Fredrik Hübinette (Hubbe) | | #endif
|
a7c144 | 1999-10-30 | Mirar (Pontus Hagland) | | raw = Stdio.read_file(raw);
|
37ffea | 1998-04-06 | Fredrik Hübinette (Hubbe) | |
|
a7c144 | 1999-10-30 | Mirar (Pontus Hagland) | | etc_hosts = ([ "localhost":"127.0.0.1" ]);
|
37ffea | 1998-04-06 | Fredrik Hübinette (Hubbe) | |
|
a7c144 | 1999-10-30 | Mirar (Pontus Hagland) | | if (raw && sizeof(raw)) {
foreach(raw/"\n"-({""}), string line) {
line = lower_case(replace((line/"#")[0], "\t", " "));
array arr = (line/" ") - ({ "" });
|
37ffea | 1998-04-06 | Fredrik Hübinette (Hubbe) | |
|
a7c144 | 1999-10-30 | Mirar (Pontus Hagland) | | if (sizeof(arr) > 1) {
if (is_ip(arr[0])) {
foreach(arr[1..], string name) {
etc_hosts[name] = arr[0];
}
} else {
|
667644 | 1998-01-25 | Henrik Grubbström (Grubba) | | }
}
}
|
a7c144 | 1999-10-30 | Mirar (Pontus Hagland) | | } else {
|
667644 | 1998-01-25 | Henrik Grubbström (Grubba) | | }
}
|
a7c144 | 1999-10-30 | Mirar (Pontus Hagland) | | return(etc_hosts[lower_case(host)]);
}
|
667644 | 1998-01-25 | Henrik Grubbström (Grubba) | |
|
625627 | 1998-01-11 | Henrik Grubbström (Grubba) | | array(string) nameservers = ({});
|
7f8292 | 1997-12-10 | David Hedbor | | array domains = ({});
|
cd6ed0 | 1998-07-17 | Fredrik Hübinette (Hubbe) | | void create(void|string|array(string) server, void|int|array(string) domain)
|
bb02cc | 1997-09-15 | Fredrik Hübinette (Hubbe) | | {
if(!server)
{
|
27c685 | 1997-12-10 | David Hedbor | | string domain;
|
37ffea | 1998-04-06 | Fredrik Hübinette (Hubbe) | |
|
d35779 | 1999-04-14 | Henrik Grubbström (Grubba) | | #if __NT__
|
37ffea | 1998-04-06 | Fredrik Hübinette (Hubbe) | | domain=get_tcpip_param("Domain");
|
be6a97 | 1999-08-06 | Johan Schön | | nameservers = get_tcpip_param("NameServer") / " ";
|
37ffea | 1998-04-06 | Fredrik Hübinette (Hubbe) | | domains=get_tcpip_param("SearchList") / " "- ({""});
|
9f6f4d | 1998-02-27 | Fredrik Hübinette (Hubbe) | | #else
|
9f7040 | 1998-11-13 | Marcus Comstedt | | string resolv_conf;
foreach(({"/etc/resolv.conf", "/amitcp/db/resolv.conf"}), string resolv_loc)
if ((resolv_conf = Stdio.read_file(resolv_loc)))
break;
|
625627 | 1998-01-11 | Henrik Grubbström (Grubba) | |
if (!resolv_conf) {
|
d35779 | 1999-04-14 | Henrik Grubbström (Grubba) | |
|
625627 | 1998-01-11 | Henrik Grubbström (Grubba) | | throw(({ "Protocols.DNS.client(): No /etc/resolv.conf!\n",
backtrace() }));
}
foreach(resolv_conf/"\n", string line)
|
27c685 | 1997-12-10 | David Hedbor | | {
string rest;
sscanf(line,"%s#",line);
sscanf(line,"%*[\r \t]%s",line);
line=reverse(line);
sscanf(line,"%*[\r \t]%s",line);
line=reverse(line);
sscanf(line,"%s%*[ \t]%s",line,rest);
switch(line)
|
bb02cc | 1997-09-15 | Fredrik Hübinette (Hubbe) | | {
|
a7c144 | 1999-10-30 | Mirar (Pontus Hagland) | | case "domain":
domain = sizeof(rest) && rest;
break;
case "search":
rest = replace(rest, "\t", " ");
domains += ((rest/" ") - ({""}));
break;
|
bb02cc | 1997-09-15 | Fredrik Hübinette (Hubbe) | |
|
a7c144 | 1999-10-30 | Mirar (Pontus Hagland) | | case "nameserver":
if (!is_ip(rest)) {
string host = rest;
if (!(rest = match_etc_hosts(host))) {
werror(sprintf("Protocols.DNS.client(): "
"Can't resolv nameserver \"%s\"\n", host));
break;
}
|
667644 | 1998-01-25 | Henrik Grubbström (Grubba) | | }
|
a7c144 | 1999-10-30 | Mirar (Pontus Hagland) | | if (sizeof(rest)) {
nameservers += ({ rest });
}
break;
|
bb02cc | 1997-09-15 | Fredrik Hübinette (Hubbe) | | }
|
27c685 | 1997-12-10 | David Hedbor | | }
|
9f6f4d | 1998-02-27 | Fredrik Hübinette (Hubbe) | | #endif
|
625627 | 1998-01-11 | Henrik Grubbström (Grubba) | | if (!sizeof(nameservers)) {
|
d35779 | 1999-04-14 | Henrik Grubbström (Grubba) | |
|
625627 | 1998-01-11 | Henrik Grubbström (Grubba) | | nameservers = ({ "127.0.0.1" });
}
|
27c685 | 1997-12-10 | David Hedbor | | if(domain)
|
7f8292 | 1997-12-10 | David Hedbor | | domains = ({ domain }) + domains;
|
dc01b5 | 1999-06-01 | Henrik Grubbström (Grubba) | | domains -= ({ "" });
|
b9b6ec | 1998-01-25 | Henrik Grubbström (Grubba) | | domains = Array.map(domains, lambda(string d) {
if (d[-1] == '.') {
return d[..sizeof(d)-2];
}
return d;
});
|
a7c144 | 1999-10-30 | Mirar (Pontus Hagland) | | }
else
{
|
cd6ed0 | 1998-07-17 | Fredrik Hübinette (Hubbe) | | if(arrayp(server))
|
940cb7 | 1998-07-17 | Henrik Grubbström (Grubba) | | nameservers = server;
|
cd6ed0 | 1998-07-17 | Fredrik Hübinette (Hubbe) | | else
|
940cb7 | 1998-07-17 | Henrik Grubbström (Grubba) | | nameservers = ({ server });
|
cd6ed0 | 1998-07-17 | Fredrik Hübinette (Hubbe) | |
if(arrayp(domain))
domains = domain;
else
|
522de1 | 1998-07-17 | Fredrik Hübinette (Hubbe) | | if(stringp(domain))
|
4f3ea4 | 1998-07-17 | Fredrik Hübinette (Hubbe) | | domains = ({ domain });
|
bb02cc | 1997-09-15 | Fredrik Hübinette (Hubbe) | | }
}
mapping do_sync_query(string s)
{
|
eabcd5 | 1999-07-25 | Henrik Grubbström (Grubba) | | object udp = Stdio.UDP();
|
bb02cc | 1997-09-15 | Fredrik Hübinette (Hubbe) | | udp->bind(0);
mapping m;
|
eabcd5 | 1999-07-25 | Henrik Grubbström (Grubba) | | int i;
for (i=0; i < RETRIES; i++) {
udp->send(nameservers[i % sizeof(nameservers)], 53, s);
while (udp->wait(RETRY_DELAY)) {
|
9df989 | 2000-05-07 | Henrik Grubbström (Grubba) | |
catch {
m = udp->read();
if ((m->port == 53) &&
(m->data[0..1] == s[0..1]) &&
(search(nameservers, m->ip) != -1)) {
return decode_res(m->data);
}
};
|
eabcd5 | 1999-07-25 | Henrik Grubbström (Grubba) | | }
}
return 0;
|
bb02cc | 1997-09-15 | Fredrik Hübinette (Hubbe) | | }
|
a7c144 | 1999-10-30 | Mirar (Pontus Hagland) | |
|
bb02cc | 1997-09-15 | Fredrik Hübinette (Hubbe) | |
|
4755b7 | 2000-02-18 | Henrik Grubbström (Grubba) | | array gethostbyname(string s)
|
bb02cc | 1997-09-15 | Fredrik Hübinette (Hubbe) | | {
|
27c685 | 1997-12-10 | David Hedbor | | mapping m;
|
7f8292 | 1997-12-10 | David Hedbor | | if(sizeof(domains) && s[-1] != '.' && sizeof(s/".") < 3) {
|
eabcd5 | 1999-07-25 | Henrik Grubbström (Grubba) | | m = do_sync_query(mkquery(s, C_IN, T_A));
|
27c685 | 1997-12-10 | David Hedbor | | if(!m || !m->an || !sizeof(m->an))
|
7f8292 | 1997-12-10 | David Hedbor | | foreach(domains, string domain)
|
27c685 | 1997-12-10 | David Hedbor | | {
|
eabcd5 | 1999-07-25 | Henrik Grubbström (Grubba) | | m = do_sync_query(mkquery(s+"."+domain, C_IN, T_A));
|
27c685 | 1997-12-10 | David Hedbor | | if(m && m->an && sizeof(m->an))
break;
}
} else {
|
eabcd5 | 1999-07-25 | Henrik Grubbström (Grubba) | | m = do_sync_query(mkquery(s, C_IN, T_A));
|
27c685 | 1997-12-10 | David Hedbor | | }
|
eabcd5 | 1999-07-25 | Henrik Grubbström (Grubba) | |
if (!m) {
return ({ 0, ({}), ({}) });
}
|
4755b7 | 2000-02-18 | Henrik Grubbström (Grubba) | | array(string) names=({});
array(string) ips=({});
|
bb02cc | 1997-09-15 | Fredrik Hübinette (Hubbe) | | foreach(m->an, mapping x)
|
27c685 | 1997-12-10 | David Hedbor | | {
if(x->name)
names+=({x->name});
if(x->a)
ips+=({x->a});
}
|
bb02cc | 1997-09-15 | Fredrik Hübinette (Hubbe) | | return ({
sizeof(names)?names[0]:0,
|
eabcd5 | 1999-07-25 | Henrik Grubbström (Grubba) | | ips,
names,
});
|
bb02cc | 1997-09-15 | Fredrik Hübinette (Hubbe) | | }
|
f971bd | 1997-09-15 | Fredrik Hübinette (Hubbe) | |
string arpa_from_ip(string ip)
{
return reverse(ip/".")*"."+".IN-ADDR.ARPA";
}
string ip_from_arpa(string arpa)
{
return reverse(arpa/".")[2..]*".";
}
|
a7c144 | 1999-10-30 | Mirar (Pontus Hagland) | |
|
4755b7 | 2000-02-18 | Henrik Grubbström (Grubba) | | array gethostbyaddr(string s)
|
bb02cc | 1997-09-15 | Fredrik Hübinette (Hubbe) | | {
|
f971bd | 1997-09-15 | Fredrik Hübinette (Hubbe) | | mapping m=do_sync_query(mkquery(arpa_from_ip(s), C_IN, T_PTR));
|
eabcd5 | 1999-07-25 | Henrik Grubbström (Grubba) | | if (m) {
|
4755b7 | 2000-02-18 | Henrik Grubbström (Grubba) | | array(string) names=({});
array(string) ips=({});
|
eabcd5 | 1999-07-25 | Henrik Grubbström (Grubba) | |
foreach(m->an, mapping x)
|
27c685 | 1997-12-10 | David Hedbor | | {
|
eabcd5 | 1999-07-25 | Henrik Grubbström (Grubba) | | if(x->ptr)
names+=({x->ptr});
if(x->name)
{
ips+=({ip_from_arpa(x->name)});
}
|
bb02cc | 1997-09-15 | Fredrik Hübinette (Hubbe) | | }
|
eabcd5 | 1999-07-25 | Henrik Grubbström (Grubba) | | return ({
sizeof(names)?names[0]:0,
|
bb02cc | 1997-09-15 | Fredrik Hübinette (Hubbe) | | ips,
names,
|
eabcd5 | 1999-07-25 | Henrik Grubbström (Grubba) | | });
} else {
return({ 0, ({}), ({}) });
}
|
bb02cc | 1997-09-15 | Fredrik Hübinette (Hubbe) | | }
|
a7c144 | 1999-10-30 | Mirar (Pontus Hagland) | |
|
bb02cc | 1997-09-15 | Fredrik Hübinette (Hubbe) | | string get_primary_mx(string host)
{
|
27c685 | 1997-12-10 | David Hedbor | | mapping m;
|
7f8292 | 1997-12-10 | David Hedbor | | if(sizeof(domains) && host[-1] != '.' && sizeof(host/".") < 3) {
|
27c685 | 1997-12-10 | David Hedbor | | m=do_sync_query(mkquery(host, C_IN, T_MX));
if(!m || !m->an || !sizeof(m->an))
|
7f8292 | 1997-12-10 | David Hedbor | | foreach(domains, string domain)
|
27c685 | 1997-12-10 | David Hedbor | | {
m=do_sync_query(mkquery(host+"."+domain, C_IN, T_MX));
if(m && m->an && sizeof(m->an))
break;
}
} else {
m=do_sync_query(mkquery(host, C_IN, T_MX));
}
|
eabcd5 | 1999-07-25 | Henrik Grubbström (Grubba) | | if (!m) {
return 0;
}
|
bb02cc | 1997-09-15 | Fredrik Hübinette (Hubbe) | | int minpref=29372974;
string ret;
foreach(m->an, mapping m2)
|
27c685 | 1997-12-10 | David Hedbor | | {
if(m2->preference<minpref)
|
bb02cc | 1997-09-15 | Fredrik Hübinette (Hubbe) | | {
|
27c685 | 1997-12-10 | David Hedbor | | ret=m2->mx;
minpref=m2->preference;
|
bb02cc | 1997-09-15 | Fredrik Hübinette (Hubbe) | | }
|
27c685 | 1997-12-10 | David Hedbor | | }
|
bb02cc | 1997-09-15 | Fredrik Hübinette (Hubbe) | | return ret;
}
|
383392 | 1999-12-22 | Henrik Grubbström (Grubba) | |
|
4755b7 | 2000-02-18 | Henrik Grubbström (Grubba) | | array(string) get_mx(string host)
|
383392 | 1999-12-22 | Henrik Grubbström (Grubba) | | {
mapping m;
if(sizeof(domains) && host[-1] != '.' && sizeof(host/".") < 3) {
m = do_sync_query(mkquery(host, C_IN, T_MX));
if(!m || !m->an || !sizeof(m->an)) {
foreach(domains, string domain)
{
m = do_sync_query(mkquery(host+"."+domain, C_IN, T_MX));
if(m && m->an && sizeof(m->an))
break;
}
}
} else {
m = do_sync_query(mkquery(host, C_IN, T_MX));
}
if (!m) {
return 0;
}
array a = m->an;
array(string) b = column( a, "mx");
sort( column( a, "preference"), b);
return b;
}
|
bb02cc | 1997-09-15 | Fredrik Hübinette (Hubbe) | | }
|
cd6ed0 | 1998-07-17 | Fredrik Hübinette (Hubbe) | | #define REMOVE_DELAY 120
#define GIVE_UP_DELAY (RETRIES * RETRY_DELAY + REMOVE_DELAY)*2
|
bb02cc | 1997-09-15 | Fredrik Hübinette (Hubbe) | |
class async_client
{
inherit client;
|
854091 | 1999-07-21 | David Hedbor | | inherit Stdio.UDP : udp;
|
cd6ed0 | 1998-07-17 | Fredrik Hübinette (Hubbe) | | async_client next_client;
|
bb02cc | 1997-09-15 | Fredrik Hübinette (Hubbe) | |
|
3d3ed7 | 1997-09-15 | Fredrik Hübinette (Hubbe) | | class Request
|
bb02cc | 1997-09-15 | Fredrik Hübinette (Hubbe) | | {
|
3d3ed7 | 1997-09-15 | Fredrik Hübinette (Hubbe) | | string req;
|
f971bd | 1997-09-15 | Fredrik Hübinette (Hubbe) | | string domain;
function callback;
|
3d3ed7 | 1997-09-15 | Fredrik Hübinette (Hubbe) | | int retries;
|
cd6ed0 | 1998-07-17 | Fredrik Hübinette (Hubbe) | | int timestamp;
|
4755b7 | 2000-02-18 | Henrik Grubbström (Grubba) | | array args;
|
bb02cc | 1997-09-15 | Fredrik Hübinette (Hubbe) | | };
|
3d3ed7 | 1997-09-15 | Fredrik Hübinette (Hubbe) | |
|
f971bd | 1997-09-15 | Fredrik Hübinette (Hubbe) | | mapping requests=([]);
|
3d3ed7 | 1997-09-15 | Fredrik Hübinette (Hubbe) | |
static private void remove(object(Request) r)
{
if(!r) return;
sscanf(r->req,"%2c",int id);
m_delete(requests,id);
|
f971bd | 1997-09-15 | Fredrik Hübinette (Hubbe) | | r->callback(r->domain,0,@r->args);
|
3d3ed7 | 1997-09-15 | Fredrik Hübinette (Hubbe) | | destruct(r);
}
|
625627 | 1998-01-11 | Henrik Grubbström (Grubba) | | void retry(object(Request) r, void|int nsno)
|
3d3ed7 | 1997-09-15 | Fredrik Hübinette (Hubbe) | | {
if(!r) return;
|
625627 | 1998-01-11 | Henrik Grubbström (Grubba) | | if (nsno >= sizeof(nameservers)) {
|
cd6ed0 | 1998-07-17 | Fredrik Hübinette (Hubbe) | | if(r->retries++ > RETRIES)
|
625627 | 1998-01-11 | Henrik Grubbström (Grubba) | | {
|
cd6ed0 | 1998-07-17 | Fredrik Hübinette (Hubbe) | | call_out(remove,REMOVE_DELAY,r);
|
625627 | 1998-01-11 | Henrik Grubbström (Grubba) | | return;
} else {
nsno = 0;
}
|
3d3ed7 | 1997-09-15 | Fredrik Hübinette (Hubbe) | | }
|
625627 | 1998-01-11 | Henrik Grubbström (Grubba) | |
send(nameservers[nsno],53,r->req);
|
cd6ed0 | 1998-07-17 | Fredrik Hübinette (Hubbe) | | call_out(retry,RETRY_DELAY,r,nsno+1);
|
3d3ed7 | 1997-09-15 | Fredrik Hübinette (Hubbe) | | }
void do_query(string domain, int cl, int type,
|
f971bd | 1997-09-15 | Fredrik Hübinette (Hubbe) | | function(string,mapping,mixed...:void) callback,
|
3d3ed7 | 1997-09-15 | Fredrik Hübinette (Hubbe) | | mixed ... args)
{
|
75c66d | 1998-07-17 | Fredrik Hübinette (Hubbe) | | for(int e=next_client ? 5 : 256;e>=0;e--)
|
cd6ed0 | 1998-07-17 | Fredrik Hübinette (Hubbe) | | {
|
75c66d | 1998-07-17 | Fredrik Hübinette (Hubbe) | | int lid = random(65536);
if(!catch { requests[lid]++; })
|
cd6ed0 | 1998-07-17 | Fredrik Hübinette (Hubbe) | | {
|
75c66d | 1998-07-17 | Fredrik Hübinette (Hubbe) | | string req=low_mkquery(lid,domain,cl,type);
object r=Request();
r->req=req;
r->domain=domain;
r->callback=callback;
r->args=args;
r->timestamp=time();
requests[lid]=r;
udp::send(nameservers[0],53,r->req);
call_out(retry,RETRY_DELAY,r,1);
return;
|
cd6ed0 | 1998-07-17 | Fredrik Hübinette (Hubbe) | | }
}
|
75c66d | 1998-07-17 | Fredrik Hübinette (Hubbe) | |
if(!next_client)
next_client=async_client(nameservers,domains);
next_client->do_query(domain, cl, type, callback, @args);
|
3d3ed7 | 1997-09-15 | Fredrik Hübinette (Hubbe) | | }
|
529b30 | 1999-07-25 | David Hedbor | | static private void rec_data(mapping m)
|
3d3ed7 | 1997-09-15 | Fredrik Hübinette (Hubbe) | | {
|
6aa4b7 | 1998-06-13 | Henrik Grubbström (Grubba) | | mixed err;
if (err = catch {
if(m->port != 53 || search(nameservers, m->ip) == -1) return;
sscanf(m->data,"%2c",int id);
object r=requests[id];
if(!r) return;
m_delete(requests,id);
r->callback(r->domain,decode_res(m->data),@r->args);
destruct(r);
}) {
werror(sprintf("DNS: Failed to read UDP packet. Connection refused?\n"
"%s\n",
describe_backtrace(err)));
}
|
3d3ed7 | 1997-09-15 | Fredrik Hübinette (Hubbe) | | }
|
f971bd | 1997-09-15 | Fredrik Hübinette (Hubbe) | |
static private void generic_get(string d,
mapping answer,
|
27c685 | 1997-12-10 | David Hedbor | | int multi,
|
901dde | 1998-09-13 | Henrik Grubbström (Grubba) | | int all,
int type,
|
f971bd | 1997-09-15 | Fredrik Hübinette (Hubbe) | | string field,
string domain,
function callback,
mixed ... args)
{
if(!answer || !answer->an || !sizeof(answer->an))
{
|
7f8292 | 1997-12-10 | David Hedbor | | if(multi == -1 || multi >= sizeof(domains)) {
|
27c685 | 1997-12-10 | David Hedbor | |
callback(domain,0,@args);
} else {
|
901dde | 1998-09-13 | Henrik Grubbström (Grubba) | | do_query(domain+"."+domains[multi], C_IN, type,
generic_get, ++multi, all, type, field, domain,
callback, @args);
|
27c685 | 1997-12-10 | David Hedbor | | }
} else {
|
901dde | 1998-09-13 | Henrik Grubbström (Grubba) | | if (all) {
callback(domain, answer->an, @args);
} else {
foreach(answer->an, array an)
if(an[field])
{
callback(domain, an[field], @args);
return;
}
callback(domain,0,@args);
return;
}
|
f971bd | 1997-09-15 | Fredrik Hübinette (Hubbe) | | }
}
void host_to_ip(string host, function callback, mixed ... args)
{
|
7f8292 | 1997-12-10 | David Hedbor | | if(sizeof(domains) && host[-1] != '.' && sizeof(host/".") < 3) {
|
27c685 | 1997-12-10 | David Hedbor | | do_query(host, C_IN, T_A,
|
901dde | 1998-09-13 | Henrik Grubbström (Grubba) | | generic_get, 0, 0, T_A, "a", host, callback, @args );
|
27c685 | 1997-12-10 | David Hedbor | | } else {
do_query(host, C_IN, T_A,
|
c9d48d | 1998-09-13 | Henrik Grubbström (Grubba) | | generic_get, -1, 0, T_A, "a",
|
27c685 | 1997-12-10 | David Hedbor | | host, callback, @args);
}
|
f971bd | 1997-09-15 | Fredrik Hübinette (Hubbe) | | }
void ip_to_host(string ip, function callback, mixed ... args)
|
3d3ed7 | 1997-09-15 | Fredrik Hübinette (Hubbe) | | {
|
f971bd | 1997-09-15 | Fredrik Hübinette (Hubbe) | | do_query(arpa_from_ip(ip), C_IN, T_PTR,
|
901dde | 1998-09-13 | Henrik Grubbström (Grubba) | | generic_get, -1, 0, T_PTR, "ptr",
|
f971bd | 1997-09-15 | Fredrik Hübinette (Hubbe) | | ip, callback,
@args);
}
|
901dde | 1998-09-13 | Henrik Grubbström (Grubba) | | void get_mx_all(string host, function callback, mixed ... args)
{
mapping m;
if(sizeof(domains) && host[-1] != '.' && sizeof(host/".") < 3) {
do_query(host, C_IN, T_MX,
generic_get, 0, 1, T_MX, "mx", host, callback, @args);
} else {
do_query(host, C_IN, T_MX,
generic_get, -1, 1, T_MX, "mx", host, callback, @args);
}
}
void get_mx(string host, function callback, mixed ... args)
{
get_mx_all(host,
lambda(string domain, array(mapping) mx,
function callback, mixed ... args) {
array a;
if (mx) {
a = column(mx, "mx");
sort(column(mx, "preference"), a);
}
callback(a, @args);
|
79ee22 | 1998-09-18 | Henrik Grubbström (Grubba) | | }, callback, @args);
|
901dde | 1998-09-13 | Henrik Grubbström (Grubba) | | }
|
cd6ed0 | 1998-07-17 | Fredrik Hübinette (Hubbe) | | void create(void|string|array(string) server, void|string|array(string) domain)
|
f971bd | 1997-09-15 | Fredrik Hübinette (Hubbe) | | {
if(!udp::bind(0))
throw(({"DNS: failed to bind a port.\n",backtrace()}));
udp::set_read_callback(rec_data);
|
cd6ed0 | 1998-07-17 | Fredrik Hübinette (Hubbe) | | ::create(server,domain);
|
3d3ed7 | 1997-09-15 | Fredrik Hübinette (Hubbe) | | }
|
bb02cc | 1997-09-15 | Fredrik Hübinette (Hubbe) | | };
|