ref: 4ba285fb7dc477013ce1fca68adf59244fc42bd4
dir: /sys/src/9/ip/arp.c/
#include "u.h"
#include "../port/lib.h"
#include "mem.h"
#include "dat.h"
#include "fns.h"
#include "../port/error.h"
#include "ip.h"
#include "ipv6.h"
/*
* address resolution tables
*/
enum
{
NHASH = (1<<6),
NCACHE = 256,
AOK = 1,
AWAIT = 2,
MAXAGE_TIMER = 15*60*1000,
};
char *arpstate[] =
{
"UNUSED",
"OK",
"WAIT",
};
/*
* one per Fs
*/
struct Arp
{
RWlock;
Fs *f;
Arpent *hash[NHASH];
Arpent cache[NCACHE];
Arpent *rxmt[2];
Proc *rxmitp; /* neib sol re-transmit proc */
Rendez rxmtq;
Block *dropf, *dropl;
};
char *Ebadarp = "bad arp";
/* quick hash for ip addresses */
#define hashipa(a) (((a)[IPaddrlen-2] + (a)[IPaddrlen-1])%NHASH)
static void rxmitproc(void*);
void
arpinit(Fs *f)
{
f->arp = smalloc(sizeof(Arp));
f->arp->f = f;
f->arp->rxmt[0] = nil;
f->arp->rxmt[1] = nil;
f->arp->dropf = f->arp->dropl = nil;
kproc("rxmitproc", rxmitproc, f->arp);
}
static void
freeblistchain(Block *bp)
{
Block *next;
while(bp != nil){
next = bp->list;
freeblist(bp);
bp = next;
}
}
/* take out of re-transmit chain */
static Arpent**
rxmtunchain(Arp *arp, Arpent *a)
{
Arpent **l;
for(l = &arp->rxmt[isv4(a->ip) != 0]; *l != nil; l = &((*l)->nextrxt)){
if(*l == a){
*l = a->nextrxt;
break;
}
}
a->nextrxt = nil;
return l;
}
static void
cleanarpent(Arp *arp, Arpent *a)
{
Arpent **l;
Block *bp;
/* take out of current chain */
for(l = &arp->hash[hashipa(a->ip)]; *l != nil; l = &((*l)->hash)){
if(*l == a){
*l = a->hash;
break;
}
}
a->hash = nil;
/* remove from retransmit / timout chain */
rxmtunchain(arp, a);
/* queue packets for icmp unreachable for rxmitproc later on, w/o arp lock */
if((bp = a->hold) != nil){
if(arp->dropf == nil)
arp->dropf = bp;
else
arp->dropl->list = bp;
arp->dropl = a->last;
if(bp == arp->dropf)
wakeup(&arp->rxmtq);
}
a->hold = nil;
a->last = nil;
a->ifc = nil;
a->ifcid = 0;
a->state = 0;
a->rxtsrem = 0;
a->utime = 0;
a->ctime = 0;
}
/*
* create a new arp entry for an ip address on ifc.
*/
static Arpent*
newarpent(Arp *arp, uchar *ip, Ipifc *ifc)
{
Arpent *a, *e, *f, **l;
ulong t;
/* find oldest entry */
e = &arp->cache[NCACHE];
a = arp->cache;
t = a->utime;
for(f = a+1; t > 0 && f < e; f++){
if(f->utime < t){
t = f->utime;
a = f;
}
}
cleanarpent(arp, a);
ipmove(a->ip, ip);
a->ifc = ifc;
a->ifcid = ifc->ifcid;
/* insert into new chain */
l = &arp->hash[hashipa(ip)];
a->hash = *l;
*l = a;
return a;
}
static Arpent*
arplookup(Arp *arp, Ipifc *ifc, uchar *ip)
{
Arpent *a;
for(a = arp->hash[hashipa(ip)]; a != nil; a = a->hash){
if(a->ifc == ifc && a->ifcid == ifc->ifcid && ipcmp(ip, a->ip) == 0)
return a;
}
return nil;
}
/*
* fill in the media address if we have it. Otherwise return an
* Arpent that represents the state of the address resolution FSM
* for ip. Add the packet to be sent onto the list of packets
* waiting for ip->mac to be resolved.
*/
Arpent*
arpget(Arp *arp, Block *bp, int version, Ipifc *ifc, uchar *ip, uchar *mac, Routehint *rh)
{
uchar v6ip[IPaddrlen];
Arpent *a;
if(version == V4){
v4tov6(v6ip, ip);
ip = v6ip;
}
if(rh != nil
&& (a = rh->a) != nil
&& a->state == AOK
&& a->ifc == ifc
&& a->ifcid == ifc->ifcid
&& ipcmp(ip, a->ip) == 0){
memmove(mac, a->mac, ifc->m->maclen);
a->utime = NOW;
if(a->utime - a->ctime < MAXAGE_TIMER)
return nil;
}
rlock(arp);
if((a = arplookup(arp, ifc, ip)) != nil && a->state == AOK){
memmove(mac, a->mac, ifc->m->maclen);
a->utime = NOW;
if(a->utime - a->ctime < MAXAGE_TIMER){
if(rh != nil)
rh->a = a;
runlock(arp);
return nil;
}
}
if(rh != nil)
rh->a = nil;
runlock(arp);
wlock(arp);
if((a = arplookup(arp, ifc, ip)) == nil)
a = newarpent(arp, ip, ifc);
a->state = AWAIT;
a->utime = NOW;
if(bp != nil){
/* needs to be a single block */
assert(bp->list == nil);
if(a->hold == nil)
a->hold = bp;
else
a->last->list = bp;
a->last = bp;
}
return a; /* return with arp locked */
}
/*
* continue address resolution for the entry,
* schedule it on the retransmit / timeout chains
* and unlock the arp cache.
*/
void
arpcontinue(Arp *arp, Arpent *a)
{
Arpent **l;
Block *bp;
/* remove all but the last message */
while((bp = a->hold) != nil){
if(bp == a->last)
break;
a->hold = bp->list;
freeblist(bp);
}
/* try to keep it around for a second more */
a->ctime = a->utime = NOW;
/* put on end of re-transmit / timeout chain */
for(l = rxmtunchain(arp, a); *l != nil; l = &(*l)->nextrxt)
;
*l = a;
if(l == &arp->rxmt[0] || l == &arp->rxmt[1])
wakeup(&arp->rxmtq);
wunlock(arp);
}
/*
* called with arp locked
*/
void
arprelease(Arp *arp, Arpent*)
{
wunlock(arp);
}
/*
* Copy out the mac address from the Arpent. Return the
* block waiting to get sent to this mac address.
*
* called with arp locked
*/
Block*
arpresolve(Arp *arp, Arpent *a, uchar *mac, Routehint *rh)
{
Block *bp;
memmove(a->mac, mac, a->ifc->m->maclen);
if(a->state == AWAIT) {
rxmtunchain(arp, a);
a->rxtsrem = 0;
}
bp = a->hold;
a->hold = a->last = nil;
a->ctime = a->utime = NOW;
a->state = AOK;
if(rh != nil)
rh->a = a;
wunlock(arp);
return bp;
}
int
arpenter(Fs *fs, int version, uchar *ip, uchar *mac, int n, uchar *ia, Ipifc *ifc, int refresh)
{
Routehint rh;
uchar v6ip[IPaddrlen];
Block *bp, *next;
Arpent *a;
Route *r;
Arp *arp;
if(ifc->m == nil || ifc->m->maclen != n || ifc->m->maclen == 0)
return -1;
rh.r = nil;
rh.a = nil;
switch(version){
case V4:
r = v4lookup(fs, ip, ia, &rh);
v4tov6(v6ip, ip);
ip = v6ip;
break;
case V6:
r = v6lookup(fs, ip, ia, &rh);
break;
default:
panic("arpenter: version %d", version);
return -1; /* to supress warnings */
}
if(r == nil || r->ifc != ifc || (r->type & (Rbcast|Rmulti)) != 0)
return -1;
arp = fs->arp;
wlock(arp);
if((a = arplookup(arp, ifc, ip)) == nil){
if(refresh){
wunlock(arp);
return 0;
}
a = newarpent(arp, ip, ifc);
}
bp = arpresolve(arp, a, mac, &rh); /* unlocks arp */
if(version == V4)
ip += IPv4off;
for(; bp != nil; bp = next){
next = bp->list;
bp->list = nil;
if(waserror()){
freeblistchain(next);
break;
}
ipifcoput(ifc, bp, version, ip, &rh);
poperror();
}
return 1;
}
int
arpwrite(Fs *fs, char *s, int len)
{
int n;
Arp *arp;
Arpent *a, *x;
Medium *m;
Ipifc *ifc;
char *f[5], buf[256];
uchar ip[IPaddrlen], ia[IPaddrlen], mac[MAClen];
arp = fs->arp;
if(len == 0)
error(Ebadarp);
if(len >= sizeof(buf))
len = sizeof(buf)-1;
strncpy(buf, s, len);
buf[len] = 0;
if(len > 0 && buf[len-1] == '\n')
buf[len-1] = 0;
n = getfields(buf, f, nelem(f), 1, " ");
if(strcmp(f[0], "flush") == 0){
wlock(arp);
for(a = arp->cache; a < &arp->cache[NCACHE]; a++){
a->hash = nil;
a->nextrxt = nil;
a->ifc = nil;
a->ifcid = 0;
a->state = 0;
a->rxtsrem = 0;
a->ctime = 0;
a->utime = 0;
freeblistchain(a->hold);
a->hold = a->last = nil;
}
memset(arp->hash, 0, sizeof(arp->hash));
freeblistchain(arp->dropf);
arp->dropf = arp->dropl = nil;
arp->rxmt[0] = arp->rxmt[1] = nil;
wunlock(arp);
} else if(strcmp(f[0], "add") == 0){
switch(n){
default:
error(Ebadarg);
case 3:
if(parseip(ip, f[1]) == -1)
error(Ebadip);
if((n = parsemac(mac, f[2], sizeof(mac))) <= 0)
error(Ebadarp);
findlocalip(fs, ia, ip);
break;
case 4:
m = ipfindmedium(f[1]);
if(m == nil || m->maclen == 0)
error(Ebadarp);
if(parseip(ip, f[2]) == -1)
error(Ebadip);
if((n = parsemac(mac, f[3], sizeof(mac))) != m->maclen)
error(Ebadarp);
findlocalip(fs, ia, ip);
break;
case 5:
m = ipfindmedium(f[1]);
if(m == nil || m->maclen == 0)
error(Ebadarp);
if(parseip(ip, f[2]) == -1)
error(Ebadip);
if((n = parsemac(mac, f[3], sizeof(mac))) != m->maclen)
error(Ebadarp);
if(parseip(ia, f[4]) == -1)
error(Ebadip);
break;
}
if((ifc = findipifc(fs, ia, ia, Runi)) == nil)
error("no interface");
rlock(ifc);
if(!ipv6local(ifc, ia, 0, ip) || arpenter(fs, V6, ip, mac, n, ia, ifc, 0) < 0){
runlock(ifc);
error("destination unreachable");
}
runlock(ifc);
} else if(strcmp(f[0], "del") == 0){
if (n != 2)
error(Ebadarg);
if (parseip(ip, f[1]) == -1)
error(Ebadip);
wlock(arp);
for(a = arp->hash[hashipa(ip)]; a != nil; a = x){
x = a->hash;
if(ipcmp(ip, a->ip) == 0)
cleanarpent(arp, a);
}
wunlock(arp);
} else
error(Ebadarp);
return len;
}
static void
convmac(char *p, uchar *mac, int n)
{
while(n-- > 0)
p += sprint(p, "%2.2ux", *mac++);
}
int
arpread(Arp *arp, char *s, ulong offset, int len)
{
char mac[2*MAClen+1], *state, *mname, *p;
uchar ip[IPaddrlen], ia[IPaddrlen];
Ipifc *ifc;
Arpent *a;
long n, o;
p = s;
o = -offset;
for(a = arp->cache; len > 0 && a < &arp->cache[NCACHE]; a++){
if(a->state == 0 || (ifc = a->ifc) == nil)
continue;
rlock(ifc);
rlock(arp);
ipmove(ip, a->ip);
if(ifc->m == nil || a->state == 0 || a->ifc != ifc || a->ifcid != ifc->ifcid || !ipv6local(ifc, ia, 0, ip)){
runlock(arp);
runlock(ifc);
continue;
}
mname = ifc->m->name;
state = arpstate[a->state];
convmac(mac, a->mac, ifc->m->maclen);
runlock(arp);
runlock(ifc);
n = snprint(up->genbuf, sizeof up->genbuf,
"%-6.6s %-4.4s %-40.40I %-16.16s %I\n",
mname, state, ip, mac, ia);
o += n;
if(o <= 0)
continue;
if(n > len)
break;
memmove(p, up->genbuf, n);
len -= n;
p += n;
}
return p - s;
}
void
ndpsendsol(Fs *f, Arpent *a)
{
Ipifc *ifc = a->ifc;
uchar targ[IPaddrlen], src[IPaddrlen];
if(a->rxtsrem == 0)
a->rxtsrem = MAX_MULTICAST_SOLICIT;
else
a->rxtsrem--;
/* try to use source address of original packet */
ipmove(targ, a->ip);
if(a->last != nil){
ipmove(src, ((Ip6hdr*)a->last->rp)->src);
arpcontinue(f->arp, a);
if(iplocalonifc(ifc, src) != nil || ipproxyifc(f, ifc, src))
goto send;
} else {
arpcontinue(f->arp, a);
}
if(!ipv6local(ifc, src, 0, targ))
return;
send:
if(!waserror()){
icmpns(f, src, SRC_UNI, targ, TARG_MULTI, ifc->mac);
poperror();
}
}
static Block*
rxmt(Arp *arp)
{
Arpent *a;
Block *bp;
Ipifc *ifc;
wlock(arp);
/* retransmit ipv6 solicitations */
while((a = arp->rxmt[0]) != nil && NOW - a->ctime > 3*RETRANS_TIMER/4){
if(a->rxtsrem > 0 && (ifc = a->ifc) != nil && canrlock(ifc)){
if(a->ifcid == ifc->ifcid){
ndpsendsol(arp->f, a); /* unlocks arp */
runlock(ifc);
wlock(arp);
continue;
}
runlock(ifc);
}
cleanarpent(arp, a);
}
/* timeout waiting ipv4 arp entries */
while((a = arp->rxmt[1]) != nil && NOW - a->ctime > 3*RETRANS_TIMER)
cleanarpent(arp, a);
bp = arp->dropf;
arp->dropf = arp->dropl = nil;
wunlock(arp);
return bp;
}
static void
drop(Fs *f, Block *bp)
{
Block *next;
Ipifc *ifc;
Route *r;
for(; bp != nil; bp = next){
next = bp->list;
bp->list = nil;
if((bp->rp[0]&0xF0) == IP_VER4)
r = v4lookup(f, ((Ip4hdr*)bp->rp)->src, ((Ip4hdr*)bp->rp)->dst, nil);
else
r = v6lookup(f, ((Ip6hdr*)bp->rp)->src, ((Ip6hdr*)bp->rp)->dst, nil);
if(r != nil && (ifc = r->ifc) != nil && canrlock(ifc)){
if(!waserror()){
if((bp->rp[0]&0xF0) == IP_VER4)
icmpnohost(f, ifc, bp);
else
icmphostunr6(f, ifc, bp, Icmp6_adr_unreach, (r->type & Runi) != 0);
poperror();
}
runlock(ifc);
}
freeblist(bp);
}
}
static int
rxready(void *v)
{
Arp *arp = (Arp *)v;
return arp->rxmt[0] != nil || arp->rxmt[1] != nil || arp->dropf != nil;
}
static void
rxmitproc(void *v)
{
Arp *arp = v;
arp->rxmitp = up;
if(waserror()){
arp->rxmitp = nil;
pexit("hangup", 1);
}
for(;;){
sleep(&arp->rxmtq, rxready, v);
drop(arp->f, rxmt(arp));
tsleep(&arp->rxmtq, return0, nil, RETRANS_TIMER/4);
}
}