/*************************************************************************
 *
 * HTCd - Copyright (C) 1998-2006 Henrik Rydberg
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
 */

#include <mt/lookup.h>
#include <mt/mvec.h>
#include <mt/lock.h>
#include <arpa/inet.h>
#include <netinet/in.h>
#include <sys/socket.h>

const time_t VALIDSEC=120;
static msvec<IPnode> cache;
static mutex_t mutex;

int Lookup(IPnode& node,const in_addr& ia)
{
  MLOCK(mutex);
  IPnode* p=cache.begin();
  time_t now=time(0);
  while(p<cache.end()) {
    if(now>p->valid) cache.erase(p);
    else if(memcmp(&p->addr,&ia,sizeof(ia))==0) {p->valid=now+VALIDSEC;node=*p;return 1;}
    else p++;
  }
  hostent hent;
  sref_t buf;
  int herr;
  hostent* hp=0;
  gethostbyaddr_r((const char*)&ia,sizeof(ia),AF_INET,
                  &hent,
                  buf,sizeof(sref_t),
                  &hp,
                  &herr);
  if(!hp) return 0;
  memcpy(&node.addr,hp->h_addr,sizeof(hp->h_length));
  node.ip=inet_ntoa(node.addr);
  node.name=hp->h_name;
  node.valid=now+VALIDSEC;
  cache.push_back(node);
  return 1;
}

int Lookup(IPnode& node,const sref& s)
{
  MLOCK(mutex);
  IPnode* p=cache.begin();
  time_t now=time(0);
  while(p<cache.end()) {
    if(now>p->valid) cache.erase(p);
    else if(p->ip==s||p->name==s) { p->valid=now+VALIDSEC; node=*p; return 1; }
    else p++;
  }
  hostent hent;
  sref_t sbuf,buf;
  int herr;
  hostent* hp=0;
  for(int i=0;s[i];i++) if(s[i]!='.'&&(s[i]<'0'||s[i]>'9')) {
    gethostbyname_r(s.copyto(sbuf),&hent,buf,sizeof(sref_t),&hp,&herr);
    break;
  }
  if(!hp) {
    unsigned long a=inet_addr(s.copyto(buf));
    gethostbyaddr_r((char*)&a,sizeof(a),AF_INET,&hent,buf,sizeof(sref_t),&hp,&herr);
  }
  if(!hp) return 0;
  memcpy(&node.addr,hp->h_addr,sizeof(hp->h_length));
  node.ip=inet_ntoa(node.addr);
  node.name=hp->h_name;
  node.valid=now+VALIDSEC;
  cache.push_back(node);
  return 1;
}